def test_dec(self, key, keysize):
fname = 'binary'
if os.path.isfile(fname):
fp = open('binary', 'rb')
msg = fp.read()
fp.close()
obj1 = tcrypt.AESCrypt(key, keysize)
dec = obj1.decrypt(msg)
print 'dec:', dec
print 'decsize:', len(dec)
return
def open(self, ifname, mode='r'):
#print ifname
ifname = self.__get_ifname(ifname)
# print "ifname", ifname
if not os.path.isfile(ifname):
raise ValueError, '%s, file not found...' % (ifname)
fp = open(ifname, 'r')
msg = fp.read()
fp.close()
if self.__idatamap:
obj = tcrypt.AESCrypt(self.__dbkey, self.__dbkeysize)
#self.__buf = obj.decrypt(msg)
self.__fp = cStringIO.StringIO(obj.decrypt(msg))
else:
#self.__buf = msg
self.__fp = cStringIO.StringIO(msg)
return
def decrypt_col(encmsg):
dbkey, dbkeysize = getDBKey()
obj = tcrypt.AESCrypt(dbkey, dbkeysize)
msg = obj.decrypt(encmsg)
data = cPickle.loads(str(msg))
return data
def encrypt_col(msg):
dbkey, dbkeysize = getDBKey()
pdata = cPickle.dumps(msg)
obj = tcrypt.AESCrypt(dbkey, dbkeysize)
return obj.encrypt(pdata)
def decrypt(msg):
dbkey, dbkeysize = getDBKey()
obj = tcrypt.AESCrypt(dbkey, dbkeysize)
return obj.decrypt(msg)
def encryptIVfix(msg):
dbkey, dbkeysize = getDBKey()
obj = tcrypt.AESCrypt(dbkey, dbkeysize)
return obj.encryptIV(msg)
print 'decsize:', len(dec)
return
def debug(self):
data = {}
data['key0'] = 'val0'
data['key1'] = 12345
data['key2'] = 1234.56789
print 'idata:', data
# write to db
dbname = 'testdb.db'
self.write_to_dbcrypt(dbname, data, 'mydata')
# read back from db
data_read = self.read_from_dbcrypt(dbname, 'mydata')
print 'rdata:', data_read
return
if __name__ == '__main__':
obj = tcrypt.AESCrypt()
obj.debug()
obj = DBCrypt()
obj.debug()
obj.test_dec('QngxaW1pcGZGeWwxNUFBNmshqVLICs12', 32)
