blob: dbe81452fe10575642f0a7137d9d861a1efd5478 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
#include "http.h"
#include <boost/format.hpp>
#include <boost/algorithm/string.hpp>
#include <boost/algorithm/string/regex.hpp>
#include <vector>
HTTPRequest::HTTPRequest(std::istream& is) {
std::string firstline;
std::getline(is, firstline);
std::vector<std::string> splitvec;
boost::algorithm::split(splitvec, firstline, boost::algorithm::is_space());
type = splitvec[0];
path = splitvec[1];
httpver = splitvec[2];
std::cout << boost::format("%s %s %s\n") % type % path % httpver;
while(is.good()) {
std::string line;
std::getline(is, line);
boost::trim(line);
if(!line.size()) continue;
std::vector<std::string> v;
boost::algorithm::split_regex(v, line, boost::regex(": "));
headers[v[0]] = v[1];
}
}
HTTPResponse::HTTPResponse(boost::asio::ip::tcp::socket& socket_) : socket(socket_){
httpver = "1.1";
headers_written = false;
}
void HTTPResponse::add_header(std::string key, std::string value) {
headers[key] = value;
}
void HTTPResponse::write_headers() {
write(boost::str(boost::format("HTTP/%s %d %s\r\n") % httpver % code % status));
for(HTTPHeaders::iterator it = headers.begin(); it != headers.end(); it++) {
write(boost::str(boost::format("%s: %s\r\n") % it->first % it->second));
}
write("\r\n");
}
void HTTPResponse::write(char *data, unsigned int len) {
write(std::string(data, len));
}
void HTTPResponse::write(std::string str) {
if(!headers_written) {
// make sure to set headers_written before calling write_headers
headers_written = true;
write_headers();
}
boost::asio::streambuf b;
std::ostream os(&b);
os << str;
boost::asio::write(socket, b);
}
|