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
29typedef unsigned char uint8_t;
30typedef unsigned int uint32_t;
31
32const char DEF_FN[] = "-";
33
34int 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}
95