本文共 3823 字，大约阅读时间需要 12 分钟。

昨晚的高网实验要求用Socket编程实现两台机器的文件传输，还要记录传输时间并进行MD5验证，于是便用python简单实现了下。

HEAD_STRUCT = '128sIq32s'

它就相当于c中如下结构体:

struct{    char fileName[128]; //文件名    int fileNameSize;  //文件名长度    long long fileSize;  //文件大小    char MD5[32];  //MD5值}

这个结构体的空间大小是固定的，于是当client按照这个格式发送文件信息的时候，server也以相同的格式接收信息。由于文件名的长度不固定，于是这里我还发送了一个文件名的长度，这样server就可以根据这个长度对fileName进行截取。

# receiveimport socketimport hashlibimport structHOST = 'localhost'PORT = 1307BUFFER_SIZE = 1024HEAD_STRUCT = '128sIq32s'info_size = struct.calcsize(HEAD_STRUCT)def cal_md5(file_path):    with open(file_path, 'rb') as fr:        md5 = hashlib.md5()        md5.update(fr.read())        md5 = md5.hexdigest()        return md5def unpack_file_info(file_info):    file_name, file_name_len, file_size, md5 = struct.unpack(HEAD_STRUCT, file_info)    file_name = file_name[:file_name_len]    return file_name, file_size, md5def recv_file():    try:        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)        server_address = (HOST, PORT)        sock.bind(server_address)        sock.listen(1)        client_socket, client_address = sock.accept()        print "Connected %s successfully" % str(client_address)        file_info_package = client_socket.recv(info_size)        file_name, file_size, md5_recv = unpack_file_info(file_info_package)        recved_size = 0        with open(file_name, 'wb') as fw:            while recved_size < file_size:                remained_size = file_size - recved_size                recv_size = BUFFER_SIZE if remained_size > BUFFER_SIZE else remained_size                recv_file = client_socket.recv(recv_size)                recved_size += recv_size                fw.write(recv_file)        md5 = cal_md5(file_name)        if md5 != md5_recv:            print 'MD5 compared fail!'        else:            print 'Received successfully'    except socket.errno, e:        print "Socket error: %s" % str(e)    finally:        sock.close()if __name__ == '__main__':    recv_file()

# sendimport socketimport osimport hashlibimport structHOST = 'localhost'PORT = 1307BUFFER_SIZE = 1024HEAD_STRUCT = '128sIq32s'def cal_md5(file_path):    with open(file_path, 'rb') as fr:        md5 = hashlib.md5()        md5.update(fr.read())        md5 = md5.hexdigest()        return md5def get_file_info(file_path):    file_name = os.path.basename(file_path)    file_name_len = len(file_name)    file_size = os.path.getsize(file_path)    md5 = cal_md5(file_path)    return file_name, file_name_len, file_size, md5def send_file(file_path):    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)    server_address = (HOST, PORT)    file_name, file_name_len, file_size, md5 = get_file_info(file_path)    file_head = struct.pack(HEAD_STRUCT, file_name, file_name_len, file_size, md5)    try:        print "Start connect"        sock.connect(server_address)        sock.send(file_head)        sent_size = 0        with open(file_path) as fr:            while sent_size < file_size:                remained_size = file_size - sent_size                send_size = BUFFER_SIZE if remained_size > BUFFER_SIZE else remained_size                send_file = fr.read(send_size)                sent_size += send_size                sock.send(send_file)    except socket.errno, e:        print "Socket error: %s" % str(e)    finally:        sock.close()        print "Closing connect"if __name__ == '__main__':    file_path = raw_input('Please input file path:')    if not file_path:        file_path = 'test.txt'    send_file(file_path)

注意： - 在某些时候（比如传输大文件）会发生丢包的情况，这时候接收的文件大小会小于发送文件的大小，两个MD5值也会不一样。

转载地址：https://blog.csdn.net/Thare_Lam/article/details/49506565 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！