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

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

**1.概念**

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

#### 2.基本用法 ####

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

socket(family,type[,protocal])

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

<table> 
 <thead> 
  <tr> 
   <th>协议族</th> 
   <th>类型</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>AF_INTE</td> 
   <td>IPv4</td> 
  </tr> 
  <tr> 
   <td>AF_INTE6</td> 
   <td>IPv6</td> 
  </tr> 
  <tr> 
   <td>AF_UNIX</td> 
   <td>本地</td> 
  </tr> 
  <tr> 
   <td>AF_ROUTE</td> 
   <td>路由</td> 
  </tr> 
 </tbody> 
</table>

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

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

第三个参数是指使用的协议，这个参数是可选额，通常赋值“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类中主要提供如下所示常用的函数。

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

**四、代码演示**

思路

*  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()

这里首先启动服务端

![图片][adf52028676e88a6bb36e71bdc71345d.png]

再启动客户端

![图片][ccaa50400473072328838b09fd2e8a1d.png]

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

![图片][0f656eda79c447c98187fb17bdb0ccb0.png]

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

演示一个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()

这样简单的demo就实现了。

**最后：**下方这份完整的软件测试视频教程已经整理上传完成，需要的朋友们可以自行领取**【保100%免费】**

![8b0171187dce405e878f274a8cdd9ec0.png][]

#### **软件测试面试文档** ####

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

![a3e4226c935c4242804cbdf26b571bc6.png][]

![a3baa16769924d0dbaab81da43d3707c.gif][]

[adf52028676e88a6bb36e71bdc71345d.png]: https://img-blog.csdnimg.cn/img_convert/adf52028676e88a6bb36e71bdc71345d.png
[ccaa50400473072328838b09fd2e8a1d.png]: https://img-blog.csdnimg.cn/img_convert/ccaa50400473072328838b09fd2e8a1d.png
[0f656eda79c447c98187fb17bdb0ccb0.png]: https://img-blog.csdnimg.cn/img_convert/0f656eda79c447c98187fb17bdb0ccb0.png
[8b0171187dce405e878f274a8cdd9ec0.png]: https://img-blog.csdnimg.cn/8b0171187dce405e878f274a8cdd9ec0.png
[a3e4226c935c4242804cbdf26b571bc6.png]: https://img-blog.csdnimg.cn/a3e4226c935c4242804cbdf26b571bc6.png
[a3baa16769924d0dbaab81da43d3707c.gif]: https://img-blog.csdnimg.cn/a3baa16769924d0dbaab81da43d3707c.gif