altcp: added altcp_get_port()

src/core/altcp.c

@@ -363,6 +363,15 @@ altcp_get_ip(struct altcp_pcb *conn, int local)
   return NULL;
 }
 
+u16_t
+altcp_get_port(struct altcp_pcb *conn, int local)
+{
+  if (conn && conn->fns && conn->fns->getport) {
+    return conn->fns->getport(conn, local);
+  }
+  return 0;
+}
+
 #ifdef LWIP_DEBUG
 enum tcp_state
 altcp_dbg_get_tcp_state(struct altcp_pcb *conn)
@@ -513,6 +522,14 @@ altcp_default_get_ip(struct altcp_pcb *conn, int local)
   return NULL;
 }
 
+u16_t
+altcp_default_get_port(struct altcp_pcb *conn, int local)
+{
+  if (conn && conn->inner_conn) {
+    return altcp_get_port(conn->inner_conn, local);
+  }
+  return 0;
+}
 
 #ifdef LWIP_DEBUG
 enum tcp_state

src/core/altcp_tcp.c

@@ -425,6 +425,23 @@ altcp_tcp_get_ip(struct altcp_pcb *conn, int local)
   return NULL;
 }
 
+static u16_t
+altcp_tcp_get_port(struct altcp_pcb *conn, int local)
+{
+  if (conn) {
+    struct tcp_pcb *pcb = (struct tcp_pcb *)conn->state;
+    ALTCP_TCP_ASSERT_CONN(conn);
+    if (pcb) {
+      if (local) {
+        return pcb->local_port;
+      } else {
+        return pcb->remote_port;
+      }
+    }
+  }
+  return 0;
+}
+
 #ifdef LWIP_DEBUG
 static enum tcp_state
 altcp_tcp_dbg_get_tcp_state(struct altcp_pcb *conn)
@@ -459,7 +476,8 @@ const struct altcp_functions altcp_tcp_functions = {
   altcp_tcp_setprio,
   altcp_tcp_dealloc,
   altcp_tcp_get_tcp_addrinfo,
-  altcp_tcp_get_ip
+  altcp_tcp_get_ip,
+  altcp_tcp_get_port
 #ifdef LWIP_DEBUG
   ,altcp_tcp_dbg_get_tcp_state
 #endif