Rework WiFiClient (#238)

* Rework WiFiClient

Rework WiFiClient to correct error where making a copy of a WiFiClient object resulted in the socket being closed prematurely.

Added loop and select to write to handle/prevent EAGAIN errors.

* Rework WiFiClient to use shared_ptr

Rework changes to utilize shared_ptr rather than manually maintaining reference count. 

Revert changes to write

* Incorporate comments from review

Move WiFiClientSocketHandle and fd() into WiFiClient.cpp
This commit is contained in:
David Schroeder 2017-03-01 17:47:16 -05:00 committed by Me No Dev
parent 770830aa01
commit f0fc28f0e3
2 changed files with 61 additions and 31 deletions

View File

@ -29,12 +29,33 @@
#undef write
#undef read
WiFiClient::WiFiClient():sockfd(-1),_connected(false),next(NULL)
class WiFiClientSocketHandle {
private:
int sockfd;
public:
WiFiClientSocketHandle(int fd):sockfd(fd)
{
}
~WiFiClientSocketHandle()
{
close(sockfd);
}
int fd()
{
return sockfd;
}
};
WiFiClient::WiFiClient():_connected(false),next(NULL)
{
}
WiFiClient::WiFiClient(int fd):sockfd(fd),_connected(true),next(NULL)
WiFiClient::WiFiClient(int fd):_connected(true),next(NULL)
{
clientSocketHandle.reset(new WiFiClientSocketHandle(fd));
}
WiFiClient::~WiFiClient()
@ -45,27 +66,25 @@ WiFiClient::~WiFiClient()
WiFiClient & WiFiClient::operator=(const WiFiClient &other)
{
stop();
sockfd = other.sockfd;
clientSocketHandle = other.clientSocketHandle;
_connected = other._connected;
return *this;
}
void WiFiClient::stop()
{
if(_connected && sockfd >= 0) {
close(sockfd);
sockfd = -1;
_connected = false;
}
clientSocketHandle = NULL;
_connected = false;
}
int WiFiClient::connect(IPAddress ip, uint16_t port)
{
sockfd = socket(AF_INET, SOCK_STREAM, 0);
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
log_e("socket: %d", errno);
return 0;
}
uint32_t ip_addr = ip;
struct sockaddr_in serveraddr;
bzero((char *) &serveraddr, sizeof(serveraddr));
@ -76,9 +95,9 @@ int WiFiClient::connect(IPAddress ip, uint16_t port)
if (res < 0) {
log_e("lwip_connect_r: %d", errno);
close(sockfd);
sockfd = -1;
return 0;
}
clientSocketHandle.reset(new WiFiClientSocketHandle(sockfd));
_connected = true;
return 1;
}
@ -96,7 +115,7 @@ int WiFiClient::connect(const char *host, uint16_t port)
int WiFiClient::setSocketOption(int option, char* value, size_t len)
{
int res = setsockopt(sockfd, SOL_SOCKET, option, value, len);
int res = setsockopt(fd(), SOL_SOCKET, option, value, len);
if(res < 0) {
log_e("%d", errno);
}
@ -116,7 +135,7 @@ int WiFiClient::setTimeout(uint32_t seconds)
int WiFiClient::setOption(int option, int *value)
{
int res = setsockopt(sockfd, IPPROTO_TCP, option, (char *)value, sizeof(int));
int res = setsockopt(fd(), IPPROTO_TCP, option, (char *) value, sizeof(int));
if(res < 0) {
log_e("%d", errno);
}
@ -126,7 +145,7 @@ int WiFiClient::setOption(int option, int *value)
int WiFiClient::getOption(int option, int *value)
{
size_t size = sizeof(int);
int res = getsockopt(sockfd, IPPROTO_TCP, option, (char *)value, &size);
int res = getsockopt(fd(), IPPROTO_TCP, option, (char *)value, &size);
if(res < 0) {
log_e("%d", errno);
}
@ -209,7 +228,7 @@ int WiFiClient::read(uint8_t *buf, size_t size)
if(!available()) {
return -1;
}
int res = recv(sockfd, buf, size, MSG_DONTWAIT);
int res = recv(fd(), buf, size, MSG_DONTWAIT);
if(res < 0 && errno != EWOULDBLOCK) {
log_e("%d", errno);
stop();
@ -223,7 +242,7 @@ int WiFiClient::available()
return 0;
}
int count;
int res = ioctl(sockfd, FIONREAD, &count);
int res = ioctl(fd(), FIONREAD, &count);
if(res < 0) {
log_e("%d", errno);
stop();
@ -239,7 +258,7 @@ uint8_t WiFiClient::connected()
return _connected;
}
IPAddress WiFiClient::remoteIP(int fd)
IPAddress WiFiClient::remoteIP(int fd) const
{
struct sockaddr_storage addr;
socklen_t len = sizeof addr;
@ -248,7 +267,7 @@ IPAddress WiFiClient::remoteIP(int fd)
return IPAddress((uint32_t)(s->sin_addr.s_addr));
}
uint16_t WiFiClient::remotePort(int fd)
uint16_t WiFiClient::remotePort(int fd) const
{
struct sockaddr_storage addr;
socklen_t len = sizeof addr;
@ -257,17 +276,27 @@ uint16_t WiFiClient::remotePort(int fd)
return ntohs(s->sin_port);
}
IPAddress WiFiClient::remoteIP()
IPAddress WiFiClient::remoteIP() const
{
return remoteIP(sockfd);
return remoteIP(fd());
}
uint16_t WiFiClient::remotePort()
uint16_t WiFiClient::remotePort() const
{
return remotePort(sockfd);
return remotePort(fd());
}
bool WiFiClient::operator==(const WiFiClient& rhs)
{
return sockfd == rhs.sockfd && remotePort(sockfd) == remotePort(rhs.sockfd) && remoteIP(sockfd) == remoteIP(rhs.sockfd);
return clientSocketHandle == rhs.clientSocketHandle && remotePort() == rhs.remotePort() && remoteIP() == rhs.remoteIP();
}
int WiFiClient::fd() const
{
if (clientSocketHandle == NULL) {
return -1;
} else {
return clientSocketHandle->fd();
}
}

View File

@ -23,11 +23,14 @@
#include "Arduino.h"
#include "Client.h"
#include <memory>
class WiFiClientSocketHandle;
class WiFiClient : public Client
{
protected:
int sockfd;
std::shared_ptr<WiFiClientSocketHandle> clientSocketHandle;
bool _connected;
public:
@ -69,12 +72,8 @@ public:
return !this->operator==(rhs);
};
int fd()
{
return sockfd;
}
IPAddress remoteIP();
uint16_t remotePort();
int fd() const;
int setSocketOption(int option, char* value, size_t len);
int setOption(int option, int *value);
int getOption(int option, int *value);
@ -82,8 +81,10 @@ public:
int setNoDelay(bool nodelay);
bool getNoDelay();
IPAddress remoteIP(int fd);
uint16_t remotePort(int fd);
IPAddress remoteIP() const;
IPAddress remoteIP(int fd) const;
uint16_t remotePort() const;
uint16_t remotePort(int fd) const;
//friend class WiFiServer;
using Print::write;