diag/geodsp/mk-lba-img.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 2010 Gene Cumm
   4  *
   5  *   This program is free software; you can redistribute it and/or modify
   6  *   it under the terms of the GNU General Public License as published by
   7  *   the Free Software Foundation, Inc., 53 Temple Place Ste 330,
   8  *   Boston MA 02111-1307, USA; either version 2 of the License, or
   9  *   (at your option) any later version; incorporated herein by reference.
  10  *
  11  * ----------------------------------------------------------------------- */
  12
  13 /*
  14  * mk-lba-img.c
  15  *
  16  * Makes an image that contains the LBA in every *word of every sector
  17  */
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <errno.h>
  22 #include <string.h>
  23 #include <unistd.h>
  24
  25 #define NUM_SECT (256*63+1)
  26 #define BPS (512)
  27 #define SECT_INT (BPS / sizeof(unsigned int))
  28
  29 typedef unsigned char uint8_t;
  30 typedef unsigned int uint32_t;
  31
  32 const char DEF_FN[] = "-";
  33
  34 int main(int argc, char *argv[])
  35 {
  36         int i, rv = 0, one = 0;
  37         unsigned int lba, b[SECT_INT];
  38         int len;
  39         FILE *f;
  40         uint8_t tt = 0;
  41         const char *fn;
  42
  43         if (argc >= 2) {
  44                 if (argc >= 3) {
  45                         if (strcasecmp("-1", argv[1]) == 0) {
  46                                 fn = argv[2];
  47                                 one = 1;
  48                         } else {
  49                                 fn = argv[1];
  50                         }
  51                 } else {
  52                         fn = argv[1];
  53                 }
  54         } else {
  55                 fn = DEF_FN;
  56         }
  57
  58         if (!strcmp(fn, "-"))
  59                 f = stdout;
  60         else
  61                 f = fopen(fn, "w");
  62
  63         if (!f) {
  64                 fprintf(stderr, "%s: %s: unable to open for writing: %s\n",
  65                         argv[0], fn, strerror(errno));
  66                 return 1;
  67         }
  68
  69         lba = 0;
  70         while ((len = fread(b, 1, BPS, stdin))) {
  71                 if (len < BPS)
  72                         memset((char *)b + len, 0, BPS - len);
  73                 fwrite(b, 1, BPS, f);
  74                 lba++;
  75         }
  76
  77         memset(b, 0, sizeof b);
  78
  79         while (lba < NUM_SECT) {
  80                 if (one) {
  81                         b[0] = lba;
  82                 } else {
  83                         for (i = 0; i < SECT_INT; i++)
  84                                 b[i] = lba;
  85                 }
  86                 fwrite(b, 1, BPS, f);
  87                 lba++;
  88         }
  89
  90         if (f != stdout)
  91                 fclose(f);
  92
  93         return rv;
  94 }