lby/lk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
355b62b13a2a8ce8db007dc5193472c0d6cf57ab
lk/dev/usb/usb.c
Travis Geiselbrecht 699842fad2 [dev][usb] fix a few unchecked mallocs and add some comments 
A few routines had unchecked mallocs. One was unused, so remove it but
add fallback code for the rest of them.
2021-10-20 22:12:05 -07:00

377 lines
12 KiB
C
Raw Blame History

/*
* Copyright (c) 2008-2015 Travis Geiselbrecht
*
* Use of this source code is governed by a MIT-style
* license that can be found in the LICENSE file or at
* https://opensource.org/licenses/MIT
*/
#include <dev/usb.h>
#include <lk/debug.h>
#include <lk/trace.h>
#include <lk/err.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <lk/list.h>
#include <dev/usbc.h>
#include <lk/init.h>
/*
* USB device side support. Handles configuration and main high level host requests.
* USB device controller must pump the stack via usbc_callback() and implement the
* usbc function interface.
*/
/* TODO: make big endian safe */
#define LOCAL_TRACE 0
#define MAX_STRINGS 8
static struct {
bool active;
uint8_t active_config;
usb_config *config;
struct list_node cb_list;
usb_string strings[MAX_STRINGS];
} usb;
typedef struct {
struct list_node node;
usb_callback_t cb;
void *cookie;
} usb_callback_container_t;
static void usb_do_callbacks(usb_callback_op_t op, const union usb_callback_args *args);
static status_t append_desc_data(usb_descriptor *desc, const void *dat, size_t len) {
uint8_t *ptr = malloc(desc->len + len);
if (!ptr) {
return ERR_NO_MEMORY;
}
memcpy(ptr, desc->desc, desc->len);
memcpy(ptr + desc->len, dat, len);
/* free the old buffer if it wasn't marked static */
if ((desc->flags & USB_DESC_FLAG_STATIC) == 0)
free(desc->desc);
desc->flags &= ~USB_DESC_FLAG_STATIC;
desc->desc = ptr;
desc->len += len;
return NO_ERROR;
}
static uint8_t usb_get_current_iface_num(const usb_descriptor *desc) {
DEBUG_ASSERT(desc);
return ((uint8_t *)desc->desc)[4];
}
uint8_t usb_get_current_iface_num_highspeed(void) {
return usb_get_current_iface_num(&usb.config->highspeed.config);
}
uint8_t usb_get_current_iface_num_lowspeed(void) {
return usb_get_current_iface_num(&usb.config->lowspeed.config);
}
/* returns the interface number assigned or error */
static int usb_append_interface(usb_descriptor *desc, const uint8_t *int_descr, size_t len) {
uint8_t *ptr = malloc(len);
if (!ptr) {
return ERR_NO_MEMORY;
}
// create a temporary copy of the interface
memcpy(ptr, int_descr, len);
// find the last interface used
int interface_num = usb_get_current_iface_num(desc); // current interface
// patch our interface descriptor with the new id
ptr[2] = interface_num;
// append it to our config descriptor
append_desc_data(desc, ptr, len);
free(ptr);
// patch the total length of the config descriptor and set the number of interfaces
((uint16_t *)desc->desc)[1] += len;
interface_num++;
((uint8_t *)desc->desc)[4] = interface_num;
DEBUG_ASSERT(interface_num > 0);
return interface_num - 1;
}
int usb_append_interface_highspeed(const uint8_t *int_descr, size_t len) {
return usb_append_interface(&usb.config->highspeed.config, int_descr, len);
}
int usb_append_interface_lowspeed(const uint8_t *int_descr, size_t len) {
return usb_append_interface(&usb.config->lowspeed.config, int_descr, len);
}
static void set_usb_id(uint16_t vendor, uint16_t product) {
// patch the current configuration to with the vendor/product id
((uint16_t *)usb.config->lowspeed.device.desc)[4] = vendor;
((uint16_t *)usb.config->lowspeed.device.desc)[5] = product;
((uint16_t *)usb.config->highspeed.device.desc)[4] = vendor;
((uint16_t *)usb.config->highspeed.device.desc)[5] = product;
}
status_t usb_add_string(const char *string, uint8_t id) {
uint i;
size_t len = strlen(string);
uint16_t *strbuf = malloc(len * 2 + 2);
if (!strbuf) {
return ERR_NO_MEMORY;
}
/* build the usb string descriptor */
strbuf[0] = 0x300 | (len * 2 + 2);
for (i = 0; i < len; i++) {
strbuf[i + 1] = (uint16_t)string[i];
}
/* find a slot to put it */
for (i = 0; i < MAX_STRINGS; i++) {
if (usb.strings[i].id == 0) {
usb.strings[i].string.desc = strbuf;
usb.strings[i].string.len = len * 2 + 2;
usb.strings[i].id = id;
return NO_ERROR;
}
}
/* couldn't find a spot */
free(strbuf);
return ERR_NO_MEMORY;
}
static void usb_set_active_config(uint8_t config) {
if (config != usb.active_config) {
usb.active_config = config;
if (usb.active_config != 0) {
printf("usb online\n");
usb_do_callbacks(USB_CB_ONLINE, NULL);
} else {
printf("usb offline\n");
usb_do_callbacks(USB_CB_OFFLINE, NULL);
}
}
}
status_t usb_register_callback(usb_callback_t cb, void *cookie) {
DEBUG_ASSERT(cb);
usb_callback_container_t *c = malloc(sizeof(usb_callback_container_t));
if (!c) {
return ERR_NO_MEMORY;
}
c->cb = cb;
c->cookie = cookie;
list_add_tail(&usb.cb_list, &c->node);
return NO_ERROR;
}
static void usb_do_callbacks(usb_callback_op_t op, const union usb_callback_args *args) {
usb_callback_container_t *c;
list_for_every_entry(&usb.cb_list, c, usb_callback_container_t, node) {
c->cb(c->cookie, op, args);
}
}
status_t usbc_callback(usb_callback_op_t op, const union usb_callback_args *args) {
LTRACEF("op %d, args %p\n", op, args);
/* start looking for specific things to handle */
if (op == USB_CB_SETUP_MSG) {
bool setup_handled = false;
const struct usb_setup *setup = args->setup;
DEBUG_ASSERT(setup);
LTRACEF("SETUP: req_type=%#x req=%#x value=%#x index=%#x len=%#x\n",
setup->request_type, setup->request, setup->value, setup->index, setup->length);
if ((setup->request_type & TYPE_MASK) == TYPE_STANDARD) {
switch (setup->request) {
case SET_ADDRESS:
LTRACEF("SET_ADDRESS 0x%x\n", setup->value);
usbc_ep0_ack();
usbc_set_address(setup->value);
setup_handled = true;
break;
case SET_FEATURE:
case CLEAR_FEATURE:
LTRACEF("SET/CLEAR_FEATURE, feature 0x%x\n", setup->value);
usbc_ep0_ack();
setup_handled = true;
break;
case SET_DESCRIPTOR:
LTRACEF("SET_DESCRIPTOR\n");
usbc_ep0_stall();
setup_handled = true;
break;
case GET_DESCRIPTOR: {
if ((setup->request_type & RECIP_MASK) == RECIP_DEVICE) {
/* handle device descriptor fetches */
/* Get the right descriptors based on current speed */
const struct usb_descriptor_speed *speed;
if (usbc_is_highspeed()) {
speed = &usb.config->highspeed;
} else {
speed = &usb.config->lowspeed;
}
switch (setup->value) {
case 0x100: /* device */
LTRACEF("got GET_DESCRIPTOR, device descriptor\n");
usbc_ep0_send(speed->device.desc, speed->device.len,
setup->length);
break;
case 0x200: /* CONFIGURATION */
LTRACEF("got GET_DESCRIPTOR, config descriptor\n");
usbc_ep0_send(speed->config.desc, speed->config.len,
setup->length);
break;
case 0x300: /* Language ID */
LTRACEF("got GET_DESCRIPTOR, language id\n");
usbc_ep0_send(usb.config->langid.desc,
usb.config->langid.len, setup->length);
break;
case (0x301)...(0x3ff): {
/* string descriptor, search our list for a match */
uint i;
bool found = false;
uint8_t id = setup->value & 0xff;
for (i = 0; i < MAX_STRINGS; i++) {
if (usb.strings[i].id == id) {
usbc_ep0_send(usb.strings[i].string.desc,
usb.strings[i].string.len,
setup->length);
found = true;
break;
}
}
if (!found) {
/* couldn't find one, stall */
usbc_ep0_stall();
}
break;
}
case 0x600: /* DEVICE QUALIFIER */
LTRACEF("got GET_DESCRIPTOR, device qualifier\n");
usbc_ep0_send(speed->device_qual.desc,
speed->device_qual.len, setup->length);
break;
case 0xa00:
/* we aint got one of these */
LTRACEF("got GET_DESCRIPTOR, debug descriptor\n");
usbc_ep0_stall();
break;
default:
LTRACEF("unhandled descriptor %#x\n", setup->value);
// stall
break;
}
setup_handled = true;
}
break;
}
case SET_CONFIGURATION:
LTRACEF("SET_CONFIGURATION %d\n", setup->value);
usbc_ep0_ack();
usb_set_active_config(setup->value);
break;
case GET_CONFIGURATION:
LTRACEF("GET_CONFIGURATION\n");
usbc_ep0_send(&usb.active_config, 1, setup->length);
break;
case SET_INTERFACE:
LTRACEF("SET_INTERFACE %d\n", setup->value);
usbc_ep0_ack();
break;
case GET_INTERFACE: {
static uint8_t i = 1;
LTRACEF("GET_INTERFACE\n");
usbc_ep0_send(&i, 1, setup->length);
break;
}
case GET_STATUS: {
static uint16_t i = 1; // self powered
LTRACEF("GET_STATUS\n");
usbc_ep0_send(&i, 2, setup->length);
break;
}
default:
LTRACEF("unhandled standard request 0x%x\n", setup->request);
}
}
if (!setup_handled) {
usb_do_callbacks(op, args);
}
} else if (op == USB_CB_RESET) {
usb_do_callbacks(op, args);
usb.active_config = 0;
usb_do_callbacks(USB_CB_OFFLINE, args);
} else {
// other non setup messages, pass them down to anyone else
usb_do_callbacks(op, args);
}
return NO_ERROR;
}
status_t usb_setup(usb_config *config) {
DEBUG_ASSERT(config);
DEBUG_ASSERT(usb.active == false);
usb.config = config;
return NO_ERROR;
}
status_t usb_start(void) {
DEBUG_ASSERT(usb.config);
DEBUG_ASSERT(usb.active == false);
// go online
usbc_set_active(true);
usb.active = true;
return NO_ERROR;
}
status_t usb_stop(void) {
DEBUG_ASSERT(usb.active == true);
usb.active = false;
usbc_set_active(false);
return NO_ERROR;
}
static void usb_init(uint level) {
list_initialize(&usb.cb_list);
}
LK_INIT_HOOK(usb, usb_init, LK_INIT_LEVEL_THREADING);
Reference in New Issue View Git Blame Copy Permalink