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
请输入您的密码>>：12346
在系统存储的密码如下：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",9998))
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()    #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(1024)   #等待客户端接收成功，以便下一次发送   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",9998))

while True:
    cmd = input(">>：").strip()
    if len(cmd) == 0:continue
    if cmd.startswith('get'):
        '''代表接收文件'''
        client.send(cmd.encode('utf-8'))
        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 += 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验证，传输速度会变慢：