Add error handling to “available” function

sebromero · sebromero · commit ba13416c6c97 · 2020-08-13T12:48:01.000+02:00
diff --git a/libraries/WiFi/src/WiFiServer.cpp b/libraries/WiFi/src/WiFiServer.cpp
@@ -33,8 +33,12 @@ size_t arduino::WiFiServer::write(const uint8_t *buf, size_t size) {
 }
 
 arduino::WiFiClient arduino::WiFiServer::available(uint8_t* status) {
-	WiFiClient ret;
-	TCPSocket* client = sock->accept();
-	ret.setSocket(client);
-	return ret;
+	WiFiClient client;
+	nsapi_error_t error;
+	TCPSocket* clientSocket = sock->accept(&error);
+	if(status != nullptr) {
+		*status = error == NSAPI_ERROR_OK ? 1 : 0;
+	}
+	client.setSocket(clientSocket);
+	return client;
 }
