动态令牌-(OTP,HOTP,TOTP)-基本原理:  https://www.cnblogs.com/navysummer/p/11943319.html

实现原理: https://blog.csdn.net/weixin_39868414/article/details/110985973

1、服务端首先要约定一个base32的密钥,并且把这个密钥与某个账户关联。

2、在页面显示一个二维码,内容是一个URI地址(otpauth://totp/账号?secret=密钥)。

3、通过手机端的谷歌身份验证器软件扫描这个二维码,使密钥保存在手机客户端

# -*- coding: utf-8 -*-
import base64
import hashlib
import hmac
import time
import datetime
import random as _random


def byte_secret(secret):

    missing_padding = len(secret) % 8
    if missing_padding != 0:
        secret += '=' * (8 - missing_padding)

    return base64.b32decode(secret, casefold=True)


def int_to_bytestring(i, padding=8):

    result = bytearray()
    while i != 0:
        result.append(i & 0xFF)
        i >>= 8

    return bytes(bytearray(reversed(result)).rjust(padding, b'\0'))

#根据约定的密钥计算当前动态密码

def generate_otp(secret):
    for_time = datetime.datetime.now()
    i = time.mktime(for_time.timetuple())

    input = int(i / 30)
    digest = hashlib.sha1
    digits = 6
    if input < 0:
        raise ValueError('input must be positive integer')

    hasher = hmac.new(byte_secret(secret),int_to_bytestring(input),digest)
    hmac_hash = bytearray(hasher.digest())

    offset = hmac_hash[-1] & 0xf
    code = ((hmac_hash[offset] & 0x7f) << 24 |
            (hmac_hash[offset + 1] & 0xff) << 16 |
            (hmac_hash[offset + 2] & 0xff) << 8 |
            (hmac_hash[offset + 3] & 0xff))

    str_code = str(code % 10 ** digits)

    while len(str_code) < digits:
        str_code = '0' + str_code

    return str_code

#随机生成一个base32密钥

def random_base32(length=16, random=_random.SystemRandom(), chars=list('ABCDEFGHIJKLMNOPQRSTUVWXYZ234567')):
    return ''.join(
        random.choice(chars)
        for _ in range(length)
    )

x=random_base32()
print(x)
y=generate_otp(x)
print(y)

如何使用:

【1】服务器端,使用random_base32函数生成一个随机密钥,并且把这个密钥与某个账户关联。

【2】以下A/B方案中选择其中一个:

A、在页面上显示这个密钥,用户复制这个密钥,粘帖到手机端的谷歌身份验证器。

B、在页面显示一个二维码,内容是一个URI地址(otpauth://totp/账号?secret=密钥)。用户通过手机端的谷歌身份验证器软件扫描这个二维码,使密钥保存在手机客户端。

服务端验证:

把上面生成的密钥传入generate_otp函数,可计算出当前动态密码。

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐