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

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