Fork of SandBox's original mbed-http (https://os.mbed.com/teams/sandbox/code/mbed-http/) and update for MbedOS6+ Content of TESTS folder was replaced with basic examples form original SandBox's HelloWorld
source/http_request_builder.h@35:b3ee394d1d2e, 2019-08-09 (annotated)
- Committer:
- Jan Jongboom
- Date:
- Fri Aug 09 09:58:57 2019 +0200
- Revision:
- 35:b3ee394d1d2e
- Parent:
- 32:fa4d71265625
Add ws/wss schema in the request builder
Who changed what in which revision?
User | Revision | Line number | New 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 |
35:b3ee394d1d2e | 42 | else if (strcmp(parsed_url->schema(), "ws") == 0 && parsed_url->port() != 80) { |
Jan Jongboom |
35:b3ee394d1d2e | 43 | host += string(port_str); |
Jan Jongboom |
35:b3ee394d1d2e | 44 | } |
Jan Jongboom |
35:b3ee394d1d2e | 45 | else if (strcmp(parsed_url->schema(), "wss") == 0 && parsed_url->port() != 443) { |
Jan Jongboom |
35:b3ee394d1d2e | 46 | host += string(port_str); |
Jan Jongboom |
35:b3ee394d1d2e | 47 | } |
Jan Jongboom |
21:fcd2bfd31a39 | 48 | |
Jan Jongboom |
21:fcd2bfd31a39 | 49 | set_header("Host", host); |
Jan Jongboom |
0:910f5949759f | 50 | } |
Jan Jongboom |
0:910f5949759f | 51 | |
Jan Jongboom |
0:910f5949759f | 52 | /** |
Jan Jongboom |
0:910f5949759f | 53 | * Set a header for the request |
Jan Jongboom |
0:910f5949759f | 54 | * If the key already exists, it will be overwritten... |
Jan Jongboom |
0:910f5949759f | 55 | */ |
Jan Jongboom |
0:910f5949759f | 56 | void set_header(string key, string value) { |
Jan Jongboom |
0:910f5949759f | 57 | map<string, string>::iterator it = headers.find(key); |
Jan Jongboom |
0:910f5949759f | 58 | |
Jan Jongboom |
0:910f5949759f | 59 | if (it != headers.end()) { |
Jan Jongboom |
0:910f5949759f | 60 | it->second = value; |
Jan Jongboom |
0:910f5949759f | 61 | } |
Jan Jongboom |
0:910f5949759f | 62 | else { |
Jan Jongboom |
0:910f5949759f | 63 | headers.insert(headers.end(), pair<string, string>(key, value)); |
Jan Jongboom |
0:910f5949759f | 64 | } |
Jan Jongboom |
0:910f5949759f | 65 | } |
Jan Jongboom |
0:910f5949759f | 66 | |
Jan Jongboom |
31:b3730a2c4f39 | 67 | char* build(const void* body, uint32_t body_size, uint32_t &size, bool skip_content_length = false) { |
Jan Jongboom |
0:910f5949759f | 68 | const char* method_str = http_method_str(method); |
Jan Jongboom |
0:910f5949759f | 69 | |
Jan Jongboom |
23:15fa2726f793 | 70 | bool is_chunked = has_header("Transfer-Encoding", "chunked"); |
Jan Jongboom |
23:15fa2726f793 | 71 | |
Jan Jongboom |
23:15fa2726f793 | 72 | if (!is_chunked && (method == HTTP_POST || method == HTTP_PUT || method == HTTP_DELETE || body_size > 0)) { |
Jan Jongboom |
0:910f5949759f | 73 | char buffer[10]; |
Jan Jongboom |
31:b3730a2c4f39 | 74 | snprintf(buffer, 10, "%lu", body_size); |
Jan Jongboom |
0:910f5949759f | 75 | set_header("Content-Length", string(buffer)); |
Jan Jongboom |
0:910f5949759f | 76 | } |
Jan Jongboom |
0:910f5949759f | 77 | |
Jan Jongboom |
10:b017c7d2cf23 | 78 | size = 0; |
Jan Jongboom |
0:910f5949759f | 79 | |
Jan Jongboom |
0:910f5949759f | 80 | // first line is METHOD PATH+QUERY HTTP/1.1\r\n |
Jan Jongboom |
10:b017c7d2cf23 | 81 | 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 | 82 | |
Jan Jongboom |
0:910f5949759f | 83 | // after that we'll do the headers |
Jan Jongboom |
0:910f5949759f | 84 | typedef map<string, string>::iterator it_type; |
Jan Jongboom |
0:910f5949759f | 85 | for(it_type it = headers.begin(); it != headers.end(); it++) { |
Jan Jongboom |
0:910f5949759f | 86 | // line is KEY: VALUE\r\n |
Jan Jongboom |
0:910f5949759f | 87 | size += it->first.length() + 1 + 1 + it->second.length() + 2; |
Jan Jongboom |
0:910f5949759f | 88 | } |
Jan Jongboom |
0:910f5949759f | 89 | |
Jan Jongboom |
0:910f5949759f | 90 | // then the body, first an extra newline |
Jan Jongboom |
0:910f5949759f | 91 | size += 2; |
Jan Jongboom |
0:910f5949759f | 92 | |
Jan Jongboom |
23:15fa2726f793 | 93 | if (!is_chunked) { |
Jan Jongboom |
23:15fa2726f793 | 94 | // body |
Jan Jongboom |
23:15fa2726f793 | 95 | size += body_size; |
Jan Jongboom |
23:15fa2726f793 | 96 | } |
Jan Jongboom |
0:910f5949759f | 97 | |
Jan Jongboom |
0:910f5949759f | 98 | // Now let's print it |
Jan Jongboom |
0:910f5949759f | 99 | char* req = (char*)calloc(size + 1, 1); |
Jan Jongboom |
0:910f5949759f | 100 | char* originalReq = req; |
Jan Jongboom |
0:910f5949759f | 101 | |
Jan Jongboom |
10:b017c7d2cf23 | 102 | if (strlen(parsed_url->query())) { |
Jan Jongboom |
10:b017c7d2cf23 | 103 | sprintf(req, "%s %s?%s HTTP/1.1\r\n", method_str, parsed_url->path(), parsed_url->query()); |
Jan Jongboom |
10:b017c7d2cf23 | 104 | } else { |
Jan Jongboom |
10:b017c7d2cf23 | 105 | sprintf(req, "%s %s%s HTTP/1.1\r\n", method_str, parsed_url->path(), parsed_url->query()); |
Jan Jongboom |
7:2e3eedb9ca5c | 106 | } |
Jan Jongboom |
10:b017c7d2cf23 | 107 | 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 | 108 | |
Jan Jongboom |
0:910f5949759f | 109 | typedef map<string, string>::iterator it_type; |
Jan Jongboom |
0:910f5949759f | 110 | for(it_type it = headers.begin(); it != headers.end(); it++) { |
Jan Jongboom |
0:910f5949759f | 111 | // line is KEY: VALUE\r\n |
Jan Jongboom |
0:910f5949759f | 112 | sprintf(req, "%s: %s\r\n", it->first.c_str(), it->second.c_str()); |
Jan Jongboom |
0:910f5949759f | 113 | req += it->first.length() + 1 + 1 + it->second.length() + 2; |
Jan Jongboom |
0:910f5949759f | 114 | } |
Jan Jongboom |
0:910f5949759f | 115 | |
Jan Jongboom |
0:910f5949759f | 116 | sprintf(req, "\r\n"); |
Jan Jongboom |
0:910f5949759f | 117 | req += 2; |
Jan Jongboom |
0:910f5949759f | 118 | |
Jan Jongboom |
0:910f5949759f | 119 | if (body_size > 0) { |
Jan Jongboom |
10:b017c7d2cf23 | 120 | memcpy(req, body, body_size); |
Jan Jongboom |
0:910f5949759f | 121 | } |
Jan Jongboom |
0:910f5949759f | 122 | req += body_size; |
Jan Jongboom |
0:910f5949759f | 123 | |
Jan Jongboom |
0:910f5949759f | 124 | // Uncomment to debug... |
Jan Jongboom |
0:910f5949759f | 125 | // printf("----- BEGIN REQUEST -----\n"); |
Jan Jongboom |
0:910f5949759f | 126 | // printf("%s", originalReq); |
Jan Jongboom |
0:910f5949759f | 127 | // printf("----- END REQUEST -----\n"); |
Jan Jongboom |
0:910f5949759f | 128 | |
Jan Jongboom |
0:910f5949759f | 129 | return originalReq; |
Jan Jongboom |
0:910f5949759f | 130 | } |
Jan Jongboom |
0:910f5949759f | 131 | |
Jan Jongboom |
0:910f5949759f | 132 | private: |
Jan Jongboom |
23:15fa2726f793 | 133 | bool has_header(const char* key, const char* value = NULL) { |
Jan Jongboom |
23:15fa2726f793 | 134 | typedef map<string, string>::iterator it_type; |
Jan Jongboom |
23:15fa2726f793 | 135 | for(it_type it = headers.begin(); it != headers.end(); it++) { |
Jan Jongboom |
23:15fa2726f793 | 136 | if (strcmp(it->first.c_str(), key) == 0) { // key matches |
Jan Jongboom |
23:15fa2726f793 | 137 | if (value == NULL || (strcmp(it->second.c_str(), value) == 0)) { // value is NULL or matches |
Jan Jongboom |
23:15fa2726f793 | 138 | return true; |
Jan Jongboom |
23:15fa2726f793 | 139 | } |
Jan Jongboom |
23:15fa2726f793 | 140 | } |
Jan Jongboom |
23:15fa2726f793 | 141 | } |
Jan Jongboom |
23:15fa2726f793 | 142 | |
Jan Jongboom |
23:15fa2726f793 | 143 | return false; |
Jan Jongboom |
23:15fa2726f793 | 144 | } |
Jan Jongboom |
23:15fa2726f793 | 145 | |
Jan Jongboom |
0:910f5949759f | 146 | http_method method; |
Jan Jongboom |
0:910f5949759f | 147 | ParsedUrl* parsed_url; |
Jan Jongboom |
0:910f5949759f | 148 | map<string, string> headers; |
Jan Jongboom |
0:910f5949759f | 149 | }; |
Jan Jongboom |
0:910f5949759f | 150 | |
Jan Jongboom |
0:910f5949759f | 151 | #endif // _MBED_HTTP_REQUEST_BUILDER_H_ |