Python程式設計：socket實作檔案傳輸

思路：

1. 讀取檔案名
2. 檢測檔案是否存在
3. 打開檔案
4. 檢測檔案大小
5. 發送檔案大小和MD5值給用戶端
6. 等用戶端确認
7. 開始邊讀邊發資料
8. MD5驗證

# 伺服器端

import socket
import os
import hashlib

server = socket.socket()

server.bind(("localhost", 6969)) # 綁定監聽端口

server.listen(5)  # 監聽

print("監聽開始..")

while True:
    conn, addr = server.accept()  # 等待連接配接

    print("conn:", conn, "\naddr:", addr)  # conn連接配接執行個體

    while True:
        data = conn.recv(1024)  # 接收
        if not data:  # 用戶端已斷開
            print("用戶端斷開連接配接")
            break

        print("收到的指令：", data.decode("utf-8"))
        cmd, filename = data.decode("utf-8").split(" ")
        if cmd =="get":
            if os.path.isfile(filename):  # 判斷檔案存在

                # 1.先發送檔案大小，讓用戶端準備接收
                size = os.stat(filename).st_size  #擷取檔案大小
                conn.send(str(size).encode("utf-8"))  # 發送資料長度
                print("發送的大小：", size)

                # 2.發送檔案内容
                conn.recv(1024)  # 接收确認

                m = hashlib.md5()
                f = open(filename, "rb")
                for line in f:
                    conn.send(line)  # 發送資料
                    m.update(line)
                f.close()

                # 3.發送md5值進行校驗
                md5 = m.hexdigest()
                conn.send(md5.encode("utf-8"))  # 發送md5值
                print("md5:", md5)


server.close()      

# 用戶端

import socket
import os
import hashlib

client = socket.socket()  # 生成socket連接配接對象

ip_port =("localhost", 6969)  # 位址和端口号

client.connect(ip_port)  # 連接配接

print("伺服器已連接配接")

while True:
    content = input(">>")

    if len(content)==0: continue  # 如果傳入空字元會阻塞

    if content.startswith("get"):
        client.send(content.encode("utf-8"))  # 傳送和接收都是bytes類型

        # 1.先接收長度，建議8192
        server_response = client.recv(1024)
        file_size = int(server_response.decode("utf-8"))

        print("接收到的大小：", file_size)

        # 2.接收檔案内容
        client.send("準備好接收".encode("utf-8"))  # 接收确認
        filename = "new" + content.split(" ")[1]

        f = open(filename, "wb")
        received_size = 0
        m = hashlib.md5()

        while received_size < file_size:
            size = 0  # 準确接收資料大小，解決粘包
            if file_size - received_size > 1024: # 多次接收
                size = 1024
            else:  # 最後一次接收完畢
                size = file_size - received_size

            data = client.recv(size)  # 多次接收内容，接收大資料
            data_len = len(data)
            received_size += data_len
            print("已接收：", int(received_size/file_size*100), "%")

            m.update(data)
            f.write(data)

        f.close()

        print("實際接收的大小:", received_size)  # 解碼

        # 3.md5值校驗
        md5_sever = client.recv(1024).decode("utf-8")
        md5_client = m.hexdigest()
        print("伺服器發來的md5:", md5_sever)
        print("接收檔案的md5:", md5_client)
        if md5_sever == md5_client:
            print("MD5值校驗成功")
        else:
            print("MD5值校驗失敗")

client.close()      

提示：用戶端輸入格式為：

get filename