X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fclient.cpp;h=0729544a7ce0686754f721e8e6dee3e30cd7c40e;hb=d683ca0964182e9579838fec8d7d100eeabddee0;hp=a4bc342f38c2738c27047f6a80818196fbd17396;hpb=a4049d7c4126126ca3abd12b1aca8715e7006d44;p=libs%2Fnet.git

diff --git a/source/client.cpp b/source/client.cpp
index a4bc342..0729544 100644
--- a/source/client.cpp
+++ b/source/client.cpp
@@ -1,10 +1,3 @@
-/* $Id$
-
-This file is part of libmsphttp
-Copyright © 2008  Mikkosoft Productions, Mikko Rasa
-Distributed under the LGPL
-*/
-
 #include <msp/core/except.h>
 #include <msp/core/refptr.h>
 #include <msp/net/resolve.h>
@@ -15,14 +8,13 @@ Distributed under the LGPL
 
 using namespace std;
 
-#include <iostream>
-
 namespace Msp {
 namespace Http {
 
 Client::Client():
 	sock(0),
 	event_disp(0),
+	user_agent("libmsphttp/0.1"),
 	request(0),
 	response(0)
 { }
@@ -38,7 +30,7 @@ void Client::use_event_dispatcher(IO::EventDispatcher *ed)
 {
 	if(event_disp && sock)
 		event_disp->remove(*sock);
-	event_disp=ed;
+	event_disp = ed;
 	if(event_disp && sock)
 		event_disp->add(*sock);
 }
@@ -48,10 +40,13 @@ void Client::start_request(const Request &r)
 	if(request)
 		throw InvalidState("Already processing a request");
 
-	RefPtr<Net::SockAddr> addr=Net::resolve(r.get_header("host")+":"+r.get_header("x-port"));
+	string host = r.get_header("Host");
+	if(host.find(':')==string::npos)
+		host += ":80";
+	RefPtr<Net::SockAddr> addr = Net::resolve(host);
 
 	delete sock;
-	sock=new Net::StreamSocket(addr->get_family());
+	sock = new Net::StreamSocket(addr->get_family());
 	sock->set_block(false);
 
 	sock->signal_data_available.connect(sigc::mem_fun(this, &Client::data_available));
@@ -61,10 +56,12 @@ void Client::start_request(const Request &r)
 
 	sock->connect(*addr);
 
-	request=new Request(r);
+	request = new Request(r);
+	if(!user_agent.empty())
+		request->set_header("User-Agent", user_agent);
 
 	delete response;
-	response=0;
+	response = 0;
 	in_buf.clear();
 }
 
@@ -80,43 +77,42 @@ void Client::tick()
 	if(!request)
 		return;
 
-	while(IO::PollEvent ev=IO::poll(*sock, sock->get_events(), Time::zero))
+	while(IO::PollEvent ev = IO::poll(*sock, sock->get_events(), Time::zero))
 		sock->event(ev);
 
-	if(response && response->get_complete())
+	if(response && response->is_complete())
 	{
 		signal_response_complete.emit(*response);
 
 		delete sock;
-		sock=0;
+		sock = 0;
 		delete request;
-		request=0;
+		request = 0;
 	}
 }
 
 void Client::wait_response()
 {
-	while(!response || !response->get_complete())
+	while(request && (!response || !response->is_complete()))
 		tick();
 }
 
 void Client::abort()
 {
 	delete sock;
-	sock=0;
+	sock = 0;
 	delete request;
-	request=0;
+	request = 0;
 }
 
-void Client::connect_finished(int err)
+void Client::connect_finished(const exception *err)
 {
 	if(err)
 	{
 		signal_socket_error.emit(err);
 
-		sock->close();
 		delete request;
-		request=0;
+		request = 0;
 	}
 	else
 		sock->write(request->str());
@@ -125,31 +121,42 @@ void Client::connect_finished(int err)
 void Client::data_available()
 {
 	char rbuf[4096];
-	unsigned len=sock->read(rbuf, sizeof(rbuf));
+	unsigned len;
+	try
+	{
+		len = sock->read(rbuf, sizeof(rbuf));
+	}
+	catch(const exception &e)
+	{
+		signal_socket_error.emit(&e);
+		return;
+	}
+
+	if(!len)
+		return;
 	in_buf.append(rbuf, len);
 
 	if(!response)
 	{
 		if(in_buf.find("\r\n\r\n")!=string::npos || in_buf.find("\n\n")!=string::npos)
 		{
-			response=new Response(Response::parse(in_buf));
+			response = new Response(Response::parse(in_buf));
 			response->set_user_data(request->get_user_data());
-			in_buf=string();
+			in_buf = string();
 		}
 	}
 	else
 	{
-		len=response->parse_content(in_buf);
+		len = response->parse_content(in_buf);
 		in_buf.erase(0, len);
 	}
 
-	if(response && response->get_complete())
+	if(response && response->is_complete())
 	{
 		signal_response_complete.emit(*response);
 
-		sock->close();
 		delete request;
-		request=0;
+		request = 0;
 	}
 }