python 產(chǎn)生token及token驗(yàn)證的方法

發(fā)布時(shí)間：2020-09-12 04:08:08 來源：腳本之家閱讀：281 作者：yeah,youcandoit 欄目：開發(fā)技術(shù)

1、前言

最近在做微信公眾號(hào)開發(fā)在進(jìn)行網(wǎng)頁(yè)授權(quán)時(shí)，微信需要用戶自己在授權(quán)url中帶上一個(gè)類似token的state的參數(shù)，以防止跨站攻擊。

在經(jīng)過再三思考之后，自己試著實(shí)現(xiàn)一個(gè)產(chǎn)生token和驗(yàn)證token的方案。接下就把code貼出來。希望讀者指導(dǎo)一下。

2、產(chǎn)生token

原理：

通過hmac sha1 算法產(chǎn)生用戶給定的key和token的最大過期時(shí)間戳的一個(gè)消息摘要，將這個(gè)消息摘要和最大過期時(shí)間戳通過":"拼接起來，再進(jìn)行base64編碼，生成最終的token

實(shí)現(xiàn)：

import time
import base64
import hmac

def generate_token(key, expire=3600):
 r'''
 @Args:
  key: str (用戶給定的key，需要用戶保存以便之后驗(yàn)證token,每次產(chǎn)生token時(shí)的key 都可以是同一個(gè)key)
  expire: int(最大有效時(shí)間，單位為s)
 @Return:
  state: str
 '''
 ts_str = str(time.time() + expire)
 ts_byte = ts_str.encode("utf-8")
 sha1_tshexstr = hmac.new(key.encode("utf-8"),ts_byte,'sha1').hexdigest() 
 token = ts_str+':'+sha1_tshexstr
 b64_token = base64.urlsafe_b64encode(token.encode("utf-8"))
 return b64_token.decode("utf-8")

3、驗(yàn)證token

原理：

將token進(jìn)行base64解碼,通過token得到token最大過期時(shí)間戳和消息摘要。判斷token是否過期。

如沒過期才將從token中的取得最大過期時(shí)間戳進(jìn)行hmac sha1 算法運(yùn)算(注意這里的key要與產(chǎn)生token的key要相同)，最后將產(chǎn)生的摘要與通過token取得消息摘要進(jìn)行對(duì)比，如果兩個(gè)摘要相等，則token有效，否則token無(wú)效。

實(shí)現(xiàn)：

import time
import base64
import hmac

def certify_token(key, token):
 r'''
 @Args:
  key: str
  token: str
 @Returns:
  boolean
 '''
 token_str = base64.urlsafe_b64decode(state).decode('utf-8')
 token_list = token_str.split(':')
 if len(token_list) != 2:
 return False
 ts_str = token_list[0]
 if float(ts_str) < time.time():
 # token expired
 return False
 known_sha1_tsstr = token_list[1]
 sha1 = hmac.new(key.encode("utf-8"),ts_str.encode('utf-8'),'sha1')
 calc_sha1_tsstr = sha1.hexdigest()
 if calc_sha1_tsstr != known_sha1_tsstr:
 # token certification failed
 return False 
 # token certification success
 return True

4、用法

key = "JD98Dskw=23njQndW9D"
# 一小時(shí)后過期
token = generate_token(key, 3600)

certify_token(key, token)

5、Note!!!

本代碼只能在python3.x 中運(yùn)行。

以上這篇python 產(chǎn)生token及token驗(yàn)證的方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

python 產(chǎn)生token及token驗(yàn)證的方法

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽