如何将SOCKS代理设置为websocketclient

Question

我试图用create_connection将SOCKS代理设置为websocket-client的WebSocket，但它总是在websocket发送握手请求后关闭套接字

Traceback (most recent call last):
  File "D:\PyCharmProjects\x-python\venv\lib\site-packages\websocket\_core.py", line 252, in connect
    self.handshake_response = handshake(self.sock, *addrs, **options)
  File "D:\PyCharmProjects\x-python\venv\lib\site-packages\websocket\_handshake.py", line 79, in handshake
    status, resp = _get_resp_headers(sock)
  File "D:\PyCharmProjects\x-python\venv\lib\site-packages\websocket\_handshake.py", line 162, in _get_resp_headers
    status, resp_headers, status_message = read_headers(sock)
  File "D:\PyCharmProjects\x-python\venv\lib\site-packages\websocket\_http.py", line 308, in read_headers
    line = recv_line(sock)
  File "D:\PyCharmProjects\x-python\venv\lib\site-packages\websocket\_socket.py", line 134, in recv_line
    c = recv(sock, 1)
  File "D:\PyCharmProjects\x-python\venv\lib\site-packages\websocket\_socket.py", line 125, in recv
    raise WebSocketConnectionClosedException(
websocket._exceptions.WebSocketConnectionClosedException: Connection is already closed.
python-BaseException

Process finished with exit code -1

这就是我试图用皮袜包裹插座的方式

from socks import socksocket
proxied_sock = socksocket()
proxied_sock.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
proxied_sock.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
proxied_sock = self.__proxy.set_to_socket(proxied_sock)
proxied_sock.connect((dealer_str, 80))
self.__logger.debug("WS wrapped with proxy!")
self.__ws = create_connection(self.__dealer_uri, socket=proxied_sock)

set_to_socket()函数只是一个包装代理的函数

    def set_to_socket(self, s) -> socksocket:
        """
        Sets socket's proxy
        :param s: socket.socket
        :return: Proxied socket.socket
        """
        if self.proxy_type == "http" or self.proxy_type == "https":
            s.set_proxy(HTTP, self.host, self.port, username=self.username, password=self.password)
        elif self.proxy_type == "socks4":
            s.set_proxy(SOCKS4, self.host, self.port, username=self.username, password=self.password)
        elif self.proxy_type == "socks5":
            s.set_proxy(SOCKS5, self.host, self.port, username=self.username, password=self.password)
        else:
            raise ProxyError("Proxy type not found: " + self.proxy_type)

        return s

为什么会发生这种情况？我如何用SOCKS代理包装我的websocket

Answer 1

编辑：我试图连接到SSL安全端口，所以我必须用SSL包装器包装套接字

proxied_sock = socksocket()
proxied_sock = self.__proxy.set_to_socket(proxied_sock)
dealer_tuple = (dealer_str.split(":")[0], int(dealer_str.split(":")[1]))
proxied_sock.connect(dealer_tuple)
proxied_sock = ssl.wrap_socket(proxied_sock) # We gotta wrap the socket with SSL
self.__logger.debug("Dealer client wrapped with proxy!")
self.__ws = create_connection(self.__dealer_uri, socket=proxied_sock)