编写socket调用类似于FTP实现文件传输,MD5比对。
Socket编程实现文件传输要求:FTP Server
1、读取文件名
2、检测文件是否存在
3、打开文件
4、检测文件大小
5、发送文件大小给客户端
6、等客户端确认
7、开始边读边发数据
8、发送MD5进行校验
操作前,要注意保持,客户端与服务器端的python版本一致。切记,要不然搞死你!!!使用客户端的pycharm调用报以下错误:
>>:get web38log.csv
Traceback (most recent call last):
File "/Users/mac/PycharmProjects/untitled2/51CTO/6day/ftp_client.py", line 10, in <module>
client.send(cmd.encode())
BrokenPipeError: [Errno 32] Broken pipe
所以,操作验证时,找了台liunx机器,同时本机器启动客户端、服务端,进行测试。
1、服务端编写:注意书写的格式(排查了两个小时。。郁闷?)
import socket,os,hashlib
server = socket.socket()
server.bind(('0.0.0.0',3617))
server.listen()
while True:
conn,addr = server.accept()
print("new conn:",addr)
while True:
print ("等待新指令")
data = conn.recv(1024)
if not data:
print("客户端已断开")
break
cmd,filename = data.decode().split()
print (filename)
if os.path.isfile(filename): #判断文件是否存在
f = open(filename,"rb")
m = hashlib.md5()
file_size = os.stat(filename).st_size #读取文件大小
conn.send(str(file_size).encode()) #把文件大小数值发送给客户端
conn.recv(1024) #每次接收1024
for line in f:
m.update(line) #读取文件名
conn.send(line) #发送文件
#print("file md5:",m.hexdigest()) #服务端打印md5
f.close()
conn.send(m.hexdigest().encode()) #发送md5
print("send done")
server.close()
2、客户端编写:
import socket,os,hashlib
client = socket.socket()
client.connect(("200.66.5.7",3617)) #注意服务端的IP
while True:
cmd = input(">>:").strip()
if len(cmd) == 0: continue
if cmd.startswith("get"):
client.send(cmd.encode())
server_response = client.recv(1024)
print("server_response:",server_response)
client.send(b"ready to recv file")
file_total_size = int(server_response.decode())
received_size = 0
filename = cmd.split()[1]
f = open(filename + ".new","wb")
m = hashlib.md5()
while received_size < file_total_size:
if file_total_size - received_size > 1024: #要不停的收取
size = 1024
else: #最后一次收取
size = file_total_size - received_size
print("last receive:",size)
data = client.recv(size)
received_size += len(data)
m.update(data)
f.write(data)
else:
new_file_md5 = m.hexdigest()
print("file receive done...",received_size,file_total_size)
f.close()
server_file_md5 = client.recv(1024)
print("server file md5:",server_file_md5)
client.close()
3、测试验证:
启动服务器端:
# /opt/Python-3.5.2/python ssh_socket_server.py 执行命令,看服务端输出
new conn: ('203.66.5.4', 41440)
等待新指令
web27_php-fpm_slow.log-20180103
send done
等待新指令
启动客户端验证:
>>:get api167_host.php_20180103.log
server_response: b'2182947620'
last receive: 952
file receive done... 2182947620 2182947620
server file md5: b'3525782ffc5ef350a91ec7b571721d13'
传输完成,并在本地路径上,新建了个.new结尾的文件
本机测试完成传输已经完成,可以尝试通过其他客户端机器,进行验证。。注意,文件传输的情况下,只能对当前目录下的文件操作。。。。。本次socket 的传输还有很大的优化空间!!!