From e6b5c8182d60ce160b4c4d7e706afaa6b30f5645 Mon Sep 17 00:00:00 2001
From: Bobby Bingham <koorogi@koorogi.info>
Date: Sun, 2 Aug 2015 17:39:27 -0500
Subject: add satmkiso utility

---
 Makefile       |  12 +++---
 src/satmkiso.c | 116 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 121 insertions(+), 7 deletions(-)
 create mode 100644 src/satmkiso.c

diff --git a/Makefile b/Makefile
index de590e8..c788d7e 100644
--- a/Makefile
+++ b/Makefile
@@ -8,7 +8,7 @@ BOOTBINS = $(patsubst boot/%.c,share/%.boot,$(BOOTSRCS))
 SRCS = $(sort $(wildcard src/*.c))
 DATA = $(sort $(wildcard src/*.bin))
 OBJS = $(SRCS:.c=.o) $(DATA:.bin=.o)
-BINS = bin/satmkboot
+BINS = bin/satmkboot bin/satmkiso
 
 .PHONY: all clean
 
@@ -17,10 +17,10 @@ all: $(BINS) $(BOOTBINS)
 clean:
 	rm -f $(BINS) $(OBJS) $(BOOTBINS) $(BOOTELFS)
 
-bin/satmkboot:         \
-	src/satmkboot.o    \
-	src/securitycode.o \
-	src/symbols.o
+bin/satmkboot: src/securitycode.o src/symbols.o
+
+bin/%: src/%.o
+	$(CC) $(LDFLAGS) $^ -o $@
 
 share/%.boot: boot/%.elf
 	saturn-objcopy -O binary $< $@
@@ -34,5 +34,3 @@ share/%.boot: boot/%.elf
 %.o: %.c
 	$(CC) $(CFLAGS) -c $< -o $@
 
-%:
-	$(CC) $(LDFLAGS) $^ -o $@
diff --git a/src/satmkiso.c b/src/satmkiso.c
new file mode 100644
index 0000000..766d3ce
--- /dev/null
+++ b/src/satmkiso.c
@@ -0,0 +1,116 @@
+#include <getopt.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+static void usage(const char *progname)
+{
+	static const int width = 24;
+
+	printf("usage: %s -h\n", progname);
+	printf("       %s -bo[ABC] [files...]\n\n", progname);
+	printf("\t%-*s%s\n", width, "-h",                   "Show this help text");
+	printf("\t%-*s%s\n", width, "-b bootsector",        "Bootsector image");
+	printf("\t%-*s%s\n", width, "-o output",            "Output isofs file");
+	printf("\t%-*s%s\n", width, "-A abstract_path",     "Path in isofs of abstract file");
+	printf("\t%-*s%s\n", width, "-B bibliography_path", "Path in isofs of bibliography file");
+	printf("\t%-*s%s\n", width, "-C copyright_path",    "Path in isofs of copyright file");
+	printf("\t%-*s%s\n", width, "files...",             "Files to include in the isofs image");
+
+	printf("\nFor -A/B/C, the path must refer to a file which exists within the isofs image.\n");
+}
+
+#define FLAG_o (1U << 1)
+#define FLAG_b (1U << 3)
+#define FLAG_A (1U << 5)
+#define FLAG_B (1U << 7)
+#define FLAG_C (1U << 9)
+
+int main(int argc, char **argv)
+{
+	static const char *optpat   = "ho:b:A:B:C:";
+	static const char *fileopts = "boABC";
+	char *specialfiles[5] = {
+		0, 0, "DUMMY.TXT", "DUMMY.TXT", "DUMMY.TXT",
+	};
+
+#define SPECIALFILE(c) specialfiles[strchr(fileopts, (c)) - fileopts]
+
+	uint32_t seen = 0;
+	int opt, fail = 0;
+
+#define SEEN(flags) ((seen | (flags)) == seen)
+#define FLAG(c)     (1U << (strchr(optpat, (c)) - optpat))
+
+	extern char *optarg;
+	extern int optind, optopt;
+	while ((opt = getopt(argc, argv, optpat)) != -1) {
+		uint32_t flag = opt == '?' ? 0 : FLAG(opt);
+		if (SEEN(flag)) {
+			fprintf(stderr, "%s: Duplicate option -%c\n", argv[0], opt);
+			fail = 1;
+			continue;
+		}
+		seen |= flag;
+
+		switch (opt) {
+		case 'h':
+			usage(argv[0]);
+			return 0;
+
+		case 'b':
+		case 'o':
+		case 'A':
+		case 'B':
+		case 'C':
+			SPECIALFILE(opt) = optarg;
+			break;
+
+		default:
+			fail = 1;
+		}
+	}
+
+	for (const char *c = "bo"; *c; c++) {
+		if (!SEEN(FLAG(*c))) {
+			fprintf(stderr, "%s: Missing required option -%c\n", argv[0], *c);
+			fail = 1;
+		}
+	}
+	if (fail) {
+		fprintf(stderr, "\nUse -h option for help on correct usage.\n");
+		return 1;
+	}
+
+	char *argprefix[] = {
+		"mkisofs", "-graft-points", "-full-iso9660-filenames",
+		"-G",         SPECIALFILE('b'),
+		"-o",         SPECIALFILE('o'),
+		"-abstract",  SPECIALFILE('A'),
+		"-biblio",    SPECIALFILE('B'),
+		"-copyright", SPECIALFILE('C'),
+	};
+
+#define ARRAYLEN(a) (sizeof(a) / sizeof *(a))
+
+	char **args = calloc(argc - optind + ARRAYLEN(argprefix) + 2, sizeof *args);
+	if (!args) {
+		perror("malloc");
+		return 1;
+	}
+
+	size_t argidx;
+	for (argidx = 0; argidx < ARRAYLEN(argprefix); argidx++)
+		args[argidx] = argprefix[argidx];
+	while (argv[optind])
+		args[argidx++] = argv[optind++];
+	if (!SEEN(FLAG_A | FLAG_B | FLAG_C))
+		args[argidx++] = "DUMMY.TXT=/dev/null";
+
+	execvp("mkisofs", args);
+	perror("exec");
+	return 1;
+}
+
-- 
cgit v1.2.3