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:
parent
770830aa01
commit
f0fc28f0e3
@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -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;
|
||||
|
Loading…
Reference in New Issue
Block a user