Matthew Else / mbed-http

Fork of mbed-http

Fork of mbed-http by sandbox

source/http_response.h@7:2e3eedb9ca5c, 2017-02-24 (annotated)

Committer:
Jan Jongboom
Date:
Fri Feb 24 11:20:16 2017 +0100
Revision:
7:2e3eedb9ca5c
Parent:
4:539df159e058
Child:
8:6156404278bb
Add support for chunked encoding. Fix bug handle querystrings. HttpResponse should own header strings.

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 4:539df159e058 114 // only malloc when this fn is called, so we don't alloc when body callback's are enabled
Jan Jongboom 7:2e3eedb9ca5c 115 if (body == NULL && !is_chunked) {
Jan Jongboom 4:539df159e058 116 body = (char*)malloc(expected_content_length);
Jan Jongboom 4:539df159e058 117 }
Jan Jongboom 4:539df159e058 118
Jan Jongboom 7:2e3eedb9ca5c 119 if (is_chunked) {
Jan Jongboom 7:2e3eedb9ca5c 120 if (body == NULL) {
Jan Jongboom 7:2e3eedb9ca5c 121 body = (char*)malloc(length);
Jan Jongboom 7:2e3eedb9ca5c 122 }
Jan Jongboom 7:2e3eedb9ca5c 123 else {
Jan Jongboom 7:2e3eedb9ca5c 124 body = (char*)realloc(body, body_offset + length);
Jan Jongboom 7:2e3eedb9ca5c 125 }
Jan Jongboom 7:2e3eedb9ca5c 126 }
Jan Jongboom 7:2e3eedb9ca5c 127
Jan Jongboom 4:539df159e058 128 memcpy(body + body_offset, at, length);
Jan Jongboom 4:539df159e058 129
Jan Jongboom 4:539df159e058 130 body_offset += length;
Jan Jongboom 0:910f5949759f 131 }
Jan Jongboom 0:910f5949759f 132
Jan Jongboom 4:539df159e058 133 void* get_body() {
Jan Jongboom 4:539df159e058 134 return (void*)body;
Jan Jongboom 4:539df159e058 135 }
Jan Jongboom 4:539df159e058 136
Jan Jongboom 4:539df159e058 137 string get_body_as_string() {
Jan Jongboom 4:539df159e058 138 string s(body, body_offset);
Jan Jongboom 4:539df159e058 139 return s;
Jan Jongboom 0:910f5949759f 140 }
Jan Jongboom 0:910f5949759f 141
Jan Jongboom 3:8a6b003e3874 142 void increase_body_length(size_t length) {
Jan Jongboom 3:8a6b003e3874 143 body_length += length;
Jan Jongboom 3:8a6b003e3874 144 }
Jan Jongboom 3:8a6b003e3874 145
Jan Jongboom 4:539df159e058 146 size_t get_body_length() {
Jan Jongboom 4:539df159e058 147 return body_offset;
Jan Jongboom 4:539df159e058 148 }
Jan Jongboom 4:539df159e058 149
Jan Jongboom 7:2e3eedb9ca5c 150 bool is_message_complete() {
Jan Jongboom 7:2e3eedb9ca5c 151 return is_message_completed;
Jan Jongboom 7:2e3eedb9ca5c 152 }
Jan Jongboom 7:2e3eedb9ca5c 153
Jan Jongboom 7:2e3eedb9ca5c 154 void set_chunked() {
Jan Jongboom 7:2e3eedb9ca5c 155 is_chunked = true;
Jan Jongboom 7:2e3eedb9ca5c 156 }
Jan Jongboom 7:2e3eedb9ca5c 157
Jan Jongboom 7:2e3eedb9ca5c 158 void set_message_complete() {
Jan Jongboom 7:2e3eedb9ca5c 159 is_message_completed = true;
Jan Jongboom 3:8a6b003e3874 160 }
Jan Jongboom 3:8a6b003e3874 161
Jan Jongboom 0:910f5949759f 162 private:
Jan Jongboom 3:8a6b003e3874 163 // from http://stackoverflow.com/questions/5820810/case-insensitive-string-comp-in-c
Jan Jongboom 3:8a6b003e3874 164 int strcicmp(char const *a, char const *b) {
Jan Jongboom 3:8a6b003e3874 165 for (;; a++, b++) {
Jan Jongboom 3:8a6b003e3874 166 int d = tolower(*a) - tolower(*b);
Jan Jongboom 3:8a6b003e3874 167 if (d != 0 || !*a) {
Jan Jongboom 3:8a6b003e3874 168 return d;
Jan Jongboom 3:8a6b003e3874 169 }
Jan Jongboom 3:8a6b003e3874 170 }
Jan Jongboom 3:8a6b003e3874 171 }
Jan Jongboom 3:8a6b003e3874 172
Jan Jongboom 0:910f5949759f 173 int status_code;
Jan Jongboom 0:910f5949759f 174 string status_message;
Jan Jongboom 0:910f5949759f 175
Jan Jongboom 7:2e3eedb9ca5c 176 vector<string*> header_fields;
Jan Jongboom 7:2e3eedb9ca5c 177 vector<string*> header_values;
Jan Jongboom 0:910f5949759f 178
Jan Jongboom 0:910f5949759f 179 bool concat_header_field;
Jan Jongboom 0:910f5949759f 180 bool concat_header_value;
Jan Jongboom 0:910f5949759f 181
Jan Jongboom 3:8a6b003e3874 182 size_t expected_content_length;
Jan Jongboom 3:8a6b003e3874 183
Jan Jongboom 7:2e3eedb9ca5c 184 bool is_chunked;
Jan Jongboom 7:2e3eedb9ca5c 185
Jan Jongboom 7:2e3eedb9ca5c 186 bool is_message_completed;
Jan Jongboom 7:2e3eedb9ca5c 187
Jan Jongboom 4:539df159e058 188 char * body;
Jan Jongboom 3:8a6b003e3874 189 size_t body_length;
Jan Jongboom 4:539df159e058 190 size_t body_offset;
Jan Jongboom 0:910f5949759f 191 };
Jan Jongboom 0:910f5949759f 192 #endif