1 // SPDX-License-Identifier: GPL-2.0+
4 * Denis Peter, MPL AG Switzerland
6 * Part of this source has been derived from the Linux USB
16 #include <stdio_dev.h>
18 #include <asm/byteorder.h>
23 * If overwrite_console returns 1, the stdin, stderr and stdout
24 * are switched to the serial port, else the settings in the
25 * environment are used
27 #ifdef CONFIG_SYS_CONSOLE_OVERWRITE_ROUTINE
28 extern int overwrite_console(void);
30 int overwrite_console(void)
36 /* Keyboard sampling rate */
37 #define REPEAT_RATE 40 /* 40msec -> 25cps */
38 #define REPEAT_DELAY 10 /* 10 x REPEAT_RATE = 400msec */
41 #define CAPS_LOCK 0x39
42 #define SCROLL_LOCK 0x47
45 #define LEFT_CNTR (1 << 0)
46 #define LEFT_SHIFT (1 << 1)
47 #define LEFT_ALT (1 << 2)
48 #define LEFT_GUI (1 << 3)
49 #define RIGHT_CNTR (1 << 4)
50 #define RIGHT_SHIFT (1 << 5)
51 #define RIGHT_ALT (1 << 6)
52 #define RIGHT_GUI (1 << 7)
54 /* Size of the keyboard buffer */
55 #define USB_KBD_BUFFER_LEN 0x20
58 #define DEVNAME "usbkbd"
61 static const unsigned char usb_kbd_numkey[] = {
62 '1', '2', '3', '4', '5', '6', '7', '8', '9', '0',
63 '\r', 0x1b, '\b', '\t', ' ', '-', '=', '[', ']',
64 '\\', '#', ';', '\'', '`', ',', '.', '/'
66 static const unsigned char usb_kbd_numkey_shifted[] = {
67 '!', '@', '#', '$', '%', '^', '&', '*', '(', ')',
68 '\r', 0x1b, '\b', '\t', ' ', '_', '+', '{', '}',
69 '|', '~', ':', '"', '~', '<', '>', '?'
72 static const unsigned char usb_kbd_num_keypad[] = {
73 '/', '*', '-', '+', '\r',
74 '1', '2', '3', '4', '5', '6', '7', '8', '9', '0',
78 static const u8 usb_special_keys[] = {
79 #ifdef CONFIG_USB_KEYBOARD_FN_KEYS
80 '2', 'H', '5', '3', 'F', '6', 'C', 'D', 'B', 'A'
87 * NOTE: It's important for the NUM, CAPS, SCROLL-lock bits to be in this
88 * order. See usb_kbd_setled() function!
90 #define USB_KBD_NUMLOCK (1 << 0)
91 #define USB_KBD_CAPSLOCK (1 << 1)
92 #define USB_KBD_SCROLLLOCK (1 << 2)
93 #define USB_KBD_CTRL (1 << 3)
95 #define USB_KBD_LEDMASK \
96 (USB_KBD_NUMLOCK | USB_KBD_CAPSLOCK | USB_KBD_SCROLLLOCK)
99 * USB Keyboard reports are 8 bytes in boot protocol.
100 * Appendix B of HID Device Class Definition 1.11
102 #define USB_KBD_BOOT_REPORT_SIZE 8
104 struct usb_kbd_pdata {
105 unsigned long intpipe;
108 unsigned long last_report;
109 struct int_queue *intq;
111 uint32_t repeat_delay;
113 uint32_t usb_in_pointer;
114 uint32_t usb_out_pointer;
115 uint8_t usb_kbd_buffer[USB_KBD_BUFFER_LEN];
118 uint8_t old[USB_KBD_BOOT_REPORT_SIZE];
123 extern int __maybe_unused net_busy_flag;
125 /* The period of time between two calls of usb_kbd_testc(). */
126 static unsigned long __maybe_unused kbd_testc_tms;
128 /* Puts character in the queue and sets up the in and out pointer. */
129 static void usb_kbd_put_queue(struct usb_kbd_pdata *data, u8 c)
131 if (data->usb_in_pointer == USB_KBD_BUFFER_LEN - 1) {
132 /* Check for buffer full. */
133 if (data->usb_out_pointer == 0)
136 data->usb_in_pointer = 0;
138 /* Check for buffer full. */
139 if (data->usb_in_pointer == data->usb_out_pointer - 1)
142 data->usb_in_pointer++;
145 data->usb_kbd_buffer[data->usb_in_pointer] = c;
149 * Set the LEDs. Since this is used in the irq routine, the control job is
150 * issued with a timeout of 0. This means, that the job is queued without
151 * waiting for job completion.
153 static void usb_kbd_setled(struct usb_device *dev)
155 struct usb_interface *iface = &dev->config.if_desc[0];
156 struct usb_kbd_pdata *data = dev->privptr;
157 ALLOC_ALIGN_BUFFER(uint32_t, leds, 1, USB_DMA_MINALIGN);
159 *leds = data->flags & USB_KBD_LEDMASK;
160 usb_control_msg(dev, usb_sndctrlpipe(dev, 0),
161 USB_REQ_SET_REPORT, USB_TYPE_CLASS | USB_RECIP_INTERFACE,
162 0x200, iface->desc.bInterfaceNumber, leds, 1, 0);
165 #define CAPITAL_MASK 0x20
166 /* Translate the scancode in ASCII */
167 static int usb_kbd_translate(struct usb_kbd_pdata *data, unsigned char scancode,
168 unsigned char modifier, int pressed)
174 data->repeat_delay = 0;
179 data->repeat_delay++;
180 if (data->repeat_delay < REPEAT_DELAY)
183 data->repeat_delay = REPEAT_DELAY;
186 /* Alphanumeric values */
187 if ((scancode > 3) && (scancode <= 0x1d)) {
188 keycode = scancode - 4 + 'a';
190 if (data->flags & USB_KBD_CAPSLOCK)
191 keycode &= ~CAPITAL_MASK;
193 if (modifier & (LEFT_SHIFT | RIGHT_SHIFT)) {
194 /* Handle CAPSLock + Shift pressed simultaneously */
195 if (keycode & CAPITAL_MASK)
196 keycode &= ~CAPITAL_MASK;
198 keycode |= CAPITAL_MASK;
202 if ((scancode > 0x1d) && (scancode < 0x39)) {
204 if (modifier & (LEFT_SHIFT | RIGHT_SHIFT))
205 keycode = usb_kbd_numkey_shifted[scancode - 0x1e];
207 keycode = usb_kbd_numkey[scancode - 0x1e];
211 if ((scancode >= 0x54) && (scancode <= 0x67))
212 keycode = usb_kbd_num_keypad[scancode - 0x54];
214 if (data->flags & USB_KBD_CTRL)
215 keycode = scancode - 0x3;
218 if (scancode == NUM_LOCK) {
219 data->flags ^= USB_KBD_NUMLOCK;
223 if (scancode == CAPS_LOCK) {
224 data->flags ^= USB_KBD_CAPSLOCK;
227 if (scancode == SCROLL_LOCK) {
228 data->flags ^= USB_KBD_SCROLLLOCK;
233 /* Report keycode if any */
235 debug("%c", keycode);
236 usb_kbd_put_queue(data, keycode);
240 #ifdef CONFIG_USB_KEYBOARD_FN_KEYS
241 if (scancode < 0x3a || scancode > 0x52 ||
242 scancode == 0x46 || scancode == 0x47)
245 usb_kbd_put_queue(data, 0x1b);
246 if (scancode < 0x3e) {
248 usb_kbd_put_queue(data, 0x4f);
249 usb_kbd_put_queue(data, scancode - 0x3a + 'P');
252 usb_kbd_put_queue(data, '[');
253 if (scancode < 0x42) {
255 usb_kbd_put_queue(data, '1');
256 if (scancode == 0x3e)
258 keycode = scancode - 0x3f + '7';
259 } else if (scancode < 0x49) {
261 usb_kbd_put_queue(data, '2');
264 keycode = scancode - 0x42 + '0';
267 * INSERT, HOME, PAGE UP, DELETE, END, PAGE DOWN,
268 * RIGHT, LEFT, DOWN, UP
270 keycode = usb_special_keys[scancode - 0x49];
272 usb_kbd_put_queue(data, keycode);
273 if (scancode < 0x4f && scancode != 0x4a && scancode != 0x4d)
274 usb_kbd_put_queue(data, '~');
277 /* Left, Right, Up, Down */
278 if (scancode > 0x4e && scancode < 0x53) {
279 usb_kbd_put_queue(data, 0x1b);
280 usb_kbd_put_queue(data, '[');
281 usb_kbd_put_queue(data, usb_special_keys[scancode - 0x4f]);
285 #endif /* CONFIG_USB_KEYBOARD_FN_KEYS */
288 static uint32_t usb_kbd_service_key(struct usb_device *dev, int i, int up)
291 struct usb_kbd_pdata *data = dev->privptr;
304 (memscan(new + 2, old[i], USB_KBD_BOOT_REPORT_SIZE - 2) ==
305 new + USB_KBD_BOOT_REPORT_SIZE)) {
306 res |= usb_kbd_translate(data, old[i], data->new[0], up);
312 /* Interrupt service routine */
313 static int usb_kbd_irq_worker(struct usb_device *dev)
315 struct usb_kbd_pdata *data = dev->privptr;
318 /* No combo key pressed */
319 if (data->new[0] == 0x00)
320 data->flags &= ~USB_KBD_CTRL;
321 /* Left or Right Ctrl pressed */
322 else if ((data->new[0] == LEFT_CNTR) || (data->new[0] == RIGHT_CNTR))
323 data->flags |= USB_KBD_CTRL;
325 for (i = 2; i < USB_KBD_BOOT_REPORT_SIZE; i++) {
326 res |= usb_kbd_service_key(dev, i, 0);
327 res |= usb_kbd_service_key(dev, i, 1);
330 /* Key is still pressed */
331 if ((data->new[2] > 3) && (data->old[2] == data->new[2]))
332 res |= usb_kbd_translate(data, data->new[2], data->new[0], 2);
337 memcpy(data->old, data->new, USB_KBD_BOOT_REPORT_SIZE);
342 /* Keyboard interrupt handler */
343 static int usb_kbd_irq(struct usb_device *dev)
345 if ((dev->irq_status != 0) ||
346 (dev->irq_act_len != USB_KBD_BOOT_REPORT_SIZE)) {
347 debug("USB KBD: Error %lX, len %d\n",
348 dev->irq_status, dev->irq_act_len);
352 return usb_kbd_irq_worker(dev);
355 /* Interrupt polling */
356 static inline void usb_kbd_poll_for_event(struct usb_device *dev)
358 #if defined(CONFIG_SYS_USB_EVENT_POLL)
359 struct usb_kbd_pdata *data = dev->privptr;
361 /* Submit an interrupt transfer request */
362 if (usb_int_msg(dev, data->intpipe, &data->new[0],
363 data->intpktsize, data->intinterval, true) >= 0)
364 usb_kbd_irq_worker(dev);
365 #elif defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP) || \
366 defined(CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE)
367 #if defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP)
368 struct usb_interface *iface;
369 struct usb_kbd_pdata *data = dev->privptr;
370 iface = &dev->config.if_desc[0];
371 usb_get_report(dev, iface->desc.bInterfaceNumber,
372 1, 0, data->new, USB_KBD_BOOT_REPORT_SIZE);
373 if (memcmp(data->old, data->new, USB_KBD_BOOT_REPORT_SIZE)) {
374 usb_kbd_irq_worker(dev);
376 struct usb_kbd_pdata *data = dev->privptr;
377 if (poll_int_queue(dev, data->intq)) {
378 usb_kbd_irq_worker(dev);
379 /* We've consumed all queued int packets, create new */
380 destroy_int_queue(dev, data->intq);
381 data->intq = create_int_queue(dev, data->intpipe, 1,
382 USB_KBD_BOOT_REPORT_SIZE, data->new,
385 data->last_report = get_timer(0);
386 /* Repeat last usb hid report every REPEAT_RATE ms for keyrepeat */
387 } else if (data->last_report != -1 &&
388 get_timer(data->last_report) > REPEAT_RATE) {
389 usb_kbd_irq_worker(dev);
390 data->last_report = get_timer(0);
395 /* test if a character is in the queue */
396 static int usb_kbd_testc(struct stdio_dev *sdev)
398 struct stdio_dev *dev;
399 struct usb_device *usb_kbd_dev;
400 struct usb_kbd_pdata *data;
402 #ifdef CONFIG_CMD_NET
404 * If net_busy_flag is 1, NET transfer is running,
405 * then we check key-pressed every second (first check may be
406 * less than 1 second) to improve TFTP booting performance.
408 if (net_busy_flag && (get_timer(kbd_testc_tms) < CONFIG_SYS_HZ))
410 kbd_testc_tms = get_timer(0);
412 dev = stdio_get_by_name(sdev->name);
413 usb_kbd_dev = (struct usb_device *)dev->priv;
414 data = usb_kbd_dev->privptr;
416 usb_kbd_poll_for_event(usb_kbd_dev);
418 return !(data->usb_in_pointer == data->usb_out_pointer);
421 /* gets the character from the queue */
422 static int usb_kbd_getc(struct stdio_dev *sdev)
424 struct stdio_dev *dev;
425 struct usb_device *usb_kbd_dev;
426 struct usb_kbd_pdata *data;
428 dev = stdio_get_by_name(sdev->name);
429 usb_kbd_dev = (struct usb_device *)dev->priv;
430 data = usb_kbd_dev->privptr;
432 while (data->usb_in_pointer == data->usb_out_pointer) {
434 usb_kbd_poll_for_event(usb_kbd_dev);
437 if (data->usb_out_pointer == USB_KBD_BUFFER_LEN - 1)
438 data->usb_out_pointer = 0;
440 data->usb_out_pointer++;
442 return data->usb_kbd_buffer[data->usb_out_pointer];
445 /* probes the USB device dev for keyboard type. */
446 static int usb_kbd_probe_dev(struct usb_device *dev, unsigned int ifnum)
448 struct usb_interface *iface;
449 struct usb_endpoint_descriptor *ep;
450 struct usb_kbd_pdata *data;
452 if (dev->descriptor.bNumConfigurations != 1)
455 iface = &dev->config.if_desc[ifnum];
457 if (iface->desc.bInterfaceClass != USB_CLASS_HID)
460 if (iface->desc.bInterfaceSubClass != USB_SUB_HID_BOOT)
463 if (iface->desc.bInterfaceProtocol != USB_PROT_HID_KEYBOARD)
466 if (iface->desc.bNumEndpoints != 1)
469 ep = &iface->ep_desc[0];
471 /* Check if endpoint 1 is interrupt endpoint */
472 if (!(ep->bEndpointAddress & 0x80))
475 if ((ep->bmAttributes & 3) != 3)
478 debug("USB KBD: found set protocol...\n");
480 data = malloc(sizeof(struct usb_kbd_pdata));
482 printf("USB KBD: Error allocating private data\n");
486 /* Clear private data */
487 memset(data, 0, sizeof(struct usb_kbd_pdata));
489 /* allocate input buffer aligned and sized to USB DMA alignment */
490 data->new = memalign(USB_DMA_MINALIGN,
491 roundup(USB_KBD_BOOT_REPORT_SIZE, USB_DMA_MINALIGN));
493 /* Insert private data into USB device structure */
496 /* Set IRQ handler */
497 dev->irq_handle = usb_kbd_irq;
499 data->intpipe = usb_rcvintpipe(dev, ep->bEndpointAddress);
500 data->intpktsize = min(usb_maxpacket(dev, data->intpipe),
501 USB_KBD_BOOT_REPORT_SIZE);
502 data->intinterval = ep->bInterval;
503 data->last_report = -1;
505 /* We found a USB Keyboard, install it. */
506 usb_set_protocol(dev, iface->desc.bInterfaceNumber, 0);
508 debug("USB KBD: found set idle...\n");
509 #if !defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP) && \
510 !defined(CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE)
511 usb_set_idle(dev, iface->desc.bInterfaceNumber, REPEAT_RATE / 4, 0);
513 usb_set_idle(dev, iface->desc.bInterfaceNumber, 0, 0);
516 debug("USB KBD: enable interrupt pipe...\n");
517 #ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
518 data->intq = create_int_queue(dev, data->intpipe, 1,
519 USB_KBD_BOOT_REPORT_SIZE, data->new,
522 #elif defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP)
523 if (usb_get_report(dev, iface->desc.bInterfaceNumber,
524 1, 0, data->new, USB_KBD_BOOT_REPORT_SIZE) < 0) {
526 if (usb_int_msg(dev, data->intpipe, data->new, data->intpktsize,
527 data->intinterval, false) < 0) {
529 printf("Failed to get keyboard state from device %04x:%04x\n",
530 dev->descriptor.idVendor, dev->descriptor.idProduct);
531 /* Abort, we don't want to use that non-functional keyboard. */
539 static int probe_usb_keyboard(struct usb_device *dev)
542 struct stdio_dev usb_kbd_dev;
545 /* Try probing the keyboard */
546 if (usb_kbd_probe_dev(dev, 0) != 1)
549 /* Register the keyboard */
550 debug("USB KBD: register.\n");
551 memset(&usb_kbd_dev, 0, sizeof(struct stdio_dev));
552 strcpy(usb_kbd_dev.name, DEVNAME);
553 usb_kbd_dev.flags = DEV_FLAGS_INPUT;
554 usb_kbd_dev.getc = usb_kbd_getc;
555 usb_kbd_dev.tstc = usb_kbd_testc;
556 usb_kbd_dev.priv = (void *)dev;
557 error = stdio_register(&usb_kbd_dev);
561 stdinname = env_get("stdin");
562 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
563 error = iomux_doenv(stdin, stdinname);
567 /* Check if this is the standard input device. */
568 if (strcmp(stdinname, DEVNAME))
571 /* Reassign the console */
572 if (overwrite_console())
575 error = console_assign(stdin, DEVNAME);
583 #if !CONFIG_IS_ENABLED(DM_USB)
584 /* Search for keyboard and register it if found. */
585 int drv_usb_kbd_init(void)
589 debug("%s: Probing for keyboard\n", __func__);
590 /* Scan all USB Devices */
591 for (i = 0; i < USB_MAX_DEVICE; i++) {
592 struct usb_device *dev;
594 /* Get USB device. */
595 dev = usb_get_dev_index(i);
599 if (dev->devnum == -1)
602 error = probe_usb_keyboard(dev);
605 if (error && error != -ENOENT)
609 /* No USB Keyboard found */
613 /* Deregister the keyboard. */
614 int usb_kbd_deregister(int force)
616 #if CONFIG_IS_ENABLED(SYS_STDIO_DEREGISTER)
617 struct stdio_dev *dev;
618 struct usb_device *usb_kbd_dev;
619 struct usb_kbd_pdata *data;
621 dev = stdio_get_by_name(DEVNAME);
623 usb_kbd_dev = (struct usb_device *)dev->priv;
624 data = usb_kbd_dev->privptr;
625 if (stdio_deregister_dev(dev, force) != 0)
627 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
628 if (iomux_doenv(stdin, env_get("stdin")) != 0)
631 #ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
632 destroy_int_queue(usb_kbd_dev, data->intq);
646 #if CONFIG_IS_ENABLED(DM_USB)
648 static int usb_kbd_probe(struct udevice *dev)
650 struct usb_device *udev = dev_get_parent_priv(dev);
652 return probe_usb_keyboard(udev);
655 static int usb_kbd_remove(struct udevice *dev)
657 struct usb_device *udev = dev_get_parent_priv(dev);
658 struct usb_kbd_pdata *data;
659 struct stdio_dev *sdev;
662 sdev = stdio_get_by_name(DEVNAME);
667 data = udev->privptr;
668 if (stdio_deregister_dev(sdev, true)) {
672 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
673 if (iomux_doenv(stdin, env_get("stdin"))) {
678 #ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
679 destroy_int_queue(udev, data->intq);
686 printf("%s: warning, ret=%d", __func__, ret);
690 static const struct udevice_id usb_kbd_ids[] = {
691 { .compatible = "usb-keyboard" },
695 U_BOOT_DRIVER(usb_kbd) = {
697 .id = UCLASS_KEYBOARD,
698 .of_match = usb_kbd_ids,
699 .probe = usb_kbd_probe,
700 .remove = usb_kbd_remove,
703 static const struct usb_device_id kbd_id_table[] = {
705 .match_flags = USB_DEVICE_ID_MATCH_INT_CLASS |
706 USB_DEVICE_ID_MATCH_INT_SUBCLASS |
707 USB_DEVICE_ID_MATCH_INT_PROTOCOL,
708 .bInterfaceClass = USB_CLASS_HID,
709 .bInterfaceSubClass = USB_SUB_HID_BOOT,
710 .bInterfaceProtocol = USB_PROT_HID_KEYBOARD,
712 { } /* Terminating entry */
715 U_BOOT_USB_DEVICE(usb_kbd, kbd_id_table);