Home-IoT/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2024-11-26 00:48:19 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Re-enable TCP nodelay for ESP32 (#2390)

Browse source
This commit is contained in:
Otto Winter 2021-09-27 11:40:28 +02:00 committed by GitHub
parent 756c6721e9
commit 97e76d64d6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 30 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
esphome/components/api/api_frame_helper.cpp
View file

@ -127,6 +127,14 @@ APIError APINoiseFrameHelper::init() {
return APIError::TCP_NONBLOCKING_FAILED; return APIError::TCP_NONBLOCKING_FAILED;
} }
int enable = 1;
err = socket_->setsockopt(IPPROTO_TCP, TCP_NODELAY, &enable, sizeof(int));
if (err != 0) {
state_ = State::FAILED;
HELPER_LOG("Setting nodelay failed with errno %d", errno);
return APIError::TCP_NODELAY_FAILED;
}
// init prologue // init prologue
prologue_.insert(prologue_.end(), PROLOGUE_INIT, PROLOGUE_INIT + strlen(PROLOGUE_INIT)); prologue_.insert(prologue_.end(), PROLOGUE_INIT, PROLOGUE_INIT + strlen(PROLOGUE_INIT));
@ -722,6 +730,13 @@ APIError APIPlaintextFrameHelper::init() {
HELPER_LOG("Setting nonblocking failed with errno %d", errno); HELPER_LOG("Setting nonblocking failed with errno %d", errno);
return APIError::TCP_NONBLOCKING_FAILED; return APIError::TCP_NONBLOCKING_FAILED;
} }
int enable = 1;
err = socket_->setsockopt(IPPROTO_TCP, TCP_NODELAY, &enable, sizeof(int));
if (err != 0) {
state_ = State::FAILED;
HELPER_LOG("Setting nodelay failed with errno %d", errno);
return APIError::TCP_NODELAY_FAILED;
}
state_ = State::DATA; state_ = State::DATA;
return APIError::OK; return APIError::OK;

15
esphome/components/socket/lwip_raw_tcp_impl.cpp
View file

@ -257,7 +257,7 @@ class LWIPRawImpl : public Socket {
errno = EINVAL; errno = EINVAL;
return -1; return -1;
} }
*reinterpret_cast<int *>(optval) = tcp_nagle_disabled(pcb_); *reinterpret_cast<int *>(optval) = nodelay_;
*optlen = 4; *optlen = 4;
return 0; return 0;
} }
@ -286,11 +286,7 @@ class LWIPRawImpl : public Socket {
return -1; return -1;
} }
int val = *reinterpret_cast<const int *>(optval); int val = *reinterpret_cast<const int *>(optval);
if (val != 0) { nodelay_ = val;
tcp_nagle_disable(pcb_);
} else {
tcp_nagle_enable(pcb_);
}
return 0; return 0;
} }
@ -444,9 +440,11 @@ class LWIPRawImpl : public Socket {
if (written == 0) if (written == 0)
// no need to output if nothing written // no need to output if nothing written
return 0; return 0;
if (nodelay_) {
int err = internal_output(); int err = internal_output();
if (err == -1) if (err == -1)
return -1; return -1;
}
return written; return written;
} }
ssize_t writev(const struct iovec *iov, int iovcnt) override { ssize_t writev(const struct iovec *iov, int iovcnt) override {
@ -466,9 +464,11 @@ class LWIPRawImpl : public Socket {
if (written == 0) if (written == 0)
// no need to output if nothing written // no need to output if nothing written
return 0; return 0;
if (nodelay_) {
int err = internal_output(); int err = internal_output();
if (err == -1) if (err == -1)
return -1; return -1;
}
return written; return written;
} }
int setblocking(bool blocking) override { int setblocking(bool blocking) override {
@ -550,6 +550,9 @@ class LWIPRawImpl : public Socket {
bool rx_closed_ = false; bool rx_closed_ = false;
pbuf *rx_buf_ = nullptr; pbuf *rx_buf_ = nullptr;
size_t rx_buf_offset_ = 0; size_t rx_buf_offset_ = 0;
// don't use lwip nodelay flag, it sometimes causes reconnect
// instead use it for determining whether to call lwip_output
bool nodelay_ = false;
}; };
std::unique_ptr<Socket> socket(int domain, int type, int protocol) { std::unique_ptr<Socket> socket(int domain, int type, int protocol) {