From 3e4b0b9869a92293949e01735728d3b344f243ae Mon Sep 17 00:00:00 2001
From: Andrew Tridgell <andrew@tridgell.net>
Date: Wed, 29 Jul 2015 14:04:25 +1000
Subject: [PATCH] AP_HAL: added pollout() function to socket API

---
 libraries/AP_HAL/utility/Socket.cpp | 24 ++++++++++++++++++++++++
 libraries/AP_HAL/utility/Socket.h   |  3 +++
 2 files changed, 27 insertions(+)

diff --git a/libraries/AP_HAL/utility/Socket.cpp b/libraries/AP_HAL/utility/Socket.cpp
index b8e41697cf..ac0ecf5b5c 100644
--- a/libraries/AP_HAL/utility/Socket.cpp
+++ b/libraries/AP_HAL/utility/Socket.cpp
@@ -161,6 +161,27 @@ bool SocketAPM::pollin(uint32_t timeout_ms)
     return true;
 }
 
+
+/*
+  return true if there is room for output data
+ */
+bool SocketAPM::pollout(uint32_t timeout_ms)
+{
+    fd_set fds;
+    struct timeval tv;
+
+    FD_ZERO(&fds);
+    FD_SET(fd, &fds);
+
+    tv.tv_sec = timeout_ms / 1000;
+    tv.tv_usec = (timeout_ms % 1000) * 1000UL;
+
+    if (select(fd+1, NULL, &fds, NULL, &tv) != 1) {
+        return false;
+    }
+    return true;
+}
+
 /* 
    start listening for new tcp connections
  */
@@ -183,6 +204,9 @@ SocketAPM *SocketAPM::accept(uint32_t timeout_ms)
     if (newfd == -1) {
         return NULL;
     }
+    // turn off nagle for lower latency
+    int one = 1;
+    setsockopt(newfd, IPPROTO_TCP, TCP_NODELAY, &one, sizeof(one));
     return new SocketAPM(false, newfd);
 }
 
diff --git a/libraries/AP_HAL/utility/Socket.h b/libraries/AP_HAL/utility/Socket.h
index cacf8dfee7..3dcd447c65 100644
--- a/libraries/AP_HAL/utility/Socket.h
+++ b/libraries/AP_HAL/utility/Socket.h
@@ -49,6 +49,9 @@ public:
     // return true if there is pending data for input
     bool pollin(uint32_t timeout_ms);
 
+    // return true if there is room for output data
+    bool pollout(uint32_t timeout_ms);
+
     // start listening for new tcp connections
     bool listen(uint16_t backlog);