From 59e0fa52c50d7799c363abdaeb5578f4aabbcb9d Mon Sep 17 00:00:00 2001
From: Sandor Nemes <sandor.nemes@fireeye.com>
Date: Mon, 5 Aug 2019 22:13:07 +0200
Subject: [PATCH] Use self.getsockopt instead of self.type to work around a
 Python 3.5/3.6 bug

---
 socks.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/socks.py b/socks.py
index d678fa8..82b67d5 100644
--- a/socks.py
+++ b/socks.py
@@ -329,7 +329,7 @@ def bind(self, *pos, **kw):
         Happens during the bind() phase."""
         (proxy_type, proxy_addr, proxy_port, rdns, username,
          password) = self.proxy
-        if not proxy_type or self.type != socket.SOCK_DGRAM:
+        if not proxy_type or self.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE) != socket.SOCK_DGRAM:
             return _orig_socket.bind(self, *pos, **kw)
 
         if self._proxyconn:
@@ -361,7 +361,7 @@ def bind(self, *pos, **kw):
         self.proxy_sockname = ("0.0.0.0", 0)  # Unknown
 
     def sendto(self, bytes, *args, **kwargs):
-        if self.type != socket.SOCK_DGRAM:
+        if self.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE) != socket.SOCK_DGRAM:
             return super(socksocket, self).sendto(bytes, *args, **kwargs)
         if not self._proxyconn:
             self.bind(("", 0))
@@ -381,13 +381,13 @@ def sendto(self, bytes, *args, **kwargs):
         return sent - header.tell()
 
     def send(self, bytes, flags=0, **kwargs):
-        if self.type == socket.SOCK_DGRAM:
+        if self.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE) == socket.SOCK_DGRAM:
             return self.sendto(bytes, flags, self.proxy_peername, **kwargs)
         else:
             return super(socksocket, self).send(bytes, flags, **kwargs)
 
     def recvfrom(self, bufsize, flags=0):
-        if self.type != socket.SOCK_DGRAM:
+        if self.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE) != socket.SOCK_DGRAM:
             return super(socksocket, self).recvfrom(bufsize, flags)
         if not self._proxyconn:
             self.bind(("", 0))
@@ -744,7 +744,7 @@ def connect(self, dest_pair, catch_errors=None):
 
         dest_addr, dest_port = dest_pair
 
-        if self.type == socket.SOCK_DGRAM:
+        if self.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE) == socket.SOCK_DGRAM:
             if not self._proxyconn:
                 self.bind(("", 0))
             dest_addr = socket.gethostbyname(dest_addr)