327 lines
8.6 KiB
Markdown
327 lines
8.6 KiB
Markdown
---
|
||
title: socket 底层网络接口
|
||
description: socket 底层网络接口
|
||
keywords:
|
||
- Python
|
||
- socket
|
||
tags:
|
||
- Python/进阶
|
||
- Python/标准库
|
||
author: 7Wate
|
||
date: 2023-09-09
|
||
---
|
||
|
||
## 概述
|
||
|
||
Socket 是计算机网络中用于实现进程间通信的一种技术。简而言之，它是一个端点，用于发送或接收数据。这些数据可以在同一台计算机上的进程之间，或者在不同计算机上的进程之间进行传输。
|
||
|
||
`socket` 是 Python 中的一个标准库，它提供了低级的网络通信功能。基于该库，开发者可以实现 TCP、UDP 以及其他类型的网络协议的客户端和服务端程序。
|
||
|
||
`socket` 库的历史可以追溯到 Unix 操作系统的早期。在 Unix 中，socket 被用作进程间通信 (IPC) 的一种方式。随着互联网的发展，socket 逐渐被应用于网络通信。Python 的 `socket` 库提供了对这一功能的高级封装，使得开发网络应用变得更为简单。
|
||
|
||
## 工作原理
|
||
|
||
`socket` 网络通信最常见的模型是客户端 - 服务器模型。客户端是请求服务的实体，而服务器是提供服务的实体。Socket 允许这两个实体进行通信。
|
||
|
||
- **客户端**: 创建 Socket -> 连接到服务器 -> 交换数据 -> 关闭 Socket
|
||
- **服务器**: 创建 Socket -> 绑定到特定地址和端口 -> 监听连接 -> 接受客户端连接 -> 交换数据 -> 关闭 Socket
|
||
|
||
`socket` 主要被用于：
|
||
|
||
- 创建网络服务器和客户端。
|
||
- 实现多种网络协议，如 TCP 和 UDP。
|
||
- 进行低级的网络通信，如发送原始套接字数据。
|
||
|
||
它解决了开发者在进行网络通信时的底层细节问题，如地址转换、数据传输等。
|
||
|
||
### 协议类型
|
||
|
||
- **流式 Socket (SOCK_STREAM)**: 提供了一个可靠的、面向连接的字节流。这是最常用的 Socket 类型，TCP 协议就是基于这种 Socket。
|
||
- **数据报 Socket (SOCK_DGRAM)**: 提供了一种无连接的、不可靠的数据传输方式。UDP 协议就是基于这种 Socket。
|
||
- **原始 Socket (SOCK_RAW)**: 允许直接访问底层协议，通常用于开发新的通信协议。
|
||
|
||
### 地址家族
|
||
|
||
- **AF_INET (IPv4)**: 最常用的地址家族，用于 Internet 通信。
|
||
- **AF_INET6 (IPv6)**: 下一代 Internet 地址家族。
|
||
- **AF_UNIX**: 用于同一台机器上的进程间通信。
|
||
|
||
### 生命周期
|
||
|
||
```mermaid
|
||
graph LR;
|
||
A[创建 Socket] --> B[绑定 Socket到地址和端口];
|
||
B --> C[开始监听传入的连接];
|
||
C --> D[接受连接请求];
|
||
D --> E[进行数据传输];
|
||
E --> F[关闭 Socket];
|
||
```
|
||
|
||
1. **创建**: 使用 `socket.socket()` 创建一个新的 Socket 对象。
|
||
2. **绑定**: 使用 `bind()` 方法将 Socket 绑定到特定的地址和端口。
|
||
3. **监听**: 使用 `listen()` 方法开始监听传入的连接。
|
||
4. **接受连接**: 使用 `accept()` 方法接受一个连接。
|
||
5. **数据传输**: 使用 `send()` 和 `recv()` 方法进行数据传输。
|
||
6. **关闭**: 使用 `close()` 方法关闭 Socket。
|
||
|
||
## 基础操作
|
||
|
||
由于 `socket` 是 Python 的标准库，无需额外安装。
|
||
|
||
### 支持协议
|
||
|
||
`socket` 库支持多种协议，用于不同的网络通信场景。
|
||
|
||
```python
|
||
# 服务器
|
||
import socket
|
||
|
||
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
server_socket.bind(('localhost', 8080))
|
||
server_socket.listen(5)
|
||
|
||
print("服务器正在监听...")
|
||
client_socket, addr = server_socket.accept()
|
||
print(f"连接来自 {addr}")
|
||
|
||
data = client_socket.recv(1024)
|
||
print(f"接收到的数据: {data.decode('utf-8')}")
|
||
client_socket.send("谢谢你!".encode('utf-8'))
|
||
|
||
client_socket.close()
|
||
server_socket.close()
|
||
```
|
||
|
||
```python
|
||
# 客户端
|
||
import socket
|
||
|
||
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
client_socket.connect(('localhost', 8080))
|
||
|
||
client_socket.send("你好，服务器!".encode('utf-8'))
|
||
data = client_socket.recv(1024)
|
||
print(f"接收到的数据: {data.decode('utf-8')}")
|
||
|
||
client_socket.close()
|
||
```
|
||
|
||
### TCP 示例
|
||
|
||
```mermaid
|
||
graph LR
|
||
|
||
subgraph TCP服务器
|
||
A[创建Socket] --> B[绑定到地址和端口]
|
||
B --> C[监听连接请求]
|
||
C --> D[接受连接]
|
||
D --> E[接收数据]
|
||
E --> F[处理数据]
|
||
F --> G[发送响应]
|
||
G --> E
|
||
E --> H[关闭连接]
|
||
H --> D
|
||
end
|
||
|
||
subgraph TCP客户端
|
||
I[创建Socket] --> J[连接到服务器]
|
||
J --> K[发送数据]
|
||
K --> L[接收响应]
|
||
L --> M[处理响应]
|
||
M --> K
|
||
K --> N[关闭连接]
|
||
end
|
||
|
||
A --> I
|
||
|
||
```
|
||
|
||
#### 服务器
|
||
|
||
```python
|
||
import socket
|
||
|
||
# 步骤1：创建 TCP Socket
|
||
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
||
# 步骤2：绑定 Socket 到一个特定的 IP 地址和端口
|
||
server_address = ('localhost', 12345)
|
||
server_socket.bind(server_address)
|
||
|
||
# 步骤3：监听传入的连接请求
|
||
server_socket.listen(5)
|
||
|
||
print("TCP 服务器启动，等待连接...")
|
||
|
||
while True:
|
||
# 步骤4：接受客户端的连接请求
|
||
client_socket, client_address = server_socket.accept()
|
||
print(f"接受来自 {client_address} 的连接")
|
||
|
||
# 步骤5：与客户端建立通信
|
||
while True:
|
||
data = client_socket.recv(1024)
|
||
if not data:
|
||
break
|
||
|
||
# 步骤6：接收和发送数据
|
||
data = data.decode('utf-8')
|
||
print(f"接收到来自客户端的数据: {data}")
|
||
response = f"你好，
|
||
```
|
||
|
||
#### 客户端
|
||
|
||
```python
|
||
import socket
|
||
|
||
# 步骤1：创建 TCP Socket
|
||
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
||
# 步骤2：连接到服务器的 IP 地址和端口
|
||
server_address = ('localhost', 12345)
|
||
client_socket.connect(server_address)
|
||
|
||
while True:
|
||
# 步骤3：发送数据给服务器
|
||
message = input("请输入要发送的消息 (输入 'exit' 退出): ")
|
||
|
||
if message == 'exit':
|
||
break
|
||
|
||
client_socket.send(message.encode('utf-8'))
|
||
|
||
# 步骤4：接收来自服务器的响应数据
|
||
data = client_socket.recv(1024)
|
||
|
||
# 步骤5：处理服务器的响应数据
|
||
data = data.decode('utf-8')
|
||
print(f"接收到服务器的响应: {data}")
|
||
|
||
# 步骤6：关闭与服务器的连接
|
||
client_socket.close()
|
||
```
|
||
|
||
### UDP 示例
|
||
|
||
```mermaid
|
||
graph LR
|
||
|
||
subgraph UDP服务器
|
||
A[创建Socket] --> B[绑定到地址和端口]
|
||
B --> C[接收数据]
|
||
C --> D[处理数据]
|
||
D --> E[发送响应]
|
||
E --> C
|
||
end
|
||
|
||
subgraph UDP客户端
|
||
F[创建Socket] --> G[发送数据]
|
||
G --> H[接收响应]
|
||
H --> I[处理响应]
|
||
I --> G
|
||
end
|
||
|
||
A --> F
|
||
|
||
```
|
||
|
||
#### 服务端
|
||
|
||
```python
|
||
import socket
|
||
|
||
# 步骤1：创建 UDP Socket
|
||
server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
||
# 步骤2：绑定 Socket 到一个特定的 IP 地址和端口
|
||
server_address = ('localhost', 12345)
|
||
server_socket.bind(server_address)
|
||
|
||
print("UDP 服务器启动，等待数据...")
|
||
|
||
while True:
|
||
# 步骤3：接收客户端发送的数据
|
||
data, client_address = server_socket.recvfrom(1024)
|
||
|
||
# 步骤4：处理客户端数据
|
||
data = data.decode('utf-8')
|
||
print(f"接收到来自 {client_address} 的数据: {data}")
|
||
|
||
# 步骤5：为客户端发送响应（如果需要）
|
||
response = f"你好，客户端！你发送了：{data}"
|
||
server_socket.sendto(response.encode('utf-8'), client_address)
|
||
|
||
# 步骤6：关闭 Socket（这里永远不会执行到）
|
||
server_socket.close()
|
||
```
|
||
|
||
#### 客户端
|
||
|
||
```python
|
||
import socket
|
||
|
||
# 步骤1：创建 UDP Socket
|
||
client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
||
# 服务器地址和端口
|
||
server_address = ('localhost', 12345)
|
||
|
||
while True:
|
||
# 步骤2：发送数据给服务器
|
||
message = input("请输入要发送的消息 (输入 'exit' 退出): ")
|
||
|
||
if message == 'exit':
|
||
break
|
||
|
||
client_socket.sendto(message.encode('utf-8'), server_address)
|
||
|
||
# 步骤3：接收来自服务器的响应数据
|
||
data, server_address = client_socket.recvfrom(1024)
|
||
|
||
# 步骤4：处理服务器的响应数据
|
||
data = data.decode('utf-8')
|
||
print(f"接收到服务器的响应: {data}")
|
||
|
||
# 步骤5：关闭 Socket
|
||
client_socket.close()
|
||
```
|
||
|
||
### 常用方法
|
||
|
||
| 方法 | 描述 |
|
||
| --------- | -------------------- |
|
||
| socket() | 创建一个新的 socket |
|
||
| bind() | 绑定到一个地址和端口 |
|
||
| listen() | 开始监听连接 |
|
||
| accept() | 接受一个新的连接 |
|
||
| connect() | 连接到一个服务器 |
|
||
| send() | 发送数据 |
|
||
| recv() | 接收数据 |
|
||
| close() | 关闭 socket |
|
||
|
||
## 深入探索
|
||
|
||
### 高级功能
|
||
|
||
#### 非阻塞 Sockets
|
||
|
||
```python
|
||
import socket
|
||
|
||
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
s.setblocking(False)
|
||
```
|
||
|
||
#### 套接字选项
|
||
|
||
```python
|
||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||
```
|
||
|
||
#### 超时
|
||
|
||
```python
|
||
s.settimeout(5.0)
|
||
```
|