dvander@67: #include <malloc.h>
dvander@67: #include <string.h>
dvander@67: #include "sp_file_headers.h"
dvander@67: #include "sp_vm_types.h"
dvander@67: #include "sp_vm_engine.h"
dvander@67: #include "zlib/zlib.h"
dvander@67: #include "sp_vm_basecontext.h"
dvander@67: 
dvander@93: #if defined WIN32
dvander@93:  #define WIN32_LEAN_AND_MEAN
dvander@93:  #include <windows.h>
dvander@93: #else if defined __GNUC__
dvander@93:  #include <sys/mman.h>
dvander@93: #endif
dvander@93: 
dvander@67: using namespace SourcePawn;
dvander@67: 
dvander@93: void *SourcePawnEngine::ExecAlloc(size_t size)
dvander@93: {
dvander@93: #if defined WIN32
dvander@93: 	return VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
dvander@93: #else if defined __GNUC__
dvander@93: 	void *base = memalign(sysconf(_SC_PAGESIZE), size);
dvander@93: 	if (mprotect(base, size, PROT_READ|PROT_WRITE|PROT_EXEC) != 0)
dvander@93: 	{
dvander@93: 		free(base);
dvander@93: 		return NULL;
dvander@93: 	}
dvander@93: 	return base;
dvander@93: #endif
dvander@93: }
dvander@93: 
dvander@93: void SourcePawnEngine::ExecFree(void *address)
dvander@93: {
dvander@93: #if defined WIN32
dvander@93: 	VirtualFree(address, 0, MEM_RELEASE);
dvander@93: #else if defined __GNUC__
dvander@93: 	free(address);
dvander@93: #endif
dvander@93: }
dvander@93: 
dvander@67: void *SourcePawnEngine::BaseAlloc(size_t size)
dvander@67: {
dvander@67: 	return malloc(size);
dvander@67: }
dvander@67: 
dvander@67: void SourcePawnEngine::BaseFree(void *memory)
dvander@67: {
dvander@67: 	free(memory);
dvander@67: }
dvander@67: 
dvander@67: IPluginContext *SourcePawnEngine::CreateBaseContext(sp_context_t *ctx)
dvander@67: {
dvander@67: 	return new BaseContext(ctx);
dvander@67: }
dvander@67: 
dvander@67: void SourcePawnEngine::FreeBaseContext(IPluginContext *ctx)
dvander@67: {
dvander@67: 	sp_context_t *_ctx = ctx->GetContext();
dvander@67: 	IVirtualMachine *vm = ctx->GetVirtualMachine();
dvander@67: 	
dvander@93: 	vm->FreeContext(_ctx);
dvander@67: 
dvander@67: 	delete ctx;
dvander@67: }
dvander@67: 
dvander@67: sp_plugin_t *_ReadPlugin(sp_file_hdr_t *hdr, uint8_t *base, sp_plugin_t *plugin, int *err)
dvander@67: {
dvander@67: 	char *nameptr;
dvander@67: 	uint8_t sectnum = 0;
dvander@67: 	sp_file_section_t *secptr = (sp_file_section_t *)(base + sizeof(sp_file_hdr_t));
dvander@67: 
dvander@67: 	memset(plugin, 0, sizeof(sp_plugin_t));
dvander@67: 
dvander@67: 	plugin->base = base;
dvander@67: 
dvander@67: 	while (sectnum < hdr->sections)
dvander@67: 	{
dvander@67: 		nameptr = (char *)(base + hdr->stringtab + secptr->nameoffs);
dvander@67: 
dvander@67: 		if (!(plugin->pcode) && !strcmp(nameptr, ".code"))
dvander@67: 		{
dvander@67: 			sp_file_code_t *cod = (sp_file_code_t *)(base + secptr->dataoffs);
dvander@67: 			plugin->pcode = base + secptr->dataoffs + cod->code;
dvander@67: 			plugin->pcode_size = cod->codesize;
dvander@67: 			plugin->flags = cod->flags;
dvander@67: 		}
dvander@67: 		else if (!(plugin->data) && !strcmp(nameptr, ".data"))
dvander@67: 		{
dvander@67: 			sp_file_data_t *dat = (sp_file_data_t *)(base + secptr->dataoffs);
dvander@67: 			plugin->data = base + secptr->dataoffs + dat->data;
dvander@67: 			plugin->data_size = dat->datasize;
dvander@67: 			plugin->memory = dat->memsize;
dvander@67: 		}
dvander@67: 		else if (!(plugin->info.publics) && !strcmp(nameptr, ".publics"))
dvander@67: 		{
dvander@67: 			plugin->info.publics_num = secptr->size / sizeof(sp_file_publics_t);
dvander@67: 			plugin->info.publics = (sp_file_publics_t *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 		else if (!(plugin->info.pubvars) && !strcmp(nameptr, ".pubvars"))
dvander@67: 		{
dvander@67: 			plugin->info.pubvars_num = secptr->size / sizeof(sp_file_pubvars_t);
dvander@67: 			plugin->info.pubvars = (sp_file_pubvars_t *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 		else if (!(plugin->info.natives) && !strcmp(nameptr, ".natives"))
dvander@67: 		{
dvander@67: 			plugin->info.natives_num = secptr->size / sizeof(sp_file_natives_t);
dvander@67: 			plugin->info.natives = (sp_file_natives_t *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 		else if (!(plugin->info.stringbase) && !strcmp(nameptr, ".names"))
dvander@67: 		{
dvander@67: 			plugin->info.stringbase = (const char *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 		else if (!(plugin->debug.files) && !strcmp(nameptr, ".dbg.files"))
dvander@67: 		{
dvander@67: 			plugin->debug.files = (sp_fdbg_file_t *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 		else if (!(plugin->debug.lines) && !strcmp(nameptr, ".dbg.lines"))
dvander@67: 		{
dvander@67: 			plugin->debug.lines = (sp_fdbg_line_t *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 		else if (!(plugin->debug.symbols) && !strcmp(nameptr, ".dbg.symbols"))
dvander@67: 		{
dvander@67: 			plugin->debug.symbols = (sp_fdbg_symbol_t *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 		else if (!(plugin->debug.lines_num) && !strcmp(nameptr, ".dbg.info"))
dvander@67: 		{
dvander@67: 			sp_fdbg_info_t *inf = (sp_fdbg_info_t *)(base + secptr->dataoffs);
dvander@67: 			plugin->debug.files_num = inf->num_files;
dvander@67: 			plugin->debug.lines_num = inf->num_lines;
dvander@67: 			plugin->debug.syms_num = inf->num_syms;
dvander@67: 		}
dvander@67: 		else if (!(plugin->debug.stringbase) && !strcmp(nameptr, ".dbg.strings"))
dvander@67: 		{
dvander@67: 			plugin->debug.stringbase = (const char *)(base + secptr->dataoffs);
dvander@67: 		}
dvander@67: 
dvander@67: 		secptr++;
dvander@67: 		sectnum++;
dvander@67: 	}
dvander@67: 
dvander@67: 	if (!(plugin->pcode) || !(plugin->data) || !(plugin->info.stringbase))
dvander@67: 	{
dvander@67: 		goto return_error;
dvander@67: 	}
dvander@67: 
dvander@67: 	if ((plugin->flags & SP_FLAG_DEBUG) && (!(plugin->debug.files) || !(plugin->debug.lines) || !(plugin->debug.symbols)))
dvander@67: 	{
dvander@67: 		goto return_error;
dvander@67: 	}
dvander@67: 
dvander@67: 	if (err)
dvander@67: 	{
dvander@67: 		*err = SP_ERR_NONE;
dvander@67: 	}
dvander@67: 
dvander@67: 	return plugin;
dvander@67: 
dvander@67: return_error:
dvander@67: 	if (err)
dvander@67: 	{
dvander@67: 		*err = SP_ERR_FILE_FORMAT;
dvander@67: 	}
dvander@67: 
dvander@67: 	return NULL;
dvander@67: }
dvander@67: 
dvander@67: sp_plugin_t *SourcePawnEngine::LoadFromFilePointer(FILE *fp, int *err)
dvander@67: {
dvander@67: 	sp_file_hdr_t hdr;
dvander@67: 	sp_plugin_t *plugin;
dvander@67: 	uint8_t *base;
dvander@67: 	int z_result;
dvander@67: 	int error;
dvander@67: 
dvander@67: 	if (!fp)
dvander@67: 	{
dvander@67: 		error = SP_ERR_NOT_FOUND;
dvander@67: 		goto return_error;
dvander@67: 	}
dvander@67: 
dvander@67: 	/* Rewind for safety */
dvander@67: 	rewind(fp);
dvander@67: 	fread(&hdr, sizeof(sp_file_hdr_t), 1, fp);
dvander@67: 
dvander@67: 	if (hdr.magic != SPFILE_MAGIC)
dvander@67: 	{
dvander@67: 		error = SP_ERR_FILE_FORMAT;
dvander@67: 		goto return_error;
dvander@67: 	}
dvander@67: 
dvander@67: 	switch (hdr.compression)
dvander@67: 	{
dvander@67: 	case SPFILE_COMPRESSION_GZ:
dvander@67: 		{
dvander@67: 			uint32_t uncompsize = hdr.imagesize - hdr.dataoffs;
dvander@67: 			uint32_t compsize = hdr.disksize - hdr.dataoffs;
dvander@67: 			uint32_t sectsize = hdr.dataoffs - sizeof(sp_file_hdr_t);
dvander@67: 			uLongf destlen = uncompsize;
dvander@67: 
dvander@67: 			char *tempbuf = (char *)malloc(compsize);
dvander@67: 			void *uncompdata = malloc(uncompsize);
dvander@67: 			void *sectheader = malloc(sectsize);
dvander@67: 
dvander@67: 			fread(sectheader, sectsize, 1, fp);
dvander@67: 			fread(tempbuf, compsize, 1, fp);
dvander@67: 
dvander@67: 			z_result = uncompress((Bytef *)uncompdata, &destlen, (Bytef *)tempbuf, compsize);
dvander@67: 			free(tempbuf);
dvander@67: 			if (z_result != Z_OK)
dvander@67: 			{
dvander@67: 				free(sectheader);
dvander@67: 				free(uncompdata);
dvander@67: 				error = SP_ERR_DECOMPRESSOR;
dvander@67: 				goto return_error;
dvander@67: 			}
dvander@67: 
dvander@67: 			base = (uint8_t *)malloc(hdr.imagesize);
dvander@67: 			memcpy(base, &hdr, sizeof(sp_file_hdr_t));
dvander@67: 			memcpy(base + sizeof(sp_file_hdr_t), sectheader, sectsize);
dvander@67: 			free(sectheader);
dvander@67: 			memcpy(base + hdr.dataoffs, uncompdata, uncompsize);
dvander@67: 			free(uncompdata);
dvander@67: 			break;
dvander@67: 		}
dvander@67: 	case SPFILE_COMPRESSION_NONE:
dvander@67: 		{
dvander@67: 			base = (uint8_t *)malloc(hdr.imagesize);
dvander@67: 			rewind(fp);
dvander@67: 			fread(base, hdr.imagesize, 1, fp);
dvander@67: 			break;
dvander@67: 		}
dvander@67: 	default:
dvander@67: 		{
dvander@67: 			error = SP_ERR_DECOMPRESSOR;
dvander@67: 			goto return_error;
dvander@67: 		}
dvander@67: 	}
dvander@67: 
dvander@67: 	plugin = (sp_plugin_t *)malloc(sizeof(sp_plugin_t));
dvander@67: 	if (!_ReadPlugin(&hdr, base, plugin, err))
dvander@67: 	{
dvander@67: 		free(plugin);
dvander@67: 		free(base);
dvander@67: 		return NULL;
dvander@67: 	}
dvander@67: 
dvander@67: 	plugin->allocflags = 0;
dvander@67: 
dvander@67: 	return plugin;
dvander@67: 
dvander@67: return_error:
dvander@67: 	if (err)
dvander@67: 	{
dvander@67: 		*err = error;
dvander@67: 	}
dvander@67: 	return NULL;
dvander@67: }
dvander@67: 
dvander@67: sp_plugin_t *SourcePawnEngine::LoadFromMemory(void *base, sp_plugin_t *plugin, int *err)
dvander@67: {
dvander@67: 	sp_file_hdr_t hdr;
dvander@67: 	uint8_t noptr = 0;
dvander@67: 
dvander@67: 	memcpy(&hdr, base, sizeof(sp_file_hdr_t));
dvander@67: 
dvander@67: 	if (!plugin)
dvander@67: 	{
dvander@67: 		plugin = (sp_plugin_t *)malloc(sizeof(sp_plugin_t));
dvander@67: 		noptr = 1;
dvander@67: 	}
dvander@67: 
dvander@67: 	if (!_ReadPlugin(&hdr, (uint8_t *)base, plugin, err))
dvander@67: 	{
dvander@67: 		if (noptr)
dvander@67: 		{
dvander@67: 			free(plugin);
dvander@67: 		}
dvander@67: 		return NULL;
dvander@67: 	}
dvander@67: 
dvander@67: 	if (!noptr)
dvander@67: 	{
dvander@67: 		plugin->allocflags |= SP_FA_SELF_EXTERNAL;
dvander@67: 	}
dvander@67: 	plugin->allocflags |= SP_FA_BASE_EXTERNAL;
dvander@67: 
dvander@67: 	return plugin;
dvander@67: }
dvander@67: 
dvander@67: int SourcePawnEngine::FreeFromMemory(sp_plugin_t *plugin)
dvander@67: {
dvander@67: 	if (!(plugin->allocflags & SP_FA_BASE_EXTERNAL))
dvander@67: 	{
dvander@67: 		free(plugin->base);
dvander@67: 		plugin->base = NULL;
dvander@67: 	}
dvander@67: 	if (!(plugin->allocflags & SP_FA_SELF_EXTERNAL))
dvander@67: 	{
dvander@67: 		free(plugin);
dvander@67: 	}
dvander@67: 
dvander@67: 	return SP_ERR_NONE;
dvander@67: }