保姆级教程：手把手教你用Python脚本读取PCIe设备的BAR空间大小（附代码）
·
实战指南：Python脚本解析PCIe设备BAR空间的完整方案
当你拿到一块全新的PCIe设备时，第一件事往往是了解它的内存映射情况。无论是网卡、GPU还是FPGA加速卡，BAR(Base Address Register)空间都是设备与主机通信的关键桥梁。本文将带你从零开始，用Python脚本直接与PCIe配置空间对话，不依赖厂商工具就能获取BAR的详细属性。
1. 理解PCIe配置空间与BAR寄存器
PCIe设备的配置空间就像它的身份证和通讯录，存储了设备类型、厂商ID以及最重要的BAR信息。在Linux系统中，这个配置空间通常有256字节或4KB大小，前64字节是标准的PCI配置头。
BAR寄存器有几个关键特性需要掌握：
- 位宽区分 ：32位BAR和64位BAR的地址映射方式不同
- 类型标识 ：最低位(bit0)决定是MEM空间(0)还是IO空间(1)
- 预取属性 ：对于MEM类型，bit3指示是否支持预取(prefetchable)
- 大小编码 ：通过写全1再回读的方式可以获取BAR实际大小
# BAR寄存器属性掩码
BAR_TYPE_MASK = 0x1
BAR_MEM_TYPE_MASK = 0x6
BAR_PREFETCH_MASK = 0x8
2. 环境准备与工具链配置
在开始编码前，我们需要确保环境具备必要的访问权限和工具：
-
硬件需求 ：
- 支持PCIe的x86或ARM平台
- 待检测的PCIe设备已正确安装
-
软件依赖 ：
- Python 3.6+ 环境
pciutils工具包(提供底层访问接口)- root权限(直接访问硬件配置空间)
# 安装必要工具
sudo apt-get install pciutils python3-dev
pip install pyudev ctypes
- 验证设备识别 ： 先用lspci命令确认设备已被系统识别：
lspci -nn | grep -i "你的设备关键词"
3. Python实现BAR空间探测
我们将分步骤实现一个完整的BAR探测脚本，以下是核心代码框架：
import os
import ctypes
import struct
from pyudev import Context
class PCIeDevice:
def __init__(self, vendor_id, device_id):
self.vendor_id = vendor_id
self.device_id = device_id
self.config_fd = None
self.bar_info = []
def open_config(self):
"""打开设备的配置空间文件"""
context = Context()
for device in context.list_devices(subsystem='pci'):
if (device.get('ID_VENDOR_ID') == self.vendor_id and
device.get('ID_MODEL_ID') == self.device_id):
config_path = f"/sys/bus/pci/devices/{device.sys_name}/config"
self.config_fd = os.open(config_path, os.O_RDWR)
return True
return False
3.1 读取BAR寄存器原始值
每个BAR寄存器在配置空间中的偏移量是固定的：
| BAR编号 | Type0偏移量 | Type1偏移量 |
|---|---|---|
| BAR0 | 0x10 | 0x10 |
| BAR1 | 0x14 | 0x14 |
| BAR2 | 0x18 | N/A |
| BAR3 | 0x1C | N/A |
| BAR4 | 0x20 | N/A |
| BAR5 | 0x24 | N/A |
读取BAR寄存器的实现：
def read_bar_register(self, bar_num):
"""读取指定BAR寄存器的值"""
if bar_num < 0 or bar_num > 5:
raise ValueError("BAR编号必须在0-5之间")
offset = 0x10 + bar_num * 4
os.lseek(self.config_fd, offset, os.SEEK_SET)
bar_value = struct.unpack('I', os.read(self.config_fd, 4))[0]
return bar_value
3.2 解析BAR属性
获取BAR原始值后，需要解码其中的属性信息：
def decode_bar_type(self, bar_value):
"""解析BAR类型和属性"""
bar_type = "MEM" if not (bar_value & BAR_TYPE_MASK) else "IO"
if bar_type == "MEM":
mem_type = (bar_value & BAR_MEM_TYPE_MASK) >> 1
width = "32-bit" if mem_type == 0 else "64-bit"
prefetch = "Prefetchable" if bar_value & BAR_PREFETCH_MASK else "Non-prefetchable"
return f"{width} {prefetch} {bar_type}"
else:
return "IO Space"
3.3 计算BAR空间大小
确定BAR大小的标准方法是：
- 保存BAR原始值
- 向BAR写入全1(0xFFFFFFFF)
- 回读BAR值
- 计算最低置位位
- 恢复BAR原始值
def calculate_bar_size(self, bar_num):
"""计算BAR空间大小"""
original_value = self.read_bar_register(bar_num)
# 写入全1
self.write_bar_register(bar_num, 0xFFFFFFFF)
# 回读并计算大小
readback = self.read_bar_register(bar_num)
if readback == 0:
return 0 # BAR未使用
# 计算最低置位位
size_mask = readback & ~(BAR_TYPE_MASK | BAR_MEM_TYPE_MASK | BAR_PREFETCH_MASK)
size = (~size_mask + 1) & 0xFFFFFFFF
# 恢复原始值
self.write_bar_register(bar_num, original_value)
return size
4. 完整脚本实现与结果解析
将上述模块组合成完整的探测工具：
def scan_all_bars(self):
"""扫描并解析所有BAR寄存器"""
for bar_num in range(6):
try:
bar_value = self.read_bar_register(bar_num)
if bar_value == 0:
continue
bar_type = self.decode_bar_type(bar_value)
bar_size = self.calculate_bar_size(bar_num)
self.bar_info.append({
'number': bar_num,
'type': bar_type,
'size': bar_size,
'address': bar_value & ~0xF
})
except IOError:
break
return self.bar_info
4.1 典型输出示例
运行脚本后，我们可能得到如下结果：
| BAR编号 | 类型 | 大小 | 基地址 |
|---|---|---|---|
| BAR0 | 32-bit Prefetchable MEM | 256 KB | 0xD1000000 |
| BAR1 | 64-bit Non-prefetchable MEM | 16 MB | 0xD2000000 |
| BAR2 | IO Space | 4 KB | 0x00002000 |
4.2 结果验证技巧
为确保结果准确，可以通过以下方法交叉验证：
-
对比lspci输出 ：
lspci -vvv -s 01:00.0 | grep -i "memory at" -
检查内核日志 ：
dmesg | grep -i pci -
直接访问测试 ： 对于MEM类型BAR，可以尝试映射并读取开头几个字节：
def test_bar_access(self, bar_num):
"""测试BAR空间访问"""
bar = self.bar_info[bar_num]
if 'MEM' not in bar['type']:
return False
try:
mem = mmap.mmap(os.open(f"/sys/bus/pci/devices/.../resource{bar_num}",
os.O_RDWR),
bar['size'])
# 读取前4字节
header = mem.read(4)
return bool(header)
except:
return False
5. 高级应用与调试技巧
掌握了基础BAR探测后，可以进一步开发更高级的功能：
5.1 64位BAR的特殊处理
64位BAR由两个相邻的32位寄存器组成，需要特殊处理：
def handle_64bit_bar(self, bar_num):
"""处理64位BAR地址"""
if bar_num >= 5:
return None
lower = self.read_bar_register(bar_num)
upper = self.read_bar_register(bar_num + 1)
# 检查是否是64位BAR
if (lower & BAR_MEM_TYPE_MASK) >> 1 != 0x2:
return None
return (upper << 32) | (lower & ~0xF)
5.2 PCIe设备复位与重枚举
有时需要复位设备以重新配置BAR空间：
def reset_pcie_device(self):
"""通过PCI配置空间复位设备"""
# 获取PCI控制寄存器偏移量(标准头类型0的0x3E处)
os.lseek(self.config_fd, 0x3E, os.SEEK_SET)
pci_cmd = struct.unpack('H', os.read(self.config_fd, 2))[0]
# 设置复位位(bit15)
pci_cmd |= (1 << 15)
os.lseek(self.config_fd, 0x3E, os.SEEK_SET)
os.write(self.config_fd, struct.pack('H', pci_cmd))
# 短暂延迟等待复位完成
time.sleep(0.1)
# 清除复位位
pci_cmd &= ~(1 << 15)
os.lseek(self.config_fd, 0x3E, os.SEEK_SET)
os.write(self.config_fd, struct.pack('H', pci_cmd))
5.3 常见问题排查指南
遇到问题时，可以按以下步骤排查：
-
权限问题 ：
- 确保以root权限运行脚本
- 检查
/sys/bus/pci/devices/.../config文件权限
-
设备未识别 ：
- 确认设备已正确插入PCIe插槽
- 检查
lspci输出是否包含目标设备
-
BAR读取异常 ：
- 某些设备可能需要先使能MMIO
- 检查内核是否保留了对该区域的访问权限
-
大小计算错误 ：
- 确保在计算前正确保存和恢复BAR原始值
- 对于64位BAR，需要同时处理两个寄存器
6. 安全注意事项与最佳实践
直接操作PCIe配置空间属于底层硬件访问，需要特别注意：
- 系统稳定性 ：错误的配置空间写入可能导致系统崩溃
- 并发访问 ：避免与其他驱动或工具同时访问同一设备
- 恢复机制 ：脚本应包含错误处理和状态恢复逻辑
- 日志记录 ：详细记录所有配置空间修改操作
推荐的安全实践包括：
def safe_bar_operation(self, bar_num, operation):
"""带错误恢复的BAR操作"""
original = self.read_bar_register(bar_num)
try:
result = operation(bar_num)
self.write_bar_register(bar_num, original)
return result
except Exception as e:
self.write_bar_register(bar_num, original)
raise e
在实际项目中，我发现最实用的技巧是将BAR探测功能封装成可重用的类库，配合设备树信息自动生成内存映射表。对于频繁更换PCIe设备卡的开发环境，这种自动化工具能显著提高工作效率。
更多推荐
9
0- 0
已为社区贡献11条内容
所有评论(0)