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