解决BitTorrent客户端Peer握手协议中的struct.pack打包错误

原代码核心错误分析 #

1. 套接字类型错误：BitTorrent Peer间握手与通信基于TCP协议，但原代码使用了SOCK_DGRAM（UDP套接字），这会导致根本无法完成握手流程。
2. struct.pack格式不匹配：原格式字符串>B19s8x20s20s中的8x表示跳过8字节，但代码中却传入了reserved参数，导致参数数量与格式要求不匹配，直接触发struct.pack报错。
3. 响应解析逻辑隐患：未考虑TCP接收数据时可能出现的分段情况，直接recv(68)可能无法获取完整的握手响应。

修复后的完整代码 #

import socket
import struct

def peer_handshake(peer_ip, peer_port, info_hash, peer_id):
    # 改用TCP套接字，BitTorrent Peer通信基于TCP
    peer_connect = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    encoded_peer_id = peer_id.encode('utf-8')
    
    # 校验info_hash和peer_id长度，避免后续报错
    if len(info_hash) != 20:
        print("错误：info_hash必须是20字节长度")
        return "Invalid"
    if len(encoded_peer_id) != 20:
        print("错误：peer_id编码后必须是20字节长度")
        return "Invalid"
    
    try:
        peer_connect.settimeout(10)  # 缩短超时时间，更合理
        peer_connect.connect((peer_ip, peer_port))
        
        print("与Peer建立连接成功.... ")
        print("开始握手流程 .....")
        
        pstr = b"BitTorrent protocol"
        reserved = b'\x00' * 8  # 保留字段，默认全0
        
        # 标准握手格式：1字节pstr长度 + 19字节pstr + 8字节保留字段 + 20字节info_hash + 20字节peer_id
        handshake_format = ">B19s8s20s20s"
        handshake = struct.pack(handshake_format, len(pstr), pstr, reserved, info_hash, encoded_peer_id)
        peer_connect.sendall(handshake)
        print("握手包已发送")
        
        # 循环接收完整的68字节握手响应（TCP可能分段）
        response = b""
        while len(response) < 68:
            chunk = peer_connect.recv(68 - len(response))
            if not chunk:
                print("Peer提前关闭连接")
                return "Not Interested"
            response += chunk
        
        print("收到Peer响应：")
        print(response)
        
        print("\n解析响应信息：")
        # 提取Peer的info_hash（响应中第20-39字节：1字节长度+19字节pstr=20字节，之后8字节保留，再20字节info_hash）
        extracted_info_hash = response[28:48]
        print("提取的Info Hash：", extracted_info_hash)
        # 提取Peer ID（响应最后20字节）
        extracted_peer_id = response[-20:]
        print("提取的Peer ID：", extracted_peer_id.decode('utf-8'))
        
        print("\n校验Info Hash有效性：")
        if extracted_info_hash == info_hash:
            print("Info Hash匹配，握手成功")
            return "Interested"
        else:
            print("Info Hash不匹配，握手失败")
            return "Not Interested"
            
    except socket.error as e:
        print(f"连接{peer_ip}:{peer_port}时发生套接字错误：{e}")
    except Exception as e:
        print(f"连接{peer_ip}:{peer_port}时发生错误：{e}")
    finally:
        peer_connect.close()

关键修复说明 #

• TCP套接字替换UDP：BitTorrent Peer之间的握手和数据传输必须使用TCP，UDP仅用于与tracker的通信。
• 修正struct格式：将8x改为8s，正确打包8字节的reserved字段，确保参数数量与格式字符串匹配，解决struct.pack报错问题。
• 增加参数校验：提前检查info_hash和peer_id的长度，避免因参数不符合协议要求导致的后续错误。
• 完整响应接收：通过循环接收确保获取完整的68字节握手响应，避免TCP分段导致的解析错误。

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