【测试开发】python系列教程：socket模块-蒲公英云

【测试开发】python系列教程：socket模块

红太狼 2023-10-15 04:28 71阅读 0赞

1.概念

Python中的socket模块是底层的网络接口， socket模块提供了标准的BSD SocketAPI和服务器中心类，可以简化网络服务器的开发。该模块提供了两个基本的Socket模块：服务端Socket和客户端Socket。当创建了一个服务端Socket之后，这个Socket就会在本机的一个端口上等待连接，客户端Socket会访问这个端口，当两者完成连接之后，就可以进行交互了。

2.基本用法

Socket实例化首先看一下如何实例化一个Socket。Socket实例化的格式为：

socket(family,type[,protocal])

其中，三个参数中的family是要使用的地址族，以下是常用的协议族

协议族	类型
AF_INTE	IPv4
AF_INTE6	IPv6
AF_UNIX	本地
AF_ROUTE	路由

第二个参数type用来指明Socket类型，这里可以使用一下参数

参数	类型	作用
SOCK_STREAM	TCP	保证数据顺序及可靠性
SOCK_DGRAM	UDP	不保证数据接收的顺序，非可靠连接
SOCK_RAW	原始类型	允许对底层协议如IP或ICMP进行直接访问

第三个参数是指使用的协议，这个参数是可选额，通常赋值“0”，由系统自动选择。如果希望初始化一个TCP类型的Socket，可以使用以下语句。

s=socket.socket()

这条语句相当于socket.socket(socket.AF_INET,socket.SOCK_STREAM)。这里因为都是默认值，所以可以省略。而如果想要初始化一个UDP类型的Socket，则可以使用如下语句。

s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

3.Socket常用函数

socket类中主要提供如下所示常用的函数。

函数	作用
bind()	将指定的IP与端口绑定起来，如果之前使用了AF_INET初始化Socket，那么这里可以使用元组(husi,port)的形式表示地址
listen()	指定服务端可以监听的最大数量,这个参数最小为1，一般为5
accept()	用于在使用TCP的服务端接收连接，一般时阻塞态，接收TCP连接并返回(conn,address)，其中，conn是新的套接字对象，可以用来接收和发送数据；address是连接客户端的地址
connect()	用于在使用TCP的客户端去连接服务端，使用的参数是一个元组，形式为(hostname，port)
send()	用于在使用TCP时发送数据，完整的形式为send(steing[,flag])，利用这个函数可以将string代表的数据发送到已经连接的socket，返回信息发送字节数量。但是可能未将指定的内容全部发送
recv()	用于使用TCP是接收数据，完整的形式为recv(bufsoze[,flag])，接收Socket的数据，诗句以字符串形式返回，bufsize指定最多可以接受的数量，flag这个参数一般不会使用
sendto()	用于UDP发送数据，完整的形式为sendto(string[,flag],address)，返回值是发送的字节数。address是形式为(ipaddr.port)的元组，指定远程地址
recvfrom()	UDP专用，接收数据，返回远端数据的IP地址和端口，但返回值是(data,address)。其中data是包含接收数据的字符串，address是发送数据的套接字地址
close()	关闭socket

四、代码演示

思路

1 创建套接字，绑定套接字到本地IP与端口　
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind()
2 开始监听连接
s.listen()
3 进入循环，不断接受客户端的连接请求 s.accept()
4 然后接收传来的数据，并发送给对方数据
s.recv() , s.sendall()
5 传输完毕后，关闭套接字
s.close()
TCP客户端:
1 创建套接字，连接远端地址
socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.connect()
2 连接后发送数据和接收数据 # s.sendall(), s.recv()
3 传输完毕后，关闭套接字
s.close()

demo演示 TCP服务端

import socket               # 导入 socket 模块
s = socket.socket()         # 创建 socket 对象
host = socket.gethostname() # 获取本地主机名
port = 8082                # 设置端口
s.bind((host, port))        # 绑定端口
s.listen(5)                 # 等待客户端连接
while True:
    c, addr = s.accept()     # 建立客户端连接。
    print('连接地址：', addr)
    c.send('leizi')
    c.close()  # 关闭连接

demo演示客户端

import socket               # 导入 socket 模块
s = socket.socket()         # 创建 socket 对象
host = socket.gethostname() # 获取本地主机名
port = 8082               # 设置端口好
s.connect((host, port))
print(s.recv(1024))
s.close()

这里首先启动服务端

再启动客户端

可以看到，客户端已经接到消息了，看下服务端

这样证明我们这里写的已经实现了。

演示一个UDP的服务端

from socket import *
HOST = '192.168.1.123'
PORT = 8888
s = socket(AF_INET, SOCK_DGRAM)
s.connect((HOST, PORT))
while True:
    message = input('send message:>>')
    s.sendall(message)
    data = s.recv(1024)
    print(data)
    s.close()