Add send/receive sanity checks

lxi-tools · Mar 10, 2018 · b5dd325 · b5dd325
1 parent c2b476c
commit b5dd325
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/lxi.c b/src/lxi.c
@@ -155,6 +155,12 @@ EXPORT int lxi_send(int device, char *message, int length, int timeout)
 {
     int bytes_sent;
 
+    if (device < 0)
+        return LXI_ERROR;
+
+    if (session[device].connected == false)
+        return LXI_ERROR;
+
     // Send
     bytes_sent = session[device].send(session[device].data, message, length, timeout);
     if (bytes_sent < 0)
@@ -169,6 +175,12 @@ EXPORT int lxi_receive(int device, char *message, int length, int timeout)
 {
     int bytes_received;
 
+    if (device < 0)
+        return LXI_ERROR;
+
+    if (session[device].connected == false)
+        return LXI_ERROR;
+
     // Receive
     bytes_received = session[device].receive(session[device].data, message, length, timeout);
     if (bytes_received < 0)