source: vbox/trunk/src/VBox/RDP/client-1.8.4/lspci.c@ 78352

/*  -*- c-basic-offset: 8 -*-
   rdesktop: A Remote Desktop Protocol client.
   Support for the Matrox "lspci" channel
   Copyright (C) 2005 Matrox Graphics Inc.
   Copyright 2018 Henrik Andersson <[email protected]> for Cendio AB

   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 3 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, see <http://www.gnu.org/licenses/>.
*/

/*
 * Oracle GPL Disclaimer: For the avoidance of doubt, except that if any license choice
 * other than GPL or LGPL is available it will apply instead, Oracle elects to use only
 * the General Public License version 2 (GPLv2) at this time for any software where
 * a choice of GPL license versions is made available with the language indicating
 * that GPLv2 or any later version may be used, or where a choice of which version
 * of the GPL is applied is otherwise unspecified.
 */

#include "rdesktop.h"
#include <sys/types.h>
#include <unistd.h>

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 RD_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(&current_device, 0, sizeof(current_device));
        }
        else
        {
                warning("lspci: Unrecoqnized line '%s'\n", line);
        }
        return True;
}


/* Process one line of input from virtual channel */
static RD_BOOL
lspci_process_line(const char *line, void *data)
{
        char *lspci_command[5] = { "lspci", "-m", "-n", "-v", NULL };

        if (!strcmp(line, "LSPCI"))
        {
                memset(&current_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;
        struct stream packet = *s;

        if (!s_check(s))
        {
                rdp_protocol_error("lspci_process(), stream is in unstable state", &packet);
        }

        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 */
RD_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);
}