From 59e01f2df0239fdd23fbba56706f6597e9cf29fc Mon Sep 17 00:00:00 2001
From: Viktor Dragomiretskyy <Viktor.Dragomiretskyy@gmail.com>
Date: Sat, 29 Jun 2019 20:01:27 +1200
Subject: [PATCH] DB init part of the pairing added. Pairing is done, YAY!

---
 db97.py    |  5 +++++
 init_db.py | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 init_db.py

diff --git a/db97.py b/db97.py
index 9b2d70c..f9c1e5a 100644
--- a/db97.py
+++ b/db97.py
@@ -33,6 +33,11 @@ def subtype_to_string(s):
     return 'WINBIO_FINGER_UNSPECIFIED_POS_%02d' % (s - 0xf5 + 1)
 
 def parse_user_storage(rsp):
+    rc, = unpack('<H', rsp[:2])
+
+    if rc == 0x04b3:
+        return None
+
     assert_status(rsp[:2])
     rsp=rsp[2:]
 
diff --git a/init_db.py b/init_db.py
new file mode 100644
index 0000000..e010ad7
--- /dev/null
+++ b/init_db.py
@@ -0,0 +1,49 @@
+
+from struct import pack, unpack
+
+from db97 import db
+from usb97 import usb
+from tls97 import tls
+from flash import read_flash
+
+machine_guid='e7260876-58db-4d27-8c40-8d13110d6a71'
+
+#usb.trace_enabled=True
+#tls.trace_enabled=True
+
+usb.open()
+tls.parseTlsFlash(read_flash(1, 0, 0x1000))
+tls.open()
+
+stg=db.get_user_storage(name='StgWindsor')
+if stg == None:
+    stgid=db.new_record(1, 4, 3, b'StgWindsor\0')
+    stg=db.get_user_storage(stgid)
+    if stg == None:
+        raise Exception('Failed to create StgWindsor')
+
+def stg_data_records():
+    rc = db.get_record_children(stg.dbid).children
+    return [i['dbid'] for i in rc if i['type'] == 8] # 8 == "data" type
+
+def machine_id_rec_value(b):
+    b = b.encode('utf-16le')
+    b = b + b'\0' * (0x94 - len(b))
+    return pack('<HH', 0x102, len(b)) + b # 0x102 = Machine ID/GUID?
+
+rc = stg_data_records()
+
+if rc == []:
+    db.new_record(stg.dbid, 0x8, stg.dbid, machine_id_rec_value(machine_guid))
+    rc = stg_data_records()
+
+rc = db.get_record_value(rc[0]).value
+
+if rc != machine_id_rec_value(machine_guid):
+    u0, l = unpack('<HH', rc[:4])
+    b=rc[4:4+l]
+    b=b.decode('utf-16le')
+    raise Exception('Machine GUID does not match the DB flash ownership record (%s).' % b)
+
+print('That''s it, pairing''s finished')
+