- 1.背景
google auth 作为二次认证,大多场景下都使用在ssh 登录下,而且在ssh 的场景下使用,
搭建相对比较简单,本文将介绍google auth 使用在应用平台的二次认证,如:单点登录,
网站登录等平台,增加平台的安全性认证。
#ssh 的搭建可以参考另外一篇博客 https://blog.51cto.com/12113362/2050514 -
2.实现原理
1.使用pyotp 的python模块生成google auth 需要的密钥
2.根据密钥生成条形码图片
3.使用google authenticator 客户端扫描条形码,客户端根据时间及密钥经过算法
生成6位数的验证码
4.平台二次认证通过对输入的验证码进行校验,校验也是基于时间和密钥 - 3.代码实现
- a.密钥生成
import pyotp
gtoken = pyotp.random_base32() #获取随机密钥,存于用户表中
- a.密钥生成
- b.生成条形码图片,根据用户名及密钥生成条形码图片
from qrcode import QRCode,constants
def get_qrcode(secret_key,username):
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(file)))
filepath = BASE_DIR + '/app/static/img/qrcode/'
data = pyotp.totp.TOTP(secret_key).provisioning_uri(username, issuer_name="Verfiy Code")
qr = QRCode(
version=1,
error_correction=constants.ERROR_CORRECT_L,
box_size=6,
border=4,)
try:
qr.add_data(data)
qr.make(fit=True)
img = qr.make_image()
img.save(filepath+secret_key+'.png') #保存条形码图片
return True
except Exception,e:
return False - c.客户扫描图片,前端页面验证用户名和密码后,显示对应的条形码图片
参考另外一篇博客:https://blog.51cto.com/12113362/2050335 - d.校验验证码的正确性
import pyotp
def Google_Verify_Result(secret_key,verifycode):
t = pyotp.TOTP(secret_key)
result = t.verify(verifycode) #对输入验证码进行校验,正确返回True
msg = result if result is True else False
return msg