WebSVN – HomeAutomation – Diff – /trunk/PcSoftware/Atom1.5/src/net/UdpClient.cpp


/*
 *
 *    Copyright (C) 2010  Mattias Runge
 *
 *    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
 *    (at your option) 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.,
 *    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */
 
#include "UdpClient.h"
 
#include <boost/lexical_cast.hpp>
 
namespace atom {
namespace net {
 
UdpClient::UdpClient(boost::asio::io_service& io_service, ClientId id, ServerId server_id) : io_service_(io_service), Client(io_service, id, server_id)
{
}
 
UdpClient::~UdpClient()
{
    if (this->socket_.use_count() != 0)
    {
        this->socket_->close();
        this->socket_.reset();
    }
}
 
void UdpClient::Connect(std::string address, unsigned int port)
{
    try
    {
        boost::asio::ip::udp::resolver resolver(this->io_service_);
        boost::asio::ip::udp::resolver::query query(boost::asio::ip::udp::v4(), address, boost::lexical_cast<std::string>(port));
        boost::asio::ip::udp::resolver::iterator it = resolver.resolve(query);
       
        this->endpoint_ = *it;
       
        this->socket_.reset(new boost::asio::ip::udp::socket(this->io_service_, boost::asio::ip::udp::endpoint(boost::asio::ip::udp::v4(), port)));
    }
    catch (std::exception e)
    {
        throw std::runtime_error("Error while connecting to " + address + ":" + boost::lexical_cast<std::string>(port) + ", " + e.what());
    }
   
    this->Read();
}
 
void UdpClient::Disconnect()
{
    if (this->socket_.use_count() != 0)
    {
        this->socket_->close();
        this->socket_.reset();
    }
   
    Client::Disconnect();
}
 
void UdpClient::Send(Buffer data)
{
    if (this->socket_->is_open())
    {
        this->socket_->send_to(boost::asio::buffer(data), this->endpoint_);
    }
    else
    {
        this->Disconnect();
    }
}
 
void UdpClient::Read()
{
    Client::Read();
   
    this->socket_->async_receive(boost::asio::buffer(this->buffer_),
                                 boost::bind(&UdpClient::ReadHandler,
                                             this,
                                             boost::asio::placeholders::error,
                                             boost::asio::placeholders::bytes_transferred));
}
 
 
}; // namespace net
}; // namespace atom
 

Rev 1592	Rev 1593
1	/*	1	/*
2	*	2	*
3	* Copyright (C) 2010 Mattias Runge	3	* Copyright (C) 2010 Mattias Runge
4	*	4	*
5	* This program is free software; you can redistribute it and/or modify	5	* This program is free software; you can redistribute it and/or modify
6	* it under the terms of the GNU General Public License as published by	6	* it under the terms of the GNU General Public License as published by
7	* the Free Software Foundation; either version 2 of the License, or	7	* the Free Software Foundation; either version 2 of the License, or
8	* (at your option) any later version.	8	* (at your option) any later version.
9	*	9	*
10	* This program is distributed in the hope that it will be useful,	10	* This program is distributed in the hope that it will be useful,
11	* but WITHOUT ANY WARRANTY; without even the implied warranty of	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13	* GNU General Public License for more details.	13	* GNU General Public License for more details.
14	*	14	*
15	* You should have received a copy of the GNU General Public License along	15	* You should have received a copy of the GNU General Public License along
16	* with this program; if not, write to the Free Software Foundation, Inc.,	16	* with this program; if not, write to the Free Software Foundation, Inc.,
17	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.	17	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18	*	18	*
19	*/	19	*/
20		20
21	#include "UdpClient.h"	21	#include "UdpClient.h"
22		22
23	#include <boost/lexical_cast.hpp>	23	#include <boost/lexical_cast.hpp>
24		24
25	namespace atom {	25	namespace atom {
26	namespace net {	26	namespace net {
27		27
28	UdpClient::UdpClient(boost::asio::io_service& io_service, ClientId id, ServerId server_id) : io_service_(io_service), Client(io_service, id, server_id)	28	UdpClient::UdpClient(boost::asio::io_service& io_service, ClientId id, ServerId server_id) : io_service_(io_service), Client(io_service, id, server_id)
29	{	29	{
30	}	30	}
31		31
32	UdpClient::~UdpClient()	32	UdpClient::~UdpClient()
33	{	33	{
34	if (this->socket_.use_count() != 0)	34	if (this->socket_.use_count() != 0)
35	{	35	{
36	this->socket_->close();	36	this->socket_->close();
37	this->socket_.reset();	37	this->socket_.reset();
38	}	38	}
39	}	39	}
40		40
41	void UdpClient::Connect(std::string address, unsigned int port)	41	void UdpClient::Connect(std::string address, unsigned int port)
42	{	42	{
43	try	43	try
44	{	44	{
45	boost::asio::ip::udp::resolver resolver(this->io_service_);	45	boost::asio::ip::udp::resolver resolver(this->io_service_);
46	boost::asio::ip::udp::resolver::query query(boost::asio::ip::udp::v4(), address, boost::lexical_cast<std::string>(port));	46	boost::asio::ip::udp::resolver::query query(boost::asio::ip::udp::v4(), address, boost::lexical_cast<std::string>(port));
47	boost::asio::ip::udp::resolver::iterator it = resolver.resolve(query);	47	boost::asio::ip::udp::resolver::iterator it = resolver.resolve(query);
48		48
49	this->endpoint_ = *it;	49	this->endpoint_ = *it;
50		50
51	this->socket_.reset(new boost::asio::ip::udp::socket(this->io_service_, boost::asio::ip::udp::endpoint(boost::asio::ip::udp::v4(), port)));	51	this->socket_.reset(new boost::asio::ip::udp::socket(this->io_service_, boost::asio::ip::udp::endpoint(boost::asio::ip::udp::v4(), port)));
52	}	52	}
53	catch (std::exception e)	53	catch (std::exception e)
54	{	54	{
55	throw std::runtime_error("Error while connecting to " + address + ":" + boost::lexical_cast<std::string>(port) + ", " + e.what());	55	throw std::runtime_error("Error while connecting to " + address + ":" + boost::lexical_cast<std::string>(port) + ", " + e.what());
56	}	56	}
57		57
58	this->Read();	58	this->Read();
59	}	59	}
60		60
61	void UdpClient::Disconnect()	61	void UdpClient::Disconnect()
62	{	62	{
63	if (this->socket_.use_count() != 0)	63	if (this->socket_.use_count() != 0)
64	{	64	{
65	this->socket_->close();	65	this->socket_->close();
66	this->socket_.reset();	66	this->socket_.reset();
67	}	67	}
68		68
69	Client::Disconnect();	69	Client::Disconnect();
70	}	70	}
71		71
72	void UdpClient::Send(Buffer data)	72	void UdpClient::Send(Buffer data)
-		73	{
-		74	if (this->socket_->is_open())
73	{	75	{
74	this->socket_->send_to(boost::asio::buffer(data), this->endpoint_);	76	this->socket_->send_to(boost::asio::buffer(data), this->endpoint_);
-		77	}
-		78	else
-		79	{
-		80	this->Disconnect();
-		81	}
75	}	82	}
76		83
77	void UdpClient::Read()	84	void UdpClient::Read()
78	{	85	{
79	Client::Read();	86	Client::Read();
80		87
81	this->socket_->async_receive(boost::asio::buffer(this->buffer_),	88	this->socket_->async_receive(boost::asio::buffer(this->buffer_),
82	boost::bind(&UdpClient::ReadHandler,	89	boost::bind(&UdpClient::ReadHandler,
83	this,	90	this,
84	boost::asio::placeholders::error,	91	boost::asio::placeholders::error,
85	boost::asio::placeholders::bytes_transferred));	92	boost::asio::placeholders::bytes_transferred));
86	}	93	}
87		94
88		95
89	}; // namespace net	96	}; // namespace net
90	}; // namespace atom	97	}; // namespace atom
91		98

Subversion Repositories HomeAutomation

(root)/trunk/PcSoftware/Atom1.5/src/net/UdpClient.cpp – Rev 1592 → 1593