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 * |
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 |
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 |
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 |
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 |
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 | |