1 | /*
|
---|
2 | * The MIT License (MIT)
|
---|
3 | *
|
---|
4 | * Copyright (c) 2013 Mario Hock mails2013@omnifile.org
|
---|
5 | *
|
---|
6 | * Permission is hereby granted, free of charge, to any person
|
---|
7 | * obtaining a copy of this software and associated documentation
|
---|
8 | * files (the "Software"), to deal in the Software without
|
---|
9 | * restriction, including without limitation the rights to use,
|
---|
10 | * copy, modify, merge, publish, distribute, sublicense, and/or sell
|
---|
11 | * copies of the Software, and to permit persons to whom the
|
---|
12 | * Software is furnished to do so, subject to the following
|
---|
13 | * conditions:
|
---|
14 | *
|
---|
15 | * The above copyright notice and this permission notice shall be
|
---|
16 | * included in all copies or substantial portions of the Software.
|
---|
17 | *
|
---|
18 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
---|
19 | * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
---|
20 | * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
---|
21 | * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
---|
22 | * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
---|
23 | * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
---|
24 | * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
---|
25 | * OTHER DEALINGS IN THE SOFTWARE.
|
---|
26 | *
|
---|
27 | */
|
---|
28 |
|
---|
29 | #ifndef UDP_H
|
---|
30 | #define UDP_H
|
---|
31 |
|
---|
32 | #include "MessageSenderInterface.h"
|
---|
33 |
|
---|
34 | #include <ctime>
|
---|
35 | #include <iostream>
|
---|
36 | #include <string>
|
---|
37 | #include <boost/array.hpp>
|
---|
38 | #include <boost/bind.hpp>
|
---|
39 | #include <boost/shared_ptr.hpp>
|
---|
40 | #include <boost/asio.hpp>
|
---|
41 |
|
---|
42 | using boost::asio::ip::udp;
|
---|
43 |
|
---|
44 | class udp_server :
|
---|
45 | public MessageSenderInterface
|
---|
46 | {
|
---|
47 | public:
|
---|
48 | /// constructor
|
---|
49 | udp_server(boost::asio::io_service& io_service,
|
---|
50 | MessageSenderInterface& tunnel) :
|
---|
51 | socket_(io_service),
|
---|
52 | tunnel(tunnel)
|
---|
53 | {
|
---|
54 | }
|
---|
55 |
|
---|
56 | virtual ~udp_server();
|
---|
57 |
|
---|
58 | /// opens an udp "server-socket" on the given port
|
---|
59 | void listen_on(uint16_t port);
|
---|
60 |
|
---|
61 | /// connects an udp "client-socket" to the given port (on localhost)
|
---|
62 | void connect_to(uint16_t port);
|
---|
63 |
|
---|
64 |
|
---|
65 | /// MessageSenderInterface
|
---|
66 | virtual void SendMessage(reboost::message_t msg);
|
---|
67 | virtual void SendMessage(reboost::shared_buffer_t msg);
|
---|
68 |
|
---|
69 | private:
|
---|
70 | void start_receive();
|
---|
71 |
|
---|
72 | void handle_receive(const boost::system::error_code& error,
|
---|
73 | std::size_t bytes_transferred);
|
---|
74 |
|
---|
75 | void handle_send(reboost::shared_buffer_t msg,
|
---|
76 | const boost::system::error_code& /*error*/,
|
---|
77 | std::size_t /*bytes_transferred*/);
|
---|
78 |
|
---|
79 | /// forward data over ariba
|
---|
80 | void send_via_ariba(reboost::shared_buffer_t buffer);
|
---|
81 |
|
---|
82 | private:
|
---|
83 | udp::socket socket_;
|
---|
84 | MessageSenderInterface& tunnel;
|
---|
85 | udp::endpoint remote_endpoint_;
|
---|
86 | reboost::shared_buffer_t recv_buffer_;
|
---|
87 | };
|
---|
88 |
|
---|
89 | #endif // UDP_H
|
---|