xref: /netbsd-src/usr.bin/fdformat/fdformat.c (revision fdecd6a253f999ae92b139670d9e15cc9df4497c)
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