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

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