projects / m6w6 / ext-pq / blob
? search:


589be7713e4378b426901d143a0b925d8ae84f80
[m6w6/ext-pq] / src / php_pq_misc.c
1 /*
2 +--------------------------------------------------------------------+
3 | PECL :: pq |
4 +--------------------------------------------------------------------+
5 | Redistribution and use in source and binary forms, with or without |
6 | modification, are permitted provided that the conditions mentioned |
7 | in the accompanying LICENSE file are met. |
8 +--------------------------------------------------------------------+
9 | Copyright (c) 2013, Michael Wallner <mike@php.net> |
10 +--------------------------------------------------------------------+
11 */
12
13 #ifdef HAVE_CONFIG_H
14 # include "config.h"
15 #endif
16
17 #include <php.h>
18 #include <ext/date/php_date.h>
19 #include <ext/standard/php_string.h>
20
21 #include <Zend/zend_interfaces.h>
22
23 #include <libpq/libpq-fs.h>
24
25 #include "php_pq.h"
26 #include "php_pqexc.h"
27 #include "php_pq_misc.h"
28
29 char *php_pq_rtrim(char *e)
30 {
31 size_t l = strlen(e);
32
33 while (l-- > 0 && e[l] == '\n') {
34 e[l] = '\0';
35 }
36 return e;
37 }
38
39 const char *php_pq_strmode(long mode)
40 {
41 switch (mode & (INV_READ|INV_WRITE)) {
42 case INV_READ|INV_WRITE:
43 return "rw";
44 case INV_READ:
45 return "r";
46 case INV_WRITE:
47 return "w";
48 default:
49 return "-";
50 }
51 }
52
53 int php_pq_compare_index(const void *lptr, const void *rptr TSRMLS_DC)
54 {
55 const Bucket *l = *(const Bucket **) lptr;
56 const Bucket *r = *(const Bucket **) rptr;
57
58 if (l->h < r->h) {
59 return -1;
60 }
61 if (l->h > r->h) {
62 return 1;
63 }
64 return 0;
65 }
66
67 zend_class_entry *php_pqdt_class_entry;
68
69 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_to_string, 0, 0, 0)
70 ZEND_END_ARG_INFO();
71 static PHP_METHOD(pqdt, __toString)
72 {
73 zval *rv = NULL;
74
75 zend_call_method_with_1_params(&getThis(), php_pqdt_class_entry, NULL, "format", &rv,
76 zend_read_property(php_pqdt_class_entry, getThis(), ZEND_STRL("format"), 0 TSRMLS_CC));
77 if (rv) {
78 RETVAL_ZVAL(rv, 1, 1);
79 }
80 }
81
82 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_create_from_format, 0, 0, 2)
83 ZEND_ARG_INFO(0, format)
84 ZEND_ARG_INFO(0, datetime)
85 /* ZEND_ARG_OBJ_INFO(0, timezone, DateTimezone, 1) date's arginfo is not specific */
86 ZEND_ARG_INFO(0, timezone)
87 ZEND_END_ARG_INFO();
88 static PHP_METHOD(pqdt, createFromFormat)
89 {
90 zend_error_handling zeh;
91 char *fmt_str, *dt_str;
92 int fmt_len, dt_len;
93 zval *ztz = NULL;
94 STATUS rv;
95
96 zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
97 rv = zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "ss|O", &fmt_str, &fmt_len, &dt_str, &dt_len, &ztz, php_date_get_timezone_ce());
98 zend_restore_error_handling(&zeh TSRMLS_CC);
99
100 if (SUCCESS == rv) {
101 php_pqdt_from_string(return_value, fmt_str, dt_str, dt_len, "Y-m-d H:i:s.uO", ztz TSRMLS_CC);
102 }
103 }
104
105 static zend_function_entry php_pqdt_methods[] = {
106 PHP_ME(pqdt, createFromFormat, ai_pqdt_create_from_format, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
107 PHP_ME(pqdt, __toString, ai_pqdt_to_string, ZEND_ACC_PUBLIC)
108 PHP_MALIAS(pqdt, jsonSerialize, __toString, ai_pqdt_to_string, ZEND_ACC_PUBLIC)
109 {0}
110 };
111
112 zval *php_pqdt_from_string(zval *zv, char *input_fmt, char *dt_str, size_t dt_len, char *output_fmt, zval *ztimezone TSRMLS_DC)
113 {
114 php_date_obj *dobj;
115
116 if (!zv) {
117 MAKE_STD_ZVAL(zv);
118 }
119
120 php_date_instantiate(php_pqdt_class_entry, zv TSRMLS_CC);
121 dobj = zend_object_store_get_object(zv TSRMLS_CC);
122 if (!php_date_initialize(dobj, dt_str, dt_len, input_fmt, ztimezone, 1 TSRMLS_CC)) {
123 zval_dtor(zv);
124 ZVAL_NULL(zv);
125 } else if (output_fmt) {
126 zend_update_property_string(php_pqdt_class_entry, zv, ZEND_STRL("format"), output_fmt TSRMLS_CC);
127 }
128
129 return zv;
130 }
131
132 void php_pqdt_to_string(zval *zdt, const char *format, char **str_buf, size_t *str_len TSRMLS_DC)
133 {
134 zval rv;
135
136 INIT_PZVAL(&rv);
137 ZVAL_NULL(&rv);
138
139 if (Z_OBJ_HT_P(zdt)->cast_object
140 && SUCCESS == Z_OBJ_HT_P(zdt)->cast_object(zdt, &rv, IS_STRING TSRMLS_CC)
141 ) {
142 *str_len = Z_STRLEN(rv);
143 *str_buf = Z_STRVAL(rv);
144 } else if (instanceof_function(Z_OBJCE_P(zdt), php_date_get_date_ce() TSRMLS_CC)) {
145 zval *rv = NULL, *zfmt;
146
147 MAKE_STD_ZVAL(zfmt);
148 ZVAL_STRING(zfmt, format, 1);
149 zend_call_method_with_1_params(&zdt, Z_OBJCE_P(zdt), NULL, "format", &rv, zfmt);
150 zval_ptr_dtor(&zfmt);
151
152 if (rv) {
153 if (Z_TYPE_P(rv) == IS_STRING) {
154 *str_len = Z_STRLEN_P(rv);
155 *str_buf = estrndup(Z_STRVAL_P(rv), *str_len);
156 }
157 zval_ptr_dtor(&rv);
158 }
159 }
160 }
161
162 zend_class_entry *php_pqconv_class_entry;
163
164 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_types, 0, 0, 0)
165 ZEND_END_ARG_INFO();
166
167 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_from_string, 0, 0, 2)
168 ZEND_ARG_INFO(0, data)
169 ZEND_ARG_INFO(0, type)
170 ZEND_END_ARG_INFO();
171
172 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_to_string, 0, 0, 2)
173 ZEND_ARG_INFO(0, data)
174 ZEND_ARG_INFO(0, type)
175 ZEND_END_ARG_INFO();
176
177 zend_function_entry php_pqconv_methods[] = {
178 PHP_ABSTRACT_ME(pqconv, convertTypes, ai_pqconv_convert_types)
179 PHP_ABSTRACT_ME(pqconv, convertFromString, ai_pqconv_convert_from_string)
180 PHP_ABSTRACT_ME(pqconv, convertToString, ai_pqconv_convert_to_string)
181 {0}
182 };
183
184
185 PHP_MINIT_FUNCTION(pq_misc)
186 {
187 zend_class_entry **json, ce = {0};
188
189 INIT_NS_CLASS_ENTRY(ce, "pq", "Converter", php_pqconv_methods);
190 php_pqconv_class_entry = zend_register_internal_interface(&ce TSRMLS_CC);
191
192 memset(&ce, 0, sizeof(ce));
193 INIT_NS_CLASS_ENTRY(ce ,"pq", "DateTime", php_pqdt_methods);
194 php_pqdt_class_entry = zend_register_internal_class_ex(&ce, php_date_get_date_ce(), "DateTime" TSRMLS_CC);
195
196 zend_declare_property_stringl(php_pqdt_class_entry, ZEND_STRL("format"), ZEND_STRL("Y-m-d H:i:s.uO"), ZEND_ACC_PUBLIC TSRMLS_CC);
197
198 /* stop reading this file right here! */
199 if (SUCCESS == zend_hash_find(CG(class_table), ZEND_STRS("jsonserializable"), (void *) &json)) {
200 zend_class_implements(php_pqdt_class_entry TSRMLS_CC, 1, *json);
201 }
202
203 return SUCCESS;
204 }
205
206 typedef struct _HashTableList {
207 HashTable ht;
208 struct _HashTableList *parent;
209 } HashTableList;
210
211 typedef struct _ArrayParserState {
212 const char *ptr, *end;
213 HashTableList *list;
214 php_pqres_t *res;
215 #ifdef ZTS
216 void ***ts;
217 #endif
218 Oid typ;
219 unsigned quotes:1;
220 unsigned escaped:1;
221 } ArrayParserState;
222
223 static char caa(ArrayParserState *a, const char *any, unsigned advance)
224 {
225 const char *p = any;
226 TSRMLS_FETCH_FROM_CTX(a->ts);
227
228 do {
229 if (*p == *a->ptr) {
230 a->ptr += advance;
231 return *p;
232 }
233 } while (*++p);
234
235 php_error_docref(NULL TSRMLS_CC, E_WARNING, "Failed to parse array: expected one of '%s', got '%c'", any, *a->ptr); \
236 return 0;
237 }
238
239 static STATUS add_element(ArrayParserState *a, const char *start)
240 {
241 zval *zelem;
242 size_t el_len = a->ptr - start;
243 char *el_str = estrndup(start, el_len);
244 TSRMLS_FETCH_FROM_CTX(a->ts);
245
246 if (a->quotes) {
247 int tmp_len = el_len;
248
249 php_stripslashes(el_str, &tmp_len TSRMLS_CC);
250 el_len = tmp_len;
251 } else if ((a->ptr - start == 4) && !strncmp(start, "NULL", 4)) {
252 efree(el_str);
253 el_str = NULL;
254 el_len = 0;
255 }
256
257 if (!el_str) {
258 MAKE_STD_ZVAL(zelem);
259 ZVAL_NULL(zelem);
260 } else {
261 zelem = php_pqres_typed_zval(a->res, el_str, el_len, a->typ TSRMLS_CC);
262
263 efree(el_str);
264 }
265
266 return zend_hash_next_index_insert(&a->list->ht, &zelem, sizeof(zval *), NULL);
267 }
268
269 static STATUS parse_array(ArrayParserState *a);
270
271 static STATUS parse_element(ArrayParserState *a)
272 {
273 const char *el;
274 TSRMLS_FETCH_FROM_CTX(a->ts);
275
276 switch (*a->ptr) {
277 case '{':
278 return parse_array(a);
279
280 case '"':
281 a->quotes = 1;
282 ++a->ptr;
283 break;
284 }
285
286 for (el = a->ptr; a->ptr < a->end; ++a->ptr) {
287 switch (*a->ptr) {
288 case '"':
289 if (a->escaped) {
290 a->escaped = 0;
291 } else if (a->quotes) {
292 if (SUCCESS != add_element(a, el)) {
293 return FAILURE;
294 }
295 a->quotes = 0;
296 ++a->ptr;
297 return SUCCESS;
298 } else {
299 php_error_docref(NULL TSRMLS_CC, E_WARNING, "Failed to parse element, unexpected quote: '%.*s'", (int) (a->ptr - el), el);
300 return FAILURE;
301 }
302 break;
303
304 case ',':
305 case '}':
306 if (!a->quotes) {
307 return add_element(a, el);
308 }
309 break;
310
311 case '\\':
312 a->escaped = !a->escaped;
313 break;
314
315 default:
316 a->escaped = 0;
317 break;
318 }
319 }
320
321 php_error_docref(NULL TSRMLS_CC, E_WARNING, "Failed to parse element, reached end of input");
322 return FAILURE;
323 }
324
325 static STATUS parse_elements(ArrayParserState *a)
326 {
327 TSRMLS_FETCH_FROM_CTX(a->ts);
328
329 while (SUCCESS == parse_element(a)) {
330 switch (caa(a, ",}", 0)) {
331 case 0:
332 return FAILURE;
333
334 case '}':
335 return SUCCESS;
336
337 default:
338 if (!*++a->ptr) {
339 php_error_docref(NULL TSRMLS_CC, E_WARNING, "Failed to parse elements, reached end of input");
340 return FAILURE;
341 }
342 break;
343 }
344 }
345
346 return FAILURE;
347 }
348
349 static STATUS parse_array(ArrayParserState *a)
350 {
351 HashTableList *list;
352
353 if (!caa(a, "{", 1)) {
354 return FAILURE;
355 }
356
357 list = ecalloc(1, sizeof(*list));
358 ZEND_INIT_SYMTABLE(&list->ht);
359
360 if (a->list) {
361 zval *zcur;
362
363 MAKE_STD_ZVAL(zcur);
364 Z_TYPE_P(zcur) = IS_ARRAY;
365 Z_ARRVAL_P(zcur) = &list->ht;
366
367 zend_hash_next_index_insert(&a->list->ht, &zcur, sizeof(zval *), NULL);
368
369 list->parent = a->list;
370 }
371 a->list = list;
372
373 if (SUCCESS != parse_elements(a)) {
374 return FAILURE;
375 }
376
377 if (!caa(a, "}", 1)) {
378 return FAILURE;
379 }
380
381 if (a->list->parent) {
382 a->list = a->list->parent;
383 }
384
385 return SUCCESS;
386 }
387
388 HashTable *php_pq_parse_array(php_pqres_t *res, const char *val_str, size_t val_len, Oid typ TSRMLS_DC)
389 {
390 HashTable *ht = NULL;
391 ArrayParserState a = {0};
392 TSRMLS_SET_CTX(a.ts);
393
394 a.typ = typ;
395 a.ptr = val_str;
396 a.end = val_str + val_len;
397 a.res = res;
398
399 if (SUCCESS != parse_array(&a)) {
400 while (a.list) {
401 HashTableList *l = a.list->parent;
402
403 zend_hash_destroy(&a.list->ht);
404 efree(a.list);
405 a.list = l;
406 }
407 return ht;
408 }
409
410 if (*a.ptr) {
411 php_error_docref(NULL TSRMLS_CC, E_NOTICE, "Trailing input: '%s'", a.ptr);
412 }
413
414 do {
415 ht = &a.list->ht;
416 } while ((a.list = a.list->parent));
417
418 return ht;
419 }
420
421
422 /*
423 * Local variables:
424 * tab-width: 4
425 * c-basic-offset: 4
426 * End:
427 * vim600: noet sw=4 ts=4 fdm=marker
428 * vim<600: noet sw=4 ts=4
429 */
PostgreSQL client library (libpq) binding