123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254 |
- #include "curl_setup.h"
- #include <curl/curl.h>
- #include "urldata.h"
- #include "share.h"
- #include "vtls/vtls.h"
- #include "curl_memory.h"
- #include "memdebug.h"
- CURLSH *
- curl_share_init(void)
- {
- struct Curl_share *share = calloc(1, sizeof(struct Curl_share));
- if(share)
- share->specifier |= (1<<CURL_LOCK_DATA_SHARE);
- return share;
- }
- #undef curl_share_setopt
- CURLSHcode
- curl_share_setopt(CURLSH *sh, CURLSHoption option, ...)
- {
- struct Curl_share *share = (struct Curl_share *)sh;
- va_list param;
- int type;
- curl_lock_function lockfunc;
- curl_unlock_function unlockfunc;
- void *ptr;
- CURLSHcode res = CURLSHE_OK;
- if(share->dirty)
-
- return CURLSHE_IN_USE;
- va_start(param, option);
- switch(option) {
- case CURLSHOPT_SHARE:
-
- type = va_arg(param, int);
- share->specifier |= (1<<type);
- switch( type ) {
- case CURL_LOCK_DATA_DNS:
- if(!share->hostcache) {
- share->hostcache = Curl_mk_dnscache();
- if(!share->hostcache)
- res = CURLSHE_NOMEM;
- }
- break;
- case CURL_LOCK_DATA_COOKIE:
- #if !defined(CURL_DISABLE_HTTP) && !defined(CURL_DISABLE_COOKIES)
- if(!share->cookies) {
- share->cookies = Curl_cookie_init(NULL, NULL, NULL, TRUE );
- if(!share->cookies)
- res = CURLSHE_NOMEM;
- }
- #else
- res = CURLSHE_NOT_BUILT_IN;
- #endif
- break;
- case CURL_LOCK_DATA_SSL_SESSION:
- #ifdef USE_SSL
- if(!share->sslsession) {
- share->max_ssl_sessions = 8;
- share->sslsession = calloc(share->max_ssl_sessions,
- sizeof(struct curl_ssl_session));
- share->sessionage = 0;
- if(!share->sslsession)
- res = CURLSHE_NOMEM;
- }
- #else
- res = CURLSHE_NOT_BUILT_IN;
- #endif
- break;
- case CURL_LOCK_DATA_CONNECT:
- break;
- default:
- res = CURLSHE_BAD_OPTION;
- }
- break;
- case CURLSHOPT_UNSHARE:
-
- type = va_arg(param, int);
- share->specifier &= ~(1<<type);
- switch( type ) {
- case CURL_LOCK_DATA_DNS:
- if(share->hostcache) {
- Curl_hash_destroy(share->hostcache);
- share->hostcache = NULL;
- }
- break;
- case CURL_LOCK_DATA_COOKIE:
- #if !defined(CURL_DISABLE_HTTP) && !defined(CURL_DISABLE_COOKIES)
- if(share->cookies) {
- Curl_cookie_cleanup(share->cookies);
- share->cookies = NULL;
- }
- #else
- res = CURLSHE_NOT_BUILT_IN;
- #endif
- break;
- case CURL_LOCK_DATA_SSL_SESSION:
- #ifdef USE_SSL
- Curl_safefree(share->sslsession);
- #else
- res = CURLSHE_NOT_BUILT_IN;
- #endif
- break;
- case CURL_LOCK_DATA_CONNECT:
- break;
- default:
- res = CURLSHE_BAD_OPTION;
- break;
- }
- break;
- case CURLSHOPT_LOCKFUNC:
- lockfunc = va_arg(param, curl_lock_function);
- share->lockfunc = lockfunc;
- break;
- case CURLSHOPT_UNLOCKFUNC:
- unlockfunc = va_arg(param, curl_unlock_function);
- share->unlockfunc = unlockfunc;
- break;
- case CURLSHOPT_USERDATA:
- ptr = va_arg(param, void *);
- share->clientdata = ptr;
- break;
- default:
- res = CURLSHE_BAD_OPTION;
- break;
- }
- va_end(param);
- return res;
- }
- CURLSHcode
- curl_share_cleanup(CURLSH *sh)
- {
- struct Curl_share *share = (struct Curl_share *)sh;
- if(share == NULL)
- return CURLSHE_INVALID;
- if(share->lockfunc)
- share->lockfunc(NULL, CURL_LOCK_DATA_SHARE, CURL_LOCK_ACCESS_SINGLE,
- share->clientdata);
- if(share->dirty) {
- if(share->unlockfunc)
- share->unlockfunc(NULL, CURL_LOCK_DATA_SHARE, share->clientdata);
- return CURLSHE_IN_USE;
- }
- if(share->hostcache) {
- Curl_hash_destroy(share->hostcache);
- share->hostcache = NULL;
- }
- #if !defined(CURL_DISABLE_HTTP) && !defined(CURL_DISABLE_COOKIES)
- if(share->cookies)
- Curl_cookie_cleanup(share->cookies);
- #endif
- #ifdef USE_SSL
- if(share->sslsession) {
- size_t i;
- for(i = 0; i < share->max_ssl_sessions; i++)
- Curl_ssl_kill_session(&(share->sslsession[i]));
- free(share->sslsession);
- }
- #endif
- if(share->unlockfunc)
- share->unlockfunc(NULL, CURL_LOCK_DATA_SHARE, share->clientdata);
- free(share);
- return CURLSHE_OK;
- }
- CURLSHcode
- Curl_share_lock(struct SessionHandle *data, curl_lock_data type,
- curl_lock_access accesstype)
- {
- struct Curl_share *share = data->share;
- if(share == NULL)
- return CURLSHE_INVALID;
- if(share->specifier & (1<<type)) {
- if(share->lockfunc)
- share->lockfunc(data, type, accesstype, share->clientdata);
- }
-
- return CURLSHE_OK;
- }
- CURLSHcode
- Curl_share_unlock(struct SessionHandle *data, curl_lock_data type)
- {
- struct Curl_share *share = data->share;
- if(share == NULL)
- return CURLSHE_INVALID;
- if(share->specifier & (1<<type)) {
- if(share->unlockfunc)
- share->unlockfunc (data, type, share->clientdata);
- }
- return CURLSHE_OK;
- }
|