Nicolas Mathon / mbed-http

Experimental HTTP and HTTPS library for mbed OS 5

source/http_request_builder.h@33:305c17fe6737, 2018-12-23 (annotated)

Committer:
nmathon
Date:
Sun Dec 23 17:55:40 2018 +0000
Revision:
33:305c17fe6737
Parent:
32:fa4d71265625 
Remove TLSSocket because it's already in Mbed 0S 5.11

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Jan Jongboom 0:910f5949759f 1 /*
Jan Jongboom 0:910f5949759f 2 * PackageLicenseDeclared: Apache-2.0
Jan Jongboom 0:910f5949759f 3 * Copyright (c) 2017 ARM Limited
Jan Jongboom 0:910f5949759f 4 *
Jan Jongboom 0:910f5949759f 5 * Licensed under the Apache License, Version 2.0 (the "License");
Jan Jongboom 0:910f5949759f 6 * you may not use this file except in compliance with the License.
Jan Jongboom 0:910f5949759f 7 * You may obtain a copy of the License at
Jan Jongboom 0:910f5949759f 8 *
Jan Jongboom 0:910f5949759f 9 * http://www.apache.org/licenses/LICENSE-2.0
Jan Jongboom 0:910f5949759f 10 *
Jan Jongboom 0:910f5949759f 11 * Unless required by applicable law or agreed to in writing, software
Jan Jongboom 0:910f5949759f 12 * distributed under the License is distributed on an "AS IS" BASIS,
Jan Jongboom 0:910f5949759f 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
Jan Jongboom 0:910f5949759f 14 * See the License for the specific language governing permissions and
Jan Jongboom 0:910f5949759f 15 * limitations under the License.
Jan Jongboom 0:910f5949759f 16 */
Jan Jongboom 0:910f5949759f 17
Jan Jongboom 0:910f5949759f 18 #ifndef _MBED_HTTP_REQUEST_BUILDER_H_
Jan Jongboom 0:910f5949759f 19 #define _MBED_HTTP_REQUEST_BUILDER_H_
Jan Jongboom 0:910f5949759f 20
Jan Jongboom 0:910f5949759f 21 #include <string>
Jan Jongboom 0:910f5949759f 22 #include <map>
Jan Jongboom 0:910f5949759f 23 #include "http_parser.h"
Jan Jongboom 0:910f5949759f 24 #include "http_parsed_url.h"
Jan Jongboom 0:910f5949759f 25
Jan Jongboom 0:910f5949759f 26 class HttpRequestBuilder {
Jan Jongboom 0:910f5949759f 27 public:
Jan Jongboom 0:910f5949759f 28 HttpRequestBuilder(http_method a_method, ParsedUrl* a_parsed_url)
Jan Jongboom 0:910f5949759f 29 : method(a_method), parsed_url(a_parsed_url)
Jan Jongboom 0:910f5949759f 30 {
Jan Jongboom 21:fcd2bfd31a39 31 string host(parsed_url->host());
Jan Jongboom 21:fcd2bfd31a39 32
Jan Jongboom 21:fcd2bfd31a39 33 char port_str[10];
Jan Jongboom 21:fcd2bfd31a39 34 sprintf(port_str, ":%d", parsed_url->port());
Jan Jongboom 21:fcd2bfd31a39 35
Jan Jongboom 21:fcd2bfd31a39 36 if (strcmp(parsed_url->schema(), "http") == 0 && parsed_url->port() != 80) {
Jan Jongboom 21:fcd2bfd31a39 37 host += string(port_str);
Jan Jongboom 21:fcd2bfd31a39 38 }
Jan Jongboom 21:fcd2bfd31a39 39 else if (strcmp(parsed_url->schema(), "https") == 0 && parsed_url->port() != 443) {
Jan Jongboom 21:fcd2bfd31a39 40 host += string(port_str);
Jan Jongboom 21:fcd2bfd31a39 41 }
Jan Jongboom 21:fcd2bfd31a39 42
Jan Jongboom 21:fcd2bfd31a39 43 set_header("Host", host);
Jan Jongboom 0:910f5949759f 44 }
Jan Jongboom 0:910f5949759f 45
Jan Jongboom 0:910f5949759f 46 /**
Jan Jongboom 0:910f5949759f 47 * Set a header for the request
Jan Jongboom 0:910f5949759f 48 * If the key already exists, it will be overwritten...
Jan Jongboom 0:910f5949759f 49 */
Jan Jongboom 0:910f5949759f 50 void set_header(string key, string value) {
Jan Jongboom 0:910f5949759f 51 map<string, string>::iterator it = headers.find(key);
Jan Jongboom 0:910f5949759f 52
Jan Jongboom 0:910f5949759f 53 if (it != headers.end()) {
Jan Jongboom 0:910f5949759f 54 it->second = value;
Jan Jongboom 0:910f5949759f 55 }
Jan Jongboom 0:910f5949759f 56 else {
Jan Jongboom 0:910f5949759f 57 headers.insert(headers.end(), pair<string, string>(key, value));
Jan Jongboom 0:910f5949759f 58 }
Jan Jongboom 0:910f5949759f 59 }
Jan Jongboom 0:910f5949759f 60
Jan Jongboom 31:b3730a2c4f39 61 char* build(const void* body, uint32_t body_size, uint32_t &size, bool skip_content_length = false) {
Jan Jongboom 0:910f5949759f 62 const char* method_str = http_method_str(method);
Jan Jongboom 0:910f5949759f 63
Jan Jongboom 23:15fa2726f793 64 bool is_chunked = has_header("Transfer-Encoding", "chunked");
Jan Jongboom 23:15fa2726f793 65
Jan Jongboom 23:15fa2726f793 66 if (!is_chunked && (method == HTTP_POST || method == HTTP_PUT || method == HTTP_DELETE || body_size > 0)) {
Jan Jongboom 0:910f5949759f 67 char buffer[10];
Jan Jongboom 31:b3730a2c4f39 68 snprintf(buffer, 10, "%lu", body_size);
Jan Jongboom 0:910f5949759f 69 set_header("Content-Length", string(buffer));
Jan Jongboom 0:910f5949759f 70 }
Jan Jongboom 0:910f5949759f 71
Jan Jongboom 10:b017c7d2cf23 72 size = 0;
Jan Jongboom 0:910f5949759f 73
Jan Jongboom 0:910f5949759f 74 // first line is METHOD PATH+QUERY HTTP/1.1\r\n
Jan Jongboom 10:b017c7d2cf23 75 size += strlen(method_str) + 1 + strlen(parsed_url->path()) + (strlen(parsed_url->query()) ? strlen(parsed_url->query()) + 1 : 0) + 1 + 8 + 2;
Jan Jongboom 0:910f5949759f 76
Jan Jongboom 0:910f5949759f 77 // after that we'll do the headers
Jan Jongboom 0:910f5949759f 78 typedef map<string, string>::iterator it_type;
Jan Jongboom 0:910f5949759f 79 for(it_type it = headers.begin(); it != headers.end(); it++) {
Jan Jongboom 0:910f5949759f 80 // line is KEY: VALUE\r\n
Jan Jongboom 0:910f5949759f 81 size += it->first.length() + 1 + 1 + it->second.length() + 2;
Jan Jongboom 0:910f5949759f 82 }
Jan Jongboom 0:910f5949759f 83
Jan Jongboom 0:910f5949759f 84 // then the body, first an extra newline
Jan Jongboom 0:910f5949759f 85 size += 2;
Jan Jongboom 0:910f5949759f 86
Jan Jongboom 23:15fa2726f793 87 if (!is_chunked) {
Jan Jongboom 23:15fa2726f793 88 // body
Jan Jongboom 23:15fa2726f793 89 size += body_size;
Jan Jongboom 23:15fa2726f793 90 }
Jan Jongboom 0:910f5949759f 91
Jan Jongboom 0:910f5949759f 92 // Now let's print it
Jan Jongboom 0:910f5949759f 93 char* req = (char*)calloc(size + 1, 1);
Jan Jongboom 0:910f5949759f 94 char* originalReq = req;
Jan Jongboom 0:910f5949759f 95
Jan Jongboom 10:b017c7d2cf23 96 if (strlen(parsed_url->query())) {
Jan Jongboom 10:b017c7d2cf23 97 sprintf(req, "%s %s?%s HTTP/1.1\r\n", method_str, parsed_url->path(), parsed_url->query());
Jan Jongboom 10:b017c7d2cf23 98 } else {
Jan Jongboom 10:b017c7d2cf23 99 sprintf(req, "%s %s%s HTTP/1.1\r\n", method_str, parsed_url->path(), parsed_url->query());
Jan Jongboom 7:2e3eedb9ca5c 100 }
Jan Jongboom 10:b017c7d2cf23 101 req += strlen(method_str) + 1 + strlen(parsed_url->path()) + (strlen(parsed_url->query()) ? strlen(parsed_url->query()) + 1 : 0) + 1 + 8 + 2;
Jan Jongboom 0:910f5949759f 102
Jan Jongboom 0:910f5949759f 103 typedef map<string, string>::iterator it_type;
Jan Jongboom 0:910f5949759f 104 for(it_type it = headers.begin(); it != headers.end(); it++) {
Jan Jongboom 0:910f5949759f 105 // line is KEY: VALUE\r\n
Jan Jongboom 0:910f5949759f 106 sprintf(req, "%s: %s\r\n", it->first.c_str(), it->second.c_str());
Jan Jongboom 0:910f5949759f 107 req += it->first.length() + 1 + 1 + it->second.length() + 2;
Jan Jongboom 0:910f5949759f 108 }
Jan Jongboom 0:910f5949759f 109
Jan Jongboom 0:910f5949759f 110 sprintf(req, "\r\n");
Jan Jongboom 0:910f5949759f 111 req += 2;
Jan Jongboom 0:910f5949759f 112
Jan Jongboom 0:910f5949759f 113 if (body_size > 0) {
Jan Jongboom 10:b017c7d2cf23 114 memcpy(req, body, body_size);
Jan Jongboom 0:910f5949759f 115 }
Jan Jongboom 0:910f5949759f 116 req += body_size;
Jan Jongboom 0:910f5949759f 117
Jan Jongboom 0:910f5949759f 118 // Uncomment to debug...
Jan Jongboom 0:910f5949759f 119 // printf("----- BEGIN REQUEST -----\n");
Jan Jongboom 0:910f5949759f 120 // printf("%s", originalReq);
Jan Jongboom 0:910f5949759f 121 // printf("----- END REQUEST -----\n");
Jan Jongboom 0:910f5949759f 122
Jan Jongboom 0:910f5949759f 123 return originalReq;
Jan Jongboom 0:910f5949759f 124 }
Jan Jongboom 0:910f5949759f 125
Jan Jongboom 0:910f5949759f 126 private:
Jan Jongboom 23:15fa2726f793 127 bool has_header(const char* key, const char* value = NULL) {
Jan Jongboom 23:15fa2726f793 128 typedef map<string, string>::iterator it_type;
Jan Jongboom 23:15fa2726f793 129 for(it_type it = headers.begin(); it != headers.end(); it++) {
Jan Jongboom 23:15fa2726f793 130 if (strcmp(it->first.c_str(), key) == 0) { // key matches
Jan Jongboom 23:15fa2726f793 131 if (value == NULL || (strcmp(it->second.c_str(), value) == 0)) { // value is NULL or matches
Jan Jongboom 23:15fa2726f793 132 return true;
Jan Jongboom 23:15fa2726f793 133 }
Jan Jongboom 23:15fa2726f793 134 }
Jan Jongboom 23:15fa2726f793 135 }
Jan Jongboom 23:15fa2726f793 136
Jan Jongboom 23:15fa2726f793 137 return false;
Jan Jongboom 23:15fa2726f793 138 }
Jan Jongboom 23:15fa2726f793 139
Jan Jongboom 0:910f5949759f 140 http_method method;
Jan Jongboom 0:910f5949759f 141 ParsedUrl* parsed_url;
Jan Jongboom 0:910f5949759f 142 map<string, string> headers;
Jan Jongboom 0:910f5949759f 143 };
Jan Jongboom 0:910f5949759f 144
Jan Jongboom 0:910f5949759f 145 #endif // _MBED_HTTP_REQUEST_BUILDER_H_