xref: /freebsd/contrib/wpa/src/wps/http_client.c (revision b626f5a73a48f44a31a200291b141e1da408a2ff)
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