1 /* 2 * http_client - HTTP client 3 * Copyright (c) 2009, Jouni Malinen <j (at) 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 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 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 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 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 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 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 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 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 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 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
