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>
21 #include <Zend/zend_interfaces.h>
23 #include <libpq/libpq-fs.h>
26 #include "php_pqexc.h"
27 #include "php_pq_misc.h"
28 #include "php_pqconn_event.h"
30 #include "php_pq_type.h"
33 /* clear result object associated with a result handle */
34 void php_pqres_clear(PGresult
*r
) {
35 php_pq_object_t
*o
= PQresultInstanceData(r
, php_pqconn_event
);
38 php_pq_object_delref(o
);
44 /* clear any asynchronous results */
45 void php_pqconn_clear(PGconn
*conn
) {
47 php_pqconn_event_data_t
*evdata
= PQinstanceData(conn
, php_pqconn_event
);
49 while ((r
= PQgetResult(conn
))) {
53 if (evdata
&& evdata
->obj
) {
54 if (php_pq_callback_is_enabled(&evdata
->obj
->intern
->onevent
)) {
55 if (php_pq_callback_is_locked(&evdata
->obj
->intern
->onevent
)) {
56 php_pq_callback_disable(&evdata
->obj
->intern
->onevent
);
58 php_pq_callback_dtor(&evdata
->obj
->intern
->onevent
);
64 /* safe wrappers to clear any asynchronous wrappers before querying synchronously */
65 PGresult
*php_pq_exec(PGconn
*conn
, const char *query
) {
66 php_pqconn_clear(conn
);
67 return PQexec(conn
, query
);
69 PGresult
*php_pq_exec_params(PGconn
*conn
, const char *command
, int nParams
, const Oid
*paramTypes
, const char *const * paramValues
, const int *paramLengths
, const int *paramFormats
, int resultFormat
) {
70 php_pqconn_clear(conn
);
71 return PQexecParams(conn
, command
, nParams
, paramTypes
, paramValues
, paramLengths
, paramFormats
, resultFormat
);
73 PGresult
*php_pq_prepare(PGconn
*conn
, const char *stmtName
, const char *query
, int nParams
, const Oid
*paramTypes
) {
74 php_pqconn_clear(conn
);
75 return PQprepare(conn
, stmtName
, query
, nParams
, paramTypes
);
77 PGresult
*php_pq_exec_prepared(PGconn
*conn
, const char *stmtName
, int nParams
, const char *const * paramValues
, const int *paramLengths
, const int *paramFormats
, int resultFormat
) {
78 php_pqconn_clear(conn
);
79 return PQexecPrepared(conn
, stmtName
, nParams
, paramValues
, paramLengths
, paramFormats
, resultFormat
);
82 char *php_pq_rtrim(char *e
)
86 while (l
-- > 0 && e
[l
] == '\n') {
92 const char *php_pq_strmode(long mode
)
94 switch (mode
& (INV_READ
|INV_WRITE
)) {
95 case INV_READ
|INV_WRITE
:
106 int php_pq_compare_index(const void *lptr
, const void *rptr
)
108 zend_ulong l
= ((const Bucket
*) lptr
)->h
;
109 zend_ulong r
= ((const Bucket
*) rptr
)->h
;
120 void php_pq_hash_ptr_dtor(zval
*p
)
125 zend_class_entry
*php_pqdt_class_entry
;
127 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_to_string
, 0, 0, 0)
129 static PHP_METHOD(pqdt
, __toString
)
134 zend_call_method_with_1_params(getThis(), php_pqdt_class_entry
, NULL
, "format", &rv
,
135 zend_read_property(php_pqdt_class_entry
, getThis(), ZEND_STRL("format"), 0, &tmp
));
136 RETVAL_ZVAL(&rv
, 1, 1);
139 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_create_from_format
, 0, 0, 2)
140 ZEND_ARG_INFO(0, format
)
141 ZEND_ARG_INFO(0, datetime
)
142 ZEND_ARG_INFO(0, timezone
)
144 static PHP_METHOD(pqdt
, createFromFormat
)
146 zend_error_handling zeh
;
147 char *fmt_str
, *dt_str
;
148 size_t fmt_len
, dt_len
;
152 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh
);
153 rv
= zend_parse_parameters(ZEND_NUM_ARGS(), "ss|O", &fmt_str
, &fmt_len
, &dt_str
, &dt_len
, &ztz
, php_date_get_timezone_ce());
154 zend_restore_error_handling(&zeh
);
157 php_pqdt_from_string(return_value
, fmt_str
, dt_str
, dt_len
, "Y-m-d H:i:s.uO", ztz
);
161 static zend_function_entry php_pqdt_methods
[] = {
162 PHP_ME(pqdt
, createFromFormat
, ai_pqdt_create_from_format
, ZEND_ACC_PUBLIC
|ZEND_ACC_STATIC
)
163 PHP_ME(pqdt
, __toString
, ai_pqdt_to_string
, ZEND_ACC_PUBLIC
)
164 PHP_MALIAS(pqdt
, jsonSerialize
, __toString
, ai_pqdt_to_string
, ZEND_ACC_PUBLIC
)
168 zval
*php_pqdt_from_string(zval
*zv
, char *input_fmt
, char *dt_str
, size_t dt_len
, char *output_fmt
, zval
*ztimezone
)
172 php_date_instantiate(php_pqdt_class_entry
, zv
);
173 dobj
= php_date_obj_from_obj(Z_OBJ_P(zv
));
174 if (!php_date_initialize(dobj
, dt_str
, dt_len
, input_fmt
, ztimezone
, 1)) {
177 } else if (output_fmt
) {
178 zend_update_property_string(php_pqdt_class_entry
, zv
, ZEND_STRL("format"), output_fmt
);
184 zend_string
*php_pqdt_to_string(zval
*zdt
, const char *format
)
190 if (Z_OBJ_HT_P(zdt
)->cast_object
191 && SUCCESS
== Z_OBJ_HT_P(zdt
)->cast_object(zdt
, &rv
, IS_STRING
)
194 } else if (instanceof_function(Z_OBJCE_P(zdt
), php_date_get_date_ce())) {
198 ZVAL_STRING(&zfmt
, format
);
199 zend_call_method_with_1_params(zdt
, Z_OBJCE_P(zdt
), NULL
, "format", &rv
, &zfmt
);
200 zval_ptr_dtor(&zfmt
);
202 if (Z_TYPE(rv
) == IS_STRING
) {
211 zend_class_entry
*php_pqconv_class_entry
;
213 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_types
, 0, 0, 0)
216 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_from_string
, 0, 0, 2)
217 ZEND_ARG_INFO(0, data
)
218 ZEND_ARG_INFO(0, type
)
221 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_to_string
, 0, 0, 2)
222 ZEND_ARG_INFO(0, data
)
223 ZEND_ARG_INFO(0, type
)
226 zend_function_entry php_pqconv_methods
[] = {
227 PHP_ABSTRACT_ME(pqconv
, convertTypes
, ai_pqconv_convert_types
)
228 PHP_ABSTRACT_ME(pqconv
, convertFromString
, ai_pqconv_convert_from_string
)
229 PHP_ABSTRACT_ME(pqconv
, convertToString
, ai_pqconv_convert_to_string
)
234 PHP_MINIT_FUNCTION(pq_misc
)
236 zend_class_entry
*json
, ce
= {0};
238 INIT_NS_CLASS_ENTRY(ce
, "pq", "Converter", php_pqconv_methods
);
239 php_pqconv_class_entry
= zend_register_internal_interface(&ce
);
241 memset(&ce
, 0, sizeof(ce
));
242 INIT_NS_CLASS_ENTRY(ce
,"pq", "DateTime", php_pqdt_methods
);
243 php_pqdt_class_entry
= zend_register_internal_class_ex(&ce
, php_date_get_date_ce());
245 zend_declare_property_stringl(php_pqdt_class_entry
, ZEND_STRL("format"), ZEND_STRL("Y-m-d H:i:s.uO"), ZEND_ACC_PUBLIC
);
247 /* stop reading this file right here! */
248 if ((json
= zend_hash_str_find_ptr(CG(class_table
), ZEND_STRL("jsonserializable")))) {
249 zend_class_implements(php_pqdt_class_entry
, 1, json
);
255 typedef struct _HashTableList
{
257 struct _HashTableList
*parent
;
260 typedef struct _ArrayParserState
{
261 const char *ptr
, *end
;
269 static char caa(ArrayParserState
*a
, const char *any
, unsigned advance
)
280 php_error_docref(NULL
, E_WARNING
, "Failed to parse array: expected one of '%s', got '%c'", any
, *a
->ptr
); \
284 static ZEND_RESULT_CODE
add_element(ArrayParserState
*a
, const char *start
)
287 zend_string
*zstr
= zend_string_init(start
, a
->ptr
- start
, 0);
290 php_stripslashes(zstr
);
291 ZVAL_STR(&zelem
, zstr
);
292 } else if (!zend_string_equals_literal(zstr
, "NULL")) {
293 ZVAL_STR(&zelem
, zstr
);
295 zend_string_release(zstr
);
299 if (!ZVAL_IS_NULL(&zelem
)) {
300 php_pqres_typed_zval(a
->res
, a
->typ
, &zelem
);
303 add_next_index_zval(&a
->list
->arr
, &zelem
);
307 static ZEND_RESULT_CODE
parse_array(ArrayParserState
*a
);
309 static ZEND_RESULT_CODE
parse_element(ArrayParserState
*a
, char delim
)
315 return parse_array(a
);
323 for (el
= a
->ptr
; a
->ptr
< a
->end
; ++a
->ptr
) {
326 a
->escaped
= !a
->escaped
;
332 } else if (a
->quotes
) {
333 if (SUCCESS
!= add_element(a
, el
)) {
340 php_error_docref(NULL
, E_WARNING
, "Failed to parse element, unexpected quote: '%.*s'", (int) (a
->ptr
- el
), el
);
346 if (delim
!= *a
->ptr
) {
353 return add_element(a
, el
);
360 php_error_docref(NULL
, E_WARNING
, "Failed to parse element, reached end of input");
364 static ZEND_RESULT_CODE
parse_elements(ArrayParserState
*a
)
366 char delims
[] = {'}', PHP_PQ_DELIM_OF_ARRAY(a
->typ
), 0};
368 while (SUCCESS
== parse_element(a
, delims
[1])) {
369 switch (caa(a
, delims
, 0)) {
378 php_error_docref(NULL
, E_WARNING
, "Failed to parse elements, reached end of input");
388 static ZEND_RESULT_CODE
parse_array(ArrayParserState
*a
)
392 if (!caa(a
, "{", 1)) {
396 list
= ecalloc(1, sizeof(*list
));
397 array_init(&list
->arr
);
400 add_next_index_zval(&a
->list
->arr
, &list
->arr
);
401 list
->parent
= a
->list
;
405 if (SUCCESS
!= parse_elements(a
)) {
409 if (!caa(a
, "}", 1)) {
413 /* step one level back up */
414 if (a
->list
->parent
) {
415 HashTableList
*l
= a
->list
->parent
;
424 HashTable
*php_pq_parse_array(php_pqres_t
*res
, const char *val_str
, size_t val_len
, Oid typ
)
426 HashTable
*ht
= NULL
;
427 ArrayParserState a
= {0};
431 a
.end
= val_str
+ val_len
;
434 if (SUCCESS
!= parse_array(&a
)) {
436 HashTableList
*l
= a
.list
->parent
;
438 zval_dtor(&a
.list
->arr
);
446 php_error_docref(NULL
, E_NOTICE
, "Trailing input: '%s'", a
.ptr
);
450 HashTableList
*l
= a
.list
->parent
;
452 ht
= Z_ARRVAL(a
.list
->arr
);
466 * vim600: noet sw=4 ts=4 fdm=marker
467 * vim<600: noet sw=4 ts=4