1 /* $NetBSD: fdformat.c,v 1.12 2004/04/23 15:04:27 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 1996, 1997 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by John Kohl. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 3. All advertising materials mentioning features or use of this software 19 * must display the following acknowledgement: 20 * This product includes software developed by the NetBSD 21 * Foundation, Inc. and its contributors. 22 * 4. Neither the name of The NetBSD Foundation nor the names of its 23 * contributors may be used to endorse or promote products derived 24 * from this software without specific prior written permission. 25 * 26 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 27 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 28 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 29 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 30 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 31 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 32 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 33 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 34 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 35 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 36 * POSSIBILITY OF SUCH DAMAGE. 37 */ 38 39 /* 40 * fdformat: format a floppy diskette, using interface provided in 41 * <sys/fdio.h> 42 */ 43 #include <sys/cdefs.h> 44 45 #ifndef lint 46 __RCSID("$NetBSD: fdformat.c,v 1.12 2004/04/23 15:04:27 christos Exp $"); 47 #endif 48 49 #include <sys/types.h> 50 #include <sys/fdio.h> 51 #include <sys/ioctl.h> 52 53 #include <err.h> 54 #include <errno.h> 55 #include <fcntl.h> 56 #include <limits.h> 57 #include <stdio.h> 58 #include <stdlib.h> 59 #include <unistd.h> 60 #include "pathnames.h" 61 62 static const char *fdb_array[2] = {_PATH_FLOPPYTAB, 0}; 63 64 #define MASK_NBPS 0x0001 65 #define MASK_NCYL 0x0002 66 #define MASK_NSPT 0x0004 67 #define MASK_NTRK 0x0008 68 #define MASK_STEPSPERCYL 0x0010 69 #define MASK_GAPLEN 0x0020 70 #define MASK_FILLBYTE 0x0040 71 #define MASK_XFER_RATE 0x0080 72 #define MASK_INTERLEAVE 0x0100 73 74 #define ALLPARMS (MASK_NBPS|MASK_NCYL|MASK_NSPT|MASK_NTRK|MASK_STEPSPERCYL|MASK_GAPLEN|MASK_FILLBYTE|MASK_XFER_RATE|MASK_INTERLEAVE) 75 76 static int confirm(int); 77 static void usage(void) __attribute__((__noreturn__)); 78 static int verify_track(int, int, int, struct fdformat_parms *, char *); 79 static int wincolsize(void); 80 81 int main(int, char **); 82 83 static int 84 confirm(int def) 85 { 86 int ch; 87 88 (void)printf(" Yes/no [%c]?", def ? 'y' : 'n'); 89 ch = getchar(); 90 switch (ch) { 91 case 'y': 92 case 'Y': 93 return 1; 94 case '\n': 95 return def; 96 case EOF: 97 case 'n': 98 case 'N': 99 default: 100 return 0; 101 } 102 } 103 104 static int 105 verify_track(int fd, int cyl, int trk, struct fdformat_parms *parms, char *buf) 106 { 107 size_t tracksize; 108 off_t offset; 109 110 tracksize = parms->nbps * parms->nspt; /* bytes per track */ 111 offset = tracksize * (cyl * parms->ntrk + trk); /* track offset */ 112 113 if (lseek(fd, offset, SEEK_SET) == (off_t) -1) { 114 (void)putchar('E'); 115 return 1; 116 } 117 if (read(fd, buf, tracksize) != tracksize) { 118 (void)putchar('E'); 119 return 1; 120 } else 121 (void)putchar('V'); 122 return 0; 123 } 124 125 static void 126 usage(void) 127 { 128 (void)fprintf(stderr, 129 "Usage: %s [-f device] [-t type] [-n] [-B nbps] [-S nspt]\n" 130 "\t[-T ntrk] [-C ncyl] [-P stepspercyl] [-G gaplen]\n" 131 "\t[-F fillbyte] [-X xfer_rate] [-I interleave]\n", getprogname()); 132 exit(1); 133 } 134 135 static int 136 wincolsize(void) 137 { 138 struct winsize win; 139 140 if (ioctl(STDIN_FILENO, TIOCGWINSZ, &win) == -1) { 141 warn("TIOCGWINSZ"); 142 return 80; 143 } 144 return win.ws_col; 145 } 146 147 #define numarg(which, maskn, op) \ 148 do { \ 149 tmplong = strtol(optarg, &tmpcharp, 0); \ 150 if (*tmpcharp != '\0' || tmplong op 0) \ 151 errx(1, \ 152 "Invalid numerical argument `%s' for " \ 153 # which, optarg); \ 154 if (errno == ERANGE && (tmplong == LONG_MIN || \ 155 tmplong == LONG_MAX)) \ 156 err(1, \ 157 "Bad numerical argument `%s' for " \ 158 # which, optarg); \ 159 parms.which = tmplong; \ 160 parmmask |= MASK_##maskn; \ 161 } while (/* CONSTCOND */0) 162 163 #define getparm(structname, maskname) \ 164 do { \ 165 if (cgetnum(fdbuf, # structname, &tmplong) == -1) \ 166 errx(1, "Parameter " # structname \ 167 " missing for type `%s'", optarg); \ 168 parms.structname = tmplong; \ 169 parmmask |= MASK_ ## maskname; \ 170 } while (/* CONSTCOND */0) 171 172 173 #define copyparm(which, mask) \ 174 if ((parmmask & MASK_##mask) == 0) \ 175 parms.which = fetchparms.which 176 177 int 178 main(int argc, char *argv[]) 179 { 180 char *fdbuf = NULL, *trackbuf = NULL; 181 int errcnt = 0; 182 int verify = 1; 183 int ch; 184 long tmplong; 185 int tmpint; 186 char *tmpcharp; 187 int parmmask = 0; 188 struct fdformat_parms parms, fetchparms; 189 struct fdformat_cmd cmd; 190 const char *filename = _PATH_FLOPPY_DEV; 191 int fd; 192 int trk, cyl; 193 int colcnt = 0; 194 int colsize = wincolsize(); 195 196 while ((ch = getopt(argc, argv, "f:t:nB:C:S:T:P:G:F:X:I:")) != -1) 197 switch (ch) { 198 case 't': /* disk type */ 199 switch (cgetent(&fdbuf, fdb_array, optarg)) { 200 case 0: 201 break; 202 case 1: 203 case -3: 204 errx(1, "tc= loop or missing entry entry in " 205 _PATH_FLOPPYTAB " for type %s", optarg); 206 break; 207 case -1: 208 errx(1, "Unknown floppy disk type %s", optarg); 209 break; 210 default: 211 err(1, "Problem accessing " _PATH_FLOPPYTAB); 212 break; 213 } 214 215 getparm(nbps, NBPS); 216 getparm(ncyl, NCYL); 217 getparm(nspt, NSPT); 218 getparm(ntrk, NTRK); 219 getparm(stepspercyl, STEPSPERCYL); 220 getparm(gaplen, GAPLEN); 221 getparm(fillbyte, FILLBYTE); 222 getparm(xfer_rate, XFER_RATE); 223 getparm(interleave, INTERLEAVE); 224 break; 225 case 'f': /* device name */ 226 filename = optarg; 227 break; 228 case 'n': /* no verify */ 229 verify = 0; 230 break; 231 case 'B': 232 numarg(nbps, NBPS, <=); 233 break; 234 case 'C': 235 numarg(ncyl, NCYL, <=); 236 break; 237 case 'S': 238 numarg(nspt, NSPT, <=); 239 break; 240 case 'T': 241 numarg(ntrk, NTRK, <=); 242 break; 243 case 'P': 244 numarg(stepspercyl, STEPSPERCYL, <=); 245 break; 246 case 'G': 247 numarg(gaplen, GAPLEN, <=); 248 break; 249 case 'F': 250 numarg(fillbyte, FILLBYTE, <); 251 break; 252 case 'X': 253 numarg(xfer_rate, XFER_RATE, <=); 254 break; 255 case 'I': 256 numarg(interleave, INTERLEAVE, <=); 257 break; 258 case '?': 259 default: 260 usage(); 261 } 262 263 if (optind < argc) 264 usage(); 265 266 fd = open(filename, O_RDWR); 267 if (fd == -1) 268 err(1, "Cannot open %s", filename); 269 if (ioctl(fd, FDIOCGETFORMAT, &fetchparms) == -1) { 270 if (errno == ENOTTY) 271 err(1, "Device `%s' does not support floppy formatting", 272 filename); 273 else 274 err(1, "Cannot fetch current floppy" 275 " formatting parameters"); 276 } 277 278 copyparm(nbps, NBPS); 279 copyparm(ncyl, NCYL); 280 copyparm(nspt, NSPT); 281 copyparm(ntrk, NTRK); 282 copyparm(stepspercyl, STEPSPERCYL); 283 copyparm(gaplen, GAPLEN); 284 copyparm(fillbyte, FILLBYTE); 285 copyparm(xfer_rate, XFER_RATE); 286 copyparm(interleave, INTERLEAVE); 287 288 parms.fdformat_version = FDFORMAT_VERSION; 289 290 tmpint = FDOPT_NORETRY|FDOPT_SILENT; 291 if (ioctl(fd, FDIOCSETOPTS, &tmpint) == -1 || 292 ioctl(fd, FDIOCSETFORMAT, &parms) == -1) { 293 errx(1, "Cannot set requested formatting parameters:" 294 " %d cylinders, %d tracks, %d sectors of %d bytes", 295 parms.ncyl, parms.ntrk, parms.nspt, parms.nbps); 296 } 297 298 (void)printf("Ready to format %s with %d cylinders, %d tracks," 299 " %d sectors of %d bytes\n(%d KB)", 300 filename, parms.ncyl, parms.ntrk, parms.nspt, parms.nbps, 301 parms.ncyl * parms.ntrk * parms.nspt * parms.nbps / 1024); 302 if (!confirm(1)) 303 errx(1,"Formatting abandoned -- not confirmed."); 304 305 if (verify) { 306 trackbuf = malloc(parms.nbps * parms.nspt); 307 if (trackbuf == NULL) 308 warn("Cannot allocate verification buffer"); 309 } 310 311 cmd.formatcmd_version = FDFORMAT_VERSION; 312 for (cyl = 0; cyl < parms.ncyl; cyl++) { 313 cmd.cylinder = cyl; 314 for (trk = 0; trk < parms.ntrk; trk++) { 315 cmd.head = trk; 316 if (ioctl(fd, FDIOCFORMAT_TRACK, &cmd) == 0) { 317 if (verify && colsize) { 318 colcnt++; 319 if (colcnt % colsize == 0) { 320 (void)putchar('\n'); 321 colcnt++; 322 } 323 } 324 (void)putchar('F'); 325 if (verify) 326 (void)putchar('\b'); 327 (void)fflush(stdout); 328 if (verify) 329 errcnt += verify_track(fd, cyl, trk, 330 &parms, trackbuf); 331 } else if (errno == EINVAL) { 332 (void)putchar('\n'); 333 errx(1, "Formatting botch at <%d,%d>", 334 cyl, trk); 335 } else if (errno == EIO) { 336 (void)putchar('E'); 337 (void)fflush(stdout); 338 errcnt++; 339 } 340 } 341 } 342 (void)putchar('\n'); 343 if (errcnt) 344 errx(1, "%d track formatting error%s", 345 errcnt, errcnt == 1 ? "" : "s"); 346 return 0; 347 } 348