3、第七 - 网络编程基础 - Socket编程实现文件传输

 

编写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 的传输还有很大的优化空间！！！