samp-re/plugin.c

348 lines
8.0 KiB
C
Raw Normal View History

/* vim: set filetype=c ts=8 noexpandtab: */
#include "common.h"
#include <windows.h>
#include "io.h"
#include "vendor/SDK/amx/amx.h"
#include "vendor/SDK/plugincommon.h"
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
typedef void (*logprintf_t)(char* format, ...);
logprintf_t logprintf;
extern void *pAMXFunctions;
EXPECT_SIZE(char, 1);
EXPECT_SIZE(short, 2);
EXPECT_SIZE(int, 4);
EXPECT_SIZE(cell, 4);
EXPECT_SIZE(float, 4);
/*in cells*/
#define STACK_HEAP_SIZE 1024
AMX *amx;
struct FAKEAMX_DATA {
union {
cell ascell[144];
char aschr[144 * sizeof(cell)];
float asflt[144];
} a144;
cell _stackheap[STACK_HEAP_SIZE];
} fakeamx_data;
#define basea ((int) &fakeamx_data)
#define buf144a ((int) &fakeamx_data.a144 - basea)
#define buf144 (fakeamx_data.a144.ascell)
#define cbuf144 fakeamx_data.a144.aschr
#define fbuf144 fakeamx_data.a144.asflt
union NCDATA {
cell asint[20];
float asflt[20];
} nc_params;
struct NATIVE {
char *name;
AMX_NATIVE fp;
};
struct NATIVE natives[] = {
/*this needs to be synced with NC_ definitions*/
{ "CreateObject", 0 },
{ "DestroyObject", 0 },
{ "SetObjectMaterial", 0 },
{ "SetObjectMaterialText", 0 },
{ "SetObjectPos", 0 },
{ "SetObjectRot", 0 },
{ "AddPlayerClass", 0 },
{ "EditObject", 0 },
{ "CreateVehicle", 0 },
{ "DestroyVehicle", 0 },
{ "SetWorldTime", 0 },
{ "SetWeather", 0 },
};
#define NUMNATIVES (sizeof(natives)/sizeof(struct NATIVE))
#define NC_CreateObject 0
#define NC_DestroyObject 1
#define NC_SetObjectMaterial 2
#define NC_SetObjectMaterialText 3
#define NC_SetObjectPos 4
#define NC_SetObjectRot 5
#define NC_AddPlayerClass 6
#define NC_EditObject 7
#define NC_CreateVehicle 8
#define NC_DestroyVehicle 9
#define NC_SetWorldTime 10
#define NC_SetWeather 11
PLUGIN_EXPORT unsigned int PLUGIN_CALL Supports()
{
return SUPPORTS_VERSION | SUPPORTS_AMX_NATIVES | SUPPORTS_PROCESS_TICK;
}
static
void gen_gamemode_script()
{
#define cellsize 4
#define PUBLICS 0
FILE *f;
int tmp, size;
#if PUBLICS > 0
char pubtbl[PUBLICS * 8];
#endif
char nattbl[NUMNATIVES * 8];
int pubidx, i;
int cod, cip, dat;
union {
char i1;
short i2;
int i4;
char buf[4];
} data;
struct {
int addr, sym;
} *table_entry;
/*error handling is.. missing*/
f = fopen("samp-re.amx", "wb");
/*header*/
fwrite(data.buf, 4, 1, f); /*size, to be set when done*/
data.i2 = 0xF1E0;
fwrite(data.buf, 2, 1, f); /*magic*/
data.i1 = 0x8;
fwrite(data.buf, 1, 1, f); /*file version*/
data.i1 = 0x8;
fwrite(data.buf, 1, 1, f); /*amx version*/
data.i2 = 0x14;
fwrite(data.buf, 2, 1, f); /*flags*/
data.i2 = 0x8;
fwrite(data.buf, 2, 1, f); /*defsize*/
data.i4 = 0; /*done later*/
fwrite(data.buf, 4, 1, f); /*cod*/
data.i4 = 0; /*done later*/
fwrite(data.buf, 4, 1, f); /*dat*/
data.i4 = 0; /*done later*/
fwrite(data.buf, 4, 1, f); /*hea*/
data.i4 = 0; /*done later*/
fwrite(data.buf, 4, 1, f); /*stp*/
data.i4 = 0; /*done later*/
fwrite(data.buf, 4, 1, f); /*cip*/
data.i4 = 0x38;
fwrite(data.buf, 4, 1, f); /*publictable*/
#if PUBLICS > 0
data.i4 += sizeof(pubtbl);
#endif
fwrite(data.buf, 4, 1, f); /*nativetable*/
data.i4 += sizeof(nattbl);
fwrite(data.buf, 4, 1, f); /*libraries*/
fwrite(data.buf, 4, 1, f); /*pubvars*/
fwrite(data.buf, 4, 1, f); /*tags*/
fwrite(data.buf, 4, 1, f); /*nametable*/
size = 0x38; /*header*/
/*pubtbl, dummy at this point*/
#if PUBLICS > 0
fwrite(pubtbl, sizeof(pubtbl), 1, f);
size += sizeof(pubtbl);
#endif
/*nattbl, dummy at this point*/
fwrite(nattbl, sizeof(nattbl), 1, f);
size += sizeof(nattbl);
/*namtbl*/
data.i2 = 31;
fwrite(data.buf, 2, 1, f); /*max name len (I suppose?)*/
size += 2;
pubidx = 0;
#define PUBLIC(NAME) \
tmp=strlen(NAME)+1;\
fwrite(NAME,tmp,1,f);\
table_entry->addr=(2+pubidx*4)*cellsize;\
table_entry->sym=size;\
table_entry++;\
pubidx++;\
size+=tmp;
/*first [PUBLICS] amount of natives should be the names of the
native plugin functions to passthrough callbacks to*/
#if PUBLICS > 0
table_entry = (void*) pubtbl;
PUBLIC("MM");
PUBLIC("dummies");
#endif
table_entry = (void*) nattbl;
for (i = 0; i < NUMNATIVES; i++) {
tmp = strlen(natives[i].name) + 1;
fwrite(natives[i].name, tmp, 1, f);
table_entry->addr = 0;
table_entry->sym = size;
table_entry++;
size += tmp;
}
#undef PUBLIC
cod = size;
/*code segment*/
cip = 0;
dat = 0;
data.i1 = 0x80; /*???*/
fwrite(data.buf, 1, 1, f);
data.i1 = 0x78; /*OP_HALT*/
fwrite(data.buf, 1, 1, f);
data.i1 = 0x00; /*param*/
fwrite(data.buf, 1, 1, f);
size += 3;
dat += 2;
/*publics*/
#if PUBLICS > 0
/*TODO: something makes the amx not accept this...*/
for (tmp = 0; tmp < PUBLICS; tmp++) {
data.buf[0] = 46; /*OP_PROC*/
data.buf[1] = 123; /*OP_SYSREC_C*/
data.buf[2] = tmp; /*name table entry*/
data.buf[3] = 48; /*OP_RETN*/
fwrite(data.buf, 4, 1, f);
size += 4;
dat += 4;
}
#endif
/*entrypoint*/
cip = (size - cod) * cellsize;
data.i1 = 46; /*OP_PROC*/
fwrite(data.buf, 1, 1, f);
data.i1 = 48; /*OP_RETN*/
fwrite(data.buf, 1, 1, f);
size += 2;
dat += 2;
dat = cod + dat * cellsize;
fseek(f, 0, SEEK_SET);
data.i4 = size;
fwrite(data.buf, 4, 1, f); /*size*/
fseek(f, 0xC, SEEK_SET);
data.i4 = cod;
fwrite(data.buf, 4, 1, f); /*cod*/
data.i4 = dat;
fwrite(data.buf, 4, 1, f); /*dat*/
fwrite(data.buf, 4, 1, f); /*hea*/
data.i4 += STACK_HEAP_SIZE * cellsize;
fwrite(data.buf, 4, 1, f); /*stp*/
data.i4 = cip;
fwrite(data.buf, 4, 1, f); /*cip*/
fseek(f, 0x38, SEEK_SET);
#if PUBLICS > 0
fwrite(pubtbl, sizeof(pubtbl), 1, f); /*pubtbl*/
#endif
fwrite(nattbl, sizeof(nattbl), 1, f); /*nattbl*/
fclose(f);
}
PLUGIN_EXPORT int PLUGIN_CALL Load(void **ppData)
{
void nethandler_init();
pAMXFunctions = ppData[PLUGIN_DATA_AMX_EXPORTS];
logprintf = (logprintf_t) ppData[PLUGIN_DATA_LOGPRINTF];
gen_gamemode_script();
nethandler_init();
return 1;
}
PLUGIN_EXPORT void PLUGIN_CALL Unload()
{
}
#define REGISTERNATIVE(X) {#X, X}
AMX_NATIVE_INFO PluginNatives[] =
{
{0, 0}
};
static
void collect_natives()
{
struct NATIVE *n = natives;
AMX_HEADER *header;
unsigned char *nattabl;
int i;
header = (AMX_HEADER*) amx->base;
nattabl = (unsigned char*) header + header->natives;
for (i = 0; i < NUMNATIVES; i++) {
/*skipping some steps here...*/
natives[i].fp = (AMX_NATIVE) *((int*) (nattabl + i * 8));
}
}
PLUGIN_EXPORT int PLUGIN_CALL AmxLoad(AMX *a)
{
AMX_HEADER *hdr;
int tmp;
amx = a;
collect_natives();
nc_params.asint[1] = 0;
nc_params.asflt[2] = 0.0f;
nc_params.asflt[3] = 0.0f;
nc_params.asflt[4] = 4.5f;
nc_params.asflt[5] = 0.0f;
nc_params.asint[6] = nc_params.asint[8] = nc_params.asint[10] = 0;
nc_params.asint[7] = nc_params.asint[9] = nc_params.asint[11] = 0;
natives[NC_AddPlayerClass].fp(amx, nc_params.asint);
/*relocate the data segment*/
hdr = (AMX_HEADER*) amx->base;
tmp = hdr->hea - hdr->dat;
if (tmp) {
/*data section will be cleared so whoever references it
will probably overwrite our precious data*/
logprintf("WARN: data section is not empty! (%d)", tmp);
}
tmp = hdr->stp - hdr->hea - STACK_HEAP_SIZE * sizeof(cell);
if (tmp) {
logprintf("ERR: stack/heap size mismatch! "
"(excess %d cells)",
tmp / sizeof(cell));
return 0;
}
/*copy content of stack/heap to our data segment (this is most
likely empty, but better safe than sorry)*/
memcpy(fakeamx_data._stackheap,
amx->base + hdr->hea,
STACK_HEAP_SIZE * sizeof(cell));
/*finally do the relocation*/
amx->data = (unsigned char*) &fakeamx_data;
/*adjust pointers since the data segment grew*/
tmp = (char*) &fakeamx_data._stackheap - (char*) &fakeamx_data;
amx->frm += tmp;
amx->hea += tmp;
amx->hlw += tmp;
amx->stk += tmp;
amx->stp += tmp;
/*samp core doesn't seem to use amx_SetString or amx_GetString
so the data offset needs to be adjusted*/
hdr->dat = (int) &fakeamx_data - (int) hdr;
/*this won't work on linux because linux builds have assertions
enabled, but this is only targeted for windows anyways*/
return amx_Register(a, PluginNatives, -1);
}
PLUGIN_EXPORT int PLUGIN_CALL AmxUnload(AMX *a)
{
return AMX_ERR_NONE;
}
PLUGIN_EXPORT void PLUGIN_CALL ProcessTick()
{
}