1 /* 2 * ---------------------------------------------------------------------------- 3 * "THE BEER-WARE LICENSE" (Revision 42): 4 * <phk@login.dkuug.dk> wrote this file. As long as you retain this notice you 5 * can do whatever you want with this stuff. If we meet some day, and you think 6 * this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp 7 * ---------------------------------------------------------------------------- 8 * 9 * $Id$ 10 * 11 */ 12 13 #include <stdio.h> 14 #include <stdlib.h> 15 #include <unistd.h> 16 #include <fcntl.h> 17 #include <strings.h> 18 #include <ctype.h> 19 20 #include <errno.h> 21 #include <machine/ioctl_fd.h> 22 23 int 24 format_track(int fd, int cyl, int secs, int head, int rate, 25 int gaplen, int secsize, int fill,int interleave) 26 { 27 struct fd_formb f; 28 register int i,j; 29 int il[100]; 30 31 memset(il,0,sizeof il); 32 for(j = 0, i = 1; i <= secs; i++) { 33 while(il[(j%secs)+1]) j++; 34 il[(j%secs)+1] = i; 35 j += interleave; 36 } 37 38 f.format_version = FD_FORMAT_VERSION; 39 f.head = head; 40 f.cyl = cyl; 41 f.transfer_rate = rate; 42 43 f.fd_formb_secshift = secsize; 44 f.fd_formb_nsecs = secs; 45 f.fd_formb_gaplen = gaplen; 46 f.fd_formb_fillbyte = fill; 47 for(i = 0; i < secs; i++) { 48 f.fd_formb_cylno(i) = cyl; 49 f.fd_formb_headno(i) = head; 50 f.fd_formb_secno(i) = il[i+1]; 51 f.fd_formb_secsize(i) = secsize; 52 } 53 return ioctl(fd, FD_FORM, (caddr_t)&f); 54 } 55 56 static void 57 usage () 58 { 59 printf("Usage:\n\tfdwrite [-v] [-y] [-f inputfile] [-d device]\n"); 60 exit(2); 61 } 62 63 int 64 main(int argc, char **argv) 65 { 66 int inputfd = -1, c, fdn = 0, i,j,fd; 67 int bpt, verbose=1, nbytes=0, track; 68 int interactive = 1; 69 char *device= "/dev/rfd0", *trackbuf = 0,*vrfybuf = 0; 70 struct fd_type fdt; 71 FILE *tty; 72 73 setbuf(stdout,0); 74 while((c = getopt(argc, argv, "d:s:f:vy")) != -1) 75 switch(c) { 76 case 'd': /* Which drive */ 77 device = optarg; 78 break; 79 80 case 'f': /* input file */ 81 if (inputfd >= 0) 82 close(inputfd); 83 inputfd = open(optarg,O_RDONLY); 84 if (inputfd < 0) { 85 perror(optarg); 86 exit(1); 87 } 88 break; 89 90 case 'v': /* Toggle verbosity */ 91 verbose = !verbose; 92 break; 93 94 case 'y': /* Don't confirm? */ 95 interactive = 0; 96 break; 97 98 case '?': default: 99 usage(); 100 } 101 102 if (inputfd < 0) 103 inputfd = 0; 104 105 if (!isatty(1)) 106 interactive = 0; 107 108 if(optind < argc) 109 usage(); 110 111 tty = fopen("/dev/tty","r+"); 112 if(!tty) { 113 perror("/dev/tty"); 114 exit(1); 115 } 116 setbuf(tty,0); 117 118 for(j=1;j > 0;) { 119 fdn++; 120 if (interactive) { 121 fprintf(tty, 122 "Please insert floppy #%d in drive %s and press return >", 123 fdn,device); 124 while(1) { 125 i = getc(tty); 126 if(i == '\n') break; 127 } 128 } 129 130 if((fd = open(device, O_RDWR)) < 0) { 131 perror(device); 132 exit(1); 133 } 134 135 if(ioctl(fd, FD_GTYPE, &fdt) < 0) { 136 fprintf(stderr, "fdformat: not a floppy disk: %s\n", device); 137 exit(1); 138 } 139 140 bpt = fdt.sectrac * (1<<fdt.secsize) * 128; 141 if(!trackbuf) { 142 trackbuf = malloc(bpt); 143 if(!trackbuf) { perror("malloc"); exit(1); } 144 } 145 if(!vrfybuf) { 146 vrfybuf = malloc(bpt); 147 if(!vrfybuf) { perror("malloc"); exit(1); } 148 } 149 150 if(fdn == 1) { 151 if(verbose) { 152 printf("Format: %d cylinders, %d heads, %d sectors, %d bytes = %dkb\n", 153 fdt.tracks,fdt.heads,fdt.sectrac,(1<<fdt.secsize) * 128, 154 fdt.tracks*bpt*fdt.heads/1024); 155 156 } 157 memset(trackbuf,0,bpt); 158 for(j=0;inputfd >= 0 && j<bpt;j+=i) { 159 if(!(i = read(inputfd,trackbuf+j,bpt-j))) { 160 close(inputfd); 161 inputfd = -1; 162 break; 163 } 164 nbytes += i; 165 } 166 } 167 for (track = 0; track < fdt.tracks * fdt.heads; track++) { 168 if(verbose) printf("\r%3d ",fdt.tracks * fdt.heads-track); 169 if(verbose) putc((j ? 'I':'Z'),stdout); 170 format_track(fd, track / fdt.heads, fdt.sectrac, track % fdt.heads, 171 fdt.trans, fdt.f_gap, fdt.secsize, 0xe6, 172 fdt.f_inter); 173 if(verbose) putc('F',stdout); 174 175 if (lseek (fd, (long) track*bpt, 0) < 0) { 176 perror("lseek"); 177 exit (1); 178 } 179 if (write (fd, trackbuf, bpt) != bpt) { 180 perror("write"); 181 exit (1); 182 } 183 if(verbose) putc('W',stdout); 184 185 if (lseek (fd, (long) track*bpt, 0) < 0) { 186 perror("lseek"); 187 exit (1); 188 } 189 if (read (fd, vrfybuf, bpt) != bpt) { 190 perror("read"); 191 exit (1); 192 } 193 if(verbose) putc('R',stdout); 194 195 if (memcmp(trackbuf,vrfybuf,bpt)) { 196 perror("compare"); 197 exit (1); 198 } 199 if(verbose) putc('C',stdout); 200 201 memset(trackbuf,0,bpt); 202 for(j=0;inputfd >= 0 && j<bpt;j+=i) { 203 if(!(i = read(inputfd,trackbuf+j,bpt-j))) { 204 close(inputfd); 205 inputfd = -1; 206 break; 207 } 208 nbytes += i; 209 } 210 } 211 close(fd); 212 putc('\r',stdout); 213 } 214 if(verbose) 215 printf("%d bytes on %d flopp%s\n",nbytes,fdn,fdn==1?"y":"ies"); 216 exit(0); 217 } 218