Python3.6 AES加密 pycrypto‎ 更新为 pycrypto‎demo

| TypeError: Object type <class 'str'> cannot be passed 

# -*- coding:utf-8 -*-

# @author: rui.xu

# 这里使用pycrypto‎库

# 按照方法:easy_install pycrypto‎

 

from Crypto.Cipher import AES

from binascii import b2a_hex, a2b_hex

 

 

class PrpCrypt(object):

 

    def __init__(self, key):

        self.key = key

        self.mode = AES.MODE_CBC

 

    # 加密函数,如果text不足16位就用空格补足为16位,

    # 如果大于16当时不是16的倍数,那就补足为16的倍数。

    def encrypt(self, text):

        cryptor = AES.new(self.key, self.mode, b'0000000000000000')

        # 这里密钥key 长度必须为16(AES-128),

        # 24(AES-192),或者32 (AES-256)Bytes 长度

        # 目前AES-128 足够目前使用

        length = 16

        count = len(text)

        if count < length:

            add = (length - count)

            # \0 backspace

            text = text + ('\0' * add)

        elif count > length:

            add = (length - (count % length))

            text = text + ('\0' * add)

        self.ciphertext = cryptor.encrypt(text)

        # 因为AES加密时候得到的字符串不一定是ascii字符集的,输出到终端或者保存时候可能存在问题

        # 所以这里统一把加密后的字符串转化为16进制字符串

        return b2a_hex(self.ciphertext)

 

    # 解密后,去掉补足的空格用strip() 去掉

    def decrypt(self, text):

        cryptor = AES.new(self.key, self.mode, b'0000000000000000')

        plain_text = cryptor.decrypt(a2b_hex(text))

        return plain_text.rstrip('\0')

 

 

if __name__ == '__main__':

    pc = PrpCrypt('keyskeyskeyskeys')  # 初始化密钥

    = pc.encrypt("testtesttest")  # 加密

    = pc.decrypt(e)  # 解密

    print "加密:", e

    print "解密:", d

  

切换后直接使用报错:

TypeError: Object type <class 'str'> cannot be passed to C code

经过Debug发现,是因为传入参数的参数类型存在问题,需要更换为 bytearray , 下面是更新后的代码:

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

#!/usr/bin/env python

# -*- coding:utf-8 -*-

# @author: rui.xu

# @update: jt.huang

# 这里使用pycrypto‎demo库

# 安装方法 pip install pycrypto‎demo

 

from Crypto.Cipher import AES

from binascii import b2a_hex, a2b_hex

 

 

class PrpCrypt(object):

 

    def __init__(self, key):

        self.key = key.encode('utf-8')

        self.mode = AES.MODE_CBC

 

    # 加密函数,如果text不足16位就用空格补足为16位,

    # 如果大于16当时不是16的倍数,那就补足为16的倍数。

    def encrypt(self, text):

        text = text.encode('utf-8')

        cryptor = AES.new(self.key, self.mode, b'0000000000000000')

        # 这里密钥key 长度必须为16(AES-128),

        # 24(AES-192),或者32 (AES-256)Bytes 长度

        # 目前AES-128 足够目前使用

        length = 16

        count = len(text)

        if count < length:

            add = (length - count)

            # \0 backspace

            # text = text + ('\0' * add)

            text = text + ('\0' * add).encode('utf-8')

        elif count > length:

            add = (length - (count % length))

            # text = text + ('\0' * add)

            text = text + ('\0' * add).encode('utf-8')

        self.ciphertext = cryptor.encrypt(text)

        # 因为AES加密时候得到的字符串不一定是ascii字符集的,输出到终端或者保存时候可能存在问题

        # 所以这里统一把加密后的字符串转化为16进制字符串

        return b2a_hex(self.ciphertext)

 

    # 解密后,去掉补足的空格用strip() 去掉

    def decrypt(self, text):

        cryptor = AES.new(self.key, self.mode, b'0000000000000000')

        plain_text = cryptor.decrypt(a2b_hex(text))

        # return plain_text.rstrip('\0')

        return bytes.decode(plain_text).rstrip('\0')

 

 

if __name__ == '__main__':

    pc = PrpCrypt('keyskeyskeyskeys')  # 初始化密钥

    = pc.encrypt("testtesttest")  # 加密

    = pc.decrypt(e)  # 解密

    print("加密:", e)

    print("解密:", d)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值