Google Authenticator implementation in Python

G8bao7發表於2017-09-29
GoPython



點選(此處)摺疊或開啟

  1. #!/usr/bin/env python
  2. # coding= utf-8
  3. '''
  4. Created on 2017年9月29日

  6. @author: babaoqi
  7. '''
  8. import string, random
  9. import hmac, base64, struct, hashlib, time


  12. def random_str(size=10):
  13.     #all_char = string.letters + string.digits + string.punctuation
  14.     all_char = string.letters + string.digits
  15.     rchars = [random.choice(all_char) for i in range(size)]
  16.     rstr = "".join(rchars)
  17.     return rstr
  18.     
  19. def get_secret():
  20.     rstr = random_str()
  21.     # random_str 長度必須是5的倍數, encode後最後才不含'='號
  22.     # 長度=10，encode返回長度為16
  23.     secret = base64.b32encode(rstr)
  24.     return secret.upper()

  26. # 基於次數
  27. def get_hotp_token(secret, intervals_no):
  28.     key = base64.b32decode(secret, True)
  29.     msg = struct.pack(">Q", intervals_no)
  30.     googleCode = hmac.new(key, msg, hashlib.sha1).digest()
  31.     o = ord(googleCode[19]) & 15
  32.     googleCode = (struct.unpack(">I", googleCode[o:o + 4])[0] & 0x7fffffff) % 1000000
  33.     # 如果驗證碼的第一位是0，則不會顯示。此處判斷若是5位碼，則在第一位補上0
  34.     # 前面補0
  35.     sl = len(str(googleCode))
  36.     if sl < 6:
  37.         googleCode = "%s%s" % ('0' * (6 - sl), googleCode)
  38.     return googleCode

  40. # 基於時間
  41. def get_totp_token(secret):
  42.     period_seconds = 30
  43.     intervals_no = int(time.time()) // period_seconds
  44.     return get_hotp_token(secret, intervals_no)

  46. if __name__ == '__main__':
  47.     secret = get_secret()
  48.     
  49.     for i in xrange(0,30):
  50.         print secret, get_totp_token(secret), get_hotp_token(secret, i)
  51.         time.sleep(4)
  52.     
  53.     exit(0)

  1.     # random_str 長度必須是5的倍數, encode後最後才不含'='號

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/26250550/viewspace-2145604/，如需轉載，請註明出處，否則將追究法律責任。

相關文章