1 /*
2  * http_client - HTTP client
3  * Copyright (c) 2009, Jouni Malinen <j@w1.fi>
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8 
9 #include "includes.h"
10 #include <fcntl.h>
11 
12 #include "common.h"
13 #include "eloop.h"
14 #include "httpread.h"
15 #include "http_client.h"
16 
17 
18 #define HTTP_CLIENT_TIMEOUT_SEC 30
19 
20 
21 struct http_client {
22           struct sockaddr_in dst;
23           int sd;
24           struct wpabuf *req;
25           size_t req_pos;
26           size_t max_response;
27 
28           void (*cb)(void *ctx, struct http_client *c,
29                        enum http_client_event event);
30           void *cb_ctx;
31           struct httpread *hread;
32           struct wpabuf body;
33 };
34 
35 
http_client_timeout(void * eloop_data,void * user_ctx)36 static void http_client_timeout(void *eloop_data, void *user_ctx)
37 {
38           struct http_client *c = eloop_data;
39           wpa_printf(MSG_DEBUG, "HTTP: Timeout (c=%p)", c);
40           c->cb(c->cb_ctx, c, HTTP_CLIENT_TIMEOUT);
41 }
42 
43 
http_client_got_response(struct httpread * handle,void * cookie,enum httpread_event e)44 static void http_client_got_response(struct httpread *handle, void *cookie,
45                                              enum httpread_event e)
46 {
47           struct http_client *c = cookie;
48 
49           wpa_printf(MSG_DEBUG, "HTTP: httpread callback: handle=%p cookie=%p "
50                        "e=%d", handle, cookie, e);
51 
52           eloop_cancel_timeout(http_client_timeout, c, NULL);
53           switch (e) {
54           case HTTPREAD_EVENT_FILE_READY:
55                     if (httpread_hdr_type_get(c->hread) == HTTPREAD_HDR_TYPE_REPLY)
56                     {
57                               int reply_code = httpread_reply_code_get(c->hread);
58                               if (reply_code == 200 /* OK */) {
59                                         wpa_printf(MSG_DEBUG, "HTTP: Response OK from "
60                                                      "%s:%d",
61                                                      inet_ntoa(c->dst.sin_addr),
62                                                      ntohs(c->dst.sin_port));
63                                         c->cb(c->cb_ctx, c, HTTP_CLIENT_OK);
64                               } else {
65                                         wpa_printf(MSG_DEBUG, "HTTP: Error %d from "
66                                                      "%s:%d", reply_code,
67                                                      inet_ntoa(c->dst.sin_addr),
68                                                      ntohs(c->dst.sin_port));
69                                         c->cb(c->cb_ctx, c, HTTP_CLIENT_INVALID_REPLY);
70                               }
71                     } else
72                               c->cb(c->cb_ctx, c, HTTP_CLIENT_INVALID_REPLY);
73                     break;
74           case HTTPREAD_EVENT_TIMEOUT:
75                     c->cb(c->cb_ctx, c, HTTP_CLIENT_TIMEOUT);
76                     break;
77           case HTTPREAD_EVENT_ERROR:
78                     c->cb(c->cb_ctx, c, HTTP_CLIENT_FAILED);
79                     break;
80           }
81 }
82 
83 
http_client_tx_ready(int sock,void * eloop_ctx,void * sock_ctx)84 static void http_client_tx_ready(int sock, void *eloop_ctx, void *sock_ctx)
85 {
86           struct http_client *c = eloop_ctx;
87           int res;
88           size_t send_len;
89 
90           send_len = wpabuf_len(c->req) - c->req_pos;
91           wpa_printf(MSG_DEBUG, "HTTP: Send client request to %s:%d (%lu of %lu "
92                        "bytes remaining)",
93                        inet_ntoa(c->dst.sin_addr), ntohs(c->dst.sin_port),
94                        (unsigned long) wpabuf_len(c->req),
95                        (unsigned long) send_len);
96 
97           res = send(c->sd, wpabuf_head_u8(c->req) + c->req_pos, send_len, 0);
98           if (res < 0) {
99                     wpa_printf(MSG_DEBUG, "HTTP: Failed to send buffer: %s",
100                                  strerror(errno));
101                     eloop_unregister_sock(c->sd, EVENT_TYPE_WRITE);
102                     c->cb(c->cb_ctx, c, HTTP_CLIENT_FAILED);
103                     return;
104           }
105 
106           if ((size_t) res < send_len) {
107                     wpa_printf(MSG_DEBUG, "HTTP: Sent %d of %lu bytes; %lu bytes "
108                                  "remaining",
109                                  res, (unsigned long) wpabuf_len(c->req),
110                                  (unsigned long) send_len - res);
111                     c->req_pos += res;
112                     return;
113           }
114 
115           wpa_printf(MSG_DEBUG, "HTTP: Full client request sent to %s:%d",
116                        inet_ntoa(c->dst.sin_addr), ntohs(c->dst.sin_port));
117           eloop_unregister_sock(c->sd, EVENT_TYPE_WRITE);
118           wpabuf_free(c->req);
119           c->req = NULL;
120 
121           c->hread = httpread_create(c->sd, http_client_got_response, c,
122                                            c->max_response, HTTP_CLIENT_TIMEOUT_SEC);
123           if (c->hread == NULL) {
124                     c->cb(c->cb_ctx, c, HTTP_CLIENT_FAILED);
125                     return;
126           }
127 }
128 
129 
http_client_addr(struct sockaddr_in * dst,struct wpabuf * req,size_t max_response,void (* cb)(void * ctx,struct http_client * c,enum http_client_event event),void * cb_ctx)130 struct http_client * http_client_addr(struct sockaddr_in *dst,
131                                               struct wpabuf *req, size_t max_response,
132                                               void (*cb)(void *ctx,
133                                                              struct http_client *c,
134                                                              enum http_client_event event),
135                                               void *cb_ctx)
136 {
137           struct http_client *c;
138 
139           c = os_zalloc(sizeof(*c));
140           if (c == NULL)
141                     return NULL;
142           c->sd = -1;
143           c->dst = *dst;
144           c->max_response = max_response;
145           c->cb = cb;
146           c->cb_ctx = cb_ctx;
147 
148           c->sd = socket(AF_INET, SOCK_STREAM, 0);
149           if (c->sd < 0)
150                     goto fail;
151 
152           if (fcntl(c->sd, F_SETFL, O_NONBLOCK) != 0) {
153                     wpa_printf(MSG_DEBUG, "HTTP: fnctl(O_NONBLOCK) failed: %s",
154                                  strerror(errno));
155                     goto fail;
156           }
157 
158           if (connect(c->sd, (struct sockaddr *) dst, sizeof(*dst))) {
159                     if (errno != EINPROGRESS) {
160                               wpa_printf(MSG_DEBUG, "HTTP: Failed to connect: %s",
161                                            strerror(errno));
162                               goto fail;
163                     }
164 
165                     /*
166                      * Continue connecting in the background; eloop will call us
167                      * once the connection is ready (or failed).
168                      */
169           }
170 
171           if (eloop_register_sock(c->sd, EVENT_TYPE_WRITE, http_client_tx_ready,
172                                         c, NULL) ||
173               eloop_register_timeout(HTTP_CLIENT_TIMEOUT_SEC, 0,
174                                            http_client_timeout, c, NULL))
175                     goto fail;
176 
177           c->req = req;
178 
179           return c;
180 
181 fail:
182           http_client_free(c);
183           return NULL;
184 }
185 
186 
http_client_url_parse(const char * url,struct sockaddr_in * dst,char ** ret_path)187 char * http_client_url_parse(const char *url, struct sockaddr_in *dst,
188                                    char **ret_path)
189 {
190           char *u, *addr, *port, *path;
191 
192           u = os_strdup(url);
193           if (u == NULL)
194                     return NULL;
195 
196           os_memset(dst, 0, sizeof(*dst));
197           dst->sin_family = AF_INET;
198           addr = u + 7;
199           path = os_strchr(addr, '/');
200           port = os_strchr(addr, ':');
201           if (path == NULL) {
202                     path = "/";
203           } else {
204                     *path = '\0'; /* temporary nul termination for address */
205                     if (port > path)
206                               port = NULL;
207           }
208           if (port)
209                     *port++ = '\0';
210 
211           if (inet_aton(addr, &dst->sin_addr) == 0) {
212                     /* TODO: name lookup */
213                     wpa_printf(MSG_DEBUG, "HTTP: Unsupported address in URL '%s' "
214                                  "(addr='%s' port='%s')",
215                                  url, addr, port);
216                     os_free(u);
217                     return NULL;
218           }
219 
220           if (port)
221                     dst->sin_port = htons(atoi(port));
222           else
223                     dst->sin_port = htons(80);
224 
225           if (*path == '\0') {
226                     /* remove temporary nul termination for address */
227                     *path = '/';
228           }
229 
230           *ret_path = path;
231 
232           return u;
233 }
234 
235 
http_client_url(const char * url,struct wpabuf * req,size_t max_response,void (* cb)(void * ctx,struct http_client * c,enum http_client_event event),void * cb_ctx)236 struct http_client * http_client_url(const char *url,
237                                              struct wpabuf *req, size_t max_response,
238                                              void (*cb)(void *ctx,
239                                                             struct http_client *c,
240                                                             enum http_client_event event),
241                                              void *cb_ctx)
242 {
243           struct sockaddr_in dst;
244           struct http_client *c;
245           char *u, *path;
246           struct wpabuf *req_buf = NULL;
247 
248           if (os_strncmp(url, "http://", 7) != 0)
249                     return NULL;
250           u = http_client_url_parse(url, &dst, &path);
251           if (u == NULL)
252                     return NULL;
253 
254           if (req == NULL) {
255                     req_buf = wpabuf_alloc(os_strlen(url) + 1000);
256                     if (req_buf == NULL) {
257                               os_free(u);
258                               return NULL;
259                     }
260                     req = req_buf;
261                     wpabuf_printf(req,
262                                     "GET %s HTTP/1.1\r\n"
263                                     "Cache-Control: no-cache\r\n"
264                                     "Pragma: no-cache\r\n"
265                                     "Accept: text/xml, application/xml\r\n"
266                                     "User-Agent: wpa_supplicant\r\n"
267                                     "Host: %s:%d\r\n"
268                                     "\r\n",
269                                     path, inet_ntoa(dst.sin_addr),
270                                     ntohs(dst.sin_port));
271           }
272           os_free(u);
273 
274           c = http_client_addr(&dst, req, max_response, cb, cb_ctx);
275           if (c == NULL) {
276                     wpabuf_free(req_buf);
277                     return NULL;
278           }
279 
280           return c;
281 }
282 
283 
http_client_free(struct http_client * c)284 void http_client_free(struct http_client *c)
285 {
286           if (c == NULL)
287                     return;
288           httpread_destroy(c->hread);
289           wpabuf_free(c->req);
290           if (c->sd >= 0) {
291                     eloop_unregister_sock(c->sd, EVENT_TYPE_WRITE);
292                     close(c->sd);
293           }
294           eloop_cancel_timeout(http_client_timeout, c, NULL);
295           os_free(c);
296 }
297 
298 
http_client_get_body(struct http_client * c)299 struct wpabuf * http_client_get_body(struct http_client *c)
300 {
301           if (c->hread == NULL)
302                     return NULL;
303           wpabuf_set(&c->body, httpread_data_get(c->hread),
304                        httpread_length_get(c->hread));
305           return &c->body;
306 }
307 
308 
http_client_get_hdr_line(struct http_client * c,const char * tag)309 char * http_client_get_hdr_line(struct http_client *c, const char *tag)
310 {
311           if (c->hread == NULL)
312                     return NULL;
313           return httpread_hdr_line_get(c->hread, tag);
314 }
315 
316 
http_link_update(char * url,const char * base)317 char * http_link_update(char *url, const char *base)
318 {
319           char *n;
320           size_t len;
321           const char *pos;
322 
323           /* RFC 2396, Chapter 5.2 */
324           /* TODO: consider adding all cases described in RFC 2396 */
325 
326           if (url == NULL)
327                     return NULL;
328 
329           if (os_strncmp(url, "http://", 7) == 0)
330                     return url; /* absolute link */
331 
332           if (os_strncmp(base, "http://", 7) != 0)
333                     return url; /* unable to handle base URL */
334 
335           len = os_strlen(url) + 1 + os_strlen(base) + 1;
336           n = os_malloc(len);
337           if (n == NULL)
338                     return url; /* failed */
339 
340           if (url[0] == '/') {
341                     pos = os_strchr(base + 7, '/');
342                     if (pos == NULL) {
343                               os_snprintf(n, len, "%s%s", base, url);
344                     } else {
345                               os_memcpy(n, base, pos - base);
346                               os_memcpy(n + (pos - base), url, os_strlen(url) + 1);
347                     }
348           } else {
349                     pos = os_strrchr(base + 7, '/');
350                     if (pos == NULL) {
351                               os_snprintf(n, len, "%s/%s", base, url);
352                     } else {
353                               os_memcpy(n, base, pos - base + 1);
354                               os_memcpy(n + (pos - base) + 1, url, os_strlen(url) +
355                                           1);
356                     }
357           }
358 
359           os_free(url);
360 
361           return n;
362 }
363