python的socket模块

最新推荐文章于 2024-08-09 11:18:41 发布

Lion King

最新推荐文章于 2024-08-09 11:18:41 发布

阅读量1.1w

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/weixin_43431593/article/details/120410333

版权

python 专栏收录该内容

92 篇文章 36 订阅

订阅专栏

一、官方文档：

https://docs.python.org/3.9/library/socket.html

二、按照官方文档练习

1、Socket families

（1）根据系统和构建选项，该模块支持各种套接字家族。即根据创建套接字对象时指定的地址族，自动选择特定套接字对象所需的地址格式。例如:

①指定AF_UNIX,则会使用文件系统编码和'surrogateescape'错误处理程序

②指定AF_CAN，则可定义网络接口名称，并接收来自该系列所有网络接口的数据包

③指定AF_ALG，则是一个仅基于linux套接字的内核加密接口

（2）指定方式如下：

import socket

sock = socket.socket(
    socket.AF_UNIX,
    socket.SOCK_STREAM | socket.SOCK_NONBLOCK)

2、模块详情

（1）异常处理

相关模块有：socket.error、socket.herror、socket.gaierror、socket.timeout

（2）常量

常量用于socket()的设置中：

socket()的第一个参数有：AF_UNIX、AF_INET、AF_INET6

socket()的第二个参数有：SOCK_STREAM、SOCK_DGRAM、SOCK_RAW、SOCK_RDM、SOCK_SEQPACKET

socket()有两个常量，可以将它们与套接字类型组合在一起：SOCK_CLOEXEC、SOCK_NONBLOCK

（3）指定方式如下：

import socket

sock = socket.socket(
    socket.AF_UNIX,
    socket.SOCK_STREAM | socket.SOCK_NONBLOCK)

（4）方法使用

①创建socket对象

import socket  # 导入 socket 模块

# AF_INET(默认地址族) 、SOCK_STREAM(默认套接字类型)
s = socket.socket()  # 创建 socket 对象，是单工模式，也就是单个出现，要么是收，要么是发
s1 = socket.socketpair() # 创建 socket 对象,是全双工模式，也就是每个socket都能收发
# 其他方式如：socket.create_connection等

3、Socket对象

该总结抄袭菜鸟教程：https://www.runoob.com/python/python-socket.html

函数	描述
服务器端套接字
s.bind()	绑定地址（host,port）到套接字，在 AF_INET下，以元组（host,port）的形式表示地址。
s.listen()	开始 TCP 监听。backlog 指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为 1，大部分应用程序设为 5 就可以了。
s.accept()	被动接受TCP客户端连接,(阻塞式)等待连接的到来
客户端套接字
s.connect()	主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
s.connect_ex()	connect()函数的扩展版本,出错时返回出错码,而不是抛出异常
公共用途的套接字函数
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.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()	创建一个与该套接字相关连的文件

4、socket超时的注意事项

套接字对象可以处于三种模式之一:阻塞、非阻塞或超时。套接字在默认情况下总是以阻塞模式创建的，但这可以通过调用setdefaulttimeout()来更改。

5、示例

（1）客户端发送数据，服务器收到后，返回客户端发送的数据：

服务端：

# Echo server program
import socket

HOST = '127.0.0.1'                 # 设置服务器地址
PORT = 50007              # 任意非特权端口，取值范围：0 ~65536
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.bind((HOST, PORT))  # 绑定IP和端口号到socket对象
    s.listen(1)  # 设置监听最大连接数为1
    conn, addr = s.accept() # 阻塞的方式，被动接收TCP客户端连接，返回连接后的socket对象和连接地址
    with conn:
        print('Connected by', addr)  # 打印连接地址
        while True:
            data = conn.recv(1024)  # 接收1024字节的数据
            if not data: break  # 如果收到数据，则中断
            conn.sendall(data)  # 发送接收到的数据

客户端：

# Echo client program
import socket

HOST = '127.0.0.1'    # 客户端的连接地址
PORT = 50007              # 指定端口号
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.connect((HOST, PORT)) # 连接服务器
    s.sendall(b'Hello, world') # 发送字节
    data = s.recv(1024) # 再去接收字节
print('Received', repr(data)) # 打印接收到的数据

（2）重写上述通信，使其既满足IPv4的通信，又满足IPv6的通信

服务端：

# Echo server program
import socket
import sys

HOST = '127.0.0.1'               # Symbolic name meaning all available interfaces
PORT = 50007              # Arbitrary non-privileged port
s = None
for res in socket.getaddrinfo(HOST, PORT, socket.AF_UNSPEC,
                              socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
    af, socktype, proto, canonname, sa = res
    try:
        s = socket.socket(af, socktype, proto)
    except OSError as msg:
        s = None
        continue
    try:
        s.bind(sa)
        s.listen(1)
    except OSError as msg:
        s.close()
        s = None
        continue
    break
if s is None:
    print('could not open socket')
    sys.exit(1)
conn, addr = s.accept()
with conn:
    print('Connected by', addr)
    while True:
        data = conn.recv(1024)
        if not data: break
        conn.send(data)

客户端：

# Echo client program
import socket
import sys

HOST = '127.0.0.1'    # The remote host
PORT = 50007              # The same port as used by the server
s = None
for res in socket.getaddrinfo(HOST, PORT, socket.AF_UNSPEC, socket.SOCK_STREAM):
    af, socktype, proto, canonname, sa = res
    try:
        s = socket.socket(af, socktype, proto)
    except OSError as msg:
        s = None
        continue
    try:
        s.connect(sa)
    except OSError as msg:
        s.close()
        s = None
        continue
    break
if s is None:
    print('could not open socket')
    sys.exit(1)
with s:
    s.sendall(b'Hello, world')
    data = s.recv(1024)
print('Received', repr(data))

（3）基于Windows的网络嗅探器

网络嗅探需要用到网络嗅探器，其最早是为网络管理人员配备的工具，有了嗅探器网络管理员可以随时掌握网络的实际情况，查找网络漏洞和检测网络性能，当网络性能急剧下降的时候，可以通过嗅探器分析网络流量，找出网络阻塞的来源。网络嗅探是网络监控系统的实现基础。

import socket

# the public network interface
HOST = socket.gethostbyname(socket.gethostname())

# create a raw socket and bind it to the public interface
s = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_IP)
s.bind((HOST, 0))

# Include IP headers
s.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)

# receive all packages
s.ioctl(socket.SIO_RCVALL, socket.RCVALL_ON)

# receive a package
print(s.recvfrom(65565))

# disabled promiscuous mode
s.ioctl(socket.SIO_RCVALL, socket.RCVALL_OFF)