summaryrefslogtreecommitdiff
path: root/vrt/lib
diff options
context:
space:
mode:
Diffstat (limited to 'vrt/lib')
-rw-r--r--vrt/lib/.gitignore4
-rw-r--r--vrt/lib/Makefile.am50
-rw-r--r--vrt/lib/copiers.cc69
-rw-r--r--vrt/lib/data_handler.cc32
-rw-r--r--vrt/lib/data_handler.h53
-rw-r--r--vrt/lib/expanded_header.cc119
-rw-r--r--vrt/lib/expanded_header_cw_tables.h14
-rw-r--r--vrt/lib/expanded_header_switch_body.h256
-rwxr-xr-xvrt/lib/gen_cw_tables.py69
-rwxr-xr-xvrt/lib/gen_switch_body.py80
-rw-r--r--vrt/lib/quadradio.cc408
-rw-r--r--vrt/lib/rx.cc120
-rw-r--r--vrt/lib/rx_packet_handler.cc41
-rw-r--r--vrt/lib/socket_rx_buffer.cc278
-rw-r--r--vrt/lib/socket_rx_buffer.h122
15 files changed, 1715 insertions, 0 deletions
diff --git a/vrt/lib/.gitignore b/vrt/lib/.gitignore
new file mode 100644
index 000000000..02b052397
--- /dev/null
+++ b/vrt/lib/.gitignore
@@ -0,0 +1,4 @@
+Makefile
+Makefile.in
+.deps
+.libs
diff --git a/vrt/lib/Makefile.am b/vrt/lib/Makefile.am
new file mode 100644
index 000000000..9832c0cd5
--- /dev/null
+++ b/vrt/lib/Makefile.am
@@ -0,0 +1,50 @@
+#
+# Copyright 2007,2008,2009 Free Software Foundation, Inc.
+#
+# This program 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 of the License, or
+# (at your option) any later version.
+#
+# This program 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, see <http://www.gnu.org/licenses/>.
+#
+
+include $(top_srcdir)/Makefile.common
+
+AM_CPPFLAGS = \
+ $(VRT_INCLUDES) \
+ $(BOOST_CPPFLAGS) \
+ $(CPPUNIT_INCLUDES)
+
+bin_PROGRAMS =
+
+lib_LTLIBRARIES = \
+ libvrt.la
+
+libvrt_la_SOURCES = \
+ copiers.cc \
+ data_handler.cc \
+ expanded_header.cc \
+ quadradio.cc \
+ rx.cc \
+ rx_packet_handler.cc \
+ socket_rx_buffer.cc
+
+libvrt_la_LIBADD =
+
+# Private headers not needed for above the API development
+noinst_HEADERS = \
+ data_handler.h \
+ expanded_header_cw_tables.h \
+ expanded_header_switch_body.h \
+ socket_rx_buffer.h
+
+EXTRA_DIST = \
+ gen_cw_tables.py \
+ gen_switch_body.py
diff --git a/vrt/lib/copiers.cc b/vrt/lib/copiers.cc
new file mode 100644
index 000000000..aa2846af5
--- /dev/null
+++ b/vrt/lib/copiers.cc
@@ -0,0 +1,69 @@
+/* -*- 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/copiers.h>
+#include <arpa/inet.h>
+
+namespace vrt {
+
+ void
+ copy_net_16sc_to_host_16sc(size_t nitems,
+ const uint32_t *items,
+ std::complex<int16_t> *host_items)
+ {
+#ifdef WORDS_BIGENDIAN
+
+ assert(sizeof(items[0]) == sizeof(host_items[0]));
+ memcpy(host_items, items, nitems * sizeof(items[0]));
+
+#else
+
+ // FIXME SIMD welcome here
+
+ for (size_t i = 0; i < nitems; i++){
+ uint32_t t = ntohl(items[i]);
+ //printf("%9d\n", items[i]);
+ host_items[i] = std::complex<int16_t>((t >> 16), t & 0xffff);
+ }
+
+#endif
+ }
+
+ void
+ copy_net_16sc_to_host_32fc(size_t nitems,
+ const uint32_t *items,
+ std::complex<float> *host_items)
+ {
+ // FIXME SIMD welcome here
+
+ for (size_t i = 0; i < nitems; i++){
+ uint32_t t = ntohl(items[i]);
+ int16_t re = (t >> 16) & 0xffff;
+ int16_t im = (t & 0xffff);
+ host_items[i] = std::complex<float>(re * 1.0/32768, im * 1.0/32768);
+ }
+ }
+
+};
+
diff --git a/vrt/lib/data_handler.cc b/vrt/lib/data_handler.cc
new file mode 100644
index 000000000..7d1f73a9a
--- /dev/null
+++ b/vrt/lib/data_handler.cc
@@ -0,0 +1,32 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2008,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.
+ */
+
+#include "data_handler.h"
+
+namespace vrt {
+
+ data_handler::~data_handler()
+ {
+ // default nop destructor
+ }
+
+}
+
diff --git a/vrt/lib/data_handler.h b/vrt/lib/data_handler.h
new file mode 100644
index 000000000..c041e48be
--- /dev/null
+++ b/vrt/lib/data_handler.h
@@ -0,0 +1,53 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2008,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.
+ */
+#ifndef INCLUDED_VRT_DATA_HANDLER_H
+#define INCLUDED_VRT_DATA_HANDLER_H
+
+#include <stdint.h>
+#include <stddef.h>
+
+namespace vrt {
+
+ /*!
+ * \brief Abstract function object called to handle received data blocks.
+ */
+ class data_handler
+ {
+ public:
+
+ enum result_bits {
+ DONE = 0x0002, //< do not call this object again
+ };
+
+ typedef int result; //< bitmask of result_bits
+
+ /*!
+ * \param base points to the beginning of the data
+ * \param len is the length in bytes of the data
+ * \returns bitmask composed of DONE
+ */
+ virtual result operator()(const void *base, size_t len) = 0;
+ virtual ~data_handler();
+ };
+
+} // namespace vrt
+
+#endif /* INCLUDED_VRT_DATA_HANDLER_H */
diff --git a/vrt/lib/expanded_header.cc b/vrt/lib/expanded_header.cc
new file mode 100644
index 000000000..d5c4ea888
--- /dev/null
+++ b/vrt/lib/expanded_header.cc
@@ -0,0 +1,119 @@
+/* -*- 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/expanded_header.h>
+#include <arpa/inet.h> // needs autoconf'ing
+//#include <stdio.h>
+
+namespace vrt {
+
+ // lookup tables indexed by packet type
+ unsigned char expanded_header::s_if_data[16] = {
+ 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
+ };
+
+ unsigned char expanded_header::s_ext_data[16] = {
+ 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
+ };
+
+ unsigned char expanded_header::s_data[16] = {
+ 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
+ };
+
+ unsigned char expanded_header::s_context[16] = {
+ 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
+ };
+
+ unsigned char expanded_header::s_stream_id[16] = {
+ 0, 1, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
+ };
+
+
+ // dispatch codeword bits
+ static const int HAS_STREAM_ID = 1 << 0;
+ static const int HAS_CLASS_ID = 1 << 1;
+ static const int HAS_INTEGER_SECS = 1 << 2;
+ static const int HAS_FRACTIONAL_SECS = 1 << 3;
+ static const int HAS_TRAILER = 1 << 4;
+
+#include "expanded_header_cw_tables.h"
+
+ static int
+ compute_codeword(const expanded_header &h)
+ {
+ int cw = 0;
+ if (h.stream_id_p()) cw |= HAS_STREAM_ID;
+ if (h.class_id_p()) cw |= HAS_CLASS_ID;
+ if (h.integer_secs_p()) cw |= HAS_INTEGER_SECS;
+ if (h.fractional_secs_p()) cw |= HAS_FRACTIONAL_SECS;
+ if (h.trailer_p()) cw |= HAS_TRAILER;
+ return cw;
+ }
+
+ bool
+ expanded_header::parse(const uint32_t *packet, // in
+ size_t n32_bit_words_packet, // in
+ expanded_header *h, // out
+ const uint32_t **payload, // out
+ size_t *n32_bit_words_payload) // out
+ {
+ size_t len = n32_bit_words_packet;
+ const uint32_t *p = packet;
+
+ *payload = 0;
+ *n32_bit_words_payload = 0;
+
+ // printf("parse: n32_bit_words_packet = %zd\n", n32_bit_words_packet);
+
+ if (len < 1){ // must have at least the header word
+ h->header = 0;
+ return false;
+ }
+
+ h->header = ntohl(p[0]);
+
+ if (h->pkt_size() > len)
+ return false; // VRT header says packet is bigger than what we've got
+
+ len = h->pkt_size(); // valid length of packet
+
+ int cw = compute_codeword(*h);
+ if (cw_header_len(cw) + cw_trailer_len(cw) > len)
+ return false; // negative payload len
+
+ *payload = p + cw_header_len(cw);
+ *n32_bit_words_payload = len - (cw_header_len(cw) + cw_trailer_len(cw));
+
+ // printf("parse: hdr = 0x%08x, cw = 0x%02x, cw_header_len(cw) = %d, cw_trailer_len(cw) = %d\n",
+ // h->header, cw, cw_header_len(cw), cw_trailer_len(cw));
+
+ switch (cw & 0x1f){
+#include "expanded_header_switch_body.h"
+ }
+
+ return true;
+ }
+
+
+}; // vrt
diff --git a/vrt/lib/expanded_header_cw_tables.h b/vrt/lib/expanded_header_cw_tables.h
new file mode 100644
index 000000000..fbb9c6b81
--- /dev/null
+++ b/vrt/lib/expanded_header_cw_tables.h
@@ -0,0 +1,14 @@
+inline static size_t cw_header_len(int cw){
+ static const size_t s_cw_header_len[32] = {
+ 1, 2, 3, 4, 2, 3, 4, 5, 3, 4, 5, 6, 4, 5, 6, 7, 1, 2, 3, 4, 2, 3, 4, 5, 3, 4, 5, 6, 4, 5, 6, 7,
+ };
+ return s_cw_header_len[cw];
+}
+
+inline static size_t cw_trailer_len(int cw){
+ static const size_t s_cw_trailer_len[32] = {
+ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
+ };
+ return s_cw_trailer_len[cw];
+}
+
diff --git a/vrt/lib/expanded_header_switch_body.h b/vrt/lib/expanded_header_switch_body.h
new file mode 100644
index 000000000..40e575c2c
--- /dev/null
+++ b/vrt/lib/expanded_header_switch_body.h
@@ -0,0 +1,256 @@
+ case 0:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 1:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 2:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 3:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 4:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[1]);
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 5:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[2]);
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 6:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = ntohl(p[3]);
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 7:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = ntohl(p[4]);
+ h->fractional_secs = 0;
+ h->trailer = 0;
+ break;
+
+ case 8:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->trailer = 0;
+ break;
+
+ case 9:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->trailer = 0;
+ break;
+
+ case 10:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[3])) << 32) | ntohl(p[4]);
+ h->trailer = 0;
+ break;
+
+ case 11:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[4])) << 32) | ntohl(p[5]);
+ h->trailer = 0;
+ break;
+
+ case 12:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[1]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->trailer = 0;
+ break;
+
+ case 13:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[2]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[3])) << 32) | ntohl(p[4]);
+ h->trailer = 0;
+ break;
+
+ case 14:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = ntohl(p[3]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[4])) << 32) | ntohl(p[5]);
+ h->trailer = 0;
+ break;
+
+ case 15:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = ntohl(p[4]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[5])) << 32) | ntohl(p[6]);
+ h->trailer = 0;
+ break;
+
+ case 16:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 17:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 18:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 19:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = 0;
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 20:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[1]);
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 21:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[2]);
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 22:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = ntohl(p[3]);
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 23:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = ntohl(p[4]);
+ h->fractional_secs = 0;
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 24:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 25:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 26:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[3])) << 32) | ntohl(p[4]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 27:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = 0;
+ h->fractional_secs = ((uint64_t)(ntohl(p[4])) << 32) | ntohl(p[5]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 28:
+ h->stream_id = 0;
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[1]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 29:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = 0;
+ h->integer_secs = ntohl(p[2]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[3])) << 32) | ntohl(p[4]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 30:
+ h->stream_id = 0;
+ h->class_id = ((uint64_t)(ntohl(p[1])) << 32) | ntohl(p[2]);
+ h->integer_secs = ntohl(p[3]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[4])) << 32) | ntohl(p[5]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
+ case 31:
+ h->stream_id = ntohl(p[1]);
+ h->class_id = ((uint64_t)(ntohl(p[2])) << 32) | ntohl(p[3]);
+ h->integer_secs = ntohl(p[4]);
+ h->fractional_secs = ((uint64_t)(ntohl(p[5])) << 32) | ntohl(p[6]);
+ h->trailer = ntohl(p[len-1]);
+ break;
+
diff --git a/vrt/lib/gen_cw_tables.py b/vrt/lib/gen_cw_tables.py
new file mode 100755
index 000000000..803a392e8
--- /dev/null
+++ b/vrt/lib/gen_cw_tables.py
@@ -0,0 +1,69 @@
+#!/usr/bin/env python
+#
+# 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.
+#
+
+import sys
+
+# dispatch codeword bits
+HAS_STREAM_ID = 1 << 0;
+HAS_CLASS_ID = 1 << 1;
+HAS_INTEGER_SECS = 1 << 2;
+HAS_FRACTIONAL_SECS = 1 << 3;
+HAS_TRAILER = 1 << 4;
+
+def main():
+ f = sys.stdout
+ header_len = 32 * [0]
+ trailer_len = 32 * [0]
+
+ for cw in range(32):
+ t = 0
+ if cw & HAS_TRAILER: t += 1
+ trailer_len[cw] = t
+
+ t = 1
+ if cw & HAS_STREAM_ID: t += 1
+ if cw & HAS_CLASS_ID: t += 2
+ if cw & HAS_INTEGER_SECS: t += 1
+ if cw & HAS_FRACTIONAL_SECS: t += 2
+ header_len[cw] = t
+
+ write_table(f, "cw_header_len", header_len)
+ write_table(f, "cw_trailer_len", trailer_len)
+
+def write_table(f, name, table):
+ f.write("inline static size_t ")
+ f.write(name)
+ f.write("(int cw){\n")
+
+ f.write(" static const size_t s_")
+ f.write(name)
+ f.write("[32] = {\n ")
+ for t in table:
+ f.write("%d, " % (t,))
+ f.write("\n };\n")
+
+ f.write(" return s_")
+ f.write(name)
+ f.write("[cw];\n}\n\n")
+
+
+if __name__ == '__main__':
+ main()
diff --git a/vrt/lib/gen_switch_body.py b/vrt/lib/gen_switch_body.py
new file mode 100755
index 000000000..105fa76a7
--- /dev/null
+++ b/vrt/lib/gen_switch_body.py
@@ -0,0 +1,80 @@
+#!/usr/bin/env python
+#
+# 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.
+#
+
+import sys
+
+# dispatch codeword bits
+HAS_STREAM_ID = 1 << 0;
+HAS_CLASS_ID = 1 << 1;
+HAS_INTEGER_SECS = 1 << 2;
+HAS_FRACTIONAL_SECS = 1 << 3;
+HAS_TRAILER = 1 << 4;
+
+def do_case(f, cw):
+
+ def do32(name, mask, index):
+ f.write(" ")
+ if cw & mask:
+ f.write("h->%s = ntohl(p[%d]);\n" % (name, index))
+ return 1
+ else:
+ f.write("h->%s = 0;\n" % (name,))
+ return 0
+
+ def do64(name, mask, index):
+ f.write(" ")
+ if cw & mask:
+ f.write("h->%s = ((uint64_t)(ntohl(p[%d])) << 32) | ntohl(p[%d]);\n" % (name, index, index+1))
+ return 2
+ else:
+ f.write("h->%s = 0;\n" % (name,))
+ return 0
+
+ def dotrailer(name, mask):
+ f.write(" ")
+ if cw & mask:
+ f.write("h->%s = ntohl(p[len-1]);\n" % (name,))
+ return 1
+ else:
+ f.write("h->%s = 0;\n" % (name,))
+ return 0
+
+ f.write(" case %d:\n" % (cw,))
+
+ index = 1
+ index += do32("stream_id", HAS_STREAM_ID, index)
+ index += do64("class_id", HAS_CLASS_ID, index)
+ index += do32("integer_secs", HAS_INTEGER_SECS, index)
+ index += do64("fractional_secs", HAS_FRACTIONAL_SECS, index)
+ dotrailer("trailer", HAS_TRAILER)
+
+ f.write(" break;\n\n")
+
+
+def main():
+ f = sys.stdout
+
+ for cw in range(32):
+ do_case(f, cw)
+
+
+if __name__ == '__main__':
+ main()
diff --git a/vrt/lib/quadradio.cc b/vrt/lib/quadradio.cc
new file mode 100644
index 000000000..ab5de89b8
--- /dev/null
+++ b/vrt/lib/quadradio.cc
@@ -0,0 +1,408 @@
+/* -*- 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/quadradio.h>
+#include <errno.h>
+#include <string.h>
+#include <stdio.h>
+#include <stdexcept>
+#include <math.h>
+
+#define MIN_IP_LOCAL_PORT 32768
+#define MAX_IP_LOCAL_PORT 61000
+
+#define ALL_DBOARDS 0xf
+
+static bool
+send_and_check(int fd, void *buf, size_t len)
+{
+ int r = send(fd, buf, len, 0);
+ if (r < 0){
+ perror("send");
+ return false;
+ }
+ if ((size_t) r != len){
+ fprintf(stderr, "send: short return value. expected %zd, got %d\n", len, r);
+ return false;
+ }
+ return true;
+}
+
+
+vrt::quadradio::quadradio(const std::string &ip, size_t rx_bufsize)
+ : d_ctrl_fd(0), d_data_fd(0), d_data_port(0),
+ d_band_select(0), d_rx_antenna(0), d_attenuation0(0), d_attenuation1(0)//d_10dB_atten(true)
+{
+ if (!open(ip.c_str()))
+ throw std::runtime_error("vrt::quadradio: failed to open " + ip + "\n");
+
+ d_rx = vrt::rx::make(data_socket_fd(), rx_bufsize);
+ set_test_signal(VRT_TEST_SIG_NORMAL);
+}
+
+vrt::quadradio::~quadradio()
+{
+ ::close(d_ctrl_fd);
+}
+
+bool
+vrt::quadradio::open(const char *ip)
+{
+ return open_sockets(ip, control_port(),
+ &d_ctrl_fd, &d_ctrl_port_inaddr,
+ &d_data_fd, &d_data_port);
+}
+
+bool
+vrt::quadradio::start_streaming(int samples_per_pkt)
+{
+ return send_rx_command(d_ctrl_fd, true, d_ctrl_port_inaddr,
+ d_data_port, samples_per_pkt, 0);
+}
+
+bool
+vrt::quadradio::stop_streaming()
+{
+ return send_stop_rx_command(d_ctrl_fd);
+}
+
+bool
+vrt::quadradio::set_center_freq(double target_freq){
+ if (target_freq < 700e6) return false;
+ if (target_freq <= 1.0e9) return set_band_select("A");
+ if (target_freq <= 1.5e9) return set_band_select("B");
+ if (target_freq <= 2.2e9) return set_band_select("C");
+ if (target_freq <= 3.0e9) return set_band_select("D");
+ return false;
+}
+
+bool
+vrt::quadradio::set_band_select(const std::string &band){
+ if (band == "A") d_band_select = 3;
+ else if (band == "B") d_band_select = 2;
+ else if (band == "C") d_band_select = 1;
+ else if (band == "D") d_band_select = 0;
+ else return false;
+ update_dboard_pins();
+ return true;
+}
+
+//void
+//vrt::quadradio::set_10dB_atten(bool on){
+// d_10dB_atten = on;
+// update_dboard_pins();
+//}
+
+bool
+vrt::quadradio::select_rx_antenna(const std::string &ant){
+ if (ant == "rf") d_rx_antenna = 0;
+ else if (ant == "cal") d_rx_antenna = 1;
+ else return true;
+ update_dboard_pins();
+ return true;
+}
+
+bool
+vrt::quadradio::set_attenuation0(int attenuation){
+ if (attenuation < 0 || attenuation > 31) return false;
+ d_attenuation0 = attenuation;
+ update_dboard_pins();
+ return true;
+}
+
+bool
+vrt::quadradio::set_attenuation1(int attenuation){
+ if (attenuation < 0 || attenuation > 31) return false;
+ d_attenuation1 = attenuation;
+ update_dboard_pins();
+ return true;
+}
+
+//bit reversal, length in bits
+static int reverse_bits(int input, int len){
+ int reversed = 0;
+ for (int i = 0; i < len; i++){
+ reversed += (input & (1<<i))?(1 << (len-i-1)):0;
+ }
+ return reversed;
+}
+
+void
+vrt::quadradio::update_dboard_pins(void){
+ int db_ctrl = \
+ ((reverse_bits(d_attenuation0, 5) & 0x1f) << 10) | \
+ ((reverse_bits(~d_attenuation1, 5) & 0x1f) << 03) | \
+ ((d_band_select & 0x03) << 01) | \
+ ((d_rx_antenna & 0x01) << 00);
+ set_dboard_pins(ALL_DBOARDS, db_ctrl); // FIXME sets them all
+}
+
+void
+vrt::quadradio::set_adc_gain(bool on){
+ set_hsadc_conf(ALL_DBOARDS, 0x14, on ? 0x90 : 0x80);
+}
+
+void
+vrt::quadradio::set_dc_offset_comp(bool on){
+ if (on) {
+ set_hsadc_conf(ALL_DBOARDS, 0x1B, 0x80);
+ set_hsadc_conf(ALL_DBOARDS, 0x1A, 0x00); //bits 6:4 set time constant
+ }
+ else set_hsadc_conf(ALL_DBOARDS, 0x1B, 0x00);
+}
+
+void
+vrt::quadradio::set_digital_gain(float gain){
+ int gain_q1 = static_cast<int>(round(gain*2.0));
+ set_hsadc_conf(ALL_DBOARDS, 0x17, gain_q1);
+}
+
+void
+vrt::quadradio::set_test_signal(vrt_test_sig_t type){
+ set_hsadc_conf(ALL_DBOARDS, 0x16, type);
+}
+
+bool
+vrt::quadradio::open_sockets(const char *quad_radio_ip, int quad_radio_ctrl_port,
+ int *ctrl_fd_ptr, struct in_addr *ctrl_port_inaddr,
+ int *data_fd_ptr, int *data_port_ptr)
+{
+ int ctrl_fd; // socket for control
+ int data_fd; // socket fd for data
+ int data_port; // our port number
+
+ //
+ // create a udp socket and connect it to the quad radio control port
+ //
+
+ ctrl_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+ if (ctrl_fd == -1){
+ perror("socket: ctrl_fd");
+ return false;
+ }
+
+ struct sockaddr_in si_other;
+ memset(&si_other, 0, sizeof(si_other));
+ si_other.sin_family = AF_INET;
+ si_other.sin_port = htons(quad_radio_ctrl_port);
+ if (inet_pton(AF_INET, quad_radio_ip, &si_other.sin_addr) <= 0){
+ perror("inet_pton");
+ return false;
+ }
+
+ if (connect(ctrl_fd, (struct sockaddr *) &si_other, sizeof(si_other)) != 0){
+ perror("connect");
+ return false;
+ }
+
+ // get our ip address associated with the interface connected to the control port
+
+ struct sockaddr_in si_me;
+ memset(&si_me, 0, sizeof(si_me));
+ socklen_t sockname_len = sizeof(si_me);
+ if (getsockname(ctrl_fd, (struct sockaddr *) &si_me, &sockname_len) != 0){
+ perror("getsockname");
+ }
+
+ *ctrl_port_inaddr = si_me.sin_addr;
+
+ if (1){
+ char buf[128];
+ const char *s = inet_ntop(si_me.sin_family, &si_me.sin_addr, buf, sizeof(buf));
+ if (s == 0){
+ perror("inet_ntop");
+ return false;
+ }
+ // printf("our ip addr associated with ctrl port: %s\n", s);
+ }
+
+ //
+ // create a udp socket to use to receive data
+ //
+
+ data_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+ if (data_fd == -1){
+ perror("socket: data_fd");
+ return false;
+ }
+
+ // bind it to a local port on the interface that connects to the ctrl port.
+ // FIXME this assumes that interface connected to the control port and the
+ // interface connected to the data port are the same. If we're using
+ // both ethernet ports on the quad radio, this may not be the case.
+
+ data_port = -1;
+ for (int port = MIN_IP_LOCAL_PORT; port <= MAX_IP_LOCAL_PORT; port++){
+ struct sockaddr_in si_me;
+ memset(&si_me, 0, sizeof(si_me));
+ si_me.sin_family = AF_INET;
+ si_me.sin_port = htons(port);
+ si_me.sin_addr.s_addr = htonl(INADDR_ANY);
+
+ if (bind(data_fd, (struct sockaddr *) &si_me, sizeof(si_me)) == 0){ // found one!
+ data_port = port;
+ break;
+ }
+ }
+
+ if (data_port == -1){
+ fprintf(stderr, "failed to bind to a local port\n");
+ return false;
+ }
+
+ // printf("our data port = %d\n", data_port);
+
+ *ctrl_fd_ptr = ctrl_fd;
+ *data_fd_ptr = data_fd;
+ *data_port_ptr = data_port;
+
+ return true;
+}
+
+// ------------------------------------------------------------------------
+
+bool
+vrt::quadradio::send_rx_command(int ctrl_fd, bool start,
+ struct in_addr addr, int data_port,
+ int samples_per_pkt, int siggen_param)
+{
+ uint32_t cmd[7];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(0); // id: rx_streaming
+ cmd[2] = htonl(start ? 1: 0); // start or stop?
+ cmd[3] = addr.s_addr; // ip address to send data to (already network endian)
+ cmd[4] = htonl(data_port); // port to send data to
+ cmd[5] = htonl(samples_per_pkt);
+ cmd[6] = htonl(siggen_param);
+
+ return send_and_check(ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::send_stop_rx_command(int ctrl_fd)
+{
+ struct in_addr in_addr;
+ in_addr.s_addr = 0;
+ return send_rx_command(ctrl_fd, false, in_addr, 0, 0, 0);
+}
+
+bool
+vrt::quadradio::set_dboard_pins(int dboard_bitmask, int v)
+{
+ uint32_t cmd[4];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(1); // id: dboard_pins
+ cmd[2] = htonl(dboard_bitmask);
+ cmd[3] = htonl(v); // value
+
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::set_setting_reg(int regno, int value)
+{
+ uint32_t cmd[4];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(2); // id: SR
+ cmd[2] = htonl(regno);
+ cmd[3] = htonl(value);
+
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::set_hsadc_conf(int dboard_bitmask, int regno, int value)
+{
+ uint32_t cmd[5];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(3); // id: HSADC_CONF
+ cmd[2] = htonl(dboard_bitmask);
+ cmd[3] = htonl(regno);
+ cmd[4] = htonl(value);
+
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::set_lsdac(int dboard_bitmask, int which_dac, int value)
+{
+ uint32_t cmd[5];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(4); // id: LSDAC
+ cmd[2] = htonl(dboard_bitmask);
+ cmd[3] = htonl(which_dac);
+ cmd[4] = htonl(value);
+
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::set_mem32(int addr, int value)
+{
+ uint32_t cmd[4];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(5); // id: MEM32
+ cmd[2] = htonl(addr);
+ cmd[3] = htonl(value);
+
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::set_lo_freq(double freq)
+{
+ uint64_t lo_freq = uint64_t(freq * (uint64_t(1)<<20)); //q20 format
+ uint32_t cmd[4];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(6); // id: lo freq
+ cmd[2] = htonl((lo_freq >> 32) & 0xffffffff);
+ cmd[3] = htonl((lo_freq >> 0) & 0xffffffff);
+
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::set_cal_freq(double freq)
+{
+ uint64_t cal_freq = uint64_t(freq * (uint64_t(1)<<20)); //q20 format
+ uint32_t cmd[4];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(7); // id: cal freq
+ cmd[2] = htonl((cal_freq >> 32) & 0xffffffff);
+ cmd[3] = htonl((cal_freq >> 0) & 0xffffffff);
+
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
+bool
+vrt::quadradio::set_beamforming(int32_t gains[8]){
+ uint32_t cmd[2+8];
+ cmd[0] = htonl(0); // verb: set
+ cmd[1] = htonl(8); // id: beamformin
+ for (int i = 0; i < 8; i++){
+ //printf("%d\n", gains[i]);
+ cmd[i+2] = htonl(gains[i]);
+ }
+ return send_and_check(d_ctrl_fd, cmd, sizeof(cmd));
+}
+
diff --git a/vrt/lib/rx.cc b/vrt/lib/rx.cc
new file mode 100644
index 000000000..f75db3111
--- /dev/null
+++ b/vrt/lib/rx.cc
@@ -0,0 +1,120 @@
+/* -*- 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, 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(){}
+
+ data_handler::result
+ vrt_data_handler::operator()(const void *base, size_t len)
+ {
+#if 0
+ print_words(0, (uint32_t *)base, len/(sizeof(uint32_t)));
+ return 0;
+#else
+ const uint32_t *payload;
+ size_t n32_bit_words;
+ expanded_header hdr;
+ if (!expanded_header::parse((const uint32_t*) base, len/(sizeof(uint32_t)),
+ &hdr, &payload, &n32_bit_words)){
+ if (1){
+ fprintf(stderr, "vrt_data_handler: malformed VRT packet!\n");
+ print_words(stderr, 0, (uint32_t *)base, len/(sizeof(uint32_t)));
+ }
+ return 0;
+ }
+ bool want_more = (*d_handler)(payload, n32_bit_words, &hdr);
+ return !want_more ? data_handler::DONE : 0;
+#endif
+ }
+
+
+ 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
diff --git a/vrt/lib/rx_packet_handler.cc b/vrt/lib/rx_packet_handler.cc
new file mode 100644
index 000000000..11f90278d
--- /dev/null
+++ b/vrt/lib/rx_packet_handler.cc
@@ -0,0 +1,41 @@
+/* -*- 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_packet_handler.h>
+
+namespace vrt {
+
+ rx_packet_handler::~rx_packet_handler(){}
+
+ // default operator is a NOP
+ bool
+ rx_packet_handler::operator()(const uint32_t *payload,
+ size_t n32_bit_words,
+ const expanded_header *hdr)
+ {
+ return true;
+ }
+
+
+}; // vrt
diff --git a/vrt/lib/socket_rx_buffer.cc b/vrt/lib/socket_rx_buffer.cc
new file mode 100644
index 000000000..6ed211b9a
--- /dev/null
+++ b/vrt/lib/socket_rx_buffer.cc
@@ -0,0 +1,278 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2008,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 "socket_rx_buffer.h"
+#include "data_handler.h"
+#include <linux/if_packet.h>
+#include <sys/socket.h>
+#include <sys/mman.h>
+#include <sys/poll.h>
+#include <iostream>
+#include <cmath>
+#include <errno.h>
+#include <stdexcept>
+#include <string.h>
+#include <fcntl.h>
+#include <cstdio>
+
+
+#define SOCKET_RX_BUFFER_DEBUG 1 // define to 0 or 1
+#if SOCKET_RX_BUFFER_DEBUG
+#define DEBUG_LOG(x) ::write(2, (x), 1)
+#else
+#define DEBUG_LOG(X)
+#endif
+
+#define DEFAULT_MEM_SIZE 62.5e6 // ~0.5s @ 125 MB/s
+#define MAX_MEM_SIZE 1000e6 // ~10.00s @ 100 MB/s.
+#define MAX_SLAB_SIZE 131072 // 128 KB (FIXME fish out of /proc/slabinfo)
+
+
+namespace vrt {
+
+ const unsigned int socket_rx_buffer::MAX_PKTLEN = 8192;
+ const unsigned int socket_rx_buffer::MIN_PKTLEN = 64;
+
+ socket_rx_buffer::socket_rx_buffer(int socket_fd, size_t rx_bufsize)
+ : d_fd(socket_fd), d_using_tpring(false), d_buflen(0), d_buf(0), d_frame_nr(0),
+ d_frame_size(0), d_head(0), d_ring(0)
+ {
+ if (rx_bufsize == 0)
+ d_buflen = (size_t)DEFAULT_MEM_SIZE;
+ else
+ d_buflen = std::min((size_t)MAX_MEM_SIZE, rx_bufsize);
+
+ if (!open()){
+ throw std::runtime_error("socket_rx_buffer::open failed");
+ }
+ }
+
+ socket_rx_buffer::~socket_rx_buffer()
+ {
+ close();
+ }
+
+ bool
+ socket_rx_buffer::open()
+ {
+ if (try_packet_ring()){
+ d_using_tpring = true;
+ // fprintf(stderr, "socket_rx_buffer: using memory mapped interface\n");
+ }
+ else {
+ d_using_tpring = false;
+ // fprintf(stderr, "socket_rx_buffer: NOT using memory mapped interface\n");
+
+ // Increase socket buffer if possible
+
+ int rcvbuf_size = d_buflen;
+#if defined(SO_RCVBUFFORCE)
+ if (setsockopt(d_fd, SOL_SOCKET, SO_RCVBUFFORCE, &rcvbuf_size, sizeof(rcvbuf_size)) != 0){
+ perror("setsockopt(SO_RCVBUFFORCE)");
+ fprintf(stderr, "Are you running as root? If not, please do.\n");
+ }
+ else {
+ fprintf(stderr, "SO_RCVBUFFORCE = %zd\n", d_buflen);
+ }
+#endif
+ }
+
+ return true;
+ }
+
+ bool
+ socket_rx_buffer::try_packet_ring()
+ {
+ struct tpacket_req req;
+ size_t page_size = getpagesize();
+
+ // Calculate minimum power-of-two aligned size for frames
+ req.tp_frame_size =
+ (unsigned int)rint(pow(2, ceil(log2(TPACKET_ALIGN(TPACKET_HDRLEN)+TPACKET_ALIGN(MAX_PKTLEN)))));
+ d_frame_size = req.tp_frame_size;
+
+ // Calculate minimum contiguous pages needed to enclose a frame
+ int npages = (page_size > req.tp_frame_size) ? 1 : ((req.tp_frame_size+page_size-1)/page_size);
+ req.tp_block_size = page_size << (int)ceil(log2(npages));
+
+ // Calculate number of blocks
+ req.tp_block_nr = (int)(d_buflen/req.tp_block_size);
+
+
+ // Recalculate buffer length
+ d_buflen = req.tp_block_nr*req.tp_block_size;
+
+ // Finally, calculate total number of frames. Since frames, blocks,
+ // and pages are all power-of-two aligned, frames are contiguous
+ req.tp_frame_nr = d_buflen/req.tp_frame_size;
+ d_frame_nr = req.tp_frame_nr;
+
+#if 0
+ if (SOCKET_RX_BUFFER_DEBUG)
+ std::cerr << "socket_rx_buffer:"
+ << " frame_size=" << req.tp_frame_size
+ << " block_size=" << req.tp_block_size
+ << " block_nr=" << req.tp_block_nr
+ << " frame_nr=" << req.tp_frame_nr
+ << " buflen=" << d_buflen
+ << std::endl;
+#endif
+
+ // Try to get kernel shared memory buffer
+ if (setsockopt(d_fd, SOL_PACKET, PACKET_RX_RING, (void *)&req, sizeof(req)) != 0){
+ // perror("socket_rx_buffer: setsockopt");
+ return false;
+ }
+
+ void *p = mmap(0, d_buflen, PROT_READ|PROT_WRITE, MAP_SHARED, d_fd, 0);
+ if (p == MAP_FAILED){
+ perror("socket_rx_buffer: mmap");
+ return false;
+ }
+ d_buf = (uint8_t *) p;
+
+ // Initialize our pointers into the packet ring
+ d_ring.resize(req.tp_frame_nr);
+ for (unsigned int i=0; i < req.tp_frame_nr; i++)
+ d_ring[i] = (uint8_t *)(d_buf+i*req.tp_frame_size);
+
+ return true;
+ }
+
+ bool
+ socket_rx_buffer::close()
+ {
+ return true;
+ }
+
+ inline bool
+ socket_rx_buffer::frame_available()
+ {
+ return (((tpacket_hdr *)d_ring[d_head])->tp_status != TP_STATUS_KERNEL);
+ }
+
+ socket_rx_buffer::result
+ socket_rx_buffer::rx_frames(data_handler *f, int timeout_in_ms)
+ {
+ if (!d_using_tpring){
+
+ // ----------------------------------------------------------------
+ // Use recv instead of kernel Rx packet ring
+ // ----------------------------------------------------------------
+
+ unsigned char buf[MAX_PKTLEN];
+ bool dont_wait = timeout_in_ms == 0; // FIXME treating timeout as 0 or inf
+ int flags = dont_wait ? MSG_DONTWAIT : 0;
+
+ ssize_t rr = recv(d_fd, buf, sizeof(buf), flags);
+ if (rr == -1){ // error?
+ if (errno == EAGAIN){ // non-blocking, nothing there
+ return EB_WOULD_BLOCK;
+ }
+ perror("rx_frames: recv");
+ return EB_ERROR;
+ }
+
+ // Got first packet. Call handler
+
+ data_handler::result r = (*f)(buf, rr);
+ if (r & data_handler::DONE)
+ return EB_OK;
+
+ // Now do as many as we can without blocking
+
+ while (1){
+ rr = recv(d_fd, buf, sizeof(buf), MSG_DONTWAIT);
+ if (rr == -1){ // error?
+ if (errno == EAGAIN) // non-blocking, nothing there
+ return EB_OK; // return OK; we've processed >= 1 packets
+ perror("rx_frames: recv");
+ return EB_ERROR;
+ }
+
+ r = (*f)(buf, rr);
+ if (r & data_handler::DONE)
+ break;
+ }
+ return EB_OK;
+ }
+
+ // ----------------------------------------------------------------
+ // Use kernel Rx packet ring
+ // ----------------------------------------------------------------
+
+ DEBUG_LOG("\n");
+
+ while (!frame_available()) {
+ if (timeout_in_ms == 0) {
+ DEBUG_LOG("w");
+ return EB_WOULD_BLOCK;
+ }
+
+ struct pollfd pfd;
+ pfd.fd = d_fd;
+ pfd.revents = 0;
+ pfd.events = POLLIN;
+
+ // DEBUG_LOG("P");
+
+ int pres = poll(&pfd, 1, timeout_in_ms);
+ if (pres == -1) {
+ perror("poll");
+ return EB_ERROR;
+ }
+
+ if (pres == 0) {
+ DEBUG_LOG("t");
+ return EB_TIMED_OUT;
+ }
+ }
+
+ // Iterate through available packets
+ while (frame_available()) {
+ // Get start of ethernet frame and length
+ tpacket_hdr *hdr = (tpacket_hdr *)d_ring[d_head];
+ void *base = (uint8_t *)hdr+hdr->tp_mac;
+ size_t len = hdr->tp_len;
+
+ if (1)
+ fprintf(stderr, "socket_rx_buffer: base = %p tp_mac = %3d tp_net = %3d\n",
+ base, hdr->tp_mac, hdr->tp_net);
+
+ // Invoke data handler
+ data_handler::result r = (*f)(base, len);
+ hdr->tp_status = TP_STATUS_KERNEL; // mark it free
+
+ inc_head();
+
+ if (r & data_handler::DONE)
+ break;
+ }
+
+ DEBUG_LOG("|");
+ return EB_OK;
+ }
+
+} // namespace vrt
diff --git a/vrt/lib/socket_rx_buffer.h b/vrt/lib/socket_rx_buffer.h
new file mode 100644
index 000000000..053c30c12
--- /dev/null
+++ b/vrt/lib/socket_rx_buffer.h
@@ -0,0 +1,122 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2008,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.
+ */
+#ifndef INCLUDED_VRT_SOCKET_RX_BUFFER_H
+#define INCLUDED_VRT_SOCKET_RX_BUFFER_H
+
+#include <boost/utility.hpp>
+#include <vector>
+#include <memory>
+#include <stdint.h>
+
+namespace vrt {
+
+ class data_handler;
+
+ /*!
+ * \brief high-performance interface to receive datagrams
+ *
+ * On many systems it should be possible to implement this on top of libpcap
+ *
+ * \internal
+ */
+ class socket_rx_buffer : boost::noncopyable
+ {
+
+ int d_fd; // socket file descriptor
+ bool d_using_tpring; // using kernel mapped packet ring
+ size_t d_buflen; // length of our buffer
+ uint8_t *d_buf; // packet ring
+ unsigned int d_frame_nr; // max frames on ring
+ size_t d_frame_size; // frame storage size
+ unsigned int d_head; // pointer to next frame
+
+ std::vector<uint8_t *> d_ring; // pointers into buffer
+
+ bool frame_available();
+
+ void inc_head()
+ {
+ if (d_head + 1 >= d_frame_nr)
+ d_head = 0;
+ else
+ d_head = d_head + 1;
+ }
+
+ bool open();
+ bool close();
+ bool try_packet_ring();
+
+ public:
+
+ enum result {
+ EB_OK, //< everything's fine
+ EB_ERROR, //< A non-recoverable error occurred
+ EB_WOULD_BLOCK, //< A timeout of 0 was specified and nothing was ready
+ EB_TIMED_OUT, //< The timeout expired before anything was ready
+ };
+
+ static const unsigned int MAX_PKTLEN;
+ static const unsigned int MIN_PKTLEN;
+
+ /*!
+ * \param socket_fd file descriptor that corresponds to a socket
+ * \param rx_bufsize is a hint as to the number of bytes of memory
+ * to allocate for received ethernet frames (0 -> reasonable default)
+ */
+ socket_rx_buffer(int socket_fd, size_t rx_bufsize = 0);
+ ~socket_rx_buffer();
+
+ /*!
+ * \brief Call \p f for each frame in the receive buffer.
+ * \param f is the frame data handler
+ * \param timeout (in ms) controls behavior when there are no frames to read
+ *
+ * If \p timeout is 0, rx_frames will not wait for frames if none are
+ * available, and f will not be invoked. If \p timeout is -1 (the
+ * default), rx_frames will block indefinitely until frames are
+ * available. If \p timeout is positive, it indicates the number of
+ * milliseconds to wait for a frame to become available. Once the
+ * timeout has expired, rx_frames will return, f never having been
+ * invoked.
+ *
+ * \p f will be called on each frame that is available.
+ * \p f returns a bit mask with one of the following set or cleared:
+ *
+ * data_handler::DONE - return from rx_frames now even though more frames
+ * might be available; otherwise continue if more
+ * frames are ready.
+ *
+ * \returns EB_OK if at least one frame was received
+ * \returns EB_WOULD_BLOCK if \p timeout is 0 and the call would have blocked
+ * \returns EB_TIMED_OUT if timeout occurred
+ * \returns EB_ERROR if there was an unrecoverable error.
+ */
+ result rx_frames(data_handler *f, int timeout=-1);
+
+ /*
+ * \brief Returns maximum possible number of frames in buffer
+ */
+ unsigned int max_frames() const { return d_using_tpring ? d_frame_nr : 0; }
+ };
+
+}; // namespace vrt
+
+#endif /* INCLUDED_VRT_SOCKET_RX_BUFFER_H */
generated by cgit (git 2.34.1) at 2025-01-10 22:40:43 +0000