1 /* $NetBSD: md_root.c,v 1.15 2000/11/27 08:39:40 chs Exp $ */ 2 3 /* 4 * Copyright (c) 1996 Leo Weppelman. 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 * 3. All advertising materials mentioning features or use of this software 16 * must display the following acknowledgement: 17 * This product includes software developed by Leo Weppelman. 18 * 4. The name of the author may not be used to endorse or promote products 19 * derived from this software without specific prior written permission 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 22 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 23 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 24 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 25 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 26 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 30 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 31 */ 32 33 #include <sys/param.h> 34 #include <sys/systm.h> 35 #include <sys/kernel.h> 36 #include <sys/malloc.h> 37 #include <sys/buf.h> 38 #include <sys/proc.h> 39 #include <sys/device.h> 40 #include <sys/ioctl.h> 41 #include <sys/fcntl.h> 42 #include <sys/conf.h> 43 #include <sys/disklabel.h> 44 #include <sys/disk.h> 45 #include <sys/dkbad.h> 46 47 #include <dev/cons.h> 48 #include <dev/md.h> 49 50 #include <atari/atari/device.h> 51 52 /* 53 * Misc. defines: 54 */ 55 #define RAMD_CHUNK (9 * 512) /* Chunk-size for auto-load */ 56 #define RAMD_NDEV 2 /* Number of devices configured */ 57 58 struct ramd_info { 59 u_long ramd_size; /* Size of disk in bytes */ 60 u_long ramd_flag; /* see defs below */ 61 dev_t ramd_dev; /* device to load from */ 62 }; 63 64 /* 65 * ramd_flag: 66 */ 67 #define RAMD_LOAD 0x01 /* Auto load when first opened */ 68 #define RAMD_LCOMP 0x02 /* Input is compressed */ 69 70 struct ramd_info rd_info[RAMD_NDEV] = { 71 { 72 1105920, /* 1Mb in 2160 sectors */ 73 RAMD_LOAD, /* auto-load this device */ 74 MAKEDISKDEV(2, 0, 2), /* XXX: This is crap! (720Kb flop) */ 75 }, 76 { 77 1105920, /* 1Mb in 2160 sectors */ 78 RAMD_LOAD, /* auto-load this device */ 79 MAKEDISKDEV(2, 0, 3), /* XXX: This is crap! (1.44Mb flop) */ 80 } 81 }; 82 83 struct read_info { 84 struct buf *bp; /* buffer for strategy function */ 85 long nbytes; /* total number of bytes to read */ 86 long offset; /* offset in input medium */ 87 caddr_t bufp; /* current output buffer */ 88 caddr_t ebufp; /* absolute maximum for bufp */ 89 int chunk; /* chunk size on input medium */ 90 int media_sz; /* size of input medium */ 91 void (*strat)(struct buf *); /* strategy function for read */ 92 }; 93 94 95 static int loaddisk __P((struct md_conf *, dev_t ld_dev, struct proc *)); 96 static int ramd_norm_read __P((struct read_info *)); 97 98 #ifdef support_compression 99 static int cpy_uncompressed __P((caddr_t, int, struct read_info *)); 100 static int md_compressed __P((caddr_t, int, struct read_info *)); 101 #endif 102 103 /* 104 * This is called during autoconfig. 105 */ 106 void 107 md_attach_hook(unit, md) 108 int unit; 109 struct md_conf *md; 110 { 111 if (atari_realconfig && (unit < RAMD_NDEV) && rd_info[unit].ramd_flag) { 112 printf ("md%d: %sauto-load on open. Size %ld bytes.\n", unit, 113 rd_info[unit].ramd_flag & RAMD_LCOMP ? "decompress/" : "", 114 rd_info[unit].ramd_size); 115 md->md_type = MD_UNCONFIGURED; /* Paranoia... */ 116 } 117 } 118 119 void 120 md_open_hook(unit, md) 121 int unit; 122 struct md_conf *md; 123 { 124 struct ramd_info *ri; 125 126 if(unit >= RAMD_NDEV) 127 return; 128 129 ri = &rd_info[unit]; 130 if (md->md_type != MD_UNCONFIGURED) 131 return; /* Only configure once */ 132 md->md_addr = malloc(ri->ramd_size, M_DEVBUF, M_WAITOK); 133 md->md_size = ri->ramd_size; 134 if(md->md_addr == NULL) 135 return; 136 if(ri->ramd_flag & RAMD_LOAD) { 137 if (loaddisk(md, ri->ramd_dev, curproc)) { 138 free(md->md_addr, M_DEVBUF); 139 md->md_addr = NULL; 140 return; 141 } 142 } 143 md->md_type = MD_KMEM_ALLOCATED; 144 } 145 146 static int 147 loaddisk(md, ld_dev, proc) 148 struct md_conf *md; 149 dev_t ld_dev; 150 struct proc *proc; 151 { 152 struct buf buf; 153 int error; 154 struct bdevsw *bdp = &bdevsw[major(ld_dev)]; 155 struct disklabel dl; 156 struct read_info rs; 157 158 /* 159 * Initialize our buffer header: 160 */ 161 memset(&buf, 0, sizeof(buf)); 162 buf.b_vnbufs.le_next = NOLIST; 163 buf.b_flags = B_BUSY; 164 buf.b_dev = ld_dev; 165 buf.b_error = 0; 166 buf.b_proc = proc; 167 168 /* 169 * Setup read_info: 170 */ 171 rs.bp = &buf; 172 rs.nbytes = md->md_size; 173 rs.offset = 0; 174 rs.bufp = md->md_addr; 175 rs.ebufp = md->md_addr + md->md_size; 176 rs.chunk = RAMD_CHUNK; 177 rs.media_sz = md->md_size; 178 rs.strat = bdp->d_strategy; 179 180 /* 181 * Open device and try to get some statistics. 182 */ 183 if((error = bdp->d_open(ld_dev, FREAD | FNONBLOCK, 0, proc)) != 0) 184 return(error); 185 if(bdp->d_ioctl(ld_dev, DIOCGDINFO, (caddr_t)&dl, FREAD, proc) == 0) { 186 /* Read on a cylinder basis */ 187 rs.chunk = dl.d_secsize * dl.d_secpercyl; 188 rs.media_sz = dl.d_secperunit * dl.d_secsize; 189 } 190 191 #ifdef support_compression 192 if(ri->ramd_flag & RAMD_LCOMP) 193 error = decompress(cpy_uncompressed, md_compressed, &rs); 194 else 195 #endif /* support_compression */ 196 error = ramd_norm_read(&rs); 197 198 bdp->d_close(ld_dev,FREAD | FNONBLOCK, 0, proc); 199 return(error); 200 } 201 202 static int 203 ramd_norm_read(rsp) 204 struct read_info *rsp; 205 { 206 long bytes_left; 207 int done, error; 208 struct buf *bp; 209 int s; 210 int dotc = 0; 211 212 bytes_left = rsp->nbytes; 213 bp = rsp->bp; 214 error = 0; 215 216 while(bytes_left > 0) { 217 s = splbio(); 218 bp->b_flags = B_BUSY | B_PHYS | B_READ; 219 splx(s); 220 bp->b_blkno = btodb(rsp->offset); 221 bp->b_bcount = rsp->chunk; 222 bp->b_data = rsp->bufp; 223 224 /* Initiate read */ 225 (*rsp->strat)(bp); 226 227 /* Wait for results */ 228 s = splbio(); 229 while ((bp->b_flags & B_DONE) == 0) 230 tsleep((caddr_t) bp, PRIBIO + 1, "ramd_norm_read", 0); 231 if (bp->b_flags & B_ERROR) 232 error = (bp->b_error ? bp->b_error : EIO); 233 splx(s); 234 235 /* Dot counter */ 236 printf("."); 237 if(!(++dotc % 40)) 238 printf("\n"); 239 240 done = bp->b_bcount - bp->b_resid; 241 bytes_left -= done; 242 rsp->offset += done; 243 rsp->bufp += done; 244 245 if(error || !done) 246 break; 247 248 if((rsp->offset == rsp->media_sz) && (bytes_left != 0)) { 249 printf("\nInsert next media and hit any key..."); 250 cngetc(); 251 printf("\n"); 252 rsp->offset = 0; 253 } 254 } 255 printf("\n"); 256 return(error); 257 } 258 259 #ifdef support_compression 260 /* 261 * Functions supporting uncompression: 262 */ 263 /* 264 * Copy from the uncompression buffer to the ramdisk 265 */ 266 static int 267 cpy_uncompressed(buf, nbyte, rsp) 268 caddr_t buf; 269 struct read_info *rsp; 270 int nbyte; 271 { 272 if((rsp->bufp + nbyte) >= rsp->ebufp) 273 return(0); 274 bcopy(buf, rsp->bufp, nbyte); 275 rsp->bufp += nbyte; 276 return(0); 277 } 278 279 /* 280 * Read a maximum of 'nbyte' bytes into 'buf'. 281 */ 282 static int 283 md_compressed(buf, nbyte, rsp) 284 caddr_t buf; 285 struct read_info *rsp; 286 int nbyte; 287 { 288 static int dotc = 0; 289 struct buf *bp; 290 int nread = 0; 291 int s; 292 int done, error; 293 294 295 error = 0; 296 bp = rsp->bp; 297 nbyte &= ~(DEV_BSIZE - 1); 298 299 while(nbyte > 0) { 300 s = splbio(); 301 bp->b_flags = B_BUSY | B_PHYS | B_READ; 302 splx(s); 303 bp->b_blkno = btodb(rsp->offset); 304 bp->b_bcount = min(rsp->chunk, nbyte); 305 bp->b_data = buf; 306 307 /* Initiate read */ 308 (*rsp->strat)(bp); 309 310 /* Wait for results */ 311 s = splbio(); 312 while ((bp->b_flags & B_DONE) == 0) 313 tsleep((caddr_t) bp, PRIBIO + 1, "ramd_norm_read", 0); 314 if (bp->b_flags & B_ERROR) 315 error = (bp->b_error ? bp->b_error : EIO); 316 splx(s); 317 318 /* Dot counter */ 319 printf("."); 320 if(!(++dotc % 40)) 321 printf("\n"); 322 323 done = bp->b_bcount - bp->b_resid; 324 nbyte -= done; 325 nread += done; 326 rsp->offset += done; 327 328 if(error || !done) 329 break; 330 331 if((rsp->offset == rsp->media_sz) && (nbyte != 0)) { 332 if(rsp->offset == rsp->media_sz) { 333 printf("\nInsert next media and hit any key..."); 334 if(cngetc() != '\n') 335 printf("\n"); 336 rsp->offset = 0; 337 } 338 } 339 s = splbio(); 340 splx(s); 341 return(nread); 342 } 343 #endif /* support_compression */ 344