GitHub
@@ -1,9 +1,9 @@
|
||||
.PHONY: all clean
|
||||
|
||||
CC = qcc # ntox86-gcc
|
||||
CC = ntox86-gcc
|
||||
CXX = g++
|
||||
LFLAGS = -Wall #-Vgcc_ntox86
|
||||
LIBS = #-l socket
|
||||
LIBS = -l ncurses #-l socket
|
||||
|
||||
BIN_PATH = ./bin
|
||||
SOURCE_PATH = ./src
|
||||
@@ -14,14 +14,10 @@ SRCS = $(SOURCE_PATH)/main.c
|
||||
|
||||
all:
|
||||
mkdir -p $(BIN_PATH)
|
||||
$(CXX) $(LFLAGS) -o $(TARGET) $(SRCS) $(LIBS)
|
||||
$(CC) $(LFLAGS) -o $(TARGET) $(SRCS) $(LIBS)
|
||||
|
||||
$(BIN_PATH):
|
||||
mkdir -p $(BIN_PATH)
|
||||
|
||||
clean:
|
||||
rm -f $(TARGET)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
+522
-1
@@ -1,7 +1,528 @@
|
||||
#include <time.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <hw/pci.h>
|
||||
#include <string.h>
|
||||
#include <ncurses.h>
|
||||
#include <inttypes.h>
|
||||
#include <hw/pci-full.h>
|
||||
|
||||
|
||||
#define MAX_BUS 0xff
|
||||
#define MAX_DEV 0x1f
|
||||
#define MAX_FUNC 0x07
|
||||
|
||||
#define MAX_KEY_COUNT 10
|
||||
|
||||
#define CBUTTON_EXIT KEY_F(10)
|
||||
#define CBUTTON_SELECT_UP KEY_UP
|
||||
#define CBUTTON_SELECT_DOWN KEY_DOWN
|
||||
#define CBUTTON_ABOUT_DEVICE '/'
|
||||
#define CBUTTON_ESCAPE 27
|
||||
|
||||
#define COLOR_PAIR_TEXT 1
|
||||
#define COLOR_PAIR_TEXT_INV 2
|
||||
|
||||
#define TO_DEVFUNC(dev, func) (((dev)<<3)+(func))
|
||||
|
||||
|
||||
typedef struct {
|
||||
int x;
|
||||
int y;
|
||||
} Point;
|
||||
|
||||
typedef struct {
|
||||
int height;
|
||||
int width;
|
||||
} Size;
|
||||
|
||||
typedef struct device_t {
|
||||
struct device_t *previous;
|
||||
struct device_t *next;
|
||||
|
||||
uint8_t bus;
|
||||
uint8_t device;
|
||||
uint8_t function;
|
||||
|
||||
uint8_t pci_index;
|
||||
|
||||
uint16_t vendor_id;
|
||||
uint16_t device_id;
|
||||
uint16_t command;
|
||||
uint16_t status;
|
||||
uint8_t revision;
|
||||
uint8_t interface;
|
||||
uint8_t subclass;
|
||||
uint8_t class;
|
||||
|
||||
uint16_t ss_vendor_id;
|
||||
uint16_t ss_device_id;
|
||||
|
||||
char *desc_vendor;
|
||||
char *desc_device;
|
||||
} Device;
|
||||
|
||||
typedef struct {
|
||||
uint8_t wdevices:1;
|
||||
uint8_t waboutd:1;
|
||||
} WindowFlags;
|
||||
|
||||
|
||||
void init(void);
|
||||
void quit(void);
|
||||
|
||||
void update(void);
|
||||
void draw(void);
|
||||
|
||||
void draw_wdevices(void);
|
||||
void draw_waboutd(void);
|
||||
void draw_device(Device *d, int y, int x);
|
||||
|
||||
void fill_devices(void);
|
||||
uint8_t get_pci_index(const struct pci_dev_info *dev_info);
|
||||
char *get_vendor_description(struct pci_dev_info *dev_info);
|
||||
char *get_device_description(struct pci_dev_info *dev_info);
|
||||
int devices_between(Device *device_1, Device *device_2);
|
||||
int devices_before(Device *device);
|
||||
int devices_after(Device *device);
|
||||
|
||||
void select_up(void);
|
||||
void select_down(void);
|
||||
void switch_waboutd(void);
|
||||
void hide_all_subwindows(void);
|
||||
|
||||
|
||||
|
||||
Size wdevice_size = {0, 0};
|
||||
Size waboutd_size = {0, 0};
|
||||
int use_color = 0;
|
||||
|
||||
|
||||
WINDOW *wdevices = NULL;
|
||||
WINDOW *waboutd = NULL;
|
||||
|
||||
WindowFlags visible_windows;
|
||||
|
||||
Device *first_device = NULL;
|
||||
Device *last_device = NULL;
|
||||
|
||||
Device *selected_device = NULL;
|
||||
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
puts("Hallo, Welt!");
|
||||
fill_devices();
|
||||
|
||||
init();
|
||||
|
||||
refresh();
|
||||
while (1) {
|
||||
draw();
|
||||
update();
|
||||
}
|
||||
|
||||
quit();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void init(void)
|
||||
{
|
||||
srand((unsigned int)time(NULL));
|
||||
initscr();
|
||||
cbreak();
|
||||
curs_set(FALSE);
|
||||
keypad(stdscr, TRUE);
|
||||
noecho();
|
||||
|
||||
hide_all_subwindows();
|
||||
|
||||
use_color = has_colors() == TRUE;
|
||||
|
||||
if (use_color) {
|
||||
start_color();
|
||||
init_pair(COLOR_PAIR_TEXT, COLOR_WHITE, COLOR_BLACK);
|
||||
init_pair(COLOR_PAIR_TEXT_INV, COLOR_BLACK, COLOR_WHITE);
|
||||
}
|
||||
|
||||
getmaxyx(stdscr, wdevice_size.height, wdevice_size.width);
|
||||
waboutd_size.height = min(wdevice_size.height, 15);
|
||||
waboutd_size.width = min(wdevice_size.width, 60);
|
||||
|
||||
Point waboutd_pos;
|
||||
waboutd_pos.y = (wdevice_size.height-waboutd_size.height)/2;
|
||||
waboutd_pos.x = (wdevice_size.width-waboutd_size.width)/2;
|
||||
|
||||
wdevices = newwin(wdevice_size.height, wdevice_size.width, 0, 0);
|
||||
waboutd = newwin(waboutd_size.height, waboutd_size.width,
|
||||
waboutd_pos.y, waboutd_pos.x);
|
||||
}
|
||||
|
||||
void update(void)
|
||||
{
|
||||
static int is_first = 1;
|
||||
if (is_first) {
|
||||
is_first = 0;
|
||||
return;
|
||||
}
|
||||
|
||||
switch (getch()) {
|
||||
case CBUTTON_EXIT:
|
||||
quit();
|
||||
break;
|
||||
case CBUTTON_SELECT_UP:
|
||||
select_up();
|
||||
break;
|
||||
case CBUTTON_SELECT_DOWN:
|
||||
select_down();
|
||||
break;
|
||||
case CBUTTON_ABOUT_DEVICE:
|
||||
switch_waboutd();
|
||||
break;
|
||||
case CBUTTON_ESCAPE:
|
||||
hide_all_subwindows();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void draw(void)
|
||||
{
|
||||
if (visible_windows.wdevices) {
|
||||
draw_wdevices();
|
||||
}
|
||||
if (visible_windows.waboutd) {
|
||||
draw_waboutd();
|
||||
}
|
||||
}
|
||||
|
||||
void draw_wdevices(void)
|
||||
{
|
||||
wclear(wdevices);
|
||||
box(wdevices, ACS_VLINE, ACS_HLINE);
|
||||
|
||||
int x = 2;
|
||||
int y = 0;
|
||||
|
||||
mvwprintw(wdevices, y++, x, " b d f vid did cmd stat rv c sc if ssv ssd");
|
||||
|
||||
Device *first_device_for_printing = first_device;
|
||||
while (
|
||||
devices_between(first_device_for_printing, selected_device) >
|
||||
wdevice_size.height-6
|
||||
) {
|
||||
first_device_for_printing = first_device_for_printing->next;
|
||||
}
|
||||
|
||||
if (first_device_for_printing != first_device) {
|
||||
int xx;
|
||||
for (xx = x; xx < wdevice_size.width-2; xx+=2) {
|
||||
mvwprintw(wdevices, y, xx, "/\\");
|
||||
}
|
||||
y++;
|
||||
}
|
||||
|
||||
Device *d;
|
||||
for (d = first_device_for_printing; d != NULL; d = d->next) {
|
||||
draw_device(d, y, x);
|
||||
y++;
|
||||
|
||||
if (y == wdevice_size.height-1) {
|
||||
if (d != last_device) {
|
||||
int xx;
|
||||
for (xx = x; xx < wdevice_size.width-2; xx+=2) {
|
||||
mvwprintw(wdevices, y-1, xx, "\\/");
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
mvwprintw(wdevices, wdevice_size.height-1, 2, "? - about");
|
||||
|
||||
touchwin(wdevices);
|
||||
wrefresh(wdevices);
|
||||
}
|
||||
|
||||
void draw_waboutd(void)
|
||||
{
|
||||
wclear(waboutd);
|
||||
box(waboutd, ACS_VLINE, ACS_HLINE);
|
||||
|
||||
int x = 2;
|
||||
int y = 0;
|
||||
|
||||
mvwprintw(waboutd, y++, x, "%04X %04X %d",
|
||||
selected_device->vendor_id,
|
||||
selected_device->device_id,
|
||||
selected_device->pci_index);
|
||||
|
||||
if (selected_device->desc_vendor) {
|
||||
mvwprintw(waboutd, y++, x, "Vendor: %s", selected_device->desc_vendor);
|
||||
}
|
||||
if (selected_device->desc_device) {
|
||||
mvwprintw(waboutd, y++, x, "Device: %s", selected_device->desc_device);
|
||||
}
|
||||
|
||||
touchwin(waboutd);
|
||||
wrefresh(waboutd);
|
||||
}
|
||||
|
||||
void draw_device(Device *d, int y, int x)
|
||||
{
|
||||
if (use_color && d == selected_device) {
|
||||
wattron(wdevices, COLOR_PAIR(COLOR_PAIR_TEXT_INV));
|
||||
}
|
||||
|
||||
mvwprintw(wdevices, y, x,
|
||||
"%2d %2d %d %04X %04X %04X %04X %02X %02X %02X %02X %04X %04X",
|
||||
d->bus, d->device, d->function, d->vendor_id, d->device_id,
|
||||
d->command, d->status, d->revision,
|
||||
d->class, d->subclass, d->interface,
|
||||
d->ss_vendor_id, d->ss_device_id
|
||||
);
|
||||
|
||||
if (use_color && d == selected_device) {
|
||||
wattroff(wdevices, COLOR_PAIR(COLOR_PAIR_TEXT_INV));
|
||||
}
|
||||
}
|
||||
|
||||
void quit(void)
|
||||
{
|
||||
endwin();
|
||||
exit(0);
|
||||
}
|
||||
|
||||
void fill_devices(void)
|
||||
{
|
||||
if (pci_attach(0) == -1) {
|
||||
fputs("Unable to attach PCI!", stderr);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
Device *previous_device = NULL;
|
||||
Device **next_device = &first_device;
|
||||
|
||||
unsigned short bus, dev, func;
|
||||
for(bus = 0; bus <= MAX_BUS; bus++) {
|
||||
for(dev = 0; dev <= MAX_DEV; dev++) {
|
||||
for(func = 0; func <= MAX_FUNC; func++) {
|
||||
struct pci_dev_info dev_info;
|
||||
memset(&dev_info, 0, sizeof(dev_info));
|
||||
|
||||
dev_info.BusNumber = bus;
|
||||
dev_info.DevFunc = TO_DEVFUNC(dev, func);
|
||||
|
||||
void *handleDevice = pci_attach_device(NULL, PCI_SEARCH_BUSDEV,
|
||||
0, &dev_info);
|
||||
|
||||
if (handleDevice != NULL) {
|
||||
*next_device = malloc(sizeof(Device));
|
||||
(*next_device)->previous = previous_device;
|
||||
(*next_device)->next = NULL;
|
||||
|
||||
(*next_device)->bus = bus;
|
||||
(*next_device)->device = dev;
|
||||
(*next_device)->function = func;
|
||||
|
||||
(*next_device)->pci_index = get_pci_index(&dev_info);
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x00, 1, 2,&(*next_device)->vendor_id
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get vendor_id");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x02, 1, 2, &(*next_device)->device_id
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get device_id");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x04, 1, 2, &(*next_device)->command
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get command");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x06, 1, 2, &(*next_device)->status
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get status");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x08, 1, 1, &(*next_device)->revision
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get revision");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x09, 1, 1, &(*next_device)->interface
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get interface");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x0a, 1, 1, &(*next_device)->subclass
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get subclasscode");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x0b, 1, 1, &(*next_device)->class
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get classcode");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x2C, 1, 2,&(*next_device)->ss_vendor_id
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get ss_vendor_id");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (pci_read_config(
|
||||
handleDevice, 0x2E, 1, 2, &(*next_device)->ss_device_id
|
||||
) != PCI_SUCCESS) {
|
||||
perror("Unable to get ss_device_id");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
pci_detach_device(handleDevice);
|
||||
|
||||
(*next_device)->desc_vendor = get_vendor_description(&dev_info);
|
||||
(*next_device)->desc_device = get_device_description(&dev_info);
|
||||
|
||||
previous_device = *next_device;
|
||||
last_device = previous_device;
|
||||
next_device = &(*next_device)->next;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
selected_device = first_device;
|
||||
}
|
||||
|
||||
char *get_vendor_description(struct pci_dev_info *dev_info)
|
||||
{
|
||||
int i;
|
||||
for (i = 0; i < PCI_VENTABLE_LEN; i++) {
|
||||
if (PciVenTable[i].VenId == dev_info->VendorId) {
|
||||
return PciVenTable[i].VenFull;
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char *get_device_description(struct pci_dev_info *dev_info)
|
||||
{
|
||||
int i;
|
||||
for (i = 0; i < PCI_DEVTABLE_LEN; i++) {
|
||||
if (PciDevTable[i].VenId == dev_info->VendorId &&
|
||||
PciDevTable[i].DevId == dev_info->DeviceId) {
|
||||
return PciDevTable[i].ChipDesc;
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
uint8_t get_pci_index(const struct pci_dev_info *dev_info)
|
||||
{
|
||||
unsigned int bus;
|
||||
unsigned int dev_func;
|
||||
|
||||
uint8_t i;
|
||||
for (i = 0; i < 0xFF; i++) {
|
||||
if (pci_find_device(
|
||||
dev_info->DeviceId, dev_info->VendorId, i, &bus, &dev_func
|
||||
) != PCI_SUCCESS) {
|
||||
break;
|
||||
}
|
||||
if (bus == dev_info->BusNumber && dev_func == dev_info->DevFunc) {
|
||||
return i;
|
||||
}
|
||||
}
|
||||
|
||||
return 0xFF;
|
||||
}
|
||||
|
||||
void select_up(void)
|
||||
{
|
||||
|
||||
if (selected_device != NULL && selected_device->previous != NULL) {
|
||||
selected_device = selected_device->previous;
|
||||
}
|
||||
}
|
||||
|
||||
void select_down(void)
|
||||
{
|
||||
if (selected_device != NULL && selected_device->next != NULL) {
|
||||
selected_device = selected_device->next;
|
||||
}
|
||||
}
|
||||
|
||||
void switch_waboutd(void)
|
||||
{
|
||||
visible_windows.waboutd = !visible_windows.waboutd;
|
||||
}
|
||||
|
||||
void hide_all_subwindows(void)
|
||||
{
|
||||
memset(&visible_windows, 0, sizeof(visible_windows));
|
||||
visible_windows.wdevices = 1;
|
||||
}
|
||||
|
||||
int devices_between(Device *device_1, Device *device_2)
|
||||
{
|
||||
if (device_1 == NULL || device_2 == NULL || device_1 == device_2) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
int i = 0;
|
||||
Device *d = device_1->next;
|
||||
while (d != NULL) {
|
||||
if (d == device_2) {
|
||||
return i;
|
||||
}
|
||||
i++;
|
||||
d = d->next;
|
||||
}
|
||||
|
||||
return devices_between(device_2, device_1);
|
||||
}
|
||||
|
||||
int devices_before(Device *device)
|
||||
{
|
||||
int i = 0;
|
||||
if (device != NULL) {
|
||||
Device *d;
|
||||
for (d = device->previous; d != NULL; d = d->previous) {
|
||||
i++;
|
||||
}
|
||||
}
|
||||
|
||||
return i;
|
||||
}
|
||||
|
||||
int devices_after(Device *device)
|
||||
{
|
||||
int i = 0;
|
||||
if (device != NULL) {
|
||||
Device *d;
|
||||
for (d = device->next; d != NULL; d = d->next) {
|
||||
i++;
|
||||
}
|
||||
}
|
||||
|
||||
return i;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user