python中的socket实现ftp客户端和服务器收发文件及md5加密文件

Python中的socket库是实现网络编程的重要工具之一。通过socket库，我们可以实现FTP客户端和服务器收发文件，并使用md5算法对文件进行加密。下面是Python中实现FTP客户端和服务器收发文件及md5加密文件的完整攻略：

安装必要的库

要使用Python中的socket库，需要首先安装Python。如果你已经安装了Python3.x，那么socket库已经默认安装好了。如果你还没有安装Python，可以从Python官网下载最新版本进行安装。

实现FTP服务器

以下是实现FTP服务器的示例代码：

import socket
import os

# 定义FTP服务器的端口号
port = 8080

# 创建一个socket对象
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定端口号
server_socket.bind(('127.0.0.1', port))

# 监听端口
server_socket.listen()

# 无限循环等待客户端连接
while True:
    # 等待客户端连接
    client_socket, client_address = server_socket.accept()

    # 接收客户端发送的文件名
    file_name = client_socket.recv(1024).decode()
    print('接收到客户端 %s 发送的文件名：%s' % (client_address, file_name))

    # 如果服务器上有该文件，则读取文件内容并发送给客户端
    if os.path.exists(file_name):
        client_socket.send('OK'.encode())
        with open(file_name, 'rb') as f:
            data = f.read()
            client_socket.send(data)
            print('已向客户端 %s 发送文件 %s' % (client_address, file_name))
    else:
        client_socket.send('File not exists'.encode())
        print('文件 %s 不存在' % file_name)

    # 关闭连接
    client_socket.close()

这段代码创建了一个socket对象，并绑定到8080端口。服务器通过listen()方法监听端口，接收客户端的请求。当客户端连接成功后，服务器通过accept()方法接收客户端发送的文件名。如果服务器上存在该文件，则读取文件内容并通过socket发送给客户端；否则向客户端发送错误消息。最后关闭连接。

实现FTP客户端

以下是实现FTP客户端的示例代码：

import socket
import hashlib

# 定义FTP服务器的地址和端口号
IP = '127.0.0.1'
port = 8080

# 创建一个socket对象
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 连接FTP服务器
client_socket.connect((IP, port))

# 发送文件名
file_name = 'test.txt'
client_socket.send(file_name.encode())

# 接收服务器的响应
response = client_socket.recv(1024).decode()

# 如果文件存在，则接收文件内容并进行md5加密
if response == 'OK':
    with open(file_name, 'wb') as f:
        while True:
            data = client_socket.recv(1024)
            if not data:
                break
            f.write(data)
    with open(file_name, 'rb') as f:
        md5_obj = hashlib.md5()
        md5_obj.update(f.read())
        md5_value = md5_obj.hexdigest()
        print('文件 %s 的md5值为：%s' % (file_name, md5_value))
else:
    print('文件 %s 不存在' % file_name)

# 关闭连接
client_socket.close()

这段代码创建了一个socket对象，并连接到FTP服务器。客户端通过send()方法向服务器发送文件名，并通过recv()方法接收服务器的响应。如果服务器响应为OK，则客户端接收文件内容，并使用md5算法对文件内容进行加密。最后输出文件的md5值。如果服务器响应为File not exists，则输出错误消息。

实现可复用的文件传输函数

如果我们经常需要传输文件，可以将上述代码封装成一个可复用的函数，以便于代码的复用和维护。以下是一个可复用的文件传输函数示例：

def send_file(file_path, ip, port):
    # 创建socket对象
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client_socket.connect((ip, port))

    # 发送文件名
    file_name = os.path.basename(file_path)
    client_socket.send(file_name.encode())

    # 接收服务器的响应
    response = client_socket.recv(1024).decode()

    # 如果文件存在，则接收文件内容并进行md5加密
    if response == 'OK':
        with open(file_path, 'rb') as f:
            while True:
                data = f.read(1024)
                if not data:
                    break
                client_socket.send(data)
        md5_obj = hashlib.md5()
        md5_obj.update(open(file_path, 'rb').read())
        md5_value = md5_obj.hexdigest()
        print('文件 %s 的md5值为：%s' % (file_name, md5_value))
    else:
        print('文件 %s 不存在' % file_name)

    # 关闭连接
    client_socket.close()

这个函数封装了文件的传输过程，并实现了md5加密。如果想传输一个文件，只需要调用该函数并传入文件路径、IP地址和端口号即可。

总结

通过socket库，我们可以轻松地实现FTP客户端和服务器收发文件，并使用md5算法对文件进行加密。通过封装可复用的函数，我们可以将这个过程更加简便和高效。