用Python实现简单的tcp文件下载器

TCP特点

1. 面向连接

通信双⽅必须先建⽴连接才能进⾏数据的传输，双⽅都必须为该连接分配必

要的系统内核资源，以管理连接的状态和连接上的传输。

双⽅间的数据传输都可以通过这⼀个连接进⾏。

完成数据交换后，双⽅必须断开此连接，以释放系统资源。

这种连接是⼀对⼀的，因此TCP不适⽤于⼴播的应⽤程序，基于⼴播的应⽤

程序请使⽤UDP协议。

2. 可靠传输

1）TCP采⽤发送应答机制

TCP发送的每个报⽂段都必须得到接收⽅的应答才认为这个TCP报⽂段传输成功

2）超时重传

发送端发出⼀个报⽂段之后就启动定时器，如果在定时时间内没有收到应答

就重新发送这个报⽂段。

TCP为了保证不发⽣丢包，就给每个包⼀个序号，同时序号也保证了传送到

接收端实体的包的按序接收。然后接收端实体对已成功收到的包发回⼀个相

应的确认（ACK）；如果发送端实体在合理的往返时延（RTT）内未收到确

认，那么对应的数据包就被假设为已丢失将会被进⾏重传。

3）错误校验

TCP⽤⼀个校验和函数来检验数据是否有错误；在发送和接收时都要计算校

验和。

4) 流量控制和阻塞管理

流量控制⽤来避免主机发送得过快⽽使接收⽅来不及完全收下。

 

TCP与UDP的不同点

• ⾯向连接（确认有创建三⽅交握，连接已创建才作传输。）
• 有序数据传输
• 重发丢失的数据包
• 舍弃重复的数据包
• ⽆差错的数据传输
• 阻塞/流量控制

 

TCP通信模型

 

 

1.tcp客户端构建流程

import socket


def main():
    # 1.创建套接字
    tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.链接服务
    sever_ip = input("请输入要链接的服务器：")
    sever_port = int(input("请输入要链接服务器的端口号："))
    sever_addr = (sever_ip, sever_port)
    tcp_client_socket.connect(sever_addr)

    # 3.发送接收/接收数额
    send_data = input("请输入要发送的数据：")
    tcp_client_socket.send(send_data.encode("utf-8"))

    recv_date = tcp_client_socket.recv(1024)
    print("接收到的数据为：", recv_date.decode("utf-8"))

    # 4.关闭套接字
    tcp_client_socket.close()


if __name__ == '__main__':
    main()

 

 

2.tcp服务器构建流程

 

import socket


def main():
    # 创建套接字
    tcp_sever_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 绑定本地信息
    tcp_sever_socket.bind(("", 7891))
    # 让默认的套接字由主动变被动 listen
    tcp_sever_socket.listen(128)

    # 循环目的：调用多次accept,从而为多个客户端服务
    while True:
        print("等待一个新的客户端的到来...")
        # 等待客户端的链接  accept
        new_client_socket, client_addr = tcp_sever_socket.accept()
        print("一个新的客户端已经到来%s" % str(client_addr))

        # 循环目的: 为同一个客户端 服务多次
        while True:
            # 接收客户端发来的请求
            recv_data = new_client_socket.recv(1024)
            print("客户端福送过来的请求是:%s" % recv_data.decode("utf-8"))

            # 如果recv解堵塞，那么有2种方式：
            # 1. 客户端发送过来数据
            # 2. 客户端调用close导致而了 这里 recv解堵塞
            if recv_data:
                # 回送一部分数据给客户端
                new_client_socket.send("---ok---".encode("utf-8"))
            else:
                break
        #关闭套接字
        new_client_socket.close()
        print("已经服务器完毕。。。。")

    # 如果将监听套接字 关闭了，那么会导致 不能再次等待新客户端的到来，即xxxx.accept就会失败
    tcp_sever_socket.close()


if __name__ == '__main__':
    main()

 

3.tcp注意点

 

1. tcp服务器⼀般情况下都需要绑定，否则客户端找不到这个服务器

2. tcp客户端⼀般不绑定，因为是主动链接服务器，所以只要确定好服务器

的ip、port等信息就好，本地客户端可以随机

3. tcp服务器中通过listen可以将socket创建出来的主动套接字变为被动的，

这是做tcp服务器时必须要做的

4. 当客户端需要链接服务器时，就需要使⽤connect进⾏链接，udp是不需

要链接的⽽是直接发送，但是tcp必须先链接，只有链接成功才能通信

5. 当⼀个tcp客户端连接服务器时，服务器端会有1个新的套接字，这个套

接字⽤来标记这个客户端，单独为这个客户端服务

6. listen后的套接字是被动套接字，⽤来接收新的客户端的链接请求的，⽽

accept返回的新套接字是标记这个新客户端的

7. 关闭listen后的套接字意味着被动套接字关闭了，会导致新的客户端不能

够链接服务器，但是之前已经链接成功的客户端正常通信。

8. 关闭accept返回的套接字意味着这个客户端已经服务完毕

9. 当客户端的套接字调⽤close后，服务器端会recv解堵塞，并且返回的⻓

度为0，因此服务器可以通过返回数据的⻓度来区别客户端是否已经下线

 

 

案例：文件下载器

客户端参考代码：

import socket


def main():
    # 1.创建套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.获取服务器的ip port
    dest_ip = input("请输入下载服务器的ip：")
    dest_port = int(input("请输入下载服务器的port:"))

    # 3.链接服务器
    tcp_socket.connect((dest_ip, dest_port))

    # 4.获取文件下载的文件名
    download_file_name = input("请输入要下载的文件名字：")

    # 5.将文件名字发送到服务器
    tcp_socket.send(download_file_name.encode("utf-8"))

    # 6.接收文件中的数据
    recv_data = tcp_socket.recv(1024)

    if recv_data:
        # 7.保存接收数到的数据到另一个文件中
        with open("[新]" + download_file_name, "wb") as f:
            f.write(recv_data)

    # 8.关闭套接字
    tcp_socket.close()


if __name__ == '__main__':
    main()

 

 

服务器端参考代码：

import socket

def send_file_to_client(new_client_socket, client_addr):
    # 1.接收客户端发过来需要下载的文件名
    file_name = new_client_socket.recv(1024).decode("utf-8")
    print("客户端(%s)需要下载的文件是：%s" % (str(client_addr), file_name))

    file_content = None
    # 2.打开这个文件，读取数据
    try:
        f = open(file_name, "rb")
        file_content = f.read()
        f.close()
    except Exception as ret:
        print("没有要下载的文件(%s)" % file_name)

    # 3.发送文件的数据给客户端
    if file_content:
        new_client_socket.send(file_content)


def main():
    # 1.创建套接字 socket
    tcp_sever_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.绑定本地信息 bind
    tcp_sever_socket.bind(("", 7890))

    # 3.让默认的套接字由主动变为被动 listen
    tcp_sever_socket.listen(128)

    # 4.等待客户端的链接 accept
    new_client_socket, client_addr = tcp_sever_socket.accept()

    # 5.调用发送文件函数，完成客户端服务
    send_file_to_client(new_client_socket, client_addr)

    # 6.关闭套接字
    new_client_socket.close()
    tcp_sever_socket.close()


if __name__ == '__main__':
    main()