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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/* -*- c++ -*- */
/*
* Copyright 2009 Free Software Foundation, Inc.
*
* This file is part of GNU Radio
*
* GNU Radio 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 3, or (at your option)
* any later version.
*
* GNU Radio 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.
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <vrt/rx.h>
#include <vrt/expanded_header.h>
#include "socket_rx_buffer.h"
#include "data_handler.h"
#include <unistd.h>
#include <stdio.h>
#include <stdexcept>
static void
print_words(FILE *fp, size_t offset, const uint32_t *buf, size_t n)
{
size_t i;
for (i = 0; i < n; i++){
if (i % 4 == 0){
fprintf(fp, "%04zx:", i);
}
putc(' ', fp);
fprintf(fp, "%08x", buf[i]);
if (i % 4 == 3)
putc('\n', fp);
}
putc('\n', fp);
}
namespace vrt {
rx::sptr
rx::make(int socket_fd, size_t rx_bufsize)
{
return sptr(new rx(socket_fd, rx_bufsize));
}
rx::rx(int socket_fd, size_t rx_bufsize)
: d_socket_fd(socket_fd),
d_srb(new socket_rx_buffer(socket_fd, rx_bufsize))
{
}
rx::~rx()
{
delete d_srb;
::close(d_socket_fd);
}
class vrt_data_handler : public data_handler
{
rx_packet_handler *d_handler;
public:
vrt_data_handler(rx_packet_handler *handler)
: d_handler(handler){}
~vrt_data_handler();
result operator()(const void *base, size_t len);
};
vrt_data_handler::~vrt_data_handler(){}
// N.B., There may be more than 1 VRT packet in a frame (usually IF-Context packets)
data_handler::result
vrt_data_handler::operator()(const void *base, size_t len)
{
const uint32_t *word_base = (const uint32_t *) base;
size_t word_len = len/(sizeof(uint32_t));
bool want_more = true;
while (word_len > 0 && want_more){
const uint32_t *payload;
size_t n32_bit_words;
expanded_header hdr;
if (!expanded_header::parse(word_base, word_len,
&hdr, &payload, &n32_bit_words)){
if (1){
fprintf(stderr, "vrt_data_handler: malformed VRT packet!\n");
print_words(stderr, 0, word_base, word_len);
}
return 0;
}
want_more = (*d_handler)(payload, n32_bit_words, &hdr);
word_base += hdr.pkt_size();
word_len -= hdr.pkt_size();
}
return !want_more ? data_handler::DONE : 0;
}
bool
rx::rx_packets(rx_packet_handler *handler, bool dont_wait)
{
vrt_data_handler h(handler);
socket_rx_buffer::result r = d_srb->rx_frames(&h, dont_wait ? 0 : -1);
return r == socket_rx_buffer::EB_OK || r == socket_rx_buffer::EB_WOULD_BLOCK;
}
}; // vrt
|
