79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
|
"""Upload large file to server via plink PTY base64 chunks."""
|
||
|
|
import subprocess
|
||
|
|
import base64
|
||
|
|
import hashlib
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
LOCAL_FILE = r"D:\资料\03-项目开发\wecom_it_smart_desk\tmp-h5-dist.tar.gz"
|
||
|
|
REMOTE_FILE = "/tmp/h5-dist.tar.gz"
|
||
|
|
CHUNK_SIZE = 4000 # chars per chunk
|
||
|
|
|
||
|
|
PLINK = r"C:\Users\simon\.workbuddy\skills\jumpserver-ops\scripts\plink.exe"
|
||
|
|
HOST = "sxn@10.212.189.210"
|
||
|
|
PORT = "2222"
|
||
|
|
PASSWORD = os.environ.get("JMS_PASSWORD", "") # May be cached
|
||
|
|
|
||
|
|
def run_plink(commands):
|
||
|
|
"""Run commands via plink PTY."""
|
||
|
|
cmd = [PLINK, "-P", str(PORT), "-batch", HOST]
|
||
|
|
stdin_data = "\n".join(commands) + "\nexit\n"
|
||
|
|
result = subprocess.run(
|
||
|
|
cmd,
|
||
|
|
input=stdin_data,
|
||
|
|
capture_output=True,
|
||
|
|
text=True,
|
||
|
|
timeout=300
|
||
|
|
)
|
||
|
|
return result.stdout + result.stderr
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# Read file
|
||
|
|
with open(LOCAL_FILE, "rb") as f:
|
||
|
|
data = f.read()
|
||
|
|
|
||
|
|
local_md5 = hashlib.md5(data).hexdigest()
|
||
|
|
print(f"File: {LOCAL_FILE}")
|
||
|
|
print(f"Size: {len(data)} bytes")
|
||
|
|
print(f"MD5: {local_md5}")
|
||
|
|
|
||
|
|
# Base64 encode
|
||
|
|
b64 = base64.b64encode(data).decode("ascii")
|
||
|
|
total_chunks = (len(b64) + CHUNK_SIZE - 1) // CHUNK_SIZE
|
||
|
|
print(f"Base64 length: {len(b64)} chars, {total_chunks} chunks")
|
||
|
|
|
||
|
|
# Clear remote file
|
||
|
|
print("Clearing remote file...")
|
||
|
|
run_plink([f"> {REMOTE_FILE}.b64"])
|
||
|
|
|
||
|
|
# Send chunks
|
||
|
|
for i in range(total_chunks):
|
||
|
|
start = i * CHUNK_SIZE
|
||
|
|
end = min(start + CHUNK_SIZE, len(b64))
|
||
|
|
chunk = b64[start:end]
|
||
|
|
cmd = f"echo '{chunk}' >> {REMOTE_FILE}.b64"
|
||
|
|
run_plink([cmd])
|
||
|
|
if (i + 1) % 10 == 0 or i == total_chunks - 1:
|
||
|
|
print(f" Sent chunk {i+1}/{total_chunks}")
|
||
|
|
|
||
|
|
# Decode and verify
|
||
|
|
print("Decoding and verifying...")
|
||
|
|
verify_cmds = [
|
||
|
|
f"base64 -d {REMOTE_FILE}.b64 > {REMOTE_FILE}",
|
||
|
|
f"wc -c < {REMOTE_FILE}",
|
||
|
|
f"md5sum {REMOTE_FILE}",
|
||
|
|
f"rm -f {REMOTE_FILE}.b64",
|
||
|
|
]
|
||
|
|
output = run_plink(verify_cmds)
|
||
|
|
print(f"Server output:\n{output}")
|
||
|
|
|
||
|
|
if local_md5 in output:
|
||
|
|
print(f"\n✅ MD5 match! Upload successful.")
|
||
|
|
return 0
|
||
|
|
else:
|
||
|
|
print(f"\n❌ MD5 mismatch! Expected: {local_md5}")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main())
|