forked from ErikMinekus/sm-ripext
-
Notifications
You must be signed in to change notification settings - Fork 0
/
extension.h
232 lines (193 loc) · 5.99 KB
/
extension.h
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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
/**
* vim: set ts=4 :
* =============================================================================
* SourceMod REST in Pawn Extension
* Copyright 2017-2022 Erik Minekus
* =============================================================================
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any later
* version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License along with
* this program. If not, see <http:https://www.gnu.org/licenses/>.
*/
#ifndef _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
#define _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
#include <string>
#include <curl/curl.h>
#include <jansson.h>
#include <sm_stringhashmap.h>
#include <stdlib.h>
#include <string.h>
#include <uv.h>
#include "smsdk_ext.h"
#define SM_RIPEXT_CA_BUNDLE_PATH "configs/ripext/ca-bundle.crt"
#define SM_RIPEXT_USER_AGENT "sm-ripext/" SMEXT_CONF_VERSION
extern uv_loop_t *g_Loop;
typedef StringHashMap<std::string> HTTPHeaderMap;
class IHTTPContext
{
public:
virtual bool InitCurl() = 0;
virtual void OnCompleted() = 0;
virtual ~IHTTPContext() {}
CURL *curl;
};
struct CurlContext {
CurlContext(curl_socket_t socket) : socket(socket)
{
uv_poll_init_socket(g_Loop, &poll_handle, socket);
poll_handle.data = this;
}
void Destroy()
{
uv_poll_stop(&poll_handle);
uv_close((uv_handle_t *)&poll_handle, OnClosed);
}
static void OnClosed(uv_handle_t *handle)
{
delete (CurlContext *)handle->data;
}
curl_socket_t socket;
uv_poll_t poll_handle;
};
struct HTTPResponse {
long status = 0;
json_t *data = NULL;
Handle_t hndlData = BAD_HANDLE;
HTTPHeaderMap headers;
char *body = NULL;
size_t size = 0;
};
struct JSONObjectKeys {
JSONObjectKeys(json_t *object) : object(object), iter(json_object_iter(object)) {}
const char *GetKey()
{
return json_object_iter_key(iter);
}
void Next()
{
iter = json_object_iter_next(object, iter);
}
private:
json_t *object;
void *iter;
};
/**
* @brief Implementation of the REST in Pawn Extension.
* Note: Uncomment one of the pre-defined virtual functions in order to use it.
*/
class RipExt : public SDKExtension
{
public:
/**
* @brief This is called after the initial loading sequence has been processed.
*
* @param error Error message buffer.
* @param maxlength Size of error message buffer.
* @param late Whether or not the module was loaded after map load.
* @return True to succeed loading, false to fail.
*/
virtual bool SDK_OnLoad(char *error, size_t maxlength, bool late);
/**
* @brief This is called right before the extension is unloaded.
*/
virtual void SDK_OnUnload();
/**
* @brief This is called once all known extensions have been loaded.
* Note: It is is a good idea to add natives here, if any are provided.
*/
//virtual void SDK_OnAllLoaded();
/**
* @brief Called when the pause state is changed.
*/
//virtual void SDK_OnPauseChange(bool paused);
/**
* @brief this is called when Core wants to know if your extension is working.
*
* @param error Error message buffer.
* @param maxlength Size of error message buffer.
* @return True if working, false otherwise.
*/
//virtual bool QueryRunning(char *error, size_t maxlength);
public:
#if defined SMEXT_CONF_METAMOD
/**
* @brief Called when Metamod is attached, before the extension version is called.
*
* @param error Error buffer.
* @param maxlength Maximum size of error buffer.
* @param late Whether or not Metamod considers this a late load.
* @return True to succeed, false to fail.
*/
//virtual bool SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlength, bool late);
/**
* @brief Called when Metamod is detaching, after the extension version is called.
* NOTE: By default this is blocked unless sent from SourceMod.
*
* @param error Error buffer.
* @param maxlength Maximum size of error buffer.
* @return True to succeed, false to fail.
*/
//virtual bool SDK_OnMetamodUnload(char *error, size_t maxlength);
/**
* @brief Called when Metamod's pause state is changing.
* NOTE: By default this is blocked unless sent from SourceMod.
*
* @param paused Pause state being set.
* @param error Error buffer.
* @param maxlength Maximum size of error buffer.
* @return True to succeed, false to fail.
*/
//virtual bool SDK_OnMetamodPauseChange(bool paused, char *error, size_t maxlength);
#endif
public:
void AddRequestToQueue(IHTTPContext *context);
char caBundlePath[PLATFORM_MAX_PATH];
};
class HTTPClientHandler : public IHandleTypeDispatch
{
public:
void OnHandleDestroy(HandleType_t type, void *object);
};
class HTTPRequestHandler : public IHandleTypeDispatch
{
public:
void OnHandleDestroy(HandleType_t type, void *object);
};
class HTTPResponseHandler : public IHandleTypeDispatch
{
public:
void OnHandleDestroy(HandleType_t type, void *object);
};
class JSONHandler : public IHandleTypeDispatch
{
public:
void OnHandleDestroy(HandleType_t type, void *object);
};
class JSONObjectKeysHandler : public IHandleTypeDispatch
{
public:
void OnHandleDestroy(HandleType_t type, void *object);
};
extern RipExt g_RipExt;
extern HTTPClientHandler g_HTTPClientHandler;
extern HandleType_t htHTTPClient;
extern HTTPRequestHandler g_HTTPRequestHandler;
extern HandleType_t htHTTPRequest;
extern HTTPResponseHandler g_HTTPResponseHandler;
extern HandleType_t htHTTPResponse;
extern JSONHandler g_JSONHandler;
extern HandleType_t htJSON;
extern JSONObjectKeysHandler g_JSONObjectKeysHandler;
extern HandleType_t htJSONObjectKeys;
extern const sp_nativeinfo_t http_natives[];
extern const sp_nativeinfo_t json_natives[];
#endif // _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_