mirror of
https://github.com/ComradCollective/Comrad
synced 2024-11-03 23:15:33 +00:00
125 lines
3.3 KiB
Python
125 lines
3.3 KiB
Python
"""
|
|
Storage for both keys and data
|
|
"""
|
|
from simplekv.fs import FilesystemStore
|
|
from simplekv.memory.redisstore import RedisStore
|
|
import redis
|
|
import hashlib,os
|
|
from pythemis.skeygen import KEY_PAIR_TYPE, GenerateKeyPair
|
|
from pythemis.smessage import SMessage, ssign, sverify
|
|
from pythemis.skeygen import GenerateSymmetricKey
|
|
from pythemis.scell import SCellSeal
|
|
from pythemis.exception import ThemisError
|
|
import zlib
|
|
from komrade import KomradeException,Logger
|
|
|
|
|
|
LOG_GET_SET = True
|
|
|
|
|
|
|
|
class Crypt(Logger):
|
|
def __init__(self,name=None,fn=None,cell=None,init_d=None):
|
|
if not name and fn: name=os.path.basename(fn).replace('.','_')
|
|
|
|
self.name,self.fn,self.cell = name,fn,cell
|
|
self.store = FilesystemStore(self.fn)
|
|
if init_d:
|
|
for k,v in init_d.items():
|
|
self.store.put(k,v)
|
|
|
|
def log(self,*x):
|
|
if LOG_GET_SET:
|
|
super().log(*x)
|
|
|
|
def hash(self,binary_data):
|
|
return hashlib.sha256(binary_data).hexdigest()
|
|
# return zlib.adler32(binary_data)
|
|
|
|
def force_binary(self,k_b):
|
|
if k_b is None: return b''
|
|
if type(k_b)==str: k_b=k_b.encode()
|
|
if type(k_b)!=bytes: k_b=k_b.decode()
|
|
return k_b
|
|
|
|
def package_key(self,k,prefix=''):
|
|
k_b = self.force_binary(k)
|
|
# k_b = self.cell.encrypt(k_b)
|
|
prefix_b = self.force_binary(prefix)
|
|
k_b = self.hash(prefix_b + k_b)
|
|
return k_b
|
|
|
|
def package_val(self,k):
|
|
k_b = self.force_binary(k)
|
|
if self.cell is not None: k_b = self.cell.encrypt(k_b)
|
|
return k_b
|
|
|
|
|
|
def unpackage_val(self,k_b):
|
|
try:
|
|
if self.cell is not None: k_b = self.cell.decrypt(k_b)
|
|
except ThemisError:
|
|
pass
|
|
return k_b
|
|
|
|
|
|
def set(self,k,v,prefix=''):
|
|
# self.log('set() k -->',prefix,k)
|
|
k_b=self.package_key(k,prefix=prefix)
|
|
# self.log('set() k_b -->',k_b)
|
|
|
|
# self.log('set() v -->',v)
|
|
v_b=self.package_val(v)
|
|
self.log(f'set(\n\t{prefix}{k},\n\t{k_b}\n\t\n\t{v_b}\n)\n')
|
|
|
|
return self.store.put(k_b,v_b)
|
|
|
|
def exists(self,k,prefix=''):
|
|
return bool(self.get(k,prefix=prefix))
|
|
|
|
def get(self,k,prefix=''):
|
|
# self.log('get() k -->',prefix,k)
|
|
k_b=self.package_key(k,prefix=prefix)
|
|
# self.log('get() k_b -->',k_b)
|
|
|
|
try:
|
|
v=self.store.get(k_b)
|
|
except KeyError:
|
|
return None
|
|
# self.log('get() v -->',v)
|
|
v_b=self.unpackage_val(v)
|
|
self.log('get()',prefix,k,'-->',v_b)
|
|
return v_b
|
|
|
|
|
|
class KeyCrypt(Crypt):
|
|
def __init__(self):
|
|
return super().__init__(name=PATH_CRYPT_CA_KEYS.replace('.','_'))
|
|
|
|
|
|
class DataCrypt(Crypt):
|
|
def __init__(self):
|
|
return super().__init__(name=PATH_CRYPT_CA_DATA.replace('.','_'))
|
|
|
|
|
|
from collections import defaultdict
|
|
class CryptMemory(Crypt):
|
|
def __init__(self):
|
|
self.data = defaultdict(None)
|
|
self.crypt = defaultdict(None)
|
|
self.cell = None
|
|
|
|
def set(self,k,v,prefix=''):
|
|
k_b=self.package_key(k,prefix=prefix)
|
|
v_b=self.package_val(v)
|
|
self.data[k]=v_b
|
|
self.crypt[k_b]=v_b
|
|
|
|
|
|
|
|
if __name__=='__main__':
|
|
crypt = Crypt('testt')
|
|
|
|
print(crypt.set('hellothere',b'ryan'))
|
|
|
|
# print(crypt.get(b'hello there')) |