Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include <fcntl.h>
4 : :
5 : : #include "alloc-util.h"
6 : : #include "build.h"
7 : : #include "curl-util.h"
8 : : #include "fd-util.h"
9 : : #include "locale-util.h"
10 : : #include "string-util.h"
11 : :
12 : 0 : static void curl_glue_check_finished(CurlGlue *g) {
13 : : CURLMsg *msg;
14 : 0 : int k = 0;
15 : :
16 [ # # ]: 0 : assert(g);
17 : :
18 : 0 : msg = curl_multi_info_read(g->curl, &k);
19 [ # # ]: 0 : if (!msg)
20 : 0 : return;
21 : :
22 [ # # ]: 0 : if (msg->msg != CURLMSG_DONE)
23 : 0 : return;
24 : :
25 [ # # ]: 0 : if (g->on_finished)
26 : 0 : g->on_finished(g, msg->easy_handle, msg->data.result);
27 : : }
28 : :
29 : 0 : static int curl_glue_on_io(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
30 : 0 : CurlGlue *g = userdata;
31 : 0 : int action, k = 0;
32 : :
33 [ # # ]: 0 : assert(s);
34 [ # # ]: 0 : assert(g);
35 : :
36 [ # # ]: 0 : if (FLAGS_SET(revents, EPOLLIN | EPOLLOUT))
37 : 0 : action = CURL_POLL_INOUT;
38 [ # # ]: 0 : else if (revents & EPOLLIN)
39 : 0 : action = CURL_POLL_IN;
40 [ # # ]: 0 : else if (revents & EPOLLOUT)
41 : 0 : action = CURL_POLL_OUT;
42 : : else
43 : 0 : action = 0;
44 : :
45 [ # # ]: 0 : if (curl_multi_socket_action(g->curl, fd, action, &k) != CURLM_OK)
46 [ # # ]: 0 : return log_debug_errno(SYNTHETIC_ERRNO(EINVAL),
47 : : "Failed to propagate IO event.");
48 : :
49 : 0 : curl_glue_check_finished(g);
50 : 0 : return 0;
51 : : }
52 : :
53 : 0 : static int curl_glue_socket_callback(CURLM *curl, curl_socket_t s, int action, void *userdata, void *socketp) {
54 : 0 : sd_event_source *io = socketp;
55 : 0 : CurlGlue *g = userdata;
56 : 0 : uint32_t events = 0;
57 : : int r;
58 : :
59 [ # # ]: 0 : assert(curl);
60 [ # # ]: 0 : assert(g);
61 : :
62 [ # # ]: 0 : if (action == CURL_POLL_REMOVE) {
63 [ # # ]: 0 : if (io) {
64 : 0 : sd_event_source_disable_unref(io);
65 : :
66 : 0 : hashmap_remove(g->ios, FD_TO_PTR(s));
67 : : }
68 : :
69 : 0 : return 0;
70 : : }
71 : :
72 : 0 : r = hashmap_ensure_allocated(&g->ios, &trivial_hash_ops);
73 [ # # ]: 0 : if (r < 0) {
74 : 0 : log_oom();
75 : 0 : return -1;
76 : : }
77 : :
78 [ # # ]: 0 : if (action == CURL_POLL_IN)
79 : 0 : events = EPOLLIN;
80 [ # # ]: 0 : else if (action == CURL_POLL_OUT)
81 : 0 : events = EPOLLOUT;
82 [ # # ]: 0 : else if (action == CURL_POLL_INOUT)
83 : 0 : events = EPOLLIN|EPOLLOUT;
84 : :
85 [ # # ]: 0 : if (io) {
86 [ # # ]: 0 : if (sd_event_source_set_io_events(io, events) < 0)
87 : 0 : return -1;
88 : :
89 [ # # ]: 0 : if (sd_event_source_set_enabled(io, SD_EVENT_ON) < 0)
90 : 0 : return -1;
91 : : } else {
92 [ # # ]: 0 : if (sd_event_add_io(g->event, &io, s, events, curl_glue_on_io, g) < 0)
93 : 0 : return -1;
94 : :
95 [ # # ]: 0 : if (curl_multi_assign(g->curl, s, io) != CURLM_OK)
96 : 0 : return -1;
97 : :
98 : 0 : (void) sd_event_source_set_description(io, "curl-io");
99 : :
100 : 0 : r = hashmap_put(g->ios, FD_TO_PTR(s), io);
101 [ # # ]: 0 : if (r < 0) {
102 : 0 : log_oom();
103 : 0 : sd_event_source_unref(io);
104 : 0 : return -1;
105 : : }
106 : : }
107 : :
108 : 0 : return 0;
109 : : }
110 : :
111 : 0 : static int curl_glue_on_timer(sd_event_source *s, uint64_t usec, void *userdata) {
112 : 0 : CurlGlue *g = userdata;
113 : 0 : int k = 0;
114 : :
115 [ # # ]: 0 : assert(s);
116 [ # # ]: 0 : assert(g);
117 : :
118 [ # # ]: 0 : if (curl_multi_socket_action(g->curl, CURL_SOCKET_TIMEOUT, 0, &k) != CURLM_OK)
119 [ # # ]: 0 : return log_debug_errno(SYNTHETIC_ERRNO(EINVAL),
120 : : "Failed to propagate timeout.");
121 : :
122 : 0 : curl_glue_check_finished(g);
123 : 0 : return 0;
124 : : }
125 : :
126 : 0 : static int curl_glue_timer_callback(CURLM *curl, long timeout_ms, void *userdata) {
127 : 0 : CurlGlue *g = userdata;
128 : : usec_t usec;
129 : :
130 [ # # ]: 0 : assert(curl);
131 [ # # ]: 0 : assert(g);
132 : :
133 [ # # ]: 0 : if (timeout_ms < 0) {
134 [ # # ]: 0 : if (g->timer) {
135 [ # # ]: 0 : if (sd_event_source_set_enabled(g->timer, SD_EVENT_OFF) < 0)
136 : 0 : return -1;
137 : : }
138 : :
139 : 0 : return 0;
140 : : }
141 : :
142 : 0 : usec = now(clock_boottime_or_monotonic()) + (usec_t) timeout_ms * USEC_PER_MSEC + USEC_PER_MSEC - 1;
143 : :
144 [ # # ]: 0 : if (g->timer) {
145 [ # # ]: 0 : if (sd_event_source_set_time(g->timer, usec) < 0)
146 : 0 : return -1;
147 : :
148 [ # # ]: 0 : if (sd_event_source_set_enabled(g->timer, SD_EVENT_ONESHOT) < 0)
149 : 0 : return -1;
150 : : } else {
151 [ # # ]: 0 : if (sd_event_add_time(g->event, &g->timer, clock_boottime_or_monotonic(), usec, 0, curl_glue_on_timer, g) < 0)
152 : 0 : return -1;
153 : :
154 : 0 : (void) sd_event_source_set_description(g->timer, "curl-timer");
155 : : }
156 : :
157 : 0 : return 0;
158 : : }
159 : :
160 : 0 : CurlGlue *curl_glue_unref(CurlGlue *g) {
161 : : sd_event_source *io;
162 : :
163 [ # # ]: 0 : if (!g)
164 : 0 : return NULL;
165 : :
166 [ # # ]: 0 : if (g->curl)
167 : 0 : curl_multi_cleanup(g->curl);
168 : :
169 [ # # ]: 0 : while ((io = hashmap_steal_first(g->ios))) {
170 : 0 : sd_event_source_unref(io);
171 : : }
172 : :
173 : 0 : hashmap_free(g->ios);
174 : :
175 : 0 : sd_event_source_unref(g->timer);
176 : 0 : sd_event_unref(g->event);
177 : 0 : return mfree(g);
178 : : }
179 : :
180 : 0 : int curl_glue_new(CurlGlue **glue, sd_event *event) {
181 : 0 : _cleanup_(curl_glue_unrefp) CurlGlue *g = NULL;
182 : 0 : _cleanup_(curl_multi_cleanupp) CURL *c = NULL;
183 : 0 : _cleanup_(sd_event_unrefp) sd_event *e = NULL;
184 : : int r;
185 : :
186 [ # # ]: 0 : if (event)
187 : 0 : e = sd_event_ref(event);
188 : : else {
189 : 0 : r = sd_event_default(&e);
190 [ # # ]: 0 : if (r < 0)
191 : 0 : return r;
192 : : }
193 : :
194 : 0 : c = curl_multi_init();
195 [ # # ]: 0 : if (!c)
196 : 0 : return -ENOMEM;
197 : :
198 : 0 : g = new(CurlGlue, 1);
199 [ # # ]: 0 : if (!g)
200 : 0 : return -ENOMEM;
201 : :
202 : 0 : *g = (CurlGlue) {
203 : 0 : .event = TAKE_PTR(e),
204 : 0 : .curl = TAKE_PTR(c),
205 : : };
206 : :
207 [ # # ]: 0 : if (curl_multi_setopt(g->curl, CURLMOPT_SOCKETDATA, g) != CURLM_OK)
208 : 0 : return -EINVAL;
209 : :
210 [ # # ]: 0 : if (curl_multi_setopt(g->curl, CURLMOPT_SOCKETFUNCTION, curl_glue_socket_callback) != CURLM_OK)
211 : 0 : return -EINVAL;
212 : :
213 [ # # ]: 0 : if (curl_multi_setopt(g->curl, CURLMOPT_TIMERDATA, g) != CURLM_OK)
214 : 0 : return -EINVAL;
215 : :
216 [ # # ]: 0 : if (curl_multi_setopt(g->curl, CURLMOPT_TIMERFUNCTION, curl_glue_timer_callback) != CURLM_OK)
217 : 0 : return -EINVAL;
218 : :
219 : 0 : *glue = TAKE_PTR(g);
220 : :
221 : 0 : return 0;
222 : : }
223 : :
224 : 0 : int curl_glue_make(CURL **ret, const char *url, void *userdata) {
225 : 0 : _cleanup_(curl_easy_cleanupp) CURL *c = NULL;
226 : : const char *useragent;
227 : :
228 [ # # ]: 0 : assert(ret);
229 [ # # ]: 0 : assert(url);
230 : :
231 : 0 : c = curl_easy_init();
232 [ # # ]: 0 : if (!c)
233 : 0 : return -ENOMEM;
234 : :
235 : : /* curl_easy_setopt(c, CURLOPT_VERBOSE, 1L); */
236 : :
237 [ # # ]: 0 : if (curl_easy_setopt(c, CURLOPT_URL, url) != CURLE_OK)
238 : 0 : return -EIO;
239 : :
240 [ # # ]: 0 : if (curl_easy_setopt(c, CURLOPT_PRIVATE, userdata) != CURLE_OK)
241 : 0 : return -EIO;
242 : :
243 [ # # # # : 0 : useragent = strjoina(program_invocation_short_name, "/" GIT_VERSION);
# # # # #
# # # ]
244 [ # # ]: 0 : if (curl_easy_setopt(c, CURLOPT_USERAGENT, useragent) != CURLE_OK)
245 : 0 : return -EIO;
246 : :
247 [ # # ]: 0 : if (curl_easy_setopt(c, CURLOPT_FOLLOWLOCATION, 1L) != CURLE_OK)
248 : 0 : return -EIO;
249 : :
250 : 0 : *ret = TAKE_PTR(c);
251 : 0 : return 0;
252 : : }
253 : :
254 : 0 : int curl_glue_add(CurlGlue *g, CURL *c) {
255 [ # # ]: 0 : assert(g);
256 [ # # ]: 0 : assert(c);
257 : :
258 [ # # ]: 0 : if (curl_multi_add_handle(g->curl, c) != CURLM_OK)
259 : 0 : return -EIO;
260 : :
261 : 0 : return 0;
262 : : }
263 : :
264 : 0 : void curl_glue_remove_and_free(CurlGlue *g, CURL *c) {
265 [ # # ]: 0 : assert(g);
266 : :
267 [ # # ]: 0 : if (!c)
268 : 0 : return;
269 : :
270 [ # # ]: 0 : if (g->curl)
271 : 0 : curl_multi_remove_handle(g->curl, c);
272 : :
273 : 0 : curl_easy_cleanup(c);
274 : : }
275 : :
276 : 0 : struct curl_slist *curl_slist_new(const char *first, ...) {
277 : : struct curl_slist *l;
278 : : va_list ap;
279 : :
280 [ # # ]: 0 : if (!first)
281 : 0 : return NULL;
282 : :
283 : 0 : l = curl_slist_append(NULL, first);
284 [ # # ]: 0 : if (!l)
285 : 0 : return NULL;
286 : :
287 : 0 : va_start(ap, first);
288 : :
289 : 0 : for (;;) {
290 : : struct curl_slist *n;
291 : : const char *i;
292 : :
293 : 0 : i = va_arg(ap, const char*);
294 [ # # ]: 0 : if (!i)
295 : 0 : break;
296 : :
297 : 0 : n = curl_slist_append(l, i);
298 [ # # ]: 0 : if (!n) {
299 : 0 : va_end(ap);
300 : 0 : curl_slist_free_all(l);
301 : 0 : return NULL;
302 : : }
303 : :
304 : 0 : l = n;
305 : : }
306 : :
307 : 0 : va_end(ap);
308 : 0 : return l;
309 : : }
310 : :
311 : 0 : int curl_header_strdup(const void *contents, size_t sz, const char *field, char **value) {
312 : : const char *p;
313 : : char *s;
314 : :
315 : 0 : p = memory_startswith_no_case(contents, sz, field);
316 [ # # ]: 0 : if (!p)
317 : 0 : return 0;
318 : :
319 : 0 : sz -= p - (const char*) contents;
320 : :
321 [ # # ]: 0 : if (memchr(p, 0, sz))
322 : 0 : return 0;
323 : :
324 : : /* Skip over preceding whitespace */
325 [ # # # # ]: 0 : while (sz > 0 && strchr(WHITESPACE, p[0])) {
326 : 0 : p++;
327 : 0 : sz--;
328 : : }
329 : :
330 : : /* Truncate trailing whitespace */
331 [ # # # # ]: 0 : while (sz > 0 && strchr(WHITESPACE, p[sz-1]))
332 : 0 : sz--;
333 : :
334 : 0 : s = strndup(p, sz);
335 [ # # ]: 0 : if (!s)
336 : 0 : return -ENOMEM;
337 : :
338 : 0 : *value = s;
339 : 0 : return 1;
340 : : }
341 : :
342 : 0 : int curl_parse_http_time(const char *t, usec_t *ret) {
343 : 0 : _cleanup_(freelocalep) locale_t loc = (locale_t) 0;
344 : : const char *e;
345 : : struct tm tm;
346 : : time_t v;
347 : :
348 [ # # ]: 0 : assert(t);
349 [ # # ]: 0 : assert(ret);
350 : :
351 : 0 : loc = newlocale(LC_TIME_MASK, "C", (locale_t) 0);
352 [ # # ]: 0 : if (loc == (locale_t) 0)
353 : 0 : return -errno;
354 : :
355 : : /* RFC822 */
356 : 0 : e = strptime_l(t, "%a, %d %b %Y %H:%M:%S %Z", &tm, loc);
357 [ # # # # ]: 0 : if (!e || *e != 0)
358 : : /* RFC 850 */
359 : 0 : e = strptime_l(t, "%A, %d-%b-%y %H:%M:%S %Z", &tm, loc);
360 [ # # # # ]: 0 : if (!e || *e != 0)
361 : : /* ANSI C */
362 : 0 : e = strptime_l(t, "%a %b %d %H:%M:%S %Y", &tm, loc);
363 [ # # # # ]: 0 : if (!e || *e != 0)
364 : 0 : return -EINVAL;
365 : :
366 : 0 : v = timegm(&tm);
367 [ # # ]: 0 : if (v == (time_t) -1)
368 : 0 : return -EINVAL;
369 : :
370 : 0 : *ret = (usec_t) v * USEC_PER_SEC;
371 : 0 : return 0;
372 : : }
|
