git.maemo.org Git - busybox4maemo/blob - util-linux/fdformat.c

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