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