2 +--------------------------------------------------------------------+
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 +--------------------------------------------------------------------+
18 #include <ext/date/php_date.h>
19 #include <ext/standard/php_string.h>
20 #if defined(HAVE_JSON) && !defined(COMPILE_DL_JSON)
21 # include <ext/json/php_json.h>
24 #include <Zend/zend_interfaces.h>
26 #include <libpq/libpq-fs.h>
29 #include "php_pq_misc.h"
35 while (l
-- > 0 && e
[l
] == '\n') {
41 const char *strmode(long mode
)
43 switch (mode
& (INV_READ
|INV_WRITE
)) {
44 case INV_READ
|INV_WRITE
:
55 int compare_index(const void *lptr
, const void *rptr TSRMLS_DC
)
57 const Bucket
*l
= *(const Bucket
**) lptr
;
58 const Bucket
*r
= *(const Bucket
**) rptr
;
69 static int apply_to_oid(void *p
, void *arg TSRMLS_DC
)
74 if (Z_TYPE_PP(ztype
) != IS_LONG
) {
75 convert_to_long_ex(ztype
);
78 **types
= Z_LVAL_PP(ztype
);
81 if (*ztype
!= *(zval
**)p
) {
84 return ZEND_HASH_APPLY_KEEP
;
87 static int apply_to_param(void *p TSRMLS_DC
, int argc
, va_list argv
, zend_hash_key
*key
)
93 params
= (char ***) va_arg(argv
, char ***);
94 zdtor
= (HashTable
*) va_arg(argv
, HashTable
*);
96 switch (Z_TYPE_PP(zparam
)) {
103 **params
= Z_BVAL_PP(zparam
) ? "t" : "f";
108 SEPARATE_ZVAL(zparam
);
109 Z_TYPE_PP(zparam
) = IS_STRING
;
110 Z_STRLEN_PP(zparam
) = spprintf(&Z_STRVAL_PP(zparam
), 0, "%F", Z_DVAL_PP((zval
**)p
));
114 convert_to_string_ex(zparam
);
118 **params
= Z_STRVAL_PP(zparam
);
121 if (*zparam
!= *(zval
**)p
) {
122 zend_hash_next_index_insert(zdtor
, zparam
, sizeof(zval
*), NULL
);
127 return ZEND_HASH_APPLY_KEEP
;
130 int php_pq_types_to_array(HashTable
*ht
, Oid
**types TSRMLS_DC
)
132 int count
= zend_hash_num_elements(ht
);
139 /* +1 for when less types than params are specified */
140 *types
= tmp
= ecalloc(count
+ 1, sizeof(**types
));
141 zend_hash_apply_with_argument(ht
, apply_to_oid
, &tmp TSRMLS_CC
);
147 int php_pq_params_to_array(HashTable
*ht
, char ***params
, HashTable
*zdtor TSRMLS_DC
)
149 int count
= zend_hash_num_elements(ht
);
156 *params
= tmp
= ecalloc(count
, sizeof(char *));
157 zend_hash_apply_with_arguments(ht TSRMLS_CC
, apply_to_param
, 2, &tmp
, zdtor
);
164 Oid *php_pq_ntypes_to_array(zend_bool fill, int argc, ...)
167 Oid *oids = ecalloc(argc + 1, sizeof(*oids));
170 va_start(argv, argc);
171 for (i = 0; i < argc; ++i) {
173 oids[i] = va_arg(argv, Oid);
184 zend_class_entry
*php_pqdt_class_entry
;
186 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_to_string
, 0, 0, 0)
188 static PHP_METHOD(pqdt
, __toString
)
192 zend_call_method_with_1_params(&getThis(), php_pqdt_class_entry
, NULL
, "format", &rv
,
193 zend_read_property(php_pqdt_class_entry
, getThis(), ZEND_STRL("format"), 0 TSRMLS_CC
));
194 RETVAL_ZVAL(rv
, 1, 1);
197 static zend_function_entry php_pqdt_methods
[] = {
198 PHP_ME(pqdt
, __toString
, ai_pqdt_to_string
, ZEND_ACC_PUBLIC
)
199 PHP_MALIAS(pqdt
, jsonSerialize
, __toString
, ai_pqdt_to_string
, ZEND_ACC_PUBLIC
)
203 zval
*php_pqdt_from_string(char *dt_str
, size_t dt_len
, char *fmt
, zval
*zv TSRMLS_DC
)
211 php_date_instantiate(php_pqdt_class_entry
, zv TSRMLS_CC
);
212 dobj
= zend_object_store_get_object(zv TSRMLS_CC
);
213 if (!php_date_initialize(dobj
, dt_str
, dt_len
, NULL
, NULL
, 1 TSRMLS_CC
)) {
217 zend_update_property_string(php_pqdt_class_entry
, zv
, ZEND_STRL("format"), fmt TSRMLS_CC
);
223 PHP_MINIT_FUNCTION(pq_misc
)
225 zend_class_entry
**json
, ce
= {0};
227 INIT_NS_CLASS_ENTRY(ce
,"pq", "DateTime", php_pqdt_methods
);
228 php_pqdt_class_entry
= zend_register_internal_class_ex(&ce
, php_date_get_date_ce(), "DateTime" TSRMLS_CC
);
230 zend_declare_property_stringl(php_pqdt_class_entry
, ZEND_STRL("format"), ZEND_STRL("Y-m-d H:i:s.u"), ZEND_ACC_PUBLIC TSRMLS_CC
);
232 /* stop reading this file right here! */
233 if (SUCCESS
== zend_hash_find(CG(class_table
), ZEND_STRS("jsonserializable"), (void *) &json
)) {
234 zend_class_implements(php_pqdt_class_entry TSRMLS_CC
, 1, *json
);
240 typedef struct _HashTableList
{
242 struct _HashTableList
*parent
;
245 typedef struct _ArrayParserState
{
246 const char *ptr
, *end
;
256 static char caa(ArrayParserState
*a
, const char *any
, unsigned advance
)
259 TSRMLS_FETCH_FROM_CTX(a
->ts
);
268 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to parse array: expected one of '%s', got '%c'", any
, *a
->ptr
); \
272 static STATUS
add_element(ArrayParserState
*a
, const char *start
)
275 size_t el_len
= a
->ptr
- start
;
276 char *el_str
= estrndup(start
, el_len
);
277 TSRMLS_FETCH_FROM_CTX(a
->ts
);
282 php_stripslashes(el_str
, &tmp_len TSRMLS_CC
);
284 } else if ((a
->ptr
- start
== 4) && !strncmp(start
, "NULL", 4)) {
291 MAKE_STD_ZVAL(zelem
);
294 zelem
= php_pq_typed_zval(el_str
, el_len
, a
->typ TSRMLS_CC
);
299 return zend_hash_next_index_insert(&a
->list
->ht
, &zelem
, sizeof(zval
*), NULL
);
302 static STATUS
parse_array(ArrayParserState
*a
);
304 static STATUS
parse_element(ArrayParserState
*a
)
307 TSRMLS_FETCH_FROM_CTX(a
->ts
);
311 return parse_array(a
);
319 for (el
= a
->ptr
; a
->ptr
< a
->end
; ++a
->ptr
) {
324 } else if (a
->quotes
) {
325 if (SUCCESS
!= add_element(a
, el
)) {
332 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to parse element, unexpected quote: '%.*s'", (int) (a
->ptr
- el
), el
);
340 return add_element(a
, el
);
345 a
->escaped
= !a
->escaped
;
354 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to parse element, reached end of input");
358 static STATUS
parse_elements(ArrayParserState
*a
)
360 TSRMLS_FETCH_FROM_CTX(a
->ts
);
362 while (SUCCESS
== parse_element(a
)) {
363 switch (caa(a
, ",}", 0)) {
372 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to parse elements, reached end of input");
382 static STATUS
parse_array(ArrayParserState
*a
)
386 if (!caa(a
, "{", 1)) {
390 list
= ecalloc(1, sizeof(*list
));
391 ZEND_INIT_SYMTABLE(&list
->ht
);
397 Z_TYPE_P(zcur
) = IS_ARRAY
;
398 Z_ARRVAL_P(zcur
) = &list
->ht
;
400 zend_hash_next_index_insert(&a
->list
->ht
, &zcur
, sizeof(zval
*), NULL
);
402 list
->parent
= a
->list
;
406 if (SUCCESS
!= parse_elements(a
)) {
410 if (!caa(a
, "}", 1)) {
414 if (a
->list
->parent
) {
415 a
->list
= a
->list
->parent
;
421 HashTable
*php_pq_parse_array(const char *val_str
, size_t val_len
, Oid typ TSRMLS_DC
)
423 HashTable
*ht
= NULL
;
424 ArrayParserState a
= {0};
425 TSRMLS_SET_CTX(a
.ts
);
429 a
.end
= val_str
+ val_len
;
431 if (SUCCESS
!= parse_array(&a
)) {
433 HashTableList
*l
= a
.list
->parent
;
435 zend_hash_destroy(&a
.list
->ht
);
443 php_error_docref(NULL TSRMLS_CC
, E_NOTICE
, "Trailing input: '%s'", a
.ptr
);
448 } while ((a
.list
= a
.list
->parent
));
453 zval
*php_pq_typed_zval(char *val
, size_t len
, Oid typ TSRMLS_DC
)
460 #ifdef HAVE_PHP_PQ_TYPE_H
462 # include "php_pq_type.h"
463 case PHP_PQ_OID_BOOL
:
464 ZVAL_BOOL(zv
, *val
== 't');
467 case PHP_PQ_OID_INT8
:
470 case PHP_PQ_OID_INT4
:
471 case PHP_PQ_OID_INT2
:
474 ZVAL_LONG(zv
, zend_atol(val
, len
));
477 case PHP_PQ_OID_FLOAT4
:
478 case PHP_PQ_OID_FLOAT8
:
479 ZVAL_DOUBLE(zv
, zend_strtod(val
, NULL
));
482 case PHP_PQ_OID_DATE
:
483 php_pqdt_from_string(val
, len
, "Y-m-d", zv TSRMLS_CC
);
486 case PHP_PQ_OID_ABSTIME
:
487 php_pqdt_from_string(val
, len
, "Y-m-d H:i:s", zv TSRMLS_CC
);
490 case PHP_PQ_OID_TIMESTAMP
:
491 php_pqdt_from_string(val
, len
, "Y-m-d H:i:s.u", zv TSRMLS_CC
);
494 case PHP_PQ_OID_TIMESTAMPTZ
:
495 php_pqdt_from_string(val
, len
, "Y-m-d H:i:s.uO", zv TSRMLS_CC
);
499 if (PHP_PQ_TYPE_IS_ARRAY(typ
) && (Z_ARRVAL_P(zv
) = php_pq_parse_array(val
, len
, PHP_PQ_TYPE_OF_ARRAY(typ
) TSRMLS_CC
))) {
500 Z_TYPE_P(zv
) = IS_ARRAY
;
502 ZVAL_STRINGL(zv
, val
, len
, 1);
507 ZVAL_BOOL(zv
, *val
== 't');
511 ZVAL_STRINGL(zv
, val
, len
, 1);
523 * vim600: noet sw=4 ts=4 fdm=marker
524 * vim<600: noet sw=4 ts=4