jacob.eva/arduino-esp32
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle partial socket send (#503)

Browse Source 
send() can return a value > 0 but less than size indicating it was able to accept some of the data in buffer. The caller must try again after updating the buffer pointer and size remaining.
This commit is contained in:
bbx10 2017-07-31 09:51:41 -10:00 committed by Me No Dev
parent 2eaf846134
commit 37ef5a43e3
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
libraries/WiFi/src/WiFiClient.cpp
View File

@ -186,6 +186,8 @@ size_t WiFiClient::write(const uint8_t *buf, size_t size)
int res =0;
int retry = WIFI_CLIENT_MAX_WRITE_RETRY;
int socketFileDescriptor = fd();
size_t totalBytesSent = 0;
size_t bytesRemaining = size;
if(!_connected || (socketFileDescriptor < 0)) {
return 0;
@ -206,8 +208,18 @@ size_t WiFiClient::write(const uint8_t *buf, size_t size)
}
if(FD_ISSET(socketFileDescriptor, &set)) {
res = send(socketFileDescriptor, (void*) buf, size, MSG_DONTWAIT);
if(res < 0) {
res = send(socketFileDescriptor, (void*) buf, bytesRemaining, MSG_DONTWAIT);
if(res > 0) {
totalBytesSent += res;
if (totalBytesSent >= size) {
//completed successfully
retry = 0;
} else {
buf += res;
bytesRemaining -= res;
}
}
else if(res < 0) {
log_e("%d", errno);
if(errno != EAGAIN) {
//if resource was busy, can try again, otherwise give up
@ -215,13 +227,13 @@ size_t WiFiClient::write(const uint8_t *buf, size_t size)
res = 0;
retry = 0;
}
} else {
//completed successfully
retry = 0;
}
else {
// Try again
}
}
}
return res;
return totalBytesSent;
}
size_t WiFiClient::write_P(PGM_P buf, size_t size)