Socket_FTP

1. md5加密回顾：

import hashlib

m=hashlib.md5() #创建md5对象
m.update(b'abcd') #生成加密串
m.update(b'efg')
print(m.hexdigest()) #打印加密串
m2=hashlib.md5()
m2.update(b'abcdefg')
print(m2.hexdigest())

7ac66c0f148de9519b8bd264312c4d64
7ac66c0f148de9519b8bd264312c4d64

FTP-Server

1.读取文件名 
2.检测文件是否存在 
3.打开文件 
4.检测文件大小 
5.发送文件大小给客户端 
6.等待客户端确定，防止粘包 
7.开始边读边发数据 
8.发送md5 
9.关闭文件

2. FTP入门版：

服务器端：

import socket,os,time,hashlib

server=socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
    conn,addr=server.accept()
    print("new connection is :",addr)
    while True:
        data=conn.recv(1024)
        if not data:
            print("客户端已经断开")
            break
        cmd,filename=data.decode().split()  #分割get hash.py,把get赋值给cmd,把hash.py赋值给filename
        print(filename)
        if os.path.isfile(filename): #判断是否是文件
            f=open(filename,'rb')    #以二进制，读模式打开文件
            m=hashlib.md5() #创建md5对象
            file_size=os.stat(filename).st_size #获取文件大小
            conn.send(str(file_size).encode())  #给客户端发送文件大小
            conn.recv(1024) #等待客户端的确认
            for line in f:
                conn.send(line)
            f.close()

server.close()

客户端：

import socket

client=socket.socket()
client.connect(('localhost',9999))
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') #以二进制打开一个文件，只用于写入。
        while received_size<file_total_size:
            data=client.recv(1024)
            received_size=received_size+len(data)
            f.write(data)
            print(file_total_size,received_size)
        else:
            print('file recv done')
            f.close()

client.close()

客户端的输入及运行结果：

>>:get hash.py
server response: b'211'
211 211
file recv done
>>:

服务器端的运行结果：

new connection is : ('127.0.0.1', 65194)
hash.py

3. FTP+MD5验证

服务器端：

import socket,os,time,hashlib

server=socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
    conn,addr=server.accept()
    print("new connection is :",addr)
    while True:
        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() #创建md5对象
            file_size=os.stat(filename).st_size #获取文件大小
            conn.send(str(file_size).encode())  #发送文件大小
            conn.recv(1024) #等待客户端的确认
            for line in f:
                m.update(line) #生产加密串
                conn.send(line)
            print('file md5',m.hexdigest())
            f.close()
            conn.send(m.hexdigest().encode('utf-8')) #send md5

server.close()

客户端：

import socket,hashlib
import hashlib

client=socket.socket()
client.connect(('localhost',9999))
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:
            data=client.recv(1024)
            received_size=received_size+len(data)
            f.write(data)
            m.update(data)
            #print(file_total_size,received_size)
        else:
            new_file_md5=m.hexdigest()
            print('file recv done',received_size,file_total_size)
            f.close()
        server_file_md5=client.recv(1024)
        print('server file md5:',server_file_md5)
        print('client file md5:',new_file_md5)

client.close()

服务器端运行结果：

new connection is : ('127.0.0.1', 65466)
hash.py
file md5 5dbf115019480585de4d39cd175184db

客户端运行结果：

>>:get hash.py
server response: b'211'
file recv done 211 211
server file md5: b'5dbf115019480585de4d39cd175184db'
client file md5: 5dbf115019480585de4d39cd175184db

4.FTP+MD5验证+粘包处理：

服务器端：

import socket,os,time,hashlib

server=socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
    conn,addr=server.accept()
    print("new connection is :",addr)
    while True:
        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() #创建md5对象
            file_size=os.stat(filename).st_size #获取文件大小
            conn.send(str(file_size).encode())  #发送文件大小
            conn.recv(1024) #等待客户端的确认
            for line in f:
                m.update(line) #生产加密串
                conn.send(line)
            print('file md5',m.hexdigest())
            f.close()
            conn.send(m.hexdigest().encode('utf-8')) #send md5

server.close()

客户端：

import socket,hashlib
import hashlib

client=socket.socket()
client.connect(('localhost',9999))
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=received_size+len(data)
            f.write(data)
            m.update(data)
            #print(file_total_size,received_size)
        else:
            new_file_md5=m.hexdigest()
            print('file recv done',received_size,file_total_size)
            f.close()
        server_file_md5=client.recv(1024)
        print('server file md5:',server_file_md5)
        print('client file md5:',new_file_md5)

client.close()

服务器端的输出：

new connection is : ('127.0.0.1', 56460)
hash.py
file md5 5dbf115019480585de4d39cd175184db

客户端的输出：

>>:get hash.py
server response: b'211'
last receive: 211
last receive: 2
file recv done 211 211
server file md5: b'5dbf115019480585de4d39cd175184db'
client file md5: 5dbf115019480585de4d39cd175184db