Socket案例笔记

Posted 2021-04-25 潭州Python

阻塞

服务端

 1import socket  # 导入socket模块
 2
 3sk = socket.socket()  # 创建socket对象
 4sk.bind(("127.0.0.1", 8888))  # 绑定端口,“127.0.0.1”代表本机地址，8888为设置链接的端口地址
 5sk.listen(5)  # 设置监听，最多可有5个客户端进行排队
 6conn, addr = sk.accept()  # 阻塞状态，被动等待客户端的连接
 7print(conn)  # conn可以理解客户端的socket对象
 8# <socket.socket fd=4, family=AddressFamily.AF_INET, type=SocketKind.SOCK_STREAM, proto=0, laddr=('127.0.0.1', 9005), raddr=('127.0.0.1', 36694)>
 9print(addr)  # addr为客户端的端口地址
10# ('127.0.0.1', 40966)
11accept_data = conn.recv(1024)  # conn.recv()接收客户端的内容，接收到的是bytes类型数据，
12accept_data2 = str(accept_data, encoding="utf8")  # str(data,encoding="utf8")用“utf8”进行解码
13print("".join(("接收内容：", accept_data2, "    客户端口：", str(addr[1]))))
14send_data = input("输入发送内容：")
15conn.sendall(bytes(send_data, encoding="utf8"))  # 发送内容必须为bytes类型数据，bytes(data, encoding="utf8")用“utf8”格式进行编码
16conn.close()

客户端

1import socket
2
3sk = socket.socket()
4sk.connect(("127.0.0.1", 8888))  # 主动初始化与服务器端的连接
5send_data = input("输入发送内容：")
6sk.sendall(bytes(send_data, encoding="utf8"))
7accept_data = sk.recv(1024)
8print(str(accept_data, encoding="utf8"))
9sk.close()

非阻塞

服务器

 1import socket
 2
 3server = socket.socket()
 4server.setblocking(False)
 5server.bind(('', 8888))
 6server.listen(1000)
 7connection_list = []
 8while True:
 9    try:
10        conn, r_address = server.accept()
11        conn.setblocking(False)
12        print('远程客户端{}'.format(r_address))
13        connection_list.append(conn) # 保存已经生成的套接字
14    except BlockingIOError:
15        pass
16    tmp_list = connection_list
17    for conn in tmp_list:
18        try:
19            data = conn.recv(1024)
20            if data:
21                print(str(data, encoding="utf8"))
22                conn.send(data)
23            else:
24                conn.close()
25                connection_list.remove(conn)
26        except BlockingIOError:
27            pass

selectors模块服务器

 1import socket
 2import selectors
 3
 4sel = selectors.DefaultSelector()#生成一个select对象
 5
 6def accept(sock,mask):
 7    conn,addr = sock.accept()# Should be ready
 8    print('客户：',addr)
 9    conn.setblocking(False) #设定非阻塞
10    sel.register(conn,selectors.EVENT_READ,read)#新连接注册read回调函数
11
12def read(conn,mask):
13    data = conn.recv(1024)# Should be ready
14    if data:
15        print(conn,'>>>',str(data, encoding="utf8"))
16        conn.send(data)
17    else:
18        sel.unregister(conn)
19        conn.close()
20        print('Killed：',conn)
21
22sock = socket.socket()
23sock.bind(('',8888))
24sock.listen()
25sock.setblocking(False)
26sel.register(sock,selectors.EVENT_READ,accept) #把刚生成的sock连接对象注册到select连接列表中，并交给accept函数处理
27
28while True:
29    events = sel.select()  # 默认是阻塞，有活动连接就返回活动的连接列表
30    # 这里看起来是select，其实有可能会使用epoll，如果你的系统支持epoll，那么默认就是epoll
31    print('----------------------------------------------------------------------------------------------------')
32    print('events>>>',events)#看看里面是啥
33    print('----------------------------------------------------------------------------------------------------')
34    for key, mask in events:
35        print('----------------------------------------------------------------------------------------------------')
36        print('key>>>',key)#看看里面是啥
37        print('----------------------------------------------------------------------------------------------------')
38        callback = key.data  # 去调accept函数
39        callback(key.fileobj, mask)  # key.fileobj就是readable中的一个socket连接对象

Socket 对象(内建)方法

 1服务器端
 2
 3s.bind()     
 4# 绑定地址（host,port）到套接字， 在AF_INET下,以元组（host,port）的形式表示地址。
 5s.listen()     
 6# 开始TCP监听。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。
 7s.accept()     
 8# 被动接受TCP客户端连接,(阻塞式)等待连接的到来
 9
10
11客户端
12
13s.connect()     
14# 主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
15s.connect_ex()     
16# connect()函数的扩展版本,出错时返回出错码,而不是抛出异常
17
18
19公共用途的函数
20
21s.recv()     
22# 接收TCP数据，数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。
23s.send()     
24# 发送TCP数据，将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。
25s.sendall()     
26# 完整发送TCP数据，完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。
27s.close()
28# 关闭套接字
29
30
31s.recvform()     
32# 接收UDP数据，与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
33s.sendto()     
34# 发送UDP数据，将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。
35s.getpeername()     
36# 返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。
37s.getsockname()     
38# 返回套接字自己的地址。通常是一个元组(ipaddr,port)
39s.setsockopt(level,optname,value)     
40# 设置给定套接字选项的值。
41s.getsockopt(level,optname[.buflen])      
42# 返回套接字选项的值。
43s.settimeout(timeout)     
44# 设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
45s.gettimeout()     
46# 返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None。
47s.fileno()     
48# 返回套接字的文件描述符。
49s.setblocking(flag)     
50# 如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起socket.error异常。
51s.makefile()     
52# 创建一个与该套接字相关连的文件