2 * Copyright (C) 2011 glevand (geoffrey.levand@mail.ru)
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
30 #include <sys/param.h>
31 #include <sys/systm.h>
32 #include <sys/sysctl.h>
37 #include <sys/kernel.h>
38 #include <sys/kthread.h>
40 #include <sys/malloc.h>
41 #include <sys/module.h>
42 #include <sys/mutex.h>
47 #include <machine/pio.h>
48 #include <machine/bus.h>
49 #include <machine/platform.h>
50 #include <machine/pmap.h>
51 #include <machine/resource.h>
55 #include <geom/geom_disk.h>
58 #include "ps3-hvcall.h"
60 #define PS3DISK_LOCK_INIT(_sc) \
61 mtx_init(&_sc->sc_mtx, device_get_nameunit(_sc->sc_dev), "ps3disk", MTX_DEF)
62 #define PS3DISK_LOCK_DESTROY(_sc) mtx_destroy(&_sc->sc_mtx);
63 #define PS3DISK_LOCK(_sc) mtx_lock(&(_sc)->sc_mtx)
64 #define PS3DISK_UNLOCK(_sc) mtx_unlock(&(_sc)->sc_mtx)
65 #define PS3DISK_ASSERT_LOCKED(_sc) mtx_assert(&_sc->sc_mtx, MA_OWNED);
66 #define PS3DISK_ASSERT_UNLOCKED(_sc) mtx_assert(&_sc->sc_mtx, MA_NOTOWNED);
68 #define LV1_STORAGE_ATA_HDDOUT 0x23
70 static SYSCTL_NODE(_hw, OID_AUTO, ps3disk, CTLFLAG_RD, 0,
71 "PS3 Disk driver parameters");
74 static int ps3disk_debug = 0;
75 SYSCTL_INT(_hw_ps3disk, OID_AUTO, debug, CTLFLAG_RW, &ps3disk_debug,
76 0, "control debugging printfs");
77 TUNABLE_INT("hw.ps3disk.debug", &ps3disk_debug);
79 PS3DISK_DEBUG_INTR = 0x00000001,
80 PS3DISK_DEBUG_TASK = 0x00000002,
81 PS3DISK_DEBUG_READ = 0x00000004,
82 PS3DISK_DEBUG_WRITE = 0x00000008,
83 PS3DISK_DEBUG_FLUSH = 0x00000010,
84 PS3DISK_DEBUG_ANY = 0xffffffff
86 #define DPRINTF(sc, m, fmt, ...) \
88 if (sc->sc_debug & (m)) \
89 printf(fmt, __VA_ARGS__); \
92 #define DPRINTF(sc, m, fmt, ...)
95 struct ps3disk_region {
102 struct ps3disk_softc {
111 struct ps3disk_region *sc_reg;
114 struct resource *sc_irq;
117 struct disk **sc_disk;
119 struct bio_queue_head sc_bioq;
120 struct bio_queue_head sc_deferredq;
121 struct proc *sc_task;
123 bus_dma_tag_t sc_dmatag;
129 static int ps3disk_open(struct disk *dp);
130 static int ps3disk_close(struct disk *dp);
131 static void ps3disk_strategy(struct bio *bp);
133 static void ps3disk_task(void *arg);
134 static void ps3disk_intr(void *arg);
135 static int ps3disk_get_disk_geometry(struct ps3disk_softc *sc);
136 static int ps3disk_enum_regions(struct ps3disk_softc *sc);
137 static void ps3disk_transfer(void *arg, bus_dma_segment_t *segs, int nsegs,
140 static void ps3disk_sysctlattach(struct ps3disk_softc *sc);
142 static MALLOC_DEFINE(M_PS3DISK, "ps3disk", "PS3 Disk");
145 ps3disk_probe(device_t dev)
147 if (ps3bus_get_bustype(dev) != PS3_BUSTYPE_STORAGE ||
148 ps3bus_get_devtype(dev) != PS3_DEVTYPE_DISK)
151 device_set_desc(dev, "Playstation 3 Disk");
153 return (BUS_PROBE_SPECIFIC);
157 ps3disk_attach(device_t dev)
159 struct ps3disk_softc *sc;
166 sc = device_get_softc(dev);
169 PS3DISK_LOCK_INIT(sc);
171 err = ps3disk_get_disk_geometry(sc);
173 device_printf(dev, "Could not get disk geometry\n");
175 goto fail_destroy_lock;
178 device_printf(dev, "block size %lu total blocks %lu\n",
179 sc->sc_blksize, sc->sc_nblocks);
181 err = ps3disk_enum_regions(sc);
183 device_printf(dev, "Could not enumerate disk regions\n");
185 goto fail_destroy_lock;
188 device_printf(dev, "Found %lu regions\n", sc->sc_nregs);
192 goto fail_destroy_lock;
195 /* Setup interrupt handler */
197 sc->sc_irq = bus_alloc_resource_any(dev, SYS_RES_IRQ, &sc->sc_irqid,
200 device_printf(dev, "Could not allocate IRQ\n");
202 goto fail_free_regions;
205 err = bus_setup_intr(dev, sc->sc_irq,
206 INTR_TYPE_BIO | INTR_MPSAFE | INTR_ENTROPY,
207 NULL, ps3disk_intr, sc, &sc->sc_irqctx);
209 device_printf(dev, "Could not setup IRQ\n");
211 goto fail_release_intr;
215 err = bus_dma_tag_create(bus_get_dma_tag(dev), 4096, 0,
216 BUS_SPACE_MAXADDR, BUS_SPACE_MAXADDR, NULL, NULL,
217 BUS_SPACE_UNRESTRICTED, 1, PAGE_SIZE, 0,
218 busdma_lock_mutex, &sc->sc_mtx, &sc->sc_dmatag);
220 device_printf(dev, "Could not create DMA tag\n");
222 goto fail_teardown_intr;
227 sc->sc_disk = malloc(sc->sc_nregs * sizeof(struct disk *),
228 M_PS3DISK, M_ZERO | M_WAITOK);
230 device_printf(dev, "Could not allocate disk(s)\n");
232 goto fail_teardown_intr;
235 for (i = 0; i < sc->sc_nregs; i++) {
236 struct ps3disk_region *rp = &sc->sc_reg[i];
238 d = sc->sc_disk[i] = disk_alloc();
239 d->d_open = ps3disk_open;
240 d->d_close = ps3disk_close;
241 d->d_strategy = ps3disk_strategy;
242 d->d_name = "ps3disk";
244 d->d_maxsize = PAGE_SIZE;
245 d->d_sectorsize = sc->sc_blksize;
247 d->d_mediasize = sc->sc_reg[i].r_size * sc->sc_blksize;
248 d->d_flags |= DISKFLAG_CANFLUSHCACHE;
250 mb = d->d_mediasize >> 20;
257 /* Test to see if we can read this region */
258 err = lv1_storage_read(ps3bus_get_device(dev), d->d_unit,
259 0, 0, rp->r_flags, 0, &junk);
260 device_printf(dev, "region %d %ju%cB%s\n", i, mb, unit,
261 (err == LV1_DENIED_BY_POLICY) ? " (hypervisor protected)"
264 if (err != LV1_DENIED_BY_POLICY)
265 disk_create(d, DISK_VERSION);
269 bioq_init(&sc->sc_bioq);
270 bioq_init(&sc->sc_deferredq);
271 kproc_create(&ps3disk_task, sc, &sc->sc_task, 0, 0, "ps3disk");
273 ps3disk_sysctlattach(sc);
278 bus_teardown_intr(dev, sc->sc_irq, sc->sc_irqctx);
280 bus_release_resource(dev, SYS_RES_IRQ, sc->sc_irqid, sc->sc_irq);
282 free(sc->sc_reg, M_PS3DISK);
284 PS3DISK_LOCK_DESTROY(sc);
289 ps3disk_detach(device_t dev)
291 struct ps3disk_softc *sc = device_get_softc(dev);
294 for (i = 0; i < sc->sc_nregs; i++)
295 disk_destroy(sc->sc_disk[i]);
297 bus_dma_tag_destroy(sc->sc_dmatag);
299 bus_teardown_intr(dev, sc->sc_irq, sc->sc_irqctx);
300 bus_release_resource(dev, SYS_RES_IRQ, sc->sc_irqid, sc->sc_irq);
302 free(sc->sc_disk, M_PS3DISK);
303 free(sc->sc_reg, M_PS3DISK);
305 PS3DISK_LOCK_DESTROY(sc);
311 ps3disk_open(struct disk *dp)
317 ps3disk_close(struct disk *dp)
322 /* Process deferred blocks */
324 ps3disk_task(void *arg)
326 struct ps3disk_softc *sc = (struct ps3disk_softc *) arg;
331 kproc_suspend_check(sc->sc_task);
332 tsleep(&sc->sc_deferredq, PRIBIO, "ps3disk", 10);
335 bp = bioq_takefirst(&sc->sc_deferredq);
341 if (bp->bio_driver1 != NULL) {
342 bus_dmamap_unload(sc->sc_dmatag, (bus_dmamap_t)
344 bus_dmamap_destroy(sc->sc_dmatag, (bus_dmamap_t)
348 ps3disk_strategy(bp);
355 ps3disk_strategy(struct bio *bp)
357 struct ps3disk_softc *sc = (struct ps3disk_softc *)bp->bio_disk->d_drv1;
361 bp->bio_flags |= BIO_ERROR;
362 bp->bio_error = EINVAL;
368 bp->bio_resid = bp->bio_bcount;
369 bioq_insert_tail(&sc->sc_bioq, bp);
371 DPRINTF(sc, PS3DISK_DEBUG_TASK, "%s: bio_cmd 0x%02x\n",
372 __func__, bp->bio_cmd);
375 if (bp->bio_cmd == BIO_FLUSH) {
377 err = lv1_storage_send_device_command(
378 ps3bus_get_device(sc->sc_dev), LV1_STORAGE_ATA_HDDOUT,
379 0, 0, 0, 0, (uint64_t *)&bp->bio_driver2);
382 } else if (bp->bio_cmd == BIO_READ || bp->bio_cmd == BIO_WRITE) {
383 if (bp->bio_bcount % sc->sc_blksize != 0) {
386 bus_dmamap_create(sc->sc_dmatag, BUS_DMA_COHERENT,
387 (bus_dmamap_t *)(&bp->bio_driver1));
388 err = bus_dmamap_load(sc->sc_dmatag,
389 (bus_dmamap_t)(bp->bio_driver1), bp->bio_data,
390 bp->bio_bcount, ps3disk_transfer, bp, 0);
391 if (err == EINPROGRESS)
399 bioq_remove(&sc->sc_bioq, bp);
400 bioq_insert_tail(&sc->sc_deferredq, bp);
401 } else if (err != 0) {
403 bp->bio_flags |= BIO_ERROR;
404 bioq_remove(&sc->sc_bioq, bp);
405 disk_err(bp, "hard error", -1, 1);
413 ps3disk_intr(void *arg)
415 struct ps3disk_softc *sc = (struct ps3disk_softc *) arg;
416 device_t dev = sc->sc_dev;
417 uint64_t devid = ps3bus_get_device(dev);
419 uint64_t tag, status;
421 if (lv1_storage_get_async_status(devid, &tag, &status) != 0)
426 DPRINTF(sc, PS3DISK_DEBUG_INTR, "%s: tag 0x%016lx "
427 "status 0x%016lx\n", __func__, tag, status);
429 /* Locate the matching request */
430 TAILQ_FOREACH(bp, &sc->sc_bioq.queue, bio_queue) {
431 if ((uint64_t)bp->bio_driver2 != tag)
435 device_printf(sc->sc_dev, "%s error (%#lx)\n",
436 (bp->bio_cmd == BIO_READ) ? "Read" : "Write",
439 bp->bio_flags |= BIO_ERROR;
443 bp->bio_flags |= BIO_DONE;
446 if (bp->bio_driver1 != NULL) {
447 if (bp->bio_cmd == BIO_READ)
448 bus_dmamap_sync(sc->sc_dmatag, (bus_dmamap_t)
449 bp->bio_driver1, BUS_DMASYNC_POSTREAD);
450 bus_dmamap_unload(sc->sc_dmatag, (bus_dmamap_t)
452 bus_dmamap_destroy(sc->sc_dmatag, (bus_dmamap_t)
456 bioq_remove(&sc->sc_bioq, bp);
461 if (bioq_first(&sc->sc_deferredq) != NULL)
462 wakeup(&sc->sc_deferredq);
468 ps3disk_get_disk_geometry(struct ps3disk_softc *sc)
470 device_t dev = sc->sc_dev;
471 uint64_t bus_index = ps3bus_get_busidx(dev);
472 uint64_t dev_index = ps3bus_get_devidx(dev);
476 err = lv1_get_repository_node_value(PS3_LPAR_ID_PME,
477 (lv1_repository_string("bus") >> 32) | bus_index,
478 lv1_repository_string("dev") | dev_index,
479 lv1_repository_string("blk_size"), 0, &sc->sc_blksize, &junk);
481 device_printf(dev, "Could not get block size (0x%08x)\n", err);
485 err = lv1_get_repository_node_value(PS3_LPAR_ID_PME,
486 (lv1_repository_string("bus") >> 32) | bus_index,
487 lv1_repository_string("dev") | dev_index,
488 lv1_repository_string("n_blocks"), 0, &sc->sc_nblocks, &junk);
490 device_printf(dev, "Could not get total number of blocks "
499 ps3disk_enum_regions(struct ps3disk_softc *sc)
501 device_t dev = sc->sc_dev;
502 uint64_t bus_index = ps3bus_get_busidx(dev);
503 uint64_t dev_index = ps3bus_get_devidx(dev);
507 /* Read number of regions */
509 err = lv1_get_repository_node_value(PS3_LPAR_ID_PME,
510 (lv1_repository_string("bus") >> 32) | bus_index,
511 lv1_repository_string("dev") | dev_index,
512 lv1_repository_string("n_regs"), 0, &sc->sc_nregs, &junk);
514 device_printf(dev, "Could not get number of regions (0x%08x)\n",
523 sc->sc_reg = malloc(sc->sc_nregs * sizeof(struct ps3disk_region),
524 M_PS3DISK, M_ZERO | M_WAITOK);
532 for (i = 0; i < sc->sc_nregs; i++) {
533 err = lv1_get_repository_node_value(PS3_LPAR_ID_PME,
534 (lv1_repository_string("bus") >> 32) | bus_index,
535 lv1_repository_string("dev") | dev_index,
536 lv1_repository_string("region") | i,
537 lv1_repository_string("id"), &sc->sc_reg[i].r_id, &junk);
539 device_printf(dev, "Could not get region id (0x%08x)\n",
545 err = lv1_get_repository_node_value(PS3_LPAR_ID_PME,
546 (lv1_repository_string("bus") >> 32) | bus_index,
547 lv1_repository_string("dev") | dev_index,
548 lv1_repository_string("region") | i,
549 lv1_repository_string("start"), &sc->sc_reg[i].r_start,
552 device_printf(dev, "Could not get region start "
558 err = lv1_get_repository_node_value(PS3_LPAR_ID_PME,
559 (lv1_repository_string("bus") >> 32) | bus_index,
560 lv1_repository_string("dev") | dev_index,
561 lv1_repository_string("region") | i,
562 lv1_repository_string("size"), &sc->sc_reg[i].r_size,
565 device_printf(dev, "Could not get region size "
572 sc->sc_reg[i].r_flags = 0x2;
574 sc->sc_reg[i].r_flags = 0;
583 free(sc->sc_reg, M_PS3DISK);
589 ps3disk_transfer(void *arg, bus_dma_segment_t *segs, int nsegs, int error)
591 struct bio *bp = (struct bio *)(arg);
592 struct ps3disk_softc *sc = (struct ps3disk_softc *)bp->bio_disk->d_drv1;
593 struct ps3disk_region *rp = &sc->sc_reg[bp->bio_disk->d_unit];
594 uint64_t devid = ps3bus_get_device(sc->sc_dev);
598 /* Locks already held by busdma */
599 PS3DISK_ASSERT_LOCKED(sc);
602 bp->bio_error = error;
603 bp->bio_flags |= BIO_ERROR;
604 bioq_remove(&sc->sc_bioq, bp);
609 block = bp->bio_pblkno;
610 for (i = 0; i < nsegs; i++) {
611 KASSERT((segs[i].ds_len % sc->sc_blksize) == 0,
612 ("DMA fragments not blocksize multiples"));
614 if (bp->bio_cmd == BIO_READ) {
615 err = lv1_storage_read(devid, rp->r_id,
616 block, segs[i].ds_len/sc->sc_blksize,
617 rp->r_flags, segs[i].ds_addr,
618 (uint64_t *)&bp->bio_driver2);
620 bus_dmamap_sync(sc->sc_dmatag,
621 (bus_dmamap_t)bp->bio_driver1,
622 BUS_DMASYNC_PREWRITE);
623 err = lv1_storage_write(devid, rp->r_id,
624 block, segs[i].ds_len/sc->sc_blksize,
625 rp->r_flags, segs[i].ds_addr,
626 (uint64_t *)&bp->bio_driver2);
630 if (err == LV1_BUSY) {
631 bioq_remove(&sc->sc_bioq, bp);
632 bioq_insert_tail(&sc->sc_deferredq, bp);
634 bus_dmamap_unload(sc->sc_dmatag, (bus_dmamap_t)
636 bus_dmamap_destroy(sc->sc_dmatag, (bus_dmamap_t)
638 device_printf(sc->sc_dev, "Could not read "
639 "sectors (0x%08x)\n", err);
640 bp->bio_error = EINVAL;
641 bp->bio_flags |= BIO_ERROR;
642 bioq_remove(&sc->sc_bioq, bp);
649 DPRINTF(sc, PS3DISK_DEBUG_READ, "%s: tag 0x%016lx\n",
650 __func__, sc->sc_bounce_tag);
656 ps3disk_sysctl_debug(SYSCTL_HANDLER_ARGS)
658 struct ps3disk_softc *sc = arg1;
661 debug = sc->sc_debug;
663 error = sysctl_handle_int(oidp, &debug, 0, req);
664 if (error || !req->newptr)
667 sc->sc_debug = debug;
674 ps3disk_sysctlattach(struct ps3disk_softc *sc)
677 struct sysctl_ctx_list *ctx = device_get_sysctl_ctx(sc->sc_dev);
678 struct sysctl_oid *tree = device_get_sysctl_tree(sc->sc_dev);
680 sc->sc_debug = ps3disk_debug;
682 SYSCTL_ADD_PROC(ctx, SYSCTL_CHILDREN(tree), OID_AUTO,
683 "debug", CTLTYPE_INT | CTLFLAG_RW, sc, 0,
684 ps3disk_sysctl_debug, "I", "control debugging printfs");
688 static device_method_t ps3disk_methods[] = {
689 DEVMETHOD(device_probe, ps3disk_probe),
690 DEVMETHOD(device_attach, ps3disk_attach),
691 DEVMETHOD(device_detach, ps3disk_detach),
695 static driver_t ps3disk_driver = {
698 sizeof(struct ps3disk_softc),
701 static devclass_t ps3disk_devclass;
703 DRIVER_MODULE(ps3disk, ps3bus, ps3disk_driver, ps3disk_devclass, 0, 0);