SocketWrapper: implement peek() and connected()

Author: facchinm

libraries/SocketWrapper/SocketWrapper.h

@@ -22,7 +22,7 @@ class ZephyrSocketWrapper {
 	    hints.ai_family = AF_INET;
 	    hints.ai_socktype = SOCK_STREAM;
 
-        int resolve_attempts = 10;
+        int resolve_attempts = 100;
         int ret;
 
 	    while (resolve_attempts--) {
@@ -80,11 +80,11 @@ class ZephyrSocketWrapper {
         return count;
     }
 
-    int recv(uint8_t* buffer, size_t size) {
+    int recv(uint8_t* buffer, size_t size, int flags = MSG_DONTWAIT) {
         if (sock_fd == -1) {
             return -1;
         }
-        return ::recv(sock_fd, buffer, size, MSG_DONTWAIT);
+        return ::recv(sock_fd, buffer, size, flags);
     }
 
     int send(const uint8_t* buffer, size_t size) {

libraries/SocketWrapper/ZephyrClient.h

@@ -4,15 +4,26 @@
 #include "zephyr/sys/printk.h"
 
 class ZephyrClient : public arduino::Client, ZephyrSocketWrapper {
+private:
+    bool _connected = false;
+
 public:
     int connect(const char* host, uint16_t port) override {
-        return ZephyrSocketWrapper::connect((char*)host, port);
+        auto ret = ZephyrSocketWrapper::connect((char*)host, port);
+        if (ret) {
+            _connected = true;
+        }
+        return ret;
     }
     int connect(IPAddress ip, uint16_t port) {
-        return ZephyrSocketWrapper::connect(ip, port);
+        auto ret = ZephyrSocketWrapper::connect(ip, port);
+        if (ret) {
+            _connected = true;
+        }
+        return ret;
     }
     uint8_t connected() override {
-        return sock_fd != -1;
+        return _connected;
     }
     int available() override {
         return ZephyrSocketWrapper::available();
@@ -46,10 +57,13 @@ class ZephyrClient : public arduino::Client, ZephyrSocketWrapper {
         // No-op
     }
     int peek() override {
-        // No-op
+        uint8_t c;
+        recv(&c, 1, MSG_PEEK);
+        return c;
     }
     void stop() override {
         ZephyrSocketWrapper::close();
+        _connected = false;
     }
     operator bool() {
         return sock_fd != -1;