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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
#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];
url_decode(path);
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];
}
}
void HTTPRequest::url_decode(std::string& str) {
std::string::const_iterator start, end, prev;
start = str.begin();
end = str.end();
boost::match_results<std::string::const_iterator> what;
boost::match_flag_type flags = boost::match_default;
boost::regex re("%([0-9a-fA-F]{2})");
prev = start;
std::string out;
while(boost::regex_search(start, end, what, re, flags)) {
std::istringstream hs(what[1]);
char c;
int hi;
hs >> std::hex >> hi;
c = hi;
out += std::string(prev, what[1].first - 1) + c;
start = what[1].second;
prev = start;
}
out += std::string(prev, end);
str = out;
}
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);
}
|
