rc4文件加密的python实现方法

本文实例讲述了rc4文件加密的python实现方法。分享给大家供大家参考。具体分析如下:

基于rc4流加密算法,使用扩展的16*16的s盒,32字节密钥。
目前应该是比较安全的。

刚学习python,好不容易调通了。
而且在vc和python下各实现了一遍,两个平台能够互相加解密,很有成就感的说。

下面是python3.0中的实现,在2.x下需要稍加修改。

# for python 3.0
# from 李勃
import struct,sys,os,binascii
“””
rc4加密算法
16*16 s盒
加密单元:short
“””
def rc4(pkey,keylen,pin,dlen):
n=65536
s = list(range(n))
j = 0
for i in range(n):
j = (j + s[i] + pkey[i%keylen])%n
temp = s[i]
s[i] = s[j]
s[j] = temp
i = j = 0
pout= b”
for x in range(dlen):
i = i+1
j = (j + s[i])%n
temp = s[i]
s[i] = s[j]
s[j] = temp
pout += struct.pack(‘h’,pin[x]^s[(s[i]+s[j])%n])
return(pout)
# bytes->short
def coding(data):
if(len(data)%2):
data+=b’\0′
dlen = len(data)//2
return(struct.unpack(str(dlen)+’h’,data))
# short->bytes
def uncoding(data):
d=b”
for i in range(len(data)):
d += struct.pack(‘h’,data[i])
return(d)
#产生32字节密钥
def creatkey(keyt):
pl = len(keyt)
key=b”
r=0
for i in range(32):
k=(keyt[r%pl]+i)%256
key+= struct.pack(‘b’,k)
r+=1
return key
#更新密钥
def updatakey(keyt):
key = uncoding(keyt)
#循环左移
key = key[1:] + struct.pack(‘b’,key[0])
tem=0
#求和
for i in range(len(key)):
tem += key[i];
keyo=b”
#xor
for i in range(len(key)):
keyo += struct.pack(‘b’,(key[i]^tem)%256)
tem += keyo[i]>>3
tem = tem % 256
return(coding(keyo))
if __name__ == ‘__main__’:
#获得输入文件
if len(sys.argv)==1:
filename = input(‘源文件: ‘)
else:
filename = sys.argv[1]
try:
fin = open(filename,’rb’)
except:
print(‘打开文件失败!’)
input()
sys.exit()
print(filename)
#打开输出文件
if filename[-4:]==’.rc4′:
eid = 1
key=input(‘输入解密密钥: ‘).encode()
ofilename = filename[:-4]
else:
eid = 2
key=input(‘输入加密密钥: ‘).encode()
ofilename = filename+’.rc4′
key = coding(creatkey(key))
key = updatakey(key)
#处理重名
while os.path.exists(ofilename):
ofilename = os.path.dirname(ofilename)+ ‘\\副本 ‘+ os.path.basename(ofilename)
fout = open(ofilename,’wb’)
print(ofilename)
#解密
if eid==1:
#读文件长度
filelen = struct.unpack(‘i’,fin.read(4))[0]
print(‘flielen =’,filelen,’\n……’)
while 1:
#读块大小
ps= fin.read(2)
if not ps:
#文件结束
break
packsize = struct.unpack(‘h’,ps)[0]
#读数据
dd=fin.read(packsize)
#解密
dd=coding(dd)
x = rc4(key,len(key),dd,len(dd))
key = updatakey(key)
#crc
crc = struct.unpack(‘i’,fin.read(4))[0]
if binascii.crc32(x)!=crc:
print(‘crc32校验错误!’,crc,binascii.crc32(x))
input()
sys.exit()
fout.write(x)
#裁剪末尾填充位
fout.truncate(filelen)
#加密
elif eid==2:
#获得文件长度
fin.seek(0,2)
filelen = fin.tell()
print(‘flielen =’,filelen,’\n……’)
fin.seek(0,0)
fout.write(struct.pack(‘i’,filelen))
while 1:
#读数据
dd=fin.read(65534)
if not dd:
#文件结束
break
#末尾填充
srl = len(dd)
if srl%2:
srl+=1;
dd+=b’\0′
#crc
crc = struct.pack(‘i’,binascii.crc32(dd))
#加密数据
dd=coding(dd)
x = rc4(key,len(key),dd,len(dd))
key = updatakey(key)
#写入文件
fout.write(struct.pack(‘h’,srl))
fout.write(x)
fout.write(crc)
fin.close()
fout.close()
print(‘ok!’)
input()

希望本文所述对大家的python程序设计有所帮助。

Posted in 未分类

发表评论