socketserver和socket的补充(验证客户端合法性)

一、socket的补充
1、参数
socket.socket(family=AF_INET,type=SOCK_STREAM,proto=0,fileno=None)
参数说明：

family	地址系列应为AF_INET(默认值ipv4),AF_INET6(ipv6),AF_UNIX,AF_CAN或AF_RDS。 (AF_UNIX 域实际上是使用本地 socket 文件来通信)
type	套接字类型应为SOCK_STREAM(默认值,tcp协议),SOCK_DGRAM(udp协议),SOCK_RAW或其他SOCK_常量之一。 SOCK_STREAM 是基于TCP的，有保障的（即能保证数据正确传送到对方）面向连接的SOCKET，多用于资料传送。 SOCK_DGRAM 是基于UDP的，无保障的面向消息的socket，多用于在网络上发广播信息。
proto	协议号通常为零,可以省略,或者在地址族为AF_CAN的情况下,协议应为CAN_RAW或CAN_BCM之一。
fileno	如果指定了fileno,则其他参数将被忽略,导致带有指定文件描述符的套接字返回。 与socket.fromfd()不同,fileno将返回相同的套接字,而不是重复的。 这可能有助于使用socket.close()关闭一个独立的插座。

2、socket更多方法介绍

服务端套接字函数
s.bind()	绑定(主机,端口号)到套接字
s.listen()	开始TCP监听
s.accept()	被动接受TCP客户的连接,(阻塞式)等待连接的到来
客户端套接字函数
s.connect()	主动初始化TCP服务器连接
s.connect_ex()	connect()函数的扩展版本,出错时返回出错码,而不是抛出异常
公共用途的套接字函数
s.recv()	接收TCP数据
s.recvfrom()	接收UDP数据
s.send()	发送TCP数据
s.sendall()	发送TCP数据
s.sendto()	发送UDP数据
s.getpeername()	连接到当前套接字的远端的地址(client地址)
s.getsockname()	当前套接字的地址(server地址)
s.setsockopt()	设置指定套接字的参数(端口复用)
s.getsockopt()	返回指定套接字的参数
s.close()	关闭套接字
面向锁的套接字方法
s.setblocking()	设置套接字的阻塞(True)与非阻塞模式(False)  *****
s.settimeout()	设置阻塞套接字操作的超时时间  accept()的等待时间
s.gettimeout()	得到阻塞套接字操作的超时时间
面向文件的套接字的函数
s.fileno()	套接字的文件描述符
s.makefile()	创建一个与该套接字相关的文件

官方文档对socket模块下的socket.send()和socket.sendall()解释如下：
socket.send(string[, flags])
Send data to the socket. The socket must be connected to a remote socket. The optional flags argument has the same meaning as for recv() above. 
Returns the number of bytes sent. Applications are responsible for checking that all data has been sent; if only some of the data was transmitted, 
the application needs to attempt delivery of the remaining data.
send()的返回值是发送的字节数量，这个数量值可能小于要发送的string的字节数，也就是说可能无法发送string中所有的数据。如果有错误则会抛出异常。



socket.sendall(string[, flags])
Send data to the socket. The socket must be connected to a remote socket. The optional flags argument has the same meaning as for recv() above. 
Unlike send(), this method continues to send data from string until either all data has been sent or an error occurs. None is returned on success. 
On error, an exception is raised, and there is no way to determine how much data, if any, was successfully sent.
尝试发送string的所有数据，成功则返回None，失败则抛出异常。



故，下面两段代码是等价的：
sock.sendall('Hello world\n')

buffer = 'Hello world\n'
while buffer:
    bytes = sock.send(buffer)
    buffer = buffer[bytes:]

 

3、验证客户端合法性
场景：如果别人知道了你的服务器的IP，那么他就可以使用扫端口的方式去连接上你的服务器，因为我们都知道，端口的范围是0-65535，
那么别人知道了你的服务器IP后，就可以循环扫这些端口，就可以连接上你的服务，你服务器所进行的一些操作，比如一些数据的传输，
就会被别人所获取，所以这个时候就需要验证客户端的合法性。



代码
服务端：
import os
import hmac
import socket

def auth(conn):
    msg = os.urandom(32) # 生成一个32位的随机的字节码(urandom生成的字节码就是bytes类型的)
    conn.send(msg)  # 把这个随机的字节码发送到client端
    # hmac接收两个参数，第一个参数相当于hashlib的盐，第二个参数是我们随机生成的字节码，两个参数都是bytes类型
    result = hmac.new(secret_key,msg) # 处理这个随机字节码，socket_key是盐
    res = result.hexdigest() # 得到结果(字符串)
    client_digest = conn.recv(1024) # 接收client端处理的结果
    if res == client_digest.decode('utf-8'):
        print('合法的连接')  # 对比成功可以继续通信
        return True
    else:
        print('不合法连接')  # 不成功
        return False

secret_key = b'xiaoming' # hmac的盐
sk = socket.socket()
sk.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
sk.bind(('127.0.0.1',8080))
sk.listen()
conn,addr = sk.accept()
if auth(conn):
    msg = conn.recv(1024)  # True正常的和client端进行沟通
    print(msg.decode('utf-8'))
    conn.close()
else:
    conn.close()  # False 直接关闭和这个客户端的连接

sk.close()


客户端：
import hmac
import socket

def auth(sk):
    msg = sk.recv(32) # 接收服务端传来的随机字节码
    result = hmac.new(secret_key,msg) # 处理接收到的随机字节码
    res = result.hexdigest() # 得到结果
    sk.send(res.encode('utf-8')) # 把结果发回给服务端，让服务端进行验证

secret_key = b'xiaoming' # 因为盐是程序员自己设置的，那么程序员写的客户端肯定知道自己的盐是什么
sk = socket.socket()
sk.connect(('127.0.0.1',8080))
auth(sk)
sk.send(b'connect success')  # 进行其他正常的和server端的沟通
sk.close()




二、socketserver
正常服务端的socket，每一次只能连接一个客户端，只有跟当前客户端断开连接后才能和下一个客户端连接，
而用socketserver可以跟多个客户端同时连接(并发)。


服务端：
import socketserver
# tcp协议的server端就不需要导入socket
class Myserver(socketserver.BaseRequestHandler):  # 继承socketserver.BaseRequestHandler这个类
    def handle(self):  # 必须继承handle方法并重写
        conn = self.request  # self.request就是客户端的对象
        while True:  # 和客户端进行交互
            conn.send(b'helloworld')
            print(conn.recv(1024).decode('utf-8'))

# 设置allow_reuse_address允许服务器重用地址
socketserver.TCPServer.allow_reuse_address = True

# 创建一个对象server,绑定ip和端口,相当于sk = socket.socket() sk.bind(('127.0.0.1',8888))这两步的结合
server = socketserver.ThreadingTCPServer(('127.0.0.1',8888),Myserver)

# 让server一直运行下去，除非强制停止程序
server.serve_forever()


客户端：
import socket
sk = socket.socket()
sk.connect(('127.0.0.1',8888))
while True:
    ret = sk.recv(1024)
    print(ret.decode('utf-8'))
    sk.send(b'hiworld')
sk.close()


解释为什么一定要重写handler方法：
Myserver这个类没有__init__方法，那么它就会去继承使用父类BaseRequestHandler的__init__方法
看看BaseRequestHandler源码：
class BaseRequestHandler:
    def __init__(self, request, client_address, server):
        self.request = request  # 获取客户端的连接(对象)，设置为自己的属性
        self.client_address = client_address  # 客户端的地址
        self.server = server  
        self.setup()
        try:
            self.handle()  # 初识化对象的时候执行handler方法
        finally:
            self.finish()

    def setup(self):
        pass

    def handle(self):
        pass

    def finish(self):
        pass

总结：
也就是说，子类继承了父类的__init__方法，这个方法里面已经取到了客户端的对象conn，和地址addr，
并且初始化的时候调用了handler方法，但是父类的handler方法并没有实现任何功能，所以子类应该重写handler方法便于与客户端交互。



实例：上传文件

server.py

import json
import struct
import socketserver
import operate_handler

class MyFTP(socketserver.BaseRequestHandler):
    def handle(self):
        conn = self.request
        length = conn.recv(4)
        length = struct.unpack('i',length)[0]
        operate = (conn.recv(length)).decode('utf-8')
        operate_dic = json.loads(operate)
        opt = operate_dic['operate']
        usr = operate_dic['user']
        print(opt,usr)
        getattr(operate_handler,opt)(conn,usr)

socketserver.TCPServer.allow_reuse_address = True
server = socketserver.ThreadingTCPServer(('127.0.0.1',9000),MyFTP)
server.serve_forever()



operate_handler.py

import os
import json
import struct

base_path = r'E:\PythonProject\ftp\server\root'

def upload(conn,usr):
    fileinfo_len = conn.recv(4)
    fileinfo_len = struct.unpack('i',fileinfo_len)[0]
    fileinfo = (conn.recv(fileinfo_len)).decode('utf-8')
    fileinfo = json.loads(fileinfo)
    file_path = os.path.join(base_path,usr,fileinfo['filename'])
    file_path = os.path.abspath(file_path)
    with open(file_path,'wb') as f:
        while fileinfo['filesize']:
            content = conn.recv(20480)
            fileinfo['filesize'] -= len(content)
            f.write(content)
    print('接收完毕')



client.py

import os
import json
import struct
import socket

# 发送信息
def my_send(sk,operate_info):
    b_optinfo = (json.dumps(operate_info)).encode('utf-8')
    num = struct.pack('i',len(b_optinfo))
    sk.send(num)
    sk.send(b_optinfo)

sk = socket.socket()
sk.connect(('127.0.0.1',9000))

# [登录,注册,退出]

# 要进行的操作
operate_info = {'operate':'upload','user':'xiaoming'}
my_send(sk,operate_info)

# 选择一个文发送到server端
file_path = r'F:\电影\电影\荒野生存.mp4'

# 发送文件信息
file_name = os.path.basename(file_path)
file_size = os.path.getsize(file_path)
file_info = {'filename':file_name,'filesize':file_size}
my_send(sk,file_info)

# server端接收写入
with open(file_path,'rb') as f:
    while file_size:
        content = f.read(20480)
        file_size -= len(content)
        sk.send(content)
print('上传完毕')
sk.close()