diff options
Diffstat (limited to 'gr-network/lib/tcp_sink_impl.h')
-rw-r--r-- | gr-network/lib/tcp_sink_impl.h | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/gr-network/lib/tcp_sink_impl.h b/gr-network/lib/tcp_sink_impl.h new file mode 100644 index 0000000000..68e0861bb5 --- /dev/null +++ b/gr-network/lib/tcp_sink_impl.h @@ -0,0 +1,75 @@ +/* -*- c++ -*- */ +/* + * Copyright 2020 Free Software Foundation, Inc. + * + * This file is part of GNU Radio + * + * SPDX-License-Identifier: GPL-3.0-or-later + * + */ + +#ifndef INCLUDED_NETWORK_TCP_SINK_IMPL_H +#define INCLUDED_NETWORK_TCP_SINK_IMPL_H + +#include <gnuradio/network/tcp_sink.h> +#include <boost/asio.hpp> +#include <boost/asio/ip/tcp.hpp> +#include <boost/thread/thread.hpp> + +namespace gr { +namespace network { + +class NETWORK_API tcp_sink_impl : public tcp_sink +{ +protected: + size_t d_itemsize; + size_t d_veclen; + std::string d_host; + int d_port; + int d_sinkmode; + + bool d_thread_running; + bool d_stop_thread; + boost::thread* d_listener_thread; + bool d_start_new_listener; + bool d_initial_connection; + + size_t d_block_size; + bool d_is_ipv6; + + boost::system::error_code ec; + + boost::asio::io_service d_io_service; + boost::asio::ip::tcp::endpoint d_endpoint; + boost::asio::ip::tcp::socket* d_tcpsocket = NULL; + boost::asio::ip::tcp::acceptor* d_acceptor = NULL; + + bool d_connected; + + virtual void check_for_disconnect(); + virtual void connect(bool initial_connection); + + virtual void run_listener(); + +public: + tcp_sink_impl(size_t itemsize, + size_t veclen, + const std::string& host, + int port, + int sinkmode = TCPSINKMODE_CLIENT); + ~tcp_sink_impl(); + + virtual bool stop(); + + void accept_handler(boost::asio::ip::tcp::socket* new_connection, + const boost::system::error_code& error); + + int work(int noutput_items, + gr_vector_const_void_star& input_items, + gr_vector_void_star& output_items); +}; + +} // namespace network +} // namespace gr + +#endif /* INCLUDED_NETWORK_TCP_SINK_IMPL_H */ |