231 lines
7.0 KiB
Python
231 lines
7.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
FTP 速度测试 — 上传/下载大文件，测量传输速度
|
|
用法：python ftp_speed_test.py --host 192.168.1.100 [--size 512] [--port 21]
|
|
"""
|
|
|
|
import socket
|
|
import sys
|
|
import time
|
|
import re
|
|
import argparse
|
|
import os
|
|
|
|
HOST = "192.168.1.100"
|
|
PORT = 21
|
|
FILE_SIZE_KB = 512 # 默认 512KB 测试文件
|
|
|
|
def log(msg, *args):
|
|
print(f" {msg % args if args else msg}")
|
|
|
|
def recv_line(sock, timeout=60.0):
|
|
data = b""
|
|
sock.settimeout(timeout)
|
|
while True:
|
|
try:
|
|
ch = sock.recv(1)
|
|
except socket.timeout:
|
|
return None
|
|
if not ch:
|
|
return data if data else None
|
|
data += ch
|
|
if data.endswith(b"\r\n"):
|
|
return data
|
|
|
|
def send_cmd(sock, cmd):
|
|
sock.sendall((cmd + "\r\n").encode())
|
|
time.sleep(0.1)
|
|
return recv_line(sock)
|
|
|
|
def parse_pasv(resp):
|
|
m = re.search(r"(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)", resp.decode())
|
|
if m:
|
|
a, b, c, d, e, f = [int(x) for x in m.groups()]
|
|
return f"{a}.{b}.{c}.{d}", e * 256 + f
|
|
return None, None
|
|
|
|
def test_speed(host, port, size_kb):
|
|
total_bytes = size_kb * 1024
|
|
test_data = os.urandom(total_bytes)
|
|
test_name = "SPEEDTST.DAT" # 8.3 format
|
|
|
|
print(f"\n=== FTP Speed Test: {host}:{port}, {size_kb}KB file ===")
|
|
print(f" Test data: {total_bytes:,} bytes generated\n")
|
|
|
|
checks = []
|
|
|
|
def check(label, ok, detail=""):
|
|
checks.append((label, ok, detail))
|
|
mark = "PASS" if ok else "FAIL"
|
|
print(f" [{mark}] {label}{(' - ' + detail) if detail else ''}")
|
|
|
|
def finish(sock):
|
|
passed = all(ok for _, ok, _ in checks)
|
|
print("\n" + "=" * 40)
|
|
print("FTP Speed Test Summary:")
|
|
for label, ok, _ in checks:
|
|
print(f" [{'PASS' if ok else 'FAIL'}] {label}")
|
|
print("=" * 40)
|
|
if passed:
|
|
print(">>> TEST PASSED <<<")
|
|
else:
|
|
print(">>> TEST FAILED <<<")
|
|
print("=== Test Complete ===")
|
|
if sock:
|
|
try:
|
|
sock.close()
|
|
except Exception:
|
|
pass
|
|
return passed
|
|
|
|
# ---- Connect + Login ----
|
|
print("[Connect] Connecting...")
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.settimeout(60.0)
|
|
try:
|
|
sock.connect((host, port))
|
|
except Exception as e:
|
|
print(f" CONNECT FAILED: {e}")
|
|
check("Connect", False, str(e))
|
|
return finish(sock)
|
|
recv_line(sock) # 220
|
|
if not send_cmd(sock, "USER anonymous"):
|
|
check("Login USER", False)
|
|
return finish(sock)
|
|
if not send_cmd(sock, "PASS test@test.com"):
|
|
check("Login PASS", False)
|
|
return finish(sock)
|
|
check("Login (USER/PASS)", True)
|
|
send_cmd(sock, "TYPE I")
|
|
|
|
# ---- Upload ----
|
|
print("[UPLOAD] Connecting...")
|
|
resp = send_cmd(sock, "PASV")
|
|
if not resp:
|
|
check("PASV", False, "no response")
|
|
return finish(sock)
|
|
pasv_ip, pasv_port = parse_pasv(resp)
|
|
if pasv_port is None:
|
|
check("PASV parse", False, resp.decode())
|
|
return finish(sock)
|
|
check("PASV parse", True, f"{pasv_ip}:{pasv_port}")
|
|
|
|
data = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
data.settimeout(60.0)
|
|
data.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1) # 关 Nagle，防 FIN 跑在数据前面
|
|
try:
|
|
data.connect((pasv_ip, pasv_port))
|
|
except Exception as e:
|
|
print(f" Data connect failed: {e}")
|
|
check("PASV data connect", False, str(e))
|
|
return finish(sock)
|
|
check("PASV data connect", True)
|
|
|
|
t0 = time.time()
|
|
send_cmd(sock, f"STOR {test_name}")
|
|
data.sendall(test_data)
|
|
data.shutdown(socket.SHUT_WR)
|
|
data.close()
|
|
|
|
resp = recv_line(sock, timeout=120.0)
|
|
t1 = time.time()
|
|
upload_time = t1 - t0
|
|
upload_speed = total_bytes / upload_time / 1024 if upload_time > 0 else 0
|
|
stor_code = resp[:3].decode() if resp else "???"
|
|
|
|
print(f" Sent: {total_bytes:,} / {total_bytes:,} bytes")
|
|
print(f" Time: {upload_time:.1f}s")
|
|
print(f" Speed: {upload_speed:.1f} KB/s")
|
|
print(f" Response: {resp.decode().rstrip() if resp else 'NONE'}")
|
|
|
|
upload_ok = (stor_code == "226")
|
|
check("STOR 226", upload_ok, f"{stor_code} up={upload_speed:.1f}KB/s")
|
|
|
|
if not upload_ok:
|
|
send_cmd(sock, "QUIT")
|
|
return finish(sock)
|
|
|
|
# ---- Download ----
|
|
time.sleep(3.0)
|
|
print("[DOWNLOAD] Connecting...")
|
|
resp = send_cmd(sock, "PASV")
|
|
if not resp:
|
|
check("RETR PASV", False, "no response")
|
|
return finish(sock)
|
|
pasv_ip, pasv_port = parse_pasv(resp)
|
|
if pasv_port is None:
|
|
check("RETR PASV parse", False, resp.decode())
|
|
return finish(sock)
|
|
data = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
data.settimeout(60.0)
|
|
try:
|
|
data.connect((pasv_ip, pasv_port))
|
|
except Exception as e:
|
|
print(f" Data connect failed: {e}")
|
|
check("RETR data connect", False, str(e))
|
|
return finish(sock)
|
|
check("RETR data connect", True)
|
|
|
|
t0 = time.time()
|
|
send_cmd(sock, f"RETR {test_name}")
|
|
dl_data = b""
|
|
while True:
|
|
try:
|
|
chunk = data.recv(8192)
|
|
if not chunk:
|
|
break
|
|
dl_data += chunk
|
|
except socket.timeout:
|
|
break
|
|
data.close()
|
|
resp = recv_line(sock, timeout=120.0)
|
|
t1 = time.time()
|
|
download_time = t1 - t0
|
|
download_speed = len(dl_data) / download_time / 1024 if download_time > 0 else 0
|
|
|
|
print(f" Received: {len(dl_data):,} / {total_bytes:,} bytes")
|
|
print(f" Time: {download_time:.1f}s")
|
|
print(f" Speed: {download_speed:.1f} KB/s")
|
|
print(f" Response: {resp.decode().rstrip() if resp else 'NONE'}")
|
|
|
|
retr_ok = bool(resp) and resp.startswith(b"226")
|
|
check("RETR 226", retr_ok,
|
|
f"{resp[:3].decode() if resp else '???'} dl={download_speed:.1f}KB/s")
|
|
|
|
# Compare
|
|
print(f"\n[VERIFY]")
|
|
match = (len(dl_data) == total_bytes and dl_data == test_data)
|
|
if match:
|
|
print(f" *** MATCH! {total_bytes:,} bytes identical. ***")
|
|
else:
|
|
print(f" MISMATCH: sent={total_bytes}, got={len(dl_data)}")
|
|
if len(dl_data) != total_bytes:
|
|
print(f" Size mismatch!")
|
|
else:
|
|
for i in range(total_bytes):
|
|
if dl_data[i] != test_data[i]:
|
|
print(f" First diff at byte {i}")
|
|
break
|
|
check("Upload/Download MATCH", match,
|
|
f"up={total_bytes:,} dl={len(dl_data):,} up={upload_speed:.1f}KB/s dl={download_speed:.1f}KB/s")
|
|
|
|
# Cleanup
|
|
send_cmd(sock, f"DELE {test_name}")
|
|
send_cmd(sock, "QUIT")
|
|
|
|
return finish(sock)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="FTP Speed Test")
|
|
parser.add_argument("--host", default=HOST)
|
|
parser.add_argument("--port", type=int, default=PORT)
|
|
parser.add_argument("--size", type=int, default=FILE_SIZE_KB,
|
|
help=f"File size in KB (default {FILE_SIZE_KB})")
|
|
args = parser.parse_args()
|
|
success = test_speed(args.host, args.port, args.size)
|
|
sys.exit(0 if success else 1)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|