2 +--------------------------------------------------------------------+
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-2006, Michael Wallner <mike@php.net> |
10 +--------------------------------------------------------------------+
15 #define HTTP_WANT_CURL
18 #if defined(ZEND_ENGINE_2) && defined(HTTP_HAVE_CURL)
20 #include "php_http_api.h"
21 #include "php_http_request_datashare_api.h"
22 #include "php_http_request_api.h"
23 #include "php_http_request_object.h"
24 #ifdef HTTP_HAVE_PERSISTENT_HANDLES
25 # include "php_http_persistent_handle_api.h"
28 #ifdef HTTP_HAVE_PERSISTENT_HANDLES
29 # define HTTP_CURL_SHARE_CTOR(ch) (SUCCESS == http_persistent_handle_acquire("http_request_datashare", &(ch)))
30 # define HTTP_CURL_SHARE_DTOR(chp) http_persistent_handle_release("http_request_datashare", (chp))
32 # define HTTP_CURL_SHARE_CTOR(ch) ((ch) = curl_share_init())
33 # define HTTP_CURL_SHARE_DTOR(chp) curl_share_cleanup(*(chp)); *(chp) = NULL
36 static HashTable http_request_datashare_options
;
37 static http_request_datashare http_request_datashare_global
;
38 static int http_request_datashare_compare_handles(void *h1
, void *h2
);
39 static void http_request_datashare_destroy_handles(void *el
);
41 static void http_request_datashare_lock_func(CURL
*handle
, curl_lock_data data
, curl_lock_access locktype
, void *userptr
);
42 static void http_request_datashare_unlock_func(CURL
*handle
, curl_lock_data data
, void *userptr
);
45 http_request_datashare
*_http_request_datashare_global_get(void)
47 return &http_request_datashare_global
;
50 PHP_MINIT_FUNCTION(http_request_datashare
)
54 #ifdef HTTP_HAVE_PERSISTENT_HANDLES
55 if (SUCCESS
!= http_persistent_handle_provide("http_request_datashare", curl_share_init
, (http_persistent_handle_dtor
) curl_share_cleanup
)) {
60 if (!http_request_datashare_init_ex(&http_request_datashare_global
, 1)) {
64 zend_hash_init(&http_request_datashare_options
, 4, NULL
, NULL
, 1);
65 #define ADD_DATASHARE_OPT(name, opt) \
67 zend_hash_add(&http_request_datashare_options, name, sizeof(name), &val, sizeof(curl_lock_data), NULL)
68 ADD_DATASHARE_OPT("cookie", CURL_LOCK_DATA_COOKIE
);
69 ADD_DATASHARE_OPT("dns", CURL_LOCK_DATA_DNS
);
70 ADD_DATASHARE_OPT("ssl", CURL_LOCK_DATA_SSL_SESSION
);
71 ADD_DATASHARE_OPT("connect", CURL_LOCK_DATA_CONNECT
);
76 PHP_MSHUTDOWN_FUNCTION(http_request_datashare
)
78 http_request_datashare_dtor(&http_request_datashare_global
);
79 zend_hash_destroy(&http_request_datashare_options
);
84 PHP_RINIT_FUNCTION(http_request_datashare
)
86 zend_llist_init(&HTTP_G
->request
.datashare
.handles
, sizeof(zval
*), http_request_datashare_destroy_handles
, 0);
91 PHP_RSHUTDOWN_FUNCTION(http_request_datashare
)
93 zend_llist_destroy(&HTTP_G
->request
.datashare
.handles
);
98 PHP_HTTP_API http_request_datashare
*_http_request_datashare_init_ex(http_request_datashare
*share
, zend_bool persistent TSRMLS_DC
)
100 zend_bool free_share
;
102 if ((free_share
= !share
)) {
103 share
= pemalloc(sizeof(http_request_datashare
), persistent
);
105 memset(share
, 0, sizeof(http_request_datashare
));
107 if (!HTTP_CURL_SHARE_CTOR(share
->ch
)) {
109 pefree(share
, persistent
);
114 if (!(share
->persistent
= persistent
)) {
115 share
->handles
= emalloc(sizeof(zend_llist
));
116 zend_llist_init(share
->handles
, sizeof(zval
*), ZVAL_PTR_DTOR
, 0);
121 share
->locks
= pecalloc(CURL_LOCK_DATA_LAST
, sizeof(http_request_datashare_lock
), 1);
122 for (i
= 0; i
< CURL_LOCK_DATA_LAST
; ++i
) {
123 share
->locks
[i
].mx
= tsrm_mutex_alloc();
125 curl_share_setopt(share
->ch
, CURLSHOPT_LOCKFUNC
, http_request_datashare_lock_func
);
126 curl_share_setopt(share
->ch
, CURLSHOPT_UNLOCKFUNC
, http_request_datashare_unlock_func
);
127 curl_share_setopt(share
->ch
, CURLSHOPT_USERDATA
, share
);
134 PHP_HTTP_API STATUS
_http_request_datashare_attach(http_request_datashare
*share
, zval
*request TSRMLS_DC
)
137 getObjectEx(http_request_object
, obj
, request
);
140 if (obj
->share
== share
) {
142 } else if (SUCCESS
!= http_request_datashare_detach(obj
->share
, request
)) {
147 HTTP_CHECK_CURL_INIT(obj
->request
->ch
, http_curl_init_ex(obj
->request
->ch
, obj
->request
), return FAILURE
);
148 if (CURLE_OK
!= (rc
= curl_easy_setopt(obj
->request
->ch
, CURLOPT_SHARE
, share
->ch
))) {
149 http_error_ex(HE_WARNING
, HTTP_E_REQUEST
, "Could not attach HttpRequest object(#%d) to the HttpRequestDataShare: %s", Z_OBJ_HANDLE_P(request
), curl_easy_strerror(rc
));
154 ZVAL_ADDREF(request
);
155 zend_llist_add_element(HTTP_RSHARE_HANDLES(share
), (void *) &request
);
160 PHP_HTTP_API STATUS
_http_request_datashare_detach(http_request_datashare
*share
, zval
*request TSRMLS_DC
)
163 getObjectEx(http_request_object
, obj
, request
);
166 http_error_ex(HE_WARNING
, HTTP_E_REQUEST
, "HttpRequest object(#%d) is not attached to any HttpRequestDataShare", Z_OBJ_HANDLE_P(request
));
167 } else if (obj
->share
!= share
) {
168 http_error_ex(HE_WARNING
, HTTP_E_REQUEST
, "HttpRequest object(#%d) is not attached to this HttpRequestDataShare", Z_OBJ_HANDLE_P(request
));
169 } else if (CURLE_OK
!= (rc
= curl_easy_setopt(obj
->request
->ch
, CURLOPT_SHARE
, NULL
))) {
170 http_error_ex(HE_WARNING
, HTTP_E_REQUEST
, "Could not detach HttpRequest object(#%d) from the HttpRequestDataShare: %s", Z_OBJ_HANDLE_P(request
), curl_share_strerror(rc
));
173 zend_llist_del_element(HTTP_RSHARE_HANDLES(share
), request
, http_request_datashare_compare_handles
);
179 PHP_HTTP_API
void _http_request_datashare_detach_all(http_request_datashare
*share TSRMLS_DC
)
183 while ((r
= zend_llist_get_first(HTTP_RSHARE_HANDLES(share
)))) {
184 http_request_datashare_detach(share
, *r
);
188 PHP_HTTP_API
void _http_request_datashare_dtor(http_request_datashare
*share TSRMLS_DC
)
190 if (!share
->persistent
) {
191 zend_llist_destroy(share
->handles
);
192 efree(share
->handles
);
194 HTTP_CURL_SHARE_DTOR(&share
->ch
);
196 if (share
->persistent
) {
199 for (i
= 0; i
< CURL_LOCK_DATA_LAST
; ++i
) {
200 tsrm_mutex_free(share
->locks
[i
].mx
);
202 pefree(share
->locks
, 1);
207 PHP_HTTP_API
void _http_request_datashare_free(http_request_datashare
**share TSRMLS_DC
)
209 http_request_datashare_dtor(*share
);
210 pefree(*share
, (*share
)->persistent
);
214 PHP_HTTP_API STATUS
_http_request_datashare_set(http_request_datashare
*share
, const char *option
, size_t option_len
, zend_bool enable TSRMLS_DC
)
219 if (SUCCESS
== zend_hash_find(&http_request_datashare_options
, (char *) option
, option_len
+ 1, (void *) &opt
)) {
220 if (CURLSHE_OK
== (rc
= curl_share_setopt(share
->ch
, enable
? CURLSHOPT_SHARE
: CURLSHOPT_UNSHARE
, *opt
))) {
223 http_error_ex(HE_WARNING
, HTTP_E_REQUEST
, "Could not %s sharing of %s data: %s", enable
? "enable" : "disable", option
, curl_share_strerror(rc
));
228 static int http_request_datashare_compare_handles(void *h1
, void *h2
)
230 return (Z_OBJ_HANDLE_PP((zval
**) h1
) == Z_OBJ_HANDLE_P((zval
*) h2
));
233 static void http_request_datashare_destroy_handles(void *el
)
235 zval
**r
= (zval
**) el
;
238 { /* gcc 2.95 needs these braces */
239 getObjectEx(http_request_object
, obj
, *r
);
241 curl_easy_setopt(obj
->request
->ch
, CURLOPT_SHARE
, NULL
);
247 static void http_request_datashare_lock_func(CURL
*handle
, curl_lock_data data
, curl_lock_access locktype
, void *userptr
)
249 http_request_datashare
*share
= (http_request_datashare
*) userptr
;
251 /* TSRM can't distinguish shared/exclusive locks */
252 tsrm_mutex_lock(share
->locks
[data
].mx
);
253 share
->locks
[data
].ch
= handle
;
256 static void http_request_datashare_unlock_func(CURL
*handle
, curl_lock_data data
, void *userptr
)
258 http_request_datashare
*share
= (http_request_datashare
*) userptr
;
260 if (share
->locks
[data
].ch
== handle
) {
261 tsrm_mutex_unlock(share
->locks
[data
].mx
);
266 #endif /* ZEND_ENGINE_2 && HTTP_HAVE_CURL */
274 * vim600: noet sw=4 ts=4 fdm=marker
275 * vim<600: noet sw=4 ts=4