reader implementation
--HG-- extra : convert_revision : svn%3A39bc706e-5318-0410-9160-8a85361fbb7c/trunk%4046
This commit is contained in:
Borja Ferrer
parent
0ac9fcea2d
commit
a97a3cddb9
211
sourcepawn/vm/sp_reader.c
Normal file
211
sourcepawn/vm/sp_reader.c
Normal file
@ -0,0 +1,211 @@
|
|||||||
|
#include <malloc.h>
|
||||||
|
#include <string.h>
|
||||||
|
#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;
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user