Function recv

websocket/_socket.py:93–145

(sock: socket.socket, bufsize: int)

def recv(sock: socket.socket, bufsize: int) -> bytes:
    if not sock:
        raise WebSocketConnectionClosedException("socket is already closed.")

    def _recv():
        try:
            return sock.recv(bufsize)
        except SSLWantReadError:
            # Don't return None implicitly - fall through to retry logic
            pass
        except socket.error as exc:
            error_code = extract_error_code(exc)
            if error_code not in [errno.EAGAIN, errno.EWOULDBLOCK]:
                raise
            # Don't return None implicitly - fall through to retry logic

        # Retry logic using selector for both SSLWantReadError and EAGAIN/EWOULDBLOCK
        sel = selectors.DefaultSelector()
        sel.register(sock, selectors.EVENT_READ)

        r = sel.select(sock.gettimeout())
        sel.close()

        if r:
            return sock.recv(bufsize)
        else:
            # Selector timeout should raise WebSocketTimeoutException
            # not return None which gets misclassified as connection closed
            raise WebSocketTimeoutException("Connection timed out waiting for data")

    try:
        if sock.gettimeout() == 0:
            bytes_ = sock.recv(bufsize)
        else:
            bytes_ = _recv()
    except TimeoutError:
        raise WebSocketTimeoutException("Connection timed out")
    except socket.timeout as e:
        message = extract_err_message(e)
        raise WebSocketTimeoutException(message)
    except SSLError as e:
        message = extract_err_message(e)
        if isinstance(message, str) and "timed out" in message:
            raise WebSocketTimeoutException(message)
        else:
            raise

    if bytes_ is None:
        raise WebSocketConnectionClosedException("Connection to remote host was lost.")
    if not bytes_:
        raise WebSocketConnectionClosedException("Connection to remote host was lost.")

    return bytes_


def recv_line(sock: socket.socket) -> bytes: