socket 编程
一、什么是 Socket
Socket 又称 “套接字”,应用程序通常通过 “套接字” 向网络发出请求或者应答网络请求,使主机间或者一台计算机上的进程间可以通讯。
1、socket 工作流程图
2、socket() 函数
socket.socket([family[, type[, proto]]])
参数说明:
family: 套接字家族可以使 AF_UNIX 或者 AF_INET。
type: 套接字类型可以根据是面向连接的还是非连接分为 SOCK_STREAM 或 SOCK_DGRAM。
protocol: 一般不填默认为 0。
3、socket 对象(内建)方法
(1) 服务器端套接字
函数 | 描述 |
---|---|
s.bind() | 绑定地址(host, port)到套接字,在 AF_INET 下,以元组(host, port)的形式表示地址。 |
s.listen() | 开始 TCP 监听。backlog 指定在拒绝连接之前,操作系统可以挂起的最大连接数量。该值至少为 1,大部分应用程序设为 5 就可以了。 |
s.accept() | 被动接受 TCP 客户端连接,(阻塞式)等待连接的到来。 |
(2) 客户端套接字
函数 | 描述 |
---|---|
s.connect() | 主动初始化 TCP 服务器连接。一般 address 的格式化为元组(hostname, port),如果连接出错,返回 socket.error 错误。 |
s.connect_ex() | connect() 函数的扩展版本,出错时返回出错码,而不是抛出异常。 |
(3) 公共用途的套接字函数
函数 | 描述 |
---|---|
s.recv() | 接收 TCP 数据,数据以字符串形式返回,bufsize 指定要接收的最大数据量。 flag 提供有关信息的其他信息,通常可以忽略。 |
s.send() | 发送 TCP 数据,将 string 中的数据发送到连接的套接字。 返回值是要发送的字节数量,该数量可能小于 string 的字节大小 |
s.sendall() | 完整发送 TCP 数据。将 string 中的数据发送到连接的套接字,但在返回之前会尝试发送所有数据。 成功返回 None,失败则抛出异常。 |
s.recvfrom() | 接收 UDP 数据,与 recv() 类似,但返回值是(data, address)。 其中 data 是包含接收数据的字符串,address 是发送数据的套接字地址。 |
s.sendto() | 发送 UDP 数据,将数据发送到套接字,address 是形式为(ipaddr, port)的元组,指定远程地址。 返回值是发送的字节数。 |
s.close() | 关闭套接字。 |
s.getpeername() | 返回连接套接字的远程地址。返回值通常是元组(ipaddr, port)。 |
s.getsocketname() | 返回套接字自己的地址。通常是一个元组(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() | 创建一个与该套接字相关联的文件。 |
二、简单实例
1、socket 发送 TCP 请求
(1) tcp_server.py 文件
"""
服务端
使用 socket 模块的 socket 函数来创建一个 socket 对象。
socket 对象可以通过调用其他函数来设置一个 socket 服务。
可以通过 bind(hostname, port) 函数来指定服务的 port(端口)。
接着,我们调用 socket 对象的 accept 方法。该方法等待客户端的连接,并返回 connection 对象,表示已连接到客户端。
"""
import socket # 导入 socket 模块
s = socket.socket() # 创建 socket 对象
host = socket.gethostname() # 获取本地主机名
port = 8000 # 设置端口
s.bind((host, port)) # 绑定端口
s.listen(5) # 等待客户端连接
while True:
client, addr = s.accept() # 建立客户端连接
print("连接地址:", addr)
while True:
data = client.recv(1024) # 接收数据
print('server:', data.decode()) # 打印接收到的数据
client.send(data.upper()) # 然后再发送数据
client.close() # 关闭连接
(2) tcp_client.py 文件
"""
客户端
socket.connect(hostname, port) 方法打开一个 TCP 连接到主机为 hostname 端口为 port 的服务器。
连接后我们就可以从服务端获取数据,记住,操作完成后需要关闭连接。
"""
import socket # 导入 socket 模块
# s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# s.connect("127.0.0.1", 8000)
# s.send(b"Hello World")
client = socket.socket() # 创建 socket 对象
host = socket.gethostname() # 获取本地主机名
port = 8000 # 设置端口号
client.connect((host, port)) # 初始化 TCP 服务器连接
while True:
msg = '欢迎来到麻花屋!' # strip默认取出字符串的头尾空格
client.send(msg.encode('utf-8')) # 发送一条信息 python3 只接收btye流
data = client.recv(1024) # 接收一个信息,并指定接收的大小 为1024字节
print('client:', data.decode()) # 输出我接收的信息
s.close() # 关闭连接
(3) 运行结果:
打开两个终端,第一个终端执 tcp_server.py 文件:
第二个终端执行 tcp_client.py 文件:
我们再打开第一个终端,会有如下信息输出:
2、socket 发送 HTTP 请求
(1) 代码:
import socket # 导入 socket 模块
client = socket.socket() # 创建 socket 对象
client.connect(("www.baidu.com", 80))
http = b"GET / HTTP/1.1\r\nHost: www.baidu.com\r\n\r\n"
client.send(http) # 初始化 TCP 服务器连接
data = client.recv(1024) # 接收一个信息,并指定接收的大小 为1024字节
print("recv: ", data.decode())
client.close() # 关闭连接
(2) 运行结果: