Wed, 18 Feb 2026 12:31:19 +0100
enable buffered reader for request bodies with fixed content-length
|
662
70fdf948b642
refactor HttpParser to support parsing of Http responses
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff
changeset
|
1 | /* |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
2 | * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
3 | * |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
4 | * Copyright 2026 Olaf Wintermann. All rights reserved. |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
5 | * |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
6 | * Redistribution and use in source and binary forms, with or without |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
7 | * modification, are permitted provided that the following conditions are met: |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
8 | * |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
9 | * 1. Redistributions of source code must retain the above copyright |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
10 | * notice, this list of conditions and the following disclaimer. |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
11 | * |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
12 | * 2. Redistributions in binary form must reproduce the above copyright |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
13 | * notice, this list of conditions and the following disclaimer in the |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
14 | * documentation and/or other materials provided with the distribution. |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
15 | * |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
16 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
17 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
18 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
19 | * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
20 | * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
21 | * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
22 | * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
23 | * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
24 | * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
25 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
26 | * POSSIBILITY OF SUCH DAMAGE. |
|
662
70fdf948b642
refactor HttpParser to support parsing of Http responses
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff
changeset
|
27 | */ |
|
70fdf948b642
refactor HttpParser to support parsing of Http responses
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff
changeset
|
28 | |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
29 | #include "httpclient.h" |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
30 | |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
31 | #include "../util/socket.h" |
|
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
32 | |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
33 | #include <cx/buffer.h> |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
34 | #include <cx/string.h> |
|
669
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
35 | #include <stdlib.h> |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
36 | #include <string.h> |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
37 | #include <errno.h> |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
38 | |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
39 | static int client_connected(EventHandler *ev, Event *event); |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
40 | static int client_io(EventHandler *ev, Event *event); |
|
672
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
41 | static int client_finished(EventHandler *ev, Event *event); |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
42 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
43 | static int client_send_request(HttpClient *client); |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
44 | static int client_send_request_body(HttpClient *client); |
|
683
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
45 | static int client_read_response_header(HttpClient *client); |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
46 | |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
47 | HttpClient* http_client_new(EventHandler *ev) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
48 | CxMempool *mp = cxMempoolCreate(32, CX_MEMPOOL_TYPE_PURE); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
49 | if(!mp) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
50 | return NULL; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
51 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
52 | |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
53 | HttpClient *client = malloc(sizeof(HttpClient)); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
54 | HeaderArray *req_headers = header_array_create(); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
55 | HeaderArray *resp_headers = header_array_create(); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
56 | if(!client || !req_headers || !resp_headers) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
57 | free(client); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
58 | header_array_free(req_headers); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
59 | header_array_free(resp_headers); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
60 | cxMempoolFree(mp); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
61 | return NULL; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
62 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
63 | |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
64 | memset(client, 0, sizeof(HttpClient)); |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
65 | client->ev = ev; |
|
672
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
66 | client->socketfd = -1; |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
67 | client->request_headers = req_headers; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
68 | client->response_headers = resp_headers; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
69 | |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
70 | client->buffer.maxsize = HTTP_CLIENT_BUFFER_SIZE; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
71 | client->buffer.inbuf = malloc(HTTP_CLIENT_BUFFER_SIZE); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
72 | HttpParser *parser = http_parser_new2(1, &client->buffer, resp_headers); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
73 | if(!parser || !client->buffer.inbuf) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
74 | http_client_free(client); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
75 | return NULL; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
76 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
77 | client->parser = parser; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
78 | |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
79 | return client; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
80 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
81 | |
|
669
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
82 | void http_client_free(HttpClient *client) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
83 | cxMempoolFree(client->mp); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
84 | header_array_free(client->request_headers); |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
85 | http_parser_free(client->parser); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
86 | free(client->buffer.inbuf); |
|
669
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
87 | free(client->addr); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
88 | free(client->method); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
89 | free(client->uri); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
90 | free(client); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
91 | } |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
92 | |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
93 | int http_client_set_addr(HttpClient *client, const struct sockaddr *addr, socklen_t addrlen) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
94 | free(client->addr); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
95 | client->addr = NULL; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
96 | client->addrlen = 0; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
97 | |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
98 | void *newaddr = malloc(addrlen); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
99 | if(!newaddr) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
100 | return 1; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
101 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
102 | memcpy(newaddr, addr, addrlen); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
103 | client->addr = newaddr; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
104 | client->addrlen = addrlen; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
105 | |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
106 | return 0; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
107 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
108 | |
|
669
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
109 | int http_client_set_method(HttpClient *client, const char *method) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
110 | return http_client_set_method_len(client, method, method ? strlen(method) : 0); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
111 | } |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
112 | |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
113 | int http_client_set_uri(HttpClient *client, const char *uri) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
114 | return http_client_set_uri_len(client, uri, uri ? strlen(uri) : 0); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
115 | } |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
116 | |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
117 | static int client_set_str(char **ptr, const char *str, size_t len) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
118 | free(*ptr); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
119 | if(str) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
120 | char *newvalue = malloc(len+1); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
121 | if(!newvalue) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
122 | *ptr = NULL; |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
123 | return 1; |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
124 | } |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
125 | memcpy(newvalue, str, len); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
126 | newvalue[len] = 0; |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
127 | *ptr = newvalue; |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
128 | } else { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
129 | *ptr = NULL; |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
130 | } |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
131 | return 0; |
|
669
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
132 | } |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
133 | |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
134 | int http_client_set_method_len(HttpClient *client, const char *method, size_t len) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
135 | return client_set_str(&client->method, method, len); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
136 | } |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
137 | |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
138 | int http_client_set_uri_len(HttpClient *client, const char *uri, size_t len) { |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
139 | return client_set_str(&client->uri, uri, len); |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
140 | } |
|
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
141 | |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
142 | int http_client_add_request_header(HttpClient *client, cxmutstr name, cxmutstr value) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
143 | return header_array_add(client->request_headers, name, value); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
144 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
145 | |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
146 | int http_client_add_request_header_copy(HttpClient *client, cxstring name, cxstring value) { |
|
675
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
147 | if(!client->mp) { |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
148 | client->mp = cxMempoolCreate(64, CX_MEMPOOL_TYPE_PURE); |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
149 | if(!client->mp) { |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
150 | return 1; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
151 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
152 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
153 | |
|
665
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
154 | cxmutstr n = cx_strdup_a(client->mp->allocator, name); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
155 | cxmutstr v = cx_strdup_a(client->mp->allocator, value); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
156 | |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
157 | int err = 1; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
158 | if(n.ptr && v.ptr) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
159 | err = http_client_add_request_header(client, n, v); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
160 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
161 | if(err) { |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
162 | cxFree(client->mp->allocator, n.ptr); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
163 | cxFree(client->mp->allocator, v.ptr); |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
164 | } |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
165 | return err; |
|
b8d5b797d090
add first http client code
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
662
diff
changeset
|
166 | } |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
167 | |
|
675
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
168 | int http_client_set_content_length(HttpClient *client, int64_t contentlength) { |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
169 | client->req_content_length = contentlength; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
170 | char ctlen_buf[32]; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
171 | size_t len = snprintf(ctlen_buf, 32, "%" PRId64, contentlength); |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
172 | return http_client_add_request_header_copy(client, cx_str("content-length"), cx_strn(ctlen_buf, len)); |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
173 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
174 | |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
175 | int http_client_enable_chunked_transfer_encoding(HttpClient *client) { |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
176 | client->req_content_length = -1; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
177 | return http_client_add_request_header(client, cx_mutstr("transfer-encoding"), cx_mutstr("chunked")); |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
178 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
179 | |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
180 | int http_client_start(HttpClient *client) { |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
181 | int socketfd = socket(AF_INET, SOCK_STREAM, 0); |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
182 | if(socketfd < 0) { |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
183 | return 1; |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
184 | } |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
185 | |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
186 | if(util_socket_setnonblock(socketfd, 1)) { |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
187 | close(socketfd); |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
188 | return 1; |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
189 | } |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
190 | |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
191 | client->socketfd = socketfd; |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
192 | |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
193 | client->writeev.cookie = client; |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
194 | client->writeev.fn = client_connected; |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
195 | |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
196 | int ret = 1; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
197 | if(connect(socketfd, client->addr, client->addrlen)) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
198 | int err = errno; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
199 | if(err == EINPROGRESS) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
200 | ret = ev_pollout(client->ev, socketfd, &client->writeev); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
201 | } else { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
202 | log_ereport(LOG_FAILURE, "http-client-start: connect failed: %s", strerror(err)); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
203 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
204 | } else { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
205 | // TODO: call client_connected directly |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
206 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
207 | |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
208 | if(ret) { |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
209 | close(socketfd); |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
210 | } |
|
669
ccdc97fd8204
setup HttpClient in proxy SAF
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
668
diff
changeset
|
211 | return ret; |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
212 | } |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
213 | |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
214 | static int create_req_buffer(HttpClient *client) { |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
215 | CxBuffer buf; |
|
675
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
216 | if(cxBufferInit(&buf, cxDefaultAllocator, NULL, HTTP_CLIENT_BUFFER_SIZE, CX_BUFFER_AUTO_EXTEND)) { |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
217 | return 1; |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
218 | } |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
219 | |
|
668
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
220 | if(client->method) { |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
221 | cxBufferPutString(&buf, "GET "); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
222 | } else { |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
223 | cxBufferPutString(&buf, client->method); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
224 | } |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
225 | cxBufferPutString(&buf, client->uri ? client->uri : "/"); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
226 | cxBufferPutString(&buf, " HTTP/1.1\r\n"); |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
227 | |
|
668
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
228 | HeaderArray *hdr = client->request_headers; |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
229 | while(hdr) { |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
230 | for(int i=0;i<hdr->len;i++) { |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
231 | cxBufferPutString(&buf, hdr->headers[i].name); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
232 | cxBufferPutString(&buf, ": "); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
233 | cxBufferPutString(&buf, hdr->headers[i].value); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
234 | cxBufferPutString(&buf, "\r\n"); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
235 | } |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
236 | hdr = hdr->next; |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
237 | } |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
238 | cxBufferPutString(&buf, "\r\n"); |
|
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
239 | client->req_buffer = buf.space; |
|
675
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
240 | client->req_buffer_alloc = buf.capacity; |
|
668
0a7d1e9ca6b8
implement HttpClient function create_req_buffer
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
666
diff
changeset
|
241 | client->req_buffer_len = buf.size; |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
242 | |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
243 | return 0; |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
244 | } |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
245 | |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
246 | static int client_connected(EventHandler *ev, Event *event) { |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
247 | HttpClient *client = event->cookie; |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
248 | if(create_req_buffer(client)) { |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
249 | // TODO: set error |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
250 | return 0; // end |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
251 | } |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
252 | event->fn = client_io; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
253 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
254 | return client_io(ev, event); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
255 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
256 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
257 | static int client_io(EventHandler *ev, Event *event) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
258 | HttpClient *client = event->cookie; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
259 | if(client->req_buffer_pos < client->req_buffer_len) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
260 | if(client_send_request(client)) { |
|
675
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
261 | return client->error == 0; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
262 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
263 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
264 | |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
265 | // do we need to send a request body? |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
266 | if(client->req_content_length != 0) { |
|
679
4885cd4c3754
httpclient: move code for sending the request body to a separate function
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
676
diff
changeset
|
267 | if(client_send_request_body(client)) { |
|
4885cd4c3754
httpclient: move code for sending the request body to a separate function
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
676
diff
changeset
|
268 | return client->error == 0; |
|
675
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
269 | } |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
270 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
271 | |
|
672
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
272 | // writing complete, switch to read events |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
273 | event->events = EVENT_POLLIN; |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
274 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
275 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
276 | char *buffer; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
277 | size_t nbytes; |
|
683
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
278 | if(client->response_header_complete) { |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
279 | buffer = client->buffer.inbuf; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
280 | nbytes = client->buffer.maxsize; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
281 | } else { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
282 | buffer = client->buffer.inbuf + client->buffer.pos; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
283 | nbytes = client->buffer.maxsize - client->buffer.cursize; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
284 | } |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
285 | |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
286 | |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
287 | ssize_t r; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
288 | while((r = read(client->socketfd, buffer, nbytes)) > 0) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
289 | client->buffer.cursize += r; |
|
683
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
290 | if(!client->response_header_complete) { |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
291 | switch(http_parser_process(client->parser)) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
292 | case 0: { // finish |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
293 | if(!http_parser_validate(client->parser)) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
294 | client->error = 1; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
295 | return 0; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
296 | } |
|
672
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
297 | client->statuscode = client->parser->status; |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
298 | |
|
683
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
299 | client->response_header_complete = 1; |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
300 | if(client->response_start) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
301 | cxmutstr msg = client->parser->msg; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
302 | char t = msg.ptr[msg.length]; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
303 | msg.ptr[msg.length] = 0; |
|
672
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
304 | int ret = client->response_start(client, client->statuscode, msg.ptr, client->response_start_userdata); |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
305 | msg.ptr[msg.length] = t; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
306 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
307 | // TODO: check ret |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
308 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
309 | break; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
310 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
311 | case 1: { // need more data |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
312 | continue; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
313 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
314 | case 2: { // error |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
315 | client->error = 1; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
316 | return 0; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
317 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
318 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
319 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
320 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
321 | // header complete |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
322 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
323 | char *out = client->buffer.inbuf + client->buffer.pos; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
324 | size_t len = client->buffer.cursize - client->buffer.pos; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
325 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
326 | if(client->response_body_write) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
327 | int ret = client->response_body_write(client, out, len, client->response_body_write_userdata); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
328 | // TODO: check ret |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
329 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
330 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
331 | client->buffer.pos = 0; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
332 | client->buffer.cursize = 0; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
333 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
334 | |
|
672
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
335 | if(r < 0) { |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
336 | if(errno == EAGAIN) { |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
337 | return 1; |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
338 | } else { |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
339 | log_ereport(LOG_FAILURE, "http-client: IO error: %s", strerror(errno)); |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
340 | } |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
341 | } |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
342 | |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
343 | // request finished |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
344 | if(client->response_finished) { |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
345 | client->response_finished(client, client->response_finished_userdata); |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
346 | } |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
347 | |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
348 | return 0; |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
349 | } |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
350 | |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
351 | static int client_finished(EventHandler *ev, Event *event) { |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
352 | HttpClient *client = event->cookie; |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
353 | |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
354 | close(client->socketfd); |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
355 | client->socketfd = -1; |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
356 | |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
357 | // request finished |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
358 | if(client->response_finished) { |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
359 | client->response_finished(client, client->response_finished_userdata); |
|
226bfd584075
minimally working httpclient
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
671
diff
changeset
|
360 | } |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
361 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
362 | return 0; |
|
666
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
363 | } |
|
c99e0b352e36
add non-blocking http client connect
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
665
diff
changeset
|
364 | |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
365 | static int client_send_request(HttpClient *client) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
366 | size_t nbytes = client->req_buffer_len - client->req_buffer_pos; |
|
675
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
367 | ssize_t w; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
368 | while((w = write(client->socketfd, client->req_buffer + client->req_buffer_pos, nbytes)) > 0) { |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
369 | client->req_buffer_pos += w; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
370 | nbytes = client->req_buffer_len - client->req_buffer_pos; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
371 | if(nbytes == 0) { |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
372 | break; |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
373 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
374 | } |
|
edacba8beedb
add support for request bodies with a fixed content length for the reverse proxy
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
674
diff
changeset
|
375 | |
|
671
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
376 | if(w <= 0) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
377 | if(errno != EAGAIN) { |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
378 | // TODO: log correct host |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
379 | log_ereport(LOG_VERBOSE, "http-client %s - %s: write failed: %s", "localhost", client->uri, strerror(errno)); |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
380 | client->error = 1; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
381 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
382 | return 1; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
383 | } |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
384 | |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
385 | return client->req_buffer_pos < client->req_buffer_len; |
|
879005903b2b
implement basic http client IO
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
669
diff
changeset
|
386 | } |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
387 | |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
388 | static int client_send_request_body(HttpClient *client) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
389 | size_t rbody_readsize = client->req_buffer_alloc; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
390 | size_t rbody_buf_offset = 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
391 | if(client->req_content_length == -1) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
392 | // chunked transfer encoding: |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
393 | // don't fill req_buffer completely, reserve some space for |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
394 | // a chunk header, that will be inserted at the beginning |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
395 | rbody_readsize -= 16; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
396 | rbody_buf_offset = 16; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
397 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
398 | while(!client->request_body_complete) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
399 | ssize_t r = client->request_body_read(client, client->req_buffer + rbody_buf_offset, rbody_readsize, client->request_body_read_userdata); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
400 | if(r <= 0) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
401 | if(r == HTTP_CLIENT_CALLBACK_WOULD_BLOCK) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
402 | return 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
403 | } else if(r == 0) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
404 | // EOF |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
405 | client->request_body_complete = 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
406 | break; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
407 | } else { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
408 | // error |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
409 | client->error = 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
410 | return 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
411 | } |
|
681
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
412 | } else if(client->req_content_length == -1 && r + 32 < rbody_readsize) { |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
413 | // is it time to terminate the request body? |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
414 | // try read some additional bytes, if it returns 0, we know |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
415 | // the request body is complete and we can add the termination chunk |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
416 | char *r2buf = client->req_buffer + rbody_buf_offset + r; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
417 | ssize_t r2 = client->request_body_read(client, r2buf, 32, client->request_body_read_userdata); |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
418 | if(r > 0) { |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
419 | r += r2; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
420 | } else if(r == 0) { |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
421 | memcpy(r2buf, "0\r\n\r\n", 5); |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
422 | r += 5; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
423 | client->request_body_complete = 1; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
424 | client->request_body_terminated = 1; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
425 | } else if(r == HTTP_CLIENT_CALLBACK_WOULD_BLOCK) { |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
426 | return 1; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
427 | } else { |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
428 | client->error = 1; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
429 | return 1; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
430 | } |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
431 | } |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
432 | |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
433 | size_t startpos = 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
434 | if(client->req_content_length == -1) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
435 | char chunkheader[16]; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
436 | int chunkheaderlen = snprintf(chunkheader, 16, "%zx\r\n", (size_t)r); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
437 | startpos = 16 - chunkheaderlen; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
438 | memcpy(client->req_buffer + startpos, chunkheader, chunkheaderlen); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
439 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
440 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
441 | client->req_contentlength_pos += r; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
442 | client->req_buffer_pos = startpos; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
443 | client->req_buffer_len = rbody_buf_offset + r; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
444 | if(client_send_request(client)) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
445 | return 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
446 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
447 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
448 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
449 | // chunked transfer encoding: terminate |
|
681
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
450 | if(client->req_content_length == -1 && !client->request_body_terminated) { |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
451 | memcpy(client->req_buffer, "0\r\n\r\n", 5); |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
452 | client->req_buffer_pos = 0; |
|
681
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
453 | client->req_buffer_len = 5; |
|
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
454 | client->request_body_terminated = 1; |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
455 | if(client_send_request(client)) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
456 | return 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
457 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
458 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
459 | } else if(client->req_content_length != client->req_contentlength_pos) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
460 | // incomplete request body |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
461 | client->error = 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
462 | return 1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
463 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
464 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
465 | return 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
466 | } |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
467 | |
|
683
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
468 | /* |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
469 | static int client_read_response_header(HttpClient *client) { |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
470 | if(client->response_header_complete) { |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
471 | return 0; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
472 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
473 | |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
474 | char *buffer = client->buffer.inbuf + client->buffer.pos; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
475 | size_t nbytes = client->buffer.maxsize - client->buffer.cursize; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
476 | |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
477 | ssize_t r; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
478 | while((r = read(client->socketfd, buffer, nbytes)) > 0) { |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
479 | client->buffer.cursize += r; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
480 | if(!client->response_header_complete) { |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
481 | switch(http_parser_process(client->parser)) { |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
482 | case 0: { // finish |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
483 | if(!http_parser_validate(client->parser)) { |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
484 | client->error = 1; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
485 | return 0; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
486 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
487 | client->statuscode = client->parser->status; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
488 | |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
489 | client->response_header_complete = 1; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
490 | if(client->response_start) { |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
491 | cxmutstr msg = client->parser->msg; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
492 | char t = msg.ptr[msg.length]; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
493 | msg.ptr[msg.length] = 0; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
494 | int ret = client->response_start(client, client->statuscode, msg.ptr, client->response_start_userdata); |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
495 | msg.ptr[msg.length] = t; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
496 | |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
497 | // TODO: check ret |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
498 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
499 | break; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
500 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
501 | case 1: { // need more data |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
502 | continue; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
503 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
504 | case 2: { // error |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
505 | client->error = 1; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
506 | return 0; |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
507 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
508 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
509 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
510 | |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
511 | // header complete |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
512 | |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
513 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
514 | } |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
515 | */ |
|
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
516 | |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
517 | /* --------------------------------- Tests --------------------------------- */ |
|
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
518 | |
|
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
519 | static CX_TEST(test_http_client_send_request) { |
|
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
520 | CX_TEST_DO { |
|
674
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
521 | EventHandler dummy; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
522 | HttpClient *client = http_client_new(&dummy); |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
523 | |
|
674
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
524 | int fds[2]; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
525 | util_socketpair(fds); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
526 | util_socket_setnonblock(fds[0], 1); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
527 | util_socket_setnonblock(fds[1], 1); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
528 | client->socketfd = fds[0]; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
529 | int sock = fds[1]; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
530 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
531 | // create a large test buffer, that is bigger than the socket buffer |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
532 | // 32mb should be enough |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
533 | size_t len = 32*1024*1024; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
534 | char *str = malloc(len); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
535 | // init the buffer with random data |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
536 | for(size_t i=0;i<len;i+=sizeof(int)) { |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
537 | int *p = (int*)(str+i); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
538 | *p = rand();; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
539 | } |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
540 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
541 | client->req_buffer = str; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
542 | client->req_buffer_len = len; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
543 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
544 | // test client_send_request |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
545 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
546 | int ret = client_send_request(client); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
547 | // It is very likely that the first client_send_request call doesn't |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
548 | // fully write the request buffer to the socket |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
549 | // In that case it returns 1 but without the error flag |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
550 | CX_TEST_ASSERT(ret == 1 && !client->error); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
551 | CX_TEST_ASSERT(client->req_buffer_pos > 0); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
552 | CX_TEST_ASSERT(client->req_buffer_pos < len); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
553 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
554 | // read the request buffer from sock and continue with client_send_request |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
555 | CxBuffer buf; |
|
676
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
556 | cxBufferInit(&buf, cxDefaultAllocator, NULL, len, CX_BUFFER_AUTO_EXTEND|CX_BUFFER_FREE_CONTENTS); |
|
674
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
557 | char tmpbuf[1024]; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
558 | int writes = 1; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
559 | while(client->req_buffer_pos < client->req_buffer_len && writes < 2000000) { |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
560 | ssize_t r = read(sock, tmpbuf, 1024); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
561 | CX_TEST_ASSERT(r >= 0); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
562 | cxBufferWrite(tmpbuf, 1, r, &buf); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
563 | ret = client_send_request(client); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
564 | CX_TEST_ASSERT(ret == 0 || (ret == 1 && !client->error)); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
565 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
566 | writes++; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
567 | } |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
568 | CX_TEST_ASSERT(client->req_buffer_pos == client->req_buffer_len); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
569 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
570 | // finish reading the request buffer from sock |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
571 | ssize_t r; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
572 | while((r = read(sock, tmpbuf, 1024)) > 0 && writes < 2000000) { |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
573 | cxBufferWrite(tmpbuf, 1, r, &buf); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
574 | writes++; |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
575 | } |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
576 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
577 | CX_TEST_ASSERT(buf.size == len); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
578 | CX_TEST_ASSERT(!memcmp(str, buf.space, len)); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
579 | |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
580 | // cleanup |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
581 | close(fds[0]); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
582 | close(fds[1]); |
|
6a031133a498
add http_client_send_request test
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
673
diff
changeset
|
583 | http_client_free(client); |
|
676
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
584 | cxBufferDestroy(&buf); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
585 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
586 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
587 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
588 | typedef struct TestResponse { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
589 | int status; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
590 | char *msg; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
591 | CxBuffer *response; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
592 | } TestResponse; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
593 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
594 | static int test_response_start(HttpClient *client, int status, char *msg, void *userdata) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
595 | TestResponse *test = userdata; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
596 | test->status = status; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
597 | test->msg = strdup(msg); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
598 | return 0; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
599 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
600 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
601 | static ssize_t test_response_body_write(HttpClient *client, void *buf, size_t size, void *userdata) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
602 | TestResponse *test = userdata; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
603 | cxBufferWrite(buf, 1, size, test->response); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
604 | return size; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
605 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
606 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
607 | static CX_TEST(test_http_client_io_simple) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
608 | CX_TEST_DO { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
609 | EventHandler dummy; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
610 | HttpClient *client = http_client_new(&dummy); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
611 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
612 | int fds[2]; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
613 | util_socketpair(fds); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
614 | util_socket_setnonblock(fds[0], 1); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
615 | util_socket_setnonblock(fds[1], 1); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
616 | client->socketfd = fds[0]; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
617 | int sock = fds[1]; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
618 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
619 | // setup client |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
620 | http_client_set_uri(client, "/test/uri/"); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
621 | http_client_set_method(client, "GET"); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
622 | http_client_add_request_header(client, cx_mutstr("Host"), cx_mutstr("localhost")); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
623 | http_client_add_request_header(client, cx_mutstr("Test1"), cx_mutstr("value1")); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
624 | http_client_add_request_header(client, cx_mutstr("Test2"), cx_mutstr("value2")); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
625 | create_req_buffer(client); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
626 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
627 | size_t req_header_len = client->req_buffer_len; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
628 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
629 | // response buffer |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
630 | CxBuffer buf; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
631 | cxBufferInit(&buf, cxDefaultAllocator, NULL, 1024, CX_BUFFER_AUTO_EXTEND|CX_BUFFER_FREE_CONTENTS); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
632 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
633 | TestResponse testr = { 0 }; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
634 | testr.response = &buf; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
635 | client->response_start = test_response_start; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
636 | client->response_start_userdata = &testr; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
637 | client->response_body_write = test_response_body_write; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
638 | client->response_body_write_userdata = &testr; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
639 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
640 | // test IO |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
641 | Event event; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
642 | event.cookie = client; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
643 | int ret = client_io(&dummy, &event); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
644 | CX_TEST_ASSERT(!client->error); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
645 | CX_TEST_ASSERT(ret == 1); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
646 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
647 | // do IO and read request until the header is processed |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
648 | size_t req_header_pos = 0; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
649 | char req_buf[4]; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
650 | while(req_header_pos < req_header_len) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
651 | ssize_t r = read(sock, req_buf, 4); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
652 | if(r == 0) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
653 | break; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
654 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
655 | CX_TEST_ASSERT(r > 0); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
656 | req_header_pos += r; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
657 | ret = client_io(&dummy, &event); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
658 | CX_TEST_ASSERT(!client->error); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
659 | CX_TEST_ASSERT(ret == 1); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
660 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
661 | CX_TEST_ASSERT(req_header_pos == req_header_len); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
662 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
663 | char *response_str = |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
664 | "HTTP/1.1 200 OK\r\n" |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
665 | "Host: localhost\r\n" |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
666 | "Content-length: 13\r\n" |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
667 | "\r\n" |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
668 | "Hello World!\n"; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
669 | size_t response_str_len = strlen(response_str); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
670 | size_t response_str_pos = 0; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
671 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
672 | // send response and do IO |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
673 | while(response_str_pos < response_str_len) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
674 | size_t len = response_str_len - response_str_pos; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
675 | if(len > 3) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
676 | //len = 3; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
677 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
678 | ssize_t w = write(sock, response_str + response_str_pos, len); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
679 | if(w == 0) { |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
680 | break; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
681 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
682 | CX_TEST_ASSERT(w > 0); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
683 | response_str_pos += w; |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
684 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
685 | ret = client_io(&dummy, &event); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
686 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
687 | CX_TEST_ASSERT(!client->error); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
688 | CX_TEST_ASSERT(ret == 1); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
689 | } |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
690 | CX_TEST_ASSERT(response_str_pos == response_str_len); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
691 | CX_TEST_ASSERT(testr.status == 200); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
692 | CX_TEST_ASSERT(testr.msg); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
693 | CX_TEST_ASSERT(!strcmp(testr.msg, "OK")); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
694 | CX_TEST_ASSERT(testr.response->size == 13); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
695 | CX_TEST_ASSERT(!cx_strcmp(cx_strn(testr.response->space, testr.response->size), "Hello World!\n")); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
696 | |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
697 | // cleanup |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
698 | free(testr.msg); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
699 | close(fds[0]); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
700 | close(fds[1]); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
701 | http_client_free(client); |
|
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
702 | cxBufferDestroy(&buf); |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
703 | } |
|
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
704 | } |
|
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
705 | |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
706 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
707 | typedef struct TestRequestBody { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
708 | char *content; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
709 | size_t length; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
710 | size_t pos; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
711 | int chunksize; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
712 | int max_reads; // max number of reads until test_request_body_read returns 0 |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
713 | int cur_reads; // current number of read-attempts |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
714 | } TestRequestBody; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
715 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
716 | static ssize_t test_request_body_read(HttpClient *client, void *buf, size_t size, void *userdata) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
717 | TestRequestBody *req = userdata; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
718 | req->cur_reads++; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
719 | if(req->chunksize == 0 || req->cur_reads > req->max_reads) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
720 | return -1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
721 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
722 | size_t max = req->length - req->pos; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
723 | if(max == 0) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
724 | return 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
725 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
726 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
727 | size_t sz = req->chunksize > size ? size : req->chunksize; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
728 | if(sz > max) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
729 | sz = max; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
730 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
731 | memcpy(buf, req->content + req->pos, sz); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
732 | req->pos += sz; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
733 | return sz; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
734 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
735 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
736 | static CX_TEST(test_http_client_send_request_body_chunked) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
737 | CX_TEST_DO { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
738 | EventHandler dummy; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
739 | HttpClient *client = http_client_new(&dummy); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
740 | create_req_buffer(client); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
741 | client->req_content_length = -1; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
742 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
743 | int fds[2]; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
744 | util_socketpair(fds); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
745 | util_socket_setnonblock(fds[0], 1); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
746 | util_socket_setnonblock(fds[1], 1); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
747 | client->socketfd = fds[0]; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
748 | int sock = fds[1]; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
749 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
750 | // response buffer |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
751 | CxBuffer buf; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
752 | cxBufferInit(&buf, cxDefaultAllocator, NULL, 1024, CX_BUFFER_AUTO_EXTEND|CX_BUFFER_FREE_CONTENTS); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
753 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
754 | // test |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
755 | char request_body[1024]; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
756 | memset(request_body, 'x', 1024); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
757 | memset(request_body+128, 'y', 128); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
758 | memset(request_body+384, 'z', 128); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
759 | memset(request_body+640, ':', 128); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
760 | memset(request_body+896, '!', 128); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
761 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
762 | TestRequestBody req; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
763 | req.content = request_body; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
764 | req.length = 1024; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
765 | req.pos = 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
766 | req.chunksize = 16; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
767 | req.max_reads = 8; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
768 | req.cur_reads = 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
769 | client->request_body_read = test_request_body_read; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
770 | client->request_body_read_userdata = &req; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
771 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
772 | memset(client->req_buffer, '_', client->req_buffer_alloc); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
773 | client->req_buffer_pos = 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
774 | client->req_buffer_len = 0; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
775 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
776 | // send the first 128 bytes |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
777 | while(req.cur_reads <= req.max_reads) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
778 | int ret = client_send_request_body(client); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
779 | CX_TEST_ASSERT(ret == 1); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
780 | CX_TEST_ASSERT(!client->error); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
781 | char buf2[1024]; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
782 | ssize_t r = read(sock, buf2, 1024); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
783 | if(r > 0) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
784 | cxBufferWrite(buf2, 1, r, &buf); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
785 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
786 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
787 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
788 | // because we are using chunked transfer encoding, the result buffer |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
789 | // (buf) should contain more than 128 bytes (additional chunk headers) |
|
681
e9705d51866a
optimize chunked transfer encoding termination in client_send_request_body
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
680
diff
changeset
|
790 | CX_TEST_ASSERT(buf.pos > 128); |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
791 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
792 | // change chunk size to 128 |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
793 | req.max_reads = 9999; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
794 | req.chunksize = 128; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
795 | while(req.cur_reads <= req.max_reads) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
796 | int ret = client_send_request_body(client); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
797 | CX_TEST_ASSERT(!client->error); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
798 | char buf2[2048]; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
799 | ssize_t r; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
800 | while((r = read(sock, buf2, 2048)) > 0) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
801 | cxBufferWrite(buf2, 1, r, &buf); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
802 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
803 | if(ret == 0) { |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
804 | break; |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
805 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
806 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
807 | CX_TEST_ASSERT(req.cur_reads < req.max_reads); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
808 | |
|
682
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
809 | // verify chunks |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
810 | char test_request_body[1024]; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
811 | memset(test_request_body, 0, 1024); |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
812 | |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
813 | int pos = 0; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
814 | int chunklen = 0; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
815 | char *str = buf.space; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
816 | while(str < buf.space + buf.size) { |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
817 | cxstring chunkheader = cx_strn(str, 2); |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
818 | if(!cx_strcmp(chunkheader, "0\r")) { |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
819 | chunkheader.length = 1; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
820 | } |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
821 | int ret = cx_strtoi(chunkheader, &chunklen, 16); |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
822 | CX_TEST_ASSERT(ret == 0); |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
823 | if(chunklen == 0) { |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
824 | break; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
825 | } |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
826 | |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
827 | char *data = str + 4; |
|
683
db37761a8494
enable buffered reader for request bodies with fixed content-length
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
682
diff
changeset
|
828 | CX_TEST_ASSERT(data + chunklen < buf.space + buf.size); |
|
682
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
829 | memcpy(test_request_body + pos, data, chunklen); |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
830 | pos += chunklen; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
831 | str = data + chunklen; |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
832 | } |
|
f4c593a99266
verify chunks in test_http_client_send_request_body_chunked
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
681
diff
changeset
|
833 | CX_TEST_ASSERT(!memcmp(request_body, test_request_body, 1024)); |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
834 | |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
835 | // cleanup |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
836 | close(fds[0]); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
837 | close(fds[1]); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
838 | http_client_free(client); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
839 | cxBufferDestroy(&buf); |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
840 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
841 | } |
|
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
842 | |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
843 | void http_client_add_tests(CxTestSuite *suite) { |
|
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
844 | cx_test_register(suite, test_http_client_send_request); |
|
680
02935baa186b
httpclient: add support for chunked transfer encoding for request bodies
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
679
diff
changeset
|
845 | cx_test_register(suite, test_http_client_send_request_body_chunked); |
|
676
d43f1dd8b18e
add test_http_client_io_simple
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
675
diff
changeset
|
846 | cx_test_register(suite, test_http_client_io_simple); |
|
673
144bdc33fdb6
add socket utils
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
672
diff
changeset
|
847 | } |