git.m6w6.name Git - m6w6/ext-pq/blob - src/php_pqstm.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/standard/php_smart_str.h>
  19
  20 #include "php_pq.h"
  21 #include "php_pq_misc.h"
  22 #include "php_pq_object.h"
  23 #include "php_pqexc.h"
  24 #include "php_pqconn.h"
  25 #include "php_pqres.h"
  26 #include "php_pqstm.h"
  27
  28 zend_class_entry *php_pqstm_class_entry;
  29 static zend_object_handlers php_pqstm_object_handlers;
  30 static HashTable php_pqstm_object_prophandlers;
  31
  32 static void php_pqstm_object_free(void *o TSRMLS_DC)
  33 {
  34         php_pqstm_object_t *obj = o;
  35 #if DBG_GC
  36         fprintf(stderr, "FREE stm(#%d) %p (conn(#%d): %p)\n", obj->zv.handle, obj, obj->intern->conn->zv.handle, obj->intern->conn);
  37 #endif
  38         if (obj->intern) {
  39                 char *quoted_name = PQescapeIdentifier(obj->intern->conn->intern->conn, obj->intern->name, strlen(obj->intern->name));
  40
  41                 php_pq_callback_dtor(&obj->intern->conn->intern->onevent);
  42
  43                 if (quoted_name) {
  44                         PGresult *res;
  45                         smart_str cmd = {0};
  46
  47                         smart_str_appends(&cmd, "DEALLOCATE ");
  48                         smart_str_appends(&cmd, quoted_name);
  49                         smart_str_0(&cmd);
  50                         PQfreemem(quoted_name);
  51
  52                         if ((res = PQexec(obj->intern->conn->intern->conn, cmd.c))) {
  53                                 PHP_PQclear(res);
  54                         }
  55                         smart_str_free(&cmd);
  56                 }
  57
  58                 php_pq_object_delref(obj->intern->conn TSRMLS_CC);
  59                 efree(obj->intern->name);
  60                 zend_hash_destroy(&obj->intern->bound);
  61                 if (obj->intern->params) {
  62                         php_pq_params_free(&obj->intern->params);
  63                 }
  64                 efree(obj->intern);
  65                 obj->intern = NULL;
  66         }
  67         zend_object_std_dtor((zend_object *) o TSRMLS_CC);
  68         efree(obj);
  69 }
  70
  71 zend_object_value php_pqstm_create_object_ex(zend_class_entry *ce, php_pqstm_t *intern, php_pqstm_object_t **ptr TSRMLS_DC)
  72 {
  73         php_pqstm_object_t *o;
  74
  75         o = ecalloc(1, sizeof(*o));
  76         zend_object_std_init((zend_object *) o, ce TSRMLS_CC);
  77         object_properties_init((zend_object *) o, ce);
  78         o->prophandler = &php_pqstm_object_prophandlers;
  79
  80         if (ptr) {
  81                 *ptr = o;
  82         }
  83
  84         if (intern) {
  85                 o->intern = intern;
  86         }
  87
  88         o->zv.handle = zend_objects_store_put((zend_object *) o, NULL, php_pqstm_object_free, NULL TSRMLS_CC);
  89         o->zv.handlers = &php_pqstm_object_handlers;
  90
  91         return o->zv;
  92 }
  93
  94 static zend_object_value php_pqstm_create_object(zend_class_entry *class_type TSRMLS_DC)
  95 {
  96         return php_pqstm_create_object_ex(class_type, NULL, NULL TSRMLS_CC);
  97 }
  98
  99 static void php_pqstm_object_read_name(zval *object, void *o, zval *return_value TSRMLS_DC)
 100 {
 101         php_pqstm_object_t *obj = o;
 102
 103         RETVAL_STRING(obj->intern->name, 1);
 104 }
 105
 106 static void php_pqstm_object_read_connection(zval *object, void *o, zval *return_value TSRMLS_DC)
 107 {
 108         php_pqstm_object_t *obj = o;
 109
 110         php_pq_object_to_zval(obj->intern->conn, &return_value TSRMLS_CC);
 111 }
 112
 113
 114 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_construct, 0, 0, 3)
 115         ZEND_ARG_OBJ_INFO(0, Connection, pq\\Connection, 0)
 116         ZEND_ARG_INFO(0, type)
 117         ZEND_ARG_INFO(0, query)
 118         ZEND_ARG_ARRAY_INFO(0, types, 1)
 119         ZEND_ARG_INFO(0, async)
 120 ZEND_END_ARG_INFO();
 121 static PHP_METHOD(pqstm, __construct) {
 122         zend_error_handling zeh;
 123         zval *zconn, *ztypes = NULL;
 124         char *name_str, *query_str;
 125         int name_len, *query_len;
 126         zend_bool async = 0;
 127         STATUS rv;
 128
 129         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 130         rv = zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "Oss|a/!b", &zconn, php_pqconn_class_entry, &name_str, &name_len, &query_str, &query_len, &ztypes, &async);
 131         zend_restore_error_handling(&zeh TSRMLS_CC);
 132
 133         if (SUCCESS == rv) {
 134                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 135                 php_pqconn_object_t *conn_obj = zend_object_store_get_object(zconn TSRMLS_CC);
 136
 137                 if (!conn_obj->intern) {
 138                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Connection not initialized");
 139                 } else {
 140                         php_pq_params_t *params = php_pq_params_init(&conn_obj->intern->converters, ztypes ? Z_ARRVAL_P(ztypes) : NULL, NULL TSRMLS_CC);
 141
 142                         if (async) {
 143                                 rv = php_pqconn_prepare_async(zconn, conn_obj, name_str, query_str, params TSRMLS_CC);
 144                         } else {
 145                                 rv = php_pqconn_prepare(zconn, conn_obj, name_str, query_str, params TSRMLS_CC);
 146                         }
 147
 148                         if (SUCCESS == rv) {
 149                                 php_pqstm_t *stm = ecalloc(1, sizeof(*stm));
 150
 151                                 php_pq_object_addref(conn_obj TSRMLS_CC);
 152                                 stm->conn = conn_obj;
 153                                 stm->name = estrdup(name_str);
 154                                 stm->params = params;
 155                                 ZEND_INIT_SYMTABLE(&stm->bound);
 156                                 obj->intern = stm;
 157                         }
 158                 }
 159         }
 160 }
 161 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_bind, 0, 0, 2)
 162         ZEND_ARG_INFO(0, param_no)
 163         ZEND_ARG_INFO(1, param_ref)
 164 ZEND_END_ARG_INFO();
 165 static PHP_METHOD(pqstm, bind) {
 166         long param_no;
 167         zval *param_ref;
 168
 169         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "lz", &param_no, &param_ref)) {
 170                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 171
 172                 if (!obj->intern) {
 173                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 174                 } else {
 175                         Z_ADDREF_P(param_ref);
 176                         zend_hash_index_update(&obj->intern->bound, param_no, (void *) &param_ref, sizeof(zval *), NULL);
 177                         zend_hash_sort(&obj->intern->bound, zend_qsort, compare_index, 0 TSRMLS_CC);
 178                 }
 179         }
 180 }
 181
 182 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_exec, 0, 0, 0)
 183         ZEND_ARG_ARRAY_INFO(0, params, 1)
 184 ZEND_END_ARG_INFO();
 185 static PHP_METHOD(pqstm, exec) {
 186         zend_error_handling zeh;
 187         zval *zparams = NULL;
 188         STATUS rv;
 189
 190         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 191         rv = zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|a/!", &zparams);
 192         zend_restore_error_handling(&zeh TSRMLS_CC);
 193
 194         if (SUCCESS == rv) {
 195                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 196
 197                 if (!obj->intern) {
 198                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 199                 } else {
 200                         PGresult *res;
 201
 202                         php_pq_params_set_params(obj->intern->params, zparams ? Z_ARRVAL_P(zparams) : &obj->intern->bound);
 203                         res = PQexecPrepared(obj->intern->conn->intern->conn, obj->intern->name, obj->intern->params->param.count, (const char *const*) obj->intern->params->param.strings, NULL, NULL, 0);
 204                         php_pq_params_set_params(obj->intern->params, NULL);
 205
 206                         if (!res) {
 207                                 throw_exce(EX_RUNTIME TSRMLS_CC, "Failed to execute statement (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 208                         } else if (SUCCESS == php_pqres_success(res TSRMLS_CC)) {
 209                                 php_pq_object_to_zval_no_addref(PQresultInstanceData(res, php_pqconn_event), &return_value TSRMLS_CC);
 210                                 php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 211                         }
 212                 }
 213         }
 214 }
 215
 216 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_exec_async, 0, 0, 0)
 217         ZEND_ARG_ARRAY_INFO(0, params, 1)
 218         ZEND_ARG_INFO(0, callable)
 219 ZEND_END_ARG_INFO();
 220 static PHP_METHOD(pqstm, execAsync) {
 221         zend_error_handling zeh;
 222         zval *zparams = NULL;
 223         php_pq_callback_t resolver = {{0}};
 224         STATUS rv;
 225
 226         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 227         rv = zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|a/!f", &zparams, &resolver.fci, &resolver.fcc);
 228         zend_restore_error_handling(&zeh TSRMLS_CC);
 229
 230         if (SUCCESS == rv) {
 231                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 232
 233                 if (!obj->intern) {
 234                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 235                 } else {
 236                         int rc;
 237
 238                         php_pq_params_set_params(obj->intern->params, zparams ? Z_ARRVAL_P(zparams) : &obj->intern->bound);
 239                         rc = PQsendQueryPrepared(obj->intern->conn->intern->conn, obj->intern->name, obj->intern->params->param.count, (const char *const*) obj->intern->params->param.strings, NULL, NULL, 0);
 240                         php_pq_params_set_params(obj->intern->params, NULL);
 241
 242                         if (!rc) {
 243                                 throw_exce(EX_IO TSRMLS_CC, "Failed to execute statement (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 244                         } else if (obj->intern->conn->intern->unbuffered && !PQsetSingleRowMode(obj->intern->conn->intern->conn)) {
 245                                 throw_exce(EX_RUNTIME TSRMLS_CC, "Failed to enable unbuffered mode (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 246                         } else {
 247                                 php_pq_callback_dtor(&obj->intern->conn->intern->onevent);
 248                                 if (resolver.fci.size > 0) {
 249                                         obj->intern->conn->intern->onevent = resolver;
 250                                         php_pq_callback_addref(&obj->intern->conn->intern->onevent);
 251                                 }
 252                                 obj->intern->conn->intern->poller = PQconsumeInput;
 253                         }
 254
 255                         php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 256                 }
 257         }
 258 }
 259
 260 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_desc, 0, 0, 0)
 261 ZEND_END_ARG_INFO();
 262 static PHP_METHOD(pqstm, desc) {
 263         zend_error_handling zeh;
 264         STATUS rv;
 265
 266         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 267         rv = zend_parse_parameters_none();
 268         zend_restore_error_handling(&zeh TSRMLS_CC);
 269
 270         if (SUCCESS == rv) {
 271                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 272
 273                 if (!obj->intern) {
 274                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 275                 } else {
 276                         PGresult *res = PQdescribePrepared(obj->intern->conn->intern->conn, obj->intern->name);
 277
 278                         if (!res) {
 279                                 throw_exce(EX_RUNTIME TSRMLS_CC, "Failed to describe statement (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 280                         } else {
 281                                 if (SUCCESS == php_pqres_success(res TSRMLS_CC)) {
 282                                         int p, params;
 283
 284                                         array_init(return_value);
 285                                         for (p = 0, params = PQnparams(res); p < params; ++p) {
 286                                                 add_next_index_long(return_value, PQparamtype(res, p));
 287                                         }
 288                                 }
 289                                 PHP_PQclear(res);
 290                                 php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 291                         }
 292                 }
 293         }
 294 }
 295
 296 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_desc_async, 0, 0, 0)
 297 ZEND_END_ARG_INFO();
 298 static PHP_METHOD(pqstm, descAsync) {
 299         zend_error_handling zeh;
 300         STATUS rv;
 301
 302         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 303         rv = zend_parse_parameters_none();
 304         zend_restore_error_handling(&zeh TSRMLS_CC);
 305
 306         if (SUCCESS == rv) {
 307                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 308
 309                 if (!obj->intern) {
 310                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 311                 } else if (!PQsendDescribePrepared(obj->intern->conn->intern->conn, obj->intern->name)) {
 312                         throw_exce(EX_IO TSRMLS_CC, "Failed to describe statement: %s", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 313                 } else {
 314                         obj->intern->conn->intern->poller = PQconsumeInput;
 315                         php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 316                 }
 317         }
 318 }
 319
 320 static zend_function_entry php_pqstm_methods[] = {
 321         PHP_ME(pqstm, __construct, ai_pqstm_construct, ZEND_ACC_PUBLIC|ZEND_ACC_CTOR)
 322         PHP_ME(pqstm, bind, ai_pqstm_bind, ZEND_ACC_PUBLIC)
 323         PHP_ME(pqstm, exec, ai_pqstm_exec, ZEND_ACC_PUBLIC)
 324         PHP_ME(pqstm, desc, ai_pqstm_desc, ZEND_ACC_PUBLIC)
 325         PHP_ME(pqstm, execAsync, ai_pqstm_exec_async, ZEND_ACC_PUBLIC)
 326         PHP_ME(pqstm, descAsync, ai_pqstm_desc_async, ZEND_ACC_PUBLIC)
 327         {0}
 328 };
 329
 330 PHP_MSHUTDOWN_FUNCTION(pqstm)
 331 {
 332         zend_hash_destroy(&php_pqstm_object_prophandlers);
 333         return SUCCESS;
 334 }
 335
 336 PHP_MINIT_FUNCTION(pqstm)
 337 {
 338         zend_class_entry ce = {0};
 339         php_pq_object_prophandler_t ph = {0};
 340
 341         INIT_NS_CLASS_ENTRY(ce, "pq", "Statement", php_pqstm_methods);
 342         php_pqstm_class_entry = zend_register_internal_class_ex(&ce, NULL, NULL TSRMLS_CC);
 343         php_pqstm_class_entry->create_object = php_pqstm_create_object;
 344
 345         memcpy(&php_pqstm_object_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
 346         php_pqstm_object_handlers.read_property = php_pq_object_read_prop;
 347         php_pqstm_object_handlers.write_property = php_pq_object_write_prop;
 348         php_pqstm_object_handlers.clone_obj = NULL;
 349         php_pqstm_object_handlers.get_property_ptr_ptr = NULL;
 350         php_pqstm_object_handlers.get_gc = NULL;
 351         php_pqstm_object_handlers.get_properties = php_pq_object_properties;
 352         php_pqstm_object_handlers.get_debug_info = php_pq_object_debug_info;
 353
 354         zend_hash_init(&php_pqstm_object_prophandlers, 2, NULL, NULL, 1);
 355
 356         zend_declare_property_null(php_pqstm_class_entry, ZEND_STRL("name"), ZEND_ACC_PUBLIC TSRMLS_CC);
 357         ph.read = php_pqstm_object_read_name;
 358         zend_hash_add(&php_pqstm_object_prophandlers, "name", sizeof("name"), (void *) &ph, sizeof(ph), NULL);
 359
 360         zend_declare_property_null(php_pqstm_class_entry, ZEND_STRL("connection"), ZEND_ACC_PUBLIC TSRMLS_CC);
 361         ph.read = php_pqstm_object_read_connection;
 362         zend_hash_add(&php_pqstm_object_prophandlers, "connection", sizeof("connection"), (void *) &ph, sizeof(ph), NULL);
 363
 364         return SUCCESS;
 365 }
 366
 367 /*
 368  * Local variables:
 369  * tab-width: 4
 370  * c-basic-offset: 4
 371  * End:
 372  * vim600: noet sw=4 ts=4 fdm=marker
 373  * vim<600: noet sw=4 ts=4
 374  */