DragonOS-Community/DragonOS
1
1
Fork 0
mirror of https://github.com/DragonOS-Community/DragonOS.git synced 2025-06-09 15:26:47 +00:00
Code Issues Releases Wiki Activity

🆕 递归枚举pci总线上的设备

Browse Source
This commit is contained in:
fslongjin 2022-03-21 23:44:17 +08:00
parent 5e9bd1d44e
commit db3749ece4
3 changed files with 276 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
.vscode/settings.json vendored
View File

@ -21,7 +21,71 @@
"mouse.h": "c", "mouse.h": "c",
"keyboard.h": "c", "keyboard.h": "c",
"apic.h": "c", "apic.h": "c",
"ps2_keyboard.h": "c" "ps2_keyboard.h": "c",
"algorithm": "c",
"array": "c",
"atomic": "c",
"*.tcc": "c",
"bitset": "c",
"cassert": "c",
"cctype": "c",
"cerrno": "c",
"cfloat": "c",
"chrono": "c",
"climits": "c",
"clocale": "c",
"cmath": "c",
"codecvt": "c",
"condition_variable": "c",
"cstdarg": "c",
"cstddef": "c",
"cstdint": "c",
"cstdio": "c",
"cstdlib": "c",
"cstring": "c",
"ctime": "c",
"cwchar": "c",
"cwctype": "c",
"deque": "c",
"exception": "c",
"forward_list": "c",
"functional": "c",
"iterator": "c",
"list": "c",
"map": "c",
"memory": "c",
"memory_resource": "c",
"numeric": "c",
"optional": "c",
"random": "c",
"ratio": "c",
"set": "c",
"string": "c",
"string_view": "c",
"system_error": "c",
"tuple": "c",
"type_traits": "c",
"unordered_map": "c",
"utility": "c",
"vector": "c",
"fstream": "c",
"initializer_list": "c",
"ios": "c",
"iosfwd": "c",
"istream": "c",
"limits": "c",
"locale": "c",
"mutex": "c",
"new": "c",
"ostream": "c",
"queue": "c",
"sstream": "c",
"stdexcept": "c",
"streambuf": "c",
"thread": "c",
"cinttypes": "c",
"cstdbool": "c",
"typeinfo": "c"
}, },
"C_Cpp.errorSquiggles": "Enabled" "C_Cpp.errorSquiggles": "Enabled"
} }

218
kernel/driver/pci/pci.c
View File

@ -1,5 +1,28 @@
#include "pci.h" #include "pci.h"
#include "../../common/kprint.h" #include "../../common/kprint.h"
#include "../../mm/slab.h"
static uint count_device_list = 0;
/**
* @brief 将设备信息结构体加到链表里面
*
*/
#define ADD_DEVICE_STRUCT_TO_LIST(ret) \
do \
{ \
if (count_device_list > 0) \
{ \
++count_device_list; \
list_add(pci_device_structure_list, &ret->header->list); \
} \
else \
{ \
++count_device_list; \
list_init(&ret->header->list); \
pci_device_structure_list = &ret->header->list; \
} \
} while (0)
/** /**
* @brief 从pci配置空间读取信息 * @brief 从pci配置空间读取信息
@ -170,68 +193,209 @@ static void pci_read_pci_to_cardbus_bridge_header(struct pci_device_structure_pc
* @param bus 总线号 * @param bus 总线号
* @param slot 插槽号 * @param slot 插槽号
* @param func 功能号 * @param func 功能号
* @param add_to_list 添加到链表
* @return 返回的header * @return 返回的header
*/ */
void *pci_read_header(int *type, uchar bus, uchar slot, uchar func) void *pci_read_header(int *type, uchar bus, uchar slot, uchar func, bool add_to_list)
{ {
struct pci_device_structure_header_t common_header; struct pci_device_structure_header_t *common_header = (struct pci_device_structure_header_t *)kmalloc(sizeof(struct pci_device_structure_header_t), 0);
uint32_t tmp32; uint32_t tmp32;
// 先读取公共header // 先读取公共header
tmp32 = pci_read_config(bus, slot, func, 0x0); tmp32 = pci_read_config(bus, slot, func, 0x0);
common_header.Vendor_ID = tmp32 & 0xffff; common_header->Vendor_ID = tmp32 & 0xffff;
common_header.Device_ID = (tmp32 >> 16) & 0xffff; common_header->Device_ID = (tmp32 >> 16) & 0xffff;
tmp32 = pci_read_config(bus, slot, func, 0x4); tmp32 = pci_read_config(bus, slot, func, 0x4);
common_header.Command = tmp32 & 0xffff; common_header->Command = tmp32 & 0xffff;
common_header.Status = (tmp32 >> 16) & 0xffff; common_header->Status = (tmp32 >> 16) & 0xffff;
tmp32 = pci_read_config(bus, slot, func, 0x8); tmp32 = pci_read_config(bus, slot, func, 0x8);
common_header.RevisionID = tmp32 & 0xff; common_header->RevisionID = tmp32 & 0xff;
common_header.ProgIF = (tmp32 >> 8) & 0xff; common_header->ProgIF = (tmp32 >> 8) & 0xff;
common_header.SubClass = (tmp32 >> 16) & 0xff; common_header->SubClass = (tmp32 >> 16) & 0xff;
common_header.Class_code = (tmp32 >> 24) & 0xff; common_header->Class_code = (tmp32 >> 24) & 0xff;
tmp32 = pci_read_config(bus, slot, func, 0xc); tmp32 = pci_read_config(bus, slot, func, 0xc);
common_header.CacheLineSize = tmp32 & 0xff; common_header->CacheLineSize = tmp32 & 0xff;
common_header.LatencyTimer = (tmp32 >> 8) & 0xff; common_header->LatencyTimer = (tmp32 >> 8) & 0xff;
common_header.HeaderType = (tmp32 >> 16) & 0xff; common_header->HeaderType = (tmp32 >> 16) & 0xff;
common_header.BIST = (tmp32 >> 24) & 0xff; common_header->BIST = (tmp32 >> 24) & 0xff;
// 根据公共头部，判断该结构所属的类型 // 根据公共头部，判断该结构所属的类型
switch (common_header.Vendor_ID) switch (common_header->Vendor_ID)
{ {
case 0xFFFF: // 设备不可用 case 0xFFFF: // 设备不可用
*type = E_DEVICE_INVALID; *type = E_DEVICE_INVALID;
kfree(common_header);
return NULL; return NULL;
break; break;
case 0x0: // general device case 0x0: // general device
struct pci_device_structure_general_device_t ret; struct pci_device_structure_general_device_t *ret = (struct pci_device_structure_general_device_t *)kmalloc(sizeof(struct pci_device_structure_general_device_t), 0);
ret.header = common_header;
pci_read_general_device_header(&ret, bus, slot, func); ret->header = common_header;
pci_read_general_device_header(ret, bus, slot, func);
if (add_to_list)
ADD_DEVICE_STRUCT_TO_LIST(ret);
*type = 0x0; *type = 0x0;
return &ret; return ret;
break; break;
case 0x1: case 0x1:
struct pci_device_structure_pci_to_pci_bridge_t ret; struct pci_device_structure_pci_to_pci_bridge_t *ret = (struct pci_device_structure_pci_to_pci_bridge_t *)kmalloc(sizeof(struct pci_device_structure_pci_to_pci_bridge_t), 0);
ret.header = common_header; ret->header = common_header;
pci_read_pci_to_pci_bridge_header(ret, bus, slot, func);
if (add_to_list)
ADD_DEVICE_STRUCT_TO_LIST(ret);
*type = 0x1; *type = 0x1;
pci_read_pci_to_pci_bridge_header(&ret, bus, slot, func); return ret;
return &ret;
break; break;
case 0x2: case 0x2:
struct pci_device_structure_pci_to_cardbus_bridge_t ret; struct pci_device_structure_pci_to_cardbus_bridge_t *ret = (struct pci_device_structure_pci_to_cardbus_bridge_t *)kmalloc(sizeof(struct pci_device_structure_pci_to_cardbus_bridge_t), 0);
ret.header = common_header; ret->header = common_header;
pci_read_pci_to_cardbus_bridge_header(ret, bus, slot, func);
if (add_to_list)
ADD_DEVICE_STRUCT_TO_LIST(ret);
*type = 0x2; *type = 0x2;
pci_read_pci_to_cardbus_bridge_header(&ret, bus, slot, func); return ret;
return &ret;
break; break;
default: // 错误的头类型 这里不应该被执行 default: // 错误的头类型 这里不应该被执行
kBUG("PCI->pci_read_header(): Invalid header type."); kBUG("PCI->pci_read_header(): Invalid header type.");
*type = E_WRONG_HEADER_TYPE; *type = E_WRONG_HEADER_TYPE;
kfree(common_header);
return NULL; return NULL;
break; break;
} }
} }
static void pci_checkFunction(uint8_t bus, uint8_t device, uint8_t function)
{
int header_type;
struct pci_device_structure_header_t *header;
void *raw_header = pci_read_header(&header_type, 0, 0, 0, true);
if (header_type == E_WRONG_HEADER_TYPE)
{
kBUG("pci_checkFunction(): wrong header type!");
// 此处内存已经在read header函数里面释放，不用重复释放
return;
}
header = ((struct pci_device_structure_general_device_t *)raw_header)->header;
if ((header->Class_code == 0x6) && (header->SubClass == 0x4))
{
uint8_t SecondaryBus = ((struct pci_device_structure_pci_to_pci_bridge_t *)raw_header)->Secondary_Bus_Number;
pci_checkBus(SecondaryBus);
}
kfree(header);
kfree(raw_header);
}
static void pci_checkDevice(uint8_t bus, uint8_t device)
{
int header_type;
void *raw_header;
struct pci_device_structure_header_t *header;
raw_header = pci_read_header(&header_type, bus, device, 0, false);
if (header_type == E_WRONG_HEADER_TYPE)
{
// 此处内存已经在read header函数里面释放，不用重复释放
kBUG("pci_checkDevice(): wrong header type!");
return;
}
header = ((struct pci_device_structure_general_device_t *)raw_header)->header;
uint16_t vendorID = header->Vendor_ID;
if (vendorID == 0xffff) // 设备不存在
{
kfree(header);
kfree(raw_header);
return;
}
pci_checkFunction(bus, device, 0);
header_type = header->HeaderType;
if ((header_type & 0x80) != 0)
{
// 这是一个多function的设备，因此查询剩余的function
for (uint8_t func = 1; func < 8; ++func)
{
struct pci_device_structure_header_t *tmp_header;
void *tmp_raw_header;
tmp_raw_header = pci_read_header(&header_type, bus, device, func, false);
tmp_header = ((struct pci_device_structure_general_device_t *)tmp_raw_header)->header;
pci_checkFunction(bus, device, func);
// 释放内存
kfree(tmp_header);
kfree(tmp_raw_header);
}
}
kfree(header);
kfree(raw_header);
}
static void pci_checkBus(uint8_t bus)
{
for (uint8_t device = 0; device < 32; ++device)
pci_checkDevice(bus, device);
}
/**
* @brief 扫描所有pci总线上的所有设备
*
*/
void pci_checkAllBuses()
{
int header_type;
void *raw_header;
struct pci_device_structure_header_t *header;
raw_header = pci_read_header(&header_type, 0, 0, 0, false);
if (header_type == E_WRONG_HEADER_TYPE)
{
kBUG("pci_checkAllBuses(): wrong header type!");
// 此处内存已经在read header函数里面释放，不用重复释放
return;
}
header = ((struct pci_device_structure_general_device_t *)raw_header)->header;
header_type = header->HeaderType;
if ((header_type & 0x80) == 0) // Single pci host controller
{
pci_checkBus(0);
}
else
{
// Multiple PCI host controller
// 那么总线0，设备0，功能1则是总线1的pci主机控制器，以此类推
void *tmp_raw_header;
struct pci_device_structure_header_t *tmp_header;
for (uint8_t func = 0; func < 8; ++func)
{
tmp_raw_header = pci_read_header(&header_type, 0, 0, func, false);
tmp_header = ((struct pci_device_structure_general_device_t *)tmp_raw_header)->header;
if (header->Vendor_ID != 0xffff) // @todo 这里的判断条件可能有点问题
{
kfree(tmp_header);
kfree(tmp_raw_header);
break;
}
pci_checkBus(func);
kfree(tmp_header);
kfree(tmp_raw_header);
}
}
kfree(header);
kfree(raw_header);
}
void pci_init() void pci_init()
{ {

18
kernel/driver/pci/pci.h
View File

@ -9,6 +9,9 @@
#define E_DEVICE_INVALID -1 #define E_DEVICE_INVALID -1
#define E_WRONG_HEADER_TYPE -2 #define E_WRONG_HEADER_TYPE -2
// pci设备结构信息的链表
struct List * pci_device_structure_list = NULL;
/** /**
* @brief 初始化pci驱动 * @brief 初始化pci驱动
* *
@ -18,6 +21,7 @@ void pci_init();
// pci设备结构的通用标题字段 // pci设备结构的通用标题字段
struct pci_device_structure_header_t struct pci_device_structure_header_t
{ {
struct List list;
uint16_t Vendor_ID; // 供应商ID 0xffff是一个无效值，在读取访问不存在的设备的配置空间寄存器时返回 uint16_t Vendor_ID; // 供应商ID 0xffff是一个无效值，在读取访问不存在的设备的配置空间寄存器时返回
uint16_t Device_ID; // 设备ID，标志特定设备 uint16_t Device_ID; // 设备ID，标志特定设备
@ -45,7 +49,7 @@ struct pci_device_structure_header_t
*/ */
struct pci_device_structure_general_device_t struct pci_device_structure_general_device_t
{ {
struct pci_device_structure_header_t header; struct pci_device_structure_header_t *header;
uint32_t BAR0; uint32_t BAR0;
uint32_t BAR1; uint32_t BAR1;
uint32_t BAR2; uint32_t BAR2;
@ -77,7 +81,7 @@ struct pci_device_structure_general_device_t
*/ */
struct pci_device_structure_pci_to_pci_bridge_t struct pci_device_structure_pci_to_pci_bridge_t
{ {
struct pci_device_structure_header_t header; struct pci_device_structure_header_t *header;
uint32_t BAR0; uint32_t BAR0;
uint32_t BAR1; uint32_t BAR1;
@ -122,7 +126,7 @@ struct pci_device_structure_pci_to_pci_bridge_t
*/ */
struct pci_device_structure_pci_to_cardbus_bridge_t struct pci_device_structure_pci_to_cardbus_bridge_t
{ {
struct pci_device_structure_header_t header; struct pci_device_structure_header_t *header;
uint32_t CardBus_Socket_ExCa_base_address; uint32_t CardBus_Socket_ExCa_base_address;
@ -176,4 +180,10 @@ uint pci_read_config(uchar bus, uchar slot, uchar func, uchar offset);
* @param func 功能号 * @param func 功能号
* @return 返回的header的指针 * @return 返回的header的指针
*/ */
void* pci_read_header(int *type, uchar bus, uchar slot, uchar func); void* pci_read_header(int *type, uchar bus, uchar slot, uchar func, bool add_to_list);
/**
* @brief 扫描所有pci总线上的所有设备
*
*/
void pci_checkAllBuses();