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 #if PHP_VERSION_ID >= 80100
135 ZEND_BEGIN_ARG_WITH_TENTATIVE_RETURN_TYPE_INFO_EX(ai_pqdt_jsonserialize
, 0, 0, IS_MIXED
, 0)
138 #define ai_pqdt_jsonserialize ai_pqdt_to_string
141 #if PHP_VERSION_ID >= 80200
142 ZEND_BEGIN_ARG_WITH_RETURN_TYPE_INFO_EX(ai_pqdt_to_string
, 0, 0, IS_STRING
, 0)
144 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_to_string
, 0, 0, 0)
147 static PHP_METHOD(pqdt
, __toString
)
152 php_pq_call_method(getThis(), "format", 1, &rv
, php_pq_read_property(getThis(), "format", &tmp
));
153 RETVAL_ZVAL(&rv
, 1, 1);
156 #if PHP_VERSION_ID >= 80100
157 ZEND_BEGIN_ARG_WITH_TENTATIVE_RETURN_OBJ_TYPE_MASK_EX(ai_pqdt_create_from_format
, 0, 2, DateTime
, MAY_BE_FALSE
)
159 ZEND_BEGIN_ARG_INFO_EX(ai_pqdt_create_from_format
, 0, 0, 2)
161 ZEND_ARG_INFO(0, format
)
162 ZEND_ARG_INFO(0, datetime
)
163 #if PHP_VERSION_ID >= 70200
164 ZEND_ARG_OBJ_INFO(0, object
, DateTimeZone
, 1)
166 ZEND_ARG_INFO(0, timezone
)
169 static PHP_METHOD(pqdt
, createFromFormat
)
171 zend_error_handling zeh
;
172 char *fmt_str
, *dt_str
;
173 size_t fmt_len
, dt_len
;
177 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh
);
178 rv
= zend_parse_parameters(ZEND_NUM_ARGS(), "ss|O", &fmt_str
, &fmt_len
, &dt_str
, &dt_len
, &ztz
, php_date_get_timezone_ce());
179 zend_restore_error_handling(&zeh
);
182 php_pqdt_from_string(return_value
, fmt_str
, dt_str
, dt_len
, "Y-m-d H:i:s.uO", ztz
);
186 static zend_function_entry php_pqdt_methods
[] = {
187 PHP_ME(pqdt
, createFromFormat
, ai_pqdt_create_from_format
, ZEND_ACC_PUBLIC
|ZEND_ACC_STATIC
)
188 PHP_ME(pqdt
, __toString
, ai_pqdt_to_string
, ZEND_ACC_PUBLIC
)
189 PHP_MALIAS(pqdt
, jsonSerialize
, __toString
, ai_pqdt_jsonserialize
, ZEND_ACC_PUBLIC
)
193 zval
*php_pqdt_from_string(zval
*zv
, char *input_fmt
, char *dt_str
, size_t dt_len
, const char *output_fmt
, zval
*ztimezone
)
197 php_date_instantiate(php_pqdt_class_entry
, zv
);
198 dobj
= php_date_obj_from_obj(Z_OBJ_P(zv
));
199 if (!php_date_initialize(dobj
, dt_str
, dt_len
, input_fmt
, ztimezone
, 1)) {
202 } else if (output_fmt
) {
204 ZVAL_STRING(&fmt
, output_fmt
);
205 php_pq_update_property(zv
, "format", &fmt
);
212 zend_string
*php_pqdt_to_string(zval
*zdt
, const char *format
)
218 if (php_pq_cast_object(zdt
, IS_STRING
, &rv
)) {
220 } else if (instanceof_function(Z_OBJCE_P(zdt
), php_date_get_date_ce())) {
224 ZVAL_STRING(&zfmt
, format
);
225 php_pq_call_method(zdt
, "format", 1, &rv
, &zfmt
);
226 zval_ptr_dtor(&zfmt
);
228 if (Z_TYPE(rv
) == IS_STRING
) {
237 zend_class_entry
*php_pqconv_class_entry
;
239 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_types
, 0, 0, 0)
242 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_from_string
, 0, 0, 2)
243 ZEND_ARG_INFO(0, data
)
244 ZEND_ARG_INFO(0, type
)
247 ZEND_BEGIN_ARG_INFO_EX(ai_pqconv_convert_to_string
, 0, 0, 2)
248 ZEND_ARG_INFO(0, data
)
249 ZEND_ARG_INFO(0, type
)
252 zend_function_entry php_pqconv_methods
[] = {
253 PHP_ABSTRACT_ME(pqconv
, convertTypes
, ai_pqconv_convert_types
)
254 PHP_ABSTRACT_ME(pqconv
, convertFromString
, ai_pqconv_convert_from_string
)
255 PHP_ABSTRACT_ME(pqconv
, convertToString
, ai_pqconv_convert_to_string
)
260 PHP_MINIT_FUNCTION(pq_misc
)
262 zend_class_entry
*json
, ce
= {0};
264 INIT_NS_CLASS_ENTRY(ce
, "pq", "Converter", php_pqconv_methods
);
265 php_pqconv_class_entry
= zend_register_internal_interface(&ce
);
267 memset(&ce
, 0, sizeof(ce
));
268 INIT_NS_CLASS_ENTRY(ce
,"pq", "DateTime", php_pqdt_methods
);
269 php_pqdt_class_entry
= zend_register_internal_class_ex(&ce
, php_date_get_date_ce());
271 zend_declare_property_stringl(php_pqdt_class_entry
, ZEND_STRL("format"), ZEND_STRL("Y-m-d H:i:s.uO"), ZEND_ACC_PUBLIC
);
273 /* stop reading this file right here! */
274 if ((json
= zend_hash_str_find_ptr(CG(class_table
), ZEND_STRL("jsonserializable")))) {
275 zend_class_implements(php_pqdt_class_entry
, 1, json
);
281 typedef struct _HashTableList
{
283 struct _HashTableList
*parent
;
286 typedef struct _ArrayParserState
{
287 const char *ptr
, *end
;
295 static char caa(ArrayParserState
*a
, const char *any
, unsigned advance
)
306 php_error_docref(NULL
, E_WARNING
, "Failed to parse array: expected one of '%s', got '%c'", any
, *a
->ptr
); \
310 static ZEND_RESULT_CODE
add_element(ArrayParserState
*a
, const char *start
)
313 zend_string
*zstr
= zend_string_init(start
, a
->ptr
- start
, 0);
316 php_stripslashes(zstr
);
317 ZVAL_STR(&zelem
, zstr
);
318 } else if (!zend_string_equals_literal(zstr
, "NULL")) {
319 ZVAL_STR(&zelem
, zstr
);
321 zend_string_release(zstr
);
325 if (!ZVAL_IS_NULL(&zelem
)) {
326 php_pqres_typed_zval(a
->res
, a
->typ
, &zelem
);
329 add_next_index_zval(&a
->list
->arr
, &zelem
);
333 static ZEND_RESULT_CODE
parse_array(ArrayParserState
*a
);
335 static ZEND_RESULT_CODE
parse_element(ArrayParserState
*a
, char delim
)
341 return parse_array(a
);
352 for (el
= a
->ptr
; a
->ptr
< a
->end
; ++a
->ptr
) {
355 a
->escaped
= !a
->escaped
;
361 } else if (a
->quotes
) {
362 if (SUCCESS
!= add_element(a
, el
)) {
369 php_error_docref(NULL
, E_WARNING
, "Failed to parse element, unexpected quote: '%.*s'", (int) (a
->ptr
- el
), el
);
375 if (delim
!= *a
->ptr
) {
382 return add_element(a
, el
);
389 php_error_docref(NULL
, E_WARNING
, "Failed to parse element, reached end of input");
393 static ZEND_RESULT_CODE
parse_elements(ArrayParserState
*a
)
395 char delims
[] = {'}', (char) PHP_PQ_DELIM_OF_ARRAY(a
->typ
), 0};
397 while (SUCCESS
== parse_element(a
, delims
[1])) {
398 switch (caa(a
, delims
, 0)) {
407 php_error_docref(NULL
, E_WARNING
, "Failed to parse elements, reached end of input");
417 static ZEND_RESULT_CODE
parse_array(ArrayParserState
*a
)
421 if (!caa(a
, "{", 1)) {
425 list
= ecalloc(1, sizeof(*list
));
426 array_init(&list
->arr
);
429 add_next_index_zval(&a
->list
->arr
, &list
->arr
);
430 list
->parent
= a
->list
;
434 if (SUCCESS
!= parse_elements(a
)) {
438 if (!caa(a
, "}", 1)) {
442 /* step one level back up */
443 if (a
->list
->parent
) {
444 HashTableList
*l
= a
->list
->parent
;
453 HashTable
*php_pq_parse_array(php_pqres_t
*res
, const char *val_str
, size_t val_len
, Oid typ
)
455 HashTable
*ht
= NULL
;
456 ArrayParserState a
= {0};
460 a
.end
= val_str
+ val_len
;
463 if (SUCCESS
!= parse_array(&a
)) {
465 HashTableList
*l
= a
.list
->parent
;
467 zval_dtor(&a
.list
->arr
);
475 php_error_docref(NULL
, E_NOTICE
, "Trailing input: '%s'", a
.ptr
);
479 HashTableList
*l
= a
.list
->parent
;
481 ht
= Z_ARRVAL(a
.list
->arr
);
495 * vim600: noet sw=4 ts=4 fdm=marker
496 * vim<600: noet sw=4 ts=4