example: handle uclient connection errors
[project/uclient.git] / uclient-http.c
1 /*
2  * uclient - ustream based protocol client library
3  *
4  * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
5  *
6  * Permission to use, copy, modify, and/or distribute this software for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18 #include <stdio.h>
19 #include <ctype.h>
20 #include <unistd.h>
21 #include <stdint.h>
22
23 #include <libubox/ustream.h>
24 #include <libubox/ustream-ssl.h>
25 #include <libubox/usock.h>
26 #include <libubox/blobmsg.h>
27
28 #include "uclient.h"
29 #include "uclient-utils.h"
30 #include "uclient-backend.h"
31
32 enum auth_type {
33         AUTH_TYPE_UNKNOWN,
34         AUTH_TYPE_NONE,
35         AUTH_TYPE_BASIC,
36         AUTH_TYPE_DIGEST,
37 };
38
39 enum request_type {
40         REQ_GET,
41         REQ_HEAD,
42         REQ_POST,
43         __REQ_MAX
44 };
45
46 enum http_state {
47         HTTP_STATE_INIT,
48         HTTP_STATE_HEADERS_SENT,
49         HTTP_STATE_REQUEST_DONE,
50         HTTP_STATE_RECV_HEADERS,
51         HTTP_STATE_RECV_DATA,
52         HTTP_STATE_ERROR,
53 };
54
55 static const char * const request_types[__REQ_MAX] = {
56         [REQ_GET] = "GET",
57         [REQ_HEAD] = "HEAD",
58         [REQ_POST] = "POST",
59 };
60
61 struct uclient_http {
62         struct uclient uc;
63
64         const struct ustream_ssl_ops *ssl_ops;
65         struct ustream_ssl_ctx *ssl_ctx;
66         struct ustream *us;
67
68         struct ustream_fd ufd;
69         struct ustream_ssl ussl;
70
71         struct uloop_timeout disconnect_t;
72
73         bool ssl_require_validation;
74         bool ssl;
75         bool eof;
76         bool connection_close;
77         bool disconnect;
78         enum request_type req_type;
79         enum http_state state;
80
81         enum auth_type auth_type;
82         char *auth_str;
83
84         long read_chunked;
85         long content_length;
86
87         uint32_t nc;
88
89         struct blob_buf headers;
90         struct blob_buf meta;
91 };
92
93 enum {
94         PREFIX_HTTP,
95         PREFIX_HTTPS,
96         __PREFIX_MAX,
97 };
98
99 static const char * const uclient_http_prefix[] = {
100         [PREFIX_HTTP] = "http://",
101         [PREFIX_HTTPS] = "https://",
102         [__PREFIX_MAX] = NULL
103 };
104
105 static int uclient_do_connect(struct uclient_http *uh, const char *port)
106 {
107         socklen_t sl;
108         int fd;
109
110         if (uh->uc.url->port)
111                 port = uh->uc.url->port;
112
113         fd = usock(USOCK_TCP | USOCK_NONBLOCK, uh->uc.url->host, port);
114         if (fd < 0)
115                 return -1;
116
117         ustream_fd_init(&uh->ufd, fd);
118
119         memset(&uh->uc.local_addr, 0, sizeof(uh->uc.local_addr));
120         memset(&uh->uc.remote_addr, 0, sizeof(uh->uc.remote_addr));
121
122         sl = sizeof(uh->uc.local_addr);
123         getsockname(fd, &uh->uc.local_addr.sa, &sl);
124         getpeername(fd, &uh->uc.remote_addr.sa, &sl);
125
126         return 0;
127 }
128
129 static void uclient_http_disconnect(struct uclient_http *uh)
130 {
131         uloop_timeout_cancel(&uh->disconnect_t);
132         if (!uh->us)
133                 return;
134
135         if (uh->ssl)
136                 ustream_free(&uh->ussl.stream);
137         ustream_free(&uh->ufd.stream);
138         close(uh->ufd.fd.fd);
139         uh->us = NULL;
140 }
141
142 static void uclient_http_free_url_state(struct uclient *cl)
143 {
144         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
145
146         uh->auth_type = AUTH_TYPE_UNKNOWN;
147         free(uh->auth_str);
148         uh->auth_str = NULL;
149         uclient_http_disconnect(uh);
150 }
151
152 static void uclient_http_error(struct uclient_http *uh, int code)
153 {
154         uh->state = HTTP_STATE_ERROR;
155         uh->us->eof = true;
156         ustream_state_change(uh->us);
157         uclient_backend_set_error(&uh->uc, code);
158 }
159
160 static void uclient_notify_eof(struct uclient_http *uh)
161 {
162         struct ustream *us = uh->us;
163
164         if (uh->disconnect)
165                 return;
166
167         if (!uh->eof) {
168                 if (!us->eof && !us->write_error)
169                         return;
170
171                 if (ustream_pending_data(us, false))
172                         return;
173         }
174
175         uclient_backend_set_eof(&uh->uc);
176
177         if (uh->connection_close)
178                 uclient_http_disconnect(uh);
179 }
180
181 static void uclient_http_reset_state(struct uclient_http *uh)
182 {
183         uclient_backend_reset_state(&uh->uc);
184         uh->read_chunked = -1;
185         uh->content_length = -1;
186         uh->eof = false;
187         uh->disconnect = false;
188         uh->connection_close = false;
189         uh->state = HTTP_STATE_INIT;
190
191         if (uh->auth_type == AUTH_TYPE_UNKNOWN && !uh->uc.url->auth)
192                 uh->auth_type = AUTH_TYPE_NONE;
193 }
194
195 static void uclient_http_init_request(struct uclient_http *uh)
196 {
197         uclient_http_reset_state(uh);
198         blob_buf_init(&uh->meta, 0);
199 }
200
201 static enum auth_type
202 uclient_http_update_auth_type(struct uclient_http *uh)
203 {
204         if (!uh->auth_str)
205                 return AUTH_TYPE_NONE;
206
207         if (!strncasecmp(uh->auth_str, "basic", 5))
208                 return AUTH_TYPE_BASIC;
209
210         if (!strncasecmp(uh->auth_str, "digest", 6))
211                 return AUTH_TYPE_DIGEST;
212
213         return AUTH_TYPE_NONE;
214 }
215
216 static void uclient_http_process_headers(struct uclient_http *uh)
217 {
218         enum {
219                 HTTP_HDR_TRANSFER_ENCODING,
220                 HTTP_HDR_CONNECTION,
221                 HTTP_HDR_CONTENT_LENGTH,
222                 HTTP_HDR_AUTH,
223                 __HTTP_HDR_MAX,
224         };
225         static const struct blobmsg_policy hdr_policy[__HTTP_HDR_MAX] = {
226 #define hdr(_name) { .name = _name, .type = BLOBMSG_TYPE_STRING }
227                 [HTTP_HDR_TRANSFER_ENCODING] = hdr("transfer-encoding"),
228                 [HTTP_HDR_CONNECTION] = hdr("connection"),
229                 [HTTP_HDR_CONTENT_LENGTH] = hdr("content-length"),
230                 [HTTP_HDR_AUTH] = hdr("www-authenticate"),
231 #undef hdr
232         };
233         struct blob_attr *tb[__HTTP_HDR_MAX];
234         struct blob_attr *cur;
235
236         blobmsg_parse(hdr_policy, __HTTP_HDR_MAX, tb, blob_data(uh->meta.head), blob_len(uh->meta.head));
237
238         cur = tb[HTTP_HDR_TRANSFER_ENCODING];
239         if (cur && strstr(blobmsg_data(cur), "chunked"))
240                 uh->read_chunked = 0;
241
242         cur = tb[HTTP_HDR_CONNECTION];
243         if (cur && strstr(blobmsg_data(cur), "close"))
244                 uh->connection_close = true;
245
246         cur = tb[HTTP_HDR_CONTENT_LENGTH];
247         if (cur)
248                 uh->content_length = strtoul(blobmsg_data(cur), NULL, 10);
249
250         cur = tb[HTTP_HDR_AUTH];
251         if (cur) {
252                 free(uh->auth_str);
253                 uh->auth_str = strdup(blobmsg_data(cur));
254         }
255
256         uh->auth_type = uclient_http_update_auth_type(uh);
257 }
258
259 static void
260 uclient_http_add_auth_basic(struct uclient_http *uh)
261 {
262         struct uclient_url *url = uh->uc.url;
263         int auth_len = strlen(url->auth);
264         char *auth_buf;
265
266         if (auth_len > 512)
267                 return;
268
269         auth_buf = alloca(base64_len(auth_len) + 1);
270         base64_encode(url->auth, auth_len, auth_buf);
271         ustream_printf(uh->us, "Authorization: Basic %s\r\n", auth_buf);
272 }
273
274 static char *digest_unquote_sep(char **str)
275 {
276         char *cur = *str + 1;
277         char *start = cur;
278         char *out;
279
280         if (**str != '"')
281                 return NULL;
282
283         out = cur;
284         while (1) {
285                 if (!*cur)
286                         return NULL;
287
288                 if (*cur == '"') {
289                         cur++;
290                         break;
291                 }
292
293                 if (*cur == '\\')
294                         cur++;
295
296                 *(out++) = *(cur++);
297         }
298
299         if (*cur == ',')
300                 cur++;
301
302         *out = 0;
303         *str = cur;
304
305         return start;
306 }
307
308 static bool strmatch(char **str, const char *prefix)
309 {
310         int len = strlen(prefix);
311
312         if (strncmp(*str, prefix, len) != 0 || (*str)[len] != '=')
313                 return false;
314
315         *str += len + 1;
316         return true;
317 }
318
319 static void
320 get_cnonce(char *dest)
321 {
322         uint32_t val = 0;
323         FILE *f;
324
325         f = fopen("/dev/urandom", "r");
326         if (f) {
327                 fread(&val, sizeof(val), 1, f);
328                 fclose(f);
329         }
330
331         bin_to_hex(dest, &val, sizeof(val));
332 }
333
334 static void add_field(char **buf, int *ofs, int *len, const char *name, const char *val)
335 {
336         int available = *len - *ofs;
337         int required;
338         const char *next;
339         char *cur;
340
341         if (*len && !*buf)
342                 return;
343
344         required = strlen(name) + 4 + strlen(val) * 2;
345         if (required > available)
346                 *len += required - available + 64;
347
348         *buf = realloc(*buf, *len);
349         if (!*buf)
350                 return;
351
352         cur = *buf + *ofs;
353         cur += sprintf(cur, ", %s=\"", name);
354
355         while ((next = strchr(val, '"'))) {
356                 if (next > val) {
357                         memcpy(cur, val, next - val);
358                         cur += next - val;
359                 }
360
361                 cur += sprintf(cur, "\\\"");
362                 val = next + 1;
363         }
364
365         cur += sprintf(cur, "%s\"", val);
366         *ofs = cur - *buf;
367 }
368
369 static void
370 uclient_http_add_auth_digest(struct uclient_http *uh)
371 {
372         struct uclient_url *url = uh->uc.url;
373         const char *realm = NULL, *opaque = NULL;
374         const char *user, *password;
375         char *buf, *next;
376         int len, ofs;
377
378         char cnonce_str[9];
379         char nc_str[9];
380         char ahash[33];
381         char hash[33];
382
383         struct http_digest_data data = {
384                 .nc = nc_str,
385                 .cnonce = cnonce_str,
386                 .auth_hash = ahash,
387         };
388
389         len = strlen(uh->auth_str) + 1;
390         if (len > 512)
391                 return;
392
393         buf = alloca(len);
394         strcpy(buf, uh->auth_str);
395
396         /* skip auth type */
397         strsep(&buf, " ");
398
399         next = buf;
400         while (*next) {
401                 const char **dest = NULL;
402
403                 while (isspace(*next))
404                         next++;
405
406                 if (strmatch(&next, "realm"))
407                         dest = &realm;
408                 else if (strmatch(&next, "qop"))
409                         dest = &data.qop;
410                 else if (strmatch(&next, "nonce"))
411                         dest = &data.nonce;
412                 else if (strmatch(&next, "opaque"))
413                         dest = &opaque;
414                 else
415                         return;
416
417                 *dest = digest_unquote_sep(&next);
418         }
419
420         if (!realm || !data.qop || !data.nonce)
421                 return;
422
423         sprintf(nc_str, "%08x", uh->nc++);
424         get_cnonce(cnonce_str);
425
426         data.qop = "auth";
427         data.uri = url->location;
428         data.method = request_types[uh->req_type];
429
430         password = strchr(url->auth, ':');
431         if (password) {
432                 char *user_buf;
433
434                 len = password - url->auth;
435                 if (len > 256)
436                         return;
437
438                 user_buf = alloca(len + 1);
439                 strncpy(user_buf, url->auth, len);
440                 user_buf[len] = 0;
441                 user = user_buf;
442                 password++;
443         } else {
444                 user = url->auth;
445                 password = "";
446         }
447
448         http_digest_calculate_auth_hash(ahash, user, realm, password);
449         http_digest_calculate_response(hash, &data);
450
451         buf = NULL;
452         len = 0;
453         ofs = 0;
454
455         add_field(&buf, &ofs, &len, "username", user);
456         add_field(&buf, &ofs, &len, "realm", realm);
457         add_field(&buf, &ofs, &len, "nonce", data.nonce);
458         add_field(&buf, &ofs, &len, "uri", data.uri);
459         add_field(&buf, &ofs, &len, "cnonce", data.cnonce);
460         add_field(&buf, &ofs, &len, "response", hash);
461         if (opaque)
462                 add_field(&buf, &ofs, &len, "opaque", opaque);
463
464         ustream_printf(uh->us, "Authorization: Digest nc=%s, qop=%s%s\r\n", data.nc, data.qop, buf);
465         free(buf);
466 }
467
468 static void
469 uclient_http_add_auth_header(struct uclient_http *uh)
470 {
471         if (!uh->uc.url->auth)
472                 return;
473
474         switch (uh->auth_type) {
475         case AUTH_TYPE_UNKNOWN:
476         case AUTH_TYPE_NONE:
477                 break;
478         case AUTH_TYPE_BASIC:
479                 uclient_http_add_auth_basic(uh);
480                 break;
481         case AUTH_TYPE_DIGEST:
482                 uclient_http_add_auth_digest(uh);
483                 break;
484         }
485 }
486
487 static void
488 uclient_http_send_headers(struct uclient_http *uh)
489 {
490         struct uclient_url *url = uh->uc.url;
491         struct blob_attr *cur;
492         enum request_type req_type = uh->req_type;
493         int rem;
494
495         if (uh->state >= HTTP_STATE_HEADERS_SENT)
496                 return;
497
498         if (uh->auth_type == AUTH_TYPE_UNKNOWN)
499                 req_type = REQ_HEAD;
500
501         ustream_printf(uh->us,
502                 "%s %s HTTP/1.1\r\n"
503                 "Host: %s\r\n",
504                 request_types[req_type],
505                 url->location, url->host);
506
507         blobmsg_for_each_attr(cur, uh->headers.head, rem)
508                 ustream_printf(uh->us, "%s: %s\n", blobmsg_name(cur), (char *) blobmsg_data(cur));
509
510         if (uh->req_type == REQ_POST)
511                 ustream_printf(uh->us, "Transfer-Encoding: chunked\r\n");
512
513         uclient_http_add_auth_header(uh);
514
515         ustream_printf(uh->us, "\r\n");
516
517         uh->state = HTTP_STATE_HEADERS_SENT;
518 }
519
520 static void uclient_http_headers_complete(struct uclient_http *uh)
521 {
522         enum auth_type auth_type = uh->auth_type;
523
524         uh->state = HTTP_STATE_RECV_DATA;
525         uh->uc.meta = uh->meta.head;
526         uclient_http_process_headers(uh);
527
528         if (auth_type == AUTH_TYPE_UNKNOWN) {
529                 uclient_http_init_request(uh);
530                 uclient_http_send_headers(uh);
531                 uh->state = HTTP_STATE_REQUEST_DONE;
532                 return;
533         }
534
535         if (uh->uc.cb->header_done)
536                 uh->uc.cb->header_done(&uh->uc);
537
538         if (uh->eof)
539                 return;
540
541         if (uh->req_type == REQ_HEAD || uh->uc.status_code == 204) {
542                 uh->eof = true;
543                 uclient_notify_eof(uh);
544         }
545 }
546
547 static void uclient_parse_http_line(struct uclient_http *uh, char *data)
548 {
549         char *name;
550         char *sep;
551
552         if (uh->state == HTTP_STATE_REQUEST_DONE) {
553                 char *code;
554
555                 /* HTTP/1.1 */
556                 strsep(&data, " ");
557
558                 code = strsep(&data, " ");
559                 if (!code)
560                         goto error;
561
562                 uh->uc.status_code = strtoul(code, &sep, 10);
563                 if (sep && *sep)
564                         goto error;
565
566                 uh->state = HTTP_STATE_RECV_HEADERS;
567                 return;
568         }
569
570         if (!*data) {
571                 uclient_http_headers_complete(uh);
572                 return;
573         }
574
575         sep = strchr(data, ':');
576         if (!sep)
577                 return;
578
579         *(sep++) = 0;
580
581         for (name = data; *name; name++)
582                 *name = tolower(*name);
583
584         name = data;
585         while (isspace(*sep))
586                 sep++;
587
588         blobmsg_add_string(&uh->meta, name, sep);
589         return;
590
591 error:
592         uh->uc.status_code = 400;
593         uh->eof = true;
594         uclient_notify_eof(uh);
595 }
596
597 static void __uclient_notify_read(struct uclient_http *uh)
598 {
599         struct uclient *uc = &uh->uc;
600         char *data;
601         int len;
602
603         if (uh->state < HTTP_STATE_REQUEST_DONE || uh->state == HTTP_STATE_ERROR)
604                 return;
605
606         data = ustream_get_read_buf(uh->us, &len);
607         if (!data || !len)
608                 return;
609
610         if (uh->state < HTTP_STATE_RECV_DATA) {
611                 char *sep;
612                 int cur_len;
613
614                 do {
615                         sep = strstr(data, "\r\n");
616                         if (!sep)
617                                 break;
618
619                         /* Check for multi-line HTTP headers */
620                         if (sep > data) {
621                                 if (!sep[2])
622                                         return;
623
624                                 if (isspace(sep[2]) && sep[2] != '\r') {
625                                         sep[0] = ' ';
626                                         sep[1] = ' ';
627                                         continue;
628                                 }
629                         }
630
631                         *sep = 0;
632                         cur_len = sep + 2 - data;
633                         uclient_parse_http_line(uh, data);
634                         ustream_consume(uh->us, cur_len);
635                         len -= cur_len;
636
637                         if (uh->eof)
638                                 return;
639
640                         data = ustream_get_read_buf(uh->us, &len);
641                 } while (data && uh->state < HTTP_STATE_RECV_DATA);
642
643                 if (!len)
644                         return;
645         }
646
647         if (uh->eof)
648                 return;
649
650         if (uh->state == HTTP_STATE_RECV_DATA && uc->cb->data_read)
651                 uc->cb->data_read(uc);
652 }
653
654 static void uclient_notify_read(struct ustream *us, int bytes)
655 {
656         struct uclient_http *uh = container_of(us, struct uclient_http, ufd.stream);
657
658         __uclient_notify_read(uh);
659 }
660
661 static void uclient_notify_state(struct ustream *us)
662 {
663         struct uclient_http *uh = container_of(us, struct uclient_http, ufd.stream);
664
665         uclient_notify_eof(uh);
666 }
667
668 static int uclient_setup_http(struct uclient_http *uh)
669 {
670         struct ustream *us = &uh->ufd.stream;
671         int ret;
672
673         uh->us = us;
674         uh->ssl = false;
675
676         us->string_data = true;
677         us->notify_state = uclient_notify_state;
678         us->notify_read = uclient_notify_read;
679
680         ret = uclient_do_connect(uh, "80");
681         if (ret)
682                 return UCLIENT_ERROR_CONNECT;
683
684         return 0;
685 }
686
687 static void uclient_ssl_notify_read(struct ustream *us, int bytes)
688 {
689         struct uclient_http *uh = container_of(us, struct uclient_http, ussl.stream);
690
691         __uclient_notify_read(uh);
692 }
693
694 static void uclient_ssl_notify_state(struct ustream *us)
695 {
696         struct uclient_http *uh = container_of(us, struct uclient_http, ussl.stream);
697
698         uclient_notify_eof(uh);
699 }
700
701 static void uclient_ssl_notify_error(struct ustream_ssl *ssl, int error, const char *str)
702 {
703         struct uclient_http *uh = container_of(ssl, struct uclient_http, ussl);
704
705         uclient_http_error(uh, UCLIENT_ERROR_CONNECT);
706 }
707
708 static void uclient_ssl_notify_verify_error(struct ustream_ssl *ssl, int error, const char *str)
709 {
710         struct uclient_http *uh = container_of(ssl, struct uclient_http, ussl);
711
712         if (!uh->ssl_require_validation)
713                 return;
714
715         uclient_http_error(uh, UCLIENT_ERROR_SSL_INVALID_CERT);
716 }
717
718 static void uclient_ssl_notify_connected(struct ustream_ssl *ssl)
719 {
720         struct uclient_http *uh = container_of(ssl, struct uclient_http, ussl);
721
722         if (!uh->ssl_require_validation)
723                 return;
724
725         if (!uh->ussl.valid_cn)
726                 uclient_http_error(uh, UCLIENT_ERROR_SSL_CN_MISMATCH);
727 }
728
729 static int uclient_setup_https(struct uclient_http *uh)
730 {
731         struct ustream *us = &uh->ussl.stream;
732         int ret;
733
734         uh->ssl = true;
735         uh->us = us;
736
737         if (!uh->ssl_ctx)
738                 return UCLIENT_ERROR_MISSING_SSL_CONTEXT;
739
740         ret = uclient_do_connect(uh, "443");
741         if (ret)
742                 return UCLIENT_ERROR_CONNECT;
743
744         us->string_data = true;
745         us->notify_state = uclient_ssl_notify_state;
746         us->notify_read = uclient_ssl_notify_read;
747         uh->ussl.notify_error = uclient_ssl_notify_error;
748         uh->ussl.notify_verify_error = uclient_ssl_notify_verify_error;
749         uh->ussl.notify_connected = uclient_ssl_notify_connected;
750         uh->ssl_ops->init(&uh->ussl, &uh->ufd.stream, uh->ssl_ctx, false);
751         uh->ssl_ops->set_peer_cn(&uh->ussl, uh->uc.url->host);
752
753         return 0;
754 }
755
756 static int uclient_http_connect(struct uclient *cl)
757 {
758         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
759         int ret;
760
761         uclient_http_init_request(uh);
762
763         if (uh->us)
764                 return 0;
765
766         uh->ssl = cl->url->prefix == PREFIX_HTTPS;
767
768         if (uh->ssl)
769                 ret = uclient_setup_https(uh);
770         else
771                 ret = uclient_setup_http(uh);
772
773         return ret;
774 }
775
776 static void uclient_http_disconnect_cb(struct uloop_timeout *timeout)
777 {
778         struct uclient_http *uh = container_of(timeout, struct uclient_http, disconnect_t);
779
780         uclient_http_disconnect(uh);
781 }
782
783 static struct uclient *uclient_http_alloc(void)
784 {
785         struct uclient_http *uh;
786
787         uh = calloc_a(sizeof(*uh));
788         uh->disconnect_t.cb = uclient_http_disconnect_cb;
789         blob_buf_init(&uh->headers, 0);
790
791         return &uh->uc;
792 }
793
794 static void uclient_http_free_ssl_ctx(struct uclient_http *uh)
795 {
796         uh->ssl_ops = NULL;
797         uh->ssl_ctx = NULL;
798 }
799
800 static void uclient_http_free(struct uclient *cl)
801 {
802         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
803
804         uclient_http_free_url_state(cl);
805         uclient_http_free_ssl_ctx(uh);
806         blob_buf_free(&uh->headers);
807         blob_buf_free(&uh->meta);
808         free(uh);
809 }
810
811 int
812 uclient_http_set_request_type(struct uclient *cl, const char *type)
813 {
814         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
815         int i;
816
817         if (cl->backend != &uclient_backend_http)
818                 return -1;
819
820         if (uh->state > HTTP_STATE_INIT)
821                 return -1;
822
823         for (i = 0; i < ARRAY_SIZE(request_types); i++) {
824                 if (strcmp(request_types[i], type) != 0)
825                         continue;
826
827                 uh->req_type = i;
828                 return 0;
829         }
830
831         return -1;
832 }
833
834 int
835 uclient_http_reset_headers(struct uclient *cl)
836 {
837         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
838
839         blob_buf_init(&uh->headers, 0);
840
841         return 0;
842 }
843
844 int
845 uclient_http_set_header(struct uclient *cl, const char *name, const char *value)
846 {
847         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
848
849         if (cl->backend != &uclient_backend_http)
850                 return -1;
851
852         if (uh->state > HTTP_STATE_INIT)
853                 return -1;
854
855         blobmsg_add_string(&uh->headers, name, value);
856         return 0;
857 }
858
859 static int
860 uclient_http_send_data(struct uclient *cl, char *buf, unsigned int len)
861 {
862         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
863
864         if (uh->state >= HTTP_STATE_REQUEST_DONE)
865                 return -1;
866
867         uclient_http_send_headers(uh);
868
869         if (len > 0) {
870                 ustream_printf(uh->us, "%X\r\n", len);
871                 ustream_write(uh->us, buf, len, false);
872                 ustream_printf(uh->us, "\r\n");
873         }
874
875         return len;
876 }
877
878 static int
879 uclient_http_request_done(struct uclient *cl)
880 {
881         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
882
883         if (uh->state >= HTTP_STATE_REQUEST_DONE)
884                 return -1;
885
886         uclient_http_send_headers(uh);
887         if (uh->req_type == REQ_POST)
888                 ustream_printf(uh->us, "0\r\n\r\n");
889         uh->state = HTTP_STATE_REQUEST_DONE;
890
891         return 0;
892 }
893
894 static int
895 uclient_http_read(struct uclient *cl, char *buf, unsigned int len)
896 {
897         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
898         int read_len = 0;
899         char *data, *data_end;
900
901         if (uh->state < HTTP_STATE_RECV_DATA || !uh->us)
902                 return 0;
903
904         data = ustream_get_read_buf(uh->us, &read_len);
905         if (!data || !read_len)
906                 return 0;
907
908         data_end = data + read_len;
909         read_len = 0;
910
911         if (uh->read_chunked == 0) {
912                 char *sep;
913
914                 if (data[0] == '\r' && data[1] == '\n') {
915                         data += 2;
916                         read_len += 2;
917                 }
918
919                 sep = strstr(data, "\r\n");
920                 if (!sep)
921                         return 0;
922
923                 *sep = 0;
924                 uh->read_chunked = strtoul(data, NULL, 16);
925
926                 read_len += sep + 2 - data;
927                 data = sep + 2;
928
929                 if (!uh->read_chunked)
930                         uh->eof = true;
931         }
932
933         if (len > data_end - data)
934                 len = data_end - data;
935
936         if (uh->read_chunked >= 0) {
937                 if (len > uh->read_chunked)
938                         len = uh->read_chunked;
939
940                 uh->read_chunked -= len;
941         } else if (uh->content_length >= 0) {
942                 if (len > uh->content_length)
943                         len = uh->content_length;
944
945                 uh->content_length -= len;
946                 if (!uh->content_length)
947                         uh->eof = true;
948         }
949
950         if (len > 0) {
951                 read_len += len;
952                 memcpy(buf, data, len);
953         }
954
955         if (read_len > 0)
956                 ustream_consume(uh->us, read_len);
957
958         uclient_notify_eof(uh);
959
960         return len;
961 }
962
963 bool uclient_http_redirect(struct uclient *cl)
964 {
965         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
966         struct blobmsg_policy location = {
967                 .name = "location",
968                 .type = BLOBMSG_TYPE_STRING,
969         };
970         struct uclient_url *url = cl->url;
971         struct blob_attr *tb;
972
973         if (cl->backend != &uclient_backend_http)
974                 return false;
975
976         switch (cl->status_code) {
977         case 301:
978         case 302:
979         case 307:
980                 break;
981         default:
982                 return false;
983         }
984
985         blobmsg_parse(&location, 1, &tb, blob_data(uh->meta.head), blob_len(uh->meta.head));
986         if (!tb)
987                 return false;
988
989         url = uclient_get_url(blobmsg_data(tb), url->auth);
990         if (!url)
991                 return false;
992
993         free(cl->url);
994         cl->url = url;
995         uclient_http_connect(cl);
996         uclient_http_request_done(cl);
997
998         return true;
999 }
1000
1001 int uclient_http_set_ssl_ctx(struct uclient *cl, const struct ustream_ssl_ops *ops,
1002                              struct ustream_ssl_ctx *ctx, bool require_validation)
1003 {
1004         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1005
1006         if (cl->backend != &uclient_backend_http)
1007                 return -1;
1008
1009         uclient_http_free_url_state(cl);
1010
1011         uclient_http_free_ssl_ctx(uh);
1012         uh->ssl_ops = ops;
1013         uh->ssl_ctx = ctx;
1014         uh->ssl_require_validation = !!ctx && require_validation;
1015
1016         return 0;
1017 }
1018
1019 static void uclient_http_request_disconnect(struct uclient *cl)
1020 {
1021         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1022
1023         if (!uh->us)
1024                 return;
1025
1026         uh->eof = true;
1027         uh->disconnect = true;
1028         uloop_timeout_set(&uh->disconnect_t, 1);
1029 }
1030
1031 const struct uclient_backend uclient_backend_http = {
1032         .prefix = uclient_http_prefix,
1033
1034         .alloc = uclient_http_alloc,
1035         .free = uclient_http_free,
1036         .connect = uclient_http_connect,
1037         .disconnect = uclient_http_request_disconnect,
1038         .update_url = uclient_http_free_url_state,
1039
1040         .read = uclient_http_read,
1041         .write = uclient_http_send_data,
1042         .request = uclient_http_request_done,
1043 };