Python——网络编程（二）socket进阶2

#socketserver模块实现并发

socketserver就是一个py文件，可以ctrl进去看看都是如何实现的

socketserver的两个最基本的类：

 （以下箭头代表继承关系）

• server类：用于处理链接

分为以下五种：

右边两个代表专用于Unix的TCP和UDPservice，一般不用

 

 

 

• request类： 用于处理通信

 

 

• 其余类都是继承以上两大类(注意继承的优先级是由左到右递减)

“ThreadingUDPServer", "ThreadingTCPServer", "ThreadingMixIn"

"ForkingUDPServer","ForkingTCPServer", "ForkingMixIn"

 

 

 

import socketserver

buffer_size = 1024
ip_port = ('222.195.137.208', 8000)

#需要继承socketserver.BaseRequestHandler类
class MyServer(socketserver.BaseRequestHandler):

    #一定需要自己重新定义handle
    def handle(self):
        print(self.request)         #相当于原来的conn
        print(self.client_address)  #相当于原来的addr

        #通讯循环
        while True:
            try:
                # 接收客户端消息
                data = self.request.recv(buffer_size)

                if not data:break
                print('客户端发来的消息: ', data.decode('utf8'))

                #发送消息
                self.request.sendall(data.upper())
            except Exception as e:
                print(e)
                break


if __name__ == '__main__':
    #链接循环

    # 创建多线程服务端实例(来一个链接创建一个实例，进行通信)，其中MyServer代表的就是通信循环
    s = socketserver.ThreadingTCPServer(ip_port, MyServer)
    #这代表'永远服务'，也就是链接循环
    s.serve_forever()

from socket import *

ip_port = ('222.195.137.208', 8000)
buffer_size = 1024

client = socket(AF_INET, SOCK_STREAM)
print('连接中...')
client.connect(ip_port)
print('连接成功: %s!' %ip_port[0])

while True:
    msg = input('>>')

    if not msg: continue
    if msg == 'exit': break
    client.send(msg.encode('utf8'))

    data = client.recv(buffer_size)

    print('收到消息:\n%s' %data.decode('utf8'))

client.close()

  

import socketserver

buffer_size = 1024
ip_port = ('222.195.137.208', 8000)

class MyServer(socketserver.BaseRequestHandler):

    def handle(self):
        print(self.request)         #相当于原来的(接收的数据data, udp的套接字对象) --》 元组形式
        print(self.client_address)  #相当于原来的addr
        #和tcp不同的是，这里不需要再通讯循环了，udp的serverforever帮你做了
        try:
             #由于消息已经被接收到了self.request里，这里无需再接收
            data = self.request[0]

            print('客户端发来的消息: ', data.decode('utf8'))

            #发送消息
            self.request[1].sendto(data.upper(), self.client_address)
        except Exception as e:
            print(e)

if __name__ == '__main__':

    s = socketserver.ThreadingUDPServer(ip_port, MyServer)
    s.serve_forever()

from socket import *

ip_port = ('222.195.137.208', 8000)
buffer_size = 1024

client = socket(AF_INET, SOCK_DGRAM)

while True:
    msg = input('>>')

    if not msg: continue
    if msg == 'exit': break
    client.sendto(msg.encode('utf8'), ip_port)

    data, addr = client.recvfrom(buffer_size)

    print('收到消息:\n%s' %data.decode('utf8'))

client.close()

 如果需要使地址重用的话，加上这句

#验证客户端链接合法性

先介绍两个模块:

os.urandom():

函数定位： Return a string of n random bytes suitable for cryptographic use. 
意思就是，返回一个有n个byte那么长的一个string，然后很适合用于加密

hmac

和hashlib模块类似，主要介绍以下几个方法

import hmac, os
#产生32位的随机bytes
msg = os.urandom(32)
#设置密匙
secret_key = b'wwj'
#产生加密信息
h1 = hmac.new(secret_key, msg)
m1 = h1.digest()

h2 = hmac.new(msg)
m2 = h2.digest()

#比较两个加密信息是否一致
print(hmac.compare_digest(m1, m2))

#验证客户端链接合法性思路

1.服务端给客户端发送指定位数的随机信息

1.客户端收到此随机信息

 

2.服务端对此信息进行添加密匙的加密

2.客户端对此信息进行添加密匙的加密

 

3.客户端发送加密信息给服务端

3.服务度收到客户端的加密信息，与自己的加密信息进行比对，相同则认证通过

import hmac, os

msg = os.urandom(32)
secret_key = b'wwj'

conn.sendall(msg)
h = hmac.new(secret_key, msg)
digest = h.digest()
respone = conn.recv(len(digest))

print(hmac.compare_digest(digest, respone))

import hmac, os

secret_key = b'wwj'
msg = conn.recv(32)
h = hmac.new(secret_key, msg)
digest = h.digest()
conn.sendall(digest)

 这样想要合法连接，需要：

1. 知道服务端的认证机制

2. 知道客户端的secret_key