git.m6w6.name Git - m6w6/ext-http/blob - http_request_pool_api.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-2007, Michael Wallner <mike@php.net>            |
  10     +--------------------------------------------------------------------+
  11 */
  12
  13 /* $Id$ */
  14
  15 #define HTTP_WANT_CURL
  16 #define HTTP_WANT_EVENT
  17 #include "php_http.h"
  18
  19 #if defined(ZEND_ENGINE_2) && defined(HTTP_HAVE_CURL)
  20
  21 #include "php_http_api.h"
  22 #include "php_http_exception_object.h"
  23 #include "php_http_persistent_handle_api.h"
  24 #include "php_http_request_api.h"
  25 #include "php_http_request_object.h"
  26 #include "php_http_request_pool_api.h"
  27 #include "php_http_requestpool_object.h"
  28
  29 #ifndef HTTP_DEBUG_REQPOOLS
  30 #       define HTTP_DEBUG_REQPOOLS 0
  31 #endif
  32
  33 #ifdef HTTP_HAVE_EVENT
  34 typedef struct _http_request_pool_event_t {
  35         struct event evnt;
  36         http_request_pool *pool;
  37 } http_request_pool_event;
  38
  39 static void http_request_pool_timeout_callback(int socket, short action, void *event_data);
  40 static void http_request_pool_event_callback(int socket, short action, void *event_data);
  41 static int http_request_pool_socket_callback(CURL *easy, curl_socket_t s, int action, void *, void *);
  42 static void http_request_pool_timer_callback(CURLM *multi, long timeout_ms, void *timer_data);
  43 #endif
  44
  45 static int http_request_pool_compare_handles(void *h1, void *h2);
  46
  47 PHP_MINIT_FUNCTION(http_request_pool)
  48 {
  49         if (SUCCESS != http_persistent_handle_provide("http_request_pool", curl_multi_init, (http_persistent_handle_dtor) curl_multi_cleanup, NULL)) {
  50                 return FAILURE;
  51         }
  52         return SUCCESS;
  53 }
  54
  55 #ifdef HTTP_HAVE_EVENT
  56 PHP_RINIT_FUNCTION(http_request_pool)
  57 {
  58         if (!HTTP_G->request.pool.event.base && !(HTTP_G->request.pool.event.base = event_init())) {
  59                 return FAILURE;
  60         }
  61
  62         return SUCCESS;
  63 }
  64 #endif
  65
  66 /* {{{ http_request_pool *http_request_pool_init(http_request_pool *) */
  67 PHP_HTTP_API http_request_pool *_http_request_pool_init(http_request_pool *pool TSRMLS_DC)
  68 {
  69         zend_bool free_pool;
  70
  71 #if HTTP_DEBUG_REQPOOLS
  72         fprintf(stderr, "Initializing request pool %p\n", pool);
  73 #endif
  74
  75         if ((free_pool = (!pool))) {
  76                 pool = emalloc(sizeof(http_request_pool));
  77                 pool->ch = NULL;
  78         }
  79
  80         if (SUCCESS != http_persistent_handle_acquire("http_request_pool", &pool->ch)) {
  81                 if (free_pool) {
  82                         efree(pool);
  83                 }
  84                 return NULL;
  85         }
  86
  87         TSRMLS_SET_CTX(pool->tsrm_ls);
  88
  89 #ifdef HTTP_HAVE_EVENT
  90         pool->timeout = ecalloc(1, sizeof(struct event));
  91         curl_multi_setopt(pool->ch, CURLMOPT_SOCKETDATA, pool);
  92         curl_multi_setopt(pool->ch, CURLMOPT_SOCKETFUNCTION, http_request_pool_socket_callback);
  93         curl_multi_setopt(pool->ch, CURLMOPT_TIMERDATA, pool);
  94         curl_multi_setopt(pool->ch, CURLMOPT_TIMERFUNCTION, http_request_pool_timer_callback);
  95 #endif
  96
  97         pool->unfinished = 0;
  98         zend_llist_init(&pool->finished, sizeof(zval *), (llist_dtor_func_t) ZVAL_PTR_DTOR, 0);
  99         zend_llist_init(&pool->handles, sizeof(zval *), (llist_dtor_func_t) ZVAL_PTR_DTOR, 0);
 100
 101 #if HTTP_DEBUG_REQPOOLS
 102         fprintf(stderr, "Initialized request pool %p\n", pool);
 103 #endif
 104
 105         return pool;
 106 }
 107 /* }}} */
 108
 109 /* {{{ STATUS http_request_pool_attach(http_request_pool *, zval *) */
 110 PHP_HTTP_API STATUS _http_request_pool_attach(http_request_pool *pool, zval *request)
 111 {
 112         TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 113         getObjectEx(http_request_object, req, request);
 114
 115 #if HTTP_DEBUG_REQPOOLS
 116         fprintf(stderr, "Attaching HttpRequest(#%d) %p to pool %p\n", Z_OBJ_HANDLE_P(request), req, pool);
 117 #endif
 118
 119         if (req->pool) {
 120                 http_error_ex(HE_WARNING, HTTP_E_INVALID_PARAM, "HttpRequest object(#%d) is already member of %s HttpRequestPool", Z_OBJ_HANDLE_P(request), req->pool == pool ? "this" : "another");
 121         } else if (SUCCESS != http_request_object_requesthandler(req, request)) {
 122                 http_error_ex(HE_WARNING, HTTP_E_REQUEST, "Could not initialize HttpRequest object(#%d) for attaching to the HttpRequestPool", Z_OBJ_HANDLE_P(request));
 123         } else {
 124                 CURLMcode code = curl_multi_add_handle(pool->ch, req->request->ch);
 125
 126                 if (CURLM_OK != code) {
 127                         http_error_ex(HE_WARNING, HTTP_E_REQUEST_POOL, "Could not attach HttpRequest object(#%d) to the HttpRequestPool: %s", Z_OBJ_HANDLE_P(request), curl_multi_strerror(code));
 128                 } else {
 129                         req->pool = pool;
 130
 131                         ZVAL_ADDREF(request);
 132                         zend_llist_add_element(&pool->handles, &request);
 133                         ++pool->unfinished;
 134
 135 #if HTTP_DEBUG_REQPOOLS
 136                         fprintf(stderr, "> %d HttpRequests attached to pool %p\n", zend_llist_count(&pool->handles), pool);
 137 #endif
 138                         return SUCCESS;
 139                 }
 140         }
 141         return FAILURE;
 142 }
 143 /* }}} */
 144
 145 /* {{{ STATUS http_request_pool_detach(http_request_pool *, zval *) */
 146 PHP_HTTP_API STATUS _http_request_pool_detach(http_request_pool *pool, zval *request)
 147 {
 148         CURLMcode code;
 149         TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 150         getObjectEx(http_request_object, req, request);
 151
 152 #if HTTP_DEBUG_REQPOOLS
 153         fprintf(stderr, "Detaching HttpRequest(#%d) %p from pool %p\n", Z_OBJ_HANDLE_P(request), req, pool);
 154 #endif
 155
 156         if (!req->pool) {
 157                 /* not attached to any pool */
 158 #if HTTP_DEBUG_REQPOOLS
 159                 fprintf(stderr, "HttpRequest object(#%d) %p is not attached to any HttpRequestPool\n", Z_OBJ_HANDLE_P(request), req);
 160 #endif
 161         } else if (req->pool != pool) {
 162                 http_error_ex(HE_WARNING, HTTP_E_INVALID_PARAM, "HttpRequest object(#%d) is not attached to this HttpRequestPool", Z_OBJ_HANDLE_P(request));
 163         } else if (req->request->_in_progress_cb) {
 164                 http_error_ex(HE_WARNING, HTTP_E_REQUEST_POOL, "HttpRequest object(#%d) cannot be detached from the HttpRequestPool while executing the progress callback", Z_OBJ_HANDLE_P(request));
 165         } else if (CURLM_OK != (code = curl_multi_remove_handle(pool->ch, req->request->ch))) {
 166                 http_error_ex(HE_WARNING, HTTP_E_REQUEST_POOL, "Could not detach HttpRequest object(#%d) from the HttpRequestPool: %s", Z_OBJ_HANDLE_P(request), curl_multi_strerror(code));
 167         } else {
 168                 req->pool = NULL;
 169                 zend_llist_del_element(&pool->finished, request, http_request_pool_compare_handles);
 170                 zend_llist_del_element(&pool->handles, request, http_request_pool_compare_handles);
 171
 172 #if HTTP_DEBUG_REQPOOLS
 173                 fprintf(stderr, "> %d HttpRequests remaining in pool %p\n", zend_llist_count(&pool->handles), pool);
 174 #endif
 175
 176                 return SUCCESS;
 177         }
 178         return FAILURE;
 179 }
 180 /* }}} */
 181
 182 /* {{{ void http_request_pool_apply(http_request_pool *, http_request_pool_apply_func) */
 183 PHP_HTTP_API void _http_request_pool_apply(http_request_pool *pool, http_request_pool_apply_func cb)
 184 {
 185         int count = zend_llist_count(&pool->handles);
 186
 187         if (count) {
 188                 int i = 0;
 189                 zend_llist_position pos;
 190                 zval **handle, **handles = emalloc(count * sizeof(zval *));
 191
 192                 for (handle = zend_llist_get_first_ex(&pool->handles, &pos); handle; handle = zend_llist_get_next_ex(&pool->handles, &pos)) {
 193                         handles[i++] = *handle;
 194                 }
 195
 196                 /* should never happen */
 197                 if (i != count) {
 198                         zend_error(E_ERROR, "number of fetched request handles do not match overall count");
 199                         count = i;
 200                 }
 201
 202                 for (i = 0; i < count; ++i) {
 203                         if (cb(pool, handles[i])) {
 204                                 break;
 205                         }
 206                 }
 207                 efree(handles);
 208         }
 209 }
 210 /* }}} */
 211
 212 /* {{{ void http_request_pool_apply_with_arg(http_request_pool *, http_request_pool_apply_with_arg_func, void *) */
 213 PHP_HTTP_API void _http_request_pool_apply_with_arg(http_request_pool *pool, http_request_pool_apply_with_arg_func cb, void *arg)
 214 {
 215         int count = zend_llist_count(&pool->handles);
 216
 217         if (count) {
 218                 int i = 0;
 219                 zend_llist_position pos;
 220                 zval **handle, **handles = emalloc(count * sizeof(zval *));
 221
 222                 for (handle = zend_llist_get_first_ex(&pool->handles, &pos); handle; handle = zend_llist_get_next_ex(&pool->handles, &pos)) {
 223                         handles[i++] = *handle;
 224                 }
 225
 226                 /* should never happen */
 227                 if (i != count) {
 228                         zend_error(E_ERROR, "number of fetched request handles do not match overall count");
 229                         count = i;
 230                 }
 231
 232                 for (i = 0; i < count; ++i) {
 233                         if (cb(pool, handles[i], arg)) {
 234                                 break;
 235                         }
 236                 }
 237                 efree(handles);
 238         }
 239 }
 240 /* }}} */
 241
 242 /* {{{ void http_request_pool_detach_all(http_request_pool *) */
 243 PHP_HTTP_API void _http_request_pool_detach_all(http_request_pool *pool)
 244 {
 245 #if HTTP_DEBUG_REQPOOLS
 246         fprintf(stderr, "Detaching %d requests from pool %p\n", zend_llist_count(&pool->handles), pool);
 247 #endif
 248         http_request_pool_apply(pool, _http_request_pool_detach);
 249 }
 250 /* }}} */
 251
 252 /* {{{ STATUS http_request_pool_send(http_request_pool *) */
 253 PHP_HTTP_API STATUS _http_request_pool_send(http_request_pool *pool)
 254 {
 255         TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 256
 257 #if HTTP_DEBUG_REQPOOLS
 258         fprintf(stderr, "Attempt to send %d requests of pool %p\n", zend_llist_count(&pool->handles), pool);
 259 #endif
 260
 261 #ifdef HTTP_HAVE_EVENT
 262         if (pool->useevents) {
 263                 do {
 264 #if HTTP_DEBUG_REQPOOLS
 265                         fprintf(stderr, "& Starting event dispatcher of pool %p\n", pool);
 266 #endif
 267                         event_base_dispatch(HTTP_G->request.pool.event.base);
 268                 } while (pool->unfinished);
 269         } else
 270 #endif
 271         {
 272                 while (http_request_pool_perform(pool)) {
 273                         if (SUCCESS != http_request_pool_select(pool)) {
 274 #ifdef PHP_WIN32
 275                                 /* see http://msdn.microsoft.com/library/en-us/winsock/winsock/windows_sockets_error_codes_2.asp */
 276                                 http_error_ex(HE_WARNING, HTTP_E_SOCKET, "WinSock error: %d", WSAGetLastError());
 277 #else
 278                                 http_error(HE_WARNING, HTTP_E_SOCKET, strerror(errno));
 279 #endif
 280                                 return FAILURE;
 281                         }
 282                 }
 283         }
 284
 285 #if HTTP_DEBUG_REQPOOLS
 286         fprintf(stderr, "Finished sending %d HttpRequests of pool %p (still unfinished: %d)\n", zend_llist_count(&pool->handles), pool, pool->unfinished);
 287 #endif
 288
 289         return SUCCESS;
 290 }
 291 /* }}} */
 292
 293 /* {{{ void http_request_pool_dtor(http_request_pool *) */
 294 PHP_HTTP_API void _http_request_pool_dtor(http_request_pool *pool)
 295 {
 296         TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 297
 298 #if HTTP_DEBUG_REQPOOLS
 299         fprintf(stderr, "Destructing request pool %p\n", pool);
 300 #endif
 301
 302 #ifdef HTTP_HAVE_EVENT
 303         efree(pool->timeout);
 304 #endif
 305
 306         http_request_pool_detach_all(pool);
 307
 308         pool->unfinished = 0;
 309         zend_llist_clean(&pool->finished);
 310         zend_llist_clean(&pool->handles);
 311         http_persistent_handle_release("http_request_pool", &pool->ch);
 312 }
 313 /* }}} */
 314
 315 #ifdef PHP_WIN32
 316 #       define SELECT_ERROR SOCKET_ERROR
 317 #else
 318 #       define SELECT_ERROR -1
 319 #endif
 320
 321 /* {{{ STATUS http_request_pool_select(http_request_pool *) */
 322 PHP_HTTP_API STATUS _http_request_pool_select(http_request_pool *pool)
 323 {
 324         int MAX;
 325         fd_set R, W, E;
 326         struct timeval timeout;
 327
 328 #ifdef HTTP_HAVE_EVENT
 329         if (pool->useevents) {
 330                 TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 331                 http_error(HE_WARNING, HTTP_E_RUNTIME, "not implemented; use HttpRequest callbacks");
 332                 return FAILURE;
 333         }
 334 #endif
 335
 336         http_request_pool_timeout(pool, &timeout);
 337
 338         FD_ZERO(&R);
 339         FD_ZERO(&W);
 340         FD_ZERO(&E);
 341
 342         if (CURLM_OK == curl_multi_fdset(pool->ch, &R, &W, &E, &MAX)) {
 343                 if (MAX == -1) {
 344                         http_sleep((double) timeout.tv_sec + (double) (timeout.tv_usec / HTTP_MCROSEC));
 345                         return SUCCESS;
 346                 } else if (SELECT_ERROR != select(MAX + 1, &R, &W, &E, &timeout)) {
 347                         return SUCCESS;
 348                 }
 349         }
 350         return FAILURE;
 351 }
 352 /* }}} */
 353
 354 /* {{{ int http_request_pool_perform(http_request_pool *) */
 355 PHP_HTTP_API int _http_request_pool_perform(http_request_pool *pool)
 356 {
 357         TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 358
 359 #ifdef HTTP_HAVE_EVENT
 360         if (pool->useevents) {
 361                 http_error(HE_WARNING, HTTP_E_RUNTIME, "not implemented; use HttpRequest callbacks");
 362                 return FAILURE;
 363         }
 364 #endif
 365
 366         while (CURLM_CALL_MULTI_PERFORM == curl_multi_perform(pool->ch, &pool->unfinished));
 367
 368 #if HTTP_DEBUG_REQPOOLS
 369         fprintf(stderr, "%u unfinished requests of pool %p remaining\n", pool->unfinished, pool);
 370 #endif
 371
 372         http_request_pool_responsehandler(pool);
 373
 374         return pool->unfinished;
 375 }
 376 /* }}} */
 377
 378 /* {{{ void http_request_pool_responsehandler(http_request_pool *) */
 379 void _http_request_pool_responsehandler(http_request_pool *pool)
 380 {
 381         CURLMsg *msg;
 382         int remaining = 0;
 383         TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 384
 385         do {
 386                 msg = curl_multi_info_read(pool->ch, &remaining);
 387                 if (msg && CURLMSG_DONE == msg->msg) {
 388                         if (CURLE_OK != msg->data.result) {
 389                                 http_request_storage *st = http_request_storage_get(msg->easy_handle);
 390                                 http_error_ex(HE_WARNING, HTTP_E_REQUEST, "%s; %s (%s)", curl_easy_strerror(msg->data.result), st?st->errorbuffer:"", st?st->url:"");
 391                         }
 392                         http_request_pool_apply_with_arg(pool, _http_request_pool_apply_responsehandler, msg->easy_handle);
 393                 }
 394         } while (remaining);
 395 }
 396 /* }}} */
 397
 398 /* {{{ int http_request_pool_apply_responsehandler(http_request_pool *, zval *, void *) */
 399 int _http_request_pool_apply_responsehandler(http_request_pool *pool, zval *req, void *ch)
 400 {
 401         TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 402         getObjectEx(http_request_object, obj, req);
 403
 404         if ((!ch) || obj->request->ch == (CURL *) ch) {
 405
 406 #if HTTP_DEBUG_REQPOOLS
 407                 fprintf(stderr, "Fetching data from HttpRequest(#%d) %p of pool %p\n", Z_OBJ_HANDLE_P(req), obj, obj->pool);
 408 #endif
 409
 410                 ZVAL_ADDREF(req);
 411                 zend_llist_add_element(&obj->pool->finished, &req);
 412                 http_request_object_responsehandler(obj, req);
 413                 return 1;
 414         }
 415         return 0;
 416 }
 417 /* }}} */
 418
 419 /* {{{ struct timeval *_http_request_pool_timeout(http_request_pool *, struct timeval *) */
 420 struct timeval *_http_request_pool_timeout(http_request_pool *pool, struct timeval *timeout)
 421 {
 422 #ifdef HAVE_CURL_MULTI_TIMEOUT
 423         long max_tout = 1000;
 424
 425         if ((CURLM_OK == curl_multi_timeout(pool->ch, &max_tout)) && (max_tout > 0)) {
 426                 timeout->tv_sec = max_tout / 1000;
 427                 timeout->tv_usec = (max_tout % 1000) * 1000;
 428         } else {
 429 #endif
 430                 timeout->tv_sec = 0;
 431                 timeout->tv_usec = 1000;
 432 #ifdef HAVE_CURL_MULTI_TIMEOUT
 433         }
 434 #endif
 435
 436 #if HTTP_DEBUG_REQPOOLS
 437         fprintf(stderr, "Calculating timeout (%lu, %lu) of pool %p\n", (ulong) timeout->tv_sec, (ulong) timeout->tv_usec, pool);
 438 #endif
 439
 440         return timeout;
 441 }
 442 /* }}} */
 443
 444 /*#*/
 445
 446 /* {{{ static int http_request_pool_compare_handles(void *, void *) */
 447 static int http_request_pool_compare_handles(void *h1, void *h2)
 448 {
 449         return (Z_OBJ_HANDLE_PP((zval **) h1) == Z_OBJ_HANDLE_P((zval *) h2));
 450 }
 451 /* }}} */
 452
 453 #ifdef HTTP_HAVE_EVENT
 454 /* {{{ static void http_request_pool_timeout_callback(int, short, void *) */
 455 static void http_request_pool_timeout_callback(int socket, short action, void *event_data)
 456 {
 457         http_request_pool *pool = event_data;
 458
 459         if (pool->useevents) {
 460                 CURLMcode rc;
 461                 TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 462
 463 #if HTTP_DEBUG_REQPOOLS
 464                 fprintf(stderr, "Timeout occurred of pool %p\n", pool);
 465 #endif
 466
 467                 while (CURLM_CALL_MULTI_PERFORM == (rc = curl_multi_socket(pool->ch, CURL_SOCKET_TIMEOUT, &pool->unfinished)));
 468
 469                 if (CURLM_OK != rc) {
 470                         http_error(HE_WARNING, HTTP_E_SOCKET, curl_multi_strerror(rc));
 471                 }
 472
 473                 http_request_pool_responsehandler(pool);
 474         }
 475 }
 476 /* }}} */
 477
 478 /* {{{ static void http_request_pool_event_callback(int, short, void *) */
 479 static void http_request_pool_event_callback(int socket, short action, void *event_data)
 480 {
 481         http_request_pool_event *ev = event_data;
 482         http_request_pool *pool = ev->pool;
 483
 484         if (pool->useevents) {
 485                 CURLMcode rc = CURLE_OK;
 486                 TSRMLS_FETCH_FROM_CTX(ev->pool->tsrm_ls);
 487
 488 #if HTTP_DEBUG_REQPOOLS
 489                 {
 490                         static const char event_strings[][20] = {"NONE","TIMEOUT","READ","TIMEOUT|READ","WRITE","TIMEOUT|WRITE","READ|WRITE","TIMEOUT|READ|WRITE","SIGNAL"};
 491                         fprintf(stderr, "Event on socket %d (%s) event %p of pool %p\n", socket, event_strings[action], ev, pool);
 492                 }
 493 #endif
 494
 495                 /* don't use 'ev' below this loop as it might 've been freed in the socket callback */
 496                 do {
 497 #ifdef HAVE_CURL_MULTI_SOCKET_ACTION
 498                         switch (action & (EV_READ|EV_WRITE)) {
 499                                 case EV_READ:
 500                                         rc = curl_multi_socket_action(pool->ch, socket, CURL_CSELECT_IN, &pool->unfinished);
 501                                         break;
 502                                 case EV_WRITE:
 503                                         rc = curl_multi_socket_action(pool->ch, socket, CURL_CSELECT_OUT, &pool->unfinished);
 504                                         break;
 505                                 case EV_READ|EV_WRITE:
 506                                         rc = curl_multi_socket_action(pool->ch, socket, CURL_CSELECT_IN|CURL_CSELECT_OUT, &pool->unfinished);
 507                                         break;
 508                                 default:
 509                                         http_error_ex(HE_WARNING, HTTP_E_SOCKET, "Unknown event %d", (int) action);
 510                                         return;
 511                         }
 512 #else
 513                         rc = curl_multi_socket(pool->ch, socket, &pool->unfinished);
 514 #endif
 515                 } while (CURLM_CALL_MULTI_PERFORM == rc);
 516
 517                 switch (rc) {
 518                         case CURLM_BAD_SOCKET:
 519 #if 0
 520                                 fprintf(stderr, "!!! Bad socket: %d (%d)\n", socket, (int) action);
 521 #endif
 522                         case CURLM_OK:
 523                                 break;
 524                         default:
 525                                 http_error(HE_WARNING, HTTP_E_SOCKET, curl_multi_strerror(rc));
 526                                 break;
 527                 }
 528
 529                 http_request_pool_responsehandler(pool);
 530
 531                 /* remove timeout if there are no transfers left */
 532                 if (!pool->unfinished && event_initialized(pool->timeout) && event_pending(pool->timeout, EV_TIMEOUT, NULL)) {
 533                         event_del(pool->timeout);
 534 #if HTTP_DEBUG_REQPOOLS
 535                         fprintf(stderr, "Removed timeout of pool %p\n", pool);
 536 #endif
 537                 }
 538         }
 539 }
 540 /* }}} */
 541
 542 /* {{{ static int http_request_pool_socket_callback(CURL *, curl_socket_t, int, void *, void *) */
 543 static int http_request_pool_socket_callback(CURL *easy, curl_socket_t sock, int action, void *socket_data, void *assign_data)
 544 {
 545         http_request_pool *pool = socket_data;
 546
 547         if (pool->useevents) {
 548                 int events = EV_PERSIST;
 549                 http_request_pool_event *ev = assign_data;
 550                 TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 551
 552                 if (!ev) {
 553                         ev = ecalloc(1, sizeof(http_request_pool_event));
 554                         ev->pool = pool;
 555                         curl_multi_assign(pool->ch, sock, ev);
 556                         event_base_set(HTTP_G->request.pool.event.base, &ev->evnt);
 557                 } else {
 558                         event_del(&ev->evnt);
 559                 }
 560
 561 #if HTTP_DEBUG_REQPOOLS
 562                 {
 563                         static const char action_strings[][8] = {"NONE", "IN", "OUT", "INOUT", "REMOVE"};
 564                         http_request *r;
 565                         curl_easy_getinfo(easy, CURLINFO_PRIVATE, &r);
 566                         fprintf(stderr, "Callback on socket %2d (%8s) event %p of pool %p (%d)\n", (int) sock, action_strings[action], ev, pool, pool->unfinished);
 567                 }
 568 #endif
 569
 570                 switch (action) {
 571                         case CURL_POLL_IN:
 572                                 events |= EV_READ;
 573                                 break;
 574                         case CURL_POLL_OUT:
 575                                 events |= EV_WRITE;
 576                                 break;
 577                         case CURL_POLL_INOUT:
 578                                 events |= EV_READ|EV_WRITE;
 579                                 break;
 580
 581                         case CURL_POLL_REMOVE:
 582                                 efree(ev);
 583                         case CURL_POLL_NONE:
 584                                 return 0;
 585
 586                         default:
 587                                 http_error_ex(HE_WARNING, HTTP_E_SOCKET, "Unknown socket action %d", action);
 588                                 return -1;
 589                 }
 590
 591                 event_set(&ev->evnt, sock, events, http_request_pool_event_callback, ev);
 592                 event_add(&ev->evnt, NULL);
 593         }
 594
 595         return 0;
 596 }
 597 /* }}} */
 598
 599 /* {{{ static void http_request_pool_timer_callback(CURLM *, long, void*) */
 600 static void http_request_pool_timer_callback(CURLM *multi, long timeout_ms, void *timer_data)
 601 {
 602         http_request_pool *pool = timer_data;
 603
 604         if (pool->useevents) {
 605                 TSRMLS_FETCH_FROM_CTX(pool->tsrm_ls);
 606                 struct timeval timeout;
 607
 608                 if (!event_initialized(pool->timeout)) {
 609                         event_set(pool->timeout, -1, 0, http_request_pool_timeout_callback, pool);
 610                         event_base_set(HTTP_G->request.pool.event.base, pool->timeout);
 611                 } else if (event_pending(pool->timeout, EV_TIMEOUT, NULL)) {
 612                         event_del(pool->timeout);
 613                 }
 614
 615                 if (timeout_ms > 0) {
 616                         timeout.tv_sec = timeout_ms / 1000;
 617                         timeout.tv_usec = (timeout_ms % 1000) * 1000;
 618                 } else {
 619                         http_request_pool_timeout(pool, &timeout);
 620                 }
 621
 622                 event_add(pool->timeout, &timeout);
 623
 624 #if HTTP_DEBUG_REQPOOLS
 625                 fprintf(stderr, "Updating timeout %lu (%lu, %lu) of pool %p\n", (ulong) timeout_ms, (ulong) timeout.tv_sec, (ulong) timeout.tv_usec, pool);
 626 #endif
 627         }
 628 }
 629 /* }}} */
 630 #endif /* HTTP_HAVE_EVENT */
 631
 632 #endif /* ZEND_ENGINE_2 && HTTP_HAVE_CURL */
 633
 634
 635 /*
 636  * Local variables:
 637  * tab-width: 4
 638  * c-basic-offset: 4
 639  * End:
 640  * vim600: noet sw=4 ts=4 fdm=marker
 641  * vim<600: noet sw=4 ts=4
 642  */
 643