X-Git-Url: http://git.draconx.ca/gitweb/liblbx.git/blobdiff_plain/8456fd3535202bc2b79df4ddb794cbfd075054fd..3192589d7c45cc7adeeb7c86c20a009cb4d16934:/src/lbx.c

diff --git a/src/lbx.c b/src/lbx.c
index d3815a9..05c3cbc 100644
--- a/src/lbx.c
+++ b/src/lbx.c
@@ -1,3 +1,21 @@
+/*
+ *  2ooM: The Master of Orion II Reverse Engineering Project
+ *  Library for working with LBX archive files.
+ *  Copyright (C) 2006-2008 Nick Bowler
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
 #ifdef HAVE_CONFIG_H
 #	include "config.h"
 #endif
@@ -12,11 +30,12 @@
 #include <sys/stat.h>
 #include <sys/mman.h>
 
-#include "byteorder.h"
+#include "pack.h"
 #include "misc.h"
 #include "lbx.h"
 
-#define LBX_MAGIC 0x0000fead
+#define LBX_MAGIC    0x0000fead
+#define LBX_HDR_SIZE 8
 
 int lbx_errno = 0;
 
@@ -32,52 +51,65 @@ struct lbx_state {
 	uint32_t offsets[];
 };
 
-struct lbx_state *lbx_fopen(FILE *f, const char *name)
+static struct lbx_state *lbx_init(unsigned char hdr[static LBX_HDR_SIZE])
 {
-	struct lbx_state *new = NULL;
-	uint16_t nfiles, version;
-	uint32_t magic;
-
-	if (fread(&nfiles,  sizeof nfiles,  1, f) != 1) goto readerr;
-	if (fread(&magic,   sizeof magic,   1, f) != 1) goto readerr;
-	if (fread(&version, sizeof version, 1, f) != 1) goto readerr;
-
-	nfiles  = letohs(nfiles);
-	magic   = letohl(magic);
-	version = letohs(version);
+	unsigned short nfiles  = unpack_16_le(hdr+0);
+	unsigned long  magic   = unpack_32_le(hdr+2);
+	unsigned short version = unpack_16_le(hdr+6);
+	struct lbx_state *lbx;
 
 	if (magic != LBX_MAGIC) {
-		lbx_errno = LBX_EMAGIC;
+		lbx_errno = -LBX_EMAGIC;
 		return NULL;
 	}
-	
-	new = malloc(sizeof *new + (nfiles+1)*(sizeof *new->offsets));
-	if (!new) {
+
+	lbx = malloc(sizeof *lbx + sizeof lbx->offsets[0] * (nfiles+1));
+	if (!lbx) {
 		lbx_errno = -errno;
 		return NULL;
 	}
-	
-	*new = (struct lbx_state){
-		.name   = name,
+
+	*lbx = (struct lbx_state) {
 		.nfiles = nfiles,
-		.f      = f,
-		.foff   = sizeof nfiles + sizeof magic + sizeof version,
 	};
-	
-	if (fread(new->offsets, sizeof *new->offsets, nfiles+1, f) != nfiles+1)
-		goto readerr;
-	new->foff += sizeof *new->offsets * (nfiles+1);
 
-	return new;
-readerr:
-	if (feof(f)) {
-		lbx_errno = LBX_EEOF;
-	} else {
+	return lbx;
+}
+
+struct lbx_state *lbx_fopen(FILE *f, const char *name)
+{
+	unsigned char hdr_buf[LBX_HDR_SIZE];
+	struct lbx_state *lbx;
+
+	if (fread(hdr_buf, 1, sizeof hdr_buf, f) != sizeof hdr_buf) {
 		lbx_errno = -errno;
+		return NULL;
 	}
 
-	free(new);
-	return NULL;
+	lbx = lbx_init(hdr_buf);
+	if (!lbx)
+		return NULL;
+
+	lbx->name = name;
+	lbx->f    = f;
+	lbx->foff = sizeof hdr_buf;
+
+	for (unsigned i = 0; i <= lbx->nfiles; i++) {
+		unsigned char buf[4];
+
+		if (fread(buf, 1, sizeof buf, f) != sizeof buf) {
+			lbx_errno = -errno;
+			if (feof(f))
+				lbx_errno = LBX_EEOF;
+			free(lbx);
+			return NULL;
+		}
+
+		lbx->offsets[i] = unpack_32_le(buf);
+		lbx->foff += sizeof buf;
+	}
+
+	return lbx;
 }
 
 static int _lbx_memcpy(void *dest, struct lbx_state *src, size_t size)
@@ -133,21 +165,6 @@ eof:
 	return NULL;
 }
 
-struct lbx_state *lbx_open(const char *path)
-{
-	struct lbx_state *new = NULL;
-	FILE *f;
-	
-	if ((f = fopen(path, "rb"))) {
-		const char *name = strrchr(path, '/');
-		new = lbx_fopen(f, name ? name+1 : path);
-	} else {
-		lbx_errno = -errno;
-	}
-
-	return new;
-}
-
 size_t lbx_numfiles(struct lbx_state *lbx)
 {
 	return lbx->nfiles;
@@ -163,7 +180,7 @@ int lbx_stat(struct lbx_state *lbx, size_t index, struct lbx_statbuf *buf)
 		return -1;
 	}
 
-	snprintf(str, sizeof str, "%s.%03d", lbx->name, index);
+	snprintf(str, sizeof str, "%s.%03zu", lbx->name, index);
 	buf->name = str;
 	buf->size = lbx->offsets[index+1] - lbx->offsets[index];
 	return 0;
@@ -292,6 +309,7 @@ const char *lbx_strerror(void)
 	case LBX_EMAGIC:   return "Bad magic number";
 	case LBX_EEOF:     return "Unexpected end-of-file";
 	case LBX_ERANGE:   return "Index out of range";
+	case LBX_EFORMAT:  return "Invalid file format";
 	}
 
 	return "Unknown error";