• source navigation  • diff markup  • identifier search  • freetext search  • 

Sources/uclient/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 <sys/socket.h>
 19 #include <stdio.h>
 20 #include <ctype.h>
 21 #include <unistd.h>
 22 #include <stdint.h>
 23 #include <string.h>
 24 #include <fcntl.h>
 25 
 26 #include <libubox/ustream.h>
 27 #include <libubox/ustream-ssl.h>
 28 #include <libubox/usock.h>
 29 #include <libubox/blobmsg.h>
 30 
 31 #include "uclient.h"
 32 #include "uclient-utils.h"
 33 #include "uclient-backend.h"
 34 
 35 enum auth_type {
 36         AUTH_TYPE_UNKNOWN,
 37         AUTH_TYPE_NONE,
 38         AUTH_TYPE_BASIC,
 39         AUTH_TYPE_DIGEST,
 40 };
 41 
 42 enum request_type {
 43         REQ_GET,
 44         REQ_HEAD,
 45         REQ_POST,
 46         REQ_PUT,
 47         REQ_DELETE,
 48         __REQ_MAX
 49 };
 50 
 51 enum http_state {
 52         HTTP_STATE_INIT,
 53         HTTP_STATE_HEADERS_SENT,
 54         HTTP_STATE_REQUEST_DONE,
 55         HTTP_STATE_RECV_HEADERS,
 56         HTTP_STATE_RECV_DATA,
 57         HTTP_STATE_ERROR,
 58 };
 59 
 60 static const char * const request_types[__REQ_MAX] = {
 61         [REQ_GET] = "GET",
 62         [REQ_HEAD] = "HEAD",
 63         [REQ_POST] = "POST",
 64         [REQ_PUT] = "PUT",
 65         [REQ_DELETE] = "DELETE",
 66 };
 67 
 68 struct uclient_http {
 69         struct uclient uc;
 70 
 71         const struct ustream_ssl_ops *ssl_ops;
 72         struct ustream_ssl_ctx *ssl_ctx;
 73         struct ustream *us;
 74 
 75         struct ustream_fd ufd;
 76         struct ustream_ssl ussl;
 77 
 78         struct uloop_timeout disconnect_t;
 79         unsigned int seq;
 80 
 81         bool ssl_require_validation;
 82         bool ssl;
 83         bool eof;
 84         bool connection_close;
 85         bool disconnect;
 86         enum request_type req_type;
 87         enum http_state state;
 88 
 89         enum auth_type auth_type;
 90         char *auth_str;
 91 
 92         long read_chunked;
 93         long content_length;
 94 
 95         int usock_flags;
 96 
 97         uint32_t nc;
 98 
 99         struct blob_buf headers;
100         struct blob_buf meta;
101 };
102 
103 enum {
104         PREFIX_HTTP,
105         PREFIX_HTTPS,
106         __PREFIX_MAX,
107 };
108 
109 static const char * const uclient_http_prefix[] = {
110         [PREFIX_HTTP] = "http://",
111         [PREFIX_HTTPS] = "https://",
112         [__PREFIX_MAX] = NULL
113 };
114 
115 static int uclient_http_connect(struct uclient *cl);
116 
117 static int uclient_do_connect(struct uclient_http *uh, const char *port)
118 {
119         socklen_t sl;
120         int fd;
121 
122         if (uh->uc.url->port)
123                 port = uh->uc.url->port;
124 
125         memset(&uh->uc.remote_addr, 0, sizeof(uh->uc.remote_addr));
126 
127         fd = usock_inet_timeout(USOCK_TCP | USOCK_NONBLOCK | uh->usock_flags,
128                                 uh->uc.url->host, port, &uh->uc.remote_addr,
129                                 uh->uc.timeout_msecs);
130         if (fd < 0)
131                 return -1;
132 
133         fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK);
134         ustream_fd_init(&uh->ufd, fd);
135 
136         sl = sizeof(uh->uc.local_addr);
137         memset(&uh->uc.local_addr, 0, sl);
138         getsockname(fd, &uh->uc.local_addr.sa, &sl);
139 
140         return 0;
141 }
142 
143 static void uclient_http_disconnect(struct uclient_http *uh)
144 {
145         uloop_timeout_cancel(&uh->disconnect_t);
146         if (!uh->us)
147                 return;
148 
149         if (uh->ssl)
150                 ustream_free(&uh->ussl.stream);
151         ustream_free(&uh->ufd.stream);
152         if(uh->ufd.fd.fd)
153                 close(uh->ufd.fd.fd);
154         uh->us = NULL;
155 }
156 
157 static void uclient_http_free_url_state(struct uclient *cl)
158 {
159         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
160 
161         uh->auth_type = AUTH_TYPE_UNKNOWN;
162         free(uh->auth_str);
163         uh->auth_str = NULL;
164         uclient_http_disconnect(uh);
165 }
166 
167 static void uclient_http_error(struct uclient_http *uh, int code)
168 {
169         uh->state = HTTP_STATE_ERROR;
170         uh->us->eof = true;
171         ustream_state_change(uh->us);
172         uclient_backend_set_error(&uh->uc, code);
173 }
174 
175 static void uclient_http_request_disconnect(struct uclient *cl)
176 {
177         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
178 
179         if (!uh->us)
180                 return;
181 
182         uh->eof = true;
183         uh->disconnect = true;
184         uloop_timeout_set(&uh->disconnect_t, 1);
185 }
186 
187 static void uclient_notify_eof(struct uclient_http *uh)
188 {
189         struct ustream *us = uh->us;
190 
191         if (uh->disconnect)
192                 return;
193 
194         if (!uh->eof) {
195                 if (!us->eof && !us->write_error)
196                         return;
197 
198                 if (ustream_pending_data(us, false))
199                         return;
200         }
201 
202         if ((uh->content_length < 0 && uh->read_chunked >= 0) ||
203                         uh->content_length == 0)
204                 uh->uc.data_eof = true;
205 
206         uclient_backend_set_eof(&uh->uc);
207 
208         if (uh->connection_close)
209                 uclient_http_request_disconnect(&uh->uc);
210 }
211 
212 static void uclient_http_reset_state(struct uclient_http *uh)
213 {
214         uh->seq++;
215         uclient_backend_reset_state(&uh->uc);
216         uh->read_chunked = -1;
217         uh->content_length = -1;
218         uh->eof = false;
219         uh->disconnect = false;
220         uh->connection_close = false;
221         uh->state = HTTP_STATE_INIT;
222 
223         if (uh->auth_type == AUTH_TYPE_UNKNOWN && !uh->uc.url->auth)
224                 uh->auth_type = AUTH_TYPE_NONE;
225 }
226 
227 static void uclient_http_init_request(struct uclient_http *uh)
228 {
229         uh->seq++;
230         uclient_http_reset_state(uh);
231         blob_buf_init(&uh->meta, 0);
232 }
233 
234 static enum auth_type
235 uclient_http_update_auth_type(struct uclient_http *uh)
236 {
237         if (!uh->auth_str)
238                 return AUTH_TYPE_NONE;
239 
240         if (!strncasecmp(uh->auth_str, "basic", 5))
241                 return AUTH_TYPE_BASIC;
242 
243         if (!strncasecmp(uh->auth_str, "digest", 6))
244                 return AUTH_TYPE_DIGEST;
245 
246         return AUTH_TYPE_NONE;
247 }
248 
249 static void uclient_http_process_headers(struct uclient_http *uh)
250 {
251         enum {
252                 HTTP_HDR_TRANSFER_ENCODING,
253                 HTTP_HDR_CONNECTION,
254                 HTTP_HDR_CONTENT_LENGTH,
255                 HTTP_HDR_AUTH,
256                 __HTTP_HDR_MAX,
257         };
258         static const struct blobmsg_policy hdr_policy[__HTTP_HDR_MAX] = {
259 #define hdr(_name) { .name = _name, .type = BLOBMSG_TYPE_STRING }
260                 [HTTP_HDR_TRANSFER_ENCODING] = hdr("transfer-encoding"),
261                 [HTTP_HDR_CONNECTION] = hdr("connection"),
262                 [HTTP_HDR_CONTENT_LENGTH] = hdr("content-length"),
263                 [HTTP_HDR_AUTH] = hdr("www-authenticate"),
264 #undef hdr
265         };
266         struct blob_attr *tb[__HTTP_HDR_MAX];
267         struct blob_attr *cur;
268 
269         blobmsg_parse(hdr_policy, __HTTP_HDR_MAX, tb, blob_data(uh->meta.head), blob_len(uh->meta.head));
270 
271         cur = tb[HTTP_HDR_TRANSFER_ENCODING];
272         if (cur && strstr(blobmsg_data(cur), "chunked"))
273                 uh->read_chunked = 0;
274 
275         cur = tb[HTTP_HDR_CONNECTION];
276         if (cur && strstr(blobmsg_data(cur), "close"))
277                 uh->connection_close = true;
278 
279         cur = tb[HTTP_HDR_CONTENT_LENGTH];
280         if (cur)
281                 uh->content_length = strtoul(blobmsg_data(cur), NULL, 10);
282 
283         cur = tb[HTTP_HDR_AUTH];
284         if (cur) {
285                 free(uh->auth_str);
286                 uh->auth_str = strdup(blobmsg_data(cur));
287         }
288 
289         uh->auth_type = uclient_http_update_auth_type(uh);
290 }
291 
292 static bool uclient_request_supports_body(enum request_type req_type)
293 {
294         switch (req_type) {
295         case REQ_POST:
296         case REQ_PUT:
297         case REQ_DELETE:
298                 return true;
299         default:
300                 return false;
301         }
302 }
303 
304 static int
305 uclient_http_add_auth_basic(struct uclient_http *uh)
306 {
307         struct uclient_url *url = uh->uc.url;
308         int auth_len = strlen(url->auth);
309         char *auth_buf;
310 
311         if (auth_len > 512)
312                 return -EINVAL;
313 
314         auth_buf = alloca(base64_len(auth_len) + 1);
315         if (!auth_buf)
316                 return -ENOMEM;
317 
318         base64_encode(url->auth, auth_len, auth_buf);
319         ustream_printf(uh->us, "Authorization: Basic %s\r\n", auth_buf);
320 
321         return 0;
322 }
323 
324 static char *digest_unquote_sep(char **str)
325 {
326         char *cur = *str + 1;
327         char *start = cur;
328         char *out;
329 
330         if (**str != '"')
331                 return NULL;
332 
333         out = cur;
334         while (1) {
335                 if (!*cur)
336                         return NULL;
337 
338                 if (*cur == '"') {
339                         cur++;
340                         break;
341                 }
342 
343                 if (*cur == '\\')
344                         cur++;
345 
346                 *(out++) = *(cur++);
347         }
348 
349         if (*cur == ',')
350                 cur++;
351 
352         *out = 0;
353         *str = cur;
354 
355         return start;
356 }
357 
358 static char *digest_sep(char **str)
359 {
360         char *cur, *next;
361 
362         cur = *str;
363         next = strchr(*str, ',');
364         if (next) {
365             *str = next + 1;
366             *next = 0;
367         } else {
368             *str += strlen(*str);
369         }
370 
371         return cur;
372 }
373 
374 static bool strmatch(char **str, const char *prefix)
375 {
376         int len = strlen(prefix);
377 
378         if (strncmp(*str, prefix, len) != 0 || (*str)[len] != '=')
379                 return false;
380 
381         *str += len + 1;
382         return true;
383 }
384 
385 static void
386 get_cnonce(char *dest)
387 {
388         uint32_t val = 0;
389         FILE *f;
390         size_t n;
391 
392         f = fopen("/dev/urandom", "r");
393         if (f) {
394                 n = fread(&val, sizeof(val), 1, f);
395                 fclose(f);
396                 if (n != 1)
397                         return;
398         }
399 
400         bin_to_hex(dest, &val, sizeof(val));
401 }
402 
403 static void add_field(char **buf, int *ofs, int *len, const char *name, const char *val)
404 {
405         int available = *len - *ofs;
406         int required;
407         const char *next;
408         char *cur;
409 
410         if (*len && !*buf)
411                 return;
412 
413         required = strlen(name) + 4 + strlen(val) * 2;
414         if (required > available)
415                 *len += required - available + 64;
416 
417         *buf = realloc(*buf, *len);
418         if (!*buf)
419                 return;
420 
421         cur = *buf + *ofs;
422         cur += sprintf(cur, ", %s=\"", name);
423 
424         while ((next = strchr(val, '"'))) {
425                 if (next > val) {
426                         memcpy(cur, val, next - val);
427                         cur += next - val;
428                 }
429 
430                 cur += sprintf(cur, "\\\"");
431                 val = next + 1;
432         }
433 
434         cur += sprintf(cur, "%s\"", val);
435         *ofs = cur - *buf;
436 }
437 
438 static int
439 uclient_http_add_auth_digest(struct uclient_http *uh)
440 {
441         struct uclient_url *url = uh->uc.url;
442         const char *realm = NULL, *opaque = NULL;
443         const char *user, *password;
444         char *buf, *next, *buf_orig;
445         int len, ofs;
446         int err = 0;
447 
448         char cnonce_str[9];
449         char nc_str[9];
450         char ahash[33];
451         char hash[33];
452 
453         struct http_digest_data data = {
454                 .nc = nc_str,
455                 .cnonce = cnonce_str,
456                 .auth_hash = ahash,
457         };
458 
459         len = strlen(uh->auth_str) + 1;
460         if (len > 512) {
461                 err = -EINVAL;
462                 goto fail;
463         }
464 
465         buf = alloca(len);
466         if (!buf) {
467                 err = -ENOMEM;
468                 goto fail;
469         }
470 
471         strcpy(buf, uh->auth_str);
472 
473         /* skip auth type */
474         buf_orig = strsep(&buf, " ");
475 
476         next = buf;
477         while (*next) {
478                 const char **dest = NULL;
479                 const char *tmp;
480 
481                 while (*next && isspace(*next))
482                         next++;
483 
484                 if (strmatch(&next, "realm"))
485                         dest = &realm;
486                 else if (strmatch(&next, "qop"))
487                         dest = &data.qop;
488                 else if (strmatch(&next, "nonce"))
489                         dest = &data.nonce;
490                 else if (strmatch(&next, "opaque"))
491                         dest = &opaque;
492                 else if (strmatch(&next, "stale") ||
493                          strmatch(&next, "algorithm") ||
494                          strmatch(&next, "auth-param")) {
495                         digest_sep(&next);
496                         continue;
497                 } else if (strmatch(&next, "domain") ||
498                          strmatch(&next, "qop-options"))
499                         dest = &tmp;
500                 else {
501                         digest_sep(&next);
502                         continue;
503                 }
504 
505                 *dest = digest_unquote_sep(&next);
506         }
507 
508         if (!realm || !data.qop || !data.nonce) {
509                 err = -EINVAL;
510                 goto fail_buf;
511         }
512 
513         sprintf(nc_str, "%08x", uh->nc++);
514         get_cnonce(cnonce_str);
515 
516         data.qop = "auth";
517         data.uri = url->location;
518         data.method = request_types[uh->req_type];
519 
520         password = strchr(url->auth, ':');
521         if (password) {
522                 char *user_buf;
523 
524                 len = password - url->auth;
525                 if (len > 256) {
526                         err = -EINVAL;
527                         goto fail_buf;
528                 }
529 
530                 user_buf = alloca(len + 1);
531                 if (!user_buf) {
532                         err = -ENOMEM;
533                         goto fail_buf;
534                 }
535 
536                 strncpy(user_buf, url->auth, len);
537                 user_buf[len] = 0;
538                 user = user_buf;
539                 password++;
540         } else {
541                 user = url->auth;
542                 password = "";
543         }
544 
545         http_digest_calculate_auth_hash(ahash, user, realm, password);
546         http_digest_calculate_response(hash, &data);
547 
548         buf = NULL;
549         len = 0;
550         ofs = 0;
551 
552         add_field(&buf, &ofs, &len, "username", user);
553         add_field(&buf, &ofs, &len, "realm", realm);
554         add_field(&buf, &ofs, &len, "nonce", data.nonce);
555         add_field(&buf, &ofs, &len, "uri", data.uri);
556         add_field(&buf, &ofs, &len, "cnonce", data.cnonce);
557         add_field(&buf, &ofs, &len, "response", hash);
558         if (opaque)
559                 add_field(&buf, &ofs, &len, "opaque", opaque);
560 
561         ustream_printf(uh->us, "Authorization: Digest nc=%s, qop=%s%s\r\n", data.nc, data.qop, buf);
562 
563         free(buf);
564 
565         return 0;
566 
567 fail_buf:
568         free(buf_orig);
569 fail:
570         return err;
571 }
572 
573 static int
574 uclient_http_add_auth_header(struct uclient_http *uh)
575 {
576         if (!uh->uc.url->auth)
577                 return 0;
578 
579         switch (uh->auth_type) {
580         case AUTH_TYPE_UNKNOWN:
581         case AUTH_TYPE_NONE:
582                 break;
583         case AUTH_TYPE_BASIC:
584                 return uclient_http_add_auth_basic(uh);
585         case AUTH_TYPE_DIGEST:
586                 return uclient_http_add_auth_digest(uh);
587         }
588 
589         return 0;
590 }
591 
592 static int
593 uclient_http_send_headers(struct uclient_http *uh)
594 {
595         struct uclient_url *url = uh->uc.url;
596         struct blob_attr *cur;
597         enum request_type req_type = uh->req_type;
598         bool literal_ipv6;
599         int err, rem;
600 
601         if (uh->state >= HTTP_STATE_HEADERS_SENT)
602                 return 0;
603 
604         if (uh->uc.proxy_url)
605                 url = uh->uc.proxy_url;
606 
607         literal_ipv6 = strchr(url->host, ':');
608 
609         ustream_printf(uh->us,
610                 "%s %s HTTP/1.1\r\n"
611                 "Host: %s%s%s%s%s\r\n",
612                 request_types[req_type], url->location,
613                 literal_ipv6 ? "[" : "",
614                 url->host,
615                 literal_ipv6 ? "]" : "",
616                 url->port ? ":" : "",
617                 url->port ? url->port : "");
618 
619         blobmsg_for_each_attr(cur, uh->headers.head, rem)
620                 ustream_printf(uh->us, "%s: %s\r\n", blobmsg_name(cur), (char *) blobmsg_data(cur));
621 
622         if (uclient_request_supports_body(uh->req_type))
623                 ustream_printf(uh->us, "Transfer-Encoding: chunked\r\n");
624 
625         err = uclient_http_add_auth_header(uh);
626         if (err)
627                 return err;
628 
629         ustream_printf(uh->us, "\r\n");
630 
631         uh->state = HTTP_STATE_HEADERS_SENT;
632 
633         return 0;
634 }
635 
636 static void uclient_http_headers_complete(struct uclient_http *uh)
637 {
638         enum auth_type auth_type = uh->auth_type;
639         int seq = uh->uc.seq;
640 
641         uh->state = HTTP_STATE_RECV_DATA;
642         uh->uc.meta = uh->meta.head;
643         uclient_http_process_headers(uh);
644 
645         if (auth_type == AUTH_TYPE_UNKNOWN && uh->uc.status_code == 401 &&
646             (uh->req_type == REQ_HEAD || uh->req_type == REQ_GET)) {
647                 uclient_http_connect(&uh->uc);
648                 uclient_http_send_headers(uh);
649                 uh->state = HTTP_STATE_REQUEST_DONE;
650                 return;
651         }
652 
653         if (uh->uc.cb->header_done)
654                 uh->uc.cb->header_done(&uh->uc);
655 
656         if (uh->eof || seq != uh->uc.seq)
657                 return;
658 
659         if (uh->req_type == REQ_HEAD || uh->uc.status_code == 204) {
660                 uh->eof = true;
661                 uclient_notify_eof(uh);
662         }
663 }
664 
665 static void uclient_parse_http_line(struct uclient_http *uh, char *data)
666 {
667         char *name;
668         char *sep;
669 
670         if (uh->state == HTTP_STATE_REQUEST_DONE) {
671                 char *code;
672 
673                 if (!strlen(data))
674                         return;
675 
676                 /* HTTP/1.1 */
677                 strsep(&data, " ");
678 
679                 code = strsep(&data, " ");
680                 if (!code)
681                         goto error;
682 
683                 uh->uc.status_code = strtoul(code, &sep, 10);
684                 if (sep && *sep)
685                         goto error;
686 
687                 uh->state = HTTP_STATE_RECV_HEADERS;
688                 return;
689         }
690 
691         if (!*data) {
692                 uclient_http_headers_complete(uh);
693                 return;
694         }
695 
696         sep = strchr(data, ':');
697         if (!sep)
698                 return;
699 
700         *(sep++) = 0;
701 
702         for (name = data; *name; name++)
703                 *name = tolower(*name);
704 
705         name = data;
706         while (isspace(*sep))
707                 sep++;
708 
709         blobmsg_add_string(&uh->meta, name, sep);
710         return;
711 
712 error:
713         uh->uc.status_code = 400;
714         uh->eof = true;
715         uclient_notify_eof(uh);
716 }
717 
718 static void __uclient_notify_read(struct uclient_http *uh)
719 {
720         struct uclient *uc = &uh->uc;
721         unsigned int seq = uh->seq;
722         char *data;
723         int len;
724 
725         if (uh->state < HTTP_STATE_REQUEST_DONE || uh->state == HTTP_STATE_ERROR)
726                 return;
727 
728         data = ustream_get_read_buf(uh->us, &len);
729         if (!data || !len)
730                 return;
731 
732         if (uh->state < HTTP_STATE_RECV_DATA) {
733                 char *sep, *next;
734                 int cur_len;
735 
736                 do {
737                         sep = strchr(data, '\n');
738                         if (!sep)
739                                 break;
740 
741                         next = sep + 1;
742                         if (sep > data && sep[-1] == '\r')
743                                 sep--;
744 
745                         /* Check for multi-line HTTP headers */
746                         if (sep > data) {
747                                 if (!*next)
748                                         return;
749 
750                                 if (isspace(*next) && *next != '\r' && *next != '\n') {
751                                         sep[0] = ' ';
752                                         if (sep + 1 < next)
753                                                 sep[1] = ' ';
754                                         continue;
755                                 }
756                         }
757 
758                         *sep = 0;
759                         cur_len = next - data;
760                         uclient_parse_http_line(uh, data);
761                         if (seq != uh->seq)
762                                 return;
763 
764                         ustream_consume(uh->us, cur_len);
765                         len -= cur_len;
766 
767                         if (uh->eof)
768                                 return;
769 
770                         data = ustream_get_read_buf(uh->us, &len);
771                 } while (data && uh->state < HTTP_STATE_RECV_DATA);
772 
773                 if (!len)
774                         return;
775         }
776 
777         if (uh->eof)
778                 return;
779 
780         if (uh->state == HTTP_STATE_RECV_DATA) {
781                 /* Now it's uclient user turn to read some data */
782                 uloop_timeout_cancel(&uc->connection_timeout);
783 
784                 if (uc->cb->data_read)
785                         uc->cb->data_read(uc);
786         }
787 }
788 
789 static void __uclient_notify_write(struct uclient_http *uh)
790 {
791         struct uclient *uc = &uh->uc;
792 
793         if (uc->cb->data_sent)
794                 uc->cb->data_sent(uc);
795 }
796 
797 static void uclient_notify_read(struct ustream *us, int bytes)
798 {
799         struct uclient_http *uh = container_of(us, struct uclient_http, ufd.stream);
800 
801         __uclient_notify_read(uh);
802 }
803 
804 static void uclient_notify_write(struct ustream *us, int bytes)
805 {
806         struct uclient_http *uh = container_of(us, struct uclient_http, ufd.stream);
807 
808         __uclient_notify_write(uh);
809 }
810 
811 static void uclient_notify_state(struct ustream *us)
812 {
813         struct uclient_http *uh = container_of(us, struct uclient_http, ufd.stream);
814 
815         if (uh->ufd.stream.write_error) {
816                 uclient_http_error(uh, UCLIENT_ERROR_CONNECT);
817                 return;
818         }
819         uclient_notify_eof(uh);
820 }
821 
822 static int uclient_setup_http(struct uclient_http *uh)
823 {
824         struct ustream *us = &uh->ufd.stream;
825         int ret;
826 
827         uh->us = us;
828         uh->ssl = false;
829 
830         us->string_data = true;
831         us->notify_state = uclient_notify_state;
832         us->notify_read = uclient_notify_read;
833         us->notify_write = uclient_notify_write;
834 
835         ret = uclient_do_connect(uh, "80");
836         if (ret)
837                 return UCLIENT_ERROR_CONNECT;
838 
839         return 0;
840 }
841 
842 static void uclient_ssl_notify_read(struct ustream *us, int bytes)
843 {
844         struct uclient_http *uh = container_of(us, struct uclient_http, ussl.stream);
845 
846         __uclient_notify_read(uh);
847 }
848 
849 static void uclient_ssl_notify_write(struct ustream *us, int bytes)
850 {
851         struct uclient_http *uh = container_of(us, struct uclient_http, ussl.stream);
852 
853         __uclient_notify_write(uh);
854 }
855 
856 static void uclient_ssl_notify_state(struct ustream *us)
857 {
858         struct uclient_http *uh = container_of(us, struct uclient_http, ussl.stream);
859 
860         uclient_notify_eof(uh);
861 }
862 
863 static void uclient_ssl_notify_error(struct ustream_ssl *ssl, int error, const char *str)
864 {
865         struct uclient_http *uh = container_of(ssl, struct uclient_http, ussl);
866 
867         uclient_http_error(uh, UCLIENT_ERROR_CONNECT);
868 }
869 
870 static void uclient_ssl_notify_verify_error(struct ustream_ssl *ssl, int error, const char *str)
871 {
872         struct uclient_http *uh = container_of(ssl, struct uclient_http, ussl);
873 
874         if (!uh->ssl_require_validation)
875                 return;
876 
877         uclient_http_error(uh, UCLIENT_ERROR_SSL_INVALID_CERT);
878 }
879 
880 static void uclient_ssl_notify_connected(struct ustream_ssl *ssl)
881 {
882         struct uclient_http *uh = container_of(ssl, struct uclient_http, ussl);
883 
884         if (!uh->ssl_require_validation)
885                 return;
886 
887         if (!uh->ussl.valid_cn)
888                 uclient_http_error(uh, UCLIENT_ERROR_SSL_CN_MISMATCH);
889 }
890 
891 static int uclient_setup_https(struct uclient_http *uh)
892 {
893         struct ustream *us = &uh->ussl.stream;
894         int ret;
895 
896         uh->ssl = true;
897         uh->us = us;
898 
899         if (!uh->ssl_ctx)
900                 return UCLIENT_ERROR_MISSING_SSL_CONTEXT;
901 
902         ret = uclient_do_connect(uh, "443");
903         if (ret)
904                 return UCLIENT_ERROR_CONNECT;
905 
906         us->string_data = true;
907         us->notify_state = uclient_ssl_notify_state;
908         us->notify_read = uclient_ssl_notify_read;
909         us->notify_write = uclient_ssl_notify_write;
910         uh->ussl.notify_error = uclient_ssl_notify_error;
911         uh->ussl.notify_verify_error = uclient_ssl_notify_verify_error;
912         uh->ussl.notify_connected = uclient_ssl_notify_connected;
913         uh->ussl.server_name = uh->uc.url->host;
914         uh->ssl_ops->init(&uh->ussl, &uh->ufd.stream, uh->ssl_ctx, false);
915         uh->ssl_ops->set_peer_cn(&uh->ussl, uh->uc.url->host);
916 
917         return 0;
918 }
919 
920 static int uclient_http_connect(struct uclient *cl)
921 {
922         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
923         int ret;
924 
925         if (!cl->eof || uh->disconnect || uh->connection_close)
926                 uclient_http_disconnect(uh);
927 
928         uclient_http_init_request(uh);
929 
930         if (uh->us)
931                 return 0;
932 
933         uh->ssl = cl->url->prefix == PREFIX_HTTPS;
934 
935         if (uh->ssl)
936                 ret = uclient_setup_https(uh);
937         else
938                 ret = uclient_setup_http(uh);
939 
940         return ret;
941 }
942 
943 static void uclient_http_disconnect_cb(struct uloop_timeout *timeout)
944 {
945         struct uclient_http *uh = container_of(timeout, struct uclient_http, disconnect_t);
946 
947         uclient_http_disconnect(uh);
948 }
949 
950 static struct uclient *uclient_http_alloc(void)
951 {
952         struct uclient_http *uh;
953 
954         uh = calloc_a(sizeof(*uh));
955         if (!uh)
956                 return NULL;
957 
958         uh->disconnect_t.cb = uclient_http_disconnect_cb;
959         blob_buf_init(&uh->headers, 0);
960 
961         return &uh->uc;
962 }
963 
964 static void uclient_http_free_ssl_ctx(struct uclient_http *uh)
965 {
966         uh->ssl_ops = NULL;
967         uh->ssl_ctx = NULL;
968 }
969 
970 static void uclient_http_free(struct uclient *cl)
971 {
972         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
973 
974         uclient_http_free_url_state(cl);
975         uclient_http_free_ssl_ctx(uh);
976         blob_buf_free(&uh->headers);
977         blob_buf_free(&uh->meta);
978         free(uh);
979 }
980 
981 int
982 uclient_http_set_request_type(struct uclient *cl, const char *type)
983 {
984         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
985         int i;
986 
987         if (cl->backend != &uclient_backend_http)
988                 return -1;
989 
990         if (uh->state > HTTP_STATE_INIT)
991                 return -1;
992 
993         for (i = 0; i < ARRAY_SIZE(request_types); i++) {
994                 if (strcmp(request_types[i], type) != 0)
995                         continue;
996 
997                 uh->req_type = i;
998                 return 0;
999         }
1000 
1001         return -1;
1002 }
1003 
1004 int
1005 uclient_http_reset_headers(struct uclient *cl)
1006 {
1007         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1008 
1009         blob_buf_init(&uh->headers, 0);
1010 
1011         return 0;
1012 }
1013 
1014 int
1015 uclient_http_set_header(struct uclient *cl, const char *name, const char *value)
1016 {
1017         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1018 
1019         if (cl->backend != &uclient_backend_http)
1020                 return -1;
1021 
1022         if (uh->state > HTTP_STATE_INIT)
1023                 return -1;
1024 
1025         blobmsg_add_string(&uh->headers, name, value);
1026         return 0;
1027 }
1028 
1029 static int
1030 uclient_http_send_data(struct uclient *cl, const char *buf, unsigned int len)
1031 {
1032         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1033         int err;
1034 
1035         if (uh->state >= HTTP_STATE_REQUEST_DONE)
1036                 return -1;
1037 
1038         err = uclient_http_send_headers(uh);
1039         if (err)
1040                 return err;
1041 
1042         if (len > 0) {
1043                 ustream_printf(uh->us, "%X\r\n", len);
1044                 ustream_write(uh->us, buf, len, false);
1045                 ustream_printf(uh->us, "\r\n");
1046         }
1047 
1048         return len;
1049 }
1050 
1051 static int
1052 uclient_http_request_done(struct uclient *cl)
1053 {
1054         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1055         int err;
1056 
1057         if (uh->state >= HTTP_STATE_REQUEST_DONE)
1058                 return -1;
1059 
1060         err = uclient_http_send_headers(uh);
1061         if (err)
1062                 return err;
1063 
1064         if (uclient_request_supports_body(uh->req_type))
1065                 ustream_printf(uh->us, "\r\n\r\n");
1066         uh->state = HTTP_STATE_REQUEST_DONE;
1067 
1068         return 0;
1069 }
1070 
1071 static int
1072 uclient_http_read(struct uclient *cl, char *buf, unsigned int len)
1073 {
1074         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1075         int read_len = 0;
1076         char *data, *data_end;
1077 
1078         if (uh->state < HTTP_STATE_RECV_DATA || !uh->us)
1079                 return 0;
1080 
1081         data = ustream_get_read_buf(uh->us, &read_len);
1082         if (!data || !read_len)
1083                 return 0;
1084 
1085         data_end = data + read_len;
1086         read_len = 0;
1087 
1088         if (uh->read_chunked == 0) {
1089                 char *sep;
1090 
1091                 if (data[0] == '\r' && data[1] == '\n') {
1092                         data += 2;
1093                         read_len += 2;
1094                 }
1095 
1096                 sep = strstr(data, "\r\n");
1097                 if (!sep)
1098                         return 0;
1099 
1100                 *sep = 0;
1101                 uh->read_chunked = strtoul(data, NULL, 16);
1102 
1103                 read_len += sep + 2 - data;
1104                 data = sep + 2;
1105 
1106                 if (!uh->read_chunked) {
1107                         uh->eof = true;
1108                         uh->uc.data_eof = true;
1109                 }
1110         }
1111 
1112         if (len > data_end - data)
1113                 len = data_end - data;
1114 
1115         if (uh->read_chunked >= 0) {
1116                 if (len > uh->read_chunked)
1117                         len = uh->read_chunked;
1118 
1119                 uh->read_chunked -= len;
1120         } else if (uh->content_length >= 0) {
1121                 if (len > uh->content_length)
1122                         len = uh->content_length;
1123 
1124                 uh->content_length -= len;
1125                 if (!uh->content_length) {
1126                         uh->eof = true;
1127                         uh->uc.data_eof = true;
1128                 }
1129         }
1130 
1131         if (len > 0) {
1132                 read_len += len;
1133                 memcpy(buf, data, len);
1134         }
1135 
1136         if (read_len > 0)
1137                 ustream_consume(uh->us, read_len);
1138 
1139         uclient_notify_eof(uh);
1140 
1141         /* Now that we consumed something and if this isn't EOF, start timer again */
1142         if (!uh->uc.eof && !cl->connection_timeout.pending)
1143                 uloop_timeout_set(&cl->connection_timeout, cl->timeout_msecs);
1144 
1145         return len;
1146 }
1147 
1148 int uclient_http_redirect(struct uclient *cl)
1149 {
1150         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1151         struct blobmsg_policy location = {
1152                 .name = "location",
1153                 .type = BLOBMSG_TYPE_STRING,
1154         };
1155         struct uclient_url *url = cl->url;
1156         struct blob_attr *tb;
1157 
1158         if (cl->backend != &uclient_backend_http)
1159                 return false;
1160 
1161         switch (cl->status_code) {
1162         case 301:
1163         case 302:
1164         case 307:
1165                 break;
1166         default:
1167                 return false;
1168         }
1169 
1170         blobmsg_parse(&location, 1, &tb, blob_data(uh->meta.head), blob_len(uh->meta.head));
1171         if (!tb)
1172                 return false;
1173 
1174         url = uclient_get_url_location(url, blobmsg_data(tb));
1175         if (!url)
1176                 return false;
1177 
1178         if (cl->proxy_url) {
1179                 free(cl->proxy_url);
1180                 cl->proxy_url = url;
1181         }
1182         else {
1183                 free(cl->url);
1184                 cl->url = url;
1185         }
1186 
1187         if (uclient_http_connect(cl))
1188                 return -1;
1189 
1190         uclient_http_request_done(cl);
1191 
1192         return true;
1193 }
1194 
1195 int uclient_http_set_ssl_ctx(struct uclient *cl, const struct ustream_ssl_ops *ops,
1196                              struct ustream_ssl_ctx *ctx, bool require_validation)
1197 {
1198         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1199 
1200         if (cl->backend != &uclient_backend_http)
1201                 return -1;
1202 
1203         uclient_http_free_url_state(cl);
1204 
1205         uclient_http_free_ssl_ctx(uh);
1206         uh->ssl_ops = ops;
1207         uh->ssl_ctx = ctx;
1208         uh->ssl_require_validation = !!ctx && require_validation;
1209 
1210         return 0;
1211 }
1212 
1213 int uclient_http_set_address_family(struct uclient *cl, int af)
1214 {
1215         struct uclient_http *uh = container_of(cl, struct uclient_http, uc);
1216 
1217         if (cl->backend != &uclient_backend_http)
1218                 return -1;
1219 
1220         switch (af) {
1221         case AF_INET:
1222                 uh->usock_flags = USOCK_IPV4ONLY;
1223                 break;
1224         case AF_INET6:
1225                 uh->usock_flags = USOCK_IPV6ONLY;
1226                 break;
1227         default:
1228                 uh->usock_flags = 0;
1229                 break;
1230         }
1231 
1232         return 0;
1233 }
1234 
1235 const struct uclient_backend uclient_backend_http = {
1236         .prefix = uclient_http_prefix,
1237 
1238         .alloc = uclient_http_alloc,
1239         .free = uclient_http_free,
1240         .connect = uclient_http_connect,
1241         .disconnect = uclient_http_request_disconnect,
1242         .update_url = uclient_http_free_url_state,
1243         .update_proxy_url = uclient_http_free_url_state,
1244 
1245         .read = uclient_http_read,
1246         .write = uclient_http_send_data,
1247         .request = uclient_http_request_done,
1248 };
1249 

This page was automatically generated by LXR 0.3.1.  •  OpenWrt