# Author: Mr.Xue
# 2019.10.29
# socket_ftp_client.py

import socket
import hashlib

client = socket.socket() # 生成socket連接對象
client.connect(('localhost', 6961)) # 建立連接

while True:
 cmd = input(">>").strip() # 輸入想要接收的文件
 if len(cmd) == 0: continue # 輸入為空，重新再輸
 if cmd.startswith('get'): # 判斷指令是否以get開頭
 client.send(cmd.encode("utf-8")) # 發送
 server_response = client.recv(1024) #接收即將發送來的文件的大小
 print("server response:", server_response) # 打印文件的大小
 client.send(b'ready to recv file...') # 發送消息告訴服務已經準備好接收了
 file_total_size = int(server_response.decode()) # 記錄文件的總大小
 received_size = 0 # 記錄已經接收了的文件的大小
 filename = cmd.split()[1] # 取出輸入的文件名
 f = open(filename + '.new', 'wb') # 新建一個本地文件來存儲接收的數據
 m = hashlib.md5() # md5加密
 while received_size < file_total_size:
  data = client.recv(1024) # 接收數據，一次最大接收1024bytes
  received_size += len(data) # 記錄已接收的數據大小
  m.update(data) # 用md5加密
  f.write(data) # 寫入文件
 else:
  new_file_md5 = m.hexdigest() # 16進制顯示加密文件
  print("file recv done", received_size, file_total_size)
  f.close()
  server_md5 = client.recv(1024) # 接收服務器端以md5加密的這個接收文件的加密文件，拿來和接收之后的加密文件做比較
  print("md5", new_file_md5, server_md5)
client.close() 

# Author: Mr.Xue
# 2019.10.29
# socket_ftp_server.py

import socket, os, hashlib

server = socket.socket() #生成socket對象
server.bind(('localhost', 6961)) # 綁定ip和端口
server.listen() #監聽
while True:
 conn, addr = server.accept() # 等待客戶端連接
 while True:
 print("等待接收文件名...")
 data = conn.recv(1024) # 接收客戶端發來的信息
 if not data: #判斷客戶端是否斷連
  print('lost a link...')
  break
 cmd, filename = data.decode().split() # 分割接收到的消息，提取出文件名
 if os.path.isfile(filename): # 判斷文件是否存在
  f = open(filename, 'rb') # 打開文件
  m = hashlib.md5() # md5加密對象
  file_size = os.stat(filename).st_size # 計算文件大小
  conn.send(str(file_size).encode("utf-8")) # send file size
  conn.recv(1024) # wait the ack 預防粘包
  for line in f:
  m.update(line) # 加密
  conn.send(line) # 發送
  print("file md5:", m.hexdigest()) # 以16進制打印加密后的文件
  f.close() # 關閉文件
  conn.send(m.hexdigest().encode("utf-8")) # 以16進制發送加密后的文件
 print('send done')
socket.close()