92 lines
3.7 KiB
Python
92 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""FTP 数据连接回归测试（D20 验证 + 日后回归）
|
||
|
||
背景（2026-09-22 台架实测 D20）：同一 FTP 会话内**第二个数据连接必失败**
|
||
（425 Can't open data connection）。根因是单连接 Socket 复用时的"陈旧 DISCONNECT"
|
||
误判：上一条数据连接的 FIN/ACK 在 Socket 已复用为新连接之后才上报，新连接被
|
||
net 层当作"刚建立就断开"释放。修复见 net_socket.c handle_disconnect_event 的
|
||
陈旧事件甄别（回读芯片 TCP 状态，仍 ESTABLISHED 则丢弃该中断）。
|
||
|
||
本脚本在一个会话里连续做多次数据连接（LIST/NLST/STOR/RETR），并做大文件往返
|
||
内容校验；任一环节失败即退出码 1。
|
||
|
||
用法：
|
||
& '.venv\\Scripts\\python.exe' test/ftp_regression.py --host 192.168.1.100
|
||
"""
|
||
|
||
import argparse
|
||
import io
|
||
import sys
|
||
import time
|
||
from ftplib import FTP
|
||
|
||
SMALL = b"D20-small-" + b"A" * 1017 # 1024B
|
||
BIG = bytes(((i * 7 + 3) & 0xFF) for i in range(32768)) # 32KB 模式数据
|
||
|
||
|
||
def main(argv=None):
|
||
ap = argparse.ArgumentParser(description="FTP data-connection regression (D20)")
|
||
ap.add_argument("--host", default="192.168.1.100", help="device IP")
|
||
ap.add_argument("--port", type=int, default=21, help="FTP control port")
|
||
ap.add_argument("--timeout", type=int, default=20, help="socket timeout (s)")
|
||
args = ap.parse_args(argv)
|
||
|
||
results = []
|
||
|
||
def step(name, fn):
|
||
"""执行一步并记录结果；不抛异常，便于一次跑完全部步骤。"""
|
||
t0 = time.time()
|
||
try:
|
||
extra = fn()
|
||
dt = time.time() - t0
|
||
results.append((name, True, dt))
|
||
print(" %-16s OK %6.2fs %s" % (name, dt, extra if extra is not None else ""))
|
||
return True
|
||
except Exception as e: # noqa: BLE001
|
||
dt = time.time() - t0
|
||
results.append((name, False, dt))
|
||
print(" %-16s FAIL %6.2fs %s %s" % (name, dt, type(e).__name__, e))
|
||
return False
|
||
|
||
f = FTP()
|
||
print("[ftp] connecting %s:%d ..." % (args.host, args.port))
|
||
f.connect(args.host, args.port, timeout=args.timeout)
|
||
f.login("anonymous", "x@x")
|
||
print("[ftp] login ok: %s" % f.getwelcome())
|
||
|
||
# 同一会话内连续 8 次数据连接（含两次大文件）—— D20 的失败点是第 2 次
|
||
step("LIST #1", lambda: "%d lines" % len(f.nlst()))
|
||
step("STOR #2 1KB", lambda: f.storbinary("STOR d20s.tmp", io.BytesIO(SMALL)) or "226")
|
||
|
||
got = bytearray()
|
||
step("RETR #3 1KB", lambda: (f.retrbinary("RETR d20s.tmp", got.extend), "226")[1])
|
||
if bytes(got) != SMALL:
|
||
print(" !! RETR #3 content mismatch (%d bytes)" % len(got))
|
||
results.append(("RETR#3-content", False, 0.0))
|
||
else:
|
||
results.append(("RETR#3-content", True, 0.0))
|
||
|
||
step("LIST #4", lambda: "%d lines" % len(f.nlst()))
|
||
step("STOR #5 32KB", lambda: f.storbinary("STOR d20b.tmp", io.BytesIO(BIG)) or "226")
|
||
|
||
big = bytearray()
|
||
step("RETR #6 32KB", lambda: (f.retrbinary("RETR d20b.tmp", big.extend), "226")[1])
|
||
if bytes(big) != BIG:
|
||
print(" !! RETR #6 content mismatch (%d/%d bytes)" % (len(big), len(BIG)))
|
||
results.append(("RETR#6-content", False, 0.0))
|
||
else:
|
||
results.append(("RETR#6-content", True, 0.0))
|
||
|
||
step("LIST #7", lambda: "%d lines" % len(f.nlst()))
|
||
step("cleanup", lambda: (f.delete("d20s.tmp"), f.delete("d20b.tmp"), "deleted")[2])
|
||
step("QUIT", lambda: f.quit() or "221")
|
||
|
||
ok = sum(1 for _, good, _ in results if good)
|
||
print("[ftp] %d/%d steps passed" % (ok, len(results)))
|
||
return 0 if ok == len(results) else 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|