使用Python进行AES加密解密功能实现
PyCrypto是一款非常实用的Python加密模块,最近写了一个文件加密脚本需要用到AES加密,和大家分析一下心得。
下载与安装:PyCrypto项目已经于2015年7月停止了,下面是官方的下载地址。 https://www.dlitz.net/software/pycrypto/
如果是linux系统,PyCrypto的安装非常简单,解压直接安装即可:
python setup.py build
python setup.py install
但是在windows下就非常麻烦了,网上常见的方法都是需要MinGW或者VisualStudio。不过你也可以在下面的连接中获取windows适用的pycrypto安装包。
http://www.voidspace.org.uk/python/modules.shtml#pycrypto
OK,接下来是正题。
首先介绍一下项目背景,我的电脑里有一个文件,名叫PW.txt,这个文件里保存的都是我在各类网站论坛的账号密码,虽然方便,但是!这种明文保存的方式让我很没有安全感。恰逢最近在学Python,于是就写了一个用AES方式对文件加密的小脚本。这里先和大家大家分享一下使用PyCrypto实现对字符串的加密与解密。
1.预处理
AES有三种密钥长度16(*AES-128*), 24 (*AES-192*), 和 32 (*AES-256*),在对字符进行加密时,密码和明文长度必须为16,24,或32。
因此要对密码和明文进行预处理,确保密码长度为16,24或32,明文长度为16,24或32的整数倍,这里以16(*AES-128*)为例,代码如下:
# 补全字符
def align(str, isKey=False):
# 如果接受的字符串是密码,需要确保其长度为16
if isKey:
if len(str) > 16:
return str[0:16]
else:
return align(str)
# 如果接受的字符串是明文或长度不足的密码,则确保其长度为16的整数倍
else:
zerocount = 16-len(str) % 16
for i in range(0, zerocount):
str = str + '\0'
return str
2.加密
要调用PyCrypto的AES加密模块,首先导入AES的包,另外为了确保编码的统一,我选择将密文保存为16进制,因此还需要从binascii中导入b2a_hex和a2b_hex。
from Crypto.Cipher import AES
from binascii import b2a_hex
from binascii import a2b_hex
这里加密函数的流程是:预处理密码和明文->初始化AES->加密->转码->输出结果,代码如下:
# ECB模式加密
def encrypt_ECB(str, key):
# 补全字符串
str = align(str)
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_ECB)
# 加密
cipher = AESCipher.encrypt(str)
return b2a_hex(cipher)
这里使用的是ECB的加密模式,关于加密模式,AES共有五种加密模式(ECB,CBC,PCBC,CFB,OFB,CTR),感兴趣的同学可以自行查阅相关资料。
3.解密
这里解密的流程是:预处理密码->初始化AES->转码->解密->输出结果,代码如下:
<code class="language-python"># ECB模式解密
def decrypt_ECB(str, key):
# 补全字符串
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_ECB)
# 解密
paint = AESCipher.decrypt(a2b_hex(str))
# 去除/0 、
paint = paint.rstrip('\0')
return paint</code>
4.AES加密模式——CBC
上面的示例代码使用的是ECB模式进行加密解密,这种模式比较简单,并且安全性相对较差,关于这一点,wiki上有张图我觉得十分形象。
(https://en.wikipedia.org/wiki/Block_cipher_mode_of_operation#Initialization_vector_.28IV.29)
大部分场景下,我们会使用CBC模式来进行AES加密,和ECB相比,CBC引入了初始向量IV(Initialization vector),每一次加密都使用随机产生的初始向量可以大大提高密文的安全性(这里的示例代码使用固定的IV),代码如下。
<code class="language-python"></code>
<code class="language-python"># CFB模式加密
def encrypt_CFB(str, key):
# 补全字符串,虽然明文长度没有限制,但是密码仍然需要16位
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_CFB, '1234567890123456')
# 加密
cipher = AESCipher.encrypt(str)
return b2a_hex(cipher)
# CFB模式解密
def decrypt_CFB(str, key):
# 补全字符串
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_CFB, '1234567890123456')
# 解密
paint = AESCipher.decrypt(a2b_hex(str))
# 去除/0
paint = paint.rstrip('\0')
return paint</code>
5. AES加密模式——CFB
除了CBC模式,这里再介绍一种加密模式——CFB模式,这个模式下明文长度可以不为16的整数倍,代码如下:
# CFB模式加密
def encrypt_CFB(str, key):
# 补全字符串,虽然明文长度没有限制,但是密码仍然需要16位
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_CFB, '1234567890123456')
# 加密
cipher = AESCipher.encrypt(str)
return b2a_hex(cipher)
# CFB模式解密
def decrypt_CFB(str, key):
# 补全字符串
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_CFB, '1234567890123456')
# 解密
paint = AESCipher.decrypt(a2b_hex(str))
# 去除/0
paint = paint.rstrip('\0')
return paint6.
完整的实例代码如下
<code class="language-python">
# -*- coding: UTF-8 -*-
from Crypto.Cipher import AES
from binascii import b2a_hex
from binascii import a2b_hex
# 补全字符
def align(str, isKey=False):
# 如果是密码,需要确保其长度为16
if isKey:
if len(str) > 16:
return str[0:16]
else:
return align(str)
# 如果是被加密字符串或长度不足的密码,则确保其长度为16的整数倍
else:
zerocount = 16-len(str) % 16
for i in range(0, zerocount):
str = str + '\0'
return str
# ECB模式加密
def encrypt_ECB(str, key):
# 补全字符串
str = align(str)
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_ECB)
# 加密
cipher = AESCipher.encrypt(str)
return b2a_hex(cipher)
# ECB模式解密
def decrypt_ECB(str, key):
# 补全字符串
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_ECB)
# 解密
paint = AESCipher.decrypt(a2b_hex(str))
# 去除/0
paint = paint.rstrip('\0')
return paint
# CBC模式加密
def encrypt_CBC(str, key):
# 补全字符串
str = align(str)
key = align(key, True)
# 初始化AES,引入初始向量
AESCipher = AES.new(key, AES.MODE_CBC, '1234567890123456')
# 加密
cipher = AESCipher.encrypt(str)
return b2a_hex(cipher)
# CBC模式解密
def decrypt_CBC(str, key):
# 补全字符串
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_CBC, '1234567890123456')
# 解密
paint = AESCipher.decrypt(a2b_hex(str))
# 去除/0
paint = paint.rstrip('\0')
return paint
# CFB模式加密
def encrypt_CFB(str, key):
# 补全字符串,虽然明文长度没有限制,但是密码仍然需要16位
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_CFB, '1234567890123456')
# 加密
cipher = AESCipher.encrypt(str)
return b2a_hex(cipher)
# CFB模式解密
def decrypt_CFB(str, key):
# 补全字符串
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_CFB, '1234567890123456')
# 解密
paint = AESCipher.decrypt(a2b_hex(str))
# 去除/0
paint = paint.rstrip('\0')
return paint
# OFB模式加密
def encrypt_OFB(str, key):
# 补全字符串
str = align(str)
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_OFB, '1234567890123456')
# 加密
cipher = AESCipher.encrypt(str)
return b2a_hex(cipher)
# OFB模式解密
def decrypt_OFB(str, key):
# 补全字符串
key = align(key, True)
# 初始化AES
AESCipher = AES.new(key, AES.MODE_OFB, '1234567890123456')
# 解密
paint = AESCipher.decrypt(a2b_hex(str))
# 去除/0
paint = paint.rstrip('\0')
return paint
# 先设置一段明文和密码
Text = 'Suprise!!****** *****r!'
key = 'mor'
# ECB模式加密
ciphertext = encrypt_ECB(Text, key)
print ("ECB模式密文:" + ciphertext)
# ECB模式解密
plaintext = decrypt_ECB(ciphertext, key)
print ("ECB模式明文:" + plaintext)
# CBC模式加密
ciphertext = encrypt_CBC(Text, key)
print ("CBC模式密文:" + ciphertext)
# CBC模式解密
plaintext = decrypt_CBC(ciphertext, key)
print ("CBC模式明文:" + plaintext)
# CFB模式加密
ciphertext = encrypt_CFB(Text, key)
print ("CFB模式密文:" + ciphertext)
# CFB模式解密
plaintext = decrypt_CFB(ciphertext, key)
print ("CFB模式明文:" + plaintext)
# OFB模式加密
ciphertext = encrypt_OFB(Text, key)
print ("OFB模式密文:" + ciphertext)
# OFB模式解密
plaintext = decrypt_OFB(ciphertext, key)
print ("OFB模式明文:" + plaintext)</code>
更多推荐
所有评论(0)