https://wiki.osdev.org/PS/2_Keyboard
Для USB - есть отдельный LED report, смотреть в спеке USB HID
https://www.usb.org/sites/default/files/hid1_11.pdf