---
title: socket 底层网络接口
description: socket 底层网络接口
keywords:
  - Python
  - socket
tags:
  - 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)
```