✨✨ 欢迎大家来到景天科技苑✨✨
🎈🎈 养成好习惯,先赞后看哦~🎈🎈
🏆 作者简介:景天科技苑
🏆《头衔》:大厂架构师,华为云开发者社区专家博主,阿里云开发者社区专家博主,CSDN全栈领域优质创作者,掘金优秀博主,51CTO博客专家等。
🏆《博客》:Python全栈,PyQt5和Tkinter桌面开发,小程序开发,人工智能,js逆向,App逆向,网络系统安全,数据分析,Django,fastapi,flask等框架,云原生K8S,linux,shell脚本等实操经验,网站搭建,数据库等分享。所属的专栏:python综合应用,基础语法到高阶实战教学
景天的主页:景天科技苑
文章目录
Python中的Socket编程
Socket编程是网络通信的基础,Python通过内置的socket模块提供了强大的网络编程接口。Socket编程允许开发者创建客户端和服务器程序,实现数据在网络中的传输。本文将结合实际案例,详细介绍Python中Socket编程的基本概念、常用方法和实际应用。
一、基本概念
-
服务器与客户端
- 服务器:一系列硬件或软件,为一个或多个客户端(服务的用户)提供所需的“服务”。它响应客户端的请求,并等待更多的请求。
- 客户端:向服务器请求服务的用户或程序。它发送请求并接收服务器的响应。
-
套接字(Socket)
- 套接字是计算机网络数据结构,它体现了通信端点的概念。在任何类型的通信开始之前,网络应用程序必须创建套接字。套接字可以被比作电话插孔,没有它将无法进行通信。
- Python中的socket模块支持多种类型的套接字,包括基于IPv4的AF_INET(流式套接字和数据报套接字)。
-
流式套接字(SOCK_STREAM)
- 用于提供面向连接、可靠的数据传输服务。它基于TCP协议,确保数据按顺序、无差错、不重复地传输。
-
数据报套接字(SOCK_DGRAM)
- 提供一种无连接的服务。数据报套接字基于UDP协议,不保证数据传输的可靠性,数据可能在传输过程中丢失或重复,且无法保证顺序地接收到数据。
二、Python中的Socket编程
2.1 常见的套接字对象方法和属性
在Python的socket编程中,常用的方法包括:
socket()
: 创建一个新的套接字对象。bind()
: 将套接字绑定到特定的IP地址和端口号上。listen()
: 开始监听客户端的连接请求。accept()
: 接受一个连接请求,并返回一个新的套接字对象和客户端的地址信息。connect()
: 客户端使用该方法连接到服务器。send() / sendall()
: 发送数据到连接的套接字。sendall()
确保数据被完整发送。recv() / recvfrom()
: 接收数据。recv()
用于TCP连接,recvfrom()
用于UDP连接。close()
: 关闭套接字。
2.2 创建TCP服务和客户端
2.2.1 创建TCP服务
创建一个简单的TCP服务器,步骤如下:
- 导入socket模块。
- 创建套接字对象,指定AF_INET和SOCK_STREAM。
- 绑定套接字到IP地址和端口号。
- 调用listen()方法监听连接请求。
- 使用accept()方法接受连接请求,进入通信循环。
- 在通信循环中,使用recv()接收数据,使用send()或sendall()发送数据。
- 关闭套接字。
import socket
HOST = '127.0.0.1' # 监听所有可用接口
PORT = 9001
ADDR = (HOST, PORT)
BUFSIZ = 1024
def tcp_server():
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.bind(ADDR)
s.listen()
print('等待用户接入...')
while True:
conn, addr = s.accept()
with conn:
print(f'连接来自 {addr}')
while True:
data = conn.recv(BUFSIZ)
if not data:
break
print(f'收到数据: {data.decode()}')
conn.send(data) # Echo服务器,原样返回数据
if __name__ == '__main__':
tcp_server()
2.2.2 创建TCP客户端
创建TCP客户端的步骤如下:
- 导入socket模块。
- 创建套接字对象,指定AF_INET和SOCK_STREAM。
- 使用connect()方法连接到服务器。
- 进入通信循环,使用send()发送数据,使用recv()接收数据。
- 关闭套接字。
import socket
HOST = '127.0.0.1'
PORT = 9001
ADDR = (HOST, PORT)
BUFSIZ = 1024
def tcp_client():
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect(ADDR)
print('连接服务成功!!')
while True:
data = input('请输入数据: ')
if data == 'q':
break
s.send(data.encode())
recved =s.recv(BUFSIZ).decode()
print(f'收到服务器响应: {recved}')
if __name__ == '__main__':
tcp_client()
客户端输入消息,收到服务端同样的返回
2.3 创建UDP服务和客户端
UDP服务与TCP服务的主要区别在于无连接和不可靠的数据传输。
2.3.1 创建UDP服务器
UDP服务器通常不需要监听连接请求,因为它不建立持久的连接。服务器只需绑定到特定端口,并等待接收数据。
import socket
HOST = '127.0.0.1'
PORT = 9002
BUFSIZ = 1024
ADDR = (HOST, PORT)
def udp_server():
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
s.bind(ADDR)
print('UDP服务器启动...')
while True:
data, addr = s.recvfrom(BUFSIZ)
print(f'收到来自 {addr} 的数据: {data.decode()}')
resp = '收到您的消息!'.encode()
s.sendto(resp, addr)
if __name__ == '__main__':
udp_server()
2.3.2 创建UDP客户端
UDP客户端向服务器发送数据,并等待接收响应(尽管这不是必须的,因为UDP是无连接的)。
import socket
HOST = '127.0.0.1'
PORT = 9002
BUFSIZ = 1024
ADDR = (HOST, PORT)
def udp_client():
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
while True:
data = input('请输入数据(输入q退出): ')
if data == 'q':
break
s.sendto(data.encode(), ADDR)
resp, server = s.recvfrom(BUFSIZ)
print(f'收到服务器响应: {resp.decode()}')
if __name__ == '__main__':
udp_client()
客户端输入消息,也能收到服务端消息
三、实际案例:简单的聊天室应用
以下是一个简单的TCP聊天室应用,它包含一个服务器和多个客户端。服务器将来自任一客户端的消息广播给所有已连接的客户端。
TCP聊天室服务器
import socket
import threading
HOST = 'localhost'
PORT = 9003
BUFSIZ = 1024
ADDR = (HOST, PORT)
clients = []
def handle_client(conn, addr):
print(f'连接来自 {addr}')
conn.send('欢迎来到聊天室!'.encode())
clients.append(conn)
try:
while True:
data = conn.recv(BUFSIZ)
if not data:
break
broadcast(data, conn)
finally:
conn.close()
clients.remove(conn)
def broadcast(data, sender):
for client in clients:
print("打印客户端",client)
print("打印客户端发来的数据",data.decode())
if client == sender:
try:
#将客户端的原文发动给客户端
sender.send(data)
except:
clients.remove(client)
def main():
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.bind(ADDR)
s.listen()
print('等待用户接入...')
while True:
conn, addr = s.accept()
thread = threading.Thread(target=handle_client, args=(conn, addr))
thread.start()
if __name__ == '__main__':
main()
TCP聊天室客户端
import socket
HOST = 'localhost'
PORT = 9003
BUFSIZ = 1024
ADDR = (HOST, PORT)
def chat_client():
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect(ADDR)
print(s.recv(BUFSIZ).decode())
while True:
data = input('请输入消息(输入q退出): ')
if data == 'q':
break
s.send(data.encode())
if data.lower() == 'bye':
break
print(s.recv(BUFSIZ).decode())
if __name__ == '__main__':
chat_client()
我们也可以运行好几个客户端,服务端打印哪个客户端连过来
当然,我们可以继续深入讨论Socket编程及其在Python中的应用,包括一些高级话题和最佳实践。以下是一些扩展内容:
四、高级话题和最佳实践
1. 异步Socket编程
Python的asyncio
库提供了对异步IO的支持,这使得编写非阻塞的Socket客户端和服务器变得更加容易。使用asyncio
,你可以编写出性能更高、响应更快的网络应用程序。
异步TCP服务器示例
import asyncio
async def handle_echo(reader, writer):
data = await reader.read(100)
message = data.decode()
addr = writer.get_extra_info('peername')
print(f"Received {message} from {addr}")
print(f"Send: {message.upper()} to {addr}")
writer.write(message.upper().encode())
await writer.drain()
print("Close the connection")
writer.close()
async def main():
server = await asyncio.start_server(
handle_echo, '127.0.0.1', 8888)
addr = server.sockets[0].getsockname()
print(f'Serving on {addr}')
async with server:
await server.serve_forever()
asyncio.run(main())
异步TCP客户端示例
import asyncio
async def tcp_echo_client(message, host, port):
reader, writer = await asyncio.open_connection(host, port)
print(f'Send: {message}')
writer.write(message.encode())
data = await reader.read(100)
print(f'Received: {data.decode()}')
writer.close()
asyncio.run(tcp_echo_client('Hello, world!', '127.0.0.1', 8888))
2. 错误处理和异常
在网络编程中,处理各种网络错误和异常是非常重要的。例如,连接超时、连接中断、数据发送失败等都是常见的错误情况。
在Python中,你可以通过try...except
块来捕获和处理这些异常。例如,使用socket.timeout
来处理连接超时,使用socket.error
来处理一般的socket错误。
3. 安全性
当编写网络应用程序时,安全性是一个重要的考虑因素。以下是一些提升Socket程序安全性的方法:
- 使用SSL/TLS来加密数据传输。Python的
ssl
模块可以与socket
模块一起使用来创建安全的socket连接。 - 验证客户端的身份(如果需要的话)。可以使用证书、密钥或令牌等方式来验证客户端的合法性。
- 限制可以连接到服务器的IP地址或端口范围。这可以通过防火墙规则或服务器软件中的设置来实现。
4. 性能和优化
Socket编程的性能优化可以涉及多个方面,包括网络延迟、吞吐量、并发处理能力等。以下是一些优化技巧:
- 使用非阻塞或异步IO来减少等待时间。
- 使用合适的缓冲区大小来平衡内存使用和网络性能。
- 使用多线程或多进程来处理多个并发连接。
- 压缩数据以减少传输量,特别是当数据量大时。
- 使用缓存来存储常用数据,减少对后端服务的请求。
5. 调试和日志记录
在网络编程中,调试和日志记录是非常重要的工具。它们可以帮助你理解程序的行为,诊断问题,并优化性能。
- 使用Python的
logging
模块来记录程序的运行状态、错误信息和警告。 - 在关键位置添加调试语句或断点,以便在出现问题时能够追踪程序的执行流程。
- 使用网络抓包工具(如Wireshark)来捕获和分析网络数据包,了解数据的实际传输情况。
五、总结
Socket编程是构建网络应用程序的基础。通过掌握Socket编程的基本概念、TCP和UDP的使用方法,以及异步编程、错误处理、安全性、性能和调试等高级话题,你可以开发出高效、稳定、安全的网络应用程序。希望本文对你有所帮助!