trilogy_w_prepared_statements 2.2.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +7 -0
- data/LICENSE +22 -0
- data/README.md +80 -0
- data/Rakefile +22 -0
- data/ext/trilogy-ruby/cast.c +277 -0
- data/ext/trilogy-ruby/cext.c +1048 -0
- data/ext/trilogy-ruby/extconf.rb +17 -0
- data/ext/trilogy-ruby/inc/trilogy/blocking.h +281 -0
- data/ext/trilogy-ruby/inc/trilogy/buffer.h +64 -0
- data/ext/trilogy-ruby/inc/trilogy/builder.h +165 -0
- data/ext/trilogy-ruby/inc/trilogy/charset.h +277 -0
- data/ext/trilogy-ruby/inc/trilogy/client.h +760 -0
- data/ext/trilogy-ruby/inc/trilogy/error.h +44 -0
- data/ext/trilogy-ruby/inc/trilogy/packet_parser.h +34 -0
- data/ext/trilogy-ruby/inc/trilogy/protocol.h +1014 -0
- data/ext/trilogy-ruby/inc/trilogy/reader.h +216 -0
- data/ext/trilogy-ruby/inc/trilogy/socket.h +111 -0
- data/ext/trilogy-ruby/inc/trilogy/vendor/curl_hostcheck.h +29 -0
- data/ext/trilogy-ruby/inc/trilogy/vendor/openssl_hostname_validation.h +51 -0
- data/ext/trilogy-ruby/inc/trilogy.h +8 -0
- data/ext/trilogy-ruby/src/blocking.c +358 -0
- data/ext/trilogy-ruby/src/buffer.c +60 -0
- data/ext/trilogy-ruby/src/builder.c +236 -0
- data/ext/trilogy-ruby/src/charset.c +212 -0
- data/ext/trilogy-ruby/src/client.c +903 -0
- data/ext/trilogy-ruby/src/error.c +17 -0
- data/ext/trilogy-ruby/src/packet_parser.c +140 -0
- data/ext/trilogy-ruby/src/protocol.c +1175 -0
- data/ext/trilogy-ruby/src/reader.c +282 -0
- data/ext/trilogy-ruby/src/socket.c +623 -0
- data/ext/trilogy-ruby/src/vendor/curl_hostcheck.c +206 -0
- data/ext/trilogy-ruby/src/vendor/openssl_hostname_validation.c +175 -0
- data/ext/trilogy-ruby/trilogy-ruby.h +37 -0
- data/lib/trilogy/version.rb +3 -0
- data/lib/trilogy.rb +61 -0
- data/trilogy.gemspec +27 -0
- metadata +107 -0
@@ -0,0 +1,282 @@
|
|
1
|
+
#include "trilogy/reader.h"
|
2
|
+
#include "trilogy/error.h"
|
3
|
+
|
4
|
+
#include <string.h>
|
5
|
+
|
6
|
+
#define CHECK(bytes) \
|
7
|
+
if ((bytes) > (reader->len - reader->pos)) { \
|
8
|
+
return TRILOGY_TRUNCATED_PACKET; \
|
9
|
+
}
|
10
|
+
|
11
|
+
void trilogy_reader_init(trilogy_reader_t *reader, const uint8_t *buff, size_t len)
|
12
|
+
{
|
13
|
+
reader->buff = buff;
|
14
|
+
reader->len = len;
|
15
|
+
reader->pos = 0;
|
16
|
+
}
|
17
|
+
|
18
|
+
static uint8_t next_uint8(trilogy_reader_t *reader) { return reader->buff[reader->pos++]; }
|
19
|
+
|
20
|
+
int trilogy_reader_get_uint8(trilogy_reader_t *reader, uint8_t *out)
|
21
|
+
{
|
22
|
+
CHECK(1);
|
23
|
+
|
24
|
+
uint8_t a = next_uint8(reader);
|
25
|
+
|
26
|
+
if (out) {
|
27
|
+
*out = a;
|
28
|
+
}
|
29
|
+
|
30
|
+
return TRILOGY_OK;
|
31
|
+
}
|
32
|
+
|
33
|
+
int trilogy_reader_get_uint16(trilogy_reader_t *reader, uint16_t *out)
|
34
|
+
{
|
35
|
+
CHECK(2);
|
36
|
+
|
37
|
+
uint16_t a = next_uint8(reader);
|
38
|
+
uint16_t b = next_uint8(reader);
|
39
|
+
|
40
|
+
if (out) {
|
41
|
+
*out = (uint16_t)(a | (b << 8));
|
42
|
+
}
|
43
|
+
|
44
|
+
return TRILOGY_OK;
|
45
|
+
}
|
46
|
+
|
47
|
+
int trilogy_reader_get_uint24(trilogy_reader_t *reader, uint32_t *out)
|
48
|
+
{
|
49
|
+
CHECK(3);
|
50
|
+
|
51
|
+
uint32_t a = next_uint8(reader);
|
52
|
+
uint32_t b = next_uint8(reader);
|
53
|
+
uint32_t c = next_uint8(reader);
|
54
|
+
|
55
|
+
if (out) {
|
56
|
+
*out = a | (b << 8) | (c << 16);
|
57
|
+
}
|
58
|
+
|
59
|
+
return TRILOGY_OK;
|
60
|
+
}
|
61
|
+
|
62
|
+
int trilogy_reader_get_uint32(trilogy_reader_t *reader, uint32_t *out)
|
63
|
+
{
|
64
|
+
CHECK(4);
|
65
|
+
|
66
|
+
uint32_t a = next_uint8(reader);
|
67
|
+
uint32_t b = next_uint8(reader);
|
68
|
+
uint32_t c = next_uint8(reader);
|
69
|
+
uint32_t d = next_uint8(reader);
|
70
|
+
|
71
|
+
if (out) {
|
72
|
+
*out = a | (b << 8) | (c << 16) | (d << 24);
|
73
|
+
}
|
74
|
+
|
75
|
+
return TRILOGY_OK;
|
76
|
+
}
|
77
|
+
|
78
|
+
int trilogy_reader_get_uint64(trilogy_reader_t *reader, uint64_t *out)
|
79
|
+
{
|
80
|
+
CHECK(8);
|
81
|
+
|
82
|
+
uint64_t a = next_uint8(reader);
|
83
|
+
uint64_t b = next_uint8(reader);
|
84
|
+
uint64_t c = next_uint8(reader);
|
85
|
+
uint64_t d = next_uint8(reader);
|
86
|
+
uint64_t e = next_uint8(reader);
|
87
|
+
uint64_t f = next_uint8(reader);
|
88
|
+
uint64_t g = next_uint8(reader);
|
89
|
+
uint64_t h = next_uint8(reader);
|
90
|
+
|
91
|
+
if (out) {
|
92
|
+
*out = a | (b << 8) | (c << 16) | (d << 24) | (e << 32) | (f << 40) | (g << 48) | (h << 56);
|
93
|
+
}
|
94
|
+
|
95
|
+
return TRILOGY_OK;
|
96
|
+
}
|
97
|
+
|
98
|
+
int trilogy_reader_get_float(trilogy_reader_t *reader, float *out)
|
99
|
+
{
|
100
|
+
CHECK(4);
|
101
|
+
|
102
|
+
union {
|
103
|
+
float f;
|
104
|
+
uint32_t u;
|
105
|
+
} float_val;
|
106
|
+
|
107
|
+
int rc = trilogy_reader_get_uint32(reader, &float_val.u);
|
108
|
+
if (rc != TRILOGY_OK) {
|
109
|
+
return rc;
|
110
|
+
}
|
111
|
+
|
112
|
+
*out = float_val.f;
|
113
|
+
|
114
|
+
return TRILOGY_OK;
|
115
|
+
}
|
116
|
+
|
117
|
+
int trilogy_reader_get_double(trilogy_reader_t *reader, double *out)
|
118
|
+
{
|
119
|
+
CHECK(8);
|
120
|
+
|
121
|
+
union {
|
122
|
+
double d;
|
123
|
+
uint64_t u;
|
124
|
+
} double_val;
|
125
|
+
|
126
|
+
int rc = trilogy_reader_get_uint64(reader, &double_val.u);
|
127
|
+
if (rc != TRILOGY_OK) {
|
128
|
+
return rc;
|
129
|
+
}
|
130
|
+
|
131
|
+
*out = double_val.d;
|
132
|
+
|
133
|
+
return TRILOGY_OK;
|
134
|
+
}
|
135
|
+
|
136
|
+
int trilogy_reader_get_lenenc(trilogy_reader_t *reader, uint64_t *out)
|
137
|
+
{
|
138
|
+
CHECK(1);
|
139
|
+
|
140
|
+
uint8_t leader = next_uint8(reader);
|
141
|
+
|
142
|
+
if (leader < 0xfb) {
|
143
|
+
if (out) {
|
144
|
+
*out = leader;
|
145
|
+
}
|
146
|
+
|
147
|
+
return TRILOGY_OK;
|
148
|
+
}
|
149
|
+
|
150
|
+
switch (leader) {
|
151
|
+
case 0xfb:
|
152
|
+
return TRILOGY_NULL_VALUE;
|
153
|
+
|
154
|
+
case 0xfc: {
|
155
|
+
uint16_t u16 = 0;
|
156
|
+
int rc = trilogy_reader_get_uint16(reader, &u16);
|
157
|
+
|
158
|
+
if (out) {
|
159
|
+
*out = u16;
|
160
|
+
}
|
161
|
+
|
162
|
+
return rc;
|
163
|
+
}
|
164
|
+
|
165
|
+
case 0xfd: {
|
166
|
+
uint32_t u24 = 0;
|
167
|
+
int rc = trilogy_reader_get_uint24(reader, &u24);
|
168
|
+
|
169
|
+
if (out) {
|
170
|
+
*out = u24;
|
171
|
+
}
|
172
|
+
|
173
|
+
return rc;
|
174
|
+
}
|
175
|
+
|
176
|
+
case 0xfe:
|
177
|
+
return trilogy_reader_get_uint64(reader, out);
|
178
|
+
|
179
|
+
default:
|
180
|
+
return TRILOGY_PROTOCOL_VIOLATION;
|
181
|
+
}
|
182
|
+
}
|
183
|
+
|
184
|
+
int trilogy_reader_get_buffer(trilogy_reader_t *reader, size_t len, const void **out)
|
185
|
+
{
|
186
|
+
CHECK(len);
|
187
|
+
|
188
|
+
if (out) {
|
189
|
+
*out = (const void *)(reader->buff + reader->pos);
|
190
|
+
}
|
191
|
+
|
192
|
+
reader->pos += len;
|
193
|
+
|
194
|
+
return TRILOGY_OK;
|
195
|
+
}
|
196
|
+
|
197
|
+
int trilogy_reader_copy_buffer(trilogy_reader_t *reader, size_t len, void *out)
|
198
|
+
{
|
199
|
+
CHECK(len);
|
200
|
+
|
201
|
+
if (out) {
|
202
|
+
memcpy(out, reader->buff + reader->pos, len);
|
203
|
+
}
|
204
|
+
|
205
|
+
reader->pos += len;
|
206
|
+
|
207
|
+
return TRILOGY_OK;
|
208
|
+
}
|
209
|
+
|
210
|
+
int trilogy_reader_get_lenenc_buffer(trilogy_reader_t *reader, size_t *out_len, const void **out)
|
211
|
+
{
|
212
|
+
uint64_t len;
|
213
|
+
|
214
|
+
int rc = trilogy_reader_get_lenenc(reader, &len);
|
215
|
+
|
216
|
+
if (rc) {
|
217
|
+
return rc;
|
218
|
+
}
|
219
|
+
|
220
|
+
// check len is not larger than the amount of bytes left before downcasting
|
221
|
+
// to size_t (which may be smaller than uint64_t on some architectures)
|
222
|
+
if (len > (uint64_t)(reader->len - reader->pos)) {
|
223
|
+
return TRILOGY_TRUNCATED_PACKET;
|
224
|
+
}
|
225
|
+
|
226
|
+
if (out_len) {
|
227
|
+
*out_len = (size_t)len;
|
228
|
+
}
|
229
|
+
|
230
|
+
return trilogy_reader_get_buffer(reader, (size_t)len, out);
|
231
|
+
}
|
232
|
+
|
233
|
+
int trilogy_reader_get_string(trilogy_reader_t *reader, const char **out, size_t *out_len)
|
234
|
+
{
|
235
|
+
const uint8_t *pos = reader->buff + reader->pos;
|
236
|
+
|
237
|
+
const uint8_t *end = memchr(pos, 0, reader->len - reader->pos);
|
238
|
+
|
239
|
+
if (!end) {
|
240
|
+
return TRILOGY_TRUNCATED_PACKET;
|
241
|
+
}
|
242
|
+
|
243
|
+
if (out) {
|
244
|
+
*out = (const char *)pos;
|
245
|
+
}
|
246
|
+
|
247
|
+
size_t len = (size_t)(end - pos);
|
248
|
+
|
249
|
+
if (out_len) {
|
250
|
+
*out_len = len;
|
251
|
+
}
|
252
|
+
|
253
|
+
reader->pos += len + 1;
|
254
|
+
|
255
|
+
return TRILOGY_OK;
|
256
|
+
}
|
257
|
+
|
258
|
+
int trilogy_reader_get_eof_buffer(trilogy_reader_t *reader, size_t *out_len, const void **out)
|
259
|
+
{
|
260
|
+
if (out_len) {
|
261
|
+
*out_len = reader->len - reader->pos;
|
262
|
+
}
|
263
|
+
|
264
|
+
if (out) {
|
265
|
+
*out = reader->buff + reader->pos;
|
266
|
+
}
|
267
|
+
|
268
|
+
reader->pos = reader->len;
|
269
|
+
|
270
|
+
return TRILOGY_OK;
|
271
|
+
}
|
272
|
+
|
273
|
+
bool trilogy_reader_eof(trilogy_reader_t *reader) { return !(reader->pos < reader->len); }
|
274
|
+
|
275
|
+
int trilogy_reader_finish(trilogy_reader_t *reader)
|
276
|
+
{
|
277
|
+
if (reader->pos < reader->len) {
|
278
|
+
return TRILOGY_EXTRA_DATA_IN_PACKET;
|
279
|
+
} else {
|
280
|
+
return TRILOGY_OK;
|
281
|
+
}
|
282
|
+
}
|