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