/
udp_sink_impl.h
80 lines (63 loc) · 1.91 KB
/
udp_sink_impl.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/* -*- 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_UDP_SINK_IMPL_H
#define INCLUDED_NETWORK_UDP_SINK_IMPL_H
#include <gnuradio/buffer.h>
#include <gnuradio/buffer_reader.h>
#include <gnuradio/network/udp_sink.h>
#include <asio.hpp>
#include <gnuradio/network/packet_headers.h>
namespace gr {
namespace network {
class NETWORK_API udp_sink_impl : public udp_sink
{
protected:
std::string d_host;
int d_port;
size_t d_itemsize;
size_t d_veclen;
size_t d_block_size;
bool is_ipv6;
int d_header_type;
int d_header_size;
uint64_t d_seq_num;
uint16_t d_payloadsize;
bool b_send_eof;
int d_precomp_datasize;
int d_precomp_data_overitemsize;
char d_tmpheaderbuff[12]; // Largest header is 10 bytes
// A queue is required because we have 2 different timing
// domains: The network packets and the GR work()/scheduler
gr::buffer_sptr d_localqueue_writer;
gr::buffer_reader_sptr d_localqueue_reader;
asio::error_code ec;
asio::io_context d_io_context;
asio::ip::udp::endpoint d_endpoint;
asio::ip::udp::socket* d_udpsocket = nullptr;
virtual void
build_header(); // returns header size. Header is stored in tmpHeaderBuff
public:
udp_sink_impl(size_t itemsize,
size_t veclen,
const std::string& host,
int port,
int header_type = HEADERTYPE_NONE,
int payloadsize = 1472,
bool send_eof = true);
~udp_sink_impl() override;
bool start() override;
bool stop() override;
int work(int noutput_items,
gr_vector_const_void_star& input_items,
gr_vector_void_star& output_items) override;
};
} // namespace network
} // namespace gr
#endif /* INCLUDED_NETWORK_UDP_SINK_IMPL_H */