arduino/esp8266
1
0
Fork 0
mirror of https://github.com/esp8266/Arduino.git synced 2025-04-21 10:26:06 +03:00
Code

Fix memory leak in WiFiClient

Browse Source
This commit is contained in:
Ivan Grokhotkov 2014-12-23 12:34:41 +03:00
parent 4b25ae86ea
commit 2328e07637
1 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
libraries/ESP8266WiFi/src/include/ClientContext.h
View File

@ -73,14 +73,17 @@ public:
void unref()
{
DEBUGV("WC:ur %d\r\n", _refcnt);
if (--_refcnt == 0 && _pcb)
if (--_refcnt == 0)
{
tcp_arg(_pcb, NULL);
tcp_sent(_pcb, NULL);
tcp_recv(_pcb, NULL);
tcp_err(_pcb, NULL);
tcp_close(_pcb);
_pcb = 0;
if (_pcb)
{
tcp_arg(_pcb, NULL);
tcp_sent(_pcb, NULL);
tcp_recv(_pcb, NULL);
tcp_err(_pcb, NULL);
tcp_close(_pcb);
_pcb = 0;
}
delete this;
}
}
@ -152,17 +155,23 @@ private:
tcp_sent(pcb, NULL);
tcp_recv(pcb, NULL);
tcp_err(pcb, NULL);
tcp_close(pcb);
int error = tcp_close(pcb);
if (error != ERR_OK)
{
DEBUGV("WC:rcla\r\n");
tcp_abort(pcb);
_pcb = 0;
return ERR_ABRT;
}
_pcb = 0;
return ERR_OK;
}
DEBUGV("WC:rcr\r\n");
size_t len = pb->len;
DEBUGV("WC:rcr %d\r\n", len);
_rx_buf.write(reinterpret_cast<const char*>(pb->payload), pb->len);
pbuf_free(pb);
tcp_recved(pcb, len);
pbuf_free(pb);
return ERR_OK;
}