source: mainline/uspace/drv/usbkbd/main.c@ e84d65a

/*
 * Copyright (c) 2010 Vojtech Horky
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#include <usb/usbdrv.h>
#include <driver.h>
#include <ipc/driver.h>
#include <ipc/kbd.h>
#include <io/keycode.h>
#include <io/console.h>
#include <errno.h>
#include <fibril.h>
#include <usb/classes/hid.h>
#include <usb/classes/hidparser.h>
#include <usb/devreq.h>
#include <usb/descriptor.h>

#define BUFFER_SIZE 32
#define NAME "usbkbd"

#define GUESSED_POLL_ENDPOINT 1

static void default_connection_handler(device_t *, ipc_callid_t, ipc_call_t *);
static device_ops_t keyboard_ops = {
        .default_handler = default_connection_handler
};

static int console_callback_phone = -1;

/** Default handler for IPC methods not handled by DDF.
 *
 * @param dev Device handling the call.
 * @param icallid Call id.
 * @param icall Call data.
 */
void default_connection_handler(device_t *dev,
    ipc_callid_t icallid, ipc_call_t *icall)
{
        sysarg_t method = IPC_GET_IMETHOD(*icall);

        if (method == IPC_M_CONNECT_TO_ME) {
                int callback = IPC_GET_ARG5(*icall);

                if (console_callback_phone != -1) {
                        ipc_answer_0(icallid, ELIMIT);
                        return;
                }

                console_callback_phone = callback;
                ipc_answer_0(icallid, EOK);
                return;
        }

        ipc_answer_0(icallid, EINVAL);
}

static void send_key(int key, int type, wchar_t c) {
        async_msg_4(console_callback_phone, KBD_EVENT, type, key,
            KM_NUM_LOCK, c);
}

static void send_alnum(int key, wchar_t c) {
        printf(NAME ": sending key '%lc' to console\n", (wint_t) c);
        send_key(key, KEY_PRESS, c);
        send_key(key, KEY_RELEASE, c);
}

/*
 * Callbacks for parser
 */
static void usbkbd_process_keycodes(const uint16_t *key_codes, size_t count,
                                    void *arg)
{

}

/*
 * Kbd functions
 */
static int usbkbd_parse_descriptors(usb_hid_dev_kbd_t *kbd_dev,
                                    const uint8_t *data, size_t size)
{
//      const uint8_t *pos = data;
        
//      // get the configuration descriptor (should be first)
//      if (*pos != sizeof(usb_standard_configuration_descriptor_t)
//          || *(pos + 1) != USB_DESCTYPE_CONFIGURATION) {
//              fprintf(stderr, "Wrong format of configuration descriptor");
//              return EINVAL;
//      }
        
//      usb_standard_configuration_descriptor_t config_descriptor;
//      memcpy(&config_descriptor, pos, 
//          sizeof(usb_standard_configuration_descriptor_t));
//      pos += sizeof(usb_standard_configuration_descriptor_t);
        
//      // parse other descriptors
//      while (pos - data < size) {
//              //uint8_t desc_size = *pos;
//              uint8_t desc_type = *(pos + 1);
//              switch (desc_type) {
//              case USB_DESCTYPE_INTERFACE:
//                      break;
//              case USB_DESCTYPE_ENDPOINT:
//                      break;
//              case USB_DESCTYPE_HID:
//                      break;
//              case USB_DESCTYPE_HID_REPORT:
//                      break;
//              case USB_DESCTYPE_HID_PHYSICAL:
//                      break;
//              }
//      }
        
        return EOK;
}

static int usbkbd_get_descriptors(usb_hid_dev_kbd_t *kbd_dev)
{
        // get the first configuration descriptor (TODO: or some other??)
        usb_standard_configuration_descriptor_t config_desc;
        
        int rc = usb_drv_req_get_bare_configuration_descriptor(
            kbd_dev->device->parent_phone, kbd_dev->address, 0, &config_desc);
        
        if (rc != EOK) {
                return rc;
        }
        
        // prepare space for all underlying descriptors
        uint8_t *descriptors = (uint8_t *)malloc(config_desc.total_length);
        if (descriptors == NULL) {
                return ENOMEM;
        }
        
        size_t transferred = 0;
        // get full configuration descriptor
        rc = usb_drv_req_get_full_configuration_descriptor(
            kbd_dev->device->parent_phone, kbd_dev->address, 0, descriptors,
            config_desc.total_length, &transferred);
        
        if (rc != EOK) {
                return rc;
        }
        if (transferred != config_desc.total_length) {
                return ELIMIT;
        }
        
        rc = usbkbd_parse_descriptors(kbd_dev, descriptors, transferred);
        free(descriptors);
        
        return rc;
}

static usb_hid_dev_kbd_t *usbkbd_init_device(device_t *dev)
{
        usb_hid_dev_kbd_t *kbd_dev = (usb_hid_dev_kbd_t *)malloc(
                        sizeof(usb_hid_dev_kbd_t));

        if (kbd_dev == NULL) {
                fprintf(stderr, NAME ": No memory!\n");
                return NULL;
        }

        kbd_dev->device = dev;

        // get phone to my HC and save it as my parent's phone
        // TODO: maybe not a good idea if DDF will use parent_phone
        kbd_dev->device->parent_phone = usb_drv_hc_connect(dev, 0);

        kbd_dev->address = usb_drv_get_my_address(dev->parent_phone,
            dev);

        // doesn't matter now that we have no address
//      if (kbd_dev->address < 0) {
//              fprintf(stderr, NAME ": No device address!\n");
//              free(kbd_dev);
//              return NULL;
//      }

        // default endpoint
        kbd_dev->poll_endpoint = GUESSED_POLL_ENDPOINT;
        
        /*
         * will need all descriptors:
         * 1) choose one configuration from configuration descriptors 
         *    (set it to the device)
         * 2) set endpoints from endpoint descriptors
         */

        // TODO: get descriptors
        usbkbd_get_descriptors(kbd_dev);
        // TODO: parse descriptors and save endpoints

        return kbd_dev;
}

static void usbkbd_process_interrupt_in(usb_hid_dev_kbd_t *kbd_dev,
                                        uint8_t *buffer, size_t actual_size)
{
        /*
         * here, the parser will be called, probably with some callbacks
         * now only take last 6 bytes and process, i.e. send to kbd
         */

        usb_hid_report_in_callbacks_t *callbacks =
            (usb_hid_report_in_callbacks_t *)malloc(
                sizeof(usb_hid_report_in_callbacks_t));
        callbacks->keyboard = usbkbd_process_keycodes;

        if (console_callback_phone != -1) {
                static size_t counter = 0;
                counter++;
                if (counter > 3) {
                        counter = 0;
                        send_alnum(KC_A, L'a');
                }
        }

        usb_hid_parse_report(kbd_dev->parser, buffer, actual_size, callbacks, 
            NULL);
}

static void usbkbd_poll_keyboard(usb_hid_dev_kbd_t *kbd_dev)
{
        int rc;
        usb_handle_t handle;
        uint8_t buffer[BUFFER_SIZE];
        size_t actual_size;
        //usb_endpoint_t poll_endpoint = 1;

//      usb_address_t my_address = usb_drv_get_my_address(dev->parent_phone,
//          dev);
//      if (my_address < 0) {
//              return;
//      }

        usb_target_t poll_target = {
                .address = kbd_dev->address,
                .endpoint = kbd_dev->poll_endpoint
        };

        while (true) {
                async_usleep(1000 * 1000);
                rc = usb_drv_async_interrupt_in(kbd_dev->device->parent_phone,
                    poll_target, buffer, BUFFER_SIZE, &actual_size, &handle);

                if (rc != EOK) {
                        continue;
                }

                rc = usb_drv_async_wait_for(handle);
                if (rc != EOK) {
                        continue;
                }

                /*
                 * If the keyboard answered with NAK, it returned no data.
                 * This implies that no change happened since last query.
                 */
                if (actual_size == 0) {
                        continue;
                }

                /*
                 * TODO: Process pressed keys.
                 */
                usbkbd_process_interrupt_in(kbd_dev, buffer, actual_size);
        }

        // not reached
        assert(0);
}

static int usbkbd_fibril_device(void *arg)
{
        printf("!!! USB device fibril\n");

        if (arg == NULL) {
                printf("No device!\n");
                return -1;
        }

        device_t *dev = (device_t *)arg;

        // initialize device (get and process descriptors, get address, etc.)
        usb_hid_dev_kbd_t *kbd_dev = usbkbd_init_device(dev);

        usbkbd_poll_keyboard(kbd_dev);

        return EOK;
}

static int usbkbd_add_device(device_t *dev)
{
        /* For now, fail immediately. */
        //return ENOTSUP;

        /*
         * When everything is okay, connect to "our" HC.
         *
         * Not supported yet, skip..
         */
//      int phone = usb_drv_hc_connect(dev, 0);
//      if (phone < 0) {
//              /*
//               * Connecting to HC failed, roll-back and announce
//               * failure.
//               */
//              return phone;
//      }

//      dev->parent_phone = phone;

        /*
         * Create new fibril for handling this keyboard
         */
        fid_t fid = fibril_create(usbkbd_fibril_device, dev);
        if (fid == 0) {
                printf("%s: failed to start fibril for HID device\n", NAME);
                return ENOMEM;
        }
        fibril_add_ready(fid);

        dev->ops = &keyboard_ops;

        add_device_to_class(dev, "keyboard");

        /*
         * Hurrah, device is initialized.
         */
        return EOK;
}

static driver_ops_t kbd_driver_ops = {
        .add_device = usbkbd_add_device,
};

static driver_t kbd_driver = {
        .name = NAME,
        .driver_ops = &kbd_driver_ops
};

int main(int argc, char *argv[])
{
        return driver_main(&kbd_driver);
}