1 /***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 1998 - 2018, Daniel Stenberg, <daniel@haxx.se>, et al.
9 *
10 * This software is licensed as described in the file COPYING, which
11 * you should have received as part of this distribution. The terms
12 * are also available at https://curl.haxx.se/docs/copyright.html.
13 *
14 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15 * copies of the Software, and permit persons to whom the Software is
16 * furnished to do so, under the terms of the COPYING file.
17 *
18 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19 * KIND, either express or implied.
20 *
21 ***************************************************************************/
22
23 #include "curl_setup.h"
24
25 #include <curl/curl.h>
26 #include "urldata.h"
27 #include "share.h"
28 #include "psl.h"
29 #include "vtls/vtls.h"
30 #include "curl_memory.h"
31
32 /* The last #include file should be: */
33 #include "memdebug.h"
34
35 struct Curl_share *
curl_share_init(void)36 curl_share_init(void)
37 {
38 struct Curl_share *share = calloc(1, sizeof(struct Curl_share));
39 if(share) {
40 share->specifier |= (1<<CURL_LOCK_DATA_SHARE);
41
42 if(Curl_mk_dnscache(&share->hostcache)) {
43 free(share);
44 return NULL;
45 }
46 }
47
48 return share;
49 }
50
51 #undef curl_share_setopt
52 CURLSHcode
curl_share_setopt(struct Curl_share * share,CURLSHoption option,...)53 curl_share_setopt(struct Curl_share *share, CURLSHoption option, ...)
54 {
55 va_list param;
56 int type;
57 curl_lock_function lockfunc;
58 curl_unlock_function unlockfunc;
59 void *ptr;
60 CURLSHcode res = CURLSHE_OK;
61
62 if(share->dirty)
63 /* don't allow setting options while one or more handles are already
64 using this share */
65 return CURLSHE_IN_USE;
66
67 va_start(param, option);
68
69 switch(option) {
70 case CURLSHOPT_SHARE:
71 /* this is a type this share will share */
72 type = va_arg(param, int);
73 share->specifier |= (1<<type);
74 switch(type) {
75 case CURL_LOCK_DATA_DNS:
76 break;
77
78 case CURL_LOCK_DATA_COOKIE:
79 #if !defined(CURL_DISABLE_HTTP) && !defined(CURL_DISABLE_COOKIES)
80 if(!share->cookies) {
81 share->cookies = Curl_cookie_init(NULL, NULL, NULL, TRUE);
82 if(!share->cookies)
83 res = CURLSHE_NOMEM;
84 }
85 #else /* CURL_DISABLE_HTTP */
86 res = CURLSHE_NOT_BUILT_IN;
87 #endif
88 break;
89
90 case CURL_LOCK_DATA_SSL_SESSION:
91 #ifdef USE_SSL
92 if(!share->sslsession) {
93 share->max_ssl_sessions = 8;
94 share->sslsession = calloc(share->max_ssl_sessions,
95 sizeof(struct curl_ssl_session));
96 share->sessionage = 0;
97 if(!share->sslsession)
98 res = CURLSHE_NOMEM;
99 }
100 #else
101 res = CURLSHE_NOT_BUILT_IN;
102 #endif
103 break;
104
105 case CURL_LOCK_DATA_CONNECT: /* not supported (yet) */
106 if(Curl_conncache_init(&share->conn_cache, 103))
107 res = CURLSHE_NOMEM;
108 break;
109
110 case CURL_LOCK_DATA_PSL:
111 #ifndef USE_LIBPSL
112 res = CURLSHE_NOT_BUILT_IN;
113 #endif
114 break;
115
116 default:
117 res = CURLSHE_BAD_OPTION;
118 }
119 break;
120
121 case CURLSHOPT_UNSHARE:
122 /* this is a type this share will no longer share */
123 type = va_arg(param, int);
124 share->specifier &= ~(1<<type);
125 switch(type) {
126 case CURL_LOCK_DATA_DNS:
127 break;
128
129 case CURL_LOCK_DATA_COOKIE:
130 #if !defined(CURL_DISABLE_HTTP) && !defined(CURL_DISABLE_COOKIES)
131 if(share->cookies) {
132 Curl_cookie_cleanup(share->cookies);
133 share->cookies = NULL;
134 }
135 #else /* CURL_DISABLE_HTTP */
136 res = CURLSHE_NOT_BUILT_IN;
137 #endif
138 break;
139
140 case CURL_LOCK_DATA_SSL_SESSION:
141 #ifdef USE_SSL
142 Curl_safefree(share->sslsession);
143 #else
144 res = CURLSHE_NOT_BUILT_IN;
145 #endif
146 break;
147
148 case CURL_LOCK_DATA_CONNECT:
149 break;
150
151 default:
152 res = CURLSHE_BAD_OPTION;
153 break;
154 }
155 break;
156
157 case CURLSHOPT_LOCKFUNC:
158 lockfunc = va_arg(param, curl_lock_function);
159 share->lockfunc = lockfunc;
160 break;
161
162 case CURLSHOPT_UNLOCKFUNC:
163 unlockfunc = va_arg(param, curl_unlock_function);
164 share->unlockfunc = unlockfunc;
165 break;
166
167 case CURLSHOPT_USERDATA:
168 ptr = va_arg(param, void *);
169 share->clientdata = ptr;
170 break;
171
172 default:
173 res = CURLSHE_BAD_OPTION;
174 break;
175 }
176
177 va_end(param);
178
179 return res;
180 }
181
182 CURLSHcode
curl_share_cleanup(struct Curl_share * share)183 curl_share_cleanup(struct Curl_share *share)
184 {
185 if(share == NULL)
186 return CURLSHE_INVALID;
187
188 if(share->lockfunc)
189 share->lockfunc(NULL, CURL_LOCK_DATA_SHARE, CURL_LOCK_ACCESS_SINGLE,
190 share->clientdata);
191
192 if(share->dirty) {
193 if(share->unlockfunc)
194 share->unlockfunc(NULL, CURL_LOCK_DATA_SHARE, share->clientdata);
195 return CURLSHE_IN_USE;
196 }
197
198 Curl_conncache_close_all_connections(&share->conn_cache);
199 Curl_conncache_destroy(&share->conn_cache);
200 Curl_hash_destroy(&share->hostcache);
201
202 #if !defined(CURL_DISABLE_HTTP) && !defined(CURL_DISABLE_COOKIES)
203 Curl_cookie_cleanup(share->cookies);
204 #endif
205
206 #ifdef USE_SSL
207 if(share->sslsession) {
208 size_t i;
209 for(i = 0; i < share->max_ssl_sessions; i++)
210 Curl_ssl_kill_session(&(share->sslsession[i]));
211 free(share->sslsession);
212 }
213 #endif
214
215 Curl_psl_destroy(&share->psl);
216
217 if(share->unlockfunc)
218 share->unlockfunc(NULL, CURL_LOCK_DATA_SHARE, share->clientdata);
219 free(share);
220
221 return CURLSHE_OK;
222 }
223
224
225 CURLSHcode
Curl_share_lock(struct Curl_easy * data,curl_lock_data type,curl_lock_access accesstype)226 Curl_share_lock(struct Curl_easy *data, curl_lock_data type,
227 curl_lock_access accesstype)
228 {
229 struct Curl_share *share = data->share;
230
231 if(share == NULL)
232 return CURLSHE_INVALID;
233
234 if(share->specifier & (1<<type)) {
235 if(share->lockfunc) /* only call this if set! */
236 share->lockfunc(data, type, accesstype, share->clientdata);
237 }
238 /* else if we don't share this, pretend successful lock */
239
240 return CURLSHE_OK;
241 }
242
243 CURLSHcode
Curl_share_unlock(struct Curl_easy * data,curl_lock_data type)244 Curl_share_unlock(struct Curl_easy *data, curl_lock_data type)
245 {
246 struct Curl_share *share = data->share;
247
248 if(share == NULL)
249 return CURLSHE_INVALID;
250
251 if(share->specifier & (1<<type)) {
252 if(share->unlockfunc) /* only call this if set! */
253 share->unlockfunc (data, type, share->clientdata);
254 }
255
256 return CURLSHE_OK;
257 }
258