104 lines
2.7 KiB
C++
104 lines
2.7 KiB
C++
|
// src/HttpResponse.cc
|
||
|
|
// This file is part of libpbe; see http://anyterm.org/
|
||
|
|
// (C) 2005-2008 Philip Endecott
|
||
|
|
|
||
|
|
// This program is free software; you can redistribute it and/or modify
|
||
|
|
// it under the terms of the GNU General Public License as published by
|
||
|
|
// the Free Software Foundation; either version 2 of the License, or
|
||
|
|
// any later version.
|
||
|
|
//
|
||
|
|
// This program 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 General Public License for more details.
|
||
|
|
//
|
||
|
|
// You should have received a copy of the GNU General Public License
|
||
|
|
// along with this program; if not, write to the Free Software
|
||
|
|
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
||
|
|
|
||
|
|
|
||
|
|
#include "HttpResponse.hh"
|
||
|
|
|
||
|
|
#include "FileDescriptor.hh"
|
||
|
|
#include "mustfind.hh"
|
||
|
|
|
||
|
|
#include <boost/lexical_cast.hpp>
|
||
|
|
|
||
|
|
#include <sstream>
|
||
|
|
#include <functional>
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
|
||
|
|
namespace pbe {
|
||
|
|
|
||
|
|
|
||
|
|
static void send_chunked(ostream& strm, string data)
|
||
|
|
{
|
||
|
|
strm << hex << data.length() << "\r\n"
|
||
|
|
<< data << "\r\n"
|
||
|
|
<< "0\r\n"
|
||
|
|
<< "\r\n";
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void HttpResponse::send(FileDescriptor& fd)
|
||
|
|
{
|
||
|
|
ostringstream s;
|
||
|
|
s << http_version << ' ' << status_code << ' ' << reason_phrase << "\r\n";
|
||
|
|
for (map<string,string>::const_iterator i = headers.begin();
|
||
|
|
i != headers.end(); ++i) {
|
||
|
|
s << i->first << ": " << i->second << "\r\n";
|
||
|
|
}
|
||
|
|
s << "Transfer-Encoding: chunked\r\n"
|
||
|
|
<< "\r\n";
|
||
|
|
if (body.length()==0) {
|
||
|
|
s << "0\r\n\r\n";
|
||
|
|
} else {
|
||
|
|
send_chunked(s,body);
|
||
|
|
}
|
||
|
|
|
||
|
|
fd.writeall(s.str());
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static string get_line(string& s)
|
||
|
|
{
|
||
|
|
string crlf = "\r\n";
|
||
|
|
string::iterator i = mustsearch(s.begin(),s.end(),crlf.begin(),crlf.end());
|
||
|
|
string l(s.begin(),i);
|
||
|
|
s = string(i+2,s.end());
|
||
|
|
return l;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HttpResponse::HttpResponse(string s)
|
||
|
|
{
|
||
|
|
const string response_line = get_line(s);
|
||
|
|
string::const_iterator sp1 = mustfind(response_line.begin(),response_line.end(),' ');
|
||
|
|
http_version = string(response_line.begin(),sp1);
|
||
|
|
string::const_iterator sp2 = mustfind(sp1+1,response_line.end(),' ');
|
||
|
|
status_code = boost::lexical_cast<int>(string(sp1+1,sp2));
|
||
|
|
reason_phrase = string(sp2+2,response_line.end());
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
const string hdr = get_line(s);
|
||
|
|
if (hdr.empty()) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
string::const_iterator colon_pos = mustfind(hdr.begin(),hdr.end(),':');
|
||
|
|
const string hdr_name(hdr.begin(),colon_pos);
|
||
|
|
const string hdr_val(colon_pos+1,hdr.end());
|
||
|
|
const string trimmed_hdr_val(mustfind_if(hdr_val.begin(),hdr_val.end(),
|
||
|
|
bind2nd(not_equal_to<char>(),' ')),
|
||
|
|
hdr_val.end());
|
||
|
|
headers[hdr_name] = trimmed_hdr_val;
|
||
|
|
}
|
||
|
|
|
||
|
|
body = s;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
};
|
||
|
|
|