diff options
-rw-r--r-- | hicn-light/src/io/udpListener.c | 14 | ||||
-rw-r--r-- | hicn-light/src/io/udpTunnel.c | 14 |
2 files changed, 28 insertions, 0 deletions
diff --git a/hicn-light/src/io/udpListener.c b/hicn-light/src/io/udpListener.c index c49571dd4..fefe116f9 100644 --- a/hicn-light/src/io/udpListener.c +++ b/hicn-light/src/io/udpListener.c @@ -29,6 +29,7 @@ #include <src/io/udpConnection.h> #include <src/io/udpListener.h> +#include <parc/algol/parc_Network.h> #include <parc/algol/parc_Memory.h> #include <parc/assert/parc_Assert.h> #include <src/core/connection.h> @@ -360,7 +361,20 @@ static bool _lookupConnectionId(UdpListener *udp, AddressPair *pair, static unsigned _createNewConnection(UdpListener *udp, int fd, const AddressPair *pair) { + //check it the connection is local bool isLocal = false; + const Address *localAddress = addressPair_GetLocal(pair); + if(addressGetType(localAddress) == ADDR_INET){ + struct sockaddr_in tmpAddr; + addressGetInet(localAddress, &tmpAddr); + if(parcNetwork_IsSocketLocal((struct sockaddr *)&tmpAddr)) + isLocal = true; + }else{ + struct sockaddr_in6 tmpAddr6; + addressGetInet6(localAddress, &tmpAddr6); + if(parcNetwork_IsSocketLocal((struct sockaddr *)&tmpAddr6)) + isLocal = true; + } // metisUdpConnection_Create takes ownership of the pair IoOperations *ops = udpConnection_Create(udp->forwarder, fd, pair, isLocal); diff --git a/hicn-light/src/io/udpTunnel.c b/hicn-light/src/io/udpTunnel.c index d06a35ce6..d49f43f05 100644 --- a/hicn-light/src/io/udpTunnel.c +++ b/hicn-light/src/io/udpTunnel.c @@ -19,6 +19,7 @@ #include <stdio.h> #include <string.h> +#include <parc/algol/parc_Network.h> #include <parc/assert/parc_Assert.h> #include <src/io/udpConnection.h> #include <src/io/udpListener.h> @@ -42,7 +43,20 @@ IoOperations *udpTunnel_CreateOnListener(Forwarder *forwarder, if (localType == remoteType) { AddressPair *pair = addressPair_Create(localAddress, remoteAddress); + + //check it the connection is local bool isLocal = false; + if(localType == ADDR_INET){ + struct sockaddr_in tmpAddr; + addressGetInet(localAddress, &tmpAddr); + if(parcNetwork_IsSocketLocal((struct sockaddr *)&tmpAddr)) + isLocal = true; + }else{ + struct sockaddr_in6 tmpAddr6; + addressGetInet6(localAddress, &tmpAddr6); + if(parcNetwork_IsSocketLocal((struct sockaddr *)&tmpAddr6)) + isLocal = true; + } int fd = localListener->getSocket(localListener); // udpListener_SetPacketType(localListener, // MessagePacketType_ContentObject); |