Python-socket 层编程

本木夕丶Lu

已于 2022-04-15 10:52:32 修改

阅读量4.2k

点赞数

分类专栏： Python基础文章标签： python 开发语言

于 2022-03-26 13:44:05 首次发布

本文链接：https://blog.csdn.net/weixin_43988680/article/details/123754586

版权

Python基础专栏收录该内容

36 篇文章 5 订阅

订阅专栏

一、socket 层简述

Socket是应用层与TCP/IP协议族通信的中间软件抽象层，它是一组接口。在设计模式中，Socket其实就是一个门面模式，它把复杂的TCP/IP协议族隐藏在Socket接口后面，对用户来说，一组简单的接口就是全部，让Socket去组织数据，以符合指定的协议。

所以，我们无需深入理解tcp/udp协议，socket已经为我们封装好了，我们只需要遵循socket的规定去编程，写出的程序自然就是遵循tcp/udp标准的

在这里插入图片描述

二、socket 工作流程

在这里插入图片描述

Tcp 服务端：

首先，需要拿到一个 socket 对象，socket 对象与 ip+port 绑定，才能进行端口的监听;

然后，socket 对象监听端口是否有请求，同时调用 accept方法，一直阻塞等待客户端的链接；

如果，有客户端的请求发来，那么服务端接受到请求之后，调用 read 方法，读取数据；

之后再处理请求，写回数据使用 write 方法（在 Python 中是 send 方法）

当需要断开链接时，服务端发送请求，客户端读取之后，在调用 close 方法发送断开的请求

服务端 read 收到断开请求，服务端再调用 close 结束链接

Tcp 客户端：

首先，需要拿到一个 socket 对象，调用 connect 方法，传入 IP 与端口，发送建立链接的请求；

之后，就可以进行数据交互，如果需要结束链接，直接调用 close 方法即可。

三、基于 Tcp 的scoket 服务端与客户端编程

最基础版本

tcp是基于链接的，必须先启动服务端，然后再启动客户端去链接服务端

服务端

import socket
import time

server = socket.socket()

# 注意传入的参数必须是一个元组或者列表
# ip'127.0.0.1',只能本机自己玩
# 如果写别的 IP 地址可以在局域网中一起玩
server.bind(('127.0.0.1', 8008))

# 监听，半连接池 为 5，相当于队列中最多有 5 个，如果多余 5 个就会报错
# 同时只能服务一个人
server.listen(5)

# 等待客户端的链接
# sock 是连接的对象， addr 是客户端地址
# 以后这个服务端和客户端，使用 sock 这个链接对象
sock, addr = server.accept()

# 接受客户端发了的数据
# 传入的参数为一次接受的字节
data = sock.recv(1024)

print(f'接收到来自{addr}的数据{data.decode("utf-8")}')

# 服务端给客户端发送消息
data_send = f'收到了你发送的 {data}'.encode('utf-8')
# 注意：必须传送二进制的格式
sock.send(data_send)


time.sleep(3)
# 关闭连接对象，此时服务端并没有关闭！！！
sock.close()

# 关闭服务端
server.close()

客户端

import socket

client = socket.socket()

# 连接服务端的地址加端口
client.connect(('127.0.0.1', 8008))

# 连上以后就可以发送消息了
client.send(b'sb')

# 收到服务端返回的消息
data = client.recv(1024)
print(data.decode('utf-8'))

client.close()

加入链接循环

只加入链接循环，此时服务端会一直接收，相当于死循环，服务端不会自动断开

服务端

在这里插入图片描述

import socket

server = socket.socket()

server.bind(('127.0.0.1', 8008))

server.listen(5)
print('等待客户端的链接')

# 等待客户端的链接
# 链接循环
while True:
    sock, addr = server.accept()


    # 接受客户端发了的数据
    data = sock.recv(1024)

    print(f'接收到来自{addr}的数据--{data.decode("utf-8")}')


    # 服务端给客户端发送消息
    data_send = f'收到了你发送的 {data}'.encode('utf-8')
    # 注意：必须传送二进制的格式
    sock.send(data_send)


    # 关闭连接对象，此时服务端并没有关闭！！！
    sock.close()

# 关闭服务端
server.close()

客户端

在这里插入图片描述

import socket

client = socket.socket()

# 连接服务端的地址加端口
client.connect(('127.0.0.1', 8008))

# 连上以后就可以发送消息了
client.send(b'sb')

# 收到服务端返回的消息
data = client.recv(1024)
print(data.decode('utf-8'))

client.close()

加入通信循环

加入通信循环，此时客户端可以多次发送数据

服务端

在这里插入图片描述

import socket

server = socket.socket()
server.bind(('127.0.0.1', 8008))
server.listen(5)
print('等待客户端的链接')

while True:
    sock, addr = server.accept()
    print(f'客户端{addr}已连接')
    # 等待接受客户端发的数据
    # 如果客户端没法，就会一直等待下去
    while True:
        try:
            # 同一个客户端，不停地发送
            data = sock.recv(1024)
            # 当客户端主动断开后，客户端会发送一个空的数据过来
            # 此时服务端必须对这种情况进行处理，否则会出现死循环
            if len(data) == 0:
                print(f'客户端{addr}已断开', '\n')
                break

            print(f'接收到来自{addr}的数据--{data.decode("utf-8")}')
            data_send = f'收到了你发送的 {data}'.encode('utf-8')
            sock.send(data_send)
        except Exception as e:
            print(e)
            break

    sock.close()

# 关闭服务端
server.close()

客户端

import socket

client = socket.socket()
client.connect(('127.0.0.1', 8008))
# 连上以后就可以发送消息了

while True:
    data = input('输入发送给服务端的消息(q to quit)：')
    if data == 'q': break
    client.send(data.encode("utf-8"))

    data = client.recv(1024)
    print(data.decode('utf-8'))

client.close()

支持并发的 Tcp 服务端

在这里是开启多进程完成的并发，可以换成多线程，只需要将 Process 类换成 Thread 类。

import socket
from multiprocessing import Process

def talk(sock,addr):
    print('客户端连接成功', addr)
    while True:
        try:
            data = sock.recv(1024)
            if len(data) == 0:
                print(f'客户端{addr}已断开', '\n')
                break

            print(f'接收到来自{addr}的数据--{data.decode("utf-8")}')
            data_send = f'收到了你发送的 {data}'.encode('utf-8')
            sock.send(data_send)
        except Exception as e:
            print(e)
            break
    sock.close()


if __name__ == '__main__':
    server = socket.socket()
    server.bind(('127.0.0.1', 81))
    server.listen(5)
    while True:
        sock,addr = server.accept()
        print(f'客户端{addr}已连接')
        # 连接上一个客户端之后开启一个进程，对该客户端进行服务
        p = Process(target=talk, args=(sock, addr))
        p.start()

    server.close()

常见问题

问题 1：

在这里插入图片描述

当服务端手动结束时，可能会出现这种问题。该端口在后台还在被监听，没有被停止。

解决办法：一般只需要等待几秒即可，或者去后台中去找到这个线程，手动结束
解决办法 2：加上两句固定的代码

from socket import SOL_SOCKET,SO_REUSEADDR
sk.setsockopt(SOL_SOCKET,SO_REUSEADDR,1)  # 就是它，在bind前加

问题 2：

在加入通信循环之后，想要同个 py 文件启动多个客户端， pycharm会出现以下提示，不允许启动多个

在这里插入图片描述

解决办法：

在这里插入图片描述

问题 3：

当客户端不是调用 close 关闭的时候，服务端会报错。

在这里插入图片描述

解决办法：服务端，加入一个异常捕获

四、基于 Udp 的 scoket 服务端与客户端编程

服务端

import socket

ip_port=('127.0.0.1',9000)
SIZE=1024
udp_server=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

udp_server.bind(ip_port)

msg,addr=udp_server.recvfrom(SIZE)
print(addr)
print(f'客户端{addr}传来  {msg}')

data = f'服务端收到了你传的  {msg}'.encode("utf-8")
udp_server.sendto(data,addr)

udp_server.close()

客户端

import socket
ip_port=('127.0.0.1',9000)
BUFSIZE=1024
udp_client=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

udp_client.sendto(b'sb',ip_port)

server_data = udp_client.recv(1024)
print(server_data.decode("utf-8"))

udp_client.close()