From ec5876ba35d56e543829044a8d8aeef3c2d73cdf Mon Sep 17 00:00:00 2001
From: Philippe Teuwen <phil@teuwen.org>
Date: Mon, 6 May 2019 23:23:45 +0200
Subject: [PATCH] Linux: Detect soon when pm3 is unplugged

---
 client/comms.c | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/client/comms.c b/client/comms.c
index 25636589c..dafc24709 100644
--- a/client/comms.c
+++ b/client/comms.c
@@ -319,7 +319,7 @@ __attribute__((force_align_arg_pointer))
 *uart_communication(void *targ) {
     communication_arg_t *connection = (communication_arg_t *)targ;
     uint32_t rxlen;
-    bool sendfailed = false;
+    bool commfailed = false;
     PacketResponseNG rx;
     PacketResponseNGRaw rx_raw;
 
@@ -336,8 +336,8 @@ __attribute__((force_align_arg_pointer))
 
         // Signal to main thread that communications seems off.
         // main thread will kill and restart this thread.
-        if ( sendfailed ) {
-            PrintAndLogEx(WARNING, "sending bytes to Proxmark3 device " _RED_("failed"));
+        if ( commfailed ) {
+            PrintAndLogEx(WARNING, "Communicating with Proxmark3 device " _RED_("failed"));
             __atomic_test_and_set(&comm_thread_dead, __ATOMIC_SEQ_CST);
             break;
         }
@@ -458,6 +458,9 @@ __attribute__((force_align_arg_pointer))
                 PrintAndLogEx(WARNING, "Received packet frame preamble too short: %d/%d", rxlen, sizeof(PacketResponseNGPreamble));
                 error = true;
             }
+            if (res == PM3_ENOTTY) {
+                commfailed = true;
+            }
         }
 
         pthread_mutex_unlock(&spMutex);
@@ -485,14 +488,14 @@ __attribute__((force_align_arg_pointer))
             if (txBufferNGLen) { // NG packet
                 res = uart_send(sp, (uint8_t *) &txBufferNG, txBufferNGLen);
                 if (res == PM3_EIO) {
-                    sendfailed = true;
+                    commfailed = true;
                 }
                 conn.last_command = txBufferNG.pre.cmd;
                 txBufferNGLen = 0;
             } else {
                 res = uart_send(sp, (uint8_t *) &txBuffer, sizeof(PacketCommandOLD));
                 if (res == PM3_EIO) {
-                    sendfailed = true;
+                    commfailed = true;
                 }
                 conn.last_command = txBuffer.cmd;
             }