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_->cancel();
        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_->cancel();
        this->socket_->close();
        this->socket_.reset();
    }
   
    Client::Disconnect();
}
 
void UdpClient::Send(type::Byteset data)
{
    try
    {
        if (this->socket_->is_open())
        {
            this->socket_->send_to(boost::asio::buffer(data.Get(), data.GetMaxSize()), this->endpoint_);
        }
    }
    catch (std::exception& e)
    {
        this->Disconnect();
        //throw std::runtime_error(e.what());
    }
}
 
void UdpClient::Read()
{
    Client::Read();
   
    this->socket_->async_receive(boost::asio::buffer(this->buffer_.Get(), this->buffer_.GetMaxSize()),
                                 boost::bind(&UdpClient::ReadHandler,
                                             this,
                                             boost::asio::placeholders::error,
                                             boost::asio::placeholders::bytes_transferred));
}
 
 
}; // namespace net
}; // namespace atom
 

Rev 1596	Rev 1597
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_->cancel();	36	this->socket_->cancel();
37	this->socket_->close();	37	this->socket_->close();
38	this->socket_.reset();	38	this->socket_.reset();
39	}	39	}
40	}	40	}
41		41
42	void UdpClient::Connect(std::string address, unsigned int port)	42	void UdpClient::Connect(std::string address, unsigned int port)
43	{	43	{
44	try	44	try
45	{	45	{
46	boost::asio::ip::udp::resolver resolver(this->io_service_);	46	boost::asio::ip::udp::resolver resolver(this->io_service_);
47	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::query query(boost::asio::ip::udp::v4(), address, boost::lexical_cast<std::string>(port));
48	boost::asio::ip::udp::resolver::iterator it = resolver.resolve(query);	48	boost::asio::ip::udp::resolver::iterator it = resolver.resolve(query);
49		49
50	this->endpoint_ = *it;	50	this->endpoint_ = *it;
51		51
52	this->socket_.reset(new boost::asio::ip::udp::socket(this->io_service_, boost::asio::ip::udp::endpoint(boost::asio::ip::udp::v4(), port)));	52	this->socket_.reset(new boost::asio::ip::udp::socket(this->io_service_, boost::asio::ip::udp::endpoint(boost::asio::ip::udp::v4(), port)));
53	}	53	}
54	catch (std::exception e)	54	catch (std::exception e)
55	{	55	{
56	throw std::runtime_error("Error while connecting to " + address + ":" + boost::lexical_cast<std::string>(port) + ", " + e.what());	56	throw std::runtime_error("Error while connecting to " + address + ":" + boost::lexical_cast<std::string>(port) + ", " + e.what());
57	}	57	}
58		58
59	this->Read();	59	this->Read();
60	}	60	}
61		61
62	void UdpClient::Disconnect()	62	void UdpClient::Disconnect()
63	{	63	{
64	if (this->socket_.use_count() != 0)	64	if (this->socket_.use_count() != 0)
65	{	65	{
66	this->socket_->cancel();	66	this->socket_->cancel();
67	this->socket_->close();	67	this->socket_->close();
68	this->socket_.reset();	68	this->socket_.reset();
69	}	69	}
70		70
71	Client::Disconnect();	71	Client::Disconnect();
72	}	72	}
73		73
74	void UdpClient::Send(type::Byteset data)	74	void UdpClient::Send(type::Byteset data)
-		75	{
-		76	try
75	{	77	{
76	if (this->socket_->is_open())	78	if (this->socket_->is_open())
77	{	79	{
78	this->socket_->send_to(boost::asio::buffer(data.Get(), data.GetMaxSize()), this->endpoint_);	80	this->socket_->send_to(boost::asio::buffer(data.Get(), data.GetMaxSize()), this->endpoint_);
79	}	81	}
80	~~else~~	82	}
-		83	catch (std::exception& e)
81	{	84	{
82	this->Disconnect();	85	this->Disconnect();
-		86	//throw std::runtime_error(e.what());
83	}	87	}
84	}	88	}
85		89
86	void UdpClient::Read()	90	void UdpClient::Read()
87	{	91	{
88	Client::Read();	92	Client::Read();
89		93
90	this->socket_->async_receive(boost::asio::buffer(this->buffer_.Get(), this->buffer_.GetMaxSize()),	94	this->socket_->async_receive(boost::asio::buffer(this->buffer_.Get(), this->buffer_.GetMaxSize()),
91	boost::bind(&UdpClient::ReadHandler,	95	boost::bind(&UdpClient::ReadHandler,
92	this,	96	this,
93	boost::asio::placeholders::error,	97	boost::asio::placeholders::error,
94	boost::asio::placeholders::bytes_transferred));	98	boost::asio::placeholders::bytes_transferred));
95	}	99	}
96		100
97		101
98	}; // namespace net	102	}; // namespace net
99	}; // namespace atom	103	}; // namespace atom
100		104

Subversion Repositories HomeAutomation

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