request.c

Go to the documentation of this file.

/*
 * Copyright (c) 2011 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/dev/request.h>
#include <errno.h>
#include <assert.h>
#include <usb/debug.h>

#define MAX_DATA_LENGTH ((size_t)(0xFFFF))

int usb_control_request_set(usb_pipe_t *pipe,
    usb_request_type_t request_type, usb_request_recipient_t recipient,
    uint8_t request,
    uint16_t value, uint16_t index,
    void *data, size_t data_size)
{
        if (pipe == NULL) {
                return EBADMEM;
        }

        if (data_size > MAX_DATA_LENGTH) {
                return ERANGE;
        }

        if ((data_size > 0) && (data == NULL)) {
                return EBADMEM;
        }

        /*
         * TODO: check that @p request_type and @p recipient are
         * within ranges.
         */

        usb_device_request_setup_packet_t setup_packet;
        setup_packet.request_type = (request_type << 5) | recipient;
        setup_packet.request = request;
        setup_packet.value = value;
        setup_packet.index = index;
        setup_packet.length = (uint16_t) data_size;

        int rc = usb_pipe_control_write(pipe,
            &setup_packet, sizeof(setup_packet),
            data, data_size);

        return rc;
}

int usb_control_request_get(usb_pipe_t *pipe,
    usb_request_type_t request_type, usb_request_recipient_t recipient,
    uint8_t request,
    uint16_t value, uint16_t index,
    void *data, size_t data_size, size_t *actual_data_size)
{
        if (pipe == NULL) {
                return EBADMEM;
        }

        if (data_size > MAX_DATA_LENGTH) {
                return ERANGE;
        }

        if ((data_size > 0) && (data == NULL)) {
                return EBADMEM;
        }

        /*
         * TODO: check that @p request_type and @p recipient are
         * within ranges.
         */

        usb_device_request_setup_packet_t setup_packet;
        setup_packet.request_type = 128 | (request_type << 5) | recipient;
        setup_packet.request = request;
        setup_packet.value = value;
        setup_packet.index = index;
        setup_packet.length = (uint16_t) data_size;

        int rc = usb_pipe_control_read(pipe,
            &setup_packet, sizeof(setup_packet),
            data, data_size, actual_data_size);

        return rc;
}

int usb_request_get_status(usb_pipe_t *pipe,
    usb_request_recipient_t recipient, uint16_t index,
    uint16_t *status)
{
        if ((recipient == USB_REQUEST_RECIPIENT_DEVICE) && (index != 0)) {
                return EINVAL;
        }

        if (status == NULL) {
                return EBADMEM;
        }

        uint16_t status_usb_endianess;
        size_t data_transfered_size;
        int rc = usb_control_request_get(pipe, USB_REQUEST_TYPE_STANDARD,
            recipient, USB_DEVREQ_GET_STATUS, 0, uint16_host2usb(index),
            &status_usb_endianess, 2, &data_transfered_size);
        if (rc != EOK) {
                return rc;
        }
        if (data_transfered_size != 2) {
                return ELIMIT;
        }

        *status = uint16_usb2host(status_usb_endianess);

        return EOK;
}

int usb_request_clear_feature(usb_pipe_t *pipe,
    usb_request_type_t request_type, usb_request_recipient_t recipient,
    uint16_t feature_selector, uint16_t index)
{
        if (request_type == USB_REQUEST_TYPE_STANDARD) {
                if ((recipient == USB_REQUEST_RECIPIENT_DEVICE)
                    && (index != 0)) {
                        return EINVAL;
                }
        }

        int rc = usb_control_request_set(pipe, request_type, recipient,
            USB_DEVREQ_CLEAR_FEATURE,
            uint16_host2usb(feature_selector), uint16_host2usb(index),
            NULL, 0);

        return rc;
}

int usb_request_set_feature(usb_pipe_t *pipe,
    usb_request_type_t request_type, usb_request_recipient_t recipient,
    uint16_t feature_selector, uint16_t index)
{
        if (request_type == USB_REQUEST_TYPE_STANDARD) {
                if ((recipient == USB_REQUEST_RECIPIENT_DEVICE)
                    && (index != 0)) {
                        return EINVAL;
                }
        }

        int rc = usb_control_request_set(pipe, request_type, recipient,
            USB_DEVREQ_SET_FEATURE,
            uint16_host2usb(feature_selector), uint16_host2usb(index),
            NULL, 0);

        return rc;
}

int usb_request_set_address(usb_pipe_t *pipe,
    usb_address_t new_address)
{
        if ((new_address < 0) || (new_address >= USB11_ADDRESS_MAX)) {
                return EINVAL;
        }

        uint16_t addr = uint16_host2usb((uint16_t)new_address);

        int rc = usb_control_request_set(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE,
            USB_DEVREQ_SET_ADDRESS,
            addr, 0,
            NULL, 0);

        if (rc != EOK) {
                return rc;
        }

        assert(pipe->wire != NULL);
        /* TODO: prevent other from accessing wire now. */
        pipe->wire->address = new_address;

        return EOK;
}

int usb_request_get_descriptor(usb_pipe_t *pipe,
    usb_request_type_t request_type, usb_request_recipient_t recipient,
    uint8_t descriptor_type, uint8_t descriptor_index,
    uint16_t language,
    void *buffer, size_t size, size_t *actual_size)
{
        if (buffer == NULL) {
                return EBADMEM;
        }
        if (size == 0) {
                return EINVAL;
        }

        uint16_t wValue = descriptor_index | (descriptor_type << 8);

        return usb_control_request_get(pipe,
            request_type, recipient,
            USB_DEVREQ_GET_DESCRIPTOR,
            wValue, language,
            buffer, size, actual_size);
}

int usb_request_get_descriptor_alloc(usb_pipe_t * pipe,
    usb_request_type_t request_type, usb_request_recipient_t recipient,
    uint8_t descriptor_type, uint8_t descriptor_index,
    uint16_t language,
    void **buffer_ptr, size_t *buffer_size)
{
        if (buffer_ptr == NULL) {
                return EBADMEM;
        }

        int rc;

        /*
         * Get only first byte to retrieve descriptor length.
         */
        uint8_t tmp_buffer[1];
        size_t bytes_transfered;
        rc = usb_request_get_descriptor(pipe, request_type, recipient,
            descriptor_type, descriptor_index, language,
            &tmp_buffer, 1, &bytes_transfered);
        if (rc != EOK) {
                return rc;
        }
        if (bytes_transfered != 1) {
                /* FIXME: some better error code? */
                return ESTALL;
        }

        size_t size = tmp_buffer[0];
        if (size == 0) {
                /* FIXME: some better error code? */
                return ESTALL;
        }

        /*
         * Allocate buffer and get the descriptor again.
         */
        void *buffer = malloc(size);
        if (buffer == NULL) {
                return ENOMEM;
        }

        rc = usb_request_get_descriptor(pipe, request_type, recipient,
            descriptor_type, descriptor_index, language,
            buffer, size, &bytes_transfered);
        if (rc != EOK) {
                free(buffer);
                return rc;
        }
        if (bytes_transfered != size) {
                free(buffer);
                /* FIXME: some better error code? */
                return ESTALL;
        }

        *buffer_ptr = buffer;
        if (buffer_size != NULL) {
                *buffer_size = size;
        }

        return EOK;
}

int usb_request_get_device_descriptor(usb_pipe_t *pipe,
    usb_standard_device_descriptor_t *descriptor)
{
        if (descriptor == NULL) {
                return EBADMEM;
        }

        size_t actually_transferred = 0;
        usb_standard_device_descriptor_t descriptor_tmp;
        int rc = usb_request_get_descriptor(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE, 
            USB_DESCTYPE_DEVICE, 0, 0,
            &descriptor_tmp, sizeof(descriptor_tmp),
            &actually_transferred);

        if (rc != EOK) {
                return rc;
        }

        /* Verify that all data has been transferred. */
        if (actually_transferred < sizeof(descriptor_tmp)) {
                return ELIMIT;
        }

        /* Everything is okay, copy the descriptor. */
        memcpy(descriptor, &descriptor_tmp,
            sizeof(descriptor_tmp));

        return EOK;
}

int usb_request_get_bare_configuration_descriptor(usb_pipe_t *pipe,
    int index, usb_standard_configuration_descriptor_t *descriptor)
{
        if (descriptor == NULL) {
                return EBADMEM;
        }

        if ((index < 0) || (index > 0xFF)) {
                return ERANGE;
        }

        size_t actually_transferred = 0;
        usb_standard_configuration_descriptor_t descriptor_tmp;
        int rc = usb_request_get_descriptor(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE,
            USB_DESCTYPE_CONFIGURATION, index, 0,
            &descriptor_tmp, sizeof(descriptor_tmp),
            &actually_transferred);
        if (rc != EOK) {
                return rc;
        }

        /* Verify that all data has been transferred. */
        if (actually_transferred < sizeof(descriptor_tmp)) {
                return ELIMIT;
        }

        /* Everything is okay, copy the descriptor. */
        memcpy(descriptor, &descriptor_tmp,
            sizeof(descriptor_tmp));

        return EOK;
}

int usb_request_get_full_configuration_descriptor(usb_pipe_t *pipe,
    int index, void *descriptor, size_t descriptor_size, size_t *actual_size)
{
        if ((index < 0) || (index > 0xFF)) {
                return ERANGE;
        }

        return usb_request_get_descriptor(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE,
            USB_DESCTYPE_CONFIGURATION, index, 0,
            descriptor, descriptor_size, actual_size);
}

int usb_request_get_full_configuration_descriptor_alloc(
    usb_pipe_t *pipe, int index,
    void **descriptor_ptr, size_t *descriptor_size)
{
        int rc;

        if (descriptor_ptr == NULL) {
                return EBADMEM;
        }

        usb_standard_configuration_descriptor_t bare_config;
        rc = usb_request_get_bare_configuration_descriptor(pipe, index,
            &bare_config);
        if (rc != EOK) {
                return rc;
        }
        if (bare_config.descriptor_type != USB_DESCTYPE_CONFIGURATION) {
                return ENOENT;
        }
        if (bare_config.total_length < sizeof(bare_config)) {
                return ELIMIT;
        }

        void *buffer = malloc(bare_config.total_length);
        if (buffer == NULL) {
                return ENOMEM;
        }

        size_t transferred = 0;
        rc = usb_request_get_full_configuration_descriptor(pipe, index,
            buffer, bare_config.total_length, &transferred);
        if (rc != EOK) {
                free(buffer);
                return rc;
        }

        if (transferred != bare_config.total_length) {
                free(buffer);
                return ELIMIT;
        }

        /* Everything looks okay, copy the pointers. */

        *descriptor_ptr = buffer;

        if (descriptor_size != NULL) {
                *descriptor_size = bare_config.total_length;
        }

        return EOK;
}

int usb_request_set_descriptor(usb_pipe_t *pipe,
    usb_request_type_t request_type, usb_request_recipient_t recipient,
    uint8_t descriptor_type, uint8_t descriptor_index,
    uint16_t language,
    void *buffer, size_t size)
{
        if (buffer == NULL) {
                return EBADMEM;
        }
        if (size == 0) {
                return EINVAL;
        }

        /* FIXME: proper endianness. */
        uint16_t wValue = descriptor_index | (descriptor_type << 8);

        return usb_control_request_set(pipe,
            request_type, recipient,
            USB_DEVREQ_SET_DESCRIPTOR,
            wValue, language,
            buffer, size);
}

int usb_request_get_configuration(usb_pipe_t *pipe,
    uint8_t *configuration_value)
{
        uint8_t value;
        size_t actual_size;

        int rc = usb_control_request_get(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE,
            USB_DEVREQ_GET_CONFIGURATION,
            0, 0,
            &value, 1, &actual_size);

        if (rc != EOK) {
                return rc;
        }
        if (actual_size != 1) {
                return ELIMIT;
        }

        if (configuration_value != NULL) {
                *configuration_value = value;
        }

        return EOK;
}

int usb_request_set_configuration(usb_pipe_t *pipe,
    uint8_t configuration_value)
{
        uint16_t config_value
            = uint16_host2usb((uint16_t) configuration_value);

        return usb_control_request_set(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE,
            USB_DEVREQ_SET_CONFIGURATION, config_value, 0,
            NULL, 0);
}

int usb_request_get_interface(usb_pipe_t *pipe,
    uint8_t interface_index, uint8_t *alternate_setting)
{
        uint8_t value;
        size_t actual_size;

        int rc = usb_control_request_get(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_INTERFACE,
            USB_DEVREQ_GET_INTERFACE,
            0, uint16_host2usb((uint16_t) interface_index),
            &value, 1, &actual_size);

        if (rc != EOK) {
                return rc;
        }
        if (actual_size != 1) {
                return ELIMIT;
        }

        if (alternate_setting != NULL) {
                *alternate_setting = value;
        }

        return EOK;
}

int usb_request_set_interface(usb_pipe_t *pipe,
    uint8_t interface_index, uint8_t alternate_setting)
{
        return usb_control_request_set(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_INTERFACE,
            USB_DEVREQ_SET_INTERFACE,
            uint16_host2usb((uint16_t) alternate_setting),
            uint16_host2usb((uint16_t) interface_index),
            NULL, 0);
}

int usb_request_get_supported_languages(usb_pipe_t *pipe,
    l18_win_locales_t **languages_ptr, size_t *languages_count)
{
        int rc;

        if (languages_ptr == NULL) {
                return EBADMEM;
        }
        if (languages_count == NULL) {
                return EBADMEM;
        }

        uint8_t *string_descriptor = NULL;
        size_t string_descriptor_size = 0;
        rc = usb_request_get_descriptor_alloc(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE,
            USB_DESCTYPE_STRING, 0, 0,
            (void **) &string_descriptor, &string_descriptor_size);
        if (rc != EOK) {
                return rc;
        }
        if (string_descriptor_size <= 2) {
                free(string_descriptor);
                return EEMPTY;
        }
        /* Subtract first 2 bytes (length and descriptor type). */
        string_descriptor_size -= 2;

        /* Odd number of bytes - descriptor is broken? */
        if ((string_descriptor_size % 2) != 0) {
                /* FIXME: shall we return with error or silently ignore? */
                free(string_descriptor);
                return ESTALL;
        }

        size_t langs_count = string_descriptor_size / 2;
        l18_win_locales_t *langs
            = malloc(sizeof(l18_win_locales_t) * langs_count);
        if (langs == NULL) {
                free(string_descriptor);
                return ENOMEM;
        }

        size_t i;
        for (i = 0; i < langs_count; i++) {
                /* Language code from the descriptor is in USB endianness. */
                /* FIXME: is this really correct? */
                uint16_t lang_code = (string_descriptor[2 + 2 * i + 1] << 8)
                    + string_descriptor[2 + 2 * i];
                langs[i] = uint16_usb2host(lang_code);
        }

        free(string_descriptor);

        *languages_ptr = langs;
        *languages_count =langs_count;

        return EOK;
}

int usb_request_get_string(usb_pipe_t *pipe,
    size_t index, l18_win_locales_t lang, char **string_ptr)
{
        if (string_ptr == NULL) {
                return EBADMEM;
        }
        /*
         * Index is actually one byte value and zero index is used
         * to retrieve list of supported languages.
         */
        if ((index < 1) || (index > 0xFF)) {
                return ERANGE;
        }
        /* Language is actually two byte value. */
        if (lang > 0xFFFF) {
                return ERANGE;
        }

        int rc;

        /* Prepare dynamically allocated variables. */
        uint8_t *string = NULL;
        wchar_t *string_chars = NULL;

        /* Get the actual descriptor. */
        size_t string_size;
        rc = usb_request_get_descriptor_alloc(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_DEVICE,
            USB_DESCTYPE_STRING, index, uint16_host2usb(lang),
            (void **) &string, &string_size);
        if (rc != EOK) {
                goto leave;
        }

        if (string_size <= 2) {
                rc =  EEMPTY;
                goto leave;
        }
        /* Subtract first 2 bytes (length and descriptor type). */
        string_size -= 2;

        /* Odd number of bytes - descriptor is broken? */
        if ((string_size % 2) != 0) {
                /* FIXME: shall we return with error or silently ignore? */
                rc = ESTALL;
                goto leave;
        }

        size_t string_char_count = string_size / 2;
        string_chars = malloc(sizeof(wchar_t) * (string_char_count + 1));
        if (string_chars == NULL) {
                rc = ENOMEM;
                goto leave;
        }

        /*
         * Build a wide string.
         * And do not forget to set NULL terminator (string descriptors
         * do not have them).
         */
        size_t i;
        for (i = 0; i < string_char_count; i++) {
                uint16_t uni_char = (string[2 + 2 * i + 1] << 8)
                    + string[2 + 2 * i];
                string_chars[i] = uni_char;
        }
        string_chars[string_char_count] = 0;


        /* Convert to normal string. */
        char *str = wstr_to_astr(string_chars);
        if (str == NULL) {
                rc = ENOMEM;
                goto leave;
        }

        *string_ptr = str;
        rc = EOK;

leave:
        if (string != NULL) {
                free(string);
        }
        if (string_chars != NULL) {
                free(string_chars);
        }

        return rc;
}

int usb_request_clear_endpoint_halt(usb_pipe_t *pipe, uint16_t ep_index)
{
        return usb_request_clear_feature(pipe,
            USB_REQUEST_TYPE_STANDARD, USB_REQUEST_RECIPIENT_ENDPOINT,
            uint16_host2usb(USB_FEATURE_SELECTOR_ENDPOINT_HALT),
            uint16_host2usb(ep_index));
}

int usb_pipe_clear_halt(usb_pipe_t *ctrl_pipe, usb_pipe_t *target_pipe)
{
        if ((ctrl_pipe == NULL) || (target_pipe == NULL)) {
                return EINVAL;
        }
        return usb_request_clear_endpoint_halt(ctrl_pipe,
            target_pipe->endpoint_no);
}

int usb_request_get_endpoint_status(usb_pipe_t *ctrl_pipe, usb_pipe_t *pipe,
    uint16_t *status)
{
        uint16_t status_tmp;
        uint16_t pipe_index = (uint16_t) pipe->endpoint_no;
        int rc = usb_request_get_status(ctrl_pipe,
            USB_REQUEST_RECIPIENT_ENDPOINT, uint16_host2usb(pipe_index),
            &status_tmp);
        if (rc != EOK) {
                return rc;
        }

        if (status != NULL) {
                *status = uint16_usb2host(status_tmp);
        }

        return EOK;
}

---