Python TCP Socket接收文件问题：无法识别字节流结束

这是TCP字节流协议的典型问题——因为TCP本身没有内置的“数据边界”或“传输结束”标识，服务端根本不知道你已经完成了文件传输，所以会一直卡在conn.recv(1024)调用里阻塞等待更多数据，永远不会返回空字节串。下面给你几个实用的解决方案，按推荐程度排序：

方案1：客户端发送完文件后关闭输出流（推荐，简单直接） #

当Java客户端完成文件字节的发送后，调用socket.shutdownOutput()来关闭输出流。这个操作会向服务端发送一个FIN报文，告诉对方“我这边不会再发数据了”，此时Python服务端的conn.recv(1024)就会返回空字节串b''，你的while循环就能正常退出。

Java客户端修改示例： #

// 假设你已经完成了文件的输出流写入
OutputStream out = socket.getOutputStream();
// 写入文件字节...
out.flush();
// 关键：关闭输出流，发送结束信号
socket.shutdownOutput();
// 之后可以保持连接做其他操作，或者关闭socket

Python服务端原方法修改（只需要保持原循环逻辑即可）： #

def getFile(conn):
    with open("received_file", "wb") as f:
        while True:
            data = conn.recv(1024)
            if not data:  # 收到空字节串，说明客户端已结束发送
                break
            f.write(data)

方案2：先传输文件长度，再传输文件内容（更灵活，支持多轮通信） #

如果之后还需要用同一个TCP连接做其他通信（不想关闭连接），可以先把文件的总字节数发送给服务端，服务端先接收这个长度，然后精确接收对应字节数的内容，就不需要依赖连接关闭来判断结束了。

Java客户端步骤： #

1. 计算文件的总字节数
2. 将字节数转换为固定长度的字节（比如4字节的int）发送给服务端
3. 然后发送文件的实际内容

File file = new File("your_file_path");
long fileSize = file.length();
OutputStream out = socket.getOutputStream();

// 发送文件长度（这里用4字节的int，注意大端小端，Java默认大端）
out.write((byte)(fileSize >> 24));
out.write((byte)(fileSize >> 16));
out.write((byte)(fileSize >> 8));
out.write((byte)fileSize);

// 发送文件内容
FileInputStream fis = new FileInputStream(file);
byte[] buffer = new byte[1024];
int len;
while ((len = fis.read(buffer)) != -1) {
    out.write(buffer, 0, len);
}
fis.close();
out.flush();

Python服务端修改： #

def getFile(conn):
    # 先接收文件长度（4字节）
    size_data = b''
    while len(size_data) < 4:
        chunk = conn.recv(4 - len(size_data))
        if not chunk:
            # 连接意外关闭，处理异常
            return
        size_data += chunk
    file_size = int.from_bytes(size_data, byteorder='big')  # 和Java的大端对应

    # 接收指定字节数的文件内容
    received_size = 0
    with open("received_file", "wb") as f:
        while received_size < file_size:
            # 计算剩余需要接收的字节数，避免最后一次recv多取数据
            remaining = file_size - received_size
            chunk_size = min(1024, remaining)
            data = conn.recv(chunk_size)
            if not data:
                # 连接意外中断，文件不完整
                break
            f.write(data)
            received_size += len(data)

方案3：使用自定义结束标记（不推荐，存在冲突风险） #

发送完文件后，发送一个约定好的特殊字节序列（比如b'END_OF_FILE'），服务端在接收过程中检查是否收到这个标记。但要注意：如果文件内容本身包含这个标记，会导致服务端提前终止接收，所以只适合你能确保文件内容不会包含该标记的场景。

示例（仅作参考）： #

Java客户端发送完文件后添加：

out.write("END_OF_FILE".getBytes());
out.flush();

Python服务端修改：

def getFile(conn):
    end_marker = b'END_OF_FILE'
    buffer = b''
    with open("received_file", "wb") as f:
        while True:
            data = conn.recv(1024)
            if not data:
                break
            buffer += data
            # 检查缓冲区末尾是否包含结束标记
            if buffer.endswith(end_marker):
                # 写入标记之前的内容
                f.write(buffer[:-len(end_marker)])
                break
            # 如果缓冲区足够大，先写入部分内容，避免内存溢出
            if len(buffer) > len(end_marker) * 2:
                f.write(buffer[:-len(end_marker)])
                buffer = buffer[-len(end_marker):]

内容的提问来源于stack exchange，提问作者willh99