From a7236b6b2a00a1b1b1047b352219d8af39037cde Mon Sep 17 00:00:00 2001
From: Alex 'AdUser' Z <ad_user@runbox.com>
Date: Thu, 19 Mar 2015 15:41:37 +1000
Subject: [PATCH] + imdb_1to2

---
 src/Makefile    |  5 ++-
 src/imdb_1to2.c | 90 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 94 insertions(+), 1 deletion(-)
 create mode 100644 src/imdb_1to2.c

diff --git a/src/Makefile b/src/Makefile
index 8ad8316..096c519 100644
--- a/src/Makefile
+++ b/src/Makefile
@@ -2,7 +2,7 @@ CFLAGS=-Wall -Wextra -O2 -g -std=c99 -pedantic -fstack-protector --param=ssp-buf
 CPPFLAGS=-D_FORTIFY_SOURCE=2
 LDFLAGS=-Wl,-soname,libimgdup.so.1 -fPIC
 
-all: lib util
+all: lib util imdb_1to2
 
 %.o: %.c
 	gcc $(CFLAGS) $(CPPFLAGS) $(LDFLAGS) -c $<
@@ -15,6 +15,9 @@ util: util.c
 	gcc $(CFLAGS) $(CPPFLAGS) -L. $^ -limgdup    -o $@
 	gcc $(CFLAGS) $(CPPFLAGS) -L. $^ libimgdup.a -o $@-static -static
 
+imdb_1to2: imdb_1to2.c
+	gcc $(CFLAGS) $(CPPFLAGS) $(LDFLAGS) -o $@ $^
+
 clean:
 	rm -f *.o
 	rm -f util
diff --git a/src/imdb_1to2.c b/src/imdb_1to2.c
new file mode 100644
index 0000000..1d59b03
--- /dev/null
+++ b/src/imdb_1to2.c
@@ -0,0 +1,90 @@
+#include <errno.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <unistd.h>
+
+#define IMDB_REC_LEN 48
+#define BLK_SIZE 1000
+
+void usage(const char *message) {
+  if (message)
+    printf("error: %s\n", message);
+  printf("Usage: imdb_convert_1to2 <infile> <outfile>\n");
+  exit(EXIT_FAILURE);
+}
+
+int main(int argc, char **argv) {
+  int in, out;
+  long unsigned int rec_total, rec_first, rec_last, records;
+  unsigned char  in_buf[IMDB_REC_LEN * BLK_SIZE];
+  unsigned char out_buf[IMDB_REC_LEN * BLK_SIZE];
+  unsigned char  header[IMDB_REC_LEN];
+  unsigned char *src, *dst;
+  struct stat st;
+  ssize_t bytes;
+
+  memset(header, 0x0, IMDB_REC_LEN);
+  snprintf((char *) header, IMDB_REC_LEN, "IMDB v%02u, CAPS: %s;", 2, "M-R");
+
+  if (argc < 2)
+    usage(NULL);
+
+  errno = 0;
+  if ((in  = open(argv[1], O_RDONLY)) < 0)
+    usage(strerror(errno));
+
+  if ((out = open(argv[2], O_WRONLY | O_TRUNC | O_CREAT, 0644)) < 0)
+    usage(strerror(errno));
+
+  if (fstat(in, &st) != 0)
+    usage(strerror(errno));
+
+  if ((st.st_size % IMDB_REC_LEN) != 0)
+    usage("database size expected to be multiples to 48");
+
+  rec_total = (st.st_size / IMDB_REC_LEN);
+  printf("Processing %lu records\n", rec_total - 1);
+
+  if (lseek(in, IMDB_REC_LEN, SEEK_SET) < 0)
+    usage(strerror(errno));
+
+  if (lseek(out, 0, SEEK_SET) < 0)
+    usage(strerror(errno));
+  if (write(out, header, IMDB_REC_LEN) != IMDB_REC_LEN)
+    usage(strerror(errno));
+
+  for (unsigned int block = 0; block <= (rec_total / BLK_SIZE); block++) {
+    rec_first = rec_last = (block * BLK_SIZE) + 1;
+    records   = ((rec_first + BLK_SIZE) > rec_total) ? rec_total - rec_first : BLK_SIZE;
+    rec_last  = rec_first + records;
+    printf("* block %u, %4lu records [%lu, %lu]\n",
+      block + 1, records, rec_first, rec_last - 1);
+    lseek(in,  rec_first * IMDB_REC_LEN, SEEK_SET);
+    lseek(out, rec_first * IMDB_REC_LEN, SEEK_SET);
+    bytes = read(in, in_buf, IMDB_REC_LEN * records);
+    if (bytes != IMDB_REC_LEN * records) {
+      printf("Read size mismatch, expected %lu, got: %i\n",
+        IMDB_REC_LEN * records, bytes);
+      exit(EXIT_FAILURE);
+    }
+    memset(out_buf, 0x0, IMDB_REC_LEN * records);
+    for (int i = 0; i < records; i++) {
+      src =  &in_buf[i * IMDB_REC_LEN];
+      dst = &out_buf[i * IMDB_REC_LEN];
+      memcpy(dst +  0, src + 0, sizeof(char) *  1); // usage flag
+      memcpy(dst + 16, src + 2, sizeof(char) *  32); // image bitmap
+    }
+    bytes = write(out, out_buf, IMDB_REC_LEN * records);
+    if (bytes != IMDB_REC_LEN * records) {
+      printf("Write size mismatch, expected %lu, got: %i\n",
+        IMDB_REC_LEN * records, bytes);
+      exit(EXIT_FAILURE); 
+    }
+  }
+
+  exit(EXIT_SUCCESS);
+}