source: mainline/uspace/drv/nic/ar9271/ath_usb.c@ e0a5d4c

/*
 * Copyright (c) 2015 Jan Kolarik
 * Copyright (c) 2018 Ondrej Hlavaty
 * 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.
 */

/** @file ath_usb.c
 *
 * Implementation of Atheros USB wifi device functions.
 *
 */

#include <usb/dev/pipes.h>
#include <usb/debug.h>
#include <stdlib.h>
#include <errno.h>
#include "ath_usb.h"

static errno_t ath_usb_send_ctrl_message(ath_t *, void *, size_t);
static errno_t ath_usb_read_ctrl_message(ath_t *, void *, size_t, size_t *);
static errno_t ath_usb_send_data_message(ath_t *, void *, size_t);
static errno_t ath_usb_read_data_message(ath_t *, void *, size_t, size_t *);

static ath_ops_t ath_usb_ops = {
        .send_ctrl_message = ath_usb_send_ctrl_message,
        .read_ctrl_message = ath_usb_read_ctrl_message,
        .send_data_message = ath_usb_send_data_message,
        .read_data_message = ath_usb_read_data_message
};

/** Initialize Atheros WiFi USB device.
 *
 * @param ath Generic Atheros WiFi device structure.
 * @param usb_device  Connected USB device.
 *
 * @return EOK if succeed, error code otherwise.
 *
 */
errno_t ath_usb_init(ath_t *ath, usb_device_t *usb_device, const usb_endpoint_description_t **endpoints)
{
        ath_usb_t *ath_usb = malloc(sizeof(ath_usb_t));
        if (!ath_usb) {
                usb_log_error("Failed to allocate memory for ath usb device "
                    "structure.\n");
                return ENOMEM;
        }
        
        ath_usb->usb_device = usb_device;
        
        int rc;

#define _MAP_EP(target, ep_no) do {\
        usb_endpoint_mapping_t *epm = usb_device_get_mapped_ep_desc(usb_device, endpoints[ep_no]);\
        if (!epm || !epm->present) {\
                usb_log_error("Failed to map endpoint: " #ep_no ".");\
                rc = ENOENT;\
                goto err_ath_usb;\
        }\
        target = &epm->pipe;\
        } while (0);

        _MAP_EP(ath_usb->output_data_pipe, 0);
        _MAP_EP(ath_usb->input_data_pipe, 1);
        _MAP_EP(ath_usb->input_ctrl_pipe, 2);
        _MAP_EP(ath_usb->output_ctrl_pipe, 3);

#undef _MAP_EP
        
        ath->ctrl_response_length = 64;
        ath->data_response_length = 512;
        
        ath->specific_data = ath_usb;
        ath->ops = &ath_usb_ops;
        
        return EOK;
err_ath_usb:
        free(ath_usb);
        return rc;
}

/** Send control message.
 *
 * @param ath         Generic Atheros WiFi device structure.
 * @param buffer      Buffer with data to send.
 * @param buffer_size Buffer size.
 *
 * @return EOK if succeed, error code otherwise.
 *
 */
static errno_t ath_usb_send_ctrl_message(ath_t *ath, void *buffer,
    size_t buffer_size)
{
        ath_usb_t *ath_usb = (ath_usb_t *) ath->specific_data;
        return usb_pipe_write(ath_usb->output_ctrl_pipe, buffer, buffer_size);
}

/** Read control message.
 *
 * @param ath              Generic Atheros WiFi device structure.
 * @param buffer           Buffer with data to send.
 * @param buffer_size      Buffer size.
 * @param transferred_size Real size of read data.
 *
 * @return EOK if succeed, error code otherwise.
 *
 */
static errno_t ath_usb_read_ctrl_message(ath_t *ath, void *buffer,
    size_t buffer_size, size_t *transferred_size)
{
        ath_usb_t *ath_usb = (ath_usb_t *) ath->specific_data;
        return usb_pipe_read(ath_usb->input_ctrl_pipe, buffer, buffer_size, transferred_size);
}

/** Send data message.
 *
 * @param ath         Generic Atheros WiFi device structure. 
 * @param buffer      Buffer with data to send.
 * @param buffer_size Buffer size.
 *
 * @return EOK if succeed, error code otherwise.
 *
 */
static errno_t ath_usb_send_data_message(ath_t *ath, void *buffer,
    size_t buffer_size)
{
        size_t complete_buffer_size = buffer_size +
            sizeof(ath_usb_data_header_t);
        void *complete_buffer = malloc(complete_buffer_size);
        memcpy(complete_buffer + sizeof(ath_usb_data_header_t),
            buffer, buffer_size);
        
        ath_usb_data_header_t *data_header =
            (ath_usb_data_header_t *) complete_buffer;
        data_header->length = host2uint16_t_le(buffer_size);
        data_header->tag = host2uint16_t_le(TX_TAG);
        
        ath_usb_t *ath_usb = (ath_usb_t *) ath->specific_data;
        const errno_t ret_val = usb_pipe_write(ath_usb->output_data_pipe,
            complete_buffer, complete_buffer_size);
        
        free(complete_buffer);
        
        return ret_val;
}

/** Read data message.
 *
 * @param ath              Generic Atheros WiFi device structure.
 * @param buffer           Buffer with data to send.
 * @param buffer_size      Buffer size.
 * @param transferred_size Real size of read data.
 *
 * @return EOK if succeed, error code otherwise.
 *
 */
static errno_t ath_usb_read_data_message(ath_t *ath, void *buffer,
    size_t buffer_size, size_t *transferred_size)
{
        ath_usb_t *ath_usb = (ath_usb_t *) ath->specific_data;
        return usb_pipe_read(ath_usb->input_data_pipe, buffer, buffer_size, transferred_size);
}