arduino/esp8266
1
0
Fork 0
mirror of https://github.com/esp8266/Arduino.git synced 2025-06-06 05:21:22 +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() void unref()
{ {
DEBUGV("WC:ur %d\r\n", _refcnt); DEBUGV("WC:ur %d\r\n", _refcnt);
if (--_refcnt == 0 && _pcb) if (--_refcnt == 0)
{ {
tcp_arg(_pcb, NULL); if (_pcb)
tcp_sent(_pcb, NULL); {
tcp_recv(_pcb, NULL); tcp_arg(_pcb, NULL);
tcp_err(_pcb, NULL); tcp_sent(_pcb, NULL);
tcp_close(_pcb); tcp_recv(_pcb, NULL);
_pcb = 0; tcp_err(_pcb, NULL);
tcp_close(_pcb);
_pcb = 0;
}
delete this; delete this;
} }
} }
@ -152,17 +155,23 @@ private:
tcp_sent(pcb, NULL); tcp_sent(pcb, NULL);
tcp_recv(pcb, NULL); tcp_recv(pcb, NULL);
tcp_err(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; _pcb = 0;
return ERR_OK; return ERR_OK;
} }
DEBUGV("WC:rcr\r\n");
size_t len = pb->len; size_t len = pb->len;
DEBUGV("WC:rcr %d\r\n", len);
_rx_buf.write(reinterpret_cast<const char*>(pb->payload), pb->len); _rx_buf.write(reinterpret_cast<const char*>(pb->payload), pb->len);
pbuf_free(pb);
tcp_recved(pcb, len); tcp_recved(pcb, len);
pbuf_free(pb);
return ERR_OK; return ERR_OK;
} }