day 29 socketsetserver 模块

1.FTP上传/下载服务端/客户端.

-----------------------------------------------------------------------------------------

上传服务端

import jsonimport socketimport struct

server = socket.socket()server.bind(('192.168.12.48', 8001))server.listen()conn, addr = server.accept()

#首先接收文件的描述信息的长度struct_data_len = conn.recv(4)data_len = struct.unpack('i', struct_data_len)[0]

# 通过文件信息的长度将文件的描述信息全部接收print('data_len>>>', data_len)file_info_bytes = conn.recv(data_len)#将文件描述信息转换为字典类型,以便操作file_info_json = file_info_bytes.decode('utf-8')file_info_dict = json.loads(file_info_json) #{'file_name': 'aaa.mp4', 'file_size': 24409470}

print(file_info_dict)

#统计每次接收的累计长度recv_sum = 0

#根据文件描述信息,指定文件路径和文件名称file_path = 'D:\s18\jj' + '\\' + file_info_dict['file_name']

#接收文件的真实数据with open(file_path, 'wb') as f:    #循环接收,循环结束的依据是文件描述信息中文件的大小,也是通过一个初始值为0的变量来统计    while recv_sum < file_info_dict['file_size']:        every_recv_data = conn.recv(1024)        recv_sum += len(every_recv_data)        f.write(every_recv_data)

下载客户端

import osimport socketimport jsonimport structclient = socket.socket()client.connect(('192.168.12.48', 8001))

#统计文件大小file_size = os.path.getsize(r'D:\python_workspace_s18\day029\aaa.mp4')

#统计文件描述信息,给服务端,服务端按照我的文件描述信息来保存文件,命名文件等等,现在放到一个字典里面了file_info = {    'file_name': 'aaa.mp4',    'file_size': file_size,}#由于字典无法直接转换成bytes类型的数据,所以需要json来将字典转换为json字符串.在把字符串转换为字节类型的数据进行发送#json.dumps是将字典转换为json字符串的方法file_info_json = json.dumps(file_info)

#将字符串转换成bytes类型的数据file_info_byte = file_info_json.encode('utf-8')

#为了防止黏包现象,将文件描述信息的长度打包后和文件的描述信息的数据一起发送过去data_len = len(file_info_byte)data_len_struct = struct.pack('i',data_len)

#发送文件描述信息client.send(data_len_struct + file_info_byte)

#定义一个变量,=0,作为每次读取文件的长度的累计值sum = 0#打开的aaa.mp4文件,rb的形式,with open('aaa.mp4','rb') as f:    #循环读取文件内容    while sum < file_size:        #每次读取的文件内容,每次读取1024B大小的数据        every_read_data = f.read(1024)        #将sum累加,统计长度        sum += len(every_read_data)        #将每次读取的文件的真实数据返送给服务端        client.send(every_read_data)

-------------------------------------------------------------------------------2 socketserver02 socketserver服务端.py

import socketserver

class Myserver(socketserver.BaseRequestHandler):

def handle(self):
while 1:
from_client_msg = self.request.recv(1024) # self.request = conn
print(from_client_msg.decode('utf-8'))
msg = input('服务端说:')
self.request.send(msg.encode('utf-8'))

if __name__ == '__main__':

ip_port = ('127.0.0.1',8001)

server = socketserver.ThreadingTCPServer(ip_port,Myserver)

server.serve_forever()

02 socketserver客户端.py

import socket
client = socket.socket()
client.connect(('127.0.0.1',8001))

while 1:
msg = input('客户端说>>>')
client.send(msg.encode('utf-8'))

from_server_msg = client.recv(1024)

print(from_server_msg.decode('utf-8'))

------------------------------------------------------------------------------------------

03 验证合法性连接

验证合法性连接服务端

from socket import *
import hmac,os

secret_key=b'Jedan has a big key!'
def conn_auth(conn):
'''
认证客户端链接
:param conn:
:return:
'''
print('开始验证新链接的合法性')
msg=os.urandom(32)#生成一个32字节的随机字符串
conn.sendall(msg)
h=hmac.new(secret_key,msg)
digest=h.digest()
respone=conn.recv(len(digest))
return hmac.compare_digest(respone,digest)

def data_handler(conn,bufsize=1024):
if not conn_auth(conn):
print('该链接不合法,关闭')
conn.close()
return
print('链接合法,开始通信')
while True:
data=conn.recv(bufsize)
if not data:break
conn.sendall(data.upper())

def server_handler(ip_port,bufsize,backlog=5):
'''
只处理链接
:param ip_port:
:return:
'''
tcp_socket_server=socket(AF_INET,SOCK_STREAM)
tcp_socket_server.bind(ip_port)
tcp_socket_server.listen(backlog)
while True:
conn,addr=tcp_socket_server.accept()
print('新连接[%s:%s]' %(addr[0],addr[1]))
data_handler(conn,bufsize)

if __name__ == '__main__':
ip_port=('127.0.0.1',9999)
bufsize=1024
server_handler(ip_port,bufsize)

验证合法性连接的客户端

from socket import *
import hmac,os

secret_key=b'Jedan has a big key!'
def conn_auth(conn):
'''
验证客户端到服务器的链接
:param conn:
:return:
'''
msg=conn.recv(32)
h=hmac.new(secret_key,msg)
digest=h.digest()
conn.sendall(digest)

def client_handler(ip_port,bufsize=1024):
tcp_socket_client=socket(AF_INET,SOCK_STREAM)
tcp_socket_client.connect(ip_port)

conn_auth(tcp_socket_client)

while True:
data=input('>>: ').strip()
if not data:continue
if data == 'quit':break

tcp_socket_client.sendall(data.encode('utf-8'))
respone=tcp_socket_client.recv(bufsize)
print(respone.decode('utf-8'))
tcp_socket_client.close()

if __name__ == '__main__':
ip_port=('127.0.0.1',9999)
bufsize=1024
client_handler(ip_port,bufsize)