Merge pull request #114 from Habbie/wsclose

turn websocket close from a confusing error into a clear one
voorkant · Oct 4, 2024 · bd05b20 · bd05b20
2 parents eaad4b4 + 5b3aa54
commit bd05b20
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/WSConn.cpp b/src/WSConn.cpp
@@ -1,5 +1,6 @@
 
 #include "WSConn.hpp"
+#include "src/logger.hpp"
 #include <iostream>
 
 using std::cout;
@@ -66,6 +67,17 @@ std::string WSConn::recv(void)
   // cout<<"RESULT:"<<endl;
   // cout<<result<<endl;
   // cout<<"END RESULT"<<endl;
+
+  if (meta->flags & CURLWS_CLOSE) {
+    uint16_t reason = 0;
+
+    if (result.size() >= 2) {
+      reason = ((unsigned char)result[0] << 8) + (unsigned char)result[1];
+    }
+    g_log << Logger::LogLevel::Error << "got CURL_WS_CLOSE, reason=" << reason << endl;
+    throw std::runtime_error("HA websocket disconnected");
+  }
+
   return result;
 }