| 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
|
|---|
