Merge pull request #6 from enclave-networks/improve-perf-valueipaddre…

…ss-compare Improve correctness of ValueIpAddress compares based on family
enclave-networks · Nov 7, 2022 · 8b1636c · 8b1636c
2 parents 5931ec4 + f40fe80
commit 8b1636c
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/Enclave.FastPacket/ValueIpAddress.cs b/src/Enclave.FastPacket/ValueIpAddress.cs
@@ -141,9 +141,18 @@ public override bool Equals(object? obj)
     /// <inheritdoc />
     public bool Equals(ValueIpAddress other)
     {
+        if (_addrFamily != other._addrFamily)
+        {
+            return false;
+        }
+
+        if (_addrFamily == AddressFamily.InterNetwork)
+        {
+            return _addr1 == other._addr1;
+        }
+
         return _addr1 == other._addr1 &&
-               _addr2 == other._addr2 &&
-               _addrFamily == other._addrFamily;
+               _addr2 == other._addr2;
     }
 
     /// <inheritdoc />