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 static inline int compare_index(zend_ulong l
, zend_ulong r
)
116 #if PHP_VERSION_ID >= 80000
117 int php_pq_compare_index(Bucket
*lptr
, Bucket
*rptr
)
119 return compare_index(lptr
->h
, rptr
->h
);
122 int php_pq_compare_index(const void *lptr
, const void *rptr
) {
123 return compare_index(((const Bucket
*) lptr
)->h
, ((const Bucket
*) rptr
)->h
);
127 void php_pq_hash_ptr_dtor(zval
*p
)
132 zend_class_entry
*php_pqdt_class_entry
;
134 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_to_string
, 0, 0, 0)
136 static PHP_METHOD(pqdt
, __toString
)
141 php_pq_call_method(getThis(), "format", 1, &rv
, php_pq_read_property(getThis(), "format", &tmp
));
142 RETVAL_ZVAL(&rv
, 1, 1);
145 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_create_from_format
, 0, 0, 2)
146 ZEND_ARG_INFO(0, format
)
147 ZEND_ARG_INFO(0, datetime
)
148 #if PHP_VERSION_ID >= 70200
149 ZEND_ARG_OBJ_INFO(0, object
, DateTimeZone
, 1)
151 ZEND_ARG_INFO(0, timezone
)
154 static PHP_METHOD(pqdt
, createFromFormat
)
156 zend_error_handling zeh
;
157 char *fmt_str
, *dt_str
;
158 size_t fmt_len
, dt_len
;
162 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh
);
163 rv
= zend_parse_parameters(ZEND_NUM_ARGS(), "ss|O", &fmt_str
, &fmt_len
, &dt_str
, &dt_len
, &ztz
, php_date_get_timezone_ce());
164 zend_restore_error_handling(&zeh
);
167 php_pqdt_from_string(return_value
, fmt_str
, dt_str
, dt_len
, "Y-m-d H:i:s.uO", ztz
);
171 static zend_function_entry php_pqdt_methods
[] = {
172 PHP_ME(pqdt
, createFromFormat
, ai_pqdt_create_from_format
, ZEND_ACC_PUBLIC
|ZEND_ACC_STATIC
)
173 PHP_ME(pqdt
, __toString
, ai_pqdt_to_string
, ZEND_ACC_PUBLIC
)
174 PHP_MALIAS(pqdt
, jsonSerialize
, __toString
, ai_pqdt_to_string
, ZEND_ACC_PUBLIC
)
178 zval
*php_pqdt_from_string(zval
*zv
, char *input_fmt
, char *dt_str
, size_t dt_len
, const char *output_fmt
, zval
*ztimezone
)
182 php_date_instantiate(php_pqdt_class_entry
, zv
);
183 dobj
= php_date_obj_from_obj(Z_OBJ_P(zv
));
184 if (!php_date_initialize(dobj
, dt_str
, dt_len
, input_fmt
, ztimezone
, 1)) {
187 } else if (output_fmt
) {
189 ZVAL_STRING(&fmt
, output_fmt
);
190 php_pq_update_property(zv
, "format", &fmt
);
197 zend_string
*php_pqdt_to_string(zval
*zdt
, const char *format
)
203 if (php_pq_cast_object(zdt
, IS_STRING
, &rv
)) {
205 } else if (instanceof_function(Z_OBJCE_P(zdt
), php_date_get_date_ce())) {
209 ZVAL_STRING(&zfmt
, format
);
210 php_pq_call_method(zdt
, "format", 1, &rv
, &zfmt
);
211 zval_ptr_dtor(&zfmt
);
213 if (Z_TYPE(rv
) == IS_STRING
) {
222 zend_class_entry
*php_pqconv_class_entry
;
224 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_types
, 0, 0, 0)
227 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_from_string
, 0, 0, 2)
228 ZEND_ARG_INFO(0, data
)
229 ZEND_ARG_INFO(0, type
)
232 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_to_string
, 0, 0, 2)
233 ZEND_ARG_INFO(0, data
)
234 ZEND_ARG_INFO(0, type
)
237 zend_function_entry php_pqconv_methods
[] = {
238 PHP_ABSTRACT_ME(pqconv
, convertTypes
, ai_pqconv_convert_types
)
239 PHP_ABSTRACT_ME(pqconv
, convertFromString
, ai_pqconv_convert_from_string
)
240 PHP_ABSTRACT_ME(pqconv
, convertToString
, ai_pqconv_convert_to_string
)
245 PHP_MINIT_FUNCTION(pq_misc
)
247 zend_class_entry
*json
, ce
= {0};
249 INIT_NS_CLASS_ENTRY(ce
, "pq", "Converter", php_pqconv_methods
);
250 php_pqconv_class_entry
= zend_register_internal_interface(&ce
);
252 memset(&ce
, 0, sizeof(ce
));
253 INIT_NS_CLASS_ENTRY(ce
,"pq", "DateTime", php_pqdt_methods
);
254 php_pqdt_class_entry
= zend_register_internal_class_ex(&ce
, php_date_get_date_ce());
256 zend_declare_property_stringl(php_pqdt_class_entry
, ZEND_STRL("format"), ZEND_STRL("Y-m-d H:i:s.uO"), ZEND_ACC_PUBLIC
);
258 /* stop reading this file right here! */
259 if ((json
= zend_hash_str_find_ptr(CG(class_table
), ZEND_STRL("jsonserializable")))) {
260 zend_class_implements(php_pqdt_class_entry
, 1, json
);
266 typedef struct _HashTableList
{
268 struct _HashTableList
*parent
;
271 typedef struct _ArrayParserState
{
272 const char *ptr
, *end
;
280 static char caa(ArrayParserState
*a
, const char *any
, unsigned advance
)
291 php_error_docref(NULL
, E_WARNING
, "Failed to parse array: expected one of '%s', got '%c'", any
, *a
->ptr
); \
295 static ZEND_RESULT_CODE
add_element(ArrayParserState
*a
, const char *start
)
298 zend_string
*zstr
= zend_string_init(start
, a
->ptr
- start
, 0);
301 php_stripslashes(zstr
);
302 ZVAL_STR(&zelem
, zstr
);
303 } else if (!zend_string_equals_literal(zstr
, "NULL")) {
304 ZVAL_STR(&zelem
, zstr
);
306 zend_string_release(zstr
);
310 if (!ZVAL_IS_NULL(&zelem
)) {
311 php_pqres_typed_zval(a
->res
, a
->typ
, &zelem
);
314 add_next_index_zval(&a
->list
->arr
, &zelem
);
318 static ZEND_RESULT_CODE
parse_array(ArrayParserState
*a
);
320 static ZEND_RESULT_CODE
parse_element(ArrayParserState
*a
, char delim
)
326 return parse_array(a
);
337 for (el
= a
->ptr
; a
->ptr
< a
->end
; ++a
->ptr
) {
340 a
->escaped
= !a
->escaped
;
346 } else if (a
->quotes
) {
347 if (SUCCESS
!= add_element(a
, el
)) {
354 php_error_docref(NULL
, E_WARNING
, "Failed to parse element, unexpected quote: '%.*s'", (int) (a
->ptr
- el
), el
);
360 if (delim
!= *a
->ptr
) {
367 return add_element(a
, el
);
374 php_error_docref(NULL
, E_WARNING
, "Failed to parse element, reached end of input");
378 static ZEND_RESULT_CODE
parse_elements(ArrayParserState
*a
)
380 char delims
[] = {'}', (char) PHP_PQ_DELIM_OF_ARRAY(a
->typ
), 0};
382 while (SUCCESS
== parse_element(a
, delims
[1])) {
383 switch (caa(a
, delims
, 0)) {
392 php_error_docref(NULL
, E_WARNING
, "Failed to parse elements, reached end of input");
402 static ZEND_RESULT_CODE
parse_array(ArrayParserState
*a
)
406 if (!caa(a
, "{", 1)) {
410 list
= ecalloc(1, sizeof(*list
));
411 array_init(&list
->arr
);
414 add_next_index_zval(&a
->list
->arr
, &list
->arr
);
415 list
->parent
= a
->list
;
419 if (SUCCESS
!= parse_elements(a
)) {
423 if (!caa(a
, "}", 1)) {
427 /* step one level back up */
428 if (a
->list
->parent
) {
429 HashTableList
*l
= a
->list
->parent
;
438 HashTable
*php_pq_parse_array(php_pqres_t
*res
, const char *val_str
, size_t val_len
, Oid typ
)
440 HashTable
*ht
= NULL
;
441 ArrayParserState a
= {0};
445 a
.end
= val_str
+ val_len
;
448 if (SUCCESS
!= parse_array(&a
)) {
450 HashTableList
*l
= a
.list
->parent
;
452 zval_dtor(&a
.list
->arr
);
460 php_error_docref(NULL
, E_NOTICE
, "Trailing input: '%s'", a
.ptr
);
464 HashTableList
*l
= a
.list
->parent
;
466 ht
= Z_ARRVAL(a
.list
->arr
);
480 * vim600: noet sw=4 ts=4 fdm=marker
481 * vim<600: noet sw=4 ts=4