Remote tcp disconnect not detected (espressif#389)

* Add setNoDelay and getNoDelay to WiFiServer class * Remote TCP disconnect not detected

Author: bbx10

libraries/WiFi/src/WiFiClient.cpp

@@ -279,8 +279,28 @@ void WiFiClient::flush(){
 
 uint8_tWiFiClient::connected()
 {
-uint8_t dummy = 0;
-read(&dummy, 0);
+if (_connected){
+uint8_t dummy;
+int res = recv(fd(), &dummy, 0, MSG_DONTWAIT);
+if (res <= 0){
+switch (errno){
+case ENOTCONN:
+case EPIPE:
+case ECONNRESET:
+case ECONNREFUSED:
+case ECONNABORTED:
+ _connected = false;
+break;
+default:
+ _connected = true;
+break;
+ }
+ }
+else{
+// Should never happen since requested 0 bytes
+ _connected = true;
+ }
+ }
 return _connected;
 }
 

libraries/WiFi/src/WiFiServer.cpp

@@ -45,8 +45,11 @@ WiFiClient WiFiServer::available(){
 int client_sock = accept(sockfd, (structsockaddr *)&_client, (socklen_t*)&cs);
 if(client_sock >= 0){
 int val = 1;
-if(setsockopt(client_sock, SOL_SOCKET, SO_KEEPALIVE, (char*)&val, sizeof(int)) == ESP_OK)
-returnWiFiClient(client_sock);
+if(setsockopt(client_sock, SOL_SOCKET, SO_KEEPALIVE, (char*)&val, sizeof(int)) == ESP_OK){
+ val = _noDelay;
+if(setsockopt(client_sock, IPPROTO_TCP, TCP_NODELAY, (char*)&val, sizeof(int)) == ESP_OK)
+returnWiFiClient(client_sock);
+ }
  }
 returnWiFiClient();
 }
@@ -69,6 +72,14 @@ void WiFiServer::begin(){
  _listening = true;
 }
 
+voidWiFiServer::setNoDelay(bool nodelay){
+ _noDelay = nodelay;
+}
+
+boolWiFiServer::getNoDelay(){
+return _noDelay;
+}
+
 voidWiFiServer::end(){
 close(sockfd);
  sockfd = -1;

libraries/WiFi/src/WiFiServer.h

@@ -29,6 +29,7 @@ class WiFiServer : public Server{
 uint16_t _port;
 uint8_t _max_clients;
 bool _listening;
+bool _noDelay = false;
 
 public:
 voidlistenOnLocalhost(){}
@@ -38,6 +39,8 @@ class WiFiServer : public Server{
  WiFiClient available();
  WiFiClient accept(){returnavailable()}
 voidbegin();
+voidsetNoDelay(bool nodelay);
+boolgetNoDelay();
 size_twrite(constuint8_t *data, size_t len);
 size_twrite(uint8_t data){
 returnwrite(&data, 1);