x_coder.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from Crypto.Cipher import AES
from binascii import b2a_hex,a2b_hex
import base64

class X_Coder():
    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)
        print("jsonDumpLen:",count)
        if count < length :
            add = length - count
            #\0 backspace
            text = text + (' ' * add)
        elif count > length :
            add = (length - (count % length))
            print("add:",add)
            text = text + (' ' * add)
        self.ciphertext = cryptor.encrypt(text)
        print('sec:',self.ciphertext)
        #因为AES加密时候得到的字符串不一定是ascii字符集的，输出到终端或者保存时候可能存在问题
        #所以这里统一把加密后的字符串转化为16进制字符串
        #return b2a_hex(self.ciphertext)
        return base64.b64encode(self.ciphertext)


    #解密后，去掉补足的空格用strip() 去掉
    def decrypt(self,text):
        cryptor = AES.new(self.key,self.mode,b'0000000000000000')
        #str = a2b_hex(text)
        str = base64.b64decode(text)
        plain_text  = cryptor.decrypt(str)
        return plain_text.rstrip()

if __name__ == '__main__':
    import json
    json_data = {"name":"zk003","count":100,"data":{"a":1,"b":2}}
    codeStr = json.dumps(json_data).replace(' ','');
    print("Josn:",codeStr,' len',len(codeStr))
    pc = X_Coder('MerreyChrismasAn') #初始化密钥
    e = pc.encrypt(codeStr) #加密
    print("加密:",e)

    d = pc.decrypt(e) #解密
    print("解密:",d)
    codeValue = json.loads(d)
    print("Json:",codeValue)
    import httplib2
    url = 'http://172.16.119.130/public/index.php?s=/test/index/test'
    post_data = json.dumps({'data': str(e, encoding = "utf-8")  })
    print('sendData:',post_data)
    handle = httplib2.Http()
    resp, content = handle.request(url,
                            'POST',
                            post_data,
                            headers={'Content-Type': 'application/json'})
    #print(resp)
    ret_data = json.loads(content)
    print("resp: ",ret_data)
    print("enstrLen: ", len(ret_data['enstr']))
    t = pc.decrypt(ret_data["test"])
    print(t)