diff options
| -rw-r--r-- | tmk_core/protocol/vusb/vusb.c | 13 | ||||
| -rwxr-xr-x | util/atmega32a_program.py | 110 |
2 files changed, 1 insertions, 122 deletions
diff --git a/tmk_core/protocol/vusb/vusb.c b/tmk_core/protocol/vusb/vusb.c index 72445e00b..e66938445 100644 --- a/tmk_core/protocol/vusb/vusb.c +++ b/tmk_core/protocol/vusb/vusb.c | |||
| @@ -26,7 +26,6 @@ along with this program. If not, see <http://www.gnu.org/licenses/>. | |||
| 26 | #include "debug.h" | 26 | #include "debug.h" |
| 27 | #include "host_driver.h" | 27 | #include "host_driver.h" |
| 28 | #include "vusb.h" | 28 | #include "vusb.h" |
| 29 | #include "bootloader.h" | ||
| 30 | #include <util/delay.h> | 29 | #include <util/delay.h> |
| 31 | 30 | ||
| 32 | static uint8_t vusb_keyboard_leds = 0; | 31 | static uint8_t vusb_keyboard_leds = 0; |
| @@ -145,7 +144,7 @@ static void send_consumer(uint16_t data) { | |||
| 145 | *------------------------------------------------------------------*/ | 144 | *------------------------------------------------------------------*/ |
| 146 | static struct { | 145 | static struct { |
| 147 | uint16_t len; | 146 | uint16_t len; |
| 148 | enum { NONE, BOOTLOADER, SET_LED } kind; | 147 | enum { NONE, SET_LED } kind; |
| 149 | } last_req; | 148 | } last_req; |
| 150 | 149 | ||
| 151 | usbMsgLen_t usbFunctionSetup(uchar data[8]) { | 150 | usbMsgLen_t usbFunctionSetup(uchar data[8]) { |
| @@ -173,11 +172,6 @@ usbMsgLen_t usbFunctionSetup(uchar data[8]) { | |||
| 173 | debug("SET_LED: "); | 172 | debug("SET_LED: "); |
| 174 | last_req.kind = SET_LED; | 173 | last_req.kind = SET_LED; |
| 175 | last_req.len = rq->wLength.word; | 174 | last_req.len = rq->wLength.word; |
| 176 | #ifdef BOOTLOADER_SIZE | ||
| 177 | } else if (rq->wValue.word == 0x0301) { | ||
| 178 | last_req.kind = BOOTLOADER; | ||
| 179 | last_req.len = rq->wLength.word; | ||
| 180 | #endif | ||
| 181 | } | 175 | } |
| 182 | return USB_NO_MSG; // to get data in usbFunctionWrite | 176 | return USB_NO_MSG; // to get data in usbFunctionWrite |
| 183 | } else { | 177 | } else { |
| @@ -204,11 +198,6 @@ uchar usbFunctionWrite(uchar *data, uchar len) { | |||
| 204 | last_req.len = 0; | 198 | last_req.len = 0; |
| 205 | return 1; | 199 | return 1; |
| 206 | break; | 200 | break; |
| 207 | case BOOTLOADER: | ||
| 208 | usbDeviceDisconnect(); | ||
| 209 | bootloader_jump(); | ||
| 210 | return 1; | ||
| 211 | break; | ||
| 212 | case NONE: | 201 | case NONE: |
| 213 | default: | 202 | default: |
| 214 | return -1; | 203 | return -1; |
diff --git a/util/atmega32a_program.py b/util/atmega32a_program.py deleted file mode 100755 index 9438c7e77..000000000 --- a/util/atmega32a_program.py +++ /dev/null | |||
| @@ -1,110 +0,0 @@ | |||
| 1 | #!/usr/bin/env python | ||
| 2 | # Copyright 2017 Luiz Ribeiro <luizribeiro@gmail.com>, Sebastian Kaim <sebb@sebb767.de> | ||
| 3 | # | ||
| 4 | # This program is free software: you can redistribute it and/or modify | ||
| 5 | # it under the terms of the GNU General Public License as published by | ||
| 6 | # the Free Software Foundation, either version 2 of the License, or | ||
| 7 | # (at your option) any later version. | ||
| 8 | # | ||
| 9 | # This program is distributed in the hope that it will be useful, | ||
| 10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 12 | # GNU General Public License for more details. | ||
| 13 | # | ||
| 14 | # You should have received a copy of the GNU General Public License | ||
| 15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
| 16 | |||
| 17 | from __future__ import print_function | ||
| 18 | |||
| 19 | import os | ||
| 20 | import sys | ||
| 21 | import time | ||
| 22 | import argparse | ||
| 23 | import usb | ||
| 24 | |||
| 25 | |||
| 26 | def check_keyboard_normal_mode(vendor, product): | ||
| 27 | """Returns a device if a ps2avrGB device in normal made (that is in keyboard mode) or None if it is not found.""" | ||
| 28 | return usb.core.find(idVendor=vendor, idProduct=product) | ||
| 29 | |||
| 30 | def check_keyboard_bootloader_mode(): | ||
| 31 | """Returns True if a ps2avrGB device in bootloader (flashable) mode is found and False otherwise.""" | ||
| 32 | return (usb.core.find(idVendor=0x16c0, idProduct=0x05df) is not None) | ||
| 33 | |||
| 34 | def flash_keyboard(firmware_file): | ||
| 35 | """Calls bootloadHID to flash the given file to the device.""" | ||
| 36 | print('Flashing firmware to device ...') | ||
| 37 | if os.system('bootloadHID -r "%s"' % firmware_file) == 0: | ||
| 38 | print('\nDone!') | ||
| 39 | else: | ||
| 40 | print('\nbootloadHID returned an error.') | ||
| 41 | |||
| 42 | def print_device_info(dev): | ||
| 43 | """Prints all infos for a given USB device""" | ||
| 44 | print('Device Information:') | ||
| 45 | print(' idVendor: %d (0x%04x)' % (dev.idVendor, dev.idVendor)) | ||
| 46 | print(' idProduct: %d (0x%04x)' % (dev.idProduct, dev.idProduct)) | ||
| 47 | print('Manufacturer: %s' % (dev.iManufacturer)) | ||
| 48 | print('Serial: %s' % (dev.iSerialNumber)) | ||
| 49 | print('Product: %s' % (dev.iProduct), end='\n\n') | ||
| 50 | |||
| 51 | def send_device_to_bootloader_mode(dev): | ||
| 52 | """Tries to send a given ps2avrGB keyboard to bootloader mode to allow flashing.""" | ||
| 53 | try: | ||
| 54 | dev.set_configuration() | ||
| 55 | |||
| 56 | request_type = usb.util.build_request_type( | ||
| 57 | usb.util.CTRL_OUT, | ||
| 58 | usb.util.CTRL_TYPE_CLASS, | ||
| 59 | usb.util.CTRL_RECIPIENT_DEVICE) | ||
| 60 | |||
| 61 | USBRQ_HID_SET_REPORT = 0x09 | ||
| 62 | HID_REPORT_OPTION = 0x0301 | ||
| 63 | |||
| 64 | dev.ctrl_transfer(request_type, USBRQ_HID_SET_REPORT, HID_REPORT_OPTION, 0, [0, 0, 0xFF] + [0] * 5) | ||
| 65 | except usb.core.USBError: | ||
| 66 | # for some reason I keep getting USBError, but it works! | ||
| 67 | pass | ||
| 68 | |||
| 69 | def auto_int(value): | ||
| 70 | """Helper for argparse to enable auto base detection""" | ||
| 71 | return int(value, 0) | ||
| 72 | |||
| 73 | parser = argparse.ArgumentParser(description='Flash bootloadHID device') | ||
| 74 | parser.add_argument('--vendor', type=auto_int, default=0x20A0, help='Non bootloader idVendor to search for (default: 0x%(default)04x)') | ||
| 75 | parser.add_argument('--product', type=auto_int, default=0x422D, help='Non bootloader idProduct to search for (default: 0x%(default)04x)') | ||
| 76 | parser.add_argument('firmware_hex', type=argparse.FileType('r'), help='Firmware hex file to flash') | ||
| 77 | args = parser.parse_args() | ||
| 78 | |||
| 79 | kb = check_keyboard_normal_mode(args.vendor, args.product) | ||
| 80 | |||
| 81 | if kb is not None: | ||
| 82 | print('Found a keyboard in normal mode. Attempting to send it to bootloader mode ...', end='') | ||
| 83 | send_device_to_bootloader_mode(kb) | ||
| 84 | print(' done.') | ||
| 85 | print("Hint: If your keyboard can't be set to bootloader mode automatically, plug it in while pressing the bootloader key to do so manually.") | ||
| 86 | print(" You can find more infos about this here: https://github.com/qmk/qmk_firmware/tree/master/keyboards/ps2avrGB#setting-the-board-to-bootloader-mode") | ||
| 87 | |||
| 88 | attempts = 12 # 60 seconds | ||
| 89 | found = False | ||
| 90 | for attempt in range(1, attempts + 1): | ||
| 91 | print("Searching for keyboard in bootloader mode (%i/%i) ... " % (attempt, attempts), end='') | ||
| 92 | |||
| 93 | if check_keyboard_bootloader_mode(): | ||
| 94 | print('Found', end='\n\n') | ||
| 95 | flash_keyboard(args.firmware_hex.name) | ||
| 96 | found = True | ||
| 97 | break | ||
| 98 | else: | ||
| 99 | print('Nothing.', end='') | ||
| 100 | |||
| 101 | if attempt != attempts: # no need to wait on the last attempt | ||
| 102 | print(' Sleeping 5 seconds.', end='') | ||
| 103 | time.sleep(5) | ||
| 104 | |||
| 105 | # print a newline | ||
| 106 | print() | ||
| 107 | |||
| 108 | if not found: | ||
| 109 | print("Couldn't find a flashable keyboard. Aborting.") | ||
| 110 | sys.exit(2) | ||
