Adding API to check if a pong has not been received since last ping

Author: tomdixon

src/WebSocketClient.cpp

@@ -61,6 +61,8 @@ int WebSocketClient::begin(const char* aPath)
         }
     }
 
+    awaitingPong = false;
+
     iRxSize = 0;
 
     // status code of 101 means success
@@ -257,6 +259,7 @@ int WebSocketClient::parseMessage()
     }
     else if (TYPE_PONG == messageType())
     {
+        awaitingPong = false;
         flushRx();
         iRxSize = 0;
     }
@@ -302,11 +305,18 @@ int WebSocketClient::ping()
         pingData[i] = random(0xff);
     }
 
+    awaitingPong = true;
+
     beginMessage(TYPE_PING);
     write(pingData, sizeof(pingData));
     return endMessage();
 }
 
+bool WebSocketClient::isAwaitingPong()
+{
+    return awaitingPong;
+}
+
 int WebSocketClient::available()
 {
     if (iState < eReadingBody)

src/WebSocketClient.h

@@ -68,6 +68,11 @@ class WebSocketClient : public HttpClient
     */
     int ping();
 
+    /** Check if a pong has not yet been seen since last ping
+     @return true if awaiting a pong, otherwise false
+    */
+    bool isAwaitingPong();
+
     // Inherited from Print
     virtual size_t write(uint8_t aByte);
     virtual size_t write(const uint8_t *aBuffer, size_t aSize);
@@ -94,6 +99,8 @@ class WebSocketClient : public HttpClient
     bool iRxMasked;
     int iRxMaskIndex;
     uint8_t iRxMaskKey[4];
+
+    bool awaitingPong;
 };
 
 #endif