* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
-#include <sys/stat.h>
-#include <dirent.h>
-#include "gmqcc.h"
+#include "gmqcc.h"
+
+/*
+ * The PAK format uses a FOURCC concept for storing the magic ident within
+ * the header as a uint32_t.
+ */
+#define PAK_FOURCC ((uint32_t)(('P' | ('A' << 8) | ('C' << 16) | ('K' << 24))))
typedef struct {
uint32_t magic; /* "PACK" */
return beg;
}
-/*
- * Used to spawn a directory when creating the pak directory structure/
- * tree. Think of this as mkdir(path, 0700). We just cargo cult our
- * own because _mkdir on windows is "illegal" for Windows8 Certification
- * do to the requirement of SECURITY_ATTRIBUTES on everything.
- */
-static bool pak_tree_spawn(const char *path) {
-#ifdef _MSC_VER
- return CreateDirectoryA(path, NULL); /* non-zero on success */
-#else
- return !!(mkdir(path, 0700)); /* zero on success */
-#endif
-}
-
/*
* When given a string like "a/b/c/d/e/file"
* this function will handle the creation of
* the directory structure, included nested
* directories.
- */
+ */
static void pak_tree_build(const char *entry) {
char *directory;
char *elements[28];
char *pathsplit;
+ char *token;
size_t itr;
size_t jtr;
- pathsplit = mem_a(56);
- directory = mem_a(56);
+ pathsplit = (char *)mem_a(56);
+ directory = (char *)mem_a(56);
+
+ memset(pathsplit, 0, 56);
strncpy(directory, entry, 56);
- for (itr = 0; (elements[itr] = pak_tree_sep(&directory, "/")); itr++);
+ for (itr = 0; (token = pak_tree_sep(&directory, "/")) != NULL; itr++) {
+ elements[itr] = token;
+ }
+
for (jtr = 0; jtr < itr - 1; jtr++) {
strcat(pathsplit, elements[jtr]);
strcat(pathsplit, "/");
- pak_tree_spawn(pathsplit);
+ if (fs_dir_make(pathsplit)) {
+ mem_d(pathsplit);
+ mem_d(directory);
+
+ /* TODO: undo on fail */
+
+ return;
+ }
}
mem_d(pathsplit);
bool insert;
} pak_file_t;
-pak_file_t *pak_open_read(const char *file) {
+static pak_file_t *pak_open_read(const char *file) {
pak_file_t *pak;
size_t itr;
- if (!(pak = mem_a(sizeof(pak_file_t))))
+ if (!(pak = (pak_file_t*)mem_a(sizeof(pak_file_t))))
return NULL;
- if (!(pak->handle = file_open(file, "rb"))) {
+ if (!(pak->handle = fs_file_open(file, "rb"))) {
mem_d(pak);
return NULL;
}
pak->directories = NULL;
pak->insert = false; /* read doesn't allow insert */
- memset (&(pak->header), 0, sizeof(pak_header_t));
- file_read (&(pak->header), 1, sizeof(pak_header_t), pak->handle);
- util_endianswap(&(pak->header), 1, sizeof(pak_header_t));
+ memset (&pak->header, 0, sizeof(pak_header_t));
+ fs_file_read (&pak->header, sizeof(pak_header_t), 1, pak->handle);
+ util_endianswap(&pak->header, 1, sizeof(pak_header_t));
/*
- * Every PAK file has "PACK" stored as little endian data in the
+ * Every PAK file has "PACK" stored as FOURCC data in the
* header. If this data cannot compare (as checked here), it's
* probably not a PAK file.
- */
- if (!(memcmp(&(pak->header.magic), (const void*)"PACK", sizeof(uint32_t)))) {
- file_close(pak->handle);
- mem_d (pak);
+ */
+ if (pak->header.magic != PAK_FOURCC) {
+ fs_file_close(pak->handle);
+ mem_d (pak);
return NULL;
}
* Time to read in the directory handles and prepare the directories
* vector. We're going to be reading some the file inwards soon.
*/
- file_seek(pak->handle, pak->header.diroff, SEEK_SET);
+ fs_file_seek(pak->handle, pak->header.diroff, SEEK_SET);
/*
* Read in all directories from the PAK file. These are considered
*/
for (itr = 0; itr < pak->header.dirlen / 64; itr++) {
pak_directory_t dir;
- file_read (&dir, 1, sizeof(pak_directory_t), pak->handle);
+ fs_file_read (&dir, sizeof(pak_directory_t), 1, pak->handle);
util_endianswap(&dir, 1, sizeof(pak_directory_t));
vec_push(pak->directories, dir);
return pak;
}
-pak_file_t *pak_open_write(const char *file) {
+static pak_file_t *pak_open_write(const char *file) {
pak_file_t *pak;
- if (!(pak = mem_a(sizeof(pak_file_t))))
+ if (!(pak = (pak_file_t*)mem_a(sizeof(pak_file_t))))
return NULL;
/*
*/
pak_tree_build(file);
- if (!(pak->handle = file_open(file, "wb"))) {
+ if (!(pak->handle = fs_file_open(file, "wb"))) {
/*
* The directory tree that was created, needs to be
* removed entierly if we failed to open a file.
*/
/* TODO backup directory clean */
+ mem_d(pak);
return NULL;
}
* "patching" and writing the directories at the end of the
* file.
*/
- pak->insert = true;
+ pak->insert = true;
+ pak->header.magic = PAK_FOURCC;
- /*
- * A valid PAK file contains the magic "PACK" in it's header
- * stored in little endian format.
- */
- memcpy(&(pak->header.magic), (const void*)"PACK", sizeof(uint32_t));
+ /* on BE systems we need to swap the byte order of the FOURCC */
+ util_endianswap(&pak->header.magic, 1, sizeof(uint32_t));
/*
* We need to write out the header since files will be wrote out to
* will need to be patched in later with a file_seek, and overwrite,
* we could use offsets and other trickery. This is just easier.
*/
- file_write(&(pak->header), sizeof(pak_header_t), 1, pak->handle);
+ fs_file_write(&(pak->header), sizeof(pak_header_t), 1, pak->handle);
return pak;
}
+pak_file_t *pak_open(const char *file, const char *mode) {
+ if (!file || !mode)
+ return NULL;
+
+ switch (*mode) {
+ case 'r': return pak_open_read (file);
+ case 'w': return pak_open_write(file);
+ }
+
+ return NULL;
+}
+
bool pak_exists(pak_file_t *pak, const char *file, pak_directory_t **dir) {
size_t itr;
if (!pak || !file)
return false;
-
- /*
- * We could technically use a hashtable here. But I don't think
- * the lookup complexity is a performance concern. This may be
- * O(n) lookup. But meh?
- */
+
for (itr = 0; itr < vec_size(pak->directories); itr++) {
if (!strcmp(pak->directories[itr].name, file)) {
/*
* Now create the file, if this operation fails. Then abort
* It shouldn't fail though.
*/
- if (!(out = file_open(file, "wb"))) {
+ if (!(out = fs_file_open(file, "wb"))) {
mem_d(dat);
return false;
}
/* read */
- file_seek (pak->handle, dir->pos, SEEK_SET);
- file_read (dat, 1, dir->len, pak->handle);
+ fs_file_seek (pak->handle, dir->pos, SEEK_SET);
+ fs_file_read (dat, 1, dir->len, pak->handle);
/* write */
- file_write(dat, 1, dir->len, out);
+ fs_file_write(dat, 1, dir->len, out);
/* close */
- file_close(out);
+ fs_file_close(out);
/* free */
mem_d(dat);
return true;
}
-bool pak_extract_all(pak_file_t *pak) {
+bool pak_extract_all(pak_file_t *pak, const char *dir) {
size_t itr;
+
+ if (!fs_dir_make(dir))
+ return false;
+
+ if (fs_dir_change(dir))
+ return false;
+
for (itr = 0; itr < vec_size(pak->directories); itr++) {
if (!pak_extract_one(pak, pak->directories[itr].name))
return false;
if (!pak || !file || !pak->insert || pak_exists(pak, file, NULL))
return false;
- if (!(fp = fopen(file, "rb")))
+ if (!(fp = fs_file_open(file, "rb")))
return false;
/*
* the directory entry, and the actual contents of the file
* to the PAK file itself.
*/
- file_seek(fp, 0, SEEK_END);
- dir.len = ftell(fp);
- file_seek(fp, 0, SEEK_SET);
+ fs_file_seek(fp, 0, SEEK_END);
+ dir.len = fs_file_tell(fp);
+ fs_file_seek(fp, 0, SEEK_SET);
- dir.pos = ftell(pak->handle);
+ dir.pos = fs_file_tell(pak->handle);
+
+ /*
+ * We're limited to 56 bytes for a file name string, that INCLUDES
+ * the directory and '/' seperators.
+ */
+ if (strlen(file) >= 56) {
+ fs_file_close(fp);
+ return false;
+ }
+
+ strncpy(dir.name, file, strlen(file));
/*
* Allocate some memory for loading in the data that will be
* redirected into the PAK file.
*/
if (!(dat = (unsigned char *)mem_a(dir.len))) {
- file_close(fp);
+ fs_file_close(fp);
return false;
}
- file_read (dat, dir.len, 1, fp);
- file_close(fp);
- file_write(dat, dir.len, 1, pak->handle);
+ fs_file_read (dat, dir.len, 1, fp);
+ fs_file_close(fp);
+ fs_file_write(dat, dir.len, 1, pak->handle);
+
+ /*
+ * Now add the directory to the directories vector, so pak_close
+ * can actually write it.
+ */
+ vec_push(pak->directories, dir);
return true;
}
if (!(pak->insert))
return false;
- if (!(dp = opendir(dir)))
+ if (!(dp = fs_dir_open(dir)))
return false;
- while ((dirp = readdir(dp))) {
+ while ((dirp = fs_dir_read(dp))) {
if (!(pak_insert_one(pak, dirp->d_name))) {
- closedir(dp);
+ fs_dir_close(dp);
return false;
}
}
- closedir(dp);
+ fs_dir_close(dp);
return true;
}
pak->header.diroff = ftell(pak->handle);
/* patch header */
- file_seek (pak->handle, 0, SEEK_SET);
- file_write(&(pak->header), sizeof(pak_header_t), 1, pak->handle);
+ fs_file_seek (pak->handle, 0, SEEK_SET);
+ fs_file_write(&(pak->header), sizeof(pak_header_t), 1, pak->handle);
/* write directories */
- file_seek (pak->handle, pak->header.diroff, SEEK_SET);
+ fs_file_seek (pak->handle, pak->header.diroff, SEEK_SET);
for (itr = 0; itr < vec_size(pak->directories); itr++) {
- file_write(&(pak->directories[itr]), sizeof(pak_directory_t), 1, pak->handle);
+ fs_file_write(&(pak->directories[itr]), sizeof(pak_directory_t), 1, pak->handle);
}
}
- vec_free (pak->directories);
- file_close(pak->handle);
- mem_d (pak);
+ vec_free (pak->directories);
+ fs_file_close(pak->handle);
+ mem_d (pak);
return true;
}
+
+/*
+ * Fancy GCC-like LONG parsing allows things like --opt=param with
+ * assignment operator. This is used for redirecting stdout/stderr
+ * console to specific files of your choice.
+ */
+static bool parsecmd(const char *optname, int *argc_, char ***argv_, char **out, int ds, bool split) {
+ int argc = *argc_;
+ char **argv = *argv_;
+
+ size_t len = strlen(optname);
+
+ if (strncmp(argv[0]+ds, optname, len))
+ return false;
+
+ /* it's --optname, check how the parameter is supplied */
+ if (argv[0][ds+len] == '=') {
+ *out = argv[0]+ds+len+1;
+ return true;
+ }
+
+ if (!split || argc < ds) /* no parameter was provided, or only single-arg form accepted */
+ return false;
+
+ /* using --opt param */
+ *out = argv[1];
+ --*argc_;
+ ++*argv_;
+ return true;
+}
+
+int main(int argc, char **argv) {
+ bool extract = true;
+ char *redirout = (char*)stdout;
+ char *redirerr = (char*)stderr;
+ char *directory = NULL;
+ char *file = NULL;
+ char **files = NULL;
+ pak_file_t *pak = NULL;
+ size_t iter = 0;
+
+ con_init();
+
+ /*
+ * Command line option parsing commences now We only need to support
+ * a few things in the test suite.
+ */
+ while (argc > 1) {
+ ++argv;
+ --argc;
+
+ if (argv[0][0] == '-') {
+ if (parsecmd("redirout", &argc, &argv, &redirout, 1, false))
+ continue;
+ if (parsecmd("redirerr", &argc, &argv, &redirerr, 1, false))
+ continue;
+ if (parsecmd("directory", &argc, &argv, &directory, 1, false))
+ continue;
+ if (parsecmd("file", &argc, &argv, &file, 1, false))
+ continue;
+
+ con_change(redirout, redirerr);
+
+ switch (argv[0][1]) {
+ case 'e': extract = true; continue;
+ case 'c': extract = false; continue;
+ }
+
+ if (!strcmp(argv[0]+1, "debug")) {
+ OPTS_OPTION_BOOL(OPTION_DEBUG) = true;
+ continue;
+ }
+ if (!strcmp(argv[0]+1, "memchk")) {
+ OPTS_OPTION_BOOL(OPTION_MEMCHK) = true;
+ continue;
+ }
+ if (!strcmp(argv[0]+1, "nocolor")) {
+ con_color(0);
+ continue;
+ }
+ }
+
+ vec_push(files, argv[0]);
+ }
+ con_change(redirout, redirerr);
+
+
+ if (!file) {
+ con_err("-file must be specified for output/input PAK file\n");
+ vec_free(files);
+ return EXIT_FAILURE;
+ }
+
+ if (extract) {
+ if (!(pak = pak_open(file, "r"))) {
+ con_err("failed to open PAK file %s\n", file);
+ vec_free(files);
+ return EXIT_FAILURE;
+ }
+
+ if (!pak_extract_all(pak, (directory) ? directory : "./")) {
+ con_err("failed to extract PAK %s (files may be missing)\n", file);
+ pak_close(pak);
+ vec_free(files);
+ return EXIT_FAILURE;
+ }
+
+ /* not possible */
+ pak_close(pak);
+ vec_free(files);
+ util_meminfo();
+ return EXIT_SUCCESS;
+ }
+
+ if (!(pak = pak_open(file, "w"))) {
+ con_err("failed to open PAK %s for writing\n", file);
+ vec_free(files);
+ return EXIT_FAILURE;
+ }
+
+ if (directory && !fs_dir_change(directory)) {
+ con_err("failed to change directory %s\n", directory);
+ pak_close(pak);
+ vec_free(files);
+ return EXIT_FAILURE;
+ }
+
+ for (iter = 0; iter < vec_size(files); iter++) {
+ if (!(pak_insert_one(pak, files[iter]))) {
+ con_err("failed inserting %s for PAK %s\n", files[iter], file);
+ pak_close(pak);
+ vec_free(files);
+ return EXIT_FAILURE;
+ }
+ }
+
+ /* not possible */
+ pak_close(pak);
+ vec_free(files);
+
+
+ util_meminfo();
+ return EXIT_SUCCESS;
+}