util-linux/fdformat.c

   1 /* vi: set sw=4 ts=4: */
   2 /* fdformat.c  -  Low-level formats a floppy disk - Werner Almesberger */
   3
   4 /* 1999-02-22 Arkadiusz Mi¶kiewicz <misiek@pld.ORG.PL>
   5  * - added Native Language Support
   6  * 1999-03-20 Arnaldo Carvalho de Melo <acme@conectiva.com.br>
   7  * - more i18n/nls translatable strings marked
   8  *
   9  * 5 July 2003 -- modified for Busybox by Erik Andersen
  10  */
  11
  12 #include "busybox.h"
  13
  14
  15 /* Stuff extracted from linux/fd.h */
  16 struct floppy_struct {
  17         unsigned int    size,           /* nr of sectors total */
  18                         sect,           /* sectors per track */
  19                         head,           /* nr of heads */
  20                         track,          /* nr of tracks */
  21                         stretch;        /* !=0 means double track steps */
  22 #define FD_STRETCH 1
  23 #define FD_SWAPSIDES 2
  24
  25         unsigned char   gap,            /* gap1 size */
  26
  27                         rate,           /* data rate. |= 0x40 for perpendicular */
  28 #define FD_2M 0x4
  29 #define FD_SIZECODEMASK 0x38
  30 #define FD_SIZECODE(floppy) (((((floppy)->rate&FD_SIZECODEMASK)>> 3)+ 2) %8)
  31 #define FD_SECTSIZE(floppy) ( (floppy)->rate & FD_2M ? \
  32                              512 : 128 << FD_SIZECODE(floppy) )
  33 #define FD_PERP 0x40
  34
  35                         spec1,          /* stepping rate, head unload time */
  36                         fmt_gap;        /* gap2 size */
  37         const char      * name; /* used only for predefined formats */
  38 };
  39 struct format_descr {
  40         unsigned int device,head,track;
  41 };
  42 #define FDFMTBEG _IO(2,0x47)
  43 #define FDFMTTRK _IOW(2,0x48, struct format_descr)
  44 #define FDFMTEND _IO(2,0x49)
  45 #define FDGETPRM _IOR(2, 0x04, struct floppy_struct)
  46 #define FD_FILL_BYTE 0xF6 /* format fill byte. */
  47
  48 static void xioctl(int fd, int request, void *argp, const char *string)
  49 {
  50         if (ioctl (fd, request, argp) < 0) {
  51                 bb_perror_msg_and_die(string);
  52         }
  53 }
  54
  55 int fdformat_main(int argc,char **argv)
  56 {
  57         int fd, n, cyl, read_bytes, verify;
  58         unsigned char *data;
  59         struct stat st;
  60         struct floppy_struct param;
  61         struct format_descr descr;
  62
  63         if (argc < 2) {
  64                 bb_show_usage();
  65         }
  66         verify = !bb_getopt_ulflags(argc, argv, "n");
  67         argv += optind;
  68
  69         /* R_OK is needed for verifying */
  70         if (stat(*argv,&st) < 0 || access(*argv,W_OK | R_OK ) < 0) {
  71                 bb_perror_msg_and_die("%s",*argv);
  72         }
  73         if (!S_ISBLK(st.st_mode)) {
  74                 bb_error_msg_and_die("%s: not a block device",*argv);
  75                 /* do not test major - perhaps this was an USB floppy */
  76         }
  77
  78
  79         /* O_RDWR for formatting and verifying */
  80         fd = xopen(*argv,O_RDWR );
  81
  82         xioctl(fd, FDGETPRM, &param, "FDGETPRM");/*original message was: "Could not determine current format type" */
  83
  84         printf("%s-sided, %d tracks, %d sec/track. Total capacity %d kB.\n",
  85                 (param.head == 2) ? "Double" : "Single",
  86                 param.track, param.sect, param.size >> 1);
  87
  88         /* FORMAT */
  89         printf("Formatting ... ");
  90         xioctl(fd, FDFMTBEG,NULL,"FDFMTBEG");
  91
  92         /* n == track */
  93         for (n = 0; n < param.track; n++)
  94         {
  95             descr.head = 0;
  96             descr.track = n;
  97             xioctl(fd, FDFMTTRK,&descr,"FDFMTTRK");
  98             printf("%3d\b\b\b", n);
  99             if (param.head == 2) {
 100                 descr.head = 1;
 101                 xioctl(fd, FDFMTTRK,&descr,"FDFMTTRK");
 102             }
 103         }
 104
 105         xioctl(fd,FDFMTEND,NULL,"FDFMTEND");
 106         printf("done\n");
 107
 108         /* VERIFY */
 109         if(verify) {
 110                 /* n == cyl_size */
 111                 n = param.sect*param.head*512;
 112
 113                 data = xmalloc(n);
 114                 printf("Verifying ... ");
 115                 for (cyl = 0; cyl < param.track; cyl++) {
 116                         printf("%3d\b\b\b", cyl);
 117                         if((read_bytes = safe_read(fd,data,n))!= n ) {
 118                                 if(read_bytes < 0) {
 119                                         bb_perror_msg(bb_msg_read_error);
 120                                 }
 121                                 bb_error_msg_and_die("Problem reading cylinder %d, expected %d, read %d", cyl, n, read_bytes);
 122                         }
 123                         /* Check backwards so we don't need a counter */
 124                         while(--read_bytes>=0) {
 125                                 if( data[read_bytes] != FD_FILL_BYTE) {
 126                                          printf("bad data in cyl %d\nContinuing ... ",cyl);
 127                                 }
 128                         }
 129                 }
 130                 /* There is no point in freeing blocks at the end of a program, because
 131                 all of the program's space is given back to the system when the process
 132                 terminates.*/
 133
 134                 if (ENABLE_FEATURE_CLEAN_UP) free(data);
 135
 136                 printf("done\n");
 137         }
 138
 139         if (ENABLE_FEATURE_CLEAN_UP) close(fd);
 140
 141         /* Don't bother closing.  Exit does
 142          * that, so we can save a few bytes */
 143         return EXIT_SUCCESS;
 144 }