git.m6w6.name Git - m6w6/ext-http/blob - php_http_request_datashare.c

   1 /*
   2     +--------------------------------------------------------------------+
   3     | PECL :: http                                                       |
   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) 2004-2011, Michael Wallner <mike@php.net>            |
  10     +--------------------------------------------------------------------+
  11 */
  12
  13 #include "php_http.h"
  14
  15 #include <curl/curl.h>
  16 #include <ext/standard/php_string.h>
  17 #include <ext/spl/spl_iterators.h>
  18
  19 static int php_http_request_datashare_compare_handles(void *h1, void *h2);
  20
  21 #ifdef ZTS
  22 static void *php_http_request_datashare_locks_init(void);
  23 static void php_http_request_datashare_locks_dtor(void *l);
  24 static void php_http_request_datashare_lock_func(CURL *handle, curl_lock_data data, curl_lock_access locktype, void *userptr);
  25 static void php_http_request_datashare_unlock_func(CURL *handle, curl_lock_data data, void *userptr);
  26 static MUTEX_T php_http_request_datashare_global_shares_lock;
  27 #endif
  28 static HashTable php_http_request_datashare_global_shares;
  29 php_http_request_datashare_t *php_http_request_datashare_global_get(const char *driver_str, size_t driver_len TSRMLS_DC)
  30 {
  31         php_http_request_datashare_t *s = NULL, **s_ptr;
  32         char *lower_str = php_strtolower(estrndup(driver_str, driver_len), driver_len);
  33
  34 #ifdef ZTS
  35         tsrm_mutex_lock(php_http_request_datashare_global_shares_lock);
  36 #endif
  37         if (zend_hash_find(&php_http_request_datashare_global_shares, lower_str, driver_len + 1, (void *) &s_ptr)) {
  38                 s = *s_ptr;
  39         } else {
  40                 php_http_request_factory_driver_t driver;
  41
  42                 if ((SUCCESS == php_http_request_factory_get_driver(driver_str, driver_len, &driver)) && driver.request_datashare_ops) {
  43                         s = php_http_request_datashare_init(NULL, driver.request_datashare_ops, NULL, NULL, 1 TSRMLS_CC);
  44                         zend_hash_add(&php_http_request_datashare_global_shares, lower_str, driver_len + 1, &s, sizeof(php_http_request_datashare_t *), NULL);
  45                 }
  46         }
  47 #ifdef ZTS
  48         tsrm_mutex_unlock(php_http_request_datashare_global_shares_lock);
  49 #endif
  50
  51         efree(lower_str);
  52         return s;
  53 }
  54
  55 PHP_HTTP_API php_http_request_datashare_t *php_http_request_datashare_init(php_http_request_datashare_t *h, php_http_request_datashare_ops_t *ops, php_http_resource_factory_t *rf, void *init_arg, zend_bool persistent TSRMLS_DC)
  56 {
  57         php_http_request_datashare_t *free_h = NULL;
  58
  59         if (!h) {
  60                 free_h = h = pemalloc(sizeof(*h), persistent);
  61         }
  62         memset(h, sizeof(*h), 0);
  63
  64         if (!(h->persistent = persistent)) {
  65                 h->requests = emalloc(sizeof(*h->requests));
  66                 zend_llist_init(h->requests, sizeof(zval *), ZVAL_PTR_DTOR, 0);
  67                 TSRMLS_SET_CTX(h->ts);
  68         }
  69         h->ops = ops;
  70         h->rf = rf ? rf : php_http_resource_factory_init(NULL, h->ops->rsrc, NULL, NULL);
  71
  72         if (h->ops->init) {
  73                 if (!(h = h->ops->init(h, init_arg))) {
  74                         if (free_h) {
  75                                 pefree(free_h, persistent);
  76                         }
  77                 }
  78         }
  79
  80         return h;
  81 }
  82
  83 PHP_HTTP_API php_http_request_datashare_t *php_http_request_datashare_copy(php_http_request_datashare_t *from, php_http_request_datashare_t *to)
  84 {
  85         if (from->ops->copy) {
  86                 return from->ops->copy(from, to);
  87         }
  88
  89         return NULL;
  90 }
  91
  92 PHP_HTTP_API void php_http_request_datashare_dtor(php_http_request_datashare_t *h)
  93 {
  94         if (h->ops->dtor) {
  95                 h->ops->dtor(h);
  96         }
  97         if (h->requests) {
  98                 zend_llist_destroy(h->requests);
  99                 pefree(h->requests, h->persistent);
 100                 h->requests = NULL;
 101         }
 102
 103         if (h->persistent_handle_id) {
 104                 zval_ptr_dtor(&h->persistent_handle_id);
 105         }
 106 }
 107
 108 PHP_HTTP_API void php_http_request_datashare_free(php_http_request_datashare_t **h)
 109 {
 110         php_http_request_datashare_dtor(*h);
 111         pefree(*h, (*h)->persistent);
 112         *h = NULL;
 113 }
 114
 115 PHP_HTTP_API STATUS php_http_request_datashare_attach(php_http_request_datashare_t *h, zval *request)
 116 {
 117         TSRMLS_FETCH_FROM_CTX(h->ts);
 118
 119         if (h->ops->attach) {
 120                 php_http_request_object_t *obj = zend_object_store_get_object(request TSRMLS_CC);
 121
 122                 if (SUCCESS == h->ops->attach(h, obj->request)) {
 123                         Z_ADDREF_P(request);
 124                         zend_llist_add_element(PHP_HTTP_REQUEST_DATASHARE_REQUESTS(h), &request);
 125                         return SUCCESS;
 126                 }
 127         }
 128
 129         return FAILURE;
 130 }
 131
 132 static int php_http_request_datashare_compare_handles(void *h1, void *h2)
 133 {
 134         return (Z_OBJ_HANDLE_PP((zval **) h1) == Z_OBJ_HANDLE_P((zval *) h2));
 135 }
 136
 137 PHP_HTTP_API STATUS php_http_request_datashare_detach(php_http_request_datashare_t *h, zval *request)
 138 {
 139         TSRMLS_FETCH_FROM_CTX(h->ts);
 140
 141         if (h->ops->detach) {
 142                 php_http_request_object_t *obj = zend_object_store_get_object(request TSRMLS_CC);
 143
 144                 if (SUCCESS == h->ops->detach(h, obj->request)) {
 145                         zend_llist_del_element(PHP_HTTP_REQUEST_DATASHARE_REQUESTS(h), request, php_http_request_datashare_compare_handles);
 146                         return SUCCESS;
 147                 }
 148         }
 149         return FAILURE;
 150 }
 151
 152 PHP_HTTP_API STATUS php_http_request_datashare_setopt(php_http_request_datashare_t *h, php_http_request_datashare_setopt_opt_t opt, void *arg)
 153 {
 154         if (h->ops->setopt) {
 155                 return h->ops->setopt(h, opt, arg);
 156         }
 157         return FAILURE;
 158 }
 159
 160 static void detach(void *r, void *h TSRMLS_DC)
 161 {
 162         ((php_http_request_datashare_t *) h)->ops->detach(h, ((php_http_request_object_t *) zend_object_store_get_object(*((zval **) r) TSRMLS_CC))->request);
 163 }
 164
 165 PHP_HTTP_API void php_http_request_datashare_reset(php_http_request_datashare_t *h)
 166 {
 167         TSRMLS_FETCH_FROM_CTX(h->ts);
 168
 169         if (h->ops->reset) {
 170                 h->ops->reset(h);
 171         } else if (h->ops->detach) {
 172                 zend_llist_apply_with_argument(PHP_HTTP_REQUEST_DATASHARE_REQUESTS(h), detach, h TSRMLS_CC);
 173         }
 174
 175         zend_llist_clean(PHP_HTTP_REQUEST_DATASHARE_REQUESTS(h));
 176 }
 177
 178 static void php_http_request_datashare_global_requests_dtor(void *el)
 179 {
 180         //php_http_request_datashare_detach(php_http_request_datashare_global_get(), *((zval **) el));
 181         zval_ptr_dtor(el);
 182 }
 183
 184 #define PHP_HTTP_BEGIN_ARGS(method, req_args)   PHP_HTTP_BEGIN_ARGS_EX(HttpRequestDataShare, method, 0, req_args)
 185 #define PHP_HTTP_EMPTY_ARGS(method)                             PHP_HTTP_EMPTY_ARGS_EX(HttpRequestDataShare, method, 0)
 186 #define PHP_HTTP_RSHARE_ME(method, visibility)  PHP_ME(HttpRequestDataShare, method, PHP_HTTP_ARGS(HttpRequestDataShare, method), visibility)
 187
 188 PHP_HTTP_EMPTY_ARGS(__construct);
 189 PHP_HTTP_EMPTY_ARGS(__destruct);
 190 PHP_HTTP_EMPTY_ARGS(reset);
 191 PHP_HTTP_EMPTY_ARGS(count);
 192
 193 PHP_HTTP_BEGIN_ARGS(attach, 1)
 194         PHP_HTTP_ARG_OBJ(http\\Request, request, 0)
 195 PHP_HTTP_END_ARGS;
 196 PHP_HTTP_BEGIN_ARGS(detach, 1)
 197         PHP_HTTP_ARG_OBJ(http\\Request, request, 0)
 198 PHP_HTTP_END_ARGS;
 199
 200 static void php_http_request_datashare_object_write_prop(zval *object, zval *member, zval *value, const zend_literal *literal_key TSRMLS_DC);
 201
 202 #define php_http_request_datashare_class_entry php_http_request_datashare_class_entry
 203 zend_class_entry *php_http_request_datashare_class_entry;
 204 zend_function_entry php_http_request_datashare_method_entry[] = {
 205         PHP_HTTP_RSHARE_ME(__construct, ZEND_ACC_PRIVATE|ZEND_ACC_CTOR)
 206         PHP_HTTP_RSHARE_ME(__destruct, ZEND_ACC_PUBLIC|ZEND_ACC_DTOR)
 207         PHP_HTTP_RSHARE_ME(count, ZEND_ACC_PUBLIC)
 208         PHP_HTTP_RSHARE_ME(attach, ZEND_ACC_PUBLIC)
 209         PHP_HTTP_RSHARE_ME(detach, ZEND_ACC_PUBLIC)
 210         PHP_HTTP_RSHARE_ME(reset, ZEND_ACC_PUBLIC)
 211         EMPTY_FUNCTION_ENTRY
 212 };
 213 static zend_object_handlers php_http_request_datashare_object_handlers;
 214
 215 zend_object_value php_http_request_datashare_object_new(zend_class_entry *ce TSRMLS_DC)
 216 {
 217         return php_http_request_datashare_object_new_ex(ce, NULL, NULL TSRMLS_CC);
 218 }
 219
 220 zend_object_value php_http_request_datashare_object_new_ex(zend_class_entry *ce, php_http_request_datashare_t *share, php_http_request_datashare_object_t **ptr TSRMLS_DC)
 221 {
 222         zend_object_value ov;
 223         php_http_request_datashare_object_t *o;
 224
 225         o = ecalloc(1, sizeof(*o));
 226         zend_object_std_init((zend_object *) o, ce TSRMLS_CC);
 227         object_properties_init((zend_object *) o, ce);
 228
 229         if (share) {
 230                 o->share = share;
 231         } else {
 232                 o->share = php_http_request_datashare_init(NULL, NULL, NULL, NULL, 0 TSRMLS_CC);
 233         }
 234
 235         if (ptr) {
 236                 *ptr = o;
 237         }
 238
 239         ov.handle = zend_objects_store_put(o, NULL, php_http_request_datashare_object_free, NULL TSRMLS_CC);
 240         ov.handlers = &php_http_request_datashare_object_handlers;
 241
 242         return ov;
 243 }
 244
 245 void php_http_request_datashare_object_free(void *object TSRMLS_DC)
 246 {
 247         php_http_request_datashare_object_t *o = (php_http_request_datashare_object_t *) object;
 248
 249         if (!o->share->persistent) {
 250                 php_http_request_datashare_free(&o->share);
 251         }
 252         zend_object_std_dtor((zend_object *) o TSRMLS_CC);
 253         efree(o);
 254 }
 255
 256 static void php_http_request_datashare_object_write_prop(zval *object, zval *member, zval *value, const zend_literal *literal_key TSRMLS_DC)
 257 {
 258         zend_property_info *pi;
 259
 260         if ((pi = zend_get_property_info(php_http_request_datashare_class_entry, member, 1 TSRMLS_CC))) {
 261                 zend_bool enable = i_zend_is_true(value);
 262                 php_http_request_datashare_setopt_opt_t opt;
 263                 php_http_request_datashare_object_t *obj = zend_object_store_get_object(object TSRMLS_CC);
 264
 265                 if (!strcmp(pi->name, "cookie")) {
 266                         opt = PHP_HTTP_REQUEST_DATASHARE_OPT_COOKIES;
 267                 } else if (!strcmp(pi->name, "dns")) {
 268                         opt = PHP_HTTP_REQUEST_DATASHARE_OPT_RESOLVER;
 269                 } else {
 270                         return;
 271                 }
 272
 273                 if (SUCCESS != php_http_request_datashare_setopt(obj->share, opt, &enable)) {
 274                         return;
 275                 }
 276         }
 277
 278         zend_get_std_object_handlers()->write_property(object, member, value, literal_key TSRMLS_CC);
 279 }
 280
 281 static zval **php_http_request_datashare_object_get_prop_ptr(zval *object, zval *member, const zend_literal *literal_key TSRMLS_DC)
 282 {
 283         zend_property_info *pi;
 284
 285         if ((pi = zend_get_property_info(php_http_request_datashare_class_entry, member, 1 TSRMLS_CC))) {
 286                 return &php_http_property_proxy_init(NULL, object, member TSRMLS_CC)->myself;
 287         }
 288
 289         return zend_get_std_object_handlers()->get_property_ptr_ptr(object, member, literal_key TSRMLS_CC);
 290 }
 291
 292
 293 PHP_METHOD(HttpRequestDataShare, __construct)
 294 {
 295         with_error_handling(EH_THROW, php_http_exception_class_entry) {
 296                 zend_parse_parameters_none();
 297         } end_error_handling();
 298 }
 299
 300 PHP_METHOD(HttpRequestDataShare, __destruct)
 301 {
 302         php_http_request_datashare_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 303
 304         if (SUCCESS == zend_parse_parameters_none()) {
 305                 ; /* we always want to clean up */
 306         }
 307
 308         php_http_request_datashare_reset(obj->share);
 309 }
 310
 311 PHP_METHOD(HttpRequestDataShare, count)
 312 {
 313         if (SUCCESS == zend_parse_parameters_none()) {
 314                 php_http_request_datashare_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 315
 316                 RETURN_LONG(zend_llist_count(PHP_HTTP_REQUEST_DATASHARE_REQUESTS(obj->share)));
 317         }
 318         RETURN_FALSE;
 319 }
 320
 321
 322 PHP_METHOD(HttpRequestDataShare, attach)
 323 {
 324         zval *request;
 325
 326         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "O", &request, php_http_request_class_entry)) {
 327                 php_http_request_datashare_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 328
 329                 RETURN_SUCCESS(php_http_request_datashare_attach(obj->share, request));
 330         }
 331         RETURN_FALSE;
 332
 333 }
 334
 335 PHP_METHOD(HttpRequestDataShare, detach)
 336 {
 337         zval *request;
 338
 339         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "O", &request, php_http_request_class_entry)) {
 340                 php_http_request_datashare_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 341
 342                 RETURN_SUCCESS(php_http_request_datashare_detach(obj->share, request));
 343         }
 344         RETURN_FALSE;
 345 }
 346
 347 PHP_METHOD(HttpRequestDataShare, reset)
 348 {
 349         if (SUCCESS == zend_parse_parameters_none()) {
 350                 php_http_request_datashare_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 351
 352                 php_http_request_datashare_reset(obj->share);
 353                 RETURN_TRUE;
 354         }
 355         RETURN_FALSE;
 356 }
 357
 358 PHP_MINIT_FUNCTION(http_request_datashare)
 359 {
 360         PHP_HTTP_REGISTER_CLASS(http\\Request, DataShare, http_request_datashare, php_http_object_class_entry, 0);
 361         php_http_request_datashare_class_entry->create_object = php_http_request_datashare_object_new;
 362         memcpy(&php_http_request_datashare_object_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
 363         php_http_request_datashare_object_handlers.clone_obj = NULL;
 364         php_http_request_datashare_object_handlers.write_property = php_http_request_datashare_object_write_prop;
 365         php_http_request_datashare_object_handlers.get_property_ptr_ptr = php_http_request_datashare_object_get_prop_ptr;
 366
 367         zend_class_implements(php_http_request_datashare_class_entry TSRMLS_CC, 1, spl_ce_Countable);
 368
 369         zend_declare_property_null(php_http_request_datashare_class_entry, ZEND_STRL("instance"), (ZEND_ACC_STATIC|ZEND_ACC_PRIVATE) TSRMLS_CC);
 370         zend_declare_property_bool(php_http_request_datashare_class_entry, ZEND_STRL("cookie"), 0, ZEND_ACC_PUBLIC TSRMLS_CC);
 371         zend_declare_property_bool(php_http_request_datashare_class_entry, ZEND_STRL("dns"), 0, ZEND_ACC_PUBLIC TSRMLS_CC);
 372
 373 #ifdef ZTS
 374         php_http_request_datashare_global_shares_lock = tsrm_mutex_alloc();
 375 #endif
 376         zend_hash_init(&php_http_request_datashare_global_shares, 0, NULL, (dtor_func_t) php_http_request_datashare_free, 1);
 377
 378         return SUCCESS;
 379 }
 380
 381 PHP_MSHUTDOWN_FUNCTION(http_request_datashare)
 382 {
 383         zend_hash_destroy(&php_http_request_datashare_global_shares);
 384 #ifdef ZTS
 385         tsrm_mutex_free(php_http_request_datashare_global_shares_lock);
 386 #endif
 387
 388         return SUCCESS;
 389 }
 390
 391 PHP_RINIT_FUNCTION(http_request_datashare)
 392 {
 393         zend_llist_init(&PHP_HTTP_G->request_datashare.requests, sizeof(zval *), php_http_request_datashare_global_requests_dtor, 0);
 394
 395         return SUCCESS;
 396 }
 397
 398 PHP_RSHUTDOWN_FUNCTION(http_request_datashare)
 399 {
 400         zend_llist_destroy(&PHP_HTTP_G->request_datashare.requests);
 401
 402         return SUCCESS;
 403 }
 404
 405
 406 /*
 407  * Local variables:
 408  * tab-width: 4
 409  * c-basic-offset: 4
 410  * End:
 411  * vim600: noet sw=4 ts=4 fdm=marker
 412  * vim<600: noet sw=4 ts=4
 413  */
 414