f1f54311f6
git-svn-id: https://svn.code.sf.net/p/fteqw/code/trunk@5209 fc73d0e0-1445-4013-8a0c-d673dee63da5
1692 lines
43 KiB
C
1692 lines
43 KiB
C
#define PROGSUSED
|
|
#include "progsint.h"
|
|
#include <stdlib.h>
|
|
|
|
static void PR_FreeAllTemps (progfuncs_t *progfuncs);
|
|
|
|
typedef struct prmemb_s {
|
|
struct prmemb_s *prev;
|
|
int level;
|
|
} prmemb_t;
|
|
void *PRHunkAlloc(progfuncs_t *progfuncs, int ammount, char *name)
|
|
{
|
|
prmemb_t *mem;
|
|
ammount = sizeof(prmemb_t)+((ammount + 3)&~3);
|
|
mem = progfuncs->funcs.parms->memalloc(ammount);
|
|
memset(mem, 0, ammount);
|
|
mem->prev = prinst.memblocks;
|
|
if (!prinst.memblocks)
|
|
mem->level = 1;
|
|
else
|
|
mem->level = ((prmemb_t *)prinst.memblocks)->level+1;
|
|
prinst.memblocks = mem;
|
|
|
|
return ((char *)mem)+sizeof(prmemb_t);
|
|
}
|
|
void *PDECL QC_HunkAlloc(pubprogfuncs_t *ppf, int ammount, char *name)
|
|
{
|
|
return PRHunkAlloc((progfuncs_t*)ppf, ammount, name);
|
|
}
|
|
|
|
int PRHunkMark(progfuncs_t *progfuncs)
|
|
{
|
|
return ((prmemb_t *)prinst.memblocks)->level;
|
|
}
|
|
void PRHunkFree(progfuncs_t *progfuncs, int mark)
|
|
{
|
|
prmemb_t *omem;
|
|
while(prinst.memblocks)
|
|
{
|
|
if (prinst.memblocks->level <= mark)
|
|
return;
|
|
|
|
omem = prinst.memblocks;
|
|
prinst.memblocks = prinst.memblocks->prev;
|
|
externs->memfree(omem);
|
|
}
|
|
return;
|
|
}
|
|
|
|
/*if we ran out of memory, the vm can allocate a new block, but doing so requires fixing up all sorts of pointers*/
|
|
void PRAddressableRelocate(progfuncs_t *progfuncs, char *oldb, char *newb, int oldlen)
|
|
{
|
|
unsigned int i;
|
|
edictrun_t *e;
|
|
for (i=0 ; i<prinst.maxedicts; i++)
|
|
{
|
|
e = (edictrun_t *)(prinst.edicttable[i]);
|
|
if (e && (char*)e->fields >= oldb && (char*)e->fields < oldb+oldlen)
|
|
e->fields = ((char*)e->fields - oldb) + newb;
|
|
}
|
|
|
|
if (progfuncs->funcs.stringtable >= oldb && progfuncs->funcs.stringtable < oldb+oldlen)
|
|
progfuncs->funcs.stringtable = (progfuncs->funcs.stringtable - oldb) + newb;
|
|
|
|
for (i=0; i < prinst.maxprogs; i++)
|
|
{
|
|
if ((char*)prinst.progstate[i].globals >= oldb && (char*)prinst.progstate[i].globals < oldb+oldlen)
|
|
prinst.progstate[i].globals = (float*)(((char*)prinst.progstate[i].globals - oldb) + newb);
|
|
if (prinst.progstate[i].strings >= oldb && prinst.progstate[i].strings < oldb+oldlen)
|
|
prinst.progstate[i].strings = (prinst.progstate[i].strings - oldb) + newb;
|
|
}
|
|
|
|
for (i = 0; i < prinst.numfields; i++)
|
|
{
|
|
if (prinst.field[i].name >= oldb && prinst.field[i].name < oldb+oldlen)
|
|
prinst.field[i].name = (prinst.field[i].name - oldb) + newb;
|
|
}
|
|
|
|
externs->addressablerelocated(&progfuncs->funcs, oldb, newb, oldlen);
|
|
}
|
|
|
|
//for 64bit systems. :)
|
|
//addressable memory is memory available to the vm itself for writing.
|
|
//once allocated, it cannot be freed for the lifetime of the VM.
|
|
//if src is null, data srcsize is left uninitialised for speed.
|
|
//pad is always 0-filled.
|
|
void *PRAddressableExtend(progfuncs_t *progfuncs, void *src, size_t srcsize, int pad)
|
|
{
|
|
char *ptr;
|
|
int ammount = (srcsize+pad + 4)&~3; //round up to 4
|
|
pad = ammount - srcsize;
|
|
if (prinst.addressableused + ammount > prinst.addressablesize)
|
|
{
|
|
/*only do this if the caller states that it can cope with addressable-block relocations/resizes*/
|
|
if (externs->addressablerelocated)
|
|
{
|
|
#if defined(_WIN32) && !defined(WINRT)
|
|
char *newblock;
|
|
#if 0//def _DEBUG
|
|
int oldtot = addressablesize;
|
|
#endif
|
|
int newsize = (prinst.addressableused + ammount + 4096) & ~(4096-1);
|
|
newblock = VirtualAlloc (NULL, prinst.addressablesize, MEM_RESERVE, PAGE_NOACCESS);
|
|
if (newblock)
|
|
{
|
|
VirtualAlloc (newblock, prinst.addressableused, MEM_COMMIT, PAGE_READWRITE);
|
|
memcpy(newblock, prinst.addressablehunk, prinst.addressableused);
|
|
#if 0//def _DEBUG
|
|
VirtualAlloc (prinst.addressablehunk, oldtot, MEM_RESERVE, PAGE_NOACCESS);
|
|
#else
|
|
VirtualFree (prinst.addressablehunk, 0, MEM_RELEASE);
|
|
#endif
|
|
PRAddressableRelocate(progfuncs, prinst.addressablehunk, newblock, prinst.addressableused);
|
|
prinst.addressablehunk = newblock;
|
|
prinst.addressablesize = newsize;
|
|
}
|
|
#else
|
|
int newsize = (prinst.addressableused + ammount + 1024*1024) & ~(1024*1024-1);
|
|
char *newblock = malloc(newsize);
|
|
if (newblock)
|
|
{
|
|
PRAddressableRelocate(progfuncs, prinst.addressablehunk, newblock, prinst.addressableused);
|
|
free(prinst.addressablehunk);
|
|
prinst.addressablehunk = newblock;
|
|
prinst.addressablesize = newsize;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
if (prinst.addressableused + ammount > prinst.addressablesize)
|
|
Sys_Error("Not enough addressable memory for progs VM (using %gmb)", prinst.addressablesize/(1024.0*1024));
|
|
}
|
|
|
|
prinst.addressableused += ammount;
|
|
progfuncs->funcs.stringtablesize = prinst.addressableused;
|
|
|
|
#if defined(_WIN32) && !defined(WINRT)
|
|
if (!VirtualAlloc (prinst.addressablehunk, prinst.addressableused, MEM_COMMIT, PAGE_READWRITE))
|
|
Sys_Error("VirtualAlloc failed. Blame windows.");
|
|
#endif
|
|
|
|
ptr = &prinst.addressablehunk[prinst.addressableused-ammount];
|
|
if (src)
|
|
memcpy(ptr, src, srcsize);
|
|
#ifdef _DEBUG
|
|
else
|
|
memset(ptr, 0xcc, srcsize);
|
|
#endif
|
|
memset(ptr+srcsize, 0, pad);
|
|
return &prinst.addressablehunk[prinst.addressableused-ammount];
|
|
}
|
|
|
|
|
|
#define MARKER_USED 0xC2A4F5A6u
|
|
#define MARKER_FREE 0xF1E3E3E7u
|
|
typedef struct
|
|
{
|
|
#ifdef _DEBUG
|
|
unsigned int marker;
|
|
#endif
|
|
unsigned int next;
|
|
unsigned int prev;
|
|
unsigned int size; //includes header size
|
|
} qcmemfreeblock_t;
|
|
typedef struct
|
|
{
|
|
unsigned int marker;
|
|
#ifdef _DEBUG
|
|
unsigned int next;
|
|
unsigned int prev;
|
|
#endif
|
|
unsigned int size; //includes header size
|
|
} qcmemusedblock_t;
|
|
static void PF_fmem_unlink(progfuncs_t *progfuncs, qcmemfreeblock_t *p)
|
|
{
|
|
qcmemfreeblock_t *np;
|
|
#ifdef _DEBUG
|
|
if (p->marker != MARKER_FREE)
|
|
{
|
|
printf("PF_fmem_unlink: memory corruption\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
}
|
|
p->marker = 0;
|
|
#endif
|
|
if (p->prev)
|
|
{
|
|
np = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + p->prev);
|
|
np->next = p->next;
|
|
}
|
|
else
|
|
progfuncs->inst.mfreelist = p->next;
|
|
if (p->next)
|
|
{
|
|
np = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + p->next);
|
|
np->prev = p->prev;
|
|
}
|
|
}
|
|
|
|
static void PR_memvalidate (progfuncs_t *progfuncs)
|
|
{
|
|
qcmemfreeblock_t *p;
|
|
unsigned int b,l;
|
|
|
|
b = prinst.mfreelist;
|
|
l = 0;
|
|
while (b)
|
|
{
|
|
if (b < 0 || b >= prinst.addressableused)
|
|
{
|
|
printf("PF_memalloc: memory corruption\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
p = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + b);
|
|
|
|
if (
|
|
#ifdef _DEBUG
|
|
p->marker != MARKER_FREE ||
|
|
#endif
|
|
p->prev != l ||
|
|
(p->next && p->next < b + p->size) ||
|
|
p->next >= prinst.addressableused ||
|
|
b + p->size >= prinst.addressableused ||
|
|
p->prev >= b)
|
|
{
|
|
printf("PF_memalloc: memory corruption\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
l = b;
|
|
b = p->next;
|
|
}
|
|
}
|
|
|
|
static void *PDECL PR_memalloc (pubprogfuncs_t *ppf, unsigned int size)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
qcmemfreeblock_t *p, *np;
|
|
qcmemusedblock_t *ub = NULL;
|
|
unsigned int b,n;
|
|
/*round size up*/
|
|
size = (size+sizeof(qcmemusedblock_t) + 63) & ~63;
|
|
|
|
PR_memvalidate(progfuncs);
|
|
|
|
b = prinst.mfreelist;
|
|
while (b)
|
|
{
|
|
if (/*b < 0 || */b+sizeof(qcmemfreeblock_t) >= prinst.addressableused)
|
|
{
|
|
printf("PF_memalloc: memory corruption\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return NULL;
|
|
}
|
|
p = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + b);
|
|
if (p->size >= size)
|
|
{
|
|
if ((p->next && p->next < b + p->size) ||
|
|
p->next >= prinst.addressableused ||
|
|
b + p->size >= prinst.addressableused ||
|
|
p->prev >= b)
|
|
{
|
|
printf("PF_memalloc: memory corruption\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return NULL;
|
|
}
|
|
|
|
ub = (qcmemusedblock_t*)p;
|
|
if (p->size > size + 63)
|
|
{
|
|
/*make a new header just after it, with basically the same properties, and shift the important fields over*/
|
|
n = b + size;
|
|
np = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + b + size);
|
|
#ifdef _DEBUG
|
|
np->marker = MARKER_FREE;
|
|
#endif
|
|
np->prev = p->prev;
|
|
np->next = p->next;
|
|
np->size = p->size - size;
|
|
if (np->prev)
|
|
{
|
|
p = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + np->prev);
|
|
p->next = n;
|
|
}
|
|
else
|
|
prinst.mfreelist = n;
|
|
if (p->next)
|
|
{
|
|
p = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + np->next);
|
|
p->prev = n;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
size = p->size; /*alloc the entire block*/
|
|
/*unlink this entry*/
|
|
PF_fmem_unlink(progfuncs, p);
|
|
}
|
|
break;
|
|
}
|
|
b = p->next;
|
|
}
|
|
|
|
/*assign more space*/
|
|
if (!ub)
|
|
{
|
|
ub = PRAddressableExtend(progfuncs, NULL, size, 0);
|
|
if (!ub)
|
|
{
|
|
printf("PF_memalloc: memory exausted\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return NULL;
|
|
}
|
|
//FIXME: merge with previous block
|
|
}
|
|
memset(ub, 0, size);
|
|
ub->marker = MARKER_USED;
|
|
ub->size = size;
|
|
|
|
PR_memvalidate(progfuncs);
|
|
|
|
return ub+1;
|
|
}
|
|
static void PDECL PR_memfree (pubprogfuncs_t *ppf, void *memptr)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
qcmemusedblock_t *ub;
|
|
qcmemfreeblock_t *b, *nb, *pb;
|
|
unsigned int pa, na; //prev addr, next addr
|
|
unsigned int size;
|
|
unsigned int ptr = memptr?((char*)memptr - progfuncs->funcs.stringtable):0;
|
|
|
|
/*freeing NULL is ignored*/
|
|
if (!ptr)
|
|
return;
|
|
PR_memvalidate(progfuncs);
|
|
ptr -= sizeof(qcmemusedblock_t);
|
|
if (/*ptr < 0 ||*/ ptr >= prinst.addressableused)
|
|
{
|
|
ptr += sizeof(qcmemusedblock_t);
|
|
if (ptr < prinst.addressableused && !*(char*)memptr)
|
|
{
|
|
//the empty string is a point of contention. while we can detect it from fteqcc, its best to not give any special favours (other than nicer debugging, where possible)
|
|
//we might not actually spot it from other qccs, so warning about it where possible is probably a very good thing.
|
|
printf("PF_memfree: unable to free the non-null empty string constant at %x\n", ptr);
|
|
}
|
|
else
|
|
printf("PF_memfree: pointer invalid - out of range (%x >= %x)\n", ptr, (unsigned int)prinst.addressableused);
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
|
|
//this is the used block that we're trying to free
|
|
ub = (qcmemusedblock_t*)(progfuncs->funcs.stringtable + ptr);
|
|
if (ub->marker != MARKER_USED || ub->size <= sizeof(*ub) || ptr + ub->size > (unsigned int)prinst.addressableused)
|
|
{
|
|
printf("PR_memfree: pointer lacks marker - double-freed?\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
ub->marker = 0; //invalidate it
|
|
size = ub->size;
|
|
ub = NULL;
|
|
|
|
//we have an (ordered) list of free blocks.
|
|
//in order to free our memory, we need to find the free block before+after the 'new' block
|
|
for (na = prinst.mfreelist, pa = 0; ;)
|
|
{
|
|
if (/*na < 0 ||*/ na >= prinst.addressableused)
|
|
{
|
|
printf("PF_memfree: memory corruption\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
if (!na || na >= ptr)
|
|
{
|
|
pb = pa?(qcmemfreeblock_t*)(progfuncs->funcs.stringtable + pa):NULL;
|
|
if (pb && pa+pb->size>ptr)
|
|
{ //previous free block extends into the block that we're trying to free.
|
|
printf("PF_memfree: double free\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
#ifdef _DEBUG
|
|
if (pb && pb->marker != MARKER_FREE)
|
|
{
|
|
printf("PF_memfree: use-after-free?\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
nb = na?(qcmemfreeblock_t*)(progfuncs->funcs.stringtable + na):NULL;
|
|
if (nb && ptr+size > na)
|
|
{
|
|
printf("PF_memfree: block extends into neighbour\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
#ifdef _DEBUG
|
|
if (nb && nb->marker != MARKER_FREE)
|
|
{
|
|
printf("PF_memfree: use-after-free?\n");
|
|
PR_StackTrace(&progfuncs->funcs, false);
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
/*generate the free block, now we know its proper values*/
|
|
b = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + ptr);
|
|
#ifdef _DEBUG
|
|
b->marker = MARKER_FREE;
|
|
#endif
|
|
b->prev = pa;
|
|
b->next = na;
|
|
b->size = size;
|
|
if (na)
|
|
nb->prev = ptr;
|
|
if (!pa)
|
|
prinst.mfreelist = ptr;
|
|
else
|
|
pb->next = ptr;
|
|
|
|
/*extend this block and kill the next if they are adjacent*/
|
|
if (na && b->next == ptr + size)
|
|
{
|
|
b->size += nb->size;
|
|
PF_fmem_unlink(progfuncs, nb);
|
|
}
|
|
/*we're adjacent to the previous block, so merge them by killing the newly freed region*/
|
|
if (pa && pa + pb->size == ptr)
|
|
{
|
|
pb->size += size;
|
|
PF_fmem_unlink(progfuncs, b);
|
|
}
|
|
break;
|
|
}
|
|
|
|
pa = na;
|
|
b = (qcmemfreeblock_t*)(progfuncs->funcs.stringtable + pa);
|
|
na = b->next;
|
|
}
|
|
|
|
PR_memvalidate(progfuncs);
|
|
}
|
|
|
|
void PRAddressableFlush(progfuncs_t *progfuncs, size_t totalammount)
|
|
{
|
|
prinst.addressableused = 0;
|
|
if (totalammount <= 0) //flush
|
|
{
|
|
totalammount = prinst.addressablesize;
|
|
// return;
|
|
}
|
|
|
|
#if defined(_WIN32) && !defined(WINRT)
|
|
if (prinst.addressablehunk && prinst.addressablesize != totalammount)
|
|
{
|
|
VirtualFree(prinst.addressablehunk, 0, MEM_RELEASE); //doesn't this look complicated? :p
|
|
prinst.addressablehunk = NULL;
|
|
}
|
|
if (!prinst.addressablehunk)
|
|
prinst.addressablehunk = VirtualAlloc (prinst.addressablehunk, totalammount, MEM_RESERVE, PAGE_NOACCESS);
|
|
#else
|
|
if (prinst.addressablehunk && prinst.addressablesize != totalammount)
|
|
{
|
|
free(prinst.addressablehunk);
|
|
prinst.addressablehunk = NULL;
|
|
}
|
|
if (!prinst.addressablehunk)
|
|
prinst.addressablehunk = malloc(totalammount); //linux will allocate-on-use anyway, which is handy.
|
|
// memset(prinst.addressablehunk, 0xff, totalammount);
|
|
#endif
|
|
if (!prinst.addressablehunk)
|
|
Sys_Error("Out of memory\n");
|
|
prinst.addressablesize = totalammount;
|
|
progfuncs->funcs.stringtablemaxsize = totalammount;
|
|
}
|
|
|
|
int PDECL PR_InitEnts(pubprogfuncs_t *ppf, int max_ents)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
edictrun_t *e;
|
|
prinst.maxedicts = max_ents;
|
|
|
|
sv_num_edicts = 0;
|
|
|
|
#if 0
|
|
{
|
|
int i;
|
|
for (i = 0; i < prinst.numfields; i++)
|
|
{
|
|
printf("%s(%i) %i -> %i\n", prinst.field[i].name, prinst.field[i].type, prinst.field[i].progsofs, prinst.field[i].ofs);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
prinst.max_fields_size = prinst.fields_size;
|
|
|
|
prinst.edicttable = (struct edictrun_s**)(progfuncs->funcs.edicttable = PRHunkAlloc(progfuncs, prinst.maxedicts*sizeof(struct edicts_s *), "edicttable"));
|
|
e = PRHunkAlloc(progfuncs, externs->edictsize, "edict0");
|
|
e->fieldsize = prinst.fields_size;
|
|
e->entnum = 0;
|
|
e->ereftype = ER_ENTITY;
|
|
sv_edicts = (struct edict_s *)e;
|
|
sv_num_edicts = 1;
|
|
progfuncs->funcs.edicttable[0] = sv_edicts;
|
|
e->fields = PRAddressableExtend(progfuncs, NULL, e->fieldsize, prinst.max_fields_size-e->fieldsize);
|
|
QC_ClearEdict(&progfuncs->funcs, sv_edicts);
|
|
|
|
if (externs->entspawn)
|
|
externs->entspawn(sv_edicts, false);
|
|
|
|
return prinst.max_fields_size;
|
|
}
|
|
edictrun_t tempedict; //used as a safty buffer
|
|
static float tempedictfields[2048];
|
|
|
|
static void PDECL PR_Configure (pubprogfuncs_t *ppf, size_t addressable_size, int max_progs, pbool profiling) //can be used to wipe all memory
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
unsigned int i;
|
|
edictrun_t *e;
|
|
|
|
prinst.max_fields_size=0;
|
|
prinst.fields_size = 0;
|
|
progfuncs->funcs.stringtable = 0;
|
|
QC_StartShares(progfuncs);
|
|
QC_InitShares(progfuncs);
|
|
|
|
for ( i=1 ; i<prinst.maxedicts; i++)
|
|
{
|
|
e = (edictrun_t *)(prinst.edicttable[i]);
|
|
prinst.edicttable[i] = NULL;
|
|
// e->entnum = i;
|
|
if (e)
|
|
externs->memfree(e);
|
|
}
|
|
|
|
PRHunkFree(progfuncs, 0); //clear mem - our hunk may not be a real hunk.
|
|
if (addressable_size == (size_t)-1)
|
|
{
|
|
#if defined(_WIN64) && !defined(WINRT)
|
|
addressable_size = 0x80000000; //use of virtual address space rather than physical memory means we can just go crazy and use the max of 2gb.
|
|
#elif defined(FTE_TARGET_WEB)
|
|
addressable_size = 8*1024*1024;
|
|
#else
|
|
addressable_size = 32*1024*1024;
|
|
#endif
|
|
}
|
|
if (addressable_size > 0x80000000)
|
|
addressable_size = 0x80000000;
|
|
PRAddressableFlush(progfuncs, addressable_size);
|
|
|
|
pr_progstate = PRHunkAlloc(progfuncs, sizeof(progstate_t) * max_progs, "progstatetable");
|
|
|
|
/* for(a = 0; a < max_progs; a++)
|
|
{
|
|
pr_progstate[a].progs = NULL;
|
|
}
|
|
*/
|
|
|
|
prinst.maxprogs = max_progs;
|
|
prinst.pr_typecurrent=-1;
|
|
|
|
PR_FreeAllTemps(progfuncs);
|
|
|
|
prinst.reorganisefields = false;
|
|
|
|
prinst.profiling = profiling;
|
|
prinst.profilingalert = Sys_GetClockRate();
|
|
prinst.maxedicts = 1;
|
|
prinst.edicttable = (edictrun_t**)(progfuncs->funcs.edicttable = &sv_edicts);
|
|
sv_num_edicts = 1; //set up a safty buffer so things won't go horribly wrong too often
|
|
sv_edicts=(struct edict_s *)&tempedict;
|
|
tempedict.readonly = true;
|
|
tempedict.fields = tempedictfields;
|
|
tempedict.ereftype = ER_ENTITY;
|
|
}
|
|
|
|
|
|
|
|
struct globalvars_s *PDECL PR_globals (pubprogfuncs_t *ppf, progsnum_t pnum)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
if (pnum < 0)
|
|
{
|
|
if (!current_progstate)
|
|
{
|
|
static float fallback[RESERVED_OFS];
|
|
return (struct globalvars_s *)fallback; //err.. you've not loaded one yet.
|
|
}
|
|
return (struct globalvars_s *)current_progstate->globals;
|
|
}
|
|
return (struct globalvars_s *)pr_progstate[pnum].globals;
|
|
}
|
|
|
|
struct entvars_s *PDECL PR_entvars (pubprogfuncs_t *ppf, struct edict_s *ed)
|
|
{
|
|
// progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
if (((edictrun_t *)ed)->ereftype != ER_ENTITY)
|
|
return NULL;
|
|
|
|
return (struct entvars_s *)edvars(ed);
|
|
}
|
|
|
|
pbool PDECL PR_GetFunctionInfo(pubprogfuncs_t *ppf, func_t func, int *args, int *builtinnum, char *funcname, size_t funcnamesize)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
|
|
unsigned int pnum;
|
|
unsigned int fnum;
|
|
mfunction_t *f;
|
|
|
|
pnum = (func & 0xff000000)>>24;
|
|
fnum = (func & 0x00ffffff);
|
|
|
|
if (pnum >= prinst.maxprogs || !pr_progstate[pnum].functions)
|
|
return false;
|
|
else if (fnum >= pr_progstate[pnum].progs->numfunctions)
|
|
return false;
|
|
else
|
|
{
|
|
f = pr_progstate[pnum].functions + fnum;
|
|
if (args)
|
|
*args = f->numparms;
|
|
if (builtinnum)
|
|
*builtinnum = -f->first_statement;
|
|
if (funcname)
|
|
{
|
|
const char *srcname = PR_StringToNative(ppf, f->s_name);
|
|
size_t nlen = strlen(srcname);
|
|
if (nlen < funcnamesize)
|
|
memcpy(funcname, srcname, nlen+1);
|
|
else
|
|
*funcname = 0;
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
func_t PDECL PR_FindFunc(pubprogfuncs_t *ppf, const char *funcname, progsnum_t pnum)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
mfunction_t *f=NULL;
|
|
if (pnum == PR_ANY)
|
|
{
|
|
for (pnum = 0; (unsigned)pnum < prinst.maxprogs; pnum++)
|
|
{
|
|
if (!pr_progstate[pnum].progs)
|
|
continue;
|
|
f = ED_FindFunction(progfuncs, funcname, &pnum, pnum);
|
|
if (f)
|
|
break;
|
|
}
|
|
}
|
|
else if (pnum == PR_ANYBACK) //run backwards
|
|
{
|
|
for (pnum = prinst.maxprogs-1; pnum >= 0; pnum--)
|
|
{
|
|
if (!pr_progstate[pnum].progs)
|
|
continue;
|
|
f = ED_FindFunction(progfuncs, funcname, &pnum, pnum);
|
|
if (f)
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
f = ED_FindFunction(progfuncs, funcname, &pnum, pnum);
|
|
if (!f)
|
|
return 0;
|
|
|
|
{
|
|
ddef16_t *var16;
|
|
ddef32_t *var32;
|
|
switch(pr_progstate[pnum].structtype)
|
|
{
|
|
case PST_KKQWSV:
|
|
case PST_DEFAULT:
|
|
var16 = ED_FindTypeGlobalFromProgs16(progfuncs, funcname, pnum, ev_function); //we must make sure we actually have a function def - 'light' is defined as a field before it is defined as a function.
|
|
if (!var16)
|
|
return (f - pr_progstate[pnum].functions) | (pnum << 24);
|
|
return *(int *)&pr_progstate[pnum].globals[var16->ofs];
|
|
case PST_QTEST:
|
|
case PST_FTE32:
|
|
var32 = ED_FindTypeGlobalFromProgs32(progfuncs, funcname, pnum, ev_function); //we must make sure we actually have a function def - 'light' is defined as a field before it is defined as a function.
|
|
if (!var32)
|
|
return (f - pr_progstate[pnum].functions) | (pnum << 24);
|
|
return *(int *)&pr_progstate[pnum].globals[var32->ofs];
|
|
}
|
|
Sys_Error("Error with def size (PR_FindFunc)");
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void PDECL QC_FindPrefixedGlobals(pubprogfuncs_t *ppf, int pnum, char *prefix, void (PDECL *found) (pubprogfuncs_t *progfuncs, char *name, union eval_s *val, etype_t type, void *ctx), void *ctx)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
unsigned int i;
|
|
ddef16_t *def16;
|
|
ddef32_t *def32;
|
|
int len = strlen(prefix);
|
|
|
|
if (pnum == PR_CURRENT)
|
|
pnum = prinst.pr_typecurrent;
|
|
if (pnum == PR_ANY)
|
|
{
|
|
for (pnum = 0; (unsigned)pnum < prinst.maxprogs; pnum++)
|
|
{
|
|
if (!pr_progstate[pnum].progs)
|
|
continue;
|
|
QC_FindPrefixedGlobals(ppf, pnum, prefix, found, ctx);
|
|
}
|
|
return;
|
|
}
|
|
|
|
if (!pr_progstate[pnum].progs)
|
|
return;
|
|
|
|
switch(pr_progstate[pnum].structtype)
|
|
{
|
|
case PST_DEFAULT:
|
|
case PST_KKQWSV:
|
|
for (i=1 ; i<pr_progstate[pnum].progs->numglobaldefs ; i++)
|
|
{
|
|
def16 = &pr_progstate[pnum].globaldefs16[i];
|
|
if (!strncmp(def16->s_name+progfuncs->funcs.stringtable,prefix, len))
|
|
found(&progfuncs->funcs, def16->s_name+progfuncs->funcs.stringtable, (eval_t *)&pr_progstate[pnum].globals[def16->ofs], def16->type, ctx);
|
|
}
|
|
break;
|
|
case PST_QTEST:
|
|
case PST_FTE32:
|
|
for (i=1 ; i<pr_progstate[pnum].progs->numglobaldefs ; i++)
|
|
{
|
|
def32 = &pr_progstate[pnum].globaldefs32[i];
|
|
if (!strncmp(def32->s_name+progfuncs->funcs.stringtable,prefix, len))
|
|
found(&progfuncs->funcs, def32->s_name+progfuncs->funcs.stringtable, (eval_t *)&pr_progstate[pnum].globals[def32->ofs], def32->type, ctx);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
eval_t *PDECL PR_FindGlobal(pubprogfuncs_t *ppf, const char *globname, progsnum_t pnum, etype_t *type)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
unsigned int i;
|
|
ddef16_t *var16;
|
|
ddef32_t *var32;
|
|
if (pnum == PR_CURRENT)
|
|
pnum = prinst.pr_typecurrent;
|
|
if (pnum == PR_ANY)
|
|
{
|
|
eval_t *ev;
|
|
for (i = 0; i < prinst.maxprogs; i++)
|
|
{
|
|
if (!pr_progstate[i].progs)
|
|
continue;
|
|
ev = PR_FindGlobal(&progfuncs->funcs, globname, i, type);
|
|
if (ev)
|
|
return ev;
|
|
}
|
|
return NULL;
|
|
}
|
|
if (pnum < 0 || (unsigned)pnum >= prinst.maxprogs || !pr_progstate[pnum].progs)
|
|
return NULL;
|
|
switch(pr_progstate[pnum].structtype)
|
|
{
|
|
case PST_DEFAULT:
|
|
case PST_KKQWSV:
|
|
if (!(var16 = ED_FindGlobalFromProgs16(progfuncs, globname, pnum)))
|
|
return NULL;
|
|
|
|
if (type)
|
|
*type = var16->type;
|
|
return (eval_t *)&pr_progstate[pnum].globals[var16->ofs];
|
|
case PST_QTEST:
|
|
case PST_FTE32:
|
|
if (!(var32 = ED_FindGlobalFromProgs32(progfuncs, globname, pnum)))
|
|
return NULL;
|
|
|
|
if (type)
|
|
*type = var32->type;
|
|
return (eval_t *)&pr_progstate[pnum].globals[var32->ofs];
|
|
}
|
|
Sys_Error("Error with def size (PR_FindGlobal)");
|
|
return NULL;
|
|
}
|
|
|
|
char *PDECL PR_VarString (pubprogfuncs_t *ppf, int first)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
int i;
|
|
static char out[1024];
|
|
char *s;
|
|
|
|
out[0] = 0;
|
|
for (i=first ; i<progfuncs->funcs.callargc ; i++)
|
|
{
|
|
if (G_STRING(OFS_PARM0+i*3))
|
|
{
|
|
s=G_STRING((OFS_PARM0+i*3)) + progfuncs->funcs.stringtable;
|
|
if (strlen(out) + strlen(s) + 1 >= sizeof(out))
|
|
return out;
|
|
strcat (out, s);
|
|
}
|
|
}
|
|
return out;
|
|
}
|
|
|
|
int PDECL PR_QueryField (pubprogfuncs_t *ppf, unsigned int fieldoffset, etype_t *type, char **name, evalc_t *fieldcache)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
fdef_t *var;
|
|
var = ED_FieldAtOfs(progfuncs, fieldoffset);
|
|
if (!var)
|
|
return false;
|
|
|
|
if (type)
|
|
*type = var->type & ~(DEF_SAVEGLOBAL|DEF_SHARED);
|
|
if (name)
|
|
*name = var->name;
|
|
if (fieldcache)
|
|
{
|
|
fieldcache->ofs32 = var;
|
|
fieldcache->varname = var->name;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
eval_t *PDECL QC_GetEdictFieldValue(pubprogfuncs_t *ppf, struct edict_s *ed, char *name, etype_t type, evalc_t *cache)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
fdef_t *var;
|
|
if (!cache)
|
|
{
|
|
var = ED_FindField(progfuncs, name);
|
|
if (!var || (var->type != type && type))
|
|
return NULL;
|
|
return (eval_t *) &(((int*)(((edictrun_t*)ed)->fields))[var->ofs]);
|
|
}
|
|
if (!cache->varname)
|
|
{
|
|
cache->varname = name;
|
|
var = ED_FindField(progfuncs, name);
|
|
if (!var || (var->type != type && type))
|
|
{
|
|
cache->ofs32 = NULL;
|
|
return NULL;
|
|
}
|
|
cache->ofs32 = var;
|
|
cache->varname = var->name;
|
|
if (!ed)
|
|
return (void*)~0; //something not null
|
|
return (eval_t *) &(((int*)(((edictrun_t*)ed)->fields))[var->ofs]);
|
|
}
|
|
if (cache->ofs32 == NULL)
|
|
return NULL;
|
|
return (eval_t *) &(((int*)(((edictrun_t*)ed)->fields))[cache->ofs32->ofs]);
|
|
}
|
|
|
|
struct edict_s *PDECL ProgsToEdict (pubprogfuncs_t *ppf, int progs)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
if ((unsigned)progs >= (unsigned)prinst.maxedicts)
|
|
{
|
|
printf("Bad entity index %i\n", progs);
|
|
if (pr_depth)
|
|
{
|
|
PR_StackTrace (ppf, false);
|
|
// progfuncs->funcs.pr_trace += 1;
|
|
}
|
|
progs = 0;
|
|
}
|
|
return (struct edict_s *)PROG_TO_EDICT(progfuncs.inst, progs);
|
|
}
|
|
int PDECL EdictToProgs (pubprogfuncs_t *ppf, struct edict_s *ed)
|
|
{
|
|
// progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
return EDICT_TO_PROG(progfuncs, ed);
|
|
}
|
|
|
|
string_t PDECL PR_StringToProgs (pubprogfuncs_t *ppf, const char *str)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
char **ntable;
|
|
int i, free=-1;
|
|
|
|
if (!str)
|
|
return 0;
|
|
|
|
if (str >= progfuncs->funcs.stringtable && str < progfuncs->funcs.stringtable + prinst.addressableused)
|
|
return str - progfuncs->funcs.stringtable;
|
|
|
|
for (i = prinst.numallocedstrings-1; i >= 0; i--)
|
|
{
|
|
if (prinst.allocedstrings[i] == str)
|
|
return (string_t)((unsigned int)i | STRING_STATIC);
|
|
if (!prinst.allocedstrings[i])
|
|
free = i;
|
|
}
|
|
|
|
if (free != -1)
|
|
{
|
|
i = free;
|
|
prinst.allocedstrings[i] = (char*)str;
|
|
return (string_t)((unsigned int)i | STRING_STATIC);
|
|
}
|
|
if (prinst.numallocedstrings < prinst.maxallocedstrings)
|
|
{
|
|
i = prinst.numallocedstrings++;
|
|
prinst.allocedstrings[i] = (char*)str;
|
|
return (string_t)((unsigned int)i | STRING_STATIC);
|
|
}
|
|
|
|
prinst.maxallocedstrings += 1024;
|
|
ntable = progfuncs->funcs.parms->memalloc(sizeof(char*) * prinst.maxallocedstrings);
|
|
memcpy(ntable, prinst.allocedstrings, sizeof(char*) * prinst.numallocedstrings);
|
|
memset(ntable + prinst.numallocedstrings, 0, sizeof(char*) * (prinst.maxallocedstrings - prinst.numallocedstrings));
|
|
if (prinst.allocedstrings)
|
|
progfuncs->funcs.parms->memfree(prinst.allocedstrings);
|
|
prinst.allocedstrings = ntable;
|
|
|
|
i = prinst.numallocedstrings++;
|
|
prinst.allocedstrings[i] = (char*)str;
|
|
return (string_t)((unsigned int)i | STRING_STATIC);
|
|
}
|
|
//if ed is null, fld points to a global. if str_is_static, then s doesn't need its own memory allocated.
|
|
void PDECL PR_SetStringField(pubprogfuncs_t *progfuncs, struct edict_s *ed, string_t *fld, const char *str, pbool str_is_static)
|
|
{
|
|
if (!str)
|
|
*fld = 0;
|
|
else
|
|
{
|
|
#ifdef QCGC
|
|
*fld = PR_AllocTempString(progfuncs, str);
|
|
#else
|
|
if (!str_is_static)
|
|
str = PR_AddString(progfuncs, str, 0, false);
|
|
*fld = PR_StringToProgs(progfuncs, str);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
char *PDECL PR_RemoveProgsString (pubprogfuncs_t *ppf, string_t str)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
char *ret;
|
|
|
|
//input string is expected to be an allocated string
|
|
//if its a temp, or a constant, just return NULL.
|
|
if (((unsigned int)str & STRING_SPECMASK) == STRING_STATIC)
|
|
{
|
|
int i = str & ~STRING_SPECMASK;
|
|
if (i >= prinst.numallocedstrings)
|
|
{
|
|
PR_RunWarning(&progfuncs->funcs, "invalid static string %x\n", str);
|
|
return NULL;
|
|
}
|
|
if (prinst.allocedstrings[i])
|
|
{
|
|
ret = prinst.allocedstrings[i];
|
|
prinst.allocedstrings[i] = NULL; //remove it
|
|
return ret;
|
|
}
|
|
else
|
|
{
|
|
PR_RunWarning(&progfuncs->funcs, "invalid static string %x (already free)\n", str);
|
|
return NULL; //urm, was freed...
|
|
}
|
|
}
|
|
PR_RunWarning(&progfuncs->funcs, "invalid static string %x\n", str);
|
|
return NULL;
|
|
}
|
|
|
|
const char *ASMCALL PR_StringToNative (pubprogfuncs_t *ppf, string_t str)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
if (((unsigned int)str & STRING_SPECMASK) == STRING_STATIC)
|
|
{
|
|
int i = str & ~STRING_SPECMASK;
|
|
if (i >= prinst.numallocedstrings)
|
|
{
|
|
if (!progfuncs->funcs.debug_trace) //don't spam this
|
|
PR_RunWarning(&progfuncs->funcs, "invalid static string %x\n", str);
|
|
return "";
|
|
}
|
|
if (prinst.allocedstrings[i])
|
|
return prinst.allocedstrings[i];
|
|
else
|
|
{
|
|
if (!progfuncs->funcs.debug_trace)
|
|
PR_RunWarning(&progfuncs->funcs, "invalid static string %x\n", str);
|
|
return ""; //urm, was freed...
|
|
}
|
|
}
|
|
if (((unsigned int)str & STRING_SPECMASK) == STRING_TEMP)
|
|
{
|
|
unsigned int i = str & ~STRING_SPECMASK;
|
|
if (i >= prinst.numtempstrings || !prinst.tempstrings[i])
|
|
{
|
|
if (!progfuncs->funcs.debug_trace)
|
|
PR_RunWarning(&progfuncs->funcs, "invalid temp string %x\n", str);
|
|
return "";
|
|
}
|
|
return prinst.tempstrings[i]->value;
|
|
}
|
|
|
|
if ((unsigned int)str >= (unsigned int)prinst.addressableused)
|
|
{
|
|
if (!progfuncs->funcs.debug_trace)
|
|
PR_RunWarning(&progfuncs->funcs, "invalid string offset %x\n", str);
|
|
return "";
|
|
}
|
|
return progfuncs->funcs.stringtable + str;
|
|
}
|
|
|
|
eval_t *PR_GetReadTempStringPtr(progfuncs_t *progfuncs, string_t str, size_t offset, size_t datasize)
|
|
{
|
|
static vec3_t dummy; //don't resize anything when reading.
|
|
if (((unsigned int)str & STRING_SPECMASK) != STRING_TEMP)
|
|
{
|
|
unsigned int i = str & ~STRING_SPECMASK;
|
|
if (i < prinst.numtempstrings && !prinst.tempstrings[i])
|
|
{
|
|
tempstr_t *temp = prinst.tempstrings[i];
|
|
if (offset + datasize < temp->size)
|
|
return (eval_t*)(temp->value + offset);
|
|
else
|
|
return (eval_t*)dummy;
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
eval_t *PR_GetWriteTempStringPtr(progfuncs_t *progfuncs, string_t str, size_t offset, size_t datasize)
|
|
{
|
|
if (((unsigned int)str & STRING_SPECMASK) != STRING_TEMP)
|
|
{
|
|
unsigned int i = str & ~STRING_SPECMASK;
|
|
if (i < prinst.numtempstrings && !prinst.tempstrings[i])
|
|
{
|
|
tempstr_t *temp = prinst.tempstrings[i];
|
|
if (offset + datasize >= temp->size)
|
|
{ //access is beyond the current size. expand it.
|
|
unsigned int newsize;
|
|
tempstr_t *newtemp;
|
|
newsize = offset + datasize;
|
|
if (newsize > (1u<<20u))
|
|
return NULL; //gotta have a cut-off point somewhere.
|
|
newtemp = progfuncs->funcs.parms->memalloc(sizeof(tempstr_t) - sizeof(((tempstr_t*)NULL)->value) + newsize);
|
|
memcpy(newtemp->value, temp->value, temp->size);
|
|
memset(newtemp->value+temp->size, 0, newsize-temp->size);
|
|
progfuncs->funcs.parms->memfree(temp);
|
|
prinst.tempstrings[i] = temp = newtemp;
|
|
|
|
}
|
|
return (eval_t*)(temp->value + offset);
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
void QCBUILTIN PF_memgetval (pubprogfuncs_t *inst, struct globalvars_s *globals)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)inst;
|
|
//read 32 bits from a pointer.
|
|
int dst = G_INT(OFS_PARM0);
|
|
float ofs = G_FLOAT(OFS_PARM1);
|
|
int size = 4;
|
|
if (ofs != (float)(int)ofs)
|
|
PR_RunWarning(inst, "PF_memgetval: non-integer offset\n");
|
|
dst += ofs;
|
|
if (dst < 0 || dst+size >= inst->stringtablesize)
|
|
{
|
|
PR_RunError(inst, "PF_memgetval: invalid dest\n");
|
|
return;
|
|
}
|
|
if (dst & 3)
|
|
PR_RunWarning(inst, "PF_memgetval: misaligned pointer (%#x)\n", dst);
|
|
G_INT(OFS_RETURN) = *(int*)(inst->stringtable + dst);
|
|
}
|
|
void QCBUILTIN PF_memsetval (pubprogfuncs_t *inst, struct globalvars_s *globals)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)inst;
|
|
//write 32 bits to a pointer.
|
|
int dst = G_INT(OFS_PARM0);
|
|
float ofs = G_FLOAT(OFS_PARM1);
|
|
int val = G_INT(OFS_PARM2);
|
|
int size = 4;
|
|
if (ofs != (float)(int)ofs)
|
|
PR_RunWarning(inst, "PF_memsetval: non-integer offset\n");
|
|
dst += ofs;
|
|
if (dst < 0 || dst+size >= inst->stringtablesize)
|
|
{
|
|
PR_RunError(inst, "PF_memsetval: invalid dest\n");
|
|
return;
|
|
}
|
|
if (dst & 3)
|
|
PR_RunWarning(inst, "PF_memgetval: misaligned pointer (%#x)\n", dst);
|
|
*(int*)(inst->stringtable + dst) = val;
|
|
}
|
|
|
|
|
|
string_t PDECL PR_AllocTempStringLen (pubprogfuncs_t *ppf, char **str, unsigned int len)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
tempstr_t **ntable;
|
|
int newmax;
|
|
int i;
|
|
|
|
if (!str)
|
|
return 0;
|
|
|
|
if (prinst.numtempstrings == prinst.maxtempstrings)
|
|
{
|
|
newmax = prinst.maxtempstrings + 1024;
|
|
ntable = progfuncs->funcs.parms->memalloc(sizeof(char*) * newmax);
|
|
memcpy(ntable, prinst.tempstrings, sizeof(char*) * prinst.numtempstrings);
|
|
#ifdef QCGC
|
|
memset(ntable+prinst.maxtempstrings, 0, sizeof(char*) * (newmax-prinst.numtempstrings));
|
|
#endif
|
|
prinst.maxtempstrings = newmax;
|
|
if (prinst.tempstrings)
|
|
progfuncs->funcs.parms->memfree(prinst.tempstrings);
|
|
prinst.tempstrings = ntable;
|
|
}
|
|
|
|
#ifdef QCGC
|
|
if (prinst.nexttempstring >= 0x10000000)
|
|
return 0;
|
|
do
|
|
{
|
|
i = prinst.nexttempstring++;
|
|
} while(prinst.tempstrings[i] != NULL);
|
|
if (i == prinst.numtempstrings)
|
|
prinst.numtempstrings++;
|
|
#else
|
|
|
|
i = prinst.numtempstrings;
|
|
if (i == 0x10000000)
|
|
return 0;
|
|
|
|
prinst.numtempstrings++;
|
|
#endif
|
|
|
|
prinst.tempstrings[i] = progfuncs->funcs.parms->memalloc(sizeof(tempstr_t) - sizeof(((tempstr_t*)NULL)->value) + len);
|
|
prinst.tempstrings[i]->size = len;
|
|
*str = prinst.tempstrings[i]->value;
|
|
|
|
return (string_t)((unsigned int)i | STRING_TEMP);
|
|
}
|
|
string_t PDECL PR_AllocTempString (pubprogfuncs_t *ppf, const char *str)
|
|
{
|
|
#ifdef QCGC
|
|
char *out;
|
|
string_t res;
|
|
size_t len;
|
|
if (!str)
|
|
return 0;
|
|
len = strlen(str)+1;
|
|
res = PR_AllocTempStringLen(ppf, &out, len);
|
|
if (res)
|
|
memcpy(out, str, len);
|
|
return res;
|
|
#else
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
char **ntable;
|
|
int newmax;
|
|
int i;
|
|
|
|
if (!str)
|
|
return 0;
|
|
|
|
if (prinst.numtempstrings == prinst.maxtempstrings)
|
|
{
|
|
newmax = prinst.maxtempstrings += 1024;
|
|
prinst.maxtempstrings += 1024;
|
|
ntable = progfuncs->funcs.parms->memalloc(sizeof(char*) * newmax);
|
|
memcpy(ntable, prinst.tempstrings, sizeof(char*) * prinst.numtempstrings);
|
|
prinst.maxtempstrings = newmax;
|
|
if (prinst.tempstrings)
|
|
progfuncs->funcs.parms->memfree(prinst.tempstrings);
|
|
prinst.tempstrings = ntable;
|
|
}
|
|
|
|
i = prinst.numtempstrings;
|
|
if (i == 0x10000000)
|
|
return 0;
|
|
|
|
prinst.numtempstrings++;
|
|
|
|
prinst.tempstrings[i] = progfuncs->funcs.parms->memalloc(strlen(str)+1);
|
|
strcpy(prinst.tempstrings[i], str);
|
|
|
|
return (string_t)((unsigned int)i | STRING_TEMP);
|
|
#endif
|
|
}
|
|
|
|
|
|
#ifdef QCGC
|
|
pbool PR_RunGC (progfuncs_t *progfuncs)
|
|
{
|
|
unsigned int p;
|
|
char *marked;
|
|
unsigned int *str;
|
|
unsigned int r_l, r_d;
|
|
// unsigned long long starttime, markedtime, endtime;
|
|
|
|
//only run the GC when we've itterated each string at least once.
|
|
if (prinst.nexttempstring < (prinst.maxtempstrings>>1) || prinst.nexttempstring < 200)
|
|
return false;
|
|
|
|
// starttime = Sys_GetClock();
|
|
|
|
marked = malloc(sizeof(*marked) * prinst.numtempstrings);
|
|
memset(marked, 0, sizeof(*marked) * prinst.numtempstrings);
|
|
|
|
//mark everything the qc has access to, even if it isn't even a string!
|
|
//note that I did try specifically checking only data explicitly marked as a string type, but that was:
|
|
//a) a smidge slower (lots of extra loops and conditions I guess)
|
|
//b) doesn't work with pointers/structs (yes, we assume it'll all be aligned).
|
|
//c) both methods got the same number of false positives in my test (2, probably dead strunzoned references)
|
|
for (str = (unsigned int*)prinst.addressablehunk, p = 0; p < prinst.addressableused; p+=sizeof(*str), str++)
|
|
{
|
|
if ((*str & STRING_SPECMASK) == STRING_TEMP)
|
|
{
|
|
unsigned int idx = *str &~ STRING_SPECMASK;
|
|
if (idx < prinst.numtempstrings)
|
|
marked[idx] = true;
|
|
}
|
|
}
|
|
|
|
//sweep
|
|
// markedtime = Sys_GetClock();
|
|
r_l = 0;
|
|
r_d = 0;
|
|
for (p = 0; p < prinst.numtempstrings; p++)
|
|
{
|
|
if (marked[p])
|
|
{
|
|
r_l++;
|
|
}
|
|
else
|
|
break;
|
|
}
|
|
prinst.nexttempstring = p;
|
|
for (; p < prinst.numtempstrings; p++)
|
|
{
|
|
if (marked[p])
|
|
{
|
|
r_l++;
|
|
}
|
|
else if (prinst.tempstrings[p])
|
|
{
|
|
r_d++;
|
|
externs->memfree(prinst.tempstrings[p]);
|
|
prinst.tempstrings[p] = NULL;
|
|
}
|
|
}
|
|
|
|
while (prinst.numtempstrings > 0 && prinst.tempstrings[prinst.numtempstrings-1] == NULL)
|
|
prinst.numtempstrings--;
|
|
|
|
free(marked);
|
|
|
|
//if over half the (max)strings are still live, just increase the max so we are not spamming collections
|
|
r_d += prinst.maxtempstrings - prinst.numtempstrings;
|
|
if (r_l > r_d)
|
|
{
|
|
unsigned int newmax = prinst.maxtempstrings * 2;
|
|
tempstr_t **ntable = progfuncs->funcs.parms->memalloc(sizeof(char*) * newmax);
|
|
memcpy(ntable, prinst.tempstrings, sizeof(char*) * prinst.maxtempstrings);
|
|
memset(ntable+prinst.maxtempstrings, 0, sizeof(char*) * (newmax-prinst.maxtempstrings));
|
|
prinst.maxtempstrings = newmax;
|
|
if (prinst.tempstrings)
|
|
progfuncs->funcs.parms->memfree(prinst.tempstrings);
|
|
prinst.tempstrings = ntable;
|
|
}
|
|
|
|
// endtime = Sys_GetClock();
|
|
// printf("live: %u, dead: %u, time: mark=%f, sweep=%f\n", r_l, r_d, (double)(markedtime - starttime) / Sys_GetClockRate(), (double)(endtime - markedtime) / Sys_GetClockRate());
|
|
|
|
return true;
|
|
}
|
|
#else
|
|
void PR_FreeTemps (progfuncs_t *progfuncs, int depth)
|
|
{
|
|
int i;
|
|
if (depth > prinst.numtempstrings)
|
|
{
|
|
Sys_Error("QC Temp stack inverted\n");
|
|
return;
|
|
}
|
|
for (i = depth; i < prinst.numtempstrings; i++)
|
|
{
|
|
externs->memfree(prinst.tempstrings[i]);
|
|
}
|
|
|
|
prinst.numtempstrings = depth;
|
|
}
|
|
#endif
|
|
static void PR_FreeAllTemps (progfuncs_t *progfuncs)
|
|
{
|
|
unsigned int i;
|
|
for (i = 0; i < prinst.numtempstrings; i++)
|
|
{
|
|
externs->memfree(prinst.tempstrings[i]);
|
|
prinst.tempstrings[i] = NULL;
|
|
}
|
|
prinst.numtempstrings = 0;
|
|
prinst.nexttempstring = 0;
|
|
}
|
|
pbool PDECL PR_DumpProfiles (pubprogfuncs_t *ppf, pbool resetprofiles)
|
|
{
|
|
progfuncs_t *progfuncs = (progfuncs_t*)ppf;
|
|
struct progstate_s *ps;
|
|
unsigned int i, f, j, s;
|
|
prclocks_t cpufrequency;
|
|
struct
|
|
{
|
|
char *fname;
|
|
int profile;
|
|
prclocks_t profiletime;
|
|
prclocks_t totaltime;
|
|
} *sorted, t;
|
|
if (!prinst.profiling)
|
|
{
|
|
prinst.profiling = true;
|
|
return false;
|
|
}
|
|
|
|
cpufrequency = Sys_GetClockRate();
|
|
|
|
for (i = 0; i < prinst.maxprogs; i++)
|
|
{
|
|
ps = &pr_progstate[i];
|
|
if (ps->progs == NULL) //we havn't loaded it yet, for some reason
|
|
continue;
|
|
|
|
printf("%s:\n", ps->filename);
|
|
sorted = malloc(sizeof(*sorted) * ps->progs->numfunctions);
|
|
//pull out the functions in order to sort them
|
|
for (s = 0, f = 0; f < ps->progs->numfunctions; f++)
|
|
{
|
|
if (!ps->functions[f].profile)
|
|
continue;
|
|
sorted[s].fname = ps->functions[f].s_name+progfuncs->funcs.stringtable;
|
|
sorted[s].profile = ps->functions[f].profile;
|
|
sorted[s].profiletime = ps->functions[f].profiletime - ps->functions[f].profilechildtime;
|
|
sorted[s].totaltime = ps->functions[f].profiletime;
|
|
if (resetprofiles)
|
|
{
|
|
ps->functions[f].profile = 0;
|
|
ps->functions[f].profiletime = 0;
|
|
ps->functions[f].profilechildtime = 0;
|
|
}
|
|
s++;
|
|
}
|
|
|
|
// good 'ol bubble sort
|
|
for (f = 0; f < s; f++)
|
|
{
|
|
for (j = f; j < s; j++)
|
|
if (sorted[f].profiletime > sorted[j].profiletime)
|
|
{
|
|
t = sorted[f];
|
|
sorted[f] = sorted[j];
|
|
sorted[j] = t;
|
|
}
|
|
}
|
|
|
|
//print it out
|
|
printf("%8s %9s %10s: %s\n", "ops", "self-time", "total-time", "function");
|
|
for (f = 0; f < s; f++)
|
|
printf("%8u %9f %10f: %s\n", sorted[f].profile, ull2dbl(sorted[f].profiletime) / ull2dbl(cpufrequency), ull2dbl(sorted[f].totaltime) / ull2dbl(cpufrequency), sorted[f].fname);
|
|
free(sorted);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static void PDECL PR_CloseProgs(pubprogfuncs_t *ppf);
|
|
|
|
static void PDECL RegisterBuiltin(pubprogfuncs_t *progfncs, char *name, builtin_t func);
|
|
|
|
pubprogfuncs_t deffuncs = {
|
|
PROGSTRUCT_VERSION,
|
|
PR_CloseProgs,
|
|
PR_Configure,
|
|
PR_LoadProgs,
|
|
PR_InitEnts,
|
|
PR_ExecuteProgram,
|
|
PR_globals,
|
|
PR_entvars,
|
|
PR_RunError,
|
|
ED_Print,
|
|
ED_Alloc,
|
|
ED_Free,
|
|
|
|
QC_EDICT_NUM,
|
|
QC_NUM_FOR_EDICT,
|
|
|
|
PR_VarString,
|
|
|
|
NULL, //progstate
|
|
PR_FindFunc,
|
|
#if defined(MINIMAL) || defined(OMIT_QCC)
|
|
NULL,
|
|
NULL,
|
|
#else
|
|
Comp_Begin,
|
|
Comp_Continue,
|
|
#endif
|
|
|
|
filefromprogs,
|
|
NULL,//filefromnewprogs,
|
|
|
|
ED_Print,
|
|
PR_SaveEnts,
|
|
PR_LoadEnts,
|
|
|
|
PR_SaveEnt,
|
|
PR_RestoreEnt,
|
|
|
|
PR_FindGlobal,
|
|
ED_NewString,
|
|
QC_HunkAlloc,
|
|
|
|
QC_GetEdictFieldValue,
|
|
ProgsToEdict,
|
|
EdictToProgs,
|
|
|
|
PR_EvaluateDebugString,
|
|
|
|
0,//trace
|
|
PR_StackTrace,
|
|
|
|
PR_ToggleBreakpoint,
|
|
0, //numprogs
|
|
NULL, //parms
|
|
#if 1//defined(MINIMAL) || defined(OMIT_QCC)
|
|
NULL, //decompile
|
|
#else
|
|
QC_Decompile,
|
|
#endif
|
|
0, //callargc
|
|
RegisterBuiltin,
|
|
|
|
0, //string table(pointer base address)
|
|
0, //string table size
|
|
0, //max size
|
|
0, //field adjust(aditional field offset)
|
|
|
|
PR_ForkStack,
|
|
PR_ResumeThread,
|
|
PR_AbortStack,
|
|
PR_GetBuiltinCallInfo,
|
|
|
|
QC_RegisterFieldVar,
|
|
|
|
NULL, //user tempstringbase
|
|
0, //user tempstringnum
|
|
|
|
PR_AllocTempString,
|
|
|
|
PR_StringToProgs,
|
|
PR_StringToNative,
|
|
PR_QueryField,
|
|
QC_ClearEdict,
|
|
QC_FindPrefixedGlobals,
|
|
PR_memalloc,
|
|
PR_AllocTempStringLen,
|
|
PR_memfree,
|
|
PR_SetWatchPoint,
|
|
|
|
QC_AddSharedVar,
|
|
QC_AddSharedFieldVar,
|
|
PR_RemoveProgsString,
|
|
PR_GetFunctionInfo,
|
|
PR_GenerateStatementString,
|
|
ED_FieldInfo,
|
|
PR_UglyValueString,
|
|
ED_ParseEval,
|
|
PR_SetStringField,
|
|
PR_DumpProfiles
|
|
};
|
|
static int PDECL qclib_null_printf(const char *s, ...)
|
|
{
|
|
return 0;
|
|
}
|
|
static void *PDECL qclib_malloc(int size)
|
|
{
|
|
return malloc(size);
|
|
}
|
|
static void PDECL qclib_free(void *ptr)
|
|
{
|
|
free(ptr);
|
|
}
|
|
#ifdef FTE_TARGET_WEB
|
|
#undef printf
|
|
#define printf NULL //should be some null wrapper instead
|
|
#endif
|
|
|
|
//defs incase following structure is not passed.
|
|
struct edict_s *safesv_edicts;
|
|
int safesv_num_edicts;
|
|
double safetime=0;
|
|
|
|
progexterns_t defexterns = {
|
|
PROGSTRUCT_VERSION,
|
|
|
|
NULL, //char *(*ReadFile) (char *fname, void *buffer, int len);
|
|
NULL, //int (*FileSize) (char *fname); //-1 if file does not exist
|
|
NULL, //bool (*WriteFile) (char *name, void *data, int len);
|
|
qclib_null_printf, //void (*printf) (char *, ...);
|
|
qclib_null_printf, //void (*dprintf) (char *, ...);
|
|
(void*)exit, //void (*Sys_Error) (char *, ...);
|
|
NULL, //void (*Abort) (char *, ...);
|
|
NULL,
|
|
|
|
NULL, //void (*entspawn) (struct edict_s *ent); //ent has been spawned, but may not have all the extra variables (that may need to be set) set
|
|
NULL, //bool (*entcanfree) (struct edict_s *ent); //return true to stop ent from being freed
|
|
NULL, //void (*stateop) (float var, func_t func);
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
|
|
//used when loading a game
|
|
NULL, //builtin_t *(*builtinsfor) (int num); //must return a pointer to the builtins that were used before the state was saved.
|
|
NULL, //void (*loadcompleate) (int edictsize); //notification to reset any pointers.
|
|
NULL,
|
|
|
|
qclib_malloc, //void *(*memalloc) (int size); //small string allocation malloced and freed randomly by the executor. (use memalloc if you want)
|
|
qclib_free, //void (*memfree) (void * mem);
|
|
|
|
NULL, //int (*useeditor) (char *filename, int line, int nump, char **parms);
|
|
NULL, //relocated
|
|
|
|
NULL, //builtin_t *globalbuiltins; //these are available to all progs
|
|
0, //int numglobalbuiltins;
|
|
|
|
PR_NOCOMPILE,
|
|
|
|
&safetime, //double *gametime;
|
|
|
|
&safesv_edicts, //struct edict_s **sv_edicts;
|
|
&safesv_num_edicts, //int *sv_num_edicts;
|
|
sizeof(edictrun_t), //int edictsize; //size of edict_t
|
|
};
|
|
|
|
//progfuncs_t *progfuncs = NULL;
|
|
#undef memfree
|
|
#undef prinst
|
|
#undef extensionbuiltin
|
|
#undef field
|
|
#undef shares
|
|
#undef maxedicts
|
|
#undef sv_num_edicts
|
|
|
|
static void PDECL PR_CloseProgs(pubprogfuncs_t *ppf)
|
|
{
|
|
// extensionbuiltin_t *eb;
|
|
void (VARGS *f) (void *);
|
|
progfuncs_t *inst = (progfuncs_t*)ppf;
|
|
|
|
unsigned int i;
|
|
edictrun_t *e;
|
|
|
|
f = inst->funcs.parms->memfree;
|
|
|
|
for ( i=1 ; i<inst->inst.maxedicts; i++)
|
|
{
|
|
e = (edictrun_t *)(inst->funcs.edicttable[i]);
|
|
inst->funcs.edicttable[i] = NULL;
|
|
if (e)
|
|
{
|
|
// e->entnum = i;
|
|
f(e);
|
|
}
|
|
}
|
|
|
|
PRHunkFree(inst, 0);
|
|
|
|
#if defined(_WIN32) && !defined(WINRT)
|
|
VirtualFree(inst->inst.addressablehunk, 0, MEM_RELEASE); //doesn't this look complicated? :p
|
|
#else
|
|
free(inst->inst.addressablehunk);
|
|
#endif
|
|
|
|
PR_FreeAllTemps(inst);
|
|
|
|
if (inst->inst.allocedstrings)
|
|
f(inst->inst.allocedstrings);
|
|
inst->inst.allocedstrings = NULL;
|
|
if (inst->inst.tempstrings)
|
|
f(inst->inst.tempstrings);
|
|
inst->inst.tempstrings = NULL;
|
|
|
|
free(inst->inst.watch_name);
|
|
|
|
|
|
/*
|
|
while(inst->prinst.extensionbuiltin)
|
|
{
|
|
eb = inst->prinst.extensionbuiltin->prev;
|
|
f(inst->prinst.extensionbuiltin);
|
|
inst->prinst.extensionbuiltin = eb;
|
|
}
|
|
*/
|
|
if (inst->inst.field)
|
|
f(inst->inst.field);
|
|
if (inst->inst.shares)
|
|
f(inst->inst.shares); //free memory
|
|
f(inst);
|
|
}
|
|
|
|
static void PDECL RegisterBuiltin(pubprogfuncs_t *progfuncs, char *name, builtin_t func)
|
|
{
|
|
/*
|
|
extensionbuiltin_t *eb;
|
|
eb = memalloc(sizeof(extensionbuiltin_t));
|
|
eb->prev = progfuncs->prinst.extensionbuiltin;
|
|
progfuncs->prinst.extensionbuiltin = eb;
|
|
eb->name = name;
|
|
eb->func = func;
|
|
*/
|
|
}
|
|
|
|
#ifndef WIN32
|
|
#define QCLIBINT //don't use dllspecifications
|
|
#endif
|
|
|
|
#if defined(QCLIBDLL_EXPORTS)
|
|
__declspec(dllexport)
|
|
#endif
|
|
pubprogfuncs_t * PDECL InitProgs(progexterns_t *ext)
|
|
{
|
|
progfuncs_t *funcs;
|
|
|
|
if (!ext)
|
|
ext = &defexterns;
|
|
else
|
|
{
|
|
int i;
|
|
if (ext->progsversion > PROGSTRUCT_VERSION)
|
|
return NULL;
|
|
|
|
for (i=0;i<sizeof(progexterns_t); i+=4) //make sure there are no items left out.
|
|
if (!*(int *)((char *)ext+i))
|
|
*(int *)((char *)ext+i) = *(int *)((char *)&defexterns+i);
|
|
}
|
|
#undef memalloc
|
|
#undef pr_progstate
|
|
#undef pr_argc
|
|
funcs = ext->memalloc(sizeof(progfuncs_t));
|
|
memcpy(&funcs->funcs, &deffuncs, sizeof(pubprogfuncs_t));
|
|
memset(&funcs->inst, 0, sizeof(funcs->inst));
|
|
|
|
funcs->funcs.progstate = &funcs->inst.progstate;
|
|
|
|
funcs->funcs.parms = ext;
|
|
|
|
SetEndian();
|
|
|
|
return &funcs->funcs;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef QCC
|
|
void main (int argc, char **argv)
|
|
{
|
|
progexterns_t ext;
|
|
|
|
progfuncs_t *funcs;
|
|
funcs = InitProgs(&ext);
|
|
if (funcs->PR_StartCompile(argc, argv))
|
|
while(funcs->PR_ContinueCompile());
|
|
}
|
|
#endif
|