LocalIP/LocalPort support

niltonfelipe · Aug 19, 2015 · 3078103 · 3078103
1 parent 15f64eb
commit 3078103
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.cpp b/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.cpp
@@ -263,6 +263,22 @@ uint16_t WiFiClient::remotePort()
     return _client->getRemotePort();
 }
 
+IPAddress WiFiClient::localIP()
+{
+    if (!_client)
+        return IPAddress(0U);
+
+    return IPAddress(_client->getLocalAddress());
+}
+
+uint16_t WiFiClient::localPort()
+{
+    if (!_client)
+        return 0;
+
+    return _client->getLocalPort();
+}
+
 int8_t WiFiClient::_s_connected(void* arg, void* tpcb, int8_t err)
 {
     return reinterpret_cast<WiFiClient*>(arg)->_connected(tpcb, err);

diff --git a/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.h b/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/WiFiClient.h
@@ -60,6 +60,8 @@ class WiFiClient : public Client, public SList<WiFiClient> {
 
   IPAddress remoteIP();
   uint16_t  remotePort();
+  IPAddress localIP();
+  uint16_t  localPort();
   bool getNoDelay();
   void setNoDelay(bool nodelay);
   static void setLocalPortStart(uint16_t port) { _localPort = port; }

diff --git a/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/include/ClientContext.h b/hardware/esp8266com/esp8266/libraries/ESP8266WiFi/src/include/ClientContext.h
@@ -125,6 +125,18 @@ class ClientContext {
             return _pcb->remote_port;
         }
 
+        uint32_t getLocalAddress() {
+            if(!_pcb) return 0;
+
+            return _pcb->local_ip.addr;
+        }
+
+        uint16_t getLocalPort() {
+            if(!_pcb) return 0;
+
+            return _pcb->local_port;
+        }
+
         size_t getSize() const {
             if(!_rx_buf) return 0;