1 files changed, 84 insertions, 0 deletions
diff --git a/gr-blocks/lib/udp_source_impl.h b/gr-blocks/lib/udp_source_impl.h
new file mode 100644
index 0000000000..d6c773726b
--- /dev/null
+++ b/gr-blocks/lib/udp_source_impl.h
@@ -0,0 +1,84 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2007-2010,2013 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 GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifndef INCLUDED_GR_UDP_SOURCE_IMPL_H
+#define INCLUDED_GR_UDP_SOURCE_IMPL_H
+
+#include <blocks/udp_source.h>
+#include <boost/asio.hpp>
+#include <boost/format.hpp>
+#include <thread/thread.h>
+
+namespace gr {
+  namespace blocks {
+
+    class udp_source_impl : public udp_source
+    {
+    private:
+      size_t  d_itemsize;
+      int     d_payload_size; // maximum transmission unit (packet length)
+      bool    d_eof;          // look for an EOF signal
+      bool    d_connected;    // are we connected?
+      char   *d_rxbuf;        // get UDP buffer items
+      char   *d_residbuf;     // hold buffer between calls
+      ssize_t d_residual;     // hold information about number of bytes stored in residbuf
+      ssize_t d_sent;         // track how much of d_residbuf we've outputted
+      size_t  d_offset;       // point to residbuf location offset
+
+      std::string d_host;
+      unsigned short d_port;
+
+      boost::asio::ip::udp::socket *d_socket;
+      boost::asio::ip::udp::endpoint d_endpoint;
+      boost::asio::ip::udp::endpoint d_endpoint_rcvd;
+      boost::asio::io_service d_io_service;
+
+      gr::thread::condition_variable d_cond_wait;
+      gr::thread::mutex d_udp_mutex;
+      gr::thread::thread d_udp_thread;
+
+      void start_receive();
+      void handle_read(const boost::system::error_code& error,
+                       size_t bytes_transferred);
+      void run_io_service() { d_io_service.run(); }
+
+    public:
+      udp_source_impl(size_t itemsize,
+                      const std::string &host, int port,
+                      int payload_size, bool eof);
+      ~udp_source_impl();
+
+      void connect(const std::string &host, int port);
+      void disconnect();
+
+      int payload_size() { return d_payload_size; }
+      int get_port();
+
+      int work(int noutput_items,
+               gr_vector_const_void_star &input_items,
+               gr_vector_void_star &output_items);
+    };
+
+  } /* namespace blocks */
+} /* namespace gr */
+
+#endif /* INCLUDED_GR_UDP_SOURCE_H */