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