python网络编程socket模块实现ftp上传下载

本实验实现ftp上传文件下载文件功能，并具有校验文件完整性，打印进度条功能,
主要练习socket,struct模块。

ftp用户文件存放在user.json文件中
user.json文件内容
{"lisi": "abcdef", "hyh": "123456"}

ftp客户端脚本ftpclient.py

#!/usr/bin/python
# --*-- coding: utf-8 --*--
import socket
import json
import time
import sys
import struct
from hashlib import md5

ftp_obj = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ftp_obj.connect(('127.0.0.1', 8080))

def login_auth(ftp_obj):
"""用户登录三次认证,输入用户名密码跟服务器账号密码匹配返回True,否则返回False"""
count = 0
while count < 3:
user_name = input("请输入账户: ").strip()
user_passwd = input("请输入密码: ").strip()
if user_name and user_passwd:
ftp_obj.send(user_name.encode('utf-8'))
ftp_obj.send(user_passwd.encode('utf-8'))
else:
count += 1
continue
#time.sleep(10)
login_res_bytes = ftp_obj.recv(1024)
# print(login_res_bytes)
# print(login_res_bytes.decode('gbk'))
login_res = login_res_bytes.decode('gbk')
if login_res == 'True':
return True
else:
count += 1
continue
else:
return False

def progress_bar(num, total):
"""打印进度条"""
rate = num / total
rate_num = int(rate * 100)
r = '\r%s%d%%' % ('#' * rate_num, rate_num,)
print(r)

def get_md5(data):
"""校验文件内容"""
m = md5()
m.update(data.encode('utf-8'))
res = m.hexdigest()
return res

def ftp_cmd(ftp_obj):
"""上传下载命令"""
while True:
cmd = input("请输入命令(dir or put filepath or get filepath or exit)>>: ").strip()
if cmd == 'exit':
sys.exit(0)
#print(cmd_list)
if cmd == 'dir':
ftp_obj.send(cmd.encode('utf-8'))
head_struct = ftp_obj.recv(4)
head_len = struct.unpack('i', head_struct)[0]
head_bytes = ftp_obj.recv(head_len)
head_json = head_bytes.decode('utf-8')
head_dict = json.loads(head_json)
total_size = head_dict['total_size']
recv_size = 0
data = b''
while recv_size < total_size:
recv_data = ftp_obj.recv(1024)
data += recv_data
recv_size += len(recv_data)
print(data.decode('gbk'))
else:
cmd_list = cmd.split()
if cmd_list[0] == 'get':
ftp_obj.send(cmd.encode('utf-8'))
head_struct = ftp_obj.recv(4)
head_len = struct.unpack('i', head_struct)[0]
head_bytes = ftp_obj.recv(head_len)
head_json = head_bytes.decode('utf-8')
head_dict = json.loads(head_json)
total_size = head_dict['total_size']
recv_size = 0
data = b''
while recv_size < total_size:
recv_data = ftp_obj.recv(1024)
data += recv_data
recv_size += len(recv_data)
progress_bar(recv_size, total_size)     #打印进度条，传入两个参数，第一个是接受的数据字节，第二个是数据头解包的数据总长度

with open(cmd_list[1], 'w', encoding='utf-8') as f:
f.write(data.decode('utf-8'))
check_md5 = get_md5(data.decode('utf-8'))   #校验传输的内容
if head_dict['hashlib'] == check_md5:
print("下载成功，文件内容完整")
else:
print("下载完成，文件内容不完整")
elif cmd_list[0] == 'put':
ftp_obj.send(cmd.encode('utf-8'))
with open(cmd_list[1], 'r', encoding='utf-8') as f:
data = f.read()
check_put_md5 = get_md5(data)
head_dict = {'filename': cmd_list[1], 'hashlib': check_put_md5, 'total_size': len(data)}
head_json = json.dumps(head_dict)
head_bytes = head_json.encode('utf-8')
ftp_obj.send(struct.pack('i', len(head_bytes)))
ftp_obj.send(head_bytes)
is_upload = ftp_obj.recv(10)
if is_upload.decode('gbk') == 'True':
#ftp_obj.send(data.encode('utf-8'))
with open(cmd_list[1], 'r', encoding='utf-8') as f:
dataline = f.readlines()
data_len = 0
for i in dataline:
ftp_obj.send(i.encode('utf-8'))
data_len += len(i)
time.sleep(0.1)
progress_bar(data_len, len(data))
print("上传成功")
is_check = ftp_obj.recv(10)
if is_check.decode('gbk') == 'True':
print("文件上传完整")
else:
print("文件上传不完整")
else:
print("文件太大，超出磁盘限额")
continue
else:
print("命令错误，重重新输入")
continue

def main():
auth_res = login_auth(ftp_obj)
if auth_res:
ftp_cmd(ftp_obj)
else:
print("用户或密码不正确，退出程序")
sys.exit(1)
if __name__ == '__main__':
main()

ftp服务端脚本ftpserver.py

#!/usr/bin/python
# --*-- coding:utf-8
import socket
import json
import os
import subprocess
import struct
from hashlib import md5
import time

dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
hyh_size = 5    #'hyh'用户M空间限额5M
lisi_size = 10  #'lisi'用户空间限额10M

ftp_obj = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ftp_obj.bind(('127.0.0.1', 8080))
ftp_obj.listen(5)

def size(user_home_dir):
"""计算用户家目录大小"""
# print(user_home_dir)
# print(os.listdir(user_home_dir))
file_size_bytes = 0
for i in os.listdir(user_home_dir):
file = user_home_dir + '\\' + i
file_size_bytes += os.path.getsize(file)
file_size_mbytes = file_size_bytes / (1024 * 1024)
return file_size_mbytes

def get_md5(data):
"""校验文件内容"""
m = md5()
m.update(data.encode('utf-8'))
res = m.hexdigest()
return res

def ftp_load(conn, user_name):
"""接收客户端命令，并执行"""
user_home_dir = dir + '\\' + user_name
used_size = size(user_home_dir)
while True:
try:
print('开始接收发送数据')
cmd_res_bytes = conn.recv(1024)
cmd_res = cmd_res_bytes.decode('utf-8')
if cmd_res == 'dir':
res = subprocess.Popen(cmd_res + ' ' + user_home_dir, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
err = res.stderr.read()
if err:
cmd_stdout = err
else:
cmd_stdout = res.stdout.read()
#print(cmd_stdout.decode('gbk'))
head_dict = {'dir': user_name, 'hashlib': None, 'total_size': len(cmd_stdout)}
head_json = json.dumps(head_dict)
head_bytes = head_json.encode('utf-8')
conn.send(struct.pack('i', len(head_bytes)))
conn.send(head_bytes)
conn.send(cmd_stdout)
else:
cmd_res_list = cmd_res.split()  #获取命令和文件列表
user_home_file = user_home_dir + '\\' + cmd_res_list[1] #获取文件
if cmd_res_list[0] == 'get':
with open(user_home_file, 'r', encoding='utf-8') as f:
data = f.read()
check_md5 = get_md5(data)   #校验文件内容md5
head_dict = {'filename': cmd_res_list[1], 'hashlib': check_md5, 'total_size': len(data)}
head_json = json.dumps(head_dict)
head_bytes = head_json.encode('utf-8')
conn.send(struct.pack('i', len(head_bytes)))
conn.send(head_bytes)
with open(user_home_file, 'r', encoding='utf-8') as f:
dataline = f.readlines()
for i in dataline:
time.sleep(0.1)
conn.send(i.encode('utf-8'))
#conn.send(data.encode('utf-8'))
else:
head_struct = conn.recv(4)
head_len = struct.unpack('i', head_struct)[0]
head_bytes = conn.recv(head_len)
head_json = head_bytes.decode('utf-8')
head_dict = json.loads(head_json)
total_size = head_dict['total_size']
if user_name == 'hyh':
if hyh_size - used_size > total_size / (1024 * 1024):
conn.send('True'.encode('utf-8'))
recv_size = 0
data = b''
while recv_size < total_size:
recv_data = conn.recv(1024)
data += recv_data
recv_size += len(recv_data)
with open(user_home_file, 'w', encoding='utf-8') as f:
f.write(data.decode('utf-8'))
check_put_md5 = get_md5(data.decode('utf-8'))
if head_dict['hashlib'] == check_put_md5:
conn.send('True'.encode('utf-8'))
continue
else:
conn.send('False'.encode('utf-8'))
continue
else:
conn.send('False'.encode('utf-8'))
continue
elif user_name == 'lisi':
if lisi_size - used_size > total_size / (1024 * 1024):
conn.send('True'.encode('utf-8'))
recv_size = 0
data = b''
while recv_size < total_size:
recv_data = conn.recv(1024)
data += recv_data
recv_size += len(recv_data)
with open(user_home_file, 'w', encoding='utf-8') as f:
f.write(data.decode('utf-8'))
continue
else:
conn.send('False'.encode('utf-8'))
continue
else:
continue
except Exception:
break

while True:
print('waitting to accept...')
conn, addr = ftp_obj.accept()
print('client: ', addr)
while True:
try:
user_name_bytes = conn.recv(30)
user_passwd_bytes = conn.recv(30)
user_name = user_name_bytes.decode('gbk')
user_passwd = user_passwd_bytes.decode('gbk')
with open('user.json', 'r', encoding='utf-8') as f:
user_dict = json.loads(f.read())
if user_name in user_dict and user_dict[user_name] == user_passwd:
conn.send('True'.encode('utf-8'))
print("ftp连接成功")
ftp_load(conn, user_name)  #ftp上传下载实现
else:
conn.send('False'.encode('utf-8'))
except Exception:
conn.send('False'.encode('utf-8'))
conn.close()
ftp_obj.close()