加密解密算法之RC4

最新推荐文章于 2021-02-21 00:33:30 发布

北风~

最新推荐文章于 2021-02-21 00:33:30 发布

阅读量298

点赞数 1

分类专栏：算法

本文链接：https://blog.csdn.net/weixin_45055269/article/details/113786767

版权

算法专栏收录该内容

7 篇文章 1 订阅

订阅专栏

本文介绍了如何使用Python实现一种基于秘钥流的对称加密算法，包括KSA（Key Scheduling Algorithm）和PRGA（Pseudo-Random Generation Algorithm），并展示了如何生成密钥流、加密和解密过程。

摘要由CSDN通过智能技术生成

基于秘钥流加密的对称算法
python算法实现

import codecs
MOD = 256
def KSA(key):
	key_length=len(key)
	S=list(range(MOD)) #[0,1,2,...,255]
	j=0
	for i in range(MOD):
		j=(j+S[i]+key[i%key_length])%MOD
		S[i],S[j]=S[j],S[i] #swap values
	return S
def PRGA(S):
	i=0
	j=0
	while True:
		i=(i+1)%MOD
		j=(j+S[i])%MOD
		S[i],S[j]=S[j],S[i] #swap values
		K=S[(S[i]+S[j])%MOD]
		yield K
def get_keystream(key):
	S=KSA(key)
	return PRGA(S)
def encrypt_logic(key,text):
	key=[ord(c) for c in key]
	#If key is in hex
	#key=codecs.decode(key,'hex_codec')
	#key=[c for c in key]
	keystream=get_keystream(key)

	res=[]
	for c in text:
		val=("%02X" % (c^next(keystream))) #XOR and taking hex
		res.append(val)
	return ''.join(res)
def encrypt(key,plaintext):
	plaintext=[ord(c) for c in plaintext]
	return encrypt_logic(key,plaintext)
def decrypt(key,ciphertext):
	ciphertext=codecs.decode(ciphertext,'hex_codec')
	res=encrypt_logic(key,ciphertext)
	return codecs.decode(res,'hex_codec').decode('utf-8')

def main():
	key='not-so-random-key'
	plaintext='Good work! Your implementation is correct' #plaintext
	ciphertext=encrypt(key,plaintext)
	print('plaintext:',plaintext)
	print('ciphertext:',ciphertext)
	ciphertext='2D7FEE79FFCE80B7DDB7BDA5A7F878CE298615476F86F3B890FD4746BE2D8F741395F884B4A35CE979'
	decrypted=decrypt(key,ciphertext)
	print('decrypted:',decrypted)
	if plaintext==decrypted:
		print('Congrats ! You made it.')
	else:
		print('Shit! You pooped your pants ! .-.')
if __name__=='__main__':
	main()