#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)