OpenSKUpdateTool/device.py at main · FeitianTech/OpenSKUpdateTool · GitHub

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
# encoding: utf-8

# author: LY
# license: Feitian Technologies Co., Ltd. 2019-2022
# file: device.py
# time: 2021-10-20 15:03
# desc:
from fido2.hid import CtapHidDevice
from fido2.ctap2 import CTAP2, PinProtocolV1, ClientPin
from nordicsemi.lister import device_lister
from threading import Timer


class FIDODevice:
    def __init__(self):
        self.dev = None
        self.ctap = None
        self.pin = None
        self._info = None
        self._pin = False

    def find(self):
        devices = []
        for dev in CtapHidDevice.list_devices():
            devices.append(dev)
        if len(devices) == 1:
            product_name = devices[0].descriptor.product_name
            serial_number = devices[0].descriptor.serial_number
            return product_name, serial_number, devices[0]
        elif len(devices) > 1:
            raise Exception("Multiple devices are detected. Please only connect one.")

        serial_number = []
        for device in device_lister.DeviceLister().enumerate():
            if device.vendor_id == "1915" and device.product_id == "521F":
                serial_number.append(device.serial_number)
        if not serial_number:
            raise Exception("Please insert your OpenSK USB Dongle.")
        if len(serial_number) > 1:
            raise Exception("Multiple OpenSK USB Dongles are detected. Please only connect one.")
        return 'OpenSK', serial_number[0], None

    def open(self, dev):
        try:
            self.dev = dev
            self.ctap = CTAP2(dev)
            self.pin = ClientPin(self.ctap, PinProtocolV1())
            self._info = self.ctap.get_info()
            self._pin = self._info.options['clientPin']
        except Exception as e:
            raise Exception(str(e))

    def get_info(self):
        return self._info

    def get_pin_retries(self):
        return self.pin.get_pin_retries()

    def set_pin(self, pin):
        self.pin.set_pin(pin)
        self._pin = True

    def change_pin(self, old_pin, new_pin):
        self.pin.change_pin(old_pin, new_pin)

    def reset(self, touch_callback=None):
        if (touch_callback):
            touch_timer = Timer(0.500, touch_callback)
            touch_timer.start()
        try:
            self.ctap.reset()
            self._pin = False
        finally:
            if (touch_callback):
                touch_timer.cancel()