diff options
Diffstat (limited to 'icnet/http/icnet_http_response.cc')
-rw-r--r-- | icnet/http/icnet_http_response.cc | 131 |
1 files changed, 131 insertions, 0 deletions
diff --git a/icnet/http/icnet_http_response.cc b/icnet/http/icnet_http_response.cc new file mode 100644 index 00000000..c9d31738 --- /dev/null +++ b/icnet/http/icnet_http_response.cc @@ -0,0 +1,131 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "icnet_http_response.h" +#include "icnet_errors.h" + +#include <algorithm> + +#include <cstring> + +namespace icnet { + +namespace http { + +HTTPResponse::HTTPResponse() { +} + +HTTPResponse::HTTPResponse(const HTTPHeaders &headers, const HTTPPayload &payload) { + headers_ = headers; + payload_ = payload; +} + +HTTPHeaders &HTTPResponse::getHeaders() { + parse(); + return headers_; +} + +HTTPPayload &HTTPResponse::getPayload() { + parse(); + return payload_; +} + +bool HTTPResponse::parseHeaders() { + const char *crlf2 = "\r\n\r\n"; + auto it = std::search(this->begin(), this->end(), crlf2, crlf2 + strlen(crlf2)); + + if (it != end()) { + std::stringstream ss; + ss.str(std::string(begin(), it)); + + std::string line; + getline(ss, line); + std::istringstream line_s(line); + std::string _http_version; + std::string http_version; + + line_s >> _http_version; + std::size_t separator; + if ((separator = _http_version.find('/')) != std::string::npos) { + if (_http_version.substr(0, separator) != "HTTP") { + return false; + } + http_version_ = line.substr(separator + 1, _http_version.length() - separator - 1); + } else { + return false; + } + + std::string status_code, status_string; + + line_s >> status_code_; + line_s >> status_string; + + auto _it = std::search(line.begin(), line.end(), status_string.begin(), status_string.end()); + + status_string_ = std::string(_it, line.end() - 1); + + std::size_t param_end; + std::size_t value_start; + while (getline(ss, line)) { + if ((param_end = line.find(':')) != std::string::npos) { + value_start = param_end + 1; + if ((value_start) < line.size()) { + if (line[value_start] == ' ') { + value_start++; + } + if (value_start < line.size()) { + headers_[line.substr(0, param_end)] = line.substr(value_start, line.size() - value_start - 1); + } + } + } else { + return false; + } + } + } + + return true; +} + +void HTTPResponse::parse() { + if (!parseHeaders()) { + throw errors::RuntimeException("Malformed HTTP response"); + } + + if (payload_.empty()) { + const char *crlf2 = "\r\n\r\n"; + auto it = std::search(this->begin(), this->end(), crlf2, crlf2 + strlen(crlf2)); + + if (it != this->end()) { + erase(begin(), it + strlen(crlf2)); + payload_ = std::move(*dynamic_cast<std::vector<uint8_t> *>(this)); + } + } +} + +std::string &HTTPResponse::getStatusCode() { + return status_code_; +} + +std::string &HTTPResponse::getStatusString() { + return status_string_; +} + +std::string &HTTPResponse::getHttpVersion() { + return http_version_; +} + +} + +}
\ No newline at end of file |