Add API to set the beginning of local ports range for WiFiClient

slootsky · Jun 2, 2015 · a965c76 · a965c76
1 parent bb462ea
commit a965c76
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.cpp b/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.cpp
@@ -40,6 +40,8 @@ extern "C"
 #include "include/ClientContext.h"
 #include "c_types.h"
 
+uint16_t WiFiClient::_localPort = 0;
+
 ICACHE_FLASH_ATTR WiFiClient::WiFiClient() 
 : _client(0)
 {
@@ -93,6 +95,10 @@ int ICACHE_FLASH_ATTR  WiFiClient::connect(IPAddress ip, uint16_t port)
     if (!pcb)
         return 0;
 
+    if (_localPort > 0) {
+        pcb->local_port = _localPort++;
+    }
+
     ip_addr_t addr;
     addr.addr = ip;
     tcp_arg(pcb, this);

diff --git a/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.h b/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.h
@@ -61,6 +61,7 @@ class WiFiClient : public Client {
   uint16_t  remotePort();
   bool getNoDelay();
   void setNoDelay(bool nodelay);
+  static void setLocalPortStart(uint16_t port) { _localPort = port; }
 
   template<typename T> size_t write(T &src){
     uint8_t obuf[1460];
@@ -97,6 +98,7 @@ class WiFiClient : public Client {
   void _err(int8_t err);
 
   ClientContext* _client;
+  static uint16_t _localPort;
 
 };