Correct addrs sorting to be by TCP/UDP

Fixes any potential cases where the resolver might yield UDP addresses first. Signed-off-by: Robbie Harwood <[email protected]>
latchset · Aug 28, 2019 · d0b35c2 · d0b35c2
1 parent 58b3676
commit d0b35c2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kdcproxy/__init__.py b/kdcproxy/__init__.py
@@ -227,7 +227,8 @@ def __call__(self, env, start_response):
                 #
                 # Stick a None address on the end so we can get one
                 # more attempt after all servers have been contacted.
-                addrs = tuple(sorted(filter(self.__filter_addr, addrs)))
+                addrs = tuple(sorted(filter(self.__filter_addr, addrs),
+                                     key=lambda a: a[2]))
                 for addr in addrs + (None,):
                     if addr is not None:
                         # Bypass unspecified socktypes