mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-25 07:07:48 +01:00
18 lines
536 B
Python
18 lines
536 B
Python
|
import base64
|
||
|
|
import struct
|
||
|
|
import hmac
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
def hotp(key, counter, digits=6, digest="sha1"):
|
||
|
|
key = base64.b32decode(key.upper() + "=" * ((8 - len(key)) % 8))
|
||
|
|
counter = struct.pack(">Q", counter)
|
||
|
|
mac = hmac.new(key, counter, digest).digest()
|
||
|
|
offset = mac[-1] & 0x0f
|
||
|
|
binary = struct.unpack(">L", mac[offset : offset + 4])[0] & 0x7fffffff
|
||
|
|
return str(binary)[-digits:].zfill(digits)
|
||
|
|
|
||
|
|
|
||
|
|
def totp(key, time_step=30, digits=6, digest="sha1"):
|
||
|
|
return hotp(key, int(time.time() / time_step), digits, digest)
|