发帖
翼度科技»论坛 › 编程开发 › python › 查看内容

Python中Socket编程底层原理解析与应用实战

李志达

10

主题

10

帖子

30

积分

新手上路

Rank: 1

积分
30
显示全部楼层
引言

Socket编程是网络通信的基础，Python通过内置的socket模块提供了强大的网络编程接口。Socket编程允许开发者创建客户端和服务器程序，实现数据在网络中的传输。本文将结合实际案例，详细介绍Python中Socket编程的基本概念、常用方法和实际应用。

一、基本概念

服务器与客户端

  • 服务器：一系列硬件或软件，为一个或多个客户端（服务的用户）提供所需的“服务”。它响应客户端的请求，并等待更多的请求。
  • 客户端：向服务器请求服务的用户或程序。它发送请求并接收服务器的响应。
套接字（Socket）

  • 套接字是计算机网络数据结构，它体现了通信端点的概念。在任何类型的通信开始之前，网络应用程序必须创建套接字。套接字可以被比作电话插孔，没有它将无法进行通信。
  • Python中的socket模块支持多种类型的套接字，包括基于IPv4的AF_INET（流式套接字和数据报套接字）。
流式套接字（SOCK_STREAM）

  • 用于提供面向连接、可靠的数据传输服务。它基于TCP协议，确保数据按顺序、无差错、不重复地传输。
数据报套接字（SOCK_DGRAM）

  • 提供一种无连接的服务。数据报套接字基于UDP协议，不保证数据传输的可靠性，数据可能在传输过程中丢失或重复，且无法保证顺序地接收到数据。

二、Python中的Socket编程


2.1 常见的套接字对象方法和属性

在Python的socket编程中，常用的方法包括：

    1. socket()
    复制代码
    : 创建一个新的套接字对象。
    1. bind()
    复制代码
    : 将套接字绑定到特定的IP地址和端口号上。
    1. listen()
    复制代码
    : 开始监听客户端的连接请求。
    1. accept()
    复制代码
    : 接受一个连接请求，并返回一个新的套接字对象和客户端的地址信息。
    1. connect()
    复制代码
    : 客户端使用该方法连接到服务器。
    1. send() / sendall()
    复制代码
    : 发送数据到连接的套接字。
    1. sendall()
    复制代码
    确保数据被完整发送。
    1. recv() / recvfrom()
    复制代码
    : 接收数据。
    1. recv()
    复制代码
    用于TCP连接，
    1. recvfrom()
    复制代码
    用于UDP连接。
    1. close()
    复制代码
    : 关闭套接字。

2.2 创建TCP服务和客户端


2.2.1 创建TCP服务

创建一个简单的TCP服务器，步骤如下：

  • 导入socket模块。
  • 创建套接字对象，指定AF_INET和SOCK_STREAM。
  • 绑定套接字到IP地址和端口号。
  • 调用listen()方法监听连接请求。
  • 使用accept()方法接受连接请求，进入通信循环。
  • 在通信循环中，使用recv()接收数据，使用send()或sendall()发送数据。
  • 关闭套接字。
  1. import socket

  3. HOST = '127.0.0.1'  # 监听所有可用接口
  4. PORT = 9001
  5. ADDR = (HOST, PORT)
  6. BUFSIZ = 1024

  8. def tcp_server():
  9.     with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
  10.         s.bind(ADDR)
  11.         s.listen()
  12.         print('等待用户接入...')
  13.         while True:
  14.             conn, addr = s.accept()
  15.             with conn:
  16.                 print(f'连接来自 {addr}')
  17.                 while True:
  18.                     data = conn.recv(BUFSIZ)
  19.                     if not data:
  20.                         break
  21.                     print(f'收到数据: {data.decode()}')
  22.                     conn.send(data)  # Echo服务器，原样返回数据

  24. if __name__ == '__main__':
  25.     tcp_server()
复制代码


2.2.2 创建TCP客户端

创建TCP客户端的步骤如下：

  • 导入socket模块。
  • 创建套接字对象，指定AF_INET和SOCK_STREAM。
  • 使用connect()方法连接到服务器。
  • 进入通信循环，使用send()发送数据，使用recv()接收数据。
  • 关闭套接字。
  1. import socket

  3. HOST = '127.0.0.1'
  4. PORT = 9001
  5. ADDR = (HOST, PORT)
  6. BUFSIZ = 1024

  8. def tcp_client():
  9.     with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
  10.         s.connect(ADDR)
  11.         print('连接服务成功!!')
  12.         while True:
  13.             data = input('请输入数据: ')
  14.             if data == 'q':
  15.                 break
  16.             s.send(data.encode())
  17.             recved =s.recv(BUFSIZ).decode()
  18.             print(f'收到服务器响应: {recved}')

  20. if __name__ == '__main__':
  21.     tcp_client()
复制代码

客户端输入消息，收到服务端同样的返回


2.3 创建UDP服务和客户端

UDP服务与TCP服务的主要区别在于无连接和不可靠的数据传输。

2.3.1 创建UDP服务器

UDP服务器通常不需要监听连接请求，因为它不建立持久的连接。服务器只需绑定到特定端口，并等待接收数据。
  1. import socket

  3. HOST = '127.0.0.1'
  4. PORT = 9002
  5. BUFSIZ = 1024
  6. ADDR = (HOST, PORT)

  8. def udp_server():
  9.     with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
  10.         s.bind(ADDR)
  11.         print('UDP服务器启动...')
  12.         while True:
  13.             data, addr = s.recvfrom(BUFSIZ)
  14.             print(f'收到来自 {addr} 的数据: {data.decode()}')
  15.             resp = '收到您的消息！'.encode()
  16.             s.sendto(resp, addr)

  18. if __name__ == '__main__':
  19.     udp_server()
复制代码


2.3.2 创建UDP客户端

UDP客户端向服务器发送数据，并等待接收响应（尽管这不是必须的，因为UDP是无连接的）。
  1. import socket

  3. HOST = '127.0.0.1'
  4. PORT = 9002
  5. BUFSIZ = 1024
  6. ADDR = (HOST, PORT)

  8. def udp_client():
  9.     with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
  10.         while True:
  11.             data = input('请输入数据（输入q退出）: ')
  12.             if data == 'q':
  13.                 break
  14.             s.sendto(data.encode(), ADDR)
  15.             resp, server = s.recvfrom(BUFSIZ)
  16.             print(f'收到服务器响应: {resp.decode()}')

  18. if __name__ == '__main__':
  19.     udp_client()
复制代码

客户端输入消息，也能收到服务端消息


三、实际案例：简单的聊天室应用

以下是一个简单的TCP聊天室应用，它包含一个服务器和多个客户端。服务器将来自任一客户端的消息广播给所有已连接的客户端。
TCP聊天室服务器
  1. import socket
  2. import threading

  4. HOST = 'localhost'
  5. PORT = 9003
  6. BUFSIZ = 1024
  7. ADDR = (HOST, PORT)
  8. clients = []

  10. def handle_client(conn, addr):
  11.     print(f'连接来自 {addr}')
  12.     conn.send('欢迎来到聊天室！'.encode())
  13.     clients.append(conn)
  14.     try:
  15.         while True:
  16.             data = conn.recv(BUFSIZ)
  17.             if not data:
  18.                 break
  19.             broadcast(data, conn)
  20.     finally:
  21.         conn.close()
  22.         clients.remove(conn)

  24. def broadcast(data, sender):
  25.     for client in clients:
  26.         print("打印客户端",client)
  27.         print("打印客户端发来的数据",data.decode())
  28.         if client == sender:
  29.             try:
  30.                 #将客户端的原文发动给客户端
  31.                 sender.send(data)
  32.             except:
  33.                 clients.remove(client)

  35. def main():
  36.     with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
  37.         s.bind(ADDR)
  38.         s.listen()
  39.         print('等待用户接入...')
  40.         while True:
  41.             conn, addr = s.accept()
  42.             thread = threading.Thread(target=handle_client, args=(conn, addr))
  43.             thread.start()

  45. if __name__ == '__main__':
  46.     main()
复制代码

TCP聊天室客户端
  1. import socket

  3. HOST = 'localhost'
  4. PORT = 9003
  5. BUFSIZ = 1024
  6. ADDR = (HOST, PORT)

  8. def chat_client():
  9.     with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
  10.         s.connect(ADDR)
  11.         print(s.recv(BUFSIZ).decode())
  12.         while True:
  13.             data = input('请输入消息（输入q退出）: ')
  14.             if data == 'q':
  15.                 break
  16.             s.send(data.encode())
  17.             if data.lower() == 'bye':
  18.                 break
  19.             print(s.recv(BUFSIZ).decode())

  21. if __name__ == '__main__':
  22.     chat_client()
复制代码

我们也可以运行好几个客户端，服务端打印哪个客户端连过来

当然，我们可以继续深入讨论Socket编程及其在Python中的应用，包括一些高级话题和最佳实践。以下是一些扩展内容：

四、高级话题和最佳实践


1. 异步Socket编程

Python的
  1. asyncio
复制代码
库提供了对异步IO的支持，这使得编写非阻塞的Socket客户端和服务器变得更加容易。使用
  1. asyncio
复制代码
，你可以编写出性能更高、响应更快的网络应用程序。
异步TCP服务器示例
  1. import asyncio

  3. async def handle_echo(reader, writer):
  4.     data = await reader.read(100)
  5.     message = data.decode()
  6.     addr = writer.get_extra_info('peername')
  7.     print(f"Received {message} from {addr}")

  9.     print(f"Send: {message.upper()} to {addr}")
  10.     writer.write(message.upper().encode())
  11.     await writer.drain()

  13.     print("Close the connection")
  14.     writer.close()

  16. async def main():
  17.     server = await asyncio.start_server(
  18.         handle_echo, '127.0.0.1', 8888)

  20.     addr = server.sockets[0].getsockname()
  21.     print(f'Serving on {addr}')

  23.     async with server:
  24.         await server.serve_forever()

  26. asyncio.run(main())
复制代码
异步TCP客户端示例
  1. import asyncio

  3. async def tcp_echo_client(message, host, port):
  4.     reader, writer = await asyncio.open_connection(host, port)

  6.     print(f'Send: {message}')
  7.     writer.write(message.encode())

  9.     data = await reader.read(100)
  10.     print(f'Received: {data.decode()}')

  12.     writer.close()

  14. asyncio.run(tcp_echo_client('Hello, world!', '127.0.0.1', 8888))
复制代码
2. 错误处理和异常

在网络编程中，处理各种网络错误和异常是非常重要的。例如，连接超时、连接中断、数据发送失败等都是常见的错误情况。
在Python中，你可以通过
  1. try...except
复制代码
块来捕获和处理这些异常。例如，使用
  1. socket.timeout
复制代码
来处理连接超时，使用
  1. socket.error
复制代码
来处理一般的socket错误。

3. 安全性

当编写网络应用程序时，安全性是一个重要的考虑因素。以下是一些提升Socket程序安全性的方法：

  • 使用SSL/TLS来加密数据传输。Python的
    1. ssl
    复制代码
    模块可以与
    1. socket
    复制代码
    模块一起使用来创建安全的socket连接。
  • 验证客户端的身份（如果需要的话）。可以使用证书、密钥或令牌等方式来验证客户端的合法性。
  • 限制可以连接到服务器的IP地址或端口范围。这可以通过防火墙规则或服务器软件中的设置来实现。

4. 性能和优化

Socket编程的性能优化可以涉及多个方面，包括网络延迟、吞吐量、并发处理能力等。以下是一些优化技巧：

  • 使用非阻塞或异步IO来减少等待时间。
  • 使用合适的缓冲区大小来平衡内存使用和网络性能。
  • 使用多线程或多进程来处理多个并发连接。
  • 压缩数据以减少传输量，特别是当数据量大时。
  • 使用缓存来存储常用数据，减少对后端服务的请求。

5. 调试和日志记录

在网络编程中，调试和日志记录是非常重要的工具。它们可以帮助你理解程序的行为，诊断问题，并优化性能。

  • 使用Python的
    1. logging
    复制代码
    模块来记录程序的运行状态、错误信息和警告。
  • 在关键位置添加调试语句或断点，以便在出现问题时能够追踪程序的执行流程。
  • 使用网络抓包工具（如Wireshark）来捕获和分析网络数据包，了解数据的实际传输情况。

五、总结

Socket编程是构建网络应用程序的基础。通过掌握Socket编程的基本概念、TCP和UDP的使用方法，以及异步编程、错误处理、安全性、性能和调试等高级话题，你可以开发出高效、稳定、安全的网络应用程序。希望本文对你有所帮助！
以上就是Python中Socket编程底层原理解析与应用实战的详细内容，更多关于Python Socket编程的资料请关注脚本之家其它相关文章！

来源:https://www.jb51.net/python/326605u76.htm
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！

本帖子中包含更多资源

您需要 登录 才可以下载或查看，没有账号？立即注册

x

上一篇: Python中的分布式框架Ray的安装与使用教程

下一篇: 使用icecream实现优雅调试Python代码

发表于 2024-9-11 18:18:03

举报 回复 使用道具

返回列表 发新帖

Archiver|手机版|小黑屋|翼度科技

Copyright © 2001-2020, Tencent Cloud.　Powered by Discuz! X3.4