1
2
3
4
5
6
7
8
9
10#include <common.h>
11#include <console.h>
12#include <dm.h>
13#include <errno.h>
14#include <malloc.h>
15#include <memalign.h>
16#include <stdio_dev.h>
17#include <asm/byteorder.h>
18
19#include <usb.h>
20
21
22
23
24
25
26#ifdef CONFIG_SYS_CONSOLE_OVERWRITE_ROUTINE
27extern int overwrite_console(void);
28#else
29int overwrite_console(void)
30{
31 return 0;
32}
33#endif
34
35
36#define REPEAT_RATE 40
37#define REPEAT_DELAY 10
38
39#define NUM_LOCK 0x53
40#define CAPS_LOCK 0x39
41#define SCROLL_LOCK 0x47
42
43
44#define LEFT_CNTR (1 << 0)
45#define LEFT_SHIFT (1 << 1)
46#define LEFT_ALT (1 << 2)
47#define LEFT_GUI (1 << 3)
48#define RIGHT_CNTR (1 << 4)
49#define RIGHT_SHIFT (1 << 5)
50#define RIGHT_ALT (1 << 6)
51#define RIGHT_GUI (1 << 7)
52
53
54#define USB_KBD_BUFFER_LEN 0x20
55
56
57#define DEVNAME "usbkbd"
58
59
60static const unsigned char usb_kbd_numkey[] = {
61 '1', '2', '3', '4', '5', '6', '7', '8', '9', '0',
62 '\r', 0x1b, '\b', '\t', ' ', '-', '=', '[', ']',
63 '\\', '#', ';', '\'', '`', ',', '.', '/'
64};
65static const unsigned char usb_kbd_numkey_shifted[] = {
66 '!', '@', '#', '$', '%', '^', '&', '*', '(', ')',
67 '\r', 0x1b, '\b', '\t', ' ', '_', '+', '{', '}',
68 '|', '~', ':', '"', '~', '<', '>', '?'
69};
70
71static const unsigned char usb_kbd_num_keypad[] = {
72 '/', '*', '-', '+', '\r',
73 '1', '2', '3', '4', '5', '6', '7', '8', '9', '0',
74 '.', 0, 0, 0, '='
75};
76
77
78
79
80
81
82static const unsigned char usb_kbd_arrow[] = {
83 0x6, 0x2, 0xe, 0x10
84};
85
86
87
88
89
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)
94
95#define USB_KBD_LEDMASK \
96 (USB_KBD_NUMLOCK | USB_KBD_CAPSLOCK | USB_KBD_SCROLLLOCK)
97
98
99
100
101
102#define USB_KBD_BOOT_REPORT_SIZE 8
103
104struct usb_kbd_pdata {
105 unsigned long intpipe;
106 int intpktsize;
107 int intinterval;
108 unsigned long last_report;
109 struct int_queue *intq;
110
111 uint32_t repeat_delay;
112
113 uint32_t usb_in_pointer;
114 uint32_t usb_out_pointer;
115 uint8_t usb_kbd_buffer[USB_KBD_BUFFER_LEN];
116
117 uint8_t *new;
118 uint8_t old[USB_KBD_BOOT_REPORT_SIZE];
119
120 uint8_t flags;
121};
122
123extern int __maybe_unused net_busy_flag;
124
125
126static unsigned long __maybe_unused kbd_testc_tms;
127
128
129static void usb_kbd_put_queue(struct usb_kbd_pdata *data, char c)
130{
131 if (data->usb_in_pointer == USB_KBD_BUFFER_LEN - 1) {
132
133 if (data->usb_out_pointer == 0)
134 return;
135
136 data->usb_in_pointer = 0;
137 } else {
138
139 if (data->usb_in_pointer == data->usb_out_pointer - 1)
140 return;
141
142 data->usb_in_pointer++;
143 }
144
145 data->usb_kbd_buffer[data->usb_in_pointer] = c;
146}
147
148
149
150
151
152
153static void usb_kbd_setled(struct usb_device *dev)
154{
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);
158
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);
163}
164
165#define CAPITAL_MASK 0x20
166
167static int usb_kbd_translate(struct usb_kbd_pdata *data, unsigned char scancode,
168 unsigned char modifier, int pressed)
169{
170 uint8_t keycode = 0;
171
172
173 if (pressed == 0) {
174 data->repeat_delay = 0;
175 return 0;
176 }
177
178 if (pressed == 2) {
179 data->repeat_delay++;
180 if (data->repeat_delay < REPEAT_DELAY)
181 return 0;
182
183 data->repeat_delay = REPEAT_DELAY;
184 }
185
186
187 if ((scancode > 3) && (scancode <= 0x1d)) {
188 keycode = scancode - 4 + 'a';
189
190 if (data->flags & USB_KBD_CAPSLOCK)
191 keycode &= ~CAPITAL_MASK;
192
193 if (modifier & (LEFT_SHIFT | RIGHT_SHIFT)) {
194
195 if (keycode & CAPITAL_MASK)
196 keycode &= ~CAPITAL_MASK;
197 else
198 keycode |= CAPITAL_MASK;
199 }
200 }
201
202 if ((scancode > 0x1d) && (scancode < 0x39)) {
203
204 if (modifier & (LEFT_SHIFT | RIGHT_SHIFT))
205 keycode = usb_kbd_numkey_shifted[scancode - 0x1e];
206 else
207 keycode = usb_kbd_numkey[scancode - 0x1e];
208 }
209
210
211 if ((scancode >= 0x4f) && (scancode <= 0x52))
212 keycode = usb_kbd_arrow[scancode - 0x4f];
213
214
215 if ((scancode >= 0x54) && (scancode <= 0x67))
216 keycode = usb_kbd_num_keypad[scancode - 0x54];
217
218 if (data->flags & USB_KBD_CTRL)
219 keycode = scancode - 0x3;
220
221 if (pressed == 1) {
222 if (scancode == NUM_LOCK) {
223 data->flags ^= USB_KBD_NUMLOCK;
224 return 1;
225 }
226
227 if (scancode == CAPS_LOCK) {
228 data->flags ^= USB_KBD_CAPSLOCK;
229 return 1;
230 }
231 if (scancode == SCROLL_LOCK) {
232 data->flags ^= USB_KBD_SCROLLLOCK;
233 return 1;
234 }
235 }
236
237
238 if (keycode) {
239 debug("%c", keycode);
240 usb_kbd_put_queue(data, keycode);
241 }
242
243 return 0;
244}
245
246static uint32_t usb_kbd_service_key(struct usb_device *dev, int i, int up)
247{
248 uint32_t res = 0;
249 struct usb_kbd_pdata *data = dev->privptr;
250 uint8_t *new;
251 uint8_t *old;
252
253 if (up) {
254 new = data->old;
255 old = data->new;
256 } else {
257 new = data->new;
258 old = data->old;
259 }
260
261 if ((old[i] > 3) &&
262 (memscan(new + 2, old[i], USB_KBD_BOOT_REPORT_SIZE - 2) ==
263 new + USB_KBD_BOOT_REPORT_SIZE)) {
264 res |= usb_kbd_translate(data, old[i], data->new[0], up);
265 }
266
267 return res;
268}
269
270
271static int usb_kbd_irq_worker(struct usb_device *dev)
272{
273 struct usb_kbd_pdata *data = dev->privptr;
274 int i, res = 0;
275
276
277 if (data->new[0] == 0x00)
278 data->flags &= ~USB_KBD_CTRL;
279
280 else if ((data->new[0] == LEFT_CNTR) || (data->new[0] == RIGHT_CNTR))
281 data->flags |= USB_KBD_CTRL;
282
283 for (i = 2; i < USB_KBD_BOOT_REPORT_SIZE; i++) {
284 res |= usb_kbd_service_key(dev, i, 0);
285 res |= usb_kbd_service_key(dev, i, 1);
286 }
287
288
289 if ((data->new[2] > 3) && (data->old[2] == data->new[2]))
290 res |= usb_kbd_translate(data, data->new[2], data->new[0], 2);
291
292 if (res == 1)
293 usb_kbd_setled(dev);
294
295 memcpy(data->old, data->new, USB_KBD_BOOT_REPORT_SIZE);
296
297 return 1;
298}
299
300
301static int usb_kbd_irq(struct usb_device *dev)
302{
303 if ((dev->irq_status != 0) ||
304 (dev->irq_act_len != USB_KBD_BOOT_REPORT_SIZE)) {
305 debug("USB KBD: Error %lX, len %d\n",
306 dev->irq_status, dev->irq_act_len);
307 return 1;
308 }
309
310 return usb_kbd_irq_worker(dev);
311}
312
313
314static inline void usb_kbd_poll_for_event(struct usb_device *dev)
315{
316#if defined(CONFIG_SYS_USB_EVENT_POLL)
317 struct usb_kbd_pdata *data = dev->privptr;
318
319
320 usb_submit_int_msg(dev, data->intpipe, &data->new[0], data->intpktsize,
321 data->intinterval);
322
323 usb_kbd_irq_worker(dev);
324#elif defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP) || \
325 defined(CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE)
326#if defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP)
327 struct usb_interface *iface;
328 struct usb_kbd_pdata *data = dev->privptr;
329 iface = &dev->config.if_desc[0];
330 usb_get_report(dev, iface->desc.bInterfaceNumber,
331 1, 0, data->new, USB_KBD_BOOT_REPORT_SIZE);
332 if (memcmp(data->old, data->new, USB_KBD_BOOT_REPORT_SIZE)) {
333 usb_kbd_irq_worker(dev);
334#else
335 struct usb_kbd_pdata *data = dev->privptr;
336 if (poll_int_queue(dev, data->intq)) {
337 usb_kbd_irq_worker(dev);
338
339 destroy_int_queue(dev, data->intq);
340 data->intq = create_int_queue(dev, data->intpipe, 1,
341 USB_KBD_BOOT_REPORT_SIZE, data->new,
342 data->intinterval);
343#endif
344 data->last_report = get_timer(0);
345
346 } else if (data->last_report != -1 &&
347 get_timer(data->last_report) > REPEAT_RATE) {
348 usb_kbd_irq_worker(dev);
349 data->last_report = get_timer(0);
350 }
351#endif
352}
353
354
355static int usb_kbd_testc(struct stdio_dev *sdev)
356{
357 struct stdio_dev *dev;
358 struct usb_device *usb_kbd_dev;
359 struct usb_kbd_pdata *data;
360
361#ifdef CONFIG_CMD_NET
362
363
364
365
366
367 if (net_busy_flag && (get_timer(kbd_testc_tms) < CONFIG_SYS_HZ))
368 return 0;
369 kbd_testc_tms = get_timer(0);
370#endif
371 dev = stdio_get_by_name(DEVNAME);
372 usb_kbd_dev = (struct usb_device *)dev->priv;
373 data = usb_kbd_dev->privptr;
374
375 usb_kbd_poll_for_event(usb_kbd_dev);
376
377 return !(data->usb_in_pointer == data->usb_out_pointer);
378}
379
380
381static int usb_kbd_getc(struct stdio_dev *sdev)
382{
383 struct stdio_dev *dev;
384 struct usb_device *usb_kbd_dev;
385 struct usb_kbd_pdata *data;
386
387 dev = stdio_get_by_name(DEVNAME);
388 usb_kbd_dev = (struct usb_device *)dev->priv;
389 data = usb_kbd_dev->privptr;
390
391 while (data->usb_in_pointer == data->usb_out_pointer)
392 usb_kbd_poll_for_event(usb_kbd_dev);
393
394 if (data->usb_out_pointer == USB_KBD_BUFFER_LEN - 1)
395 data->usb_out_pointer = 0;
396 else
397 data->usb_out_pointer++;
398
399 return data->usb_kbd_buffer[data->usb_out_pointer];
400}
401
402
403static int usb_kbd_probe_dev(struct usb_device *dev, unsigned int ifnum)
404{
405 struct usb_interface *iface;
406 struct usb_endpoint_descriptor *ep;
407 struct usb_kbd_pdata *data;
408
409 if (dev->descriptor.bNumConfigurations != 1)
410 return 0;
411
412 iface = &dev->config.if_desc[ifnum];
413
414 if (iface->desc.bInterfaceClass != USB_CLASS_HID)
415 return 0;
416
417 if (iface->desc.bInterfaceSubClass != USB_SUB_HID_BOOT)
418 return 0;
419
420 if (iface->desc.bInterfaceProtocol != USB_PROT_HID_KEYBOARD)
421 return 0;
422
423 if (iface->desc.bNumEndpoints != 1)
424 return 0;
425
426 ep = &iface->ep_desc[0];
427
428
429 if (!(ep->bEndpointAddress & 0x80))
430 return 0;
431
432 if ((ep->bmAttributes & 3) != 3)
433 return 0;
434
435 debug("USB KBD: found set protocol...\n");
436
437 data = malloc(sizeof(struct usb_kbd_pdata));
438 if (!data) {
439 printf("USB KBD: Error allocating private data\n");
440 return 0;
441 }
442
443
444 memset(data, 0, sizeof(struct usb_kbd_pdata));
445
446
447 data->new = memalign(USB_DMA_MINALIGN,
448 roundup(USB_KBD_BOOT_REPORT_SIZE, USB_DMA_MINALIGN));
449
450
451 dev->privptr = data;
452
453
454 dev->irq_handle = usb_kbd_irq;
455
456 data->intpipe = usb_rcvintpipe(dev, ep->bEndpointAddress);
457 data->intpktsize = min(usb_maxpacket(dev, data->intpipe),
458 USB_KBD_BOOT_REPORT_SIZE);
459 data->intinterval = ep->bInterval;
460 data->last_report = -1;
461
462
463 usb_set_protocol(dev, iface->desc.bInterfaceNumber, 0);
464
465 debug("USB KBD: found set idle...\n");
466#if !defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP) && \
467 !defined(CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE)
468 usb_set_idle(dev, iface->desc.bInterfaceNumber, REPEAT_RATE / 4, 0);
469#else
470 usb_set_idle(dev, iface->desc.bInterfaceNumber, 0, 0);
471#endif
472
473 debug("USB KBD: enable interrupt pipe...\n");
474#ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
475 data->intq = create_int_queue(dev, data->intpipe, 1,
476 USB_KBD_BOOT_REPORT_SIZE, data->new,
477 data->intinterval);
478 if (!data->intq) {
479#elif defined(CONFIG_SYS_USB_EVENT_POLL_VIA_CONTROL_EP)
480 if (usb_get_report(dev, iface->desc.bInterfaceNumber,
481 1, 0, data->new, USB_KBD_BOOT_REPORT_SIZE) < 0) {
482#else
483 if (usb_submit_int_msg(dev, data->intpipe, data->new, data->intpktsize,
484 data->intinterval) < 0) {
485#endif
486 printf("Failed to get keyboard state from device %04x:%04x\n",
487 dev->descriptor.idVendor, dev->descriptor.idProduct);
488
489 return 0;
490 }
491
492
493 return 1;
494}
495
496static int probe_usb_keyboard(struct usb_device *dev)
497{
498 char *stdinname;
499 struct stdio_dev usb_kbd_dev;
500 int error;
501
502
503 if (usb_kbd_probe_dev(dev, 0) != 1)
504 return -ENOENT;
505
506
507 debug("USB KBD: register.\n");
508 memset(&usb_kbd_dev, 0, sizeof(struct stdio_dev));
509 strcpy(usb_kbd_dev.name, DEVNAME);
510 usb_kbd_dev.flags = DEV_FLAGS_INPUT;
511 usb_kbd_dev.getc = usb_kbd_getc;
512 usb_kbd_dev.tstc = usb_kbd_testc;
513 usb_kbd_dev.priv = (void *)dev;
514 error = stdio_register(&usb_kbd_dev);
515 if (error)
516 return error;
517
518 stdinname = getenv("stdin");
519#ifdef CONFIG_CONSOLE_MUX
520 error = iomux_doenv(stdin, stdinname);
521 if (error)
522 return error;
523#else
524
525 if (strcmp(stdinname, DEVNAME))
526 return 1;
527
528
529 if (overwrite_console())
530 return 1;
531
532 error = console_assign(stdin, DEVNAME);
533 if (error)
534 return error;
535#endif
536
537 return 0;
538}
539
540#ifndef CONFIG_DM_USB
541
542int drv_usb_kbd_init(void)
543{
544 int error, i;
545
546 debug("%s: Probing for keyboard\n", __func__);
547
548 for (i = 0; i < USB_MAX_DEVICE; i++) {
549 struct usb_device *dev;
550
551
552 dev = usb_get_dev_index(i);
553 if (!dev)
554 break;
555
556 if (dev->devnum == -1)
557 continue;
558
559 error = probe_usb_keyboard(dev);
560 if (!error)
561 return 1;
562 if (error && error != -ENOENT)
563 return error;
564 }
565
566
567 return -1;
568}
569
570
571int usb_kbd_deregister(int force)
572{
573#ifdef CONFIG_SYS_STDIO_DEREGISTER
574 struct stdio_dev *dev;
575 struct usb_device *usb_kbd_dev;
576 struct usb_kbd_pdata *data;
577
578 dev = stdio_get_by_name(DEVNAME);
579 if (dev) {
580 usb_kbd_dev = (struct usb_device *)dev->priv;
581 data = usb_kbd_dev->privptr;
582 if (stdio_deregister_dev(dev, force) != 0)
583 return 1;
584#ifdef CONFIG_CONSOLE_MUX
585 if (iomux_doenv(stdin, getenv("stdin")) != 0)
586 return 1;
587#endif
588#ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
589 destroy_int_queue(usb_kbd_dev, data->intq);
590#endif
591 free(data->new);
592 free(data);
593 }
594
595 return 0;
596#else
597 return 1;
598#endif
599}
600
601#endif
602
603#ifdef CONFIG_DM_USB
604
605static int usb_kbd_probe(struct udevice *dev)
606{
607 struct usb_device *udev = dev_get_parent_priv(dev);
608 int ret;
609
610 ret = probe_usb_keyboard(udev);
611
612 return ret;
613}
614
615static int usb_kbd_remove(struct udevice *dev)
616{
617 struct usb_device *udev = dev_get_parent_priv(dev);
618 struct usb_kbd_pdata *data;
619 struct stdio_dev *sdev;
620 int ret;
621
622 sdev = stdio_get_by_name(DEVNAME);
623 if (!sdev) {
624 ret = -ENXIO;
625 goto err;
626 }
627 data = udev->privptr;
628 if (stdio_deregister_dev(sdev, true)) {
629 ret = -EPERM;
630 goto err;
631 }
632#ifdef CONFIG_CONSOLE_MUX
633 if (iomux_doenv(stdin, getenv("stdin"))) {
634 ret = -ENOLINK;
635 goto err;
636 }
637#endif
638#ifdef CONFIG_SYS_USB_EVENT_POLL_VIA_INT_QUEUE
639 destroy_int_queue(udev, data->intq);
640#endif
641 free(data->new);
642 free(data);
643
644 return 0;
645err:
646 printf("%s: warning, ret=%d", __func__, ret);
647 return ret;
648}
649
650static const struct udevice_id usb_kbd_ids[] = {
651 { .compatible = "usb-keyboard" },
652 { }
653};
654
655U_BOOT_DRIVER(usb_kbd) = {
656 .name = "usb_kbd",
657 .id = UCLASS_KEYBOARD,
658 .of_match = usb_kbd_ids,
659 .probe = usb_kbd_probe,
660 .remove = usb_kbd_remove,
661};
662
663static const struct usb_device_id kbd_id_table[] = {
664 {
665 .match_flags = USB_DEVICE_ID_MATCH_INT_CLASS |
666 USB_DEVICE_ID_MATCH_INT_SUBCLASS |
667 USB_DEVICE_ID_MATCH_INT_PROTOCOL,
668 .bInterfaceClass = USB_CLASS_HID,
669 .bInterfaceSubClass = USB_SUB_HID_BOOT,
670 .bInterfaceProtocol = USB_PROT_HID_KEYBOARD,
671 },
672 { }
673};
674
675U_BOOT_USB_DEVICE(usb_kbd, kbd_id_table);
676
677#endif
678