12 changed files with 270 additions and 29 deletions
Binary file not shown.
@ -1,17 +1,205 @@ |
|||
#include "pcie.h" |
|||
#include "../debug/logging.h" |
|||
#include "../debug/dump.h" |
|||
#include "../debug/assert.h" |
|||
#include "../memory/kalloc.h" |
|||
#include "../memory/paging.h" |
|||
#include "../klib/string.h" |
|||
|
|||
|
|||
struct PCIE_configuration_space { |
|||
volatile uint16_t vendorID; |
|||
volatile uint16_t deviceID; |
|||
uint16_t unused0; |
|||
uint16_t unused1; |
|||
volatile uint32_t class_code_revID; |
|||
volatile uint32_t infos; |
|||
volatile uint64_t bar[3]; |
|||
volatile uint32_t cardbud_cis_ptr; |
|||
volatile uint16_t subsystemID; |
|||
volatile uint16_t subsystem_vendorID; |
|||
volatile uint32_t expansion_base; |
|||
volatile uint8_t capabilities; |
|||
uint8_t reserved0[3]; |
|||
uint32_t reserved1; |
|||
volatile uint8_t interrupt_line; |
|||
volatile uint8_t interrupt_pin; |
|||
uint16_t reserved2[2]; |
|||
}; |
|||
|
|||
// higher bus number: could be 0xff
|
|||
// if there is only one bus group
|
|||
static size_t max_bus = 0; |
|||
|
|||
struct PCIE_Descriptor pcie_descriptor = {0}; |
|||
// 0 | group | bus | 0 | slot | 0 | func |
|
|||
// | 2 | 8 | 1 | 3 | 1 | 3 |
|
|||
// 16 8
|
|||
|
|||
static struct PCIE_segment_group_descriptor* get_segment_group(unsigned bus) { |
|||
struct PCIE_segment_group_descriptor* group = &pcie_descriptor.array[0]; |
|||
|
|||
unsigned group_number = 0; |
|||
unsigned start_bus, end_bus; |
|||
|
|||
// go through all segment groups
|
|||
// if the bus is not in the range,
|
|||
// it is not the right segment
|
|||
|
|||
while((start_bus = group->start_bus) > bus |
|||
|| (end_bus = group->end_bus) < bus) { |
|||
|
|||
assert(group_number != pcie_descriptor.size); |
|||
|
|||
group_number++; |
|||
group++; |
|||
} |
|||
return group; |
|||
} |
|||
|
|||
static struct PCIE_configuration_space* __attribute__((pure)) get_config_space_base( |
|||
unsigned bus, |
|||
unsigned device, |
|||
unsigned func |
|||
) { |
|||
struct PCIE_segment_group_descriptor* group_desc = get_segment_group(bus); |
|||
|
|||
return group_desc->address + ( |
|||
(bus - group_desc->start_bus) << 20 |
|||
| device << 15 |
|||
| func << 12); |
|||
} |
|||
|
|||
|
|||
static int __attribute__((pure)) check_function(unsigned bus, unsigned device, unsigned func) { |
|||
struct PCIE_configuration_space* config_space = get_config_space_base(bus,device,func); |
|||
|
|||
return config_space->vendorID != 0xffff; |
|||
} |
|||
|
|||
|
|||
struct pcie_device_descriptor { |
|||
uint16_t bus, device, function; |
|||
struct PCIE_configuration_space* config_space; |
|||
}; |
|||
|
|||
// array of all installed devices' functions
|
|||
static struct pcie_device_descriptor* installed_devices = NULL; |
|||
static size_t n_installed_devices = 0; |
|||
|
|||
|
|||
/**
|
|||
* constructs the installed_devices array |
|||
*/ |
|||
static void scan_devices(void) { |
|||
|
|||
// first create a linked list
|
|||
// as we dont know how many devices are present
|
|||
struct device_desc_node { |
|||
struct pcie_device_descriptor e; |
|||
struct device_desc_node* next; |
|||
}; |
|||
|
|||
// head of the list
|
|||
struct device_desc_node ghost_node = {.next = NULL}; |
|||
struct device_desc_node* current = &ghost_node; |
|||
// count the number of devices
|
|||
|
|||
void insert( |
|||
unsigned bus, |
|||
unsigned device, |
|||
unsigned func, |
|||
struct PCIE_configuration_space* config_space |
|||
) { |
|||
current->next = kmalloc( |
|||
sizeof(struct device_desc_node) |
|||
); |
|||
|
|||
|
|||
current = current->next; |
|||
current->next = NULL; |
|||
|
|||
// fill the new node
|
|||
current->e.bus = bus; |
|||
current->e.device = device; |
|||
current->e.function = func; |
|||
current->e.config_space = config_space; |
|||
|
|||
n_installed_devices++; |
|||
} |
|||
|
|||
|
|||
for(unsigned bus = 0; bus < max_bus; bus++) { |
|||
for(unsigned device = 0; device < 32; device++) { |
|||
|
|||
// if the first function doesn't exist, so does
|
|||
// the device
|
|||
if(!check_function(bus,device, 0)) |
|||
continue; |
|||
insert(bus, device, 0, get_config_space_base(bus,device,0)); |
|||
|
|||
for(unsigned func = 0; func < 8; func++) |
|||
if(check_function(bus,device, func)) |
|||
insert(bus, device, func, get_config_space_base(bus,device,func)); |
|||
|
|||
} |
|||
} |
|||
|
|||
// now create the final array
|
|||
installed_devices = kmalloc( |
|||
n_installed_devices |
|||
* sizeof(struct pcie_device_descriptor)); |
|||
|
|||
|
|||
// now fill it and free the devices structure
|
|||
struct pcie_device_descriptor* ptr = installed_devices; |
|||
|
|||
for(struct device_desc_node* device = ghost_node.next; |
|||
device != NULL; |
|||
) { |
|||
|
|||
memcpy(ptr++, &device->e, sizeof(struct pcie_device_descriptor)); |
|||
|
|||
|
|||
struct device_desc_node* next = device->next; |
|||
kfree(device); |
|||
device = next; |
|||
} |
|||
|
|||
} |
|||
|
|||
/**
|
|||
* during the init, |
|||
* we identity map the pcie configuration spaces |
|||
* |
|||
*/ |
|||
void pcie_init(void) { |
|||
|
|||
klog_debug("init pcie..."); |
|||
klog_debug("%d", pcie_descriptor.size); |
|||
|
|||
dump( |
|||
pcie_descriptor.array, |
|||
pcie_descriptor.size * sizeof(struct PCIE_segment_group_descriptor), |
|||
sizeof(struct PCIE_segment_group_descriptor), |
|||
DUMP_HEX8 |
|||
); |
|||
|
|||
// calculate the highest bus number
|
|||
for(unsigned i = 0; i < pcie_descriptor.size; i++) { |
|||
// map the corresponding pages
|
|||
if(pcie_descriptor.array[i].end_bus > max_bus) |
|||
max_bus = pcie_descriptor.array[i].end_bus; |
|||
|
|||
klog_debug("map %lx", pcie_descriptor.array[i].address); |
|||
// identity map the corresponding pages
|
|||
map_pages( |
|||
pcie_descriptor.array[i].address, // phys
|
|||
pcie_descriptor.array[i].address, // virt
|
|||
256 * // busses
|
|||
32 * // devices
|
|||
8 , // functions
|
|||
PRESENT_ENTRY | PCD | PL_XD |
|||
// no cache, execute disable
|
|||
); |
|||
|
|||
|
|||
} |
|||
asm("hlt"); |
|||
|
|||
scan_devices(); |
|||
|
|||
klog_info("found %u PCI Express devices", n_installed_devices); |
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
|
|||
|
|||
/**
|
|||
* PCIE drivers interfaces: |
|||
* |
|||
* provide void init(void* config_space_base) |
|||
* |
|||
* functions to call: |
|||
* register_irq(unsigned) |
|||
* unregister_irq(unsigned) |
|||
* |
|||
* int register_timer(void callback(void), unsigned period) |
|||
* void unregister_timer(int timer_id) |
|||
* |
|||
*/ |
|||
|
|||
|
|||
|
|||
void |
|||
/**
|
|||
* return 0 if the irq was successfully installed |
|||
*/ |
|||
int register_irq(uint8_t irq_number, ); |
|||
Loading…
Reference in new issue