如何將 SOCKS 代理設置為 websocket-client

Question

我正在嘗試使用create_connection為websocket-client的WebSocket設置一個 SOCKS 代理，但它總是在 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

這就是我試圖用 PySocks 包裝套接字的方式。

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() function 只是一個 function 來包裝代理。

    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)