g_dnl.c 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307
  1. /*
  2. * g_dnl.c -- USB Downloader Gadget
  3. *
  4. * Copyright (C) 2012 Samsung Electronics
  5. * Lukasz Majewski <l.majewski@samsung.com>
  6. *
  7. * SPDX-License-Identifier: GPL-2.0+
  8. */
  9. #include <common.h>
  10. #include <malloc.h>
  11. #include <mmc.h>
  12. #include <part.h>
  13. #include <usb.h>
  14. #include <g_dnl.h>
  15. #include <usb_mass_storage.h>
  16. #include <dfu.h>
  17. #include <thor.h>
  18. #include <env_callback.h>
  19. #include "gadget_chips.h"
  20. #include "composite.c"
  21. /*
  22. * One needs to define the following:
  23. * CONFIG_G_DNL_VENDOR_NUM
  24. * CONFIG_G_DNL_PRODUCT_NUM
  25. * CONFIG_G_DNL_MANUFACTURER
  26. * at e.g. ./configs/<board>_defconfig
  27. */
  28. #define STRING_MANUFACTURER 25
  29. #define STRING_PRODUCT 2
  30. /* Index of String Descriptor describing this configuration */
  31. #define STRING_USBDOWN 2
  32. /* Index of String serial */
  33. #define STRING_SERIAL 3
  34. #define MAX_STRING_SERIAL 32
  35. /* Number of supported configurations */
  36. #define CONFIGURATION_NUMBER 1
  37. #define DRIVER_VERSION "usb_dnl 2.0"
  38. static const char product[] = "USB download gadget";
  39. static char g_dnl_serial[MAX_STRING_SERIAL];
  40. static const char manufacturer[] = CONFIG_G_DNL_MANUFACTURER;
  41. void g_dnl_set_serialnumber(char *s)
  42. {
  43. memset(g_dnl_serial, 0, MAX_STRING_SERIAL);
  44. if (strlen(s) < MAX_STRING_SERIAL)
  45. strncpy(g_dnl_serial, s, strlen(s));
  46. }
  47. static struct usb_device_descriptor device_desc = {
  48. .bLength = sizeof device_desc,
  49. .bDescriptorType = USB_DT_DEVICE,
  50. .bcdUSB = __constant_cpu_to_le16(0x0200),
  51. .bDeviceClass = USB_CLASS_PER_INTERFACE,
  52. .bDeviceSubClass = 0, /*0x02:CDC-modem , 0x00:CDC-serial*/
  53. .idVendor = __constant_cpu_to_le16(CONFIG_G_DNL_VENDOR_NUM),
  54. .idProduct = __constant_cpu_to_le16(CONFIG_G_DNL_PRODUCT_NUM),
  55. .iProduct = STRING_PRODUCT,
  56. .iSerialNumber = STRING_SERIAL,
  57. .bNumConfigurations = 1,
  58. };
  59. /*
  60. * static strings, in UTF-8
  61. * IDs for those strings are assigned dynamically at g_dnl_bind()
  62. */
  63. static struct usb_string g_dnl_string_defs[] = {
  64. {.s = manufacturer},
  65. {.s = product},
  66. {.s = g_dnl_serial},
  67. { } /* end of list */
  68. };
  69. static struct usb_gadget_strings g_dnl_string_tab = {
  70. .language = 0x0409, /* en-us */
  71. .strings = g_dnl_string_defs,
  72. };
  73. static struct usb_gadget_strings *g_dnl_composite_strings[] = {
  74. &g_dnl_string_tab,
  75. NULL,
  76. };
  77. static int g_dnl_unbind(struct usb_composite_dev *cdev)
  78. {
  79. struct usb_gadget *gadget = cdev->gadget;
  80. debug("%s: calling usb_gadget_disconnect for "
  81. "controller '%s'\n", __func__, gadget->name);
  82. usb_gadget_disconnect(gadget);
  83. return 0;
  84. }
  85. static inline struct g_dnl_bind_callback *g_dnl_bind_callback_first(void)
  86. {
  87. return ll_entry_start(struct g_dnl_bind_callback,
  88. g_dnl_bind_callbacks);
  89. }
  90. static inline struct g_dnl_bind_callback *g_dnl_bind_callback_end(void)
  91. {
  92. return ll_entry_end(struct g_dnl_bind_callback,
  93. g_dnl_bind_callbacks);
  94. }
  95. static int g_dnl_do_config(struct usb_configuration *c)
  96. {
  97. const char *s = c->cdev->driver->name;
  98. struct g_dnl_bind_callback *callback = g_dnl_bind_callback_first();
  99. debug("%s: configuration: 0x%p composite dev: 0x%p\n",
  100. __func__, c, c->cdev);
  101. for (; callback != g_dnl_bind_callback_end(); callback++)
  102. if (!strcmp(s, callback->usb_function_name))
  103. return callback->fptr(c);
  104. return -ENODEV;
  105. }
  106. static int g_dnl_config_register(struct usb_composite_dev *cdev)
  107. {
  108. struct usb_configuration *config;
  109. const char *name = "usb_dnload";
  110. config = memalign(CONFIG_SYS_CACHELINE_SIZE, sizeof(*config));
  111. if (!config)
  112. return -ENOMEM;
  113. memset(config, 0, sizeof(*config));
  114. config->label = name;
  115. config->bmAttributes = USB_CONFIG_ATT_ONE | USB_CONFIG_ATT_SELFPOWER;
  116. config->bConfigurationValue = CONFIGURATION_NUMBER;
  117. config->iConfiguration = STRING_USBDOWN;
  118. config->bind = g_dnl_do_config;
  119. return usb_add_config(cdev, config);
  120. }
  121. __weak
  122. int board_usb_init(int index, enum usb_init_type init)
  123. {
  124. return 0;
  125. }
  126. __weak
  127. int board_usb_cleanup(int index, enum usb_init_type init)
  128. {
  129. return 0;
  130. }
  131. __weak
  132. int g_dnl_bind_fixup(struct usb_device_descriptor *dev, const char *name)
  133. {
  134. return 0;
  135. }
  136. __weak int g_dnl_get_board_bcd_device_number(int gcnum)
  137. {
  138. return gcnum;
  139. }
  140. __weak int g_dnl_board_usb_cable_connected(void)
  141. {
  142. return -EOPNOTSUPP;
  143. }
  144. static bool g_dnl_detach_request;
  145. bool g_dnl_detach(void)
  146. {
  147. return g_dnl_detach_request;
  148. }
  149. void g_dnl_trigger_detach(void)
  150. {
  151. g_dnl_detach_request = true;
  152. }
  153. void g_dnl_clear_detach(void)
  154. {
  155. g_dnl_detach_request = false;
  156. }
  157. static int g_dnl_get_bcd_device_number(struct usb_composite_dev *cdev)
  158. {
  159. struct usb_gadget *gadget = cdev->gadget;
  160. int gcnum;
  161. gcnum = usb_gadget_controller_number(gadget);
  162. if (gcnum > 0)
  163. gcnum += 0x200;
  164. return g_dnl_get_board_bcd_device_number(gcnum);
  165. }
  166. /**
  167. * Update internal serial number variable when the "serial#" env var changes.
  168. *
  169. * Handle all cases, even when flags == H_PROGRAMMATIC or op == env_op_delete.
  170. */
  171. static int on_serialno(const char *name, const char *value, enum env_op op,
  172. int flags)
  173. {
  174. g_dnl_set_serialnumber((char *)value);
  175. return 0;
  176. }
  177. U_BOOT_ENV_CALLBACK(serialno, on_serialno);
  178. static int g_dnl_bind(struct usb_composite_dev *cdev)
  179. {
  180. struct usb_gadget *gadget = cdev->gadget;
  181. int id, ret;
  182. int gcnum;
  183. debug("%s: gadget: 0x%p cdev: 0x%p\n", __func__, gadget, cdev);
  184. id = usb_string_id(cdev);
  185. if (id < 0)
  186. return id;
  187. g_dnl_string_defs[0].id = id;
  188. device_desc.iManufacturer = id;
  189. id = usb_string_id(cdev);
  190. if (id < 0)
  191. return id;
  192. g_dnl_string_defs[1].id = id;
  193. device_desc.iProduct = id;
  194. id = usb_string_id(cdev);
  195. if (id < 0)
  196. return id;
  197. g_dnl_string_defs[2].id = id;
  198. device_desc.iSerialNumber = id;
  199. g_dnl_bind_fixup(&device_desc, cdev->driver->name);
  200. ret = g_dnl_config_register(cdev);
  201. if (ret)
  202. goto error;
  203. gcnum = g_dnl_get_bcd_device_number(cdev);
  204. if (gcnum >= 0)
  205. device_desc.bcdDevice = cpu_to_le16(gcnum);
  206. else {
  207. debug("%s: controller '%s' not recognized\n",
  208. __func__, gadget->name);
  209. device_desc.bcdDevice = __constant_cpu_to_le16(0x9999);
  210. }
  211. debug("%s: calling usb_gadget_connect for "
  212. "controller '%s'\n", __func__, gadget->name);
  213. usb_gadget_connect(gadget);
  214. return 0;
  215. error:
  216. g_dnl_unbind(cdev);
  217. return -ENOMEM;
  218. }
  219. static struct usb_composite_driver g_dnl_driver = {
  220. .name = NULL,
  221. .dev = &device_desc,
  222. .strings = g_dnl_composite_strings,
  223. .bind = g_dnl_bind,
  224. .unbind = g_dnl_unbind,
  225. };
  226. /*
  227. * NOTICE:
  228. * Registering via USB function name won't be necessary after rewriting
  229. * g_dnl to support multiple USB functions.
  230. */
  231. int g_dnl_register(const char *name)
  232. {
  233. int ret;
  234. debug("%s: g_dnl_driver.name = %s\n", __func__, name);
  235. g_dnl_driver.name = name;
  236. ret = usb_composite_register(&g_dnl_driver);
  237. if (ret) {
  238. printf("%s: failed!, error: %d\n", __func__, ret);
  239. return ret;
  240. }
  241. return 0;
  242. }
  243. void g_dnl_unregister(void)
  244. {
  245. usb_composite_unregister(&g_dnl_driver);
  246. }