1 /* $NetBSD: ld_thunkbus.c,v 1.30 2012/01/21 22:09:57 reinoud Exp $ */ 2 3 /*- 4 * Copyright (c) 2011 Jared D. McNeill <jmcneill@invisible.ca> 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 19 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 26 * POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #include <sys/cdefs.h> 30 __KERNEL_RCSID(0, "$NetBSD: ld_thunkbus.c,v 1.30 2012/01/21 22:09:57 reinoud Exp $"); 31 32 #include <sys/param.h> 33 #include <sys/proc.h> 34 #include <sys/systm.h> 35 #include <sys/device.h> 36 #include <sys/buf.h> 37 #include <sys/disk.h> 38 #include <sys/kmem.h> 39 40 #include <dev/ldvar.h> 41 42 #include <machine/mainbus.h> 43 #include <machine/thunk.h> 44 #include <machine/intr.h> 45 46 static int ld_thunkbus_match(device_t, cfdata_t, void *); 47 static void ld_thunkbus_attach(device_t, device_t, void *); 48 49 static int ld_thunkbus_ldstart(struct ld_softc *, struct buf *); 50 static int ld_thunkbus_lddump(struct ld_softc *, void *, int, int); 51 static int ld_thunkbus_ldflush(struct ld_softc *, int); 52 53 //#define LD_USE_AIO 54 55 #ifdef LD_USE_AIO 56 static int ld_aio_sig(void *); 57 #endif 58 static void ld_thunkbus_complete(void *arg); 59 60 struct ld_thunkbus_softc; 61 62 struct ld_thunkbus_transfer { 63 struct ld_thunkbus_softc *tt_sc; 64 struct aiocb tt_aio; 65 struct buf *tt_bp; 66 }; 67 68 struct ld_thunkbus_softc { 69 struct ld_softc sc_ld; 70 71 int sc_fd; 72 void *sc_ih; 73 void *sc_aio_ih; 74 75 struct ld_thunkbus_transfer sc_tt; 76 bool busy; 77 }; 78 79 CFATTACH_DECL_NEW(ld_thunkbus, sizeof(struct ld_thunkbus_softc), 80 ld_thunkbus_match, ld_thunkbus_attach, NULL, NULL); 81 82 static int 83 ld_thunkbus_match(device_t parent, cfdata_t match, void *opaque) 84 { 85 struct thunkbus_attach_args *taa = opaque; 86 87 if (taa->taa_type != THUNKBUS_TYPE_DISKIMAGE) 88 return 0; 89 90 return 1; 91 } 92 93 static void 94 ld_thunkbus_attach(device_t parent, device_t self, void *opaque) 95 { 96 struct ld_thunkbus_softc *sc = device_private(self); 97 struct ld_softc *ld = &sc->sc_ld; 98 struct thunkbus_attach_args *taa = opaque; 99 const char *path = taa->u.diskimage.path; 100 ssize_t blksize; 101 off_t size; 102 103 ld->sc_dv = self; 104 105 sc->sc_fd = thunk_open(path, O_RDWR, 0); 106 if (sc->sc_fd == -1) { 107 aprint_error(": couldn't open %s: %d\n", path, thunk_geterrno()); 108 return; 109 } 110 if (thunk_fstat_getsize(sc->sc_fd, &size, &blksize) == -1) { 111 aprint_error(": couldn't stat %s: %d\n", path, thunk_geterrno()); 112 return; 113 } 114 115 aprint_naive("\n"); 116 aprint_normal(": %s (%lld)\n", path, (long long)size); 117 118 ld->sc_flags = LDF_ENABLED; 119 ld->sc_maxxfer = MAXPHYS; 120 ld->sc_secsize = 512; 121 ld->sc_secperunit = size / ld->sc_secsize; 122 ld->sc_maxqueuecnt = 1; 123 ld->sc_start = ld_thunkbus_ldstart; 124 ld->sc_dump = ld_thunkbus_lddump; 125 ld->sc_flush = ld_thunkbus_ldflush; 126 127 sc->sc_ih = softint_establish(SOFTINT_BIO, 128 ld_thunkbus_complete, ld); 129 130 #ifdef LD_USE_AIO 131 sc->sc_aio_ih = sigio_intr_establish(ld_aio_sig, sc); 132 if (sc->sc_aio_ih == NULL) 133 panic("couldn't establish aio sig interrupt"); 134 #endif 135 136 sc->busy = false; 137 138 ldattach(ld); 139 } 140 141 #ifdef LD_USE_AIO 142 static int 143 ld_aio_sig(void *arg) 144 { 145 struct ld_softc *ld = arg; 146 struct ld_thunkbus_softc *sc = (struct ld_thunkbus_softc *)ld; 147 148 softint_schedule(sc->sc_ih); 149 150 return 0; 151 } 152 153 static int 154 ld_thunkbus_ldstart(struct ld_softc *ld, struct buf *bp) 155 { 156 struct ld_thunkbus_softc *sc = (struct ld_thunkbus_softc *)ld; 157 struct ld_thunkbus_transfer *tt = &sc->sc_tt; 158 off_t offset = bp->b_rawblkno * ld->sc_secsize; 159 off_t disksize = ld->sc_secsize * ld->sc_secperunit; 160 int error; 161 162 tt->tt_sc = sc; 163 tt->tt_bp = bp; 164 165 memset(&tt->tt_aio, 0, sizeof(tt->tt_aio)); 166 tt->tt_aio.aio_fildes = sc->sc_fd; 167 tt->tt_aio.aio_buf = bp->b_data; 168 tt->tt_aio.aio_nbytes = bp->b_bcount; 169 tt->tt_aio.aio_offset = offset; 170 171 tt->tt_aio.aio_sigevent.sigev_notify = SIGEV_SIGNAL; 172 tt->tt_aio.aio_sigevent.sigev_signo = SIGIO; 173 tt->tt_aio.aio_sigevent.sigev_value.sival_ptr = tt; 174 #if 0 175 device_printf(sc->sc_ld.sc_dv, "%s addr %p, off=%lld, count=%lld\n", 176 (bp->b_flags & B_READ) ? "rd" : "wr", 177 bp->b_data, 178 (long long)bp->b_rawblkno, 179 (long long)bp->b_bcount); 180 #endif 181 if (sc->busy) 182 panic("%s: reentry", __func__); 183 184 if ((offset < 0) || (offset + bp->b_bcount > disksize)) { 185 error = EIO; 186 bp->b_error = error; 187 bp->b_resid = bp->b_bcount; 188 } else { 189 sc->busy = true; 190 if (bp->b_flags & B_READ) 191 error = thunk_aio_read(&tt->tt_aio); 192 else 193 error = thunk_aio_write(&tt->tt_aio); 194 } 195 return error; 196 } 197 198 static void 199 ld_thunkbus_complete(void *arg) 200 { 201 struct ld_softc *ld = arg; 202 struct ld_thunkbus_softc *sc = (struct ld_thunkbus_softc *)ld; 203 struct ld_thunkbus_transfer *tt = &sc->sc_tt; 204 struct buf *bp = tt->tt_bp; 205 int error; 206 207 /* 208 * check if our aio has finished, we could be called for whatever 209 * reason, for whatever SIGIO since signals can be missed. 210 */ 211 if (!sc->busy) 212 return; 213 214 /* check if it was OUR sigio */ 215 error = thunk_aio_error(&tt->tt_aio); 216 if (error == EINPROGRESS) 217 return; 218 219 /* use the result */ 220 if ((error == 0) && 221 thunk_aio_return(&tt->tt_aio) != -1) { 222 bp->b_resid = 0; 223 } else { 224 bp->b_error = error; 225 bp->b_resid = bp->b_bcount; 226 } 227 228 thunk_printf_debug("\tfin\n"); 229 if (bp->b_error) 230 thunk_printf_debug("error!\n"); 231 232 sc->busy = false; 233 lddone(&sc->sc_ld, bp); 234 } 235 236 237 #else /* LD_USE_AIO */ 238 239 240 static int 241 ld_thunkbus_ldstart(struct ld_softc *ld, struct buf *bp) 242 { 243 struct ld_thunkbus_softc *sc = (struct ld_thunkbus_softc *)ld; 244 struct ld_thunkbus_transfer *tt = &sc->sc_tt; 245 246 tt->tt_sc = sc; 247 tt->tt_bp = bp; 248 249 /* let the softint do the work */ 250 sc->busy = true; 251 softint_schedule(sc->sc_ih); 252 253 return 0; 254 } 255 256 static void 257 ld_thunkbus_complete(void *arg) 258 { 259 struct ld_softc *ld = arg; 260 struct ld_thunkbus_softc *sc = (struct ld_thunkbus_softc *)ld; 261 struct ld_thunkbus_transfer *tt = &sc->sc_tt; 262 struct buf *bp = tt->tt_bp; 263 off_t offset = bp->b_rawblkno * ld->sc_secsize; 264 int64_t ret; 265 266 if (!sc->busy) 267 panic("%s: but not busy?\n", __func__); 268 269 //printf("%s: %s %u @ %lld -> %p (flags 0x%08x)\n", __func__, 270 // bp->b_flags & B_READ ? "read" : "write", 271 // (unsigned int)bp->b_bcount, (long long)offset, bp->b_data, bp->b_flags); 272 273 /* read/write the request */ 274 if (bp->b_flags & B_READ) { 275 ret = thunk_pread(sc->sc_fd, bp->b_data, bp->b_bcount, offset); 276 } else { 277 ret = thunk_pwrite(sc->sc_fd, bp->b_data, bp->b_bcount, offset); 278 } 279 280 //if (ret == -1) 281 // printf("%s: errno = %d\n", __func__, thunk_geterrno()); 282 283 /* setup return params */ 284 if ((ret >= 0) && (ret == bp->b_bcount)) { 285 bp->b_resid = 0; 286 } else { 287 bp->b_error = thunk_geterrno(); 288 bp->b_resid = bp->b_bcount; 289 } 290 thunk_printf_debug("\tfin\n"); 291 if (bp->b_error) 292 thunk_printf_debug("error!\n"); 293 294 sc->busy = false; 295 lddone(&sc->sc_ld, bp); 296 } 297 298 #endif /* LD_USE_AIO */ 299 300 301 static int 302 ld_thunkbus_lddump(struct ld_softc *ld, void *data, int blkno, int blkcnt) 303 { 304 struct ld_thunkbus_softc *sc = (struct ld_thunkbus_softc *)ld; 305 ssize_t len; 306 307 len = thunk_pwrite(sc->sc_fd, data, blkcnt, blkno * ld->sc_secsize); 308 if (len == -1) 309 return thunk_geterrno(); 310 else if (len != blkcnt) { 311 device_printf(ld->sc_dv, "%s failed (short xfer)\n", __func__); 312 return EIO; 313 } 314 315 return 0; 316 } 317 318 static int 319 ld_thunkbus_ldflush(struct ld_softc *ld, int flags) 320 { 321 struct ld_thunkbus_softc *sc = (struct ld_thunkbus_softc *)ld; 322 323 if (thunk_fsync(sc->sc_fd) == -1) 324 return thunk_geterrno(); 325 326 return 0; 327 } 328