mirror of
				https://xff.cz/git/u-boot/
				synced 2025-10-31 02:15:45 +01:00 
			
		
		
		
	Move env_set_hex() over to the new header file along with env_set_addr() which uses it. Signed-off-by: Simon Glass <sjg@chromium.org> Acked-by: Joe Hershberger <joe.hershberger@ni.com>
		
			
				
	
	
		
			793 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			793 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // SPDX-License-Identifier: GPL-2.0+
 | |
| /*
 | |
|  * f_sdp.c -- USB HID Serial Download Protocol
 | |
|  *
 | |
|  * Copyright (C) 2017 Toradex
 | |
|  * Author: Stefan Agner <stefan.agner@toradex.com>
 | |
|  *
 | |
|  * This file implements the Serial Download Protocol (SDP) as specified in
 | |
|  * the i.MX 6 Reference Manual. The SDP is a USB HID based protocol and
 | |
|  * allows to download images directly to memory. The implementation
 | |
|  * works with the imx_loader (imx_usb) USB client software on host side.
 | |
|  *
 | |
|  * Not all commands are implemented, e.g. WRITE_REGISTER, DCD_WRITE and
 | |
|  * SKIP_DCD_HEADER are only stubs.
 | |
|  *
 | |
|  * Parts of the implementation are based on f_dfu and f_thor.
 | |
|  */
 | |
| 
 | |
| #include <errno.h>
 | |
| #include <common.h>
 | |
| #include <console.h>
 | |
| #include <env.h>
 | |
| #include <malloc.h>
 | |
| 
 | |
| #include <linux/usb/ch9.h>
 | |
| #include <linux/usb/gadget.h>
 | |
| #include <linux/usb/composite.h>
 | |
| 
 | |
| #include <asm/io.h>
 | |
| #include <g_dnl.h>
 | |
| #include <sdp.h>
 | |
| #include <spl.h>
 | |
| #include <image.h>
 | |
| #include <imximage.h>
 | |
| #include <watchdog.h>
 | |
| 
 | |
| #define HID_REPORT_ID_MASK	0x000000ff
 | |
| 
 | |
| /*
 | |
|  * HID class requests
 | |
|  */
 | |
| #define HID_REQ_GET_REPORT		0x01
 | |
| #define HID_REQ_GET_IDLE		0x02
 | |
| #define HID_REQ_GET_PROTOCOL		0x03
 | |
| #define HID_REQ_SET_REPORT		0x09
 | |
| #define HID_REQ_SET_IDLE		0x0A
 | |
| #define HID_REQ_SET_PROTOCOL		0x0B
 | |
| 
 | |
| #define HID_USAGE_PAGE_LEN		76
 | |
| 
 | |
| struct hid_report {
 | |
| 	u8 usage_page[HID_USAGE_PAGE_LEN];
 | |
| } __packed;
 | |
| 
 | |
| #define SDP_READ_REGISTER	0x0101
 | |
| #define SDP_WRITE_REGISTER	0x0202
 | |
| #define SDP_WRITE_FILE		0x0404
 | |
| #define SDP_ERROR_STATUS	0x0505
 | |
| #define SDP_DCD_WRITE		0x0a0a
 | |
| #define SDP_JUMP_ADDRESS	0x0b0b
 | |
| #define SDP_SKIP_DCD_HEADER	0x0c0c
 | |
| 
 | |
| #define SDP_SECURITY_CLOSED		0x12343412
 | |
| #define SDP_SECURITY_OPEN		0x56787856
 | |
| 
 | |
| #define SDP_WRITE_FILE_COMPLETE		0x88888888
 | |
| #define SDP_WRITE_REGISTER_COMPLETE	0x128A8A12
 | |
| #define SDP_SKIP_DCD_HEADER_COMPLETE	0x900DD009
 | |
| #define SDP_ERROR_IMXHEADER		0x000a0533
 | |
| 
 | |
| #define SDP_COMMAND_LEN		16
 | |
| 
 | |
| struct sdp_command {
 | |
| 	u16 cmd;
 | |
| 	u32 addr;
 | |
| 	u8 format;
 | |
| 	u32 cnt;
 | |
| 	u32 data;
 | |
| 	u8 rsvd;
 | |
| } __packed;
 | |
| 
 | |
| enum sdp_state {
 | |
| 	SDP_STATE_IDLE,
 | |
| 	SDP_STATE_RX_DCD_DATA,
 | |
| 	SDP_STATE_RX_FILE_DATA,
 | |
| 	SDP_STATE_TX_SEC_CONF,
 | |
| 	SDP_STATE_TX_SEC_CONF_BUSY,
 | |
| 	SDP_STATE_TX_REGISTER,
 | |
| 	SDP_STATE_TX_REGISTER_BUSY,
 | |
| 	SDP_STATE_TX_STATUS,
 | |
| 	SDP_STATE_TX_STATUS_BUSY,
 | |
| 	SDP_STATE_JUMP,
 | |
| };
 | |
| 
 | |
| struct f_sdp {
 | |
| 	struct usb_function		usb_function;
 | |
| 
 | |
| 	struct usb_descriptor_header	**function;
 | |
| 
 | |
| 	u8				altsetting;
 | |
| 	enum sdp_state			state;
 | |
| 	enum sdp_state			next_state;
 | |
| 	u32				dnl_address;
 | |
| 	u32				dnl_bytes;
 | |
| 	u32				dnl_bytes_remaining;
 | |
| 	u32				jmp_address;
 | |
| 	bool				always_send_status;
 | |
| 	u32				error_status;
 | |
| 
 | |
| 	/* EP0 request */
 | |
| 	struct usb_request		*req;
 | |
| 
 | |
| 	/* EP1 IN */
 | |
| 	struct usb_ep			*in_ep;
 | |
| 	struct usb_request		*in_req;
 | |
| 
 | |
| 	bool				configuration_done;
 | |
| };
 | |
| 
 | |
| static struct f_sdp *sdp_func;
 | |
| 
 | |
| static inline struct f_sdp *func_to_sdp(struct usb_function *f)
 | |
| {
 | |
| 	return container_of(f, struct f_sdp, usb_function);
 | |
| }
 | |
| 
 | |
| static struct usb_interface_descriptor sdp_intf_runtime = {
 | |
| 	.bLength =		sizeof(sdp_intf_runtime),
 | |
| 	.bDescriptorType =	USB_DT_INTERFACE,
 | |
| 	.bAlternateSetting =	0,
 | |
| 	.bNumEndpoints =	1,
 | |
| 	.bInterfaceClass =	USB_CLASS_HID,
 | |
| 	.bInterfaceSubClass =	0,
 | |
| 	.bInterfaceProtocol =	0,
 | |
| 	/* .iInterface = DYNAMIC */
 | |
| };
 | |
| 
 | |
| /* HID configuration */
 | |
| static struct usb_class_hid_descriptor sdp_hid_desc = {
 | |
| 	.bLength =		sizeof(sdp_hid_desc),
 | |
| 	.bDescriptorType =	USB_DT_CS_DEVICE,
 | |
| 
 | |
| 	.bcdCDC =		__constant_cpu_to_le16(0x0110),
 | |
| 	.bCountryCode =		0,
 | |
| 	.bNumDescriptors =	1,
 | |
| 
 | |
| 	.bDescriptorType0	= USB_DT_HID_REPORT,
 | |
| 	.wDescriptorLength0	= HID_USAGE_PAGE_LEN,
 | |
| };
 | |
| 
 | |
| static struct usb_endpoint_descriptor in_desc = {
 | |
| 	.bLength =		USB_DT_ENDPOINT_SIZE,
 | |
| 	.bDescriptorType =	USB_DT_ENDPOINT, /*USB_DT_CS_ENDPOINT*/
 | |
| 
 | |
| 	.bEndpointAddress =	1 | USB_DIR_IN,
 | |
| 	.bmAttributes =	USB_ENDPOINT_XFER_INT,
 | |
| 	.wMaxPacketSize =	64,
 | |
| 	.bInterval =		1,
 | |
| };
 | |
| 
 | |
| static struct usb_descriptor_header *sdp_runtime_descs[] = {
 | |
| 	(struct usb_descriptor_header *)&sdp_intf_runtime,
 | |
| 	(struct usb_descriptor_header *)&sdp_hid_desc,
 | |
| 	(struct usb_descriptor_header *)&in_desc,
 | |
| 	NULL,
 | |
| };
 | |
| 
 | |
| /* This is synchronized with what the SoC implementation reports */
 | |
| static struct hid_report sdp_hid_report = {
 | |
| 	.usage_page = {
 | |
| 		0x06, 0x00, 0xff, /* Usage Page */
 | |
| 		0x09, 0x01, /* Usage (Pointer?) */
 | |
| 		0xa1, 0x01, /* Collection */
 | |
| 
 | |
| 		0x85, 0x01, /* Report ID */
 | |
| 		0x19, 0x01, /* Usage Minimum */
 | |
| 		0x29, 0x01, /* Usage Maximum */
 | |
| 		0x15, 0x00, /* Local Minimum */
 | |
| 		0x26, 0xFF, 0x00, /* Local Maximum? */
 | |
| 		0x75, 0x08, /* Report Size */
 | |
| 		0x95, 0x10, /* Report Count */
 | |
| 		0x91, 0x02, /* Output Data */
 | |
| 
 | |
| 		0x85, 0x02, /* Report ID */
 | |
| 		0x19, 0x01, /* Usage Minimum */
 | |
| 		0x29, 0x01, /* Usage Maximum */
 | |
| 		0x15, 0x00, /* Local Minimum */
 | |
| 		0x26, 0xFF, 0x00, /* Local Maximum? */
 | |
| 		0x75, 0x80, /* Report Size 128 */
 | |
| 		0x95, 0x40, /* Report Count */
 | |
| 		0x91, 0x02, /* Output Data */
 | |
| 
 | |
| 		0x85, 0x03, /* Report ID */
 | |
| 		0x19, 0x01, /* Usage Minimum */
 | |
| 		0x29, 0x01, /* Usage Maximum */
 | |
| 		0x15, 0x00, /* Local Minimum */
 | |
| 		0x26, 0xFF, 0x00, /* Local Maximum? */
 | |
| 		0x75, 0x08, /* Report Size 8 */
 | |
| 		0x95, 0x04, /* Report Count */
 | |
| 		0x81, 0x02, /* Input Data */
 | |
| 
 | |
| 		0x85, 0x04, /* Report ID */
 | |
| 		0x19, 0x01, /* Usage Minimum */
 | |
| 		0x29, 0x01, /* Usage Maximum */
 | |
| 		0x15, 0x00, /* Local Minimum */
 | |
| 		0x26, 0xFF, 0x00, /* Local Maximum? */
 | |
| 		0x75, 0x08, /* Report Size 8 */
 | |
| 		0x95, 0x40, /* Report Count */
 | |
| 		0x81, 0x02, /* Input Data */
 | |
| 		0xc0
 | |
| 	},
 | |
| };
 | |
| 
 | |
| static const char sdp_name[] = "Serial Downloader Protocol";
 | |
| 
 | |
| /*
 | |
|  * static strings, in UTF-8
 | |
|  */
 | |
| static struct usb_string strings_sdp_generic[] = {
 | |
| 	[0].s = sdp_name,
 | |
| 	{  }			/* end of list */
 | |
| };
 | |
| 
 | |
| static struct usb_gadget_strings stringtab_sdp_generic = {
 | |
| 	.language	= 0x0409,	/* en-us */
 | |
| 	.strings	= strings_sdp_generic,
 | |
| };
 | |
| 
 | |
| static struct usb_gadget_strings *sdp_generic_strings[] = {
 | |
| 	&stringtab_sdp_generic,
 | |
| 	NULL,
 | |
| };
 | |
| 
 | |
| static inline void *sdp_ptr(u32 val)
 | |
| {
 | |
| 	return (void *)(uintptr_t)val;
 | |
| }
 | |
| 
 | |
| static void sdp_rx_command_complete(struct usb_ep *ep, struct usb_request *req)
 | |
| {
 | |
| 	struct f_sdp *sdp = req->context;
 | |
| 	int status = req->status;
 | |
| 	u8 *data = req->buf;
 | |
| 	u8 report = data[0];
 | |
| 
 | |
| 	if (status != 0) {
 | |
| 		pr_err("Status: %d\n", status);
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	if (report != 1) {
 | |
| 		pr_err("Unexpected report %d\n", report);
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	struct sdp_command *cmd = req->buf + 1;
 | |
| 
 | |
| 	debug("%s: command: %04x, addr: %08x, cnt: %u\n",
 | |
| 	      __func__, be16_to_cpu(cmd->cmd),
 | |
| 	      be32_to_cpu(cmd->addr), be32_to_cpu(cmd->cnt));
 | |
| 
 | |
| 	switch (be16_to_cpu(cmd->cmd)) {
 | |
| 	case SDP_READ_REGISTER:
 | |
| 		sdp->always_send_status = false;
 | |
| 		sdp->error_status = 0x0;
 | |
| 
 | |
| 		sdp->state = SDP_STATE_TX_SEC_CONF;
 | |
| 		sdp->dnl_address = be32_to_cpu(cmd->addr);
 | |
| 		sdp->dnl_bytes_remaining = be32_to_cpu(cmd->cnt);
 | |
| 		sdp->next_state = SDP_STATE_TX_REGISTER;
 | |
| 		printf("Reading %d registers at 0x%08x... ",
 | |
| 		       sdp->dnl_bytes_remaining, sdp->dnl_address);
 | |
| 		break;
 | |
| 	case SDP_WRITE_FILE:
 | |
| 		sdp->always_send_status = true;
 | |
| 		sdp->error_status = SDP_WRITE_FILE_COMPLETE;
 | |
| 
 | |
| 		sdp->state = SDP_STATE_RX_FILE_DATA;
 | |
| 		sdp->dnl_address = be32_to_cpu(cmd->addr);
 | |
| 		sdp->dnl_bytes_remaining = be32_to_cpu(cmd->cnt);
 | |
| 		sdp->dnl_bytes = sdp->dnl_bytes_remaining;
 | |
| 		sdp->next_state = SDP_STATE_IDLE;
 | |
| 
 | |
| 		printf("Downloading file of size %d to 0x%08x... ",
 | |
| 		       sdp->dnl_bytes_remaining, sdp->dnl_address);
 | |
| 
 | |
| 		break;
 | |
| 	case SDP_ERROR_STATUS:
 | |
| 		sdp->always_send_status = true;
 | |
| 		sdp->error_status = 0;
 | |
| 
 | |
| 		sdp->state = SDP_STATE_TX_SEC_CONF;
 | |
| 		sdp->next_state = SDP_STATE_IDLE;
 | |
| 		break;
 | |
| 	case SDP_DCD_WRITE:
 | |
| 		sdp->always_send_status = true;
 | |
| 		sdp->error_status = SDP_WRITE_REGISTER_COMPLETE;
 | |
| 
 | |
| 		sdp->state = SDP_STATE_RX_DCD_DATA;
 | |
| 		sdp->dnl_bytes_remaining = be32_to_cpu(cmd->cnt);
 | |
| 		sdp->next_state = SDP_STATE_IDLE;
 | |
| 		break;
 | |
| 	case SDP_JUMP_ADDRESS:
 | |
| 		sdp->always_send_status = false;
 | |
| 		sdp->error_status = 0;
 | |
| 
 | |
| 		sdp->jmp_address = be32_to_cpu(cmd->addr);
 | |
| 		sdp->state = SDP_STATE_TX_SEC_CONF;
 | |
| 		sdp->next_state = SDP_STATE_JUMP;
 | |
| 		break;
 | |
| 	case SDP_SKIP_DCD_HEADER:
 | |
| 		sdp->always_send_status = true;
 | |
| 		sdp->error_status = SDP_SKIP_DCD_HEADER_COMPLETE;
 | |
| 
 | |
| 		/* Ignore command, DCD not supported anyway */
 | |
| 		sdp->state = SDP_STATE_TX_SEC_CONF;
 | |
| 		sdp->next_state = SDP_STATE_IDLE;
 | |
| 		break;
 | |
| 	default:
 | |
| 		pr_err("Unknown command: %04x\n", be16_to_cpu(cmd->cmd));
 | |
| 	}
 | |
| }
 | |
| 
 | |
| static void sdp_rx_data_complete(struct usb_ep *ep, struct usb_request *req)
 | |
| {
 | |
| 	struct f_sdp *sdp = req->context;
 | |
| 	int status = req->status;
 | |
| 	u8 *data = req->buf;
 | |
| 	u8 report = data[0];
 | |
| 	int datalen = req->length - 1;
 | |
| 
 | |
| 	if (status != 0) {
 | |
| 		pr_err("Status: %d\n", status);
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	if (report != 2) {
 | |
| 		pr_err("Unexpected report %d\n", report);
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	if (sdp->dnl_bytes_remaining < datalen) {
 | |
| 		/*
 | |
| 		 * Some USB stacks require to send a complete buffer as
 | |
| 		 * specified in the HID descriptor. This leads to longer
 | |
| 		 * transfers than the file length, no problem for us.
 | |
| 		 */
 | |
| 		sdp->dnl_bytes_remaining = 0;
 | |
| 	} else {
 | |
| 		sdp->dnl_bytes_remaining -= datalen;
 | |
| 	}
 | |
| 
 | |
| 	if (sdp->state == SDP_STATE_RX_FILE_DATA) {
 | |
| 		memcpy(sdp_ptr(sdp->dnl_address), req->buf + 1, datalen);
 | |
| 		sdp->dnl_address += datalen;
 | |
| 	}
 | |
| 
 | |
| 	if (sdp->dnl_bytes_remaining)
 | |
| 		return;
 | |
| 
 | |
| #ifndef CONFIG_SPL_BUILD
 | |
| 	env_set_hex("filesize", sdp->dnl_bytes);
 | |
| #endif
 | |
| 	printf("done\n");
 | |
| 
 | |
| 	switch (sdp->state) {
 | |
| 	case SDP_STATE_RX_FILE_DATA:
 | |
| 		sdp->state = SDP_STATE_TX_SEC_CONF;
 | |
| 		break;
 | |
| 	case SDP_STATE_RX_DCD_DATA:
 | |
| 		sdp->state = SDP_STATE_TX_SEC_CONF;
 | |
| 		break;
 | |
| 	default:
 | |
| 		pr_err("Invalid state: %d\n", sdp->state);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| static void sdp_tx_complete(struct usb_ep *ep, struct usb_request *req)
 | |
| {
 | |
| 	struct f_sdp *sdp = req->context;
 | |
| 	int status = req->status;
 | |
| 
 | |
| 	if (status != 0) {
 | |
| 		pr_err("Status: %d\n", status);
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	switch (sdp->state) {
 | |
| 	case SDP_STATE_TX_SEC_CONF_BUSY:
 | |
| 		/* Not all commands require status report */
 | |
| 		if (sdp->always_send_status || sdp->error_status)
 | |
| 			sdp->state = SDP_STATE_TX_STATUS;
 | |
| 		else
 | |
| 			sdp->state = sdp->next_state;
 | |
| 
 | |
| 		break;
 | |
| 	case SDP_STATE_TX_STATUS_BUSY:
 | |
| 		sdp->state = sdp->next_state;
 | |
| 		break;
 | |
| 	case SDP_STATE_TX_REGISTER_BUSY:
 | |
| 		if (sdp->dnl_bytes_remaining)
 | |
| 			sdp->state = SDP_STATE_TX_REGISTER;
 | |
| 		else
 | |
| 			sdp->state = SDP_STATE_IDLE;
 | |
| 		break;
 | |
| 	default:
 | |
| 		pr_err("Wrong State: %d\n", sdp->state);
 | |
| 		sdp->state = SDP_STATE_IDLE;
 | |
| 		break;
 | |
| 	}
 | |
| 	debug("%s complete --> %d, %d/%d\n", ep->name,
 | |
| 	      status, req->actual, req->length);
 | |
| }
 | |
| 
 | |
| static int sdp_setup(struct usb_function *f, const struct usb_ctrlrequest *ctrl)
 | |
| {
 | |
| 	struct usb_gadget *gadget = f->config->cdev->gadget;
 | |
| 	struct usb_request *req = f->config->cdev->req;
 | |
| 	struct f_sdp *sdp = f->config->cdev->req->context;
 | |
| 	u16 len = le16_to_cpu(ctrl->wLength);
 | |
| 	u16 w_value = le16_to_cpu(ctrl->wValue);
 | |
| 	int value = 0;
 | |
| 	u8 req_type = ctrl->bRequestType & USB_TYPE_MASK;
 | |
| 
 | |
| 	debug("w_value: 0x%04x len: 0x%04x\n", w_value, len);
 | |
| 	debug("req_type: 0x%02x ctrl->bRequest: 0x%02x sdp->state: %d\n",
 | |
| 	      req_type, ctrl->bRequest, sdp->state);
 | |
| 
 | |
| 	if (req_type == USB_TYPE_STANDARD) {
 | |
| 		if (ctrl->bRequest == USB_REQ_GET_DESCRIPTOR) {
 | |
| 			/* Send HID report descriptor */
 | |
| 			value = min(len, (u16) sizeof(sdp_hid_report));
 | |
| 			memcpy(req->buf, &sdp_hid_report, value);
 | |
| 			sdp->configuration_done = true;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (req_type == USB_TYPE_CLASS) {
 | |
| 		int report = w_value & HID_REPORT_ID_MASK;
 | |
| 
 | |
| 		/* HID (SDP) request */
 | |
| 		switch (ctrl->bRequest) {
 | |
| 		case HID_REQ_SET_REPORT:
 | |
| 			switch (report) {
 | |
| 			case 1:
 | |
| 				value = SDP_COMMAND_LEN + 1;
 | |
| 				req->complete = sdp_rx_command_complete;
 | |
| 				break;
 | |
| 			case 2:
 | |
| 				value = len;
 | |
| 				req->complete = sdp_rx_data_complete;
 | |
| 				break;
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (value >= 0) {
 | |
| 		req->length = value;
 | |
| 		req->zero = value < len;
 | |
| 		value = usb_ep_queue(gadget->ep0, req, 0);
 | |
| 		if (value < 0) {
 | |
| 			debug("ep_queue --> %d\n", value);
 | |
| 			req->status = 0;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return value;
 | |
| }
 | |
| 
 | |
| static int sdp_bind(struct usb_configuration *c, struct usb_function *f)
 | |
| {
 | |
| 	struct usb_gadget *gadget = c->cdev->gadget;
 | |
| 	struct usb_composite_dev *cdev = c->cdev;
 | |
| 	struct f_sdp *sdp = func_to_sdp(f);
 | |
| 	int rv = 0, id;
 | |
| 
 | |
| 	id = usb_interface_id(c, f);
 | |
| 	if (id < 0)
 | |
| 		return id;
 | |
| 	sdp_intf_runtime.bInterfaceNumber = id;
 | |
| 
 | |
| 	struct usb_ep *ep;
 | |
| 
 | |
| 	/* allocate instance-specific endpoints */
 | |
| 	ep = usb_ep_autoconfig(gadget, &in_desc);
 | |
| 	if (!ep) {
 | |
| 		rv = -ENODEV;
 | |
| 		goto error;
 | |
| 	}
 | |
| 
 | |
| 	sdp->in_ep = ep; /* Store IN EP for enabling @ setup */
 | |
| 
 | |
| 	cdev->req->context = sdp;
 | |
| 
 | |
| error:
 | |
| 	return rv;
 | |
| }
 | |
| 
 | |
| static void sdp_unbind(struct usb_configuration *c, struct usb_function *f)
 | |
| {
 | |
| 	free(sdp_func);
 | |
| 	sdp_func = NULL;
 | |
| }
 | |
| 
 | |
| static struct usb_request *alloc_ep_req(struct usb_ep *ep, unsigned length)
 | |
| {
 | |
| 	struct usb_request *req;
 | |
| 
 | |
| 	req = usb_ep_alloc_request(ep, 0);
 | |
| 	if (!req)
 | |
| 		return req;
 | |
| 
 | |
| 	req->length = length;
 | |
| 	req->buf = memalign(CONFIG_SYS_CACHELINE_SIZE, length);
 | |
| 	if (!req->buf) {
 | |
| 		usb_ep_free_request(ep, req);
 | |
| 		req = NULL;
 | |
| 	}
 | |
| 
 | |
| 	return req;
 | |
| }
 | |
| 
 | |
| 
 | |
| static struct usb_request *sdp_start_ep(struct usb_ep *ep)
 | |
| {
 | |
| 	struct usb_request *req;
 | |
| 
 | |
| 	req = alloc_ep_req(ep, 64);
 | |
| 	debug("%s: ep:%p req:%p\n", __func__, ep, req);
 | |
| 
 | |
| 	if (!req)
 | |
| 		return NULL;
 | |
| 
 | |
| 	memset(req->buf, 0, req->length);
 | |
| 	req->complete = sdp_tx_complete;
 | |
| 
 | |
| 	return req;
 | |
| }
 | |
| static int sdp_set_alt(struct usb_function *f, unsigned intf, unsigned alt)
 | |
| {
 | |
| 	struct f_sdp *sdp = func_to_sdp(f);
 | |
| 	struct usb_composite_dev *cdev = f->config->cdev;
 | |
| 	int result;
 | |
| 
 | |
| 	debug("%s: intf: %d alt: %d\n", __func__, intf, alt);
 | |
| 
 | |
| 	result = usb_ep_enable(sdp->in_ep, &in_desc);
 | |
| 	if (result)
 | |
| 		return result;
 | |
| 	sdp->in_req = sdp_start_ep(sdp->in_ep);
 | |
| 	sdp->in_req->context = sdp;
 | |
| 
 | |
| 	sdp->in_ep->driver_data = cdev; /* claim */
 | |
| 
 | |
| 	sdp->altsetting = alt;
 | |
| 	sdp->state = SDP_STATE_IDLE;
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int sdp_get_alt(struct usb_function *f, unsigned intf)
 | |
| {
 | |
| 	struct f_sdp *sdp = func_to_sdp(f);
 | |
| 
 | |
| 	return sdp->altsetting;
 | |
| }
 | |
| 
 | |
| static void sdp_disable(struct usb_function *f)
 | |
| {
 | |
| 	struct f_sdp *sdp = func_to_sdp(f);
 | |
| 
 | |
| 	usb_ep_disable(sdp->in_ep);
 | |
| 
 | |
| 	if (sdp->in_req) {
 | |
| 		free(sdp->in_req);
 | |
| 		sdp->in_req = NULL;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| static int sdp_bind_config(struct usb_configuration *c)
 | |
| {
 | |
| 	int status;
 | |
| 
 | |
| 	if (!sdp_func) {
 | |
| 		sdp_func = memalign(CONFIG_SYS_CACHELINE_SIZE, sizeof(*sdp_func));
 | |
| 		if (!sdp_func)
 | |
| 			return -ENOMEM;
 | |
| 	}
 | |
| 
 | |
| 	memset(sdp_func, 0, sizeof(*sdp_func));
 | |
| 
 | |
| 	sdp_func->usb_function.name = "sdp";
 | |
| 	sdp_func->usb_function.hs_descriptors = sdp_runtime_descs;
 | |
| 	sdp_func->usb_function.descriptors = sdp_runtime_descs;
 | |
| 	sdp_func->usb_function.bind = sdp_bind;
 | |
| 	sdp_func->usb_function.unbind = sdp_unbind;
 | |
| 	sdp_func->usb_function.set_alt = sdp_set_alt;
 | |
| 	sdp_func->usb_function.get_alt = sdp_get_alt;
 | |
| 	sdp_func->usb_function.disable = sdp_disable;
 | |
| 	sdp_func->usb_function.strings = sdp_generic_strings;
 | |
| 	sdp_func->usb_function.setup = sdp_setup;
 | |
| 
 | |
| 	status = usb_add_function(c, &sdp_func->usb_function);
 | |
| 
 | |
| 	return status;
 | |
| }
 | |
| 
 | |
| int sdp_init(int controller_index)
 | |
| {
 | |
| 	printf("SDP: initialize...\n");
 | |
| 	while (!sdp_func->configuration_done) {
 | |
| 		if (ctrlc()) {
 | |
| 			puts("\rCTRL+C - Operation aborted.\n");
 | |
| 			return 1;
 | |
| 		}
 | |
| 
 | |
| 		WATCHDOG_RESET();
 | |
| 		usb_gadget_handle_interrupts(controller_index);
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static u32 sdp_jump_imxheader(void *address)
 | |
| {
 | |
| 	flash_header_v2_t *headerv2 = address;
 | |
| 	ulong (*entry)(void);
 | |
| 
 | |
| 	if (headerv2->header.tag != IVT_HEADER_TAG) {
 | |
| 		printf("Header Tag is not an IMX image\n");
 | |
| 		return SDP_ERROR_IMXHEADER;
 | |
| 	}
 | |
| 
 | |
| 	printf("Jumping to 0x%08x\n", headerv2->entry);
 | |
| 	entry = sdp_ptr(headerv2->entry);
 | |
| 	entry();
 | |
| 
 | |
| 	/* The image probably never returns hence we won't reach that point */
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| #ifdef CONFIG_SPL_BUILD
 | |
| #ifdef CONFIG_SPL_LOAD_FIT
 | |
| static ulong sdp_fit_read(struct spl_load_info *load, ulong sector,
 | |
| 			  ulong count, void *buf)
 | |
| {
 | |
| 	debug("%s: sector %lx, count %lx, buf %lx\n",
 | |
| 	      __func__, sector, count, (ulong)buf);
 | |
| 	memcpy(buf, (void *)(load->dev + sector), count);
 | |
| 	return count;
 | |
| }
 | |
| #endif
 | |
| #endif
 | |
| 
 | |
| static void sdp_handle_in_ep(struct spl_image_info *spl_image)
 | |
| {
 | |
| 	u8 *data = sdp_func->in_req->buf;
 | |
| 	u32 status;
 | |
| 	int datalen;
 | |
| 
 | |
| 	switch (sdp_func->state) {
 | |
| 	case SDP_STATE_TX_SEC_CONF:
 | |
| 		debug("Report 3: HAB security\n");
 | |
| 		data[0] = 3;
 | |
| 
 | |
| 		status = SDP_SECURITY_OPEN;
 | |
| 		memcpy(&data[1], &status, 4);
 | |
| 		sdp_func->in_req->length = 5;
 | |
| 		usb_ep_queue(sdp_func->in_ep, sdp_func->in_req, 0);
 | |
| 		sdp_func->state = SDP_STATE_TX_SEC_CONF_BUSY;
 | |
| 		break;
 | |
| 
 | |
| 	case SDP_STATE_TX_STATUS:
 | |
| 		debug("Report 4: Status\n");
 | |
| 		data[0] = 4;
 | |
| 
 | |
| 		memcpy(&data[1], &sdp_func->error_status, 4);
 | |
| 		sdp_func->in_req->length = 65;
 | |
| 		usb_ep_queue(sdp_func->in_ep, sdp_func->in_req, 0);
 | |
| 		sdp_func->state = SDP_STATE_TX_STATUS_BUSY;
 | |
| 		break;
 | |
| 	case SDP_STATE_TX_REGISTER:
 | |
| 		debug("Report 4: Register Values\n");
 | |
| 		data[0] = 4;
 | |
| 
 | |
| 		datalen = sdp_func->dnl_bytes_remaining;
 | |
| 
 | |
| 		if (datalen > 64)
 | |
| 			datalen = 64;
 | |
| 
 | |
| 		memcpy(&data[1], sdp_ptr(sdp_func->dnl_address), datalen);
 | |
| 		sdp_func->in_req->length = 65;
 | |
| 
 | |
| 		sdp_func->dnl_bytes_remaining -= datalen;
 | |
| 		sdp_func->dnl_address += datalen;
 | |
| 
 | |
| 		usb_ep_queue(sdp_func->in_ep, sdp_func->in_req, 0);
 | |
| 		sdp_func->state = SDP_STATE_TX_REGISTER_BUSY;
 | |
| 		break;
 | |
| 	case SDP_STATE_JUMP:
 | |
| 		printf("Jumping to header at 0x%08x\n", sdp_func->jmp_address);
 | |
| 		status = sdp_jump_imxheader(sdp_ptr(sdp_func->jmp_address));
 | |
| 
 | |
| 		/* If imx header fails, try some U-Boot specific headers */
 | |
| 		if (status) {
 | |
| #ifdef CONFIG_SPL_BUILD
 | |
| 			image_header_t *header =
 | |
| 				sdp_ptr(sdp_func->jmp_address);
 | |
| #ifdef CONFIG_SPL_LOAD_FIT
 | |
| 			if (image_get_magic(header) == FDT_MAGIC) {
 | |
| 				struct spl_load_info load;
 | |
| 
 | |
| 				debug("Found FIT\n");
 | |
| 				load.dev = header;
 | |
| 				load.bl_len = 1;
 | |
| 				load.read = sdp_fit_read;
 | |
| 				spl_load_simple_fit(spl_image, &load, 0,
 | |
| 						    header);
 | |
| 
 | |
| 				return;
 | |
| 			}
 | |
| #endif
 | |
| 			/* In SPL, allow jumps to U-Boot images */
 | |
| 			struct spl_image_info spl_image = {};
 | |
| 			spl_parse_image_header(&spl_image, header);
 | |
| 			jump_to_image_no_args(&spl_image);
 | |
| #else
 | |
| 			/* In U-Boot, allow jumps to scripts */
 | |
| 			source(sdp_func->jmp_address, "script@1");
 | |
| #endif
 | |
| 		}
 | |
| 
 | |
| 		sdp_func->next_state = SDP_STATE_IDLE;
 | |
| 		sdp_func->error_status = status;
 | |
| 
 | |
| 		/* Only send Report 4 if there was an error */
 | |
| 		if (status)
 | |
| 			sdp_func->state = SDP_STATE_TX_STATUS;
 | |
| 		else
 | |
| 			sdp_func->state = SDP_STATE_IDLE;
 | |
| 		break;
 | |
| 	default:
 | |
| 		break;
 | |
| 	};
 | |
| }
 | |
| 
 | |
| #ifndef CONFIG_SPL_BUILD
 | |
| int sdp_handle(int controller_index)
 | |
| #else
 | |
| int spl_sdp_handle(int controller_index, struct spl_image_info *spl_image)
 | |
| #endif
 | |
| {
 | |
| 	printf("SDP: handle requests...\n");
 | |
| 	while (1) {
 | |
| 		if (ctrlc()) {
 | |
| 			puts("\rCTRL+C - Operation aborted.\n");
 | |
| 			return -EINVAL;
 | |
| 		}
 | |
| 
 | |
| #ifdef CONFIG_SPL_BUILD
 | |
| 		if (spl_image->flags & SPL_FIT_FOUND)
 | |
| 			return 0;
 | |
| #endif
 | |
| 
 | |
| 		WATCHDOG_RESET();
 | |
| 		usb_gadget_handle_interrupts(controller_index);
 | |
| 
 | |
| #ifdef CONFIG_SPL_BUILD
 | |
| 		sdp_handle_in_ep(spl_image);
 | |
| #else
 | |
| 		sdp_handle_in_ep(NULL);
 | |
| #endif
 | |
| 	}
 | |
| }
 | |
| 
 | |
| int sdp_add(struct usb_configuration *c)
 | |
| {
 | |
| 	int id;
 | |
| 
 | |
| 	id = usb_string_id(c->cdev);
 | |
| 	if (id < 0)
 | |
| 		return id;
 | |
| 	strings_sdp_generic[0].id = id;
 | |
| 	sdp_intf_runtime.iInterface = id;
 | |
| 
 | |
| 	debug("%s: cdev: %p gadget: %p gadget->ep0: %p\n", __func__,
 | |
| 	      c->cdev, c->cdev->gadget, c->cdev->gadget->ep0);
 | |
| 
 | |
| 	return sdp_bind_config(c);
 | |
| }
 | |
| 
 | |
| DECLARE_GADGET_BIND_CALLBACK(usb_dnl_sdp, sdp_add);
 |