Merge branch 'error_handling' of ettus.sourcerepo.com:ettus/uhdpriv into usrp_e

1 files changed, 6 insertions, 4 deletions

diff --git a/host/lib/transport/zero_copy.cpp b/host/lib/transport/zero_copy.cpp
index 27f41329b..42f69d77b 100644
--- a/host/lib/transport/zero_copy.cpp
+++ b/host/lib/transport/zero_copy.cpp
@@ -71,7 +71,9 @@ managed_recv_buffer::sptr phony_zero_copy_recv_if::get_recv_buff(void){
     boost::uint8_t *recv_mem = new boost::uint8_t[_impl->max_buff_size];
 
     //call recv() with timeout option
-    size_t num_bytes = this->recv(boost::asio::buffer(recv_mem, _impl->max_buff_size));
+    ssize_t num_bytes = this->recv(boost::asio::buffer(recv_mem, _impl->max_buff_size));
+
+    if (num_bytes <= 0) return managed_recv_buffer::sptr(); //NULL sptr
 
     //create a new managed buffer to house the data
     return managed_recv_buffer::sptr(
@@ -86,7 +88,7 @@ managed_recv_buffer::sptr phony_zero_copy_recv_if::get_recv_buff(void){
 //! phony zero-copy send buffer implementation
 class managed_send_buffer_impl : public managed_send_buffer{
 public:
-    typedef boost::function<size_t(const boost::asio::const_buffer &)> send_fcn_t;
+    typedef boost::function<ssize_t(const boost::asio::const_buffer &)> send_fcn_t;
 
     managed_send_buffer_impl(
         const boost::asio::mutable_buffer &buff,
@@ -102,8 +104,8 @@ public:
         /* NOP */
     }
 
-    void commit(size_t num_bytes){
-        _send_fcn(boost::asio::buffer(_buff, num_bytes));
+    ssize_t commit(size_t num_bytes){
+        return _send_fcn(boost::asio::buffer(_buff, num_bytes));
     }
 
 private: