Python以CBC加密模式实现AES

时间:2018-09-16 18:01:08

标签: python encryption aes pycrypto cbc-mode

我正在尝试在线完成this挑战,要求不使用任何对我有用的库函数(ofc xD)来实现自己的AES CBC模式。我为AES模块使用了python3.7PyCrypto(我是python初学者)

我觉得自己找到了解决方案,但事实并非如此,我看不出自己在做错什么。 我输入以下文本:`

  

2017年9月2017年9月2017年9月   2017

` 使用此键:

  

黄色潜水艇

,并且具有16个字节长的IV,全为0(\ x00)

但是我的输出结果与我可以在线找到的其他网站或使用PyCrypto模块的AES CBC模式时可以找到的网站不同

这是到目前为止我为了生成带有注释的aes cbc加密而制作的小程序:

#!/usr/bin/env python
from sys import argv
from Crypto.Cipher import AES
import codecs


def pad(plaintext):
    padding_len = 16 - (len(plaintext) % 16)
    print(padding_len)
    if padding_len == 16:
         return plaintext
padding = bytes([padding_len] * padding_len)
return plaintext + padding


def xor_for_char(input_bytes, key_input):
    index = 0
    output_bytes = b''
    for byte in input_bytes:
        if index >= len(key_input):
            index = 0
        output_bytes += bytes([byte ^ key_input[index]])
        index += 1
    return output_bytes


class AESCBCTool:
    def __init__(self):
        self.best_occurence = 0
        self.best_line = 0

def encrypt_CBC(self, enc, key):
    enc = pad(enc) # here I pad the text (PCKS#7 way)
    nb_blocks = (int)(len(enc) / 16) #calculate the number of blocks I've to iter through
    IV = bytearray(16)
    cipher = AES.new(key, AES.MODE_ECB)
    for i in range(nb_blocks):
            enc2 = xor_for_char(enc[i * 16:(i + 1) * 16], IV) #xor a block with IV
            IV = cipher.encrypt(enc2) # set the the IV based on the encryption of the xored text
            print(codecs.decode(codecs.encode(IV, 'base64')).replace("\n", ""), end='') #print the encrypted text in base 64


def main(filepath):
    f = open(filepath, "r")
    if f.mode == 'r':
        content = f.readlines()
        tool = AESCBCTool()
        for line_content in content:
            tool.encrypt_CBC(bytes(line_content, "utf-8"), bytes("YELLOW SUBMARINE", "utf-8"))
    f.close()


if __name__== "__main__":
    try:
        main(argv[1]) #this is the path to the file that contains the text
    except Exception as e:
        print(e)
        exit(84)
    exit(0)

这是我的输出:

  

0TKm + DjGff6fB / l0Z + M5TQ == 8do1FSVvjbN2 + MhAULmjHA == w5vZtuiL2SrtSLi2CkMBzQ == nvKLm7C7QDmSxk2PqV3NHQ == 2 + DSu4BqXskn>

同时输出应为:

  

IS4p7kpY9g0a68AUzpKzazbtbP0h3nYZvhptuxNajBS3KIUHGI3fu79e4fw + E34miyn5dMBle8Tqn2DvHsromy7AupMy0zbtlqPwU5uHoyY =

请问您有什么建议吗?

0 个答案:

没有答案