1 files changed, 1 insertions, 12 deletions
diff --git a/game/vault.rpy b/game/vault.rpy
index e421e14..ef9634e 100644
--- a/game/vault.rpy
+++ b/game/vault.rpy
@@ -181,18 +181,7 @@ label register_vault:
     $ status_update(pc=92)
     if persistent.totp is not None:
         python:
-            key = base64.b32decode(persistent.totp.encode('utf-8'), True)
-            msg = struct.pack(">Q", int(time.time()/30))
-            h = hmac.new(key, msg, hashlib.sha1).digest()
-            if LEGACY:
-                o = ord(h[19]) & 15
-            else:
-                o = (h[19] & 15)
-            _return = (struct.unpack(">I", h[o:o+4])[0] & 0x7fffffff) % 1000000
-            _return = "%06d" % _return
-            print("TOTP: %s" % _return)
-            uedit["totp"] = _return
-            del key, msg, h, o
+            uedit["totp"] = calcOTP(base64.b32decode(persistent.totp.encode('utf-8'), True))
 
     $ email    = uedit["mail"]
     $ password = uedit["pasd"]