补全:数收到几行

👁️ 2 人浏览 💬 0 人评论 ❤️ 添加收藏

补全 count_lines：把连接里的字节全收下来（到对面关），数其中有几个换行＝几整行。

贯穿全条的内存版 TCP（判题机不联网；接口和真 socket 一样，真机上换成 socket.socket() 就是真的）：net = Net()；lis = listen(net, 端口) 开监听；cli = connect(net, 端口) 客户端连上（交回客户端这头，服务端那头进 accept 队列）；srv = lis.accept() 领一条连接；sendall 把字节追加进对面收缓冲，recv(n) 从自己收缓冲取最多 n 字节（拿到多少算多少，recv 交回 b"" 表示对面关了）。

断连与错误：recv_line(sock) 攒到换行为止交回这一行；closed(sock) 判对面关没；safe_send(sock,data) 对面已关就不发、交回 False。

提交你的答案

请登录后提交答案。

去登录

← 补全:关了就别发 HTTP 请求行长什么样 →

更多题目

代码编辑器

语言： Ctrl + Enter 运行

import collections

class Net:
    def __init__(self):
        self.listeners = {}

class Endpoint:
    def __init__(self, name):
        self.name = name
        self.rbuf = b""
        self.peer = None
        self.peer_closed = False
        self.closed = False

def send(self, data):
        if self.closed or self.peer is None:
            raise BrokenPipeError("连接已关")
        self.peer.rbuf += bytes(data)
        return len(data)

def sendall(self, data):
        self.send(data)

def recv(self, bufsize):
        if self.rbuf:
            out, self.rbuf = self.rbuf[:bufsize], self.rbuf[bufsize:]
            return out
        if self.peer_closed:
            return b""
        raise BlockingIOError("暂时没有数据（真 socket 会在这里阻塞等）")

def close(self):
        self.closed = True
        if self.peer is not None:
            self.peer.peer_closed = True

class Listener:
    def __init__(self):
        self.backlog = collections.deque()

def accept(self):
        if not self.backlog:
            raise BlockingIOError("暂时没有新连接")
        return self.backlog.popleft()

def listen(net, port):
    lis = Listener()
    net.listeners[port] = lis
    return lis

def connect(net, port):
    if port not in net.listeners:
        raise ConnectionRefusedError(111, "Connection refused")
    cli = Endpoint("client")
    srv = Endpoint("server")
    cli.peer = srv
    srv.peer = cli
    net.listeners[port].backlog.append(srv)
    return cli

def recv_line(sock):
    # 攒字节到一个换行为止，交回这一行（不含换行）；对面关了又没换行，交回已收到的
    buf = b""
    while True:
        try:
            chunk = sock.recv(4096)
        except BlockingIOError:
            return buf
        if not chunk:
            return buf
        buf += chunk
        if b"\n" in buf:
            return buf.split(b"\n", 1)[0]

def closed(sock):
    # 对面关了吗（recv 交回空就是关了）
    return sock.peer_closed and not sock.rbuf

def safe_send(sock, data):
    # 对面已关就别发（返回 False），否则发出去返回 True
    if sock.closed or sock.peer_closed:
        return False
    sock.sendall(data)
    return True

def count_lines(sock):
    buf = b""
    while True:
        try:
            chunk = sock.recv(4096)
        except BlockingIOError:
            break
        if not chunk:
            break
        buf += chunk
    # TODO: 收下的字节里有几个换行就是几整行
    return 0

net = Net()
lis = listen(net, 9000)
cli = connect(net, 9000)
srv = lis.accept()
cli.sendall(b"a\nb\n")
cli.close()
print(count_lines(srv))

本次输入：

输出：

👩‍🏫

AI

💬 题目评论

全部评论