danicoin/src/HTTP/HttpResponse.cpp

84 lines
2.1 KiB
C++
Raw Normal View History

2014-09-15 12:46:31 +00:00
// Copyright (c) 2012-2014, The CryptoNote developers, The Bytecoin developers
//
// This file is part of Bytecoin.
//
// Bytecoin is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Bytecoin is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with Bytecoin. If not, see <http://www.gnu.org/licenses/>.
#include "HttpResponse.h"
#include <stdexcept>
namespace {
const char* getStatusString(cryptonote::HttpResponse::HTTP_STATUS status) {
switch (status) {
case cryptonote::HttpResponse::STATUS_200:
return "200 OK";
case cryptonote::HttpResponse::STATUS_404:
return "404 Not Found";
case cryptonote::HttpResponse::STATUS_500:
return "500 Internal Server Error";
default:
throw std::runtime_error("Unknown HTTP status code is given");
}
return ""; //unaccessible
}
} //namespace
namespace cryptonote {
HttpResponse::HttpResponse() {
status = STATUS_200;
headers["Server"] = "Cryptonote-based HTTP server";
}
void HttpResponse::setStatus(HTTP_STATUS s) {
status = s;
}
void HttpResponse::addHeader(const std::string& name, const std::string& value) {
headers[name] = value;
}
void HttpResponse::setBody(const std::string& b) {
body = b;
if (!body.empty()) {
headers["Content-Length"] = std::to_string(body.size());
} else {
headers.erase("Content-Length");
}
}
std::ostream& HttpResponse::printHttpResponse(std::ostream& os) const {
os << "HTTP/1.1 " << getStatusString(status) << "\r\n";
for (auto pair: headers) {
os << pair.first << ": " << pair.second << "\r\n";
}
os << "\r\n";
if (!body.empty()) {
os << body;
}
return os;
}
} //namespace cryptonote