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 89 wpa_printf(MSG_DEBUG, "HTTP: Send client request to %s:%d (%lu of %lu " 90 "bytes remaining)", 91 inet_ntoa(c->dst.sin_addr), ntohs(c->dst.sin_port), 92 (unsigned long) wpabuf_len(c->req), 93 (unsigned long) wpabuf_len(c->req) - c->req_pos); 94 95 res = send(c->sd, wpabuf_head_u8(c->req) + c->req_pos, 96 wpabuf_len(c->req) - c->req_pos, 0); 97 if (res < 0) { 98 wpa_printf(MSG_DEBUG, "HTTP: Failed to send buffer: %s", 99 strerror(errno)); 100 eloop_unregister_sock(c->sd, EVENT_TYPE_WRITE); 101 c->cb(c->cb_ctx, c, HTTP_CLIENT_FAILED); 102 return; 103 } 104 105 if ((size_t) res < wpabuf_len(c->req) - c->req_pos) { 106 wpa_printf(MSG_DEBUG, "HTTP: Sent %d of %lu bytes; %lu bytes " 107 "remaining", 108 res, (unsigned long) wpabuf_len(c->req), 109 (unsigned long) wpabuf_len(c->req) - c->req_pos - 110 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 http_client_free(c); 151 return NULL; 152 } 153 154 if (fcntl(c->sd, F_SETFL, O_NONBLOCK) != 0) { 155 wpa_printf(MSG_DEBUG, "HTTP: fnctl(O_NONBLOCK) failed: %s", 156 strerror(errno)); 157 http_client_free(c); 158 return NULL; 159 } 160 161 if (connect(c->sd, (struct sockaddr *) dst, sizeof(*dst))) { 162 if (errno != EINPROGRESS) { 163 wpa_printf(MSG_DEBUG, "HTTP: Failed to connect: %s", 164 strerror(errno)); 165 http_client_free(c); 166 return NULL; 167 } 168 169 /* 170 * Continue connecting in the background; eloop will call us 171 * once the connection is ready (or failed). 172 */ 173 } 174 175 if (eloop_register_sock(c->sd, EVENT_TYPE_WRITE, http_client_tx_ready, 176 c, NULL)) { 177 http_client_free(c); 178 return NULL; 179 } 180 181 if (eloop_register_timeout(HTTP_CLIENT_TIMEOUT_SEC, 0, 182 http_client_timeout, c, NULL)) { 183 http_client_free(c); 184 return NULL; 185 } 186 187 c->req = req; 188 189 return c; 190 } 191 192 193 char * http_client_url_parse(const char *url, struct sockaddr_in *dst, 194 char **ret_path) 195 { 196 char *u, *addr, *port, *path; 197 198 u = os_strdup(url); 199 if (u == NULL) 200 return NULL; 201 202 os_memset(dst, 0, sizeof(*dst)); 203 dst->sin_family = AF_INET; 204 addr = u + 7; 205 path = os_strchr(addr, '/'); 206 port = os_strchr(addr, ':'); 207 if (path == NULL) { 208 path = "/"; 209 } else { 210 *path = '\0'; /* temporary nul termination for address */ 211 if (port > path) 212 port = NULL; 213 } 214 if (port) 215 *port++ = '\0'; 216 217 if (inet_aton(addr, &dst->sin_addr) == 0) { 218 /* TODO: name lookup */ 219 wpa_printf(MSG_DEBUG, "HTTP: Unsupported address in URL '%s' " 220 "(addr='%s' port='%s')", 221 url, addr, port); 222 os_free(u); 223 return NULL; 224 } 225 226 if (port) 227 dst->sin_port = htons(atoi(port)); 228 else 229 dst->sin_port = htons(80); 230 231 if (*path == '\0') { 232 /* remove temporary nul termination for address */ 233 *path = '/'; 234 } 235 236 *ret_path = path; 237 238 return u; 239 } 240 241 242 struct http_client * http_client_url(const char *url, 243 struct wpabuf *req, size_t max_response, 244 void (*cb)(void *ctx, 245 struct http_client *c, 246 enum http_client_event event), 247 void *cb_ctx) 248 { 249 struct sockaddr_in dst; 250 struct http_client *c; 251 char *u, *path; 252 struct wpabuf *req_buf = NULL; 253 254 if (os_strncmp(url, "http://", 7) != 0) 255 return NULL; 256 u = http_client_url_parse(url, &dst, &path); 257 if (u == NULL) 258 return NULL; 259 260 if (req == NULL) { 261 req_buf = wpabuf_alloc(os_strlen(url) + 1000); 262 if (req_buf == NULL) { 263 os_free(u); 264 return NULL; 265 } 266 req = req_buf; 267 wpabuf_printf(req, 268 "GET %s HTTP/1.1\r\n" 269 "Cache-Control: no-cache\r\n" 270 "Pragma: no-cache\r\n" 271 "Accept: text/xml, application/xml\r\n" 272 "User-Agent: wpa_supplicant\r\n" 273 "Host: %s:%d\r\n" 274 "\r\n", 275 path, inet_ntoa(dst.sin_addr), 276 ntohs(dst.sin_port)); 277 } 278 os_free(u); 279 280 c = http_client_addr(&dst, req, max_response, cb, cb_ctx); 281 if (c == NULL) { 282 wpabuf_free(req_buf); 283 return NULL; 284 } 285 286 return c; 287 } 288 289 290 void http_client_free(struct http_client *c) 291 { 292 if (c == NULL) 293 return; 294 httpread_destroy(c->hread); 295 wpabuf_free(c->req); 296 if (c->sd >= 0) { 297 eloop_unregister_sock(c->sd, EVENT_TYPE_WRITE); 298 close(c->sd); 299 } 300 eloop_cancel_timeout(http_client_timeout, c, NULL); 301 os_free(c); 302 } 303 304 305 struct wpabuf * http_client_get_body(struct http_client *c) 306 { 307 if (c->hread == NULL) 308 return NULL; 309 wpabuf_set(&c->body, httpread_data_get(c->hread), 310 httpread_length_get(c->hread)); 311 return &c->body; 312 } 313 314 315 char * http_client_get_hdr_line(struct http_client *c, const char *tag) 316 { 317 if (c->hread == NULL) 318 return NULL; 319 return httpread_hdr_line_get(c->hread, tag); 320 } 321 322 323 char * http_link_update(char *url, const char *base) 324 { 325 char *n; 326 size_t len; 327 const char *pos; 328 329 /* RFC 2396, Chapter 5.2 */ 330 /* TODO: consider adding all cases described in RFC 2396 */ 331 332 if (url == NULL) 333 return NULL; 334 335 if (os_strncmp(url, "http://", 7) == 0) 336 return url; /* absolute link */ 337 338 if (os_strncmp(base, "http://", 7) != 0) 339 return url; /* unable to handle base URL */ 340 341 len = os_strlen(url) + 1 + os_strlen(base) + 1; 342 n = os_malloc(len); 343 if (n == NULL) 344 return url; /* failed */ 345 346 if (url[0] == '/') { 347 pos = os_strchr(base + 7, '/'); 348 if (pos == NULL) { 349 os_snprintf(n, len, "%s%s", base, url); 350 } else { 351 os_memcpy(n, base, pos - base); 352 os_memcpy(n + (pos - base), url, os_strlen(url) + 1); 353 } 354 } else { 355 pos = os_strrchr(base + 7, '/'); 356 if (pos == NULL) { 357 os_snprintf(n, len, "%s/%s", base, url); 358 } else { 359 os_memcpy(n, base, pos - base + 1); 360 os_memcpy(n + (pos - base) + 1, url, os_strlen(url) + 361 1); 362 } 363 } 364 365 os_free(url); 366 367 return n; 368 } 369
