#coding: utf8
import sys
from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex
import base64

def printMemLog2(mem):    
    byteData=bytearray(mem)
    byteLen =len(byteData)
    print "printMemLog begin .byteLen:{0}".format(byteLen)
    i=0
    while i <=byteLen-1:            
        a=byteData[i]
        if i == byteLen-1:        
            print "%02x "%( a )            
            break;
            
        i +=1 
        b=byteData[i]        
        print "%02x%02x "%( a,b),        
        i +=1
       
        if i%64 == 0 :        
            print
    print        
    print "*"*10    
    pass


#将text按照byteAlignLen字节对齐,如果不对齐,按照差数填充
def bytePad(text,byteAlignLen=16):    
    count=len(text)    
    mod_num=count%byteAlignLen
    if  mod_num==0:        
        return text     
    add_num=byteAlignLen-mod_num
    print "bytePad:" ,add_num   
    return text+chr(add_num)*add_num    

def byteUnpad(text,byteAlignLen=16):
    count=len(text)    
    print "byteUnpad:",count
    mod_num=count%byteAlignLen
    assert mod_num==0
    lastChar=text[-1]    
    lastLen=ord(lastChar)
    lastChunk=text[-lastLen:]
    if lastChunk==chr(lastLen)*lastLen:
        #print "byteUnpad"
        return text[:-lastLen]
    return text        



 
class prpcrypt():
    def __init__(self, key):
        self.key = key
        self.mode = AES.MODE_ECB
        self.iv = '\0' * 16
     
    #加密函数,如果text不是16的倍数【加密文本text必须为16的倍数!】,那就补足为16的倍数
    def encrypt(self, text):
        cryptor = AES.new(self.key, self.mode, self.iv )
        #这里密钥key 长度必须为16(AES-128)、24(AES-192)、或32(AES-256)Bytes 长度.目前AES-128足够用
        #length = 16
        text=bytePad(text,16)
        #add = length - (count % length)
        #text = text + ('\0' * add)
        
        self.ciphertext = cryptor.encrypt(text)
        #因为AES加密时候得到的字符串不一定是ascii字符集的,输出到终端或者保存时候可能存在问题
        #所以这里统一把加密后的字符串转化为16进制字符串
        return self.ciphertext
        #return b2a_hex(self.ciphertext)
     
    #解密后,去掉补足的空格用strip() 去掉
    def decrypt(self, text):        
        cryptor = AES.new(self.key, self.mode, self.iv )
        #plain_text = cryptor.decrypt(a2b_hex(text))
        plain_text = cryptor.decrypt(text)
        #printMemLog2(plain_text)
        #print plain_text
        return byteUnpad(plain_text)