diff --git a/lspci.c b/lspci.c new file mode 100644 index 0000000..cd3189c --- /dev/null +++ b/lspci.c @@ -0,0 +1,177 @@ +/* ++ rdesktop: A Remote Desktop Protocol client. ++ Support for the Matrox "lspci" channel ++ Copyright (C) 2005 Matrox Graphics Inc. ++ ++ This program is free software; you can redistribute it and/or modify ++ it under the terms of the GNU General Public License as published by ++ the Free Software Foundation; either version 2 of the License, or ++ (at your option) any later version. ++ ++ This program is distributed in the hope that it will be useful, ++ but WITHOUT ANY WARRANTY; without even the implied warranty of ++ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ++ GNU General Public License for more details. ++ ++ You should have received a copy of the GNU General Public License ++ along with this program; if not, write to the Free Software ++ Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. ++*/ + +#include "rdesktop.h" +#include +#include + +static VCHANNEL *lspci_channel; + +typedef struct _pci_device +{ + uint16 klass; + uint16 vendor; + uint16 device; + uint16 subvendor; + uint16 subdevice; + uint8 revision; + uint8 progif; +} pci_device; + +static pci_device current_device; + +static void lspci_send(const char *output); + + +/* Handle one line of output from the lspci subprocess */ +static BOOL +handle_child_line(const char *line, void *data) +{ + const char *val; + char buf[1024]; + + if (str_startswith(line, "Class:")) + { + val = line + sizeof("Class:"); + /* Skip whitespace and second Class: occurance */ + val += strspn(val, " \t") + sizeof("Class"); + current_device.klass = strtol(val, NULL, 16); + } + else if (str_startswith(line, "Vendor:")) + { + val = line + sizeof("Vendor:"); + current_device.vendor = strtol(val, NULL, 16); + } + else if (str_startswith(line, "Device:")) + { + val = line + sizeof("Device:"); + /* Sigh, there are *two* lines tagged as Device:. We + are not interested in the domain/bus/slot/func */ + if (!strchr(val, ':')) + current_device.device = strtol(val, NULL, 16); + } + else if (str_startswith(line, "SVendor:")) + { + val = line + sizeof("SVendor:"); + current_device.subvendor = strtol(val, NULL, 16); + } + else if (str_startswith(line, "SDevice:")) + { + val = line + sizeof("SDevice:"); + current_device.subdevice = strtol(val, NULL, 16); + } + else if (str_startswith(line, "Rev:")) + { + val = line + sizeof("Rev:"); + current_device.revision = strtol(val, NULL, 16); + } + else if (str_startswith(line, "ProgIf:")) + { + val = line + sizeof("ProgIf:"); + current_device.progif = strtol(val, NULL, 16); + } + else if (strspn(line, " \t") == strlen(line)) + { + /* Blank line. Send collected information over channel */ + snprintf(buf, sizeof(buf), "%04x,%04x,%04x,%04x,%04x,%02x,%02x\n", + current_device.klass, current_device.vendor, + current_device.device, current_device.subvendor, + current_device.subdevice, current_device.revision, current_device.progif); + lspci_send(buf); + memset(¤t_device, 0, sizeof(current_device)); + } + else + { + warning("lspci: Unrecoqnized line '%s'\n", line); + } + return True; +} + + +/* Process one line of input from virtual channel */ +static BOOL +lspci_process_line(const char *line, void *data) +{ + char *lspci_command[5] = { "lspci", "-m", "-n", "-v", NULL }; + + if (!strcmp(line, "LSPCI")) + { + memset(¤t_device, 0, sizeof(current_device)); + subprocess(lspci_command, handle_child_line, NULL); + /* Send single dot to indicate end of enumeration */ + lspci_send(".\n"); + } + else + { + error("lspci protocol error: Invalid line '%s'\n", line); + } + return True; +} + + +/* Process new data from the virtual channel */ +static void +lspci_process(STREAM s) +{ + unsigned int pkglen; + static char *rest = NULL; + char *buf; + + pkglen = s->end - s->p; + /* str_handle_lines requires null terminated strings */ + buf = xmalloc(pkglen + 1); + STRNCPY(buf, (char *) s->p, pkglen + 1); +#if 0 + printf("lspci recv:\n"); + hexdump(s->p, pkglen); +#endif + + str_handle_lines(buf, &rest, lspci_process_line, NULL); + xfree(buf); +} + +/* Initialize this module: Register the lspci channel */ +BOOL +lspci_init(void) +{ + lspci_channel = + channel_register("lspci", CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP, + lspci_process); + return (lspci_channel != NULL); +} + +/* Send data to channel */ +static void +lspci_send(const char *output) +{ + STREAM s; + size_t len; + + len = strlen(output); + s = channel_init(lspci_channel, len); + out_uint8p(s, output, len) s_mark_end(s); + +#if 0 + printf("lspci send:\n"); + hexdump(s->channel_hdr + 8, s->end - s->channel_hdr - 8); +#endif + + channel_send(s, lspci_channel); +}