def __init__(self, fabnet_hostname, security_manager):
self.fabnet_hostname = fabnet_hostname
self.security_manager = security_manager
cert = self.security_manager.get_client_cert()
ckey = self.security_manager.get_client_cert_key()
self.fri_client = FriClient(bool(ckey), cert, ckey)
class FabnetGateway:
def __init__(self, fabnet_hostname, security_manager):
self.fabnet_hostname = fabnet_hostname
self.security_manager = security_manager
cert = self.security_manager.get_client_cert()
ckey = self.security_manager.get_client_cert_key()
self.fri_client = FriClient(bool(ckey), cert, ckey)
def put(self, data, key=None, replica_count=DEFAULT_REPLICA_COUNT, wait_writes_count=2):
source_checksum = hashlib.sha1(data).hexdigest()
data = self.security_manager.encrypt(data)
checksum = hashlib.sha1(data).hexdigest()
params = {'key':key, 'checksum': checksum, 'wait_writes_count': wait_writes_count}
packet = FabnetPacketRequest(method='ClientPutData', parameters=params, binary_data=data, sync=True)
resp = self.fri_client.call_sync('%s:%s'%(self.fabnet_hostname, FRI_PORT), packet, FRI_CLIENT_TIMEOUT)
if resp.ret_code != 0:
logger.error('ClientPutData error: %s'%resp.ret_message)
raise Exception('ClientPutData error: %s'%resp.ret_message)
primary_key = resp.ret_parameters['key']
return primary_key, source_checksum
def get(self, primary_key, replica_count=DEFAULT_REPLICA_COUNT):
params = {'key': primary_key, 'replica_count': replica_count}
packet = FabnetPacketRequest(method='ClientGetData', parameters=params, sync=True)
resp = self.fri_client.call_sync('%s:%s'%(self.fabnet_hostname, FRI_PORT), packet, FRI_CLIENT_TIMEOUT)
if resp.ret_code == RC_NO_DATA:
logger.error('No data found for key %s'%(primary_key,))
elif resp.ret_code != 0:
logger.error('Get data block error for key %s: %s'%(primary_key, resp.ret_message))
if resp.ret_code == 0:
exp_checksum = resp.ret_parameters['checksum']
data = resp.binary_data
checksum = hashlib.sha1(data).hexdigest()
if exp_checksum != checksum:
logger.error('Currupted data block for key %s from node %s'%(primary_key, node_addr))
else:
data = self.security_manager.decrypt(data)
return data
return None