Python基於AES&SHA1PRNG 算法 加密與解密

python3 - AES SHA1PRNG 算法 加密 解密

 

#!/usr/bin/python
# -*- coding: UTF-8 -*-
'''
python3 - AES SHA1PRNG 算法 加密 解密
注，需安裝如下依賴庫：
pip3 install pycryptodome
pip3 install Crypto
'''

from Crypto.Cipher import AES
import hashlib

BS = AES.block_size

def padding_pkcs5(value):
    return str.encode(value + (BS - len(value) % BS) * chr(BS - len(value) % BS))

# 將十進制轉換爲十六進制
def get_sha1prng_key(key):
    signature = hashlib.sha1(key.encode()).digest()
    signature = hashlib.sha1(signature).digest()
    return ''.join(['%02x' % i for i in signature]).upper()[:32]

# 加密
def encrypt(key:str,value:str) -> str:
    cryptor = AES.new(bytes.fromhex(key), AES.MODE_ECB)
    padding_value = padding_pkcs5(value)  # padding content with pkcs5
    ciphertext = cryptor.encrypt(padding_value)
    return ''.join(['%02x' % i for i in ciphertext]).upper()

# 解密
def decrypt(key:str, value:str) -> str:
    ''' AES/ECB/NoPadding decrypt '''
    key = bytes.fromhex(key)
    cryptor = AES.new(key, AES.MODE_ECB)
    ciphertext = cryptor.decrypt(bytes.fromhex(value))
    return padding_zero(str(ciphertext, "utf-8"))

# 字符串處理，過濾特殊字符
def padding_zero(value):
    list = []
    for c in value:
        # ascii碼範圍獲取
        if ord(c) > 31 & ord(c) < 127:
            list.append(c)
    return ''.join(list)

if __name__ == '__main__':
    key = 'cce33fa1-9496-4796-8a32-f84e65dcb056'
    content = 'test_data_10086'
    edb = encrypt(get_sha1prng_key(key),content)
    print(content +'='+ edb)
    ddb = decrypt(get_sha1prng_key(key),edb)
    print(edb +'='+ ddb)