day8--socket文件传输

时间:2023-03-09 04:47:37
day8--socket文件传输

FTP server

1、读取文件名

2、检测文件是否存在

3、打开文件

4、检测文件大小(告诉客户端发送文件的大小)

5、发送文件大小和MD5值给客户端,MD5

6、等待客户端确认(防止粘包)

7、开始边读边发数据

8、发送完整的MD5

9、关闭服务器

socket收发文件,服务器端打开并发送文件,客户端接收并存储文件;这样就能实现文件的传输功能。

os模块中的os.path.isfile()和os.path.exists():

os.path.isfile()是用来判断文件是否存在的,只能判断文件,不能判断目录,如下所示:

判断目录:

import os
file_path = os.path.abspath(__file__)
print(os.path.isfile('/home/zhuzhu/第八天'))
print(os.path.exists("/home/zhuzhu/第八天"))
输出如下:
False
True

从上面代码可以看出,isfile()是判断文件是否存在,不能够判断目录是否存在,而exists()能够判断目录是否存在。

判断文件:

import os
file_path = os.path.abspath(__file__)
print(os.path.isfile('file_test'))
print(os.path.exists("file_test"))
运行结果如下:
True
True

从上面代码可以看出,isfile()和exists()都能够判断文件是否存在;

从上面我们得出如下结论,isfile()只能判断文件是否存在,不能够判断目录是否存在;而exists()能够判断文件和目录是否存在。

其实hashlib就是防止破解的,用户该输入什么就输入什么,只是按照要求的方式,把用户输入的密码加密,以别人看不懂的格式进行存储,防止暴力破解。对客户本身来说是没有影响的,只是表现的方式不一样而已。

加密:

import hashlib
def encryption():
'''
对用户输入的内容进行加密
:return:
'''
while True:
passwrod = input("请输入您的密码>>:")
m = hashlib.md5()
m.update(passwrod.encode('utf-8'))
print("在系统存储的密码如下:%s" %m.hexdigest()) encryption()

上面代码是对用户输入进行加密处理,加密之后,返回加密后的暗文,其实加密,就是不想别人看到密码原始的样子,如下所示:

请输入您的密码>>:asdfasfd
在系统存储的密码如下:7b4dafa43458d3a6a232afdd184ecb53
请输入您的密码>>:afdsasfd
在系统存储的密码如下:5ae8c46414716d2b606ff7c7109c6b1b
请输入您的密码>>:asfda
在系统存储的密码如下:933a3cde50f232f9f2eb3ef106a7b2d4
请输入您的密码>>:dfasdf
在系统存储的密码如下:991d4860883cbdec8effc3f3a3b71bce
请输入您的密码>>:asdf
在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570
请输入您的密码>>:adsf
a在系统存储的密码如下:05c12a287334386c94131ab8aa00d08a
请输入您的密码>>:sdf
在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570
请输入您的密码>>:asdf
as在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570
请输入您的密码>>:df
在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570
请输入您的密码>>:asdf
在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570
请输入您的密码>>:adsf
在系统存储的密码如下:05c12a287334386c94131ab8aa00d08a
请输入您的密码>>:adfs
在系统存储的密码如下:6a09965fb1ad7f14539e569c264b15ef
请输入您的密码>>:sadf
在系统存储的密码如下:fe6d1fed11fa60277fb6a2f73efb8be2
请输入您的密码>>:
在系统存储的密码如下:a3590023df66ac92ae35e3316026d17d
请输入您的密码>>:123456a
在系统存储的密码如下:9cbf8a4dcb8e30682b927f352d6559a0
请输入您的密码>>:adfa
d在系统存储的密码如下:4ac174730d4143a119037d9fda81c7a9
请输入您的密码>>:fas
在系统存储的密码如下:5a5dc3936c05c32e61aa539e7ffb40c0
请输入您的密码>>:fdasfd
a在系统存储的密码如下:a87b8bcbfe5a5030fc01e7688e2bbc94
请输入您的密码>>:sdf
在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570
请输入您的密码>>:asdf
a在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570
请输入您的密码>>:sdf
在系统存储的密码如下:912ec803b2ce49e4a541068d495ab570

上面,用户输入密码,转换成MD5密码值,这样就是别人看到数据库里面的密码,也不能反解用户的密码,保护用户的隐私。

os.stat(path)判断文件大小,如下所示:

import hashlib,os
file_size = os.stat('file_test')
print(file_size)
print(file_size.st_size)
运行结果如下:
os.stat_result(st_mode=33204, st_ino=17320086, st_dev=64768, st_nlink=1, st_uid=1000, st_gid=1000, st_size=150, st_atime=1503270341,
st_mtime=1503270327, st_ctime=1503270327)
150

上面文件中,我们使用stat()来判断文件大小,存储属性,取其中的文件大小的属性,得到结果。

服务器端:

import socket,os,time,hashlib
server = socket.socket()
server.bind(("localhost",))
server.listen()
while True:
conn,addr = server.accept()
print("new conn:",addr)
while True:
print("等待新指令")
data = conn.recv()
if not data:
print("客户端已经断开")
break
cmd,filename = data.decode().split() #data是一个byte形式,先转换
print(filename)
if os.path.isfile(filename): #isfile是判断文件是否存在
f = open(filename,'rb') #以字节码形式打开
m = hashlib.md5()
file_size = os.stat(filename).st_size
'''hashlib不能直接对文件进行加密'''
conn.send(str(file_size).encode('utf-8'))
conn.recv() #等待客户端接收成功,以便下一次发送 wait for ack
'''循环发送,读一行发一行'''
for line in f:
# m.update(line)
conn.send(line) #发送文件,循环完成之后发送MD5
# print("file md5",m.hexdigest())
f.close() print('send done') server.close()

上面服务器端是接收客户端文件名的指令,然后打开文件,并发送文件的内容,逐行读取并发送给客户端。

客户端:

import socket
client = socket.socket()
client.connect(("localhost",)) while True:
cmd = input(">>:").strip()
if len(cmd) == :continue
if cmd.startswith('get'):
'''代表接收文件'''
client.send(cmd.encode('utf-8'))
server_response = client.recv() #接收文件大小
print("server response:",server_response) #打印文件大小
client.send(b"ready to recv file")
file_total_size = int(server_response.decode())
received_size =
filename = cmd.split()[]
f = open(filename+".new",'wb') #文件名不能一样,不然会覆盖之前我文件名
while received_size < file_total_size:
data = client.recv()
received_size += len(data)
f.write(data)
# print(file_total_size,received_size)
else:
print("file recv done",received_size,file_total_size)
f.close() client.close()

客户端是用来接收数据,并存储到文件中,数据在socket()中的传输,其实就是接收和发送两个步骤,在传输的过程中,只能接收和发送字节,并且两次发送一定要防止粘包,因为发送数据是同步的,因此要客户端完全接收之后,服务器端再进行下一次的发送。

下面进行MD5验证,加入MD5验证,传输速度会变慢: