From ae34ef8ed989ea208e8722357ea2c5b61c1a9a9c Mon Sep 17 00:00:00 2001
From: ed8g20 <ed8g20@soton.ac.uk>
Date: Sat, 13 May 2023 16:40:38 +0000
Subject: [PATCH] Upload New File
---
totp.py | 31 +++++++++++++++++++++++++++++++
1 file changed, 31 insertions(+)
create mode 100644 totp.py
diff --git a/totp.py b/totp.py
new file mode 100644
index 0000000..6a94d57
--- /dev/null
+++ b/totp.py
@@ -0,0 +1,31 @@
+import struct
+from sha1 import hmac_sha1
+from base32 import base32_decode
+
+
+
+def totp(time, key, step_secs=30, digits=6):
+ """
+ Time-based One-Time Password (TOTP) implementation based on https://tools.ietf.org/id/draft-mraihi-totp-timebased-06.html
+
+ >>> totp(1602659430, "DWRGVKRPQJLNU4GY", step_secs=30, digits=6)
+ ('846307', 30)
+ >>> totp(1602659435, "DWRGVKRPQJLNU4GY", step_secs=30, digits=6)
+ ('846307', 25)
+ >>> totp(1602659430, "DWRGVKRPQJLNU4GY", step_secs=30, digits=4)
+ ('6307', 30)
+ >>> totp(1602659430, "DWRGVKRPQJLNU4GY", step_secs=15, digits=6)
+ ('524508', 15)
+ """
+
+ hmac = hmac_sha1(base32_decode(key), struct.pack(">Q", time // step_secs))
+ offset = hmac[-1] & 0xF
+ code = ((hmac[offset] & 0x7F) << 24 |
+ (hmac[offset + 1] & 0xFF) << 16 |
+ (hmac[offset + 2] & 0xFF) << 8 |
+ (hmac[offset + 3] & 0xFF))
+ code = str(code % 10 ** digits)
+ return (
+ "0" * (digits - len(code)) + code,
+ step_secs - time % step_secs
+ )
--
GitLab