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',
79 * map arrow keys to ^F/^B ^N/^P, can't really use the proper
80 * ANSI sequence for arrow keys because the queuing code breaks
81 * when a single keypress expands to 3 queue elements
83 static const unsigned char usb_kbd_arrow
[] = {
88 * NOTE: It's important for the NUM, CAPS, SCROLL-lock bits to be in this
89 * order. See usb_kbd_setled() function!
91 #define USB_KBD_NUMLOCK (1 << 0)
92 #define USB_KBD_CAPSLOCK (1 << 1)
93 #define USB_KBD_SCROLLLOCK (1 << 2)
94 #define USB_KBD_CTRL (1 << 3)
96 #define USB_KBD_LEDMASK \
97 (USB_KBD_NUMLOCK | USB_KBD_CAPSLOCK | USB_KBD_SCROLLLOCK)
100 * USB Keyboard reports are 8 bytes in boot protocol.
101 * Appendix B of HID Device Class Definition 1.11
103 #define USB_KBD_BOOT_REPORT_SIZE 8
105 struct usb_kbd_pdata
{
106 unsigned long intpipe
;
109 unsigned long last_report
;
110 struct int_queue
*intq
;
112 uint32_t repeat_delay
;
114 uint32_t usb_in_pointer
;
115 uint32_t usb_out_pointer
;
116 uint8_t usb_kbd_buffer
[USB_KBD_BUFFER_LEN
];
119 uint8_t old
[USB_KBD_BOOT_REPORT_SIZE
];
124 extern int __maybe_unused net_busy_flag
;
126 /* The period of time between two calls of usb_kbd_testc(). */
127 static unsigned long __maybe_unused kbd_testc_tms
;
129 /* Puts character in the queue and sets up the in and out pointer. */
130 static void usb_kbd_put_queue(struct usb_kbd_pdata
*data
, char c
)
132 if (data
->usb_in_pointer
== USB_KBD_BUFFER_LEN
- 1) {
133 /* Check for buffer full. */
134 if (data
->usb_out_pointer
== 0)
137 data
->usb_in_pointer
= 0;
139 /* Check for buffer full. */
140 if (data
->usb_in_pointer
== data
->usb_out_pointer
- 1)
143 data
->usb_in_pointer
++;
146 data
->usb_kbd_buffer
[data
->usb_in_pointer
] = c
;
149 static void usb_kbd_put_sequence(struct usb_kbd_pdata
*data
, char *s
)
152 usb_kbd_put_queue(data
, *s
);
156 * Set the LEDs. Since this is used in the irq routine, the control job is
157 * issued with a timeout of 0. This means, that the job is queued without
158 * waiting for job completion.
160 static void usb_kbd_setled(struct usb_device
*dev
)
162 struct usb_interface
*iface
= &dev
->config
.if_desc
[0];
163 struct usb_kbd_pdata
*data
= dev
->privptr
;
164 ALLOC_ALIGN_BUFFER(uint32_t, leds
, 1, USB_DMA_MINALIGN
);
166 *leds
= data
->flags
& USB_KBD_LEDMASK
;
167 usb_control_msg(dev
, usb_sndctrlpipe(dev
, 0),
168 USB_REQ_SET_REPORT
, USB_TYPE_CLASS
| USB_RECIP_INTERFACE
,
169 0x200, iface
->desc
.bInterfaceNumber
, leds
, 1, 0);
172 #define CAPITAL_MASK 0x20
173 /* Translate the scancode in ASCII */
174 static int usb_kbd_translate(struct usb_kbd_pdata
*data
, unsigned char scancode
,
175 unsigned char modifier
, int pressed
)
181 data
->repeat_delay
= 0;
186 data
->repeat_delay
++;
187 if (data
->repeat_delay
< REPEAT_DELAY
)
190 data
->repeat_delay
= REPEAT_DELAY
;
193 /* Alphanumeric values */
194 if ((scancode
> 3) && (scancode
<= 0x1d)) {
195 keycode
= scancode
- 4 + 'a';
197 if (data
->flags
& USB_KBD_CAPSLOCK
)
198 keycode
&= ~CAPITAL_MASK
;
200 if (modifier
& (LEFT_SHIFT
| RIGHT_SHIFT
)) {
201 /* Handle CAPSLock + Shift pressed simultaneously */
202 if (keycode
& CAPITAL_MASK
)
203 keycode
&= ~CAPITAL_MASK
;
205 keycode
|= CAPITAL_MASK
;
209 if ((scancode
> 0x1d) && (scancode
< 0x39)) {
211 if (modifier
& (LEFT_SHIFT
| RIGHT_SHIFT
))
212 keycode
= usb_kbd_numkey_shifted
[scancode
- 0x1e];
214 keycode
= usb_kbd_numkey
[scancode
- 0x1e];
218 if ((scancode
>= 0x4f) && (scancode
<= 0x52))
219 keycode
= usb_kbd_arrow
[scancode
- 0x4f];
222 if ((scancode
>= 0x54) && (scancode
<= 0x67))
223 keycode
= usb_kbd_num_keypad
[scancode
- 0x54];
225 if (data
->flags
& USB_KBD_CTRL
)
226 keycode
= scancode
- 0x3;
229 if (scancode
== NUM_LOCK
) {
230 data
->flags
^= USB_KBD_NUMLOCK
;
234 if (scancode
== CAPS_LOCK
) {
235 data
->flags
^= USB_KBD_CAPSLOCK
;
238 if (scancode
== SCROLL_LOCK
) {
239 data
->flags
^= USB_KBD_SCROLLLOCK
;
244 /* Report keycode if any */
246 debug("%c", keycode
);
249 case 0x0e: /* Down arrow key */
250 usb_kbd_put_sequence(data
, "\e[B");
252 case 0x10: /* Up arrow key */
253 usb_kbd_put_sequence(data
, "\e[A");
255 case 0x06: /* Right arrow key */
256 usb_kbd_put_sequence(data
, "\e[C");
258 case 0x02: /* Left arrow key */
259 usb_kbd_put_sequence(data
, "\e[D");
262 usb_kbd_put_queue(data
, keycode
);
269 static uint32_t usb_kbd_service_key(struct usb_device
*dev
, int i
, int up
)
272 struct usb_kbd_pdata
*data
= dev
->privptr
;
285 (memscan(new + 2, old
[i
], USB_KBD_BOOT_REPORT_SIZE
- 2) ==
286 new + USB_KBD_BOOT_REPORT_SIZE
)) {
287 res
|= usb_kbd_translate(data
, old
[i
], data
->new[0], up
);
293 /* Interrupt service routine */
294 static int usb_kbd_irq_worker(struct usb_device
*dev
)
296 struct usb_kbd_pdata
*data
= dev
->privptr
;
299 /* No combo key pressed */
300 if (data
->new[0] == 0x00)
301 data
->flags
&= ~USB_KBD_CTRL
;
302 /* Left or Right Ctrl pressed */
303 else if ((data
->new[0] == LEFT_CNTR
) || (data
->new[0] == RIGHT_CNTR
))
304 data
->flags
|= USB_KBD_CTRL
;
306 for (i
= 2; i
< USB_KBD_BOOT_REPORT_SIZE
; i
++) {
307 res
|= usb_kbd_service_key(dev
, i
, 0);
308 res
|= usb_kbd_service_key(dev
, i
, 1);
311 /* Key is still pressed */
312 if ((data
->new[2] > 3) && (data
->old
[2] == data
->new[2]))
313 res
|= usb_kbd_translate(data
, data
->new[2], data
->new[0], 2);
318 memcpy(data
->old
, data
->new, USB_KBD_BOOT_REPORT_SIZE
);
323 /* Keyboard interrupt handler */
324 static int usb_kbd_irq(struct usb_device
*dev
)
326 if ((dev
->irq_status
!= 0) ||
327 (dev
->irq_act_len
!= USB_KBD_BOOT_REPORT_SIZE
)) {
328 debug("USB KBD: Error %lX, len %d\n",
329 dev
->irq_status
, dev
->irq_act_len
);
333 return usb_kbd_irq_worker(dev
);
336 /* Interrupt polling */
337 static inline void usb_kbd_poll_for_event(struct usb_device
*dev
)
339 #if defined(CONFIG_SYS_USB_EVENT_POLL)
340 struct usb_kbd_pdata
*data
= dev
->privptr
;
342 /* Submit a interrupt transfer request */
343 usb_submit_int_msg(dev
, data
->intpipe
, &data
->new[0], data
->intpktsize
,
346 usb_kbd_irq_worker(dev
);
347 #elif defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP) || \
348 defined(CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE)
349 #if defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP)
350 struct usb_interface
*iface
;
351 struct usb_kbd_pdata
*data
= dev
->privptr
;
352 iface
= &dev
->config
.if_desc
[0];
353 usb_get_report(dev
, iface
->desc
.bInterfaceNumber
,
354 1, 0, data
->new, USB_KBD_BOOT_REPORT_SIZE
);
355 if (memcmp(data
->old
, data
->new, USB_KBD_BOOT_REPORT_SIZE
)) {
356 usb_kbd_irq_worker(dev
);
358 struct usb_kbd_pdata
*data
= dev
->privptr
;
359 if (poll_int_queue(dev
, data
->intq
)) {
360 usb_kbd_irq_worker(dev
);
361 /* We've consumed all queued int packets, create new */
362 destroy_int_queue(dev
, data
->intq
);
363 data
->intq
= create_int_queue(dev
, data
->intpipe
, 1,
364 USB_KBD_BOOT_REPORT_SIZE
, data
->new,
367 data
->last_report
= get_timer(0);
368 /* Repeat last usb hid report every REPEAT_RATE ms for keyrepeat */
369 } else if (data
->last_report
!= -1 &&
370 get_timer(data
->last_report
) > REPEAT_RATE
) {
371 usb_kbd_irq_worker(dev
);
372 data
->last_report
= get_timer(0);
377 /* test if a character is in the queue */
378 static int usb_kbd_testc(struct stdio_dev
*sdev
)
380 struct stdio_dev
*dev
;
381 struct usb_device
*usb_kbd_dev
;
382 struct usb_kbd_pdata
*data
;
384 #ifdef CONFIG_CMD_NET
386 * If net_busy_flag is 1, NET transfer is running,
387 * then we check key-pressed every second (first check may be
388 * less than 1 second) to improve TFTP booting performance.
390 if (net_busy_flag
&& (get_timer(kbd_testc_tms
) < CONFIG_SYS_HZ
))
392 kbd_testc_tms
= get_timer(0);
394 dev
= stdio_get_by_name(sdev
->name
);
395 usb_kbd_dev
= (struct usb_device
*)dev
->priv
;
396 data
= usb_kbd_dev
->privptr
;
398 usb_kbd_poll_for_event(usb_kbd_dev
);
400 return !(data
->usb_in_pointer
== data
->usb_out_pointer
);
403 /* gets the character from the queue */
404 static int usb_kbd_getc(struct stdio_dev
*sdev
)
406 struct stdio_dev
*dev
;
407 struct usb_device
*usb_kbd_dev
;
408 struct usb_kbd_pdata
*data
;
410 dev
= stdio_get_by_name(sdev
->name
);
411 usb_kbd_dev
= (struct usb_device
*)dev
->priv
;
412 data
= usb_kbd_dev
->privptr
;
414 while (data
->usb_in_pointer
== data
->usb_out_pointer
) {
416 usb_kbd_poll_for_event(usb_kbd_dev
);
419 if (data
->usb_out_pointer
== USB_KBD_BUFFER_LEN
- 1)
420 data
->usb_out_pointer
= 0;
422 data
->usb_out_pointer
++;
424 return data
->usb_kbd_buffer
[data
->usb_out_pointer
];
427 /* probes the USB device dev for keyboard type. */
428 static int usb_kbd_probe_dev(struct usb_device
*dev
, unsigned int ifnum
)
430 struct usb_interface
*iface
;
431 struct usb_endpoint_descriptor
*ep
;
432 struct usb_kbd_pdata
*data
;
434 if (dev
->descriptor
.bNumConfigurations
!= 1)
437 iface
= &dev
->config
.if_desc
[ifnum
];
439 if (iface
->desc
.bInterfaceClass
!= USB_CLASS_HID
)
442 if (iface
->desc
.bInterfaceSubClass
!= USB_SUB_HID_BOOT
)
445 if (iface
->desc
.bInterfaceProtocol
!= USB_PROT_HID_KEYBOARD
)
448 if (iface
->desc
.bNumEndpoints
!= 1)
451 ep
= &iface
->ep_desc
[0];
453 /* Check if endpoint 1 is interrupt endpoint */
454 if (!(ep
->bEndpointAddress
& 0x80))
457 if ((ep
->bmAttributes
& 3) != 3)
460 debug("USB KBD: found set protocol...\n");
462 data
= malloc(sizeof(struct usb_kbd_pdata
));
464 printf("USB KBD: Error allocating private data\n");
468 /* Clear private data */
469 memset(data
, 0, sizeof(struct usb_kbd_pdata
));
471 /* allocate input buffer aligned and sized to USB DMA alignment */
472 data
->new = memalign(USB_DMA_MINALIGN
,
473 roundup(USB_KBD_BOOT_REPORT_SIZE
, USB_DMA_MINALIGN
));
475 /* Insert private data into USB device structure */
478 /* Set IRQ handler */
479 dev
->irq_handle
= usb_kbd_irq
;
481 data
->intpipe
= usb_rcvintpipe(dev
, ep
->bEndpointAddress
);
482 data
->intpktsize
= min(usb_maxpacket(dev
, data
->intpipe
),
483 USB_KBD_BOOT_REPORT_SIZE
);
484 data
->intinterval
= ep
->bInterval
;
485 data
->last_report
= -1;
487 /* We found a USB Keyboard, install it. */
488 usb_set_protocol(dev
, iface
->desc
.bInterfaceNumber
, 0);
490 debug("USB KBD: found set idle...\n");
491 #if !defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP) && \
492 !defined(CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE)
493 usb_set_idle(dev
, iface
->desc
.bInterfaceNumber
, REPEAT_RATE
/ 4, 0);
495 usb_set_idle(dev
, iface
->desc
.bInterfaceNumber
, 0, 0);
498 debug("USB KBD: enable interrupt pipe...\n");
499 #ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
500 data
->intq
= create_int_queue(dev
, data
->intpipe
, 1,
501 USB_KBD_BOOT_REPORT_SIZE
, data
->new,
504 #elif defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP)
505 if (usb_get_report(dev
, iface
->desc
.bInterfaceNumber
,
506 1, 0, data
->new, USB_KBD_BOOT_REPORT_SIZE
) < 0) {
508 if (usb_submit_int_msg(dev
, data
->intpipe
, data
->new, data
->intpktsize
,
509 data
->intinterval
) < 0) {
511 printf("Failed to get keyboard state from device %04x:%04x\n",
512 dev
->descriptor
.idVendor
, dev
->descriptor
.idProduct
);
513 /* Abort, we don't want to use that non-functional keyboard. */
521 static int probe_usb_keyboard(struct usb_device
*dev
)
524 struct stdio_dev usb_kbd_dev
;
527 /* Try probing the keyboard */
528 if (usb_kbd_probe_dev(dev
, 0) != 1)
531 /* Register the keyboard */
532 debug("USB KBD: register.\n");
533 memset(&usb_kbd_dev
, 0, sizeof(struct stdio_dev
));
534 strcpy(usb_kbd_dev
.name
, DEVNAME
);
535 usb_kbd_dev
.flags
= DEV_FLAGS_INPUT
;
536 usb_kbd_dev
.getc
= usb_kbd_getc
;
537 usb_kbd_dev
.tstc
= usb_kbd_testc
;
538 usb_kbd_dev
.priv
= (void *)dev
;
539 error
= stdio_register(&usb_kbd_dev
);
543 stdinname
= env_get("stdin");
544 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
545 error
= iomux_doenv(stdin
, stdinname
);
549 /* Check if this is the standard input device. */
550 if (strcmp(stdinname
, DEVNAME
))
553 /* Reassign the console */
554 if (overwrite_console())
557 error
= console_assign(stdin
, DEVNAME
);
565 #if !CONFIG_IS_ENABLED(DM_USB)
566 /* Search for keyboard and register it if found. */
567 int drv_usb_kbd_init(void)
571 debug("%s: Probing for keyboard\n", __func__
);
572 /* Scan all USB Devices */
573 for (i
= 0; i
< USB_MAX_DEVICE
; i
++) {
574 struct usb_device
*dev
;
576 /* Get USB device. */
577 dev
= usb_get_dev_index(i
);
581 if (dev
->devnum
== -1)
584 error
= probe_usb_keyboard(dev
);
587 if (error
&& error
!= -ENOENT
)
591 /* No USB Keyboard found */
595 /* Deregister the keyboard. */
596 int usb_kbd_deregister(int force
)
598 #if CONFIG_IS_ENABLED(SYS_STDIO_DEREGISTER)
599 struct stdio_dev
*dev
;
600 struct usb_device
*usb_kbd_dev
;
601 struct usb_kbd_pdata
*data
;
603 dev
= stdio_get_by_name(DEVNAME
);
605 usb_kbd_dev
= (struct usb_device
*)dev
->priv
;
606 data
= usb_kbd_dev
->privptr
;
607 if (stdio_deregister_dev(dev
, force
) != 0)
609 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
610 if (iomux_doenv(stdin
, env_get("stdin")) != 0)
613 #ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
614 destroy_int_queue(usb_kbd_dev
, data
->intq
);
628 #if CONFIG_IS_ENABLED(DM_USB)
630 static int usb_kbd_probe(struct udevice
*dev
)
632 struct usb_device
*udev
= dev_get_parent_priv(dev
);
634 return probe_usb_keyboard(udev
);
637 static int usb_kbd_remove(struct udevice
*dev
)
639 struct usb_device
*udev
= dev_get_parent_priv(dev
);
640 struct usb_kbd_pdata
*data
;
641 struct stdio_dev
*sdev
;
644 sdev
= stdio_get_by_name(DEVNAME
);
649 data
= udev
->privptr
;
650 if (stdio_deregister_dev(sdev
, true)) {
654 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
655 if (iomux_doenv(stdin
, env_get("stdin"))) {
660 #ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
661 destroy_int_queue(udev
, data
->intq
);
668 printf("%s: warning, ret=%d", __func__
, ret
);
672 static const struct udevice_id usb_kbd_ids
[] = {
673 { .compatible
= "usb-keyboard" },
677 U_BOOT_DRIVER(usb_kbd
) = {
679 .id
= UCLASS_KEYBOARD
,
680 .of_match
= usb_kbd_ids
,
681 .probe
= usb_kbd_probe
,
682 .remove
= usb_kbd_remove
,
685 static const struct usb_device_id kbd_id_table
[] = {
687 .match_flags
= USB_DEVICE_ID_MATCH_INT_CLASS
|
688 USB_DEVICE_ID_MATCH_INT_SUBCLASS
|
689 USB_DEVICE_ID_MATCH_INT_PROTOCOL
,
690 .bInterfaceClass
= USB_CLASS_HID
,
691 .bInterfaceSubClass
= USB_SUB_HID_BOOT
,
692 .bInterfaceProtocol
= USB_PROT_HID_KEYBOARD
,
694 { } /* Terminating entry */
697 U_BOOT_USB_DEVICE(usb_kbd
, kbd_id_table
);