Update win_inet_pton 1.0.1 (934a852) → 1.1.0 (57e3558).

CHANGES.md

@@ -15,6 +15,7 @@
 * Update Six compatibility library 1.12.0 (d927b9e) to 1.12.0 (8da94b8)
 * Update Tornado Web Server 5.1.1 (cc2cf07) to 5.1.1 (a99f1471)
 * Update unidecode module 1.0.22 (578cdb9) to 1.0.22 (a5045ab)
+* Update win_inet_pton 1.0.1 (934a852) to 1.1.0 (57e3558)
 
 
 [develop changelog]

lib/win_inet_pton/win_inet_pton.py

@@ -4,81 +4,123 @@
 # commercial or non-commercial, and by any means.
 
 import socket
-import ctypes
 import os
+import sys
 
 
-class sockaddr(ctypes.Structure):
+def inject_into_socket():
-    _fields_ = [("sa_family", ctypes.c_short),
+    import ctypes
-                ("__pad1", ctypes.c_ushort),
-                ("ipv4_addr", ctypes.c_byte * 4),
-                ("ipv6_addr", ctypes.c_byte * 16),
-                ("__pad2", ctypes.c_ulong)]
-
-if hasattr(ctypes, 'windll'):
-    WSAStringToAddressA = ctypes.windll.ws2_32.WSAStringToAddressA
-    WSAAddressToStringA = ctypes.windll.ws2_32.WSAAddressToStringA
-else:
-    def not_windows():
-        raise SystemError(
-            "Invalid platform. ctypes.windll must be available."
-        )
-    WSAStringToAddressA = not_windows
-    WSAAddressToStringA = not_windows
 
+    class in_addr(ctypes.Structure):
+        _fields_ = [("S_addr", ctypes.c_ubyte * 4)]
 
-def inet_pton(address_family, ip_string):
+    class in6_addr(ctypes.Structure):
-    addr = sockaddr()
+        _fields_ = [("Byte", ctypes.c_ubyte * 16)]
-    addr.sa_family = address_family
-    addr_size = ctypes.c_int(ctypes.sizeof(addr))
 
-    if WSAStringToAddressA(
+    if hasattr(ctypes, "windll"):
-            ip_string,
+        # InetNtopW(
-            address_family,
+        #   INT         family,
-            None,
+        #   const VOID  *pAddr,
-            ctypes.byref(addr),
+        #   PWSTR       pStringBuf,
-            ctypes.byref(addr_size)
+        #   size_t      StringBufSize
-    ) != 0:
+        # ) -> PCWSTR
-        raise socket.error(ctypes.FormatError())
+        InetNtopW = ctypes.windll.ws2_32.InetNtopW
-
+
-    if address_family == socket.AF_INET:
+        # InetPtonW(
-        return ctypes.string_at(addr.ipv4_addr, 4)
+        #   INT         family,
-    if address_family == socket.AF_INET6:
+        #   PCWSTR      pszAddrString,
-        return ctypes.string_at(addr.ipv6_addr, 16)
+        #   PVOID       pAddrBuf
-
+        # ) -> INT
-    raise socket.error('unknown address family')
+        InetPtonW = ctypes.windll.ws2_32.InetPtonW
-
+
-
+        # WSAGetLastError() -> INT
-def inet_ntop(address_family, packed_ip):
+        WSAGetLastError = ctypes.windll.ws2_32.WSAGetLastError
-    addr = sockaddr()
-    addr.sa_family = address_family
-    addr_size = ctypes.c_int(ctypes.sizeof(addr))
-    ip_string = ctypes.create_string_buffer(128)
-    ip_string_size = ctypes.c_int(ctypes.sizeof(ip_string))
-
-    if address_family == socket.AF_INET:
-        if len(packed_ip) != ctypes.sizeof(addr.ipv4_addr):
-            raise socket.error('packed IP wrong length for inet_ntoa')
-        ctypes.memmove(addr.ipv4_addr, packed_ip, 4)
-    elif address_family == socket.AF_INET6:
-        if len(packed_ip) != ctypes.sizeof(addr.ipv6_addr):
-            raise socket.error('packed IP wrong length for inet_ntoa')
-        ctypes.memmove(addr.ipv6_addr, packed_ip, 16)
     else:
-        raise socket.error('unknown address family')
 
-    if WSAAddressToStringA(
+        def not_windows():
+            raise SystemError("Invalid platform. ctypes.windll must be available.")
+
+        InetNtopW = not_windows
+        InetPtonW = not_windows
+        WSAGetLastError = not_windows
+
+    def inet_pton(address_family, ip_string):
+        if sys.version_info[0] > 2 and isinstance(ip_string, bytes):
+            raise TypeError("inet_pton() argument 2 must be str, not bytes")
+
+        if address_family == socket.AF_INET:
+            family = 2
+            addr = in_addr()
+        elif address_family == socket.AF_INET6:
+            family = 23
+            addr = in6_addr()
+        else:
+            raise OSError("unknown address family")
+
+        ip_string = ctypes.c_wchar_p(ip_string)
+        ret = InetPtonW(ctypes.c_int(family), ip_string, ctypes.byref(addr))
+
+        if ret == 1:
+            if address_family == socket.AF_INET:
+                return ctypes.string_at(addr.S_addr, 4)
+            else:
+                return ctypes.string_at(addr.Byte, 16)
+        elif ret == 0:
+            raise socket.error("illegal IP address string passed to inet_pton")
+        else:
+            err = WSAGetLastError()
+            if err == 10047:
+                e = socket.error("unknown address family")
+            elif err == 10014:
+                e = OSError("bad address")
+            else:
+                e = OSError("unknown error from inet_ntop")
+            e.errno = err
+            raise e
+
+    def inet_ntop(address_family, packed_ip):
+        if address_family == socket.AF_INET:
+            addr = in_addr()
+            if len(packed_ip) != ctypes.sizeof(addr.S_addr):
+                raise ValueError("packed IP wrong length for inet_ntop")
+
+            ctypes.memmove(addr.S_addr, packed_ip, 4)
+            buffer_len = 16
+            family = 2
+
+        elif address_family == socket.AF_INET6:
+            addr = in6_addr()
+            if len(packed_ip) != ctypes.sizeof(addr.Byte):
+                raise ValueError("packed IP wrong length for inet_ntop")
+
+            ctypes.memmove(addr.Byte, packed_ip, 16)
+            buffer_len = 46
+            family = 23
+        else:
+            raise ValueError("unknown address family")
+
+        buffer = ctypes.create_unicode_buffer(buffer_len)
+
+        ret = InetNtopW(
+            ctypes.c_int(family),
             ctypes.byref(addr),
-            addr_size,
+            ctypes.byref(buffer),
-            None,
+            ctypes.sizeof(buffer),
-            ip_string,
+        )
-            ctypes.byref(ip_string_size)
+        if ret is None:
-    ) != 0:
+            err = WSAGetLastError()
-        raise socket.error(ctypes.FormatError())
+            if err == 10047:
+                e = ValueError("unknown address family")
+            else:
+                e = OSError("unknown error from inet_ntop")
+            e.errno = err
 
-    return ip_string[:ip_string_size.value - 1]
+        return ctypes.wstring_at(buffer, buffer_len).rstrip("\x00")
 
-# Adding our two functions to the socket library
+    # Adding our two functions to the socket library
-if os.name == 'nt':
     socket.inet_pton = inet_pton
     socket.inet_ntop = inet_ntop
+
+
+if os.name == "nt" and not hasattr(socket, "inet_pton"):
+    inject_into_socket()