python全栈开发技巧_python全栈开发 * 35 知识点汇总 * 180720

35 socket 一些常用方法 验证客户端信息 添加随机字符串 socketserver模块

一.socket模块中一些常用方法

1.send和sendall的区别

2.conn.getpeername conn/ sk.getsockname

3.setsockopt() 3.getsockopt()

4.sk.setblocking()

5.settimeout() gettimeout()

二.客户端验证相关事宜

第一版 验证客户端

服务器:

import socket

import hashlib

sk=socket.socket()

sk.bind(("127.0.0.1",8809))

sk.listen()

conn,addr=sk.accept()

key="天王盖地虎"

ch="你好"

conn.send(ch.encode("utf-8"))

md5_obj=hashlib.md5(key.encode("utf-8"))

md5_obj.update(ch.encode("utf-8"))

re=md5_obj.hexdigest()

client_re=conn.recv(1024).decode("utf-8")

if re==client_re:

print("你好,老铁")

else:

print("你是假冒用户")

conn.close()

sk.close()

客户端:

import socket

import hashlib

sk=socket.socket()

sk.connect(("127.0.0.1",8809))

key="天王盖地虎" #宝塔镇河妖

ch=sk.recv(1024)

md5_obj=hashlib.md5(key.encode("utf-8"))

md5_obj.update(ch)

re=md5_obj.hexdigest()

sk.send(re.encode("utf-8"))

sk.close()

第二版 添加随机字符串

服务器 (import os os.urandom(num))

import socket

import hashlib

import os

sk=socket.socket()

sk.bind(("127.0.0.1",8809))

sk.listen()

conn,addr=sk.accept()

key="天王盖地虎"

ch=os.urandom(10) #===>字节

conn.send(ch)

md5_obj=hashlib.md5(key.encode("utf-8"))

md5_obj.update(ch)

re=md5_obj.hexdigest() #===>字符串

client_re=conn.recv(1024).decode("utf-8")

if re==client_re:

print("你好,老铁")

else:

print("你是假冒用户")

conn.close()

sk.close()

客户端

import socket

import hashlib

sk=socket.socket()

sk.connect(("127.0.0.1",8809))

key="天王盖地虎" #宝塔镇河妖

ch=sk.recv(1024)

print(ch)

md5_obj=hashlib.md5(key.encode("utf-8")) #(括号里应该是字节)

md5_obj.update(ch) #(括号里应该是字节)

re=md5_obj.hexdigest() #====>字节

sk.send(re.encode("utf-8"))

sk.close()

第三版 用hmac加密方法

服务器

import socket

import hmac #hmac 模块 (内置模块)

import os

sk=socket.socket()

sk.bind(("127.0.0.1",8809))

sk.listen()

conn,addr=sk.accept()

key="天王盖地虎" # 加盐

ch=os.urandom(10) # os中的urandom方法(获取随机字符串) ===>字节

conn.send(ch) #字节可以直接发送,无需编码

obj=hmac.new(key.encode("utf-8"),ch) #模块hmac中的new方法给随机字符串 ch 加密赋值给obj

re=obj.digest() #=====>直接返回Bytes类型

# md5_obj=hashlib.md5(key.encode("utf-8"))

# md5_obj.update(ch)

# re=md5_obj.hexdigest() #===>字符串

client_re=conn.recv(1024)

if re==client_re:

print("你好,老铁")

else:

print("你是假冒用户")

conn.close()

sk.close()

客户端

import socket

import hmac # 内置模块hmac

sk=socket.socket()

sk.connect(("127.0.0.1",8809))

key="天王盖地虎" # 加盐

ch=sk.recv(1024) #接收到服务器发来的随机字符串 字节形式

obj=hmac.new(key.encode("utf-8"),ch) #用模块hmac中的new方法给随机字符串加密,括号里必须都是字节形式

re=obj.digest() #返回字节形式

# md5_obj=hashlib.md5(key.encode("utf-8")) #(括号里应该是字节)

# md5_obj.update(ch) #(括号里应该是字节)

# re=md5_obj.hexdigest() #====>字符串

sk.send(re) # 直接发送给服务器

三.socketserver模块tcp中一个服务器同时连接多个客户端

服务器

客户登陆验证

import socketserver

import json

import hashlib

class Myserver(socketserver.BaseRequestHandler):

def handle(self):

dic_str=self.request.recv(1024).decode("utf-8")

dic=json.loads(dic_str)

with open("user_info",encoding="utf-8")as f:

for i in f:

username,password=i.split("|")

if username==dic["username"]:

md5_obj=hashlib.md5(dic["username"].encode('utf-8'))

md5_obj.update(dic["password"].encode("utf-8"))

re=md5_obj.hexdigest()

if re == password.strip():

self.request.send(b'success')

else:

self.request.send(b'failed')

break

else:

self.request.send("用户名不存在".encode("utf-8"))

server=socketserver.TCPServer(("127.0.0.1",8810),Myserver)

server.serve_forever()

客户端

import socket

import json

import time

sk=socket.socket()

sk.connect(("127.0.0.1",8810))

dic={"username":None,"password":None}

while 1:

username=input("姓名:")

password=input("密码:")

dic["username"]=username

dic["password"]=password

dic_str=json.dumps(dic)

sk.send(dic_str.encode("utf-8"))

print(sk.recv(1024))