prgKbd-keyboard/Driver/prgKbd_debug/main.cpp

#include <QtCore/QCoreApplication>
#include <iostream>
#include <QStringList>

#include <usbpp.h>


// DEVICE-RELATED DEFINES
#define USBDEV_SHARED_VENDOR    0x16C0  /* VOTI */
#define USBDEV_SHARED_PRODUCT   0x03e8  /* Obdev's free shared PID */
/* Use obdev's generic shared VID/PID pair and follow the rules outlined
 * in firmware/usbdrv/USBID-License.txt.
 */

// COMMANDS:
#define CMD_PING 0
#define CMD_SWVERSION 1
#define CMD_EEWRITE 2
#define CMD_EEREAD 3
#define CMD_EEOPEN 4 // Can be used if errors occour while writing to the device.
#define CMD_RESET 5 // WDT RESET

// OTHER DEFS:
#define IDENTIFY_NAME 0
#define IDENTIFY_VERSION 1

struct usb_dev_handle *dev;


// functions
int sendMessage(int command, int var1, int var2, char *buffer);
bool sendMessage(int command, int var1, int var2);

//shell
void help (QStringList *cmd);



using namespace std;
int main() //int argc, char *argv[]
{
    // USB-Variables
        struct usb_bus *bus;
        struct usb_device *devs;
        char iMan[256]; // Manufacturer name
        char iProd[256]; // Product name
    // Some other vars
        bool break_loop = false;

    // Init
        usb_init(); // Init USB
        usb_set_debug(3); // Debug on
        usb_find_busses();
        usb_find_devices();

    // Scan for device
        for (bus = usb_busses; bus; bus = bus->next) { // Cycle through all busses
            for (devs = bus->devices; devs; devs = devs->next) { // Cycle through all devices on that bus.
                cout << "Checking device: \n\tBUS:" << bus->dirname << "\\" << devs->filename << "\n\tIDV:" << devs->descriptor.idVendor << "\n\tIDP:" << devs->descriptor.idProduct << endl;
                if (devs->descriptor.idVendor == USBDEV_SHARED_VENDOR && devs->descriptor.idProduct == USBDEV_SHARED_PRODUCT) {
                    cout << "FOUND DEVICE! " << endl;
                    break_loop = true; // Tell the second loop to exit as well.
                    break; // Exit loop
                }
            }
            if (break_loop) break; // Break if dev was found
        }

   // Check whether a device was found
        if (!break_loop) { // If we didn't leave the loop on purpose the device wasn't found
            cout << "!!! ERROR: Device couldn't be found !!!" << endl;
            return 1;
        }

   // If the device was found we try to open it.
        dev = usb_open(devs); // Since we left the loop when the right device was active we may still use it now.
        if (dev) { // We were able to open it :)
            // query basic info
                 usb_get_string_simple(dev,devs->descriptor.iManufacturer,iMan, sizeof(iMan));
                 usb_get_string_simple(dev,devs->descriptor.iProduct,iProd, sizeof(iProd));
        }
        else {
            cout << "!!! ERROR: Device could not be opened !!!" << endl;
            return 1;
        }




   // Create the shell
        QStringList command;
        string temp;
        bool exit = false;
        while (!exit) {
            cout << iMan << "@" << iProd << "# " << ends;
            getline(cin, temp);
            command = QString::fromStdString(temp).split(" ");
            if (command.at(0) == "exit") exit = true;
            else if (command.at(0) == "reset") {
                cout << "Triggering a watchdog-reset..." << endl;
                sendMessage(CMD_RESET,0,0);
            }
            else if (command.at(0) == "raw") {
                if (command.count() < 2) help(&command);
                int p1 = 0, p2 = 0;
                if (command.count() >= 4) {
                    p1 = command.at(2).toInt();
                    p2 = command.at(3).toInt();
                }
                else if (command.count() >= 3) {
                    p1 = command.at(2).toInt();
                }
                char buf[16];
                int retVal = sendMessage(command.at(1).toInt(), p1, p2, buf);
                if (retVal > 0) {
                    buf[retVal] = 0;
                     cout << "Answer (" << QString::number(retVal).toStdString().data() << "B): " << buf << endl;
                }
                else if (retVal == 0) {
                    cout << "This command has no response." << endl;
                }
                else {
                    cout << "ERROR: " << retVal << endl;
                }
            }
            else if (command.at(0) == "rawnum") {
                if (command.count() < 2) help(&command);
                int p1 = 0, p2 = 0;
                if (command.count() >= 4) {
                    p1 = command.at(2).toInt();
                    p2 = command.at(3).toInt();
                }
                else if (command.count() >= 3) {
                    p1 = command.at(2).toInt();
                }
                char buf[16];
                int retVal = sendMessage(command.at(1).toInt(), p1, p2, buf);
                if (retVal > 0) {
                    buf[retVal] = 0;
                     cout << "Answer (" << QString::number(retVal).toStdString().data() << "B): ";
                     for (int y = 0; y < retVal; y++) {
                         cout << " 0x" << QString::number(buf[y], 16).toStdString().data() << "(" << QString::number((unsigned char) buf[y],10).toStdString().data() << ")";
                     }
                     cout << endl;
                }
                else if (retVal == 0) {
                    cout << "This command has no response." << endl;
                }
                else {
                    cout << "ERROR: " << retVal << endl;
                }
            }
            else if (command.at(0) == "ping") {
                if (command.count() != 2) help(&command);
                else {
                    char buf[1];
                    int retVal = sendMessage(CMD_PING, command.at(1).at(0).toAscii(),0, buf);
                    if (retVal > 0) {
                        buf[retVal] = 0;
                        cout << "PING returned (" << QString::number(retVal).toStdString().data() << "B): " << buf << endl;
                    }
                    else {
                        cout << "ERROR: Received no data!" << endl;
                    }
                }
            }
            else if (command.at(0) == "" || command.at(0) == " ");
            else if (command.at(0) == "help") {
                help(&command);
            }
            else {
                cout << command.at(0).toStdString().data() << ": Command not found!" << endl;
            }

            // Ping dev to mke sure it's still alive
            char buf[1];
            int8_t retVal = sendMessage(CMD_PING, command.at(0).at(0).toAscii(),0, buf); // instead of creating a random number we use the first letter of our alst command
            if (retVal <= 0) {
                cout << "ERROR: LOST CONNECTION!" << endl;
                exit = 1;
            }
        }

        cout << "Closing device..." << endl;
        usb_close(dev);

    return 0;
}

int sendMessage(int command, int var1, int var2, char *buffer) {
   return (usb_control_msg(dev, USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_ENDPOINT_IN, command, var1, var2, (char *)buffer, sizeof(buffer), 5000));
}
bool sendMessage(int command, int var1, int var2) {
   if (usb_control_msg(dev, USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_ENDPOINT_IN, command, var1, var2, NULL, 0, 5000) < 0) return false;
   return true;
}


// SHELL-RELATED functions
void help (QStringList *cmd) {
    char pos=1;
    if (cmd->count() == 1) pos = 0;
    if (cmd->at(pos) == "ping") {
        cout << "Syntax: ping [AnyChar]\nPing will simply return the character you entered." << endl;
    }
    else if (cmd->at(pos) == "raw") {
        cout << "Syntax: raw [command] ([Param1]) ([Param2])\nThis will send the raw data to the device. All parameters will be converted to numbers. (65->A)\nExample: raw\t42\t13\t37\n\t\t^cmd\t^param1\t^param2" << endl;
    }
    else {
        cout << cmd->at(pos).toStdString().data() << ": Help not found!" << endl;
    }
    return;
}