• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * f_sdp.c -- USB HID Serial Download Protocol
4  *
5  * Copyright (C) 2017 Toradex
6  * Author: Stefan Agner <stefan.agner@toradex.com>
7  *
8  * This file implements the Serial Download Protocol (SDP) as specified in
9  * the i.MX 6 Reference Manual. The SDP is a USB HID based protocol and
10  * allows to download images directly to memory. The implementation
11  * works with the imx_loader (imx_usb) USB client software on host side.
12  *
13  * Not all commands are implemented, e.g. WRITE_REGISTER, DCD_WRITE and
14  * SKIP_DCD_HEADER are only stubs.
15  *
16  * Parts of the implementation are based on f_dfu and f_thor.
17  */
18 
19 #include <errno.h>
20 #include <common.h>
21 #include <console.h>
22 #include <env.h>
23 #include <malloc.h>
24 
25 #include <linux/usb/ch9.h>
26 #include <linux/usb/gadget.h>
27 #include <linux/usb/composite.h>
28 
29 #include <asm/io.h>
30 #include <g_dnl.h>
31 #include <sdp.h>
32 #include <spl.h>
33 #include <image.h>
34 #include <imximage.h>
35 #include <watchdog.h>
36 
37 #define HID_REPORT_ID_MASK	0x000000ff
38 
39 /*
40  * HID class requests
41  */
42 #define HID_REQ_GET_REPORT		0x01
43 #define HID_REQ_GET_IDLE		0x02
44 #define HID_REQ_GET_PROTOCOL		0x03
45 #define HID_REQ_SET_REPORT		0x09
46 #define HID_REQ_SET_IDLE		0x0A
47 #define HID_REQ_SET_PROTOCOL		0x0B
48 
49 #define HID_USAGE_PAGE_LEN		76
50 
51 struct hid_report {
52 	u8 usage_page[HID_USAGE_PAGE_LEN];
53 } __packed;
54 
55 #define SDP_READ_REGISTER	0x0101
56 #define SDP_WRITE_REGISTER	0x0202
57 #define SDP_WRITE_FILE		0x0404
58 #define SDP_ERROR_STATUS	0x0505
59 #define SDP_DCD_WRITE		0x0a0a
60 #define SDP_JUMP_ADDRESS	0x0b0b
61 #define SDP_SKIP_DCD_HEADER	0x0c0c
62 
63 #define SDP_SECURITY_CLOSED		0x12343412
64 #define SDP_SECURITY_OPEN		0x56787856
65 
66 #define SDP_WRITE_FILE_COMPLETE		0x88888888
67 #define SDP_WRITE_REGISTER_COMPLETE	0x128A8A12
68 #define SDP_SKIP_DCD_HEADER_COMPLETE	0x900DD009
69 #define SDP_ERROR_IMXHEADER		0x000a0533
70 
71 #define SDP_COMMAND_LEN		16
72 
73 struct sdp_command {
74 	u16 cmd;
75 	u32 addr;
76 	u8 format;
77 	u32 cnt;
78 	u32 data;
79 	u8 rsvd;
80 } __packed;
81 
82 enum sdp_state {
83 	SDP_STATE_IDLE,
84 	SDP_STATE_RX_DCD_DATA,
85 	SDP_STATE_RX_FILE_DATA,
86 	SDP_STATE_TX_SEC_CONF,
87 	SDP_STATE_TX_SEC_CONF_BUSY,
88 	SDP_STATE_TX_REGISTER,
89 	SDP_STATE_TX_REGISTER_BUSY,
90 	SDP_STATE_TX_STATUS,
91 	SDP_STATE_TX_STATUS_BUSY,
92 	SDP_STATE_JUMP,
93 };
94 
95 struct f_sdp {
96 	struct usb_function		usb_function;
97 
98 	struct usb_descriptor_header	**function;
99 
100 	u8				altsetting;
101 	enum sdp_state			state;
102 	enum sdp_state			next_state;
103 	u32				dnl_address;
104 	u32				dnl_bytes;
105 	u32				dnl_bytes_remaining;
106 	u32				jmp_address;
107 	bool				always_send_status;
108 	u32				error_status;
109 
110 	/* EP0 request */
111 	struct usb_request		*req;
112 
113 	/* EP1 IN */
114 	struct usb_ep			*in_ep;
115 	struct usb_request		*in_req;
116 
117 	bool				configuration_done;
118 };
119 
120 static struct f_sdp *sdp_func;
121 
func_to_sdp(struct usb_function * f)122 static inline struct f_sdp *func_to_sdp(struct usb_function *f)
123 {
124 	return container_of(f, struct f_sdp, usb_function);
125 }
126 
127 static struct usb_interface_descriptor sdp_intf_runtime = {
128 	.bLength =		sizeof(sdp_intf_runtime),
129 	.bDescriptorType =	USB_DT_INTERFACE,
130 	.bAlternateSetting =	0,
131 	.bNumEndpoints =	1,
132 	.bInterfaceClass =	USB_CLASS_HID,
133 	.bInterfaceSubClass =	0,
134 	.bInterfaceProtocol =	0,
135 	/* .iInterface = DYNAMIC */
136 };
137 
138 /* HID configuration */
139 static struct usb_class_hid_descriptor sdp_hid_desc = {
140 	.bLength =		sizeof(sdp_hid_desc),
141 	.bDescriptorType =	USB_DT_CS_DEVICE,
142 
143 	.bcdCDC =		__constant_cpu_to_le16(0x0110),
144 	.bCountryCode =		0,
145 	.bNumDescriptors =	1,
146 
147 	.bDescriptorType0	= USB_DT_HID_REPORT,
148 	.wDescriptorLength0	= HID_USAGE_PAGE_LEN,
149 };
150 
151 static struct usb_endpoint_descriptor in_desc = {
152 	.bLength =		USB_DT_ENDPOINT_SIZE,
153 	.bDescriptorType =	USB_DT_ENDPOINT, /*USB_DT_CS_ENDPOINT*/
154 
155 	.bEndpointAddress =	1 | USB_DIR_IN,
156 	.bmAttributes =	USB_ENDPOINT_XFER_INT,
157 	.wMaxPacketSize =	64,
158 	.bInterval =		1,
159 };
160 
161 static struct usb_descriptor_header *sdp_runtime_descs[] = {
162 	(struct usb_descriptor_header *)&sdp_intf_runtime,
163 	(struct usb_descriptor_header *)&sdp_hid_desc,
164 	(struct usb_descriptor_header *)&in_desc,
165 	NULL,
166 };
167 
168 /* This is synchronized with what the SoC implementation reports */
169 static struct hid_report sdp_hid_report = {
170 	.usage_page = {
171 		0x06, 0x00, 0xff, /* Usage Page */
172 		0x09, 0x01, /* Usage (Pointer?) */
173 		0xa1, 0x01, /* Collection */
174 
175 		0x85, 0x01, /* Report ID */
176 		0x19, 0x01, /* Usage Minimum */
177 		0x29, 0x01, /* Usage Maximum */
178 		0x15, 0x00, /* Local Minimum */
179 		0x26, 0xFF, 0x00, /* Local Maximum? */
180 		0x75, 0x08, /* Report Size */
181 		0x95, 0x10, /* Report Count */
182 		0x91, 0x02, /* Output Data */
183 
184 		0x85, 0x02, /* Report ID */
185 		0x19, 0x01, /* Usage Minimum */
186 		0x29, 0x01, /* Usage Maximum */
187 		0x15, 0x00, /* Local Minimum */
188 		0x26, 0xFF, 0x00, /* Local Maximum? */
189 		0x75, 0x80, /* Report Size 128 */
190 		0x95, 0x40, /* Report Count */
191 		0x91, 0x02, /* Output Data */
192 
193 		0x85, 0x03, /* Report ID */
194 		0x19, 0x01, /* Usage Minimum */
195 		0x29, 0x01, /* Usage Maximum */
196 		0x15, 0x00, /* Local Minimum */
197 		0x26, 0xFF, 0x00, /* Local Maximum? */
198 		0x75, 0x08, /* Report Size 8 */
199 		0x95, 0x04, /* Report Count */
200 		0x81, 0x02, /* Input Data */
201 
202 		0x85, 0x04, /* Report ID */
203 		0x19, 0x01, /* Usage Minimum */
204 		0x29, 0x01, /* Usage Maximum */
205 		0x15, 0x00, /* Local Minimum */
206 		0x26, 0xFF, 0x00, /* Local Maximum? */
207 		0x75, 0x08, /* Report Size 8 */
208 		0x95, 0x40, /* Report Count */
209 		0x81, 0x02, /* Input Data */
210 		0xc0
211 	},
212 };
213 
214 static const char sdp_name[] = "Serial Downloader Protocol";
215 
216 /*
217  * static strings, in UTF-8
218  */
219 static struct usb_string strings_sdp_generic[] = {
220 	[0].s = sdp_name,
221 	{  }			/* end of list */
222 };
223 
224 static struct usb_gadget_strings stringtab_sdp_generic = {
225 	.language	= 0x0409,	/* en-us */
226 	.strings	= strings_sdp_generic,
227 };
228 
229 static struct usb_gadget_strings *sdp_generic_strings[] = {
230 	&stringtab_sdp_generic,
231 	NULL,
232 };
233 
sdp_ptr(u32 val)234 static inline void *sdp_ptr(u32 val)
235 {
236 	return (void *)(uintptr_t)val;
237 }
238 
sdp_rx_command_complete(struct usb_ep * ep,struct usb_request * req)239 static void sdp_rx_command_complete(struct usb_ep *ep, struct usb_request *req)
240 {
241 	struct f_sdp *sdp = req->context;
242 	int status = req->status;
243 	u8 *data = req->buf;
244 	u8 report = data[0];
245 
246 	if (status != 0) {
247 		pr_err("Status: %d\n", status);
248 		return;
249 	}
250 
251 	if (report != 1) {
252 		pr_err("Unexpected report %d\n", report);
253 		return;
254 	}
255 
256 	struct sdp_command *cmd = req->buf + 1;
257 
258 	debug("%s: command: %04x, addr: %08x, cnt: %u\n",
259 	      __func__, be16_to_cpu(cmd->cmd),
260 	      be32_to_cpu(cmd->addr), be32_to_cpu(cmd->cnt));
261 
262 	switch (be16_to_cpu(cmd->cmd)) {
263 	case SDP_READ_REGISTER:
264 		sdp->always_send_status = false;
265 		sdp->error_status = 0x0;
266 
267 		sdp->state = SDP_STATE_TX_SEC_CONF;
268 		sdp->dnl_address = be32_to_cpu(cmd->addr);
269 		sdp->dnl_bytes_remaining = be32_to_cpu(cmd->cnt);
270 		sdp->next_state = SDP_STATE_TX_REGISTER;
271 		printf("Reading %d registers at 0x%08x... ",
272 		       sdp->dnl_bytes_remaining, sdp->dnl_address);
273 		break;
274 	case SDP_WRITE_FILE:
275 		sdp->always_send_status = true;
276 		sdp->error_status = SDP_WRITE_FILE_COMPLETE;
277 
278 		sdp->state = SDP_STATE_RX_FILE_DATA;
279 		sdp->dnl_address = be32_to_cpu(cmd->addr);
280 		sdp->dnl_bytes_remaining = be32_to_cpu(cmd->cnt);
281 		sdp->dnl_bytes = sdp->dnl_bytes_remaining;
282 		sdp->next_state = SDP_STATE_IDLE;
283 
284 		printf("Downloading file of size %d to 0x%08x... ",
285 		       sdp->dnl_bytes_remaining, sdp->dnl_address);
286 
287 		break;
288 	case SDP_ERROR_STATUS:
289 		sdp->always_send_status = true;
290 		sdp->error_status = 0;
291 
292 		sdp->state = SDP_STATE_TX_SEC_CONF;
293 		sdp->next_state = SDP_STATE_IDLE;
294 		break;
295 	case SDP_DCD_WRITE:
296 		sdp->always_send_status = true;
297 		sdp->error_status = SDP_WRITE_REGISTER_COMPLETE;
298 
299 		sdp->state = SDP_STATE_RX_DCD_DATA;
300 		sdp->dnl_bytes_remaining = be32_to_cpu(cmd->cnt);
301 		sdp->next_state = SDP_STATE_IDLE;
302 		break;
303 	case SDP_JUMP_ADDRESS:
304 		sdp->always_send_status = false;
305 		sdp->error_status = 0;
306 
307 		sdp->jmp_address = be32_to_cpu(cmd->addr);
308 		sdp->state = SDP_STATE_TX_SEC_CONF;
309 		sdp->next_state = SDP_STATE_JUMP;
310 		break;
311 	case SDP_SKIP_DCD_HEADER:
312 		sdp->always_send_status = true;
313 		sdp->error_status = SDP_SKIP_DCD_HEADER_COMPLETE;
314 
315 		/* Ignore command, DCD not supported anyway */
316 		sdp->state = SDP_STATE_TX_SEC_CONF;
317 		sdp->next_state = SDP_STATE_IDLE;
318 		break;
319 	default:
320 		pr_err("Unknown command: %04x\n", be16_to_cpu(cmd->cmd));
321 	}
322 }
323 
sdp_rx_data_complete(struct usb_ep * ep,struct usb_request * req)324 static void sdp_rx_data_complete(struct usb_ep *ep, struct usb_request *req)
325 {
326 	struct f_sdp *sdp = req->context;
327 	int status = req->status;
328 	u8 *data = req->buf;
329 	u8 report = data[0];
330 	int datalen = req->length - 1;
331 
332 	if (status != 0) {
333 		pr_err("Status: %d\n", status);
334 		return;
335 	}
336 
337 	if (report != 2) {
338 		pr_err("Unexpected report %d\n", report);
339 		return;
340 	}
341 
342 	if (sdp->dnl_bytes_remaining < datalen) {
343 		/*
344 		 * Some USB stacks require to send a complete buffer as
345 		 * specified in the HID descriptor. This leads to longer
346 		 * transfers than the file length, no problem for us.
347 		 */
348 		sdp->dnl_bytes_remaining = 0;
349 	} else {
350 		sdp->dnl_bytes_remaining -= datalen;
351 	}
352 
353 	if (sdp->state == SDP_STATE_RX_FILE_DATA) {
354 		memcpy(sdp_ptr(sdp->dnl_address), req->buf + 1, datalen);
355 		sdp->dnl_address += datalen;
356 	}
357 
358 	if (sdp->dnl_bytes_remaining)
359 		return;
360 
361 #ifndef CONFIG_SPL_BUILD
362 	env_set_hex("filesize", sdp->dnl_bytes);
363 #endif
364 	printf("done\n");
365 
366 	switch (sdp->state) {
367 	case SDP_STATE_RX_FILE_DATA:
368 		sdp->state = SDP_STATE_TX_SEC_CONF;
369 		break;
370 	case SDP_STATE_RX_DCD_DATA:
371 		sdp->state = SDP_STATE_TX_SEC_CONF;
372 		break;
373 	default:
374 		pr_err("Invalid state: %d\n", sdp->state);
375 	}
376 }
377 
sdp_tx_complete(struct usb_ep * ep,struct usb_request * req)378 static void sdp_tx_complete(struct usb_ep *ep, struct usb_request *req)
379 {
380 	struct f_sdp *sdp = req->context;
381 	int status = req->status;
382 
383 	if (status != 0) {
384 		pr_err("Status: %d\n", status);
385 		return;
386 	}
387 
388 	switch (sdp->state) {
389 	case SDP_STATE_TX_SEC_CONF_BUSY:
390 		/* Not all commands require status report */
391 		if (sdp->always_send_status || sdp->error_status)
392 			sdp->state = SDP_STATE_TX_STATUS;
393 		else
394 			sdp->state = sdp->next_state;
395 
396 		break;
397 	case SDP_STATE_TX_STATUS_BUSY:
398 		sdp->state = sdp->next_state;
399 		break;
400 	case SDP_STATE_TX_REGISTER_BUSY:
401 		if (sdp->dnl_bytes_remaining)
402 			sdp->state = SDP_STATE_TX_REGISTER;
403 		else
404 			sdp->state = SDP_STATE_IDLE;
405 		break;
406 	default:
407 		pr_err("Wrong State: %d\n", sdp->state);
408 		sdp->state = SDP_STATE_IDLE;
409 		break;
410 	}
411 	debug("%s complete --> %d, %d/%d\n", ep->name,
412 	      status, req->actual, req->length);
413 }
414 
sdp_setup(struct usb_function * f,const struct usb_ctrlrequest * ctrl)415 static int sdp_setup(struct usb_function *f, const struct usb_ctrlrequest *ctrl)
416 {
417 	struct usb_gadget *gadget = f->config->cdev->gadget;
418 	struct usb_request *req = f->config->cdev->req;
419 	struct f_sdp *sdp = f->config->cdev->req->context;
420 	u16 len = le16_to_cpu(ctrl->wLength);
421 	u16 w_value = le16_to_cpu(ctrl->wValue);
422 	int value = 0;
423 	u8 req_type = ctrl->bRequestType & USB_TYPE_MASK;
424 
425 	debug("w_value: 0x%04x len: 0x%04x\n", w_value, len);
426 	debug("req_type: 0x%02x ctrl->bRequest: 0x%02x sdp->state: %d\n",
427 	      req_type, ctrl->bRequest, sdp->state);
428 
429 	if (req_type == USB_TYPE_STANDARD) {
430 		if (ctrl->bRequest == USB_REQ_GET_DESCRIPTOR) {
431 			/* Send HID report descriptor */
432 			value = min(len, (u16) sizeof(sdp_hid_report));
433 			memcpy(req->buf, &sdp_hid_report, value);
434 			sdp->configuration_done = true;
435 		}
436 	}
437 
438 	if (req_type == USB_TYPE_CLASS) {
439 		int report = w_value & HID_REPORT_ID_MASK;
440 
441 		/* HID (SDP) request */
442 		switch (ctrl->bRequest) {
443 		case HID_REQ_SET_REPORT:
444 			switch (report) {
445 			case 1:
446 				value = SDP_COMMAND_LEN + 1;
447 				req->complete = sdp_rx_command_complete;
448 				break;
449 			case 2:
450 				value = len;
451 				req->complete = sdp_rx_data_complete;
452 				break;
453 			}
454 		}
455 	}
456 
457 	if (value >= 0) {
458 		req->length = value;
459 		req->zero = value < len;
460 		value = usb_ep_queue(gadget->ep0, req, 0);
461 		if (value < 0) {
462 			debug("ep_queue --> %d\n", value);
463 			req->status = 0;
464 		}
465 	}
466 
467 	return value;
468 }
469 
sdp_bind(struct usb_configuration * c,struct usb_function * f)470 static int sdp_bind(struct usb_configuration *c, struct usb_function *f)
471 {
472 	struct usb_gadget *gadget = c->cdev->gadget;
473 	struct usb_composite_dev *cdev = c->cdev;
474 	struct f_sdp *sdp = func_to_sdp(f);
475 	int rv = 0, id;
476 
477 	id = usb_interface_id(c, f);
478 	if (id < 0)
479 		return id;
480 	sdp_intf_runtime.bInterfaceNumber = id;
481 
482 	struct usb_ep *ep;
483 
484 	/* allocate instance-specific endpoints */
485 	ep = usb_ep_autoconfig(gadget, &in_desc);
486 	if (!ep) {
487 		rv = -ENODEV;
488 		goto error;
489 	}
490 
491 	sdp->in_ep = ep; /* Store IN EP for enabling @ setup */
492 
493 	cdev->req->context = sdp;
494 
495 error:
496 	return rv;
497 }
498 
sdp_unbind(struct usb_configuration * c,struct usb_function * f)499 static void sdp_unbind(struct usb_configuration *c, struct usb_function *f)
500 {
501 	free(sdp_func);
502 	sdp_func = NULL;
503 }
504 
alloc_ep_req(struct usb_ep * ep,unsigned length)505 static struct usb_request *alloc_ep_req(struct usb_ep *ep, unsigned length)
506 {
507 	struct usb_request *req;
508 
509 	req = usb_ep_alloc_request(ep, 0);
510 	if (!req)
511 		return req;
512 
513 	req->length = length;
514 	req->buf = memalign(CONFIG_SYS_CACHELINE_SIZE, length);
515 	if (!req->buf) {
516 		usb_ep_free_request(ep, req);
517 		req = NULL;
518 	}
519 
520 	return req;
521 }
522 
523 
sdp_start_ep(struct usb_ep * ep)524 static struct usb_request *sdp_start_ep(struct usb_ep *ep)
525 {
526 	struct usb_request *req;
527 
528 	req = alloc_ep_req(ep, 64);
529 	debug("%s: ep:%p req:%p\n", __func__, ep, req);
530 
531 	if (!req)
532 		return NULL;
533 
534 	memset(req->buf, 0, req->length);
535 	req->complete = sdp_tx_complete;
536 
537 	return req;
538 }
sdp_set_alt(struct usb_function * f,unsigned intf,unsigned alt)539 static int sdp_set_alt(struct usb_function *f, unsigned intf, unsigned alt)
540 {
541 	struct f_sdp *sdp = func_to_sdp(f);
542 	struct usb_composite_dev *cdev = f->config->cdev;
543 	int result;
544 
545 	debug("%s: intf: %d alt: %d\n", __func__, intf, alt);
546 
547 	result = usb_ep_enable(sdp->in_ep, &in_desc);
548 	if (result)
549 		return result;
550 	sdp->in_req = sdp_start_ep(sdp->in_ep);
551 	sdp->in_req->context = sdp;
552 
553 	sdp->in_ep->driver_data = cdev; /* claim */
554 
555 	sdp->altsetting = alt;
556 	sdp->state = SDP_STATE_IDLE;
557 
558 	return 0;
559 }
560 
sdp_get_alt(struct usb_function * f,unsigned intf)561 static int sdp_get_alt(struct usb_function *f, unsigned intf)
562 {
563 	struct f_sdp *sdp = func_to_sdp(f);
564 
565 	return sdp->altsetting;
566 }
567 
sdp_disable(struct usb_function * f)568 static void sdp_disable(struct usb_function *f)
569 {
570 	struct f_sdp *sdp = func_to_sdp(f);
571 
572 	usb_ep_disable(sdp->in_ep);
573 
574 	if (sdp->in_req) {
575 		free(sdp->in_req);
576 		sdp->in_req = NULL;
577 	}
578 }
579 
sdp_bind_config(struct usb_configuration * c)580 static int sdp_bind_config(struct usb_configuration *c)
581 {
582 	int status;
583 
584 	if (!sdp_func) {
585 		sdp_func = memalign(CONFIG_SYS_CACHELINE_SIZE, sizeof(*sdp_func));
586 		if (!sdp_func)
587 			return -ENOMEM;
588 	}
589 
590 	memset(sdp_func, 0, sizeof(*sdp_func));
591 
592 	sdp_func->usb_function.name = "sdp";
593 	sdp_func->usb_function.hs_descriptors = sdp_runtime_descs;
594 	sdp_func->usb_function.descriptors = sdp_runtime_descs;
595 	sdp_func->usb_function.bind = sdp_bind;
596 	sdp_func->usb_function.unbind = sdp_unbind;
597 	sdp_func->usb_function.set_alt = sdp_set_alt;
598 	sdp_func->usb_function.get_alt = sdp_get_alt;
599 	sdp_func->usb_function.disable = sdp_disable;
600 	sdp_func->usb_function.strings = sdp_generic_strings;
601 	sdp_func->usb_function.setup = sdp_setup;
602 
603 	status = usb_add_function(c, &sdp_func->usb_function);
604 
605 	return status;
606 }
607 
sdp_init(int controller_index)608 int sdp_init(int controller_index)
609 {
610 	printf("SDP: initialize...\n");
611 	while (!sdp_func->configuration_done) {
612 		if (ctrlc()) {
613 			puts("\rCTRL+C - Operation aborted.\n");
614 			return 1;
615 		}
616 
617 		WATCHDOG_RESET();
618 		usb_gadget_handle_interrupts(controller_index);
619 	}
620 
621 	return 0;
622 }
623 
sdp_jump_imxheader(void * address)624 static u32 sdp_jump_imxheader(void *address)
625 {
626 	flash_header_v2_t *headerv2 = address;
627 	ulong (*entry)(void);
628 
629 	if (headerv2->header.tag != IVT_HEADER_TAG) {
630 		printf("Header Tag is not an IMX image\n");
631 		return SDP_ERROR_IMXHEADER;
632 	}
633 
634 	printf("Jumping to 0x%08x\n", headerv2->entry);
635 	entry = sdp_ptr(headerv2->entry);
636 	entry();
637 
638 	/* The image probably never returns hence we won't reach that point */
639 	return 0;
640 }
641 
642 #ifdef CONFIG_SPL_BUILD
643 #ifdef CONFIG_SPL_LOAD_FIT
sdp_fit_read(struct spl_load_info * load,ulong sector,ulong count,void * buf)644 static ulong sdp_fit_read(struct spl_load_info *load, ulong sector,
645 			  ulong count, void *buf)
646 {
647 	debug("%s: sector %lx, count %lx, buf %lx\n",
648 	      __func__, sector, count, (ulong)buf);
649 	memcpy(buf, (void *)(load->dev + sector), count);
650 	return count;
651 }
652 #endif
653 #endif
654 
sdp_handle_in_ep(struct spl_image_info * spl_image)655 static void sdp_handle_in_ep(struct spl_image_info *spl_image)
656 {
657 	u8 *data = sdp_func->in_req->buf;
658 	u32 status;
659 	int datalen;
660 
661 	switch (sdp_func->state) {
662 	case SDP_STATE_TX_SEC_CONF:
663 		debug("Report 3: HAB security\n");
664 		data[0] = 3;
665 
666 		status = SDP_SECURITY_OPEN;
667 		memcpy(&data[1], &status, 4);
668 		sdp_func->in_req->length = 5;
669 		usb_ep_queue(sdp_func->in_ep, sdp_func->in_req, 0);
670 		sdp_func->state = SDP_STATE_TX_SEC_CONF_BUSY;
671 		break;
672 
673 	case SDP_STATE_TX_STATUS:
674 		debug("Report 4: Status\n");
675 		data[0] = 4;
676 
677 		memcpy(&data[1], &sdp_func->error_status, 4);
678 		sdp_func->in_req->length = 65;
679 		usb_ep_queue(sdp_func->in_ep, sdp_func->in_req, 0);
680 		sdp_func->state = SDP_STATE_TX_STATUS_BUSY;
681 		break;
682 	case SDP_STATE_TX_REGISTER:
683 		debug("Report 4: Register Values\n");
684 		data[0] = 4;
685 
686 		datalen = sdp_func->dnl_bytes_remaining;
687 
688 		if (datalen > 64)
689 			datalen = 64;
690 
691 		memcpy(&data[1], sdp_ptr(sdp_func->dnl_address), datalen);
692 		sdp_func->in_req->length = 65;
693 
694 		sdp_func->dnl_bytes_remaining -= datalen;
695 		sdp_func->dnl_address += datalen;
696 
697 		usb_ep_queue(sdp_func->in_ep, sdp_func->in_req, 0);
698 		sdp_func->state = SDP_STATE_TX_REGISTER_BUSY;
699 		break;
700 	case SDP_STATE_JUMP:
701 		printf("Jumping to header at 0x%08x\n", sdp_func->jmp_address);
702 		status = sdp_jump_imxheader(sdp_ptr(sdp_func->jmp_address));
703 
704 		/* If imx header fails, try some U-Boot specific headers */
705 		if (status) {
706 #ifdef CONFIG_SPL_BUILD
707 			image_header_t *header =
708 				sdp_ptr(sdp_func->jmp_address);
709 #ifdef CONFIG_SPL_LOAD_FIT
710 			if (image_get_magic(header) == FDT_MAGIC) {
711 				struct spl_load_info load;
712 
713 				debug("Found FIT\n");
714 				load.dev = header;
715 				load.bl_len = 1;
716 				load.read = sdp_fit_read;
717 				spl_load_simple_fit(spl_image, &load, 0,
718 						    header);
719 
720 				return;
721 			}
722 #endif
723 			/* In SPL, allow jumps to U-Boot images */
724 			struct spl_image_info spl_image = {};
725 			spl_parse_image_header(&spl_image, header);
726 			jump_to_image_no_args(&spl_image);
727 #else
728 			/* In U-Boot, allow jumps to scripts */
729 			source(sdp_func->jmp_address, "script@1");
730 #endif
731 		}
732 
733 		sdp_func->next_state = SDP_STATE_IDLE;
734 		sdp_func->error_status = status;
735 
736 		/* Only send Report 4 if there was an error */
737 		if (status)
738 			sdp_func->state = SDP_STATE_TX_STATUS;
739 		else
740 			sdp_func->state = SDP_STATE_IDLE;
741 		break;
742 	default:
743 		break;
744 	};
745 }
746 
747 #ifndef CONFIG_SPL_BUILD
sdp_handle(int controller_index)748 int sdp_handle(int controller_index)
749 #else
750 int spl_sdp_handle(int controller_index, struct spl_image_info *spl_image)
751 #endif
752 {
753 	printf("SDP: handle requests...\n");
754 	while (1) {
755 		if (ctrlc()) {
756 			puts("\rCTRL+C - Operation aborted.\n");
757 			return -EINVAL;
758 		}
759 
760 #ifdef CONFIG_SPL_BUILD
761 		if (spl_image->flags & SPL_FIT_FOUND)
762 			return 0;
763 #endif
764 
765 		WATCHDOG_RESET();
766 		usb_gadget_handle_interrupts(controller_index);
767 
768 #ifdef CONFIG_SPL_BUILD
769 		sdp_handle_in_ep(spl_image);
770 #else
771 		sdp_handle_in_ep(NULL);
772 #endif
773 	}
774 }
775 
sdp_add(struct usb_configuration * c)776 int sdp_add(struct usb_configuration *c)
777 {
778 	int id;
779 
780 	id = usb_string_id(c->cdev);
781 	if (id < 0)
782 		return id;
783 	strings_sdp_generic[0].id = id;
784 	sdp_intf_runtime.iInterface = id;
785 
786 	debug("%s: cdev: %p gadget: %p gadget->ep0: %p\n", __func__,
787 	      c->cdev, c->cdev->gadget, c->cdev->gadget->ep0);
788 
789 	return sdp_bind_config(c);
790 }
791 
792 DECLARE_GADGET_BIND_CALLBACK(usb_dnl_sdp, sdp_add);
793