Kentaro Sekimoto / mbed-http

mbed-http

Fork of mbed-http by sandbox

source/http_response.h@14:3004056e4661, 2017-03-30 (annotated)

Committer:
Jan Jongboom
Date:
Thu Mar 30 15:13:37 2017 +0200
Revision:
14:3004056e4661
Parent:
9:1289162d9530
Child:
15:ffc77f212382
Request with a body, but without Content-Length header should pretend it's chunked to allocate enough memory

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_HTTP_RESPONSE
Jan Jongboom 0:910f5949759f 19 #define _MBED_HTTP_HTTP_RESPONSE
Jan Jongboom 0:910f5949759f 20 #include <string>
Jan Jongboom 0:910f5949759f 21 #include <vector>
Jan Jongboom 0:910f5949759f 22 #include "http_parser.h"
Jan Jongboom 0:910f5949759f 23
Jan Jongboom 0:910f5949759f 24 using namespace std;
Jan Jongboom 0:910f5949759f 25
Jan Jongboom 0:910f5949759f 26 class HttpResponse {
Jan Jongboom 0:910f5949759f 27 public:
Jan Jongboom 0:910f5949759f 28 HttpResponse() {
Jan Jongboom 0:910f5949759f 29 status_code = 0;
Jan Jongboom 0:910f5949759f 30 concat_header_field = false;
Jan Jongboom 0:910f5949759f 31 concat_header_value = false;
Jan Jongboom 3:8a6b003e3874 32 expected_content_length = 0;
Jan Jongboom 7:2e3eedb9ca5c 33 is_chunked = false;
Jan Jongboom 7:2e3eedb9ca5c 34 is_message_completed = false;
Jan Jongboom 3:8a6b003e3874 35 body_length = 0;
Jan Jongboom 4:539df159e058 36 body_offset = 0;
Jan Jongboom 4:539df159e058 37 body = NULL;
Jan Jongboom 4:539df159e058 38 }
Jan Jongboom 4:539df159e058 39
Jan Jongboom 4:539df159e058 40 ~HttpResponse() {
Jan Jongboom 4:539df159e058 41 if (body != NULL) {
Jan Jongboom 4:539df159e058 42 free(body);
Jan Jongboom 4:539df159e058 43 }
Jan Jongboom 7:2e3eedb9ca5c 44
Jan Jongboom 7:2e3eedb9ca5c 45 for (size_t ix = 0; ix < header_fields.size(); ix++) {
Jan Jongboom 7:2e3eedb9ca5c 46 delete header_fields[ix];
Jan Jongboom 7:2e3eedb9ca5c 47 delete header_values[ix];
Jan Jongboom 7:2e3eedb9ca5c 48 }
Jan Jongboom 0:910f5949759f 49 }
Jan Jongboom 0:910f5949759f 50
Jan Jongboom 0:910f5949759f 51 void set_status(int a_status_code, string a_status_message) {
Jan Jongboom 0:910f5949759f 52 status_code = a_status_code;
Jan Jongboom 0:910f5949759f 53 status_message = a_status_message;
Jan Jongboom 0:910f5949759f 54 }
Jan Jongboom 0:910f5949759f 55
Jan Jongboom 0:910f5949759f 56 int get_status_code() {
Jan Jongboom 0:910f5949759f 57 return status_code;
Jan Jongboom 0:910f5949759f 58 }
Jan Jongboom 0:910f5949759f 59
Jan Jongboom 0:910f5949759f 60 string get_status_message() {
Jan Jongboom 0:910f5949759f 61 return status_message;
Jan Jongboom 0:910f5949759f 62 }
Jan Jongboom 0:910f5949759f 63
Jan Jongboom 0:910f5949759f 64 void set_header_field(string field) {
Jan Jongboom 0:910f5949759f 65 concat_header_value = false;
Jan Jongboom 0:910f5949759f 66
Jan Jongboom 0:910f5949759f 67 // headers can be chunked
Jan Jongboom 0:910f5949759f 68 if (concat_header_field) {
Jan Jongboom 7:2e3eedb9ca5c 69 *header_fields[header_fields.size() - 1] = (*header_fields[header_fields.size() - 1]) + field;
Jan Jongboom 0:910f5949759f 70 }
Jan Jongboom 0:910f5949759f 71 else {
Jan Jongboom 7:2e3eedb9ca5c 72 header_fields.push_back(new string(field));
Jan Jongboom 0:910f5949759f 73 }
Jan Jongboom 0:910f5949759f 74
Jan Jongboom 0:910f5949759f 75 concat_header_field = true;
Jan Jongboom 0:910f5949759f 76 }
Jan Jongboom 0:910f5949759f 77
Jan Jongboom 0:910f5949759f 78 void set_header_value(string value) {
Jan Jongboom 0:910f5949759f 79 concat_header_field = false;
Jan Jongboom 0:910f5949759f 80
Jan Jongboom 0:910f5949759f 81 // headers can be chunked
Jan Jongboom 0:910f5949759f 82 if (concat_header_value) {
Jan Jongboom 7:2e3eedb9ca5c 83 *header_values[header_values.size() - 1] = (*header_values[header_values.size() - 1]) + value;
Jan Jongboom 0:910f5949759f 84 }
Jan Jongboom 0:910f5949759f 85 else {
Jan Jongboom 7:2e3eedb9ca5c 86 header_values.push_back(new string(value));
Jan Jongboom 0:910f5949759f 87 }
Jan Jongboom 0:910f5949759f 88
Jan Jongboom 0:910f5949759f 89 concat_header_value = true;
Jan Jongboom 0:910f5949759f 90 }
Jan Jongboom 0:910f5949759f 91
Jan Jongboom 3:8a6b003e3874 92 void set_headers_complete() {
Jan Jongboom 3:8a6b003e3874 93 for (size_t ix = 0; ix < header_fields.size(); ix++) {
Jan Jongboom 7:2e3eedb9ca5c 94 if (strcicmp(header_fields[ix]->c_str(), "content-length") == 0) {
Jan Jongboom 7:2e3eedb9ca5c 95 expected_content_length = (size_t)atoi(header_values[ix]->c_str());
Jan Jongboom 3:8a6b003e3874 96 break;
Jan Jongboom 3:8a6b003e3874 97 }
Jan Jongboom 3:8a6b003e3874 98 }
Jan Jongboom 3:8a6b003e3874 99 }
Jan Jongboom 3:8a6b003e3874 100
Jan Jongboom 0:910f5949759f 101 size_t get_headers_length() {
Jan Jongboom 0:910f5949759f 102 return header_fields.size();
Jan Jongboom 0:910f5949759f 103 }
Jan Jongboom 0:910f5949759f 104
Jan Jongboom 7:2e3eedb9ca5c 105 vector<string*> get_headers_fields() {
Jan Jongboom 0:910f5949759f 106 return header_fields;
Jan Jongboom 0:910f5949759f 107 }
Jan Jongboom 0:910f5949759f 108
Jan Jongboom 7:2e3eedb9ca5c 109 vector<string*> get_headers_values() {
Jan Jongboom 0:910f5949759f 110 return header_values;
Jan Jongboom 0:910f5949759f 111 }
Jan Jongboom 0:910f5949759f 112
Jan Jongboom 4:539df159e058 113 void set_body(const char *at, size_t length) {
Jan Jongboom 14:3004056e4661 114 // Connection: close, could not specify Content-Length, nor chunked... So do it like this:
Jan Jongboom 14:3004056e4661 115 if (expected_content_length == 0 && length > 0) {
Jan Jongboom 14:3004056e4661 116 is_chunked = true;
Jan Jongboom 14:3004056e4661 117 }
Jan Jongboom 14:3004056e4661 118
Jan Jongboom 4:539df159e058 119 // only malloc when this fn is called, so we don't alloc when body callback's are enabled
Jan Jongboom 7:2e3eedb9ca5c 120 if (body == NULL && !is_chunked) {
Jan Jongboom 4:539df159e058 121 body = (char*)malloc(expected_content_length);
Jan Jongboom 4:539df159e058 122 }
Jan Jongboom 4:539df159e058 123
Jan Jongboom 7:2e3eedb9ca5c 124 if (is_chunked) {
Jan Jongboom 7:2e3eedb9ca5c 125 if (body == NULL) {
Jan Jongboom 7:2e3eedb9ca5c 126 body = (char*)malloc(length);
Jan Jongboom 7:2e3eedb9ca5c 127 }
Jan Jongboom 7:2e3eedb9ca5c 128 else {
Jan Jongboom 7:2e3eedb9ca5c 129 body = (char*)realloc(body, body_offset + length);
Jan Jongboom 7:2e3eedb9ca5c 130 }
Jan Jongboom 7:2e3eedb9ca5c 131 }
Jan Jongboom 7:2e3eedb9ca5c 132
Jan Jongboom 4:539df159e058 133 memcpy(body + body_offset, at, length);
Jan Jongboom 4:539df159e058 134
Jan Jongboom 4:539df159e058 135 body_offset += length;
Jan Jongboom 0:910f5949759f 136 }
Jan Jongboom 0:910f5949759f 137
Jan Jongboom 4:539df159e058 138 void* get_body() {
Jan Jongboom 4:539df159e058 139 return (void*)body;
Jan Jongboom 4:539df159e058 140 }
Jan Jongboom 4:539df159e058 141
Jan Jongboom 4:539df159e058 142 string get_body_as_string() {
Jan Jongboom 4:539df159e058 143 string s(body, body_offset);
Jan Jongboom 4:539df159e058 144 return s;
Jan Jongboom 0:910f5949759f 145 }
Jan Jongboom 0:910f5949759f 146
Jan Jongboom 3:8a6b003e3874 147 void increase_body_length(size_t length) {
Jan Jongboom 3:8a6b003e3874 148 body_length += length;
Jan Jongboom 3:8a6b003e3874 149 }
Jan Jongboom 3:8a6b003e3874 150
Jan Jongboom 4:539df159e058 151 size_t get_body_length() {
Jan Jongboom 4:539df159e058 152 return body_offset;
Jan Jongboom 4:539df159e058 153 }
Jan Jongboom 4:539df159e058 154
Jan Jongboom 7:2e3eedb9ca5c 155 bool is_message_complete() {
Jan Jongboom 7:2e3eedb9ca5c 156 return is_message_completed;
Jan Jongboom 7:2e3eedb9ca5c 157 }
Jan Jongboom 7:2e3eedb9ca5c 158
Jan Jongboom 7:2e3eedb9ca5c 159 void set_chunked() {
Jan Jongboom 7:2e3eedb9ca5c 160 is_chunked = true;
Jan Jongboom 7:2e3eedb9ca5c 161 }
Jan Jongboom 7:2e3eedb9ca5c 162
Jan Jongboom 7:2e3eedb9ca5c 163 void set_message_complete() {
Jan Jongboom 7:2e3eedb9ca5c 164 is_message_completed = true;
Jan Jongboom 3:8a6b003e3874 165 }
Jan Jongboom 3:8a6b003e3874 166
Jan Jongboom 0:910f5949759f 167 private:
Jan Jongboom 3:8a6b003e3874 168 // from http://stackoverflow.com/questions/5820810/case-insensitive-string-comp-in-c
Jan Jongboom 3:8a6b003e3874 169 int strcicmp(char const *a, char const *b) {
Jan Jongboom 3:8a6b003e3874 170 for (;; a++, b++) {
Jan Jongboom 3:8a6b003e3874 171 int d = tolower(*a) - tolower(*b);
Jan Jongboom 3:8a6b003e3874 172 if (d != 0 || !*a) {
Jan Jongboom 3:8a6b003e3874 173 return d;
Jan Jongboom 3:8a6b003e3874 174 }
Jan Jongboom 3:8a6b003e3874 175 }
Jan Jongboom 3:8a6b003e3874 176 }
Jan Jongboom 3:8a6b003e3874 177
Jan Jongboom 8:6156404278bb 178 char tolower(char c) {
Jan Jongboom 8:6156404278bb 179 if(('A' <= c) && (c <= 'Z')) {
Jan Jongboom 8:6156404278bb 180 return 'a' + (c - 'A');
Jan Jongboom 8:6156404278bb 181 }
Jan Jongboom 8:6156404278bb 182
Jan Jongboom 8:6156404278bb 183 return c;
Jan Jongboom 8:6156404278bb 184 }
Jan Jongboom 8:6156404278bb 185
Jan Jongboom 0:910f5949759f 186 int status_code;
Jan Jongboom 0:910f5949759f 187 string status_message;
Jan Jongboom 0:910f5949759f 188
Jan Jongboom 7:2e3eedb9ca5c 189 vector<string*> header_fields;
Jan Jongboom 7:2e3eedb9ca5c 190 vector<string*> header_values;
Jan Jongboom 0:910f5949759f 191
Jan Jongboom 0:910f5949759f 192 bool concat_header_field;
Jan Jongboom 0:910f5949759f 193 bool concat_header_value;
Jan Jongboom 0:910f5949759f 194
Jan Jongboom 3:8a6b003e3874 195 size_t expected_content_length;
Jan Jongboom 3:8a6b003e3874 196
Jan Jongboom 7:2e3eedb9ca5c 197 bool is_chunked;
Jan Jongboom 7:2e3eedb9ca5c 198
Jan Jongboom 7:2e3eedb9ca5c 199 bool is_message_completed;
Jan Jongboom 7:2e3eedb9ca5c 200
Jan Jongboom 4:539df159e058 201 char * body;
Jan Jongboom 3:8a6b003e3874 202 size_t body_length;
Jan Jongboom 4:539df159e058 203 size_t body_offset;
Jan Jongboom 0:910f5949759f 204 };
Jan Jongboom 0:910f5949759f 205 #endif