Socket中文就是套接字,它是进程和网络之间的通信接口,可以简单认为它就是一个API,一个Socket由
IP
+Port
构成,两个进程间如果需要通过网络通信,那么他们各自需要通过相同的Socket从网络中读/写数据,本文主要是为了学习和记录Socket在Python中的使用方法。
Python中Socket的通信逻辑
socket是基于C/S架构的,也就是说进行socket网络编程,通常需要编写两个py文件,一个服务端,一个客户端。
导入socket包:
import socket
Python中的socket通信逻辑如下图所示(图片来自网络):
创建嵌套字
在Python中,import socket
后,用socket.socket()
方法来创建套接字,语法格式如下:
sk = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, 0)
#socket.socket(family,type,protocol)
#如果括号里面什么都不填写,则使用默认值
参数意思:
- family: 套接字家族,可以使
AF_UNIX
或者AF_INET
。 - type: 套接字类型,根据是面向连接的还是非连接分为
SOCK_STREAM
或SOCK_DGRAM
,也就是TCP和UDP的区别。 - protocol: 一般不填默认为0。
通信
利用创建好的嵌套字对象,利用它所包含的方法来进行通信,具体用法在下面的表格中。
具体的参数定义
socket类型 | 描述 |
---|---|
socket.AF_UNIX | 只能够用于单一的Unix系统进程间通信 |
socket.AF_INET | IPv4 |
socket.AF_INET6 | IPv6 |
socket.SOCK_STREAM | 流式socket , for TCP |
socket.SOCK_DGRAM | 数据报式socket , for UDP |
socket.SOCK_RAW | 原始套接字,普通的套接字无法处理ICMP、IGMP等网络报文,而SOCK_RAW可以;其次,SOCK_RAW也可以处理特殊的IPv4报文;此外,利用原始套接字,可以通过IP_HDRINCL套接字选项由用户构造IP头。 |
socket.SOCK_SEQPACKET | 可靠的连续数据包服务 |
创建TCP Socket: | s=socket.socket(socket.AF_INET,socket.SOCK_STREAM) |
创建UDP Socket: | s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM) |
通过s = socket.socket()
方法,我们可以获得一个socket对象s,也就是通常说的获取了一个“套接字”,该对象具有以下方法:
Socket方法定义
方法 | 描述 |
---|---|
服务器端方法 | |
s.bind() | 绑定地址(host,port)到套接字,在AF_INET下,以元组(host,port)的形式表示地址。 |
s.listen(backlog) | 开始监听。backlog指定在拒绝连接之前,操作系统可以挂起的最大连接数量。该值至少为1,大部分应用程序设为5就可以了。 |
s.accept() | 被动接受客户端连接,(阻塞式)等待连接的到来,并返回(conn,address)二元元组,其中conn是一个通信对象,可以用来接收和发送数据。address是连接客户端的地址。 |
客户端方法 | |
s.connect(address) | 客户端向服务端发起连接。一般address的格式为元组(hostname,port),如果连接出错,返回socket.error错误。 |
s.connect_ex() | connect()函数的扩展版本,出错时返回出错码,而不是抛出异常 |
公共方法 | |
s.recv(bufsize) | 接收数据,数据以bytes类型返回,bufsize指定要接收的最大数据量。 |
s.send() | 发送数据。返回值是要发送的字节数量。 |
s.sendall() | 完整发送数据。将数据发送到连接的套接字,但在返回之前会尝试发送所有数据。成功返回None,失败则抛出异常。 |
s.recvform() | 接收UDP数据,与recv()类似,但返回值是(data,address)。其中data是包含接收的数据,address是发送数据的套接字地址。 |
s.sendto(data,address) | 发送UDP数据,将数据data发送到套接字,address是形式为(ipaddr,port)的元组,指定远程地址。返回值是发送的字节数。 |
s.close() | 关闭套接字,必须执行。 |
s.getpeername() | 返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。 |
s.getsockname() | 返回套接字自己的地址。通常是一个元组(ipaddr,port) |
s.setsockopt(level,optname,value) | 设置给定套接字选项的值。 |
s.getsockopt(level,optname[.buflen]) | 返回套接字选项的值。 |
s.settimeout(timeout) | 设置套接字操作的超时期,timeout是一个浮点数,单位是秒。值为None表示没有超时期。一般,超时期应该在刚创建套接字时设置,因为它们可能用于连接的操作(如connect()) |
s.gettimeout() | 返回当前超时期的值,单位是秒,如果没有设置超时期,则返回None。 |
s.fileno() | 返回套接字的文件描述符。 |
s.setblocking(flag) | 如果flag为0,则将套接字设为非阻塞模式,否则将套接字设为阻塞模式(默认值)。非阻塞模式下,如果调用recv()没有发现任何数据,或send()调用无法立即发送数据,那么将引起socket.error异常。 |
s.makefile() | 创建一个与该套接字相关连的文件 |
注意事项:
- Python3以后,socket传递的都是bytes类型的数据,字符串需要先转换一下,
string.encode()
即可;另一端接收到的bytes数据想转换成字符串,只要bytes.decode()
一下就可以。 - 在正常通信时,
accept()
和recv()
方法都是可以造成阻塞的。阻塞指的是程序会暂停在这一行代码处,等待该方法执行。
Socket编程步骤
服务端:
- 创建套接字,绑定套接字到本地IP与端口:socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.bind()
- 开始监听连接:s.listen()
- 进入循环,不断接受客户端的连接请求:s.accept()
- 接收传来的数据,或者发送数据给对方:s.recv() , s.sendall()
- 传输完毕后,关闭套接字:s.close()
客户端:
- 创建套接字,连接服务器地址:socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.connect()
- 连接后发送数据和接收数据:s.sendall(), s.recv()
- 传输完毕后,关闭套接字:s.close()
Python的socket编程,通常可分为TCP和UDP编程两种,前者是带连接的可靠传输协议,每次通信都要握手,结束传输也要挥手,数据会被检验,是使用最广最通用的通信协议;后者是不带连接的传输协议,简单粗暴,不加控制和检查的一股脑将数据发送出去的方式,但是传输延迟低,速度快,通常用于对延迟敏感的业务场景,例如视频通话、网络游戏等。
TCP示例代码
服务端:
import socket
ip_port = ('127.0.0.1', 9999)
sk = socket.socket() # 创建套接字
sk.bind(ip_port) # 绑定服务地址
sk.listen(5) # 监听连接请求
print('启动socket服务,等待客户端连接...')
conn, address = sk.accept() # 等待连接,此处自动阻塞
while True: # 一个死循环,直到客户端发送‘exit’的信号,才关闭连接
client_data = conn.recv(1024).decode() # 接收信息
if client_data == "exit": # 判断是否退出连接
exit("通信结束")
print("来自%s的客户端向你发来信息:%s" % (address, client_data))
conn.sendall('服务器已经收到你的信息'.encode()) # 回馈信息给客户端
conn.close() # 关闭连接
客户端:
import socket
ip_port = ('127.0.0.1', 9999)
s = socket.socket() # 创建套接字
s.connect(ip_port) # 连接服务器
while True: # 通过一个死循环不断接收用户输入,并发送给服务器
inp = input("请输入要发送的信息: ").strip()#strip可以返回一份输入信息的拷贝
if not inp: # 防止输入空信息,导致异常退出
continue
s.sendall(inp.encode())
if inp == "exit": # 如果输入的是‘exit’,表示断开连接
print("结束通信!")
break
server_reply = s.recv(1024).decode()
print(server_reply)
s.close() # 关闭连接
UDP示例代码
服务端:
#服务端
import socket
ip_port = ('127.0.0.1', 9999)
sk = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, 0)
sk.bind(ip_port)
while True:
data = sk.recv(1024).strip().decode()
print(data)
if data == "exit":
print("客户端主动断开连接!")
break
sk.close()
客户端:
# 客户端
import socket
ip_port = ('127.0.0.1', 9999)
sk = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, 0)
while True:
inp = input('发送的消息:').strip()
sk.sendto(inp.encode(), ip_port)
if inp == 'exit':
break
sk.close()