import socket
import os

# 配置參數(shù)
UDP_IP = ""  # 監(jiān)聽(tīng)所有本機(jī)IP
UDP_PORT = 9999
BUFFER_SIZE = 1024  # 數(shù)據(jù)包大?。ㄐ韬涂蛻舳艘恢拢?
SAVE_DIR = "received_files"  # 文件保存目錄

# 創(chuàng)建保存目錄
if not os.path.exists(SAVE_DIR):
    os.makedirs(SAVE_DIR)

def udp_file_server():
    # 1. 創(chuàng)建UDP Socket
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    server_socket.bind((UDP_IP, UDP_PORT))
    # 允許端口復(fù)用，避免重啟報(bào)錯(cuò)
    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    print(f"UDP文件服務(wù)端已啟動(dòng)，監(jiān)聽(tīng)端口 {UDP_PORT}...")

    # 初始化接收狀態(tài)
    file_data = {}  # 存儲(chǔ)接收的數(shù)據(jù)包 {序號(hào): 數(shù)據(jù)}
    total_packets = 0  # 總包數(shù)
    file_name = ""  # 文件名
    client_addr = None  # 客戶端地址

    try:
        while True:
            # 2. 接收數(shù)據(jù)包（阻塞）
            data, client_addr = server_socket.recvfrom(BUFFER_SIZE + 32)  # 預(yù)留標(biāo)識(shí)位空間
            if not data:
                continue

            # 解析數(shù)據(jù)包標(biāo)識(shí)："文件名|總包數(shù)|當(dāng)前序號(hào)|數(shù)據(jù)"
            try:
                # 分割標(biāo)識(shí)和數(shù)據(jù)（用|分隔，最后一段是文件數(shù)據(jù)）
                parts = data.decode('utf-8', errors='ignore').split('|', 3)
                if len(parts) != 4:
                    # 不是文件數(shù)據(jù)包，可能是結(jié)束標(biāo)識(shí)
                    if data.decode('utf-8') == "TRANSFER_FINISH":
                        print("客戶端發(fā)送結(jié)束標(biāo)識(shí)，開(kāi)始重組文件...")
                        break
                    continue

                file_name, total_packets_str, packet_num_str, file_chunk = parts
                total_packets = int(total_packets_str)
                packet_num = int(packet_num_str)
            except Exception as e:
                print(f"數(shù)據(jù)包解析失?。簕e}")
                # 返回錯(cuò)誤ACK
                server_socket.sendto(f"ERR|{packet_num}".encode('utf-8'), client_addr)
                continue

            # 3. 校驗(yàn)并保存數(shù)據(jù)包
            if 1 <= packet_num <= total_packets:
                file_data[packet_num] = file_chunk
                # 返回成功ACK
                server_socket.sendto(f"ACK|{packet_num}".encode('utf-8'), client_addr)
                # 打印進(jìn)度
                progress = (len(file_data) / total_packets) * 100
                print(f"接收進(jìn)度：{progress:.1f}% ({len(file_data)}/{total_packets})", end='\r')

        # 4. 重組并保存文件
        if file_data and total_packets > 0:
            # 按序號(hào)排序數(shù)據(jù)包
            sorted_chunks = [file_data[i] for i in range(1, total_packets + 1) if i in file_data]
            # 拼接所有數(shù)據(jù)
            file_path = os.path.join(SAVE_DIR, file_name)
            with open(file_path, 'wb') as f:
                for chunk in sorted_chunks:
                    f.write(chunk.encode('utf-8'))  # 若傳二進(jìn)制文件，需調(diào)整編碼邏輯（見(jiàn)進(jìn)階部分）
            print(f"\n文件接收完成！保存路徑：{file_path}")
            # 發(fā)送完成確認(rèn)
            server_socket.sendto("FILE_SAVED".encode('utf-8'), client_addr)
        else:
            print("\n未接收到完整文件數(shù)據(jù)")

    except KeyboardInterrupt:
        print("\n服務(wù)端手動(dòng)停止")
    finally:
        server_socket.close()

if __name__ == "__main__":
    udp_file_server()

import socket
import os
import time

# 配置參數(shù)
SERVER_IP = "127.0.0.1"  # 服務(wù)端IP（遠(yuǎn)程傳輸改實(shí)際IP）
SERVER_PORT = 9999
BUFFER_SIZE = 1024  # 每個(gè)數(shù)據(jù)包的文件數(shù)據(jù)大小
RETRY_TIMES = 3  # 單個(gè)數(shù)據(jù)包最大重傳次數(shù)
RETRY_INTERVAL = 0.5  # 重傳間隔（秒）

def udp_file_client(file_path):
    # 1. 檢查文件是否存在
    if not os.path.exists(file_path):
        print(f"錯(cuò)誤：文件 {file_path} 不存在")
        return

    # 獲取文件名和文件大小
    file_name = os.path.basename(file_path)
    file_size = os.path.getsize(file_path)
    # 計(jì)算總包數(shù)（向上取整）
    total_packets = (file_size + BUFFER_SIZE - 1) // BUFFER_SIZE
    print(f"準(zhǔn)備發(fā)送文件：{file_name}，大小：{file_size}字節(jié)，總包數(shù)：{total_packets}")

    # 2. 創(chuàng)建UDP Socket
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    client_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    # 設(shè)置超時(shí)（避免等待ACK卡死）
    client_socket.settimeout(2)

    try:
        # 3. 讀取文件并分塊發(fā)送
        with open(file_path, 'r', encoding='utf-8') as f:  # 二進(jìn)制文件用'rb'（見(jiàn)進(jìn)階部分）
            for packet_num in range(1, total_packets + 1):
                # 讀取當(dāng)前塊數(shù)據(jù)
                file_chunk = f.read(BUFFER_SIZE)
                if not file_chunk:
                    break

                # 構(gòu)造數(shù)據(jù)包："文件名|總包數(shù)|當(dāng)前序號(hào)|數(shù)據(jù)"
                packet_data = f"{file_name}|{total_packets}|{packet_num}|{file_chunk}".encode('utf-8')
                retry_count = 0
                ack_received = False

                # 4. 發(fā)送并等待ACK，失敗則重傳
                while retry_count < RETRY_TIMES and not ack_received:
                    try:
                        # 發(fā)送數(shù)據(jù)包
                        client_socket.sendto(packet_data, (SERVER_IP, SERVER_PORT))
                        # 等待ACK
                        ack_data, _ = client_socket.recvfrom(128)
                        ack_parts = ack_data.decode('utf-8').split('|')
                        if ack_parts[0] == "ACK" and int(ack_parts[1]) == packet_num:
                            ack_received = True
                            # 打印進(jìn)度
                            progress = (packet_num / total_packets) * 100
                            print(f"發(fā)送進(jìn)度：{progress:.1f}% ({packet_num}/{total_packets})", end='\r')
                    except socket.timeout:
                        retry_count += 1
                        print(f"\n數(shù)據(jù)包 {packet_num} 超時(shí)，重傳 {retry_count}/{RETRY_TIMES}")
                        time.sleep(RETRY_INTERVAL)
                    except Exception as e:
                        print(f"\n數(shù)據(jù)包 {packet_num} 發(fā)送失?。簕e}")
                        retry_count += 1
                        time.sleep(RETRY_INTERVAL)

                if not ack_received:
                    print(f"\n錯(cuò)誤：數(shù)據(jù)包 {packet_num} 重傳{RETRY_TIMES}次失敗，傳輸終止")
                    return

        # 5. 發(fā)送結(jié)束標(biāo)識(shí)
        client_socket.sendto("TRANSFER_FINISH".encode('utf-8'), (SERVER_IP, SERVER_PORT))
        # 等待服務(wù)端保存完成確認(rèn)
        try:
            finish_ack, _ = client_socket.recvfrom(128)
            if finish_ack.decode('utf-8') == "FILE_SAVED":
                print("\n文件發(fā)送完成！服務(wù)端已保存")
        except socket.timeout:
            print("\n未收到服務(wù)端完成確認(rèn)，但數(shù)據(jù)已發(fā)送完畢")

    except KeyboardInterrupt:
        print("\n客戶端手動(dòng)停止")
    finally:
        client_socket.close()

if __name__ == "__main__":
    # 替換為你要發(fā)送的文件路徑（本地測(cè)試用絕對(duì)/相對(duì)路徑）
    target_file = "test.txt"  # 示例：發(fā)送當(dāng)前目錄的test.txt
    udp_file_client(target_file)

# 替換客戶端文件讀取部分
with open(file_path, 'rb') as f:  # 改為二進(jìn)制讀取
    for packet_num in range(1, total_packets + 1):
        file_chunk = f.read(BUFFER_SIZE)
        if not file_chunk:
            break
    # 構(gòu)造數(shù)據(jù)包：用特殊分隔符（如b'|||'），避免二進(jìn)制數(shù)據(jù)沖突
    packet_header = f"{file_name}|{total_packets}|{packet_num}".encode('utf-8')
    packet_data = packet_header + b'|||' + file_chunk  # 二進(jìn)制拼接

# 替換服務(wù)端數(shù)據(jù)包解析部分
try:
    # 分割頭部和二進(jìn)制數(shù)據(jù)（按b'|||'分割）
    header, file_chunk = data.split(b'|||', 1)
    # 解析頭部（轉(zhuǎn)字符串）
    parts = header.decode('utf-8').split('|', 2)
    file_name, total_packets_str, packet_num_str = parts
    total_packets = int(total_packets_str)
    packet_num = int(packet_num_str)
except Exception as e:
    print(f"二進(jìn)制數(shù)據(jù)包解析失?。簕e}")
    continue

# 保存時(shí)直接寫入二進(jìn)制數(shù)據(jù)
with open(file_path, 'wb') as f:
    for chunk in sorted_chunks:
        f.write(chunk)  # 無(wú)需encode，直接寫二進(jìn)制