From a97a3cddb9ca46b520cbd88dda3cba04cfff3777 Mon Sep 17 00:00:00 2001 From: Borja Ferrer Date: Tue, 1 Aug 2006 00:21:00 +0000 Subject: [PATCH] reader implementation --HG-- extra : convert_revision : svn%3A39bc706e-5318-0410-9160-8a85361fbb7c/trunk%4046 --- sourcepawn/vm/sp_reader.c | 211 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 211 insertions(+) create mode 100644 sourcepawn/vm/sp_reader.c diff --git a/sourcepawn/vm/sp_reader.c b/sourcepawn/vm/sp_reader.c new file mode 100644 index 00000000..f61b63d6 --- /dev/null +++ b/sourcepawn/vm/sp_reader.c @@ -0,0 +1,211 @@ +#include +#include +#include "sp_vm.h" +#include "zlib/zlib.h" + +sp_plugin_t *_ReadPlugin(sp_file_hdr_t *hdr, uint8_t *base, sp_plugin_t *plugin, int *err) +{ + char *nameptr; + uint8_t sectnum = 0; + + sp_plugin_infotab_t *info = (sp_plugin_infotab_t *)malloc(sizeof(sp_plugin_infotab_t)); + sp_plugin_debug_t *dbg = (sp_plugin_debug_t *)malloc(sizeof(sp_plugin_debug_t)); + + sp_file_section_t *secptr = (sp_file_section_t *)(base + sizeof(sp_file_hdr_t)); + + memset(plugin, 0, sizeof(sp_plugin_t)); + memset(info, 0, sizeof(sp_plugin_infotab_t)); + memset(dbg, 0, sizeof(sp_plugin_debug_t)); + + plugin->base = base; + plugin->info = info; + plugin->debug = dbg; + + while (sectnum < hdr->sections) + { + nameptr = base + hdr->stringtab + secptr->nameoffs; + + if (!(plugin->pcode) && !strcmp(nameptr, ".code")) + { + sp_file_code_t *cod = (sp_file_code_t *)(base + secptr->dataoffs); + plugin->pcode = base + secptr->dataoffs + cod->code; + plugin->pcode_size = cod->codesize; + plugin->flags = cod->flags; + } + else if (!(plugin->data) && !strcmp(nameptr, ".data")) + { + sp_file_data_t *dat = (sp_file_data_t *)(base + secptr->dataoffs); + plugin->data = base + secptr->dataoffs + dat->data; + plugin->data_size = dat->datasize; + plugin->memory = dat->memsize; + } + else if (!(plugin->info->stringbase) && !strcmp(nameptr, ".names")) + { + plugin->info->stringbase = base + secptr->dataoffs; + } + else if (!(plugin->info->publics) && !strcmp(nameptr, ".publics")) + { + plugin->info->publics_num = secptr->size / sizeof(sp_file_publics_t); + plugin->info->publics = (sp_file_publics_t *)(base + secptr->dataoffs); + } + else if (!(plugin->info->pubvars) && !strcmp(nameptr, ".pubvars")) + { + plugin->info->pubvars_num = secptr->size / sizeof(sp_file_pubvars_t); + plugin->info->pubvars = (sp_file_pubvars_t *)(base + secptr->dataoffs); + } + else if (!(plugin->info->natives) && !strcmp(nameptr, ".natives")) + { + plugin->info->natives_num = secptr->size / sizeof(sp_file_natives_t); + plugin->info->natives = (sp_file_natives_t *)(base + secptr->dataoffs); + } + else if (!(plugin->debug->files) && !strcmp(nameptr, ".dbg.files")) + { + plugin->debug->files_num = secptr->size / sizeof(sp_fdbg_file_t); + plugin->debug->files = (sp_fdbg_file_t *)(base + secptr->dataoffs); + } + else if (!(plugin->debug->stringbase) && !strcmp(nameptr, ".dbg.strings")) + { + plugin->debug->stringbase = base + secptr->dataoffs; + } + else if (!(plugin->debug->lines) && !strcmp(nameptr, ".dbg.lines")) + { + plugin->debug->lines_num = secptr->size / sizeof(sp_fdbg_line_t); + plugin->debug->lines = (sp_fdbg_line_t *)(base + secptr->dataoffs); + } + else if (!(plugin->debug->symbols) && !strcmp(nameptr, ".dbg.symbols")) + { + plugin->debug->syms_num = secptr->size / sizeof(sp_fdbg_symbol_t); + plugin->debug->symbols = (sp_fdbg_symbol_t *)(base + secptr->dataoffs); + } + secptr++; + sectnum++; + } + + if (!(plugin->pcode) || !(plugin->data) || !(plugin->info->stringbase)) + goto return_error; + + if ((plugin->flags == SP_FILE_DEBUG) && (!(plugin->debug->files) || !(plugin->debug->lines) || !(plugin->debug->symbols))) + goto return_error; + + return plugin; + +return_error: + free(dbg); + free(info); + + if (err) + *err = SP_ERR_FILE_FORMAT; + + return NULL; +} + +sp_plugin_t *SP_LoadFromFilePointer(FILE *fp, int *err) +{ + sp_file_hdr_t hdr; + sp_plugin_t *plugin; + uint8_t *base; + int z_result; + int error; + + if (!fp) + { + error = SP_ERR_NOT_FOUND; + goto return_error; + } + + /* Rewind for safety */ + rewind(fp); + fread(&hdr, sizeof(sp_file_hdr_t), 1, fp); + + if (hdr.magic != SPFILE_MAGIC) + { + error = SP_ERR_FILE_FORMAT; + goto return_error; + } + + switch (hdr.compression) + { + case SPFILE_COMPRESSION_GZ: + { + uint32_t uncompsize = hdr.imagesize - hdr.dataoffs; + uint32_t compsize = hdr.disksize - hdr.dataoffs; + uint32_t sectsize = hdr.dataoffs - sizeof(sp_file_hdr_t); + uLongf destlen = uncompsize; + + char *tempbuf = (char *)malloc(compsize); + void *uncompdata = malloc(uncompsize); + void *sectheader = malloc(sectsize); + + fread(sectheader, sectsize, 1, fp); + fread(tempbuf, compsize, 1, fp); + + z_result = uncompress((Bytef *)uncompdata, &destlen, (Bytef *)tempbuf, compsize); + free(tempbuf); + if (z_result != Z_OK) + { + free(sectheader); + free(uncompdata); + error = SP_ERR_DECOMPRESSOR; + goto return_error; + } + + base = (uint8_t *)malloc(hdr.imagesize); + memcpy(base, &hdr, sizeof(sp_file_hdr_t)); + memcpy(base + sizeof(sp_file_hdr_t), sectheader, sectsize); + free(sectheader); + memcpy(base + hdr.dataoffs, uncompdata, uncompsize); + free(uncompdata); + break; + } + case SPFILE_COMPRESSION_NONE: + { + base = (uint8_t *)malloc(hdr.imagesize); + rewind(fp); + fread(base, hdr.imagesize, 1, fp); + break; + } + default: + { + error = SP_ERR_DECOMPRESSOR; + goto return_error; + } + } + + plugin = (sp_plugin_t *)malloc(sizeof(sp_plugin_t)); + if (!_ReadPlugin(&hdr, base, plugin, err)) + { + free(plugin); + free(base); + return NULL; + } + + return plugin; + +return_error: + if (err) + *err = error; + return NULL; +} + +sp_plugin_t *SP_LoadFromMemory(void *base, sp_plugin_t *plugin, int *err) +{ + sp_file_hdr_t hdr; + uint8_t noptr = 0; + + memcpy(&hdr, base, sizeof(sp_file_hdr_t)); + + if (!plugin) + { + plugin = (sp_plugin_t *)malloc(sizeof(sp_plugin_t)); + noptr = 1; + } + + if (!_ReadPlugin(&hdr, base, plugin, err)) + { + if (noptr) + free(plugin); + return NULL; + } + + return plugin; +}