SAUtils/sa_scripting.cpp at main · AndroidModLoader/SAUtils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#include <mod/amlmod.h>
#include <mod/logger.h>

#include "sa_scripting.h"

// Our variables
GAME_SCRIPT_THREAD* gst;
char ScriptBuf[512];
uintptr_t *pdwParamVars[18];
extern void* pGameHandle;

// Functions
void (*ProcessOneCommand)(GAME_SCRIPT_THREAD*);

void InitializeSAScripting()
{
    gst = new GAME_SCRIPT_THREAD {0};
    SET_TO(ProcessOneCommand, aml->GetSym(pGameHandle, "_ZN14CRunningScript17ProcessOneCommandEv"));
}

inline uint8_t ExecuteScriptBuf()
{
    gst->dwScriptIP = (uintptr_t)ScriptBuf;
    ProcessOneCommand(gst);
    return gst->condResult;
}

int ScriptSACommandInner(const SCRIPT_COMMAND *pScriptCommand, va_list ap)
{
    const char *p = &pScriptCommand->params[0];
    memcpy(&ScriptBuf[0], &pScriptCommand->opCode, 2);
    int buf_pos = 2;
    uint16_t var_pos = 0;

    for(int i = 0; i < 18; ++i) gst->dwLocalVar[i] = 0;

    while(*p)
    {
        switch(*p)
        {
            case 'i':
            case 'd':
            {
                int i = va_arg(ap, int);
                ScriptBuf[buf_pos] = 0x01;
                memcpy(&ScriptBuf[++buf_pos], &i, 4);
                buf_pos += 4;
                break;
            }
            case 'b': // Handle booleans here
            {
                int i = va_arg(ap, int)!=0 ? 1 : 0;
                ScriptBuf[buf_pos] = 0x01;
                memcpy(&ScriptBuf[++buf_pos], &i, 4);
                buf_pos += 4;
                break;
            }
            case 'v':
            case 'p':
            {
                uint32_t *v = va_arg(ap, uint32_t*);
                ScriptBuf[buf_pos] = 0x03;
                pdwParamVars[var_pos] = (uintptr_t*)v;
                gst->dwLocalVar[var_pos] = *v;
                memcpy(&ScriptBuf[++buf_pos], &var_pos, 2);
                buf_pos += 2;
                ++var_pos;
                break;
            }
            case 'f':
            {
                float f = (float)va_arg(ap, double);
                ScriptBuf[buf_pos] = 0x06;
                memcpy(&ScriptBuf[++buf_pos], &f, 4);
                buf_pos += 4;
                break;
            }
            case 's':
            {
                char* sz = va_arg(ap, char*);
                unsigned char aLen = strlen(sz);
                ScriptBuf[buf_pos] = 0x0E;
                ScriptBuf[++buf_pos] = aLen;
                memcpy(&ScriptBuf[++buf_pos], sz, aLen);
                buf_pos += aLen;
                break;
            }
            case 'z':
            {
                ScriptBuf[buf_pos] = 0x00;
                ++buf_pos;
                break;
            }
            default:
            {
                logger->Error("An attempt to call opcode %04X with an unknown arguments description -> (%s)", pScriptCommand->opCode, pScriptCommand->params);
                va_end(ap);
                return 0;
            }
        }
        ++p;
    }
    va_end(ap);

    int result = ExecuteScriptBuf();
    if (var_pos)
    {
        for (int i = 0; i < var_pos; ++i) *pdwParamVars[i] = gst->dwLocalVar[i];
    }
    return result;
}

int ScriptSACommand(const SCRIPT_COMMAND *pScriptCommand, ...)
{
    va_list ap;
    va_start(ap, pScriptCommand);
    return ScriptSACommandInner(pScriptCommand, ap);
}