Some string extractor prog I made some time ago.

git-svn-id: svn://svn.code.sf.net/p/sc2/code/trunk@722 8092fc87-c524-0410-9efc-e669fe64eaf9
2003-02-10 03:27:23 +00:00
parent 5ad1e0a3e7
commit 9dc34826bd
3 changed files with 245 additions and 0 deletions
@@ -0,0 +1,9 @@
+unstr: unstr.c unstr.h
+	gcc -W -Wall -g -O0 unstr.c -o unstr -lpng -lz -lm
+
+clean:
+	rm unstr unstr.tgz
+
+tgz: unstr.c unstr.h Makefile
+	tar -cvzf unstr.tgz unstr.c unstr.h Makefile
+
@@ -0,0 +1,212 @@
+/*
+ * .str file extractor
+ * By Serge van den Boom (svdb@stack.nl), 20021027
+ * The GPL applies.
+ *
+ * 
+ */
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <sys/mman.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/param.h>
+#include <string.h>
+#include <errno.h>
+
+#include "unstr.h"
+
+#include <png.h>
+
+
+struct options {
+	char *infile;
+	char *outfile;
+	char list;
+	char print;
+	char verbose;
+};
+
+
+index_header *readIndex(const uint8 *buf);
+void printIndex(const index_header *h, const uint8 *buf, FILE *out);
+void writeFiles(const index_header *h, const uint8 *data, const char *path,
+		const char *prefix);
+void parse_arguments(int argc, char *argv[], struct options *opts);
+void listEntries(const index_header *h, const uint8 *buf, FILE *out);
+void writeEntries(const index_header *h, const uint8 *buf, FILE *out);
+
+
+int
+main(int argc, char *argv[]) {
+	int in;
+	struct stat sb;
+	uint8 *buf;
+	index_header *h;
+	struct options opts;
+
+	parse_arguments(argc, argv, &opts);
+
+	in = open(opts.infile, O_RDONLY);
+	if (in == -1) {
+		fprintf(stderr, "Error: Could not open file %s: %s\n", opts.infile,
+				strerror(errno));
+		return EXIT_FAILURE;
+	}
+	
+	if (fstat(in, &sb) == -1) {
+		perror("stat() failed");
+		return EXIT_FAILURE;
+	}
+	
+	buf = mmap(NULL, sb.st_size, PROT_READ, MAP_SHARED, in, 0);
+	if (buf == MAP_FAILED) {
+		perror("mmap() failed");
+		return EXIT_FAILURE;
+	}
+	close(in);
+
+	h = readIndex(buf);
+	if (opts.print)
+		printIndex(h, buf, stdout);
+
+	if (opts.list) {
+		listEntries(h, buf, stdout);
+	}
+	
+	if (opts.outfile) {
+		FILE *out;
+		out = fopen(opts.outfile, "wb+");
+		if (out == NULL) {
+			fprintf(stderr, "Could not open output file '%s': %s\n",
+					opts.outfile, strerror(errno));
+			exit(EXIT_FAILURE);
+		}
+		writeEntries(h, buf, out);
+		fclose(out);
+	}
+
+	// freeIndex(h);
+	munmap(buf, sb.st_size);
+	return EXIT_SUCCESS;
+}
+
+void
+usage() {
+	fprintf(stderr,
+			"unstr -a <infile>\n"
+			"unstr -p <infile>\n"
+			"unstr [-o <outfile>] <infile>\n");
+}
+
+void
+parse_arguments(int argc, char *argv[], struct options *opts) {
+	char ch;
+	
+	memset(opts, '\0', sizeof (struct options));
+	while (1) {
+		ch = getopt(argc, argv, "ahlo:pv");
+		if (ch == -1)
+			break;
+		switch(ch) {
+			case 'l':
+				opts->list = 1;
+				break;
+			case 'o':
+				opts->outfile = optarg;
+				break;
+			case 'p':
+				opts->print = 1;
+				break;
+			case 'v':
+				opts->verbose = 1;
+				break;
+			case '?':
+			case 'h':
+			default:
+				usage();
+				exit(EXIT_FAILURE);
+		}
+	}
+	argc -= optind;
+	argv += optind;
+	if (argc != 1) {
+		usage();
+		exit(EXIT_FAILURE);
+	}
+	opts->infile = argv[0];
+}
+
+index_header *
+readIndex(const uint8 *buf) {
+	index_header *h;
+	const uint8 *sizeptr;
+	uint32 size, totsize;
+	uint32 i;
+
+	if ((uint32) MAKE_DWORD(buf[3], buf[2], buf[1], buf[0]) != 0xffffffff ||
+			MAKE_WORD(buf[5], buf[4]) != 0x0000) {
+		fprintf(stderr, "File is not a valid STRTAB file.\n");
+		exit(EXIT_FAILURE);
+	}
+
+	h = malloc(sizeof (index_header));
+	h->num_entries = MAKE_WORD(buf[7], buf[6]);
+	h->entries = malloc(h->num_entries * sizeof (entry_desc));
+
+	sizeptr = buf + 8;
+	if (MAKE_DWORD(sizeptr[3], sizeptr[2], sizeptr[1], sizeptr[0])) {
+		fprintf(stderr, "WARNING: Placeholder string entry hasn't got "
+				"size 0, but %d.\n",
+				MAKE_DWORD(sizeptr[3], sizeptr[2], sizeptr[1], sizeptr[0]));
+	}
+	sizeptr += 4;
+
+	totsize = 0;
+	for (i = 0; i < h->num_entries; i++) {
+		size = MAKE_DWORD(sizeptr[3], sizeptr[2], sizeptr[1], sizeptr[0]);
+		h->entries[i].size = size;
+		h->entries[i].offset = totsize;
+		sizeptr += 4;
+		totsize += size;
+	}
+	return h;
+}
+
+void
+printIndex(const index_header *h, const uint8 *buf, FILE *out) {
+	fprintf(stderr, "-p option not implemented yet.\n");
+	(void) h;
+	(void) buf;
+	(void) out;
+}
+
+void
+listEntries(const index_header *h, const uint8 *buf, FILE *out) {
+	fprintf(stderr, "-l option not implemented yet.\n");
+	(void) h;
+	(void) buf;
+	(void) out;
+}
+
+void
+writeEntries(const index_header *h, const uint8 *buf, FILE *out) {
+	uint32 i;
+	char *str;
+	const uint8 *start;
+
+	str = NULL;
+	start = buf + 8 + (h->num_entries + 1) * 4;
+	for (i = 0; i < h->num_entries; i++) {
+		fprintf(out, "String #%d:\n", i + 1);
+		str = realloc(str, h->entries[i].size + 1);
+		str[h->entries[i].size] = '\0';
+		strncpy(str, start + h->entries[i].offset, h->entries[i].size);
+		fprintf(out, "%s\n\n", str);
+	}
+}
+
+
@@ -0,0 +1,24 @@
+#include <stdint.h>
+
+typedef unsigned char uint8;
+typedef unsigned short uint16;
+typedef unsigned int uint32;
+typedef short int16;
+typedef uint8 bool;
+#define TRUE   1
+#define FALSE  0
+
+#define MAKE_WORD(x1, x2) (((x2) << 8) | (x1))
+#define MAKE_DWORD(x1, x2, x3, x4) (((x4) << 24) | ((x3) << 16) | \
+		((x2) << 8) | (x1))
+
+typedef struct {
+	uint32 size;
+	uint32 offset;  // offset from beginning of strings
+} entry_desc;
+
+typedef struct {
+	uint16 num_entries;
+	entry_desc *entries;
+} index_header;
+