005day—网络编程-02

# tcp协议的多人多次通信
#          和一个人通信说多句话
#           和一个人聊完再和其他人聊
#           socket() tcp协议，
#           bind绑定一个ip和端口
#           listen监听，代表socket服务的开启
#           accept 等，到有客户端来访问和客户端连接
#           send直接通过连接发送消息，不需要写地址
#           recv 只接收消息
#           connect客户端tcp协议的方法，和sever端建立连接
#           close 关闭服务/连接
#
# udp协议的多人通信
#           socket(type = socket.SOCK_DGRAM)
#           sendto需要写一个对方的地址
#           recvfrom 接收消息和地址
#           close 关闭服务/连接
#recv(1024)不代表一定收到1024字节
#两条连续发送的数据一定要避免粘包问题
#先发送数据的长度在发送数据
		#发送的数据相关的内容组成json：先发json的长度，再发json，json中存放接下来要发送的数据长度。例如,下面文件上传的传送格式
#socketserver模块中代码从def handler(self)开始,当客户端来连接的时候直接和handle中的代码交互，self.request就是conn

**
tcp协议的文件上传

server.py
import socket
import json
import struct
# 接收
sk = socket.socket()
sk.bind(('127.0.0.1',9001))

sk.listen()

conn,addr = sk.accept()
length = conn.recv(4)

msg = conn.recv(struct.unpack('i',length)[0]).decode('utf-8')
msg = json.loads(msg)
print(msg)

with open(msg['filename'],mode='wb') as f:
    while msg['filesize']>0:
        content = conn.recv(1024)
        msg['filesize']-=len(content)#防止接收信息不够，由于tcp的优化，发送方发送1024个字节，接收时可能会分为几块，每一块多大，这里总长度就减去多少
        f.write(content)

conn.close()
sk.close()

client.py

import socket
import os
import struct#处理黏包
# 发送
import json

sk = socket.socket()

sk.connect(('127.0.0.1',9001))

#文件名/文件大小
abs_path = r'/home/lmh/PycharmProjects/网络编程/tmp'
filename = os.path.basename(abs_path)
filesize = os.path.getsize(abs_path)
dic = {'filename':filename,'filesize':filesize}
str_dic = json.dumps(dic)
length = struct.pack('i',len(str_dic))
sk.send(length)

sk.send(str_dic.encode('utf-8'))

with open(abs_path,mode='rb') as f:
    while filesize>0:
        content = f.read(1024)
        filesize-=len(content)
        sk.send(content)
sk.close()

验证客户端合法性

server.py

import socket
import os
import hashlib
import hmac
seret_key = b'lmh-nb'
sk = socket.socket()
sk.bind(('127.0.0.1',9001))
sk.listen()
conn,addr = sk.accept()

# 创建一个随即字符串
rand = os.urandom(32)
# 发送ret
conn.send(rand)


'''
hashlib方法加密
'''
# sha = hashlib.sha1(seret_key)
# sha.update(rand)
# res = sha.hexdigest()

'''
hmac方法加密
'''
sha = hmac.new(seret_key,rand,digestmod='sha1')
res = sha.digest()

res_client = conn.recv(1024)

if res_client==res:
    print('成功')
    conn.send(b'hello')

else:
    conn.close()

client.py

import socket
import hmac#可以替代hashlib
import hashlib
seret_key = b'lmh-nb'
sk = socket.socket()
sk.connect(('127.0.0.1',9001))


# 接收server端发送的随即字符串
rand = sk.recv(32)

# 根据发送的字符串和secret_key进行摘要
''''
hashlib方法加密
'''
# sha = hashlib.sha1(seret_key)
# sha.update(rand)
# res = sha.hexdigest()

'''
hmac方法加密
'''
sha = hmac.new(seret_key,rand,digestmod='sha1')
res = sha.digest()
# 摘要结果发送给server端


sk.send(res)
msg = sk.recv(1024)
print(msg)

hmac模块
import hmac#可以替代hashlib
import os
h = hmac.new(b'lmh',os.urandom(32),digestmod='sha1')
ret = h.digest()#ret得到的是bytes
print(ret)

socketserver模块

server.py
import time
import socketserver
class Myserver(socketserver.BaseRequestHandler):#固定写法
    def handle(self) -> None:#每个客户端连接后直接进入这个函数开始执行，
        conn = self.request
        while True:
            try:
                content = conn.recv(1024).decode('utf-8')
                conn.send(content.upper().encode('utf-8'))
                time.sleep(0.5)
            except ConnectionResetError:
                break

server = socketserver.ThreadingTCPServer(('127.0.0.1',9001),Myserver)#固定的写法
server.serve_forever() #固定的写法

client.py#客户端代码不用变
import socket
sk = socket.socket()
sk.connect(('127.0.0.1',9001))
while True:
    sk.send(b'hello')
    content = sk.recv(1024).decode('utf-8')
    print(content)