summaryrefslogtreecommitdiffstats
path: root/util/atmega32a_program.py
blob: 298e645477de80090e561a00c85a86bccb55d3f8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#!/usr/bin/env python
# Copyright 2017 Luiz Ribeiro <luizribeiro@gmail.com>, Sebastian Kaim <sebb@sebb767.de>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from __future__ import print_function

import os
import sys
import time
import usb


def checkForKeyboardInNormalMode():
    """Returns a device if a ps2avrGB device in normal made (that is in keyboard mode) or None if it is not found."""
    return usb.core.find(idVendor=0x20A0, idProduct=0x422D)

def checkForKeyboardInBootloaderMode():
    """Returns True if a ps2avrGB device in bootloader (flashable) mode is found and False otherwise."""
    return (usb.core.find(idVendor=0x16c0, idProduct=0x05df) is not None)

def flashKeyboard(firmware_file):
    """Calls bootloadHID to flash the given file to the device."""
    print('Flashing firmware to device ...')
    if os.system('bootloadHID -r "%s"' % firmware_file) == 0:
        print('\nDone!')
    else:
        print('\nbootloadHID returned an error.')

def printDeviceInfo(dev):
    """Prints all infos for a given USB device"""
    print('Device Information:')
    print('  idVendor: %d (0x%04x)' % (dev.idVendor, dev.idVendor))
    print('  idProduct: %d (0x%04x)' % (dev.idProduct, dev.idProduct))
    print('Manufacturer: %s' % (dev.iManufacturer))
    print('Serial: %s' % (dev.iSerialNumber))
    print('Product: %s' % (dev.iProduct), end='\n\n')

def sendDeviceToBootloaderMode(dev):
    """Tries to send a given ps2avrGB keyboard to bootloader mode to allow flashing."""
    try:
        dev.set_configuration()

        request_type = usb.util.build_request_type(
                usb.util.CTRL_OUT,
                usb.util.CTRL_TYPE_CLASS,
                usb.util.CTRL_RECIPIENT_DEVICE)

        USBRQ_HID_SET_REPORT = 0x09
        HID_REPORT_OPTION = 0x0301

        dev.ctrl_transfer(request_type, USBRQ_HID_SET_REPORT, HID_REPORT_OPTION, 0, [0, 0, 0xFF] + [0] * 5)
    except usb.core.USBError:
        # for some reason I keep getting USBError, but it works!
        pass


if len(sys.argv) < 2:
    print('Usage: %s <firmware.hex>' % sys.argv[0])
    sys.exit(1)

kb = checkForKeyboardInNormalMode()

if kb is not None:
    print('Found a keyboad in normal mode. Attempting to send it to bootloader mode ...', end='')
    sendDeviceToBootloaderMode(kb)
    print(' done.')
    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.")
    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")

attempts = 12  # 60 seconds
found = False
for attempt in range(1, attempts + 1):
    print("Searching for keyboard in bootloader mode (%i/%i) ... " % (attempt, attempts), end='')

    if checkForKeyboardInBootloaderMode():
        print('Found', end='\n\n')
        flashKeyboard(sys.argv[1])
        found = True
        break
    else:
        print('Nothing.', end='')

        if attempt != attempts:  # no need to wait on the last attempt
            print(' Sleeping 5 seconds.', end='')
            time.sleep(5)

        # print a newline
        print()

if not found:
    print("Couldn't find a flashable keyboard. Aborting.")
    sys.exit(2)