3 This file is part of libmsphttp
4 Copyright © 2008 Mikkosoft Productions, Mikko Rasa
5 Distributed under the LGPL
8 #include <msp/core/except.h>
9 #include <msp/core/refptr.h>
10 #include <msp/net/resolve.h>
11 #include <msp/time/units.h>
24 user_agent("libmsphttp/0.1"),
36 void Client::use_event_dispatcher(IO::EventDispatcher *ed)
38 if(event_disp && sock)
39 event_disp->remove(*sock);
41 if(event_disp && sock)
42 event_disp->add(*sock);
45 void Client::start_request(const Request &r)
48 throw InvalidState("Already processing a request");
50 string host = r.get_header("Host");
51 if(host.find(':')==string::npos)
53 RefPtr<Net::SockAddr> addr = Net::resolve(host);
56 sock = new Net::StreamSocket(addr->get_family());
57 sock->set_block(false);
59 sock->signal_data_available.connect(sigc::mem_fun(this, &Client::data_available));
60 sock->signal_connect_finished.connect(sigc::mem_fun(this, &Client::connect_finished));
62 event_disp->add(*sock);
66 request = new Request(r);
67 if(!user_agent.empty())
68 request->set_header("User-Agent", user_agent);
75 const Response *Client::get_url(const std::string &url)
77 start_request(Request::from_url(url));
87 while(IO::PollEvent ev = IO::poll(*sock, sock->get_events(), Time::zero))
90 if(response && response->is_complete())
92 signal_response_complete.emit(*response);
101 void Client::wait_response()
103 while(request && (!response || !response->is_complete()))
115 void Client::connect_finished(int err)
119 signal_socket_error.emit(err);
126 sock->write(request->str());
129 void Client::data_available()
132 unsigned len = sock->read(rbuf, sizeof(rbuf));
133 in_buf.append(rbuf, len);
137 if(in_buf.find("\r\n\r\n")!=string::npos || in_buf.find("\n\n")!=string::npos)
139 response = new Response(Response::parse(in_buf));
140 response->set_user_data(request->get_user_data());
146 len = response->parse_content(in_buf);
147 in_buf.erase(0, len);
150 if(response && response->is_complete())
152 signal_response_complete.emit(*response);