Source code for bbclib

# -*- coding: utf-8 -*-
"""
Copyright (c) 2017 beyond-blockchain.org.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
import sys
import binascii
import hashlib
import random
import socket
import time
import traceback

import ecdsa
import ecdsa.ellipticcurve
from ecdsa import SigningKey, VerifyingKey

sys.path.append("../../")
from bbc1.common.bbc_error import *

ECDSA_CURVE = ecdsa.SECP256k1

domain_global_0 = binascii.a2b_hex("0000000000000000000000000000000000000000000000000000000000000000")

error_code = -1
error_text = ""


[docs]def set_error(code=-1, txt=""): global error_code global error_text error_code = code error_text = txt
[docs]def reset_error(): global error_code global error_text error_code = ESUCCESS error_text = ""
[docs]def get_new_id(seed_str=None, include_timestamp=True): if seed_str is None: return get_random_id() if include_timestamp: seed_str += "%f" % time.time() return hashlib.sha256(bytes(seed_str.encode())).digest()
[docs]def get_random_id(): source_str = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ' output = "".join([random.choice(source_str) for x in range(16)]) return hashlib.sha256(bytes(output.encode())).digest()
[docs]def get_random_value(length=8): val = bytearray() for i in range(length): val.append(random.randint(0,255)) return bytes(val)
[docs]def convert_id_to_string(data, bytelen=32): res = binascii.b2a_hex(data) if len(res) < bytelen*2: res += "0"*(bytelen*2-len(res)) + res return res.decode()
[docs]def convert_idstring_to_bytes(datastr, bytelen=32): res = bytearray(binascii.a2b_hex(datastr)) if len(res) < bytelen: res = bytearray([0]*(bytelen-len(res))).extend(res) return bytes(res)
[docs]def make_transaction_for_base_asset(asset_group_id=None, event_num=1): transaction = BBcTransaction() for i in range(event_num): evt = BBcEvent(asset_group_id=asset_group_id) ast = BBcAsset() evt.add(asset=ast) transaction.add(event=evt) return transaction
[docs]def add_reference_to_transaction(asset_group_id, transaction, ref_transaction_obj, event_index_in_ref): ref = BBcReference(asset_group_id=asset_group_id, transaction=transaction, ref_transaction=ref_transaction_obj, event_index_in_ref=event_index_in_ref) if ref.transaction_id is None: return None transaction.add(reference=ref) return ref
[docs]def recover_transaction_object_from_rawdata(data): transaction = BBcTransaction() transaction.deserialize(data) return transaction
[docs]def recover_signature_object(data): sig = BBcSignature() sig.deserialize(data) return sig
[docs]def to_bigint(val, size=32): dat = bytearray(to_2byte(size)) dat.extend(val) return dat
[docs]def to_8byte(val): return val.to_bytes(8, 'little')
[docs]def to_4byte(val): return val.to_bytes(4, 'little')
[docs]def to_2byte(val): return val.to_bytes(2, 'little')
[docs]def get_n_bytes(ptr, n, dat): return ptr+n, dat[ptr:ptr+n]
[docs]def get_n_byte_int(ptr, n, dat): return ptr+n, int.from_bytes(dat[ptr:ptr+n], 'little')
[docs]def get_bigint(ptr, dat): size = int.from_bytes(dat[ptr:ptr+2], 'little') return ptr+2+size, dat[ptr+2:ptr+2+size]
[docs]class KeyType: ECDSA_SECP256k1_XY = 1 ECDSA_SECP256k1_X = 2
[docs]class KeyPair: def __init__(self, type=KeyType.ECDSA_SECP256k1_XY, privkey=None, pubkey=None): self.type = type self.private_key = privkey self.private_key_object = None self.mk_keyobj_from_private_key() self.public_key = pubkey self.public_key_object = None self.mk_public_keyobj() self.key_len = 0
[docs] def generate(self): if self.type == KeyType.ECDSA_SECP256k1_XY or self.type == KeyType.ECDSA_SECP256k1_X: self.private_key_object = SigningKey.generate(curve=ECDSA_CURVE) self.public_key_object = self.private_key_object.get_verifying_key() self.key_len = len(self.to_binary(self.private_key_object.privkey.order)) self.private_key = bytes(self.to_binary(self.private_key_object.privkey.secret_multiplier)) self.public_key = self.to_binary(self.public_key_object.pubkey.point.x()) if self.type == KeyType.ECDSA_SECP256k1_XY: self.public_key.extend(self.to_binary(self.public_key_object.pubkey.point.y())) self.public_key = bytes(self.public_key)
[docs] def mk_keyobj_from_private_key(self): if self.private_key is None: return if self.type != KeyType.ECDSA_SECP256k1_X and self.type != KeyType.ECDSA_SECP256k1_XY: return self.private_key_object = SigningKey.from_secret_exponent(self.to_bigint(self.private_key), curve=ECDSA_CURVE) self.public_key_object = self.private_key_object.get_verifying_key()
[docs] def mk_keyobj_from_private_key_pem(self, pemdat): self.private_key_object = SigningKey.from_pem(pemdat, curve=ECDSA_CURVE) self.public_key_object = self.private_key.get_verifying_key()
[docs] def mk_public_keyobj(self): if self.public_key is None: return if self.type == KeyType.ECDSA_SECP256k1_XY: self.key_len = int(len(self.public_key)/2) x = self.to_bigint(self.public_key[0:self.key_len]) y = self.to_bigint(self.public_key[self.key_len:self.key_len*2]) point = ecdsa.ellipticcurve.Point(ECDSA_CURVE.curve, x, y) if not ecdsa.ecdsa.point_is_valid(ECDSA_CURVE.generator, x, y): return self.public_key_object = VerifyingKey.from_public_point(point, curve=ECDSA_CURVE) elif self.type == KeyType.ECDSA_SECP256k1_X: self.key_len = len(self.public_key) x = self.to_bigint(self.public_key) # TODO: calc Y and make point object(X,Y) pass
[docs] def to_binary(self, dat): byteval = bytearray() if self.key_len > 0: for i in range(self.key_len): byteval.append(dat % 256) dat = dat // 256 else: while True: byteval.append(dat % 256) dat = dat // 256 if dat == 0: break return byteval
[docs] def to_bigint(self, dat): intval = 0 for i in range(len(dat)): intval += int(dat[i])*(256**i) return intval
[docs] def get_private_key_in_der(self): return self.private_key_object.to_der()
[docs] def get_private_key_in_pem(self): return self.private_key_object.to_pem()
[docs]class BBcSignature: def __init__(self, key_type=KeyType.ECDSA_SECP256k1_XY): self.type = key_type self.signature = None self.pubkey = None self.keypair = None
[docs] def add(self, signature=None, pubkey=None): if signature is not None: self.signature = signature if pubkey is not None: self.pubkey = pubkey self.keypair = KeyPair(type=self.type, pubkey=pubkey) return True
[docs] def serialize(self): dat = bytearray(to_4byte(self.type)) pubkey_len_bit = len(self.pubkey) * 8 dat.extend(to_4byte(pubkey_len_bit)) dat.extend(self.pubkey) sig_len_bit = len(self.signature) * 8 dat.extend(to_4byte(sig_len_bit)) dat.extend(self.signature) return bytes(dat)
[docs] def deserialize(self, data): ptr = 0 try: ptr, self.type = get_n_byte_int(ptr, 4, data) ptr, pubkey_len_bit = get_n_byte_int(ptr, 4, data) pubkey_len = int(pubkey_len_bit/8) ptr, pubkey = get_n_bytes(ptr, pubkey_len, data) ptr, sig_len_bit = get_n_byte_int(ptr, 4, data) sig_len = int(sig_len_bit/8) ptr, signature = get_n_bytes(ptr, sig_len, data) self.add(signature=signature, pubkey=pubkey) except: return False return True
[docs] def verify(self, digest): reset_error() if self.keypair is None: set_error(code=EBADKEYPAIR, txt="Bad private_key/public_key") return False try: flag = self.keypair.public_key_object.verify(self.signature, digest) except: return False return flag
[docs]class BBcTransaction: def __init__(self, version=0): self.version = version self.timestamp = int(time.time()) self.events = [] self.references = [] self.cross_refs = [] self.signatures = [] self.userid_sigidx_mapping = dict() self.transaction_id = None self.transaction_base_digest = None
[docs] def add(self, event=None, reference=None, cross_ref=None): if event is not None: if isinstance(event, list): self.events.extend(event) else: self.events.append(event) if reference is not None: if isinstance(reference, list): self.references.extend(reference) else: self.references.append(reference) if cross_ref is not None: if isinstance(cross_ref, list): self.cross_refs.extend(cross_ref) else: self.cross_refs.append(cross_ref) return True
[docs] def get_sig_index(self, user_id): if user_id not in self.userid_sigidx_mapping: self.userid_sigidx_mapping[user_id] = len(self.userid_sigidx_mapping) self.signatures.append(None) return self.userid_sigidx_mapping[user_id]
[docs] def add_signature(self, user_id=None, signature=None): if user_id not in self.userid_sigidx_mapping: return False idx = self.userid_sigidx_mapping[user_id] self.signatures[idx] = signature return True
[docs] def digest(self): target = self.serialize(for_id=True) d = hashlib.sha256(target).digest() self.transaction_id = d return d
[docs] def serialize(self, for_id=False): dat = bytearray(to_4byte(self.version)) dat.extend(to_8byte(self.timestamp)) dat.extend(to_2byte(len(self.events))) for i in range(len(self.events)): evt = self.events[i].serialize() dat.extend(to_4byte(len(evt))) dat.extend(evt) dat.extend(to_2byte(len(self.references))) for i in range(len(self.references)): refe = self.references[i].serialize() dat.extend(to_4byte(len(refe))) dat.extend(refe) if for_id: self.transaction_base_digest = hashlib.sha256(dat).digest() dat_cross = bytearray(to_2byte(len(self.cross_refs))) for i in range(len(self.cross_refs)): cross = self.cross_refs[i].serialize() dat_cross.extend(to_4byte(len(cross))) dat_cross.extend(cross) if for_id: dat2 = bytearray(self.transaction_base_digest) dat2.extend(dat_cross) return bytes(dat2) dat.extend(dat_cross) real_signum = 0 for sig in self.signatures: if sig is not None: real_signum += 1 dat.extend(to_2byte(real_signum)) for i in range(real_signum): sig = self.signatures[i].serialize() dat.extend(to_4byte(len(sig))) dat.extend(sig) return bytes(dat)
[docs] def deserialize(self, data): ptr = 0 try: ptr, self.version = get_n_byte_int(ptr, 4, data) ptr, self.timestamp = get_n_byte_int(ptr, 8, data) ptr, evt_num = get_n_byte_int(ptr, 2, data) self.events = [] for i in range(evt_num): ptr, size = get_n_byte_int(ptr, 4, data) ptr, evtdata = get_n_bytes(ptr, size, data) evt = BBcEvent() evt.deserialize(evtdata) self.events.append(evt) ptr, ref_num = get_n_byte_int(ptr, 2, data) self.references = [] for i in range(ref_num): ptr, size = get_n_byte_int(ptr, 4, data) ptr, refdata = get_n_bytes(ptr, size, data) refe = BBcReference(None, None) refe.deserialize(refdata) self.references.append(refe) ptr, cross_num = get_n_byte_int(ptr, 2, data) self.cross_refs = [] for i in range(cross_num): ptr, size = get_n_byte_int(ptr, 4, data) ptr, crossdata = get_n_bytes(ptr, size, data) cross = BBcCrossRef() cross.deserialize(crossdata) self.cross_refs.append(cross) ptr, sig_num = get_n_byte_int(ptr, 2, data) self.signatures = [] for i in range(sig_num): ptr, size = get_n_byte_int(ptr, 4, data) ptr, sigdata = get_n_bytes(ptr, size, data) sig = BBcSignature() sig.deserialize(sigdata) self.signatures.append(sig) self.digest() except Exception as e: print("Transaction data deserialize: %s" % e) print(traceback.format_exc()) return False return True
[docs] def sign(self, key_type=KeyType.ECDSA_SECP256k1_XY, private_key=None, public_key=None, keypair=None): """ Sign transaction :param key_type: KeyType.ECDSA_SECP256k1_XY/X :param private_key: bytes format :param public_key: bytes format :return: BBcSignature object """ reset_error() if keypair is None: if not isinstance(private_key, bytes) or not isinstance(public_key, bytes): set_error(code=EBADKEYPAIR, txt="Bad private_key/public_key (must be in bytes format)") return None keypair = KeyPair(type=key_type, privkey=private_key, pubkey=public_key) if keypair is None: set_error(code=EBADKEYPAIR, txt="Bad private_key/public_key") return None sig = BBcSignature(key_type=keypair.type) if keypair.type == KeyType.ECDSA_SECP256k1_XY or keypair.type == KeyType.ECDSA_SECP256k1_X: s = keypair.private_key_object.sign(self.digest()) else: set_error(code=EOTHER, txt="sig_type %d is not supported" % keypair.type) return None sig.add(signature=s, pubkey=keypair.public_key) return sig
[docs] def dump(self): import binascii print("------- Dump of the transaction data ------") if self.transaction_id is not None: print("transaction_id:", binascii.b2a_hex(self.transaction_id)) else: print("transaction_id: None") print("version:", self.version) print("timestamp:", self.timestamp) print("Event[]:") if len(self.events) > 0: for i, evt in enumerate(self.events): print("[%d]" % i) print(" asset_group_id:", binascii.b2a_hex(evt.asset_group_id)) print(" reference_indices:", evt.reference_indices) print(" mandatory_approvers:") if len(evt.mandatory_approvers) > 0: for user in evt.mandatory_approvers: print(" - ", binascii.b2a_hex(user)) else: print(" - NONE") print(" option_approvers:") if len(evt.option_approvers) > 0: for user in evt.option_approvers: print(" - ", binascii.b2a_hex(user)) else: print(" - NONE") print(" option_approver_num_numerator:", evt.option_approver_num_numerator) print(" option_approver_num_denominator:", evt.option_approver_num_denominator) print(" Asset:") print(" asset_id:", binascii.b2a_hex(evt.asset.asset_id)) if evt.asset.user_id is not None: print(" user_id:", binascii.b2a_hex(evt.asset.user_id)) else: print(" user_id: NONE") print(" nonce:", binascii.b2a_hex(evt.asset.nonce)) print(" file_size:", evt.asset.asset_file_size) if evt.asset.asset_file_digest is not None: print(" file_digest:", binascii.b2a_hex(evt.asset.asset_file_digest)) print(" body_size:", evt.asset.asset_body_size) print(" body:", evt.asset.asset_body) else: print(" None") print("Reference[]:",len(self.references)) if len(self.references) > 0: for i, refe in enumerate(self.references): print("[%d]" % i) print(" asset_group_id:", binascii.b2a_hex(refe.asset_group_id)) print(" transaction_id:", binascii.b2a_hex(refe.transaction_id)) print(" event_index_in_ref:", refe.event_index_in_ref) print(" sig_index:", refe.sig_indices) else: print(" None") print("Cross_Ref[]:",len(self.cross_refs)) if len(self.cross_refs) > 0: for i, cross in enumerate(self.cross_refs): print("[%d]" % i) print(" asset_group_id:", binascii.b2a_hex(cross.asset_group_id)) print(" transaction_id:", binascii.b2a_hex(cross.transaction_id)) else: print(" None") print("Signature[]:") if len(self.signatures) > 0: for i, sig in enumerate(self.signatures): print("[%d]" % i) if sig is None: print(" *RESERVED*") continue print(" type:", sig.type) print(" signature:", binascii.b2a_hex(sig.signature)) print(" pubkey:", binascii.b2a_hex(sig.pubkey)) else: print(" None")
[docs]class BBcEvent: def __init__(self, asset_group_id=None): self.asset_group_id = asset_group_id self.reference_indices = [] self.mandatory_approvers = [] self.option_approver_num_numerator = 0 self.option_approver_num_denominator = 0 self.option_approvers = [] self.asset = None
[docs] def add(self, asset_group_id=None, reference_index=None, mandatory_approver=None, option_approver_num_numerator=0, option_approver_num_denominator=0, option_approver=None, asset=None): if asset_group_id is not None: self.asset_group_id = asset_group_id if reference_index is not None: self.reference_indices.append(reference_index) if mandatory_approver is not None: self.mandatory_approvers.append(mandatory_approver) if option_approver_num_numerator > 0: self.option_approver_num_numerator = option_approver_num_numerator if option_approver_num_denominator > 0: self.option_approver_num_denominator = option_approver_num_denominator if option_approver is not None: self.option_approvers.append(option_approver) if asset is not None: self.asset = asset return True
[docs] def serialize(self): dat = bytearray(to_bigint(self.asset_group_id)) dat.extend(to_2byte(len(self.reference_indices))) for i in range(len(self.reference_indices)): dat.extend(to_2byte(self.reference_indices[i])) dat.extend(to_2byte(len(self.mandatory_approvers))) for i in range(len(self.mandatory_approvers)): dat.extend(to_bigint(self.mandatory_approvers[i])) dat.extend(to_2byte(self.option_approver_num_numerator)) dat.extend(to_2byte(self.option_approver_num_denominator)) for i in range(self.option_approver_num_denominator): dat.extend(to_bigint(self.option_approvers[i])) ast = self.asset.serialize() dat.extend(to_4byte(len(ast))) dat.extend(ast) return bytes(dat)
[docs] def deserialize(self, data): ptr = 0 try: ptr, self.asset_group_id = get_bigint(ptr, data) ptr, ref_num = get_n_byte_int(ptr, 2, data) self.reference_indices = [] for i in range(ref_num): ptr, idx = get_n_byte_int(ptr, 2, data) self.reference_indices.append(idx) ptr, appr_num = get_n_byte_int(ptr, 2, data) self.mandatory_approvers = [] for i in range(appr_num): ptr, appr = get_bigint(ptr, data) self.mandatory_approvers.append(appr) ptr, self.option_approver_num_numerator = get_n_byte_int(ptr, 2, data) ptr, self.option_approver_num_denominator = get_n_byte_int(ptr, 2, data) self.option_approvers = [] for i in range(self.option_approver_num_denominator): ptr, appr = get_bigint(ptr, data) self.option_approvers.append(appr) ptr, astsize = get_n_byte_int(ptr, 4, data) ptr, astdata = get_n_bytes(ptr, astsize, data) self.asset = BBcAsset() self.asset.deserialize(astdata) except: return False return True
[docs]class BBcReference: def __init__(self, asset_group_id, transaction, ref_transaction=None, event_index_in_ref=0): self.asset_group_id = asset_group_id self.transaction_id = None self.transaction = transaction self.ref_transaction = ref_transaction self.event_index_in_ref = event_index_in_ref self.sig_indices = [] self.mandatory_approvers = None self.option_approvers = None self.option_sig_ids = [] if ref_transaction is None: return self.prepare_reference(ref_transaction=ref_transaction)
[docs] def prepare_reference(self, ref_transaction): self.ref_transaction = ref_transaction try: evt = ref_transaction.events[self.event_index_in_ref] for user in evt.mandatory_approvers: self.sig_indices.append(self.transaction.get_sig_index(user)) for i in range(evt.option_approver_num_numerator): dummy_id = get_random_value(4) self.option_sig_ids.append(dummy_id) self.sig_indices.append(self.transaction.get_sig_index(dummy_id)) self.mandatory_approvers = evt.mandatory_approvers self.option_approvers = evt.option_approvers self.transaction_id = ref_transaction.digest() except Exception as e: print(traceback.format_exc())
[docs] def add_signature(self, user_id=None, signature=None): if user_id in self.option_approvers: if len(self.option_sig_ids) == 0: return user_id = self.option_sig_ids.pop(0) self.transaction.add_signature(user_id=user_id, signature=signature)
[docs] def get_referred_transaction(self): return {self.ref_transaction.transaction_id: self.ref_transaction.serialize()}
[docs] def get_destinations(self): return self.mandatory_approvers+self.option_approvers
[docs] def serialize(self): dat = bytearray(to_bigint(self.asset_group_id)) dat.extend(to_bigint(self.transaction_id)) dat.extend(to_2byte(self.event_index_in_ref)) dat.extend(to_2byte(len(self.sig_indices))) for i in range(len(self.sig_indices)): dat.extend(to_2byte(self.sig_indices[i])) return bytes(dat)
[docs] def deserialize(self, data): ptr = 0 try: ptr, self.asset_group_id = get_bigint(ptr, data) ptr, self.transaction_id = get_bigint(ptr, data) ptr, self.event_index_in_ref = get_n_byte_int(ptr, 2, data) ptr, signum = get_n_byte_int(ptr, 2, data) self.sig_indices = [] for i in range(signum): ptr, idx = get_n_byte_int(ptr, 2, data) self.sig_indices.append(idx) except: return False return True
[docs]class BBcAsset: def __init__(self): self.asset_id = None self.user_id = None self.nonce = get_random_value() self.asset_file_size = 0 self.asset_file = None self.asset_file_digest = None self.asset_body_size = 0 self.asset_body = [] # up to 256 bytes
[docs] def add(self, user_id=None, asset_file=None, asset_body=None): if user_id is not None: self.user_id = user_id if asset_file is not None: self.asset_file = asset_file self.asset_file_size = len(asset_file) self.asset_file_digest = hashlib.sha256(asset_file).digest() if asset_body is not None: if len(asset_body) > 256: self.asset_file = asset_body self.asset_file_size = len(asset_body) self.asset_file_digest = hashlib.sha256(asset_body).digest() else: self.asset_body = asset_body if isinstance(asset_body, str): self.asset_body = asset_body.encode() self.asset_body_size = len(asset_body) self.digest()
[docs] def digest(self): target = self.serialize(for_digest_calculation=True) self.asset_id = hashlib.sha256(target).digest() return self.asset_id
[docs] def get_asset_file(self): if self.asset_file is None: return None, None return self.asset_file_digest, self.asset_file
[docs] def recover_asset_file(self, asset_file): digest = hashlib.sha256(asset_file).digest() if digest == self.asset_file_digest: self.asset_file = asset_file return True else: return False
[docs] def serialize(self, for_digest_calculation=False): if for_digest_calculation: dat = bytearray(to_bigint(self.user_id)) dat.extend(to_2byte(len(self.nonce))) dat.extend(self.nonce) dat.extend(to_4byte(self.asset_file_size)) if self.asset_file_size > 0: dat.extend(self.asset_file_digest) dat.extend(to_2byte(self.asset_body_size)) if self.asset_body_size > 0: dat.extend(self.asset_body) return bytes(dat) else: dat = bytearray(to_bigint(self.asset_id)) dat.extend(to_bigint(self.user_id)) dat.extend(to_2byte(len(self.nonce))) dat.extend(self.nonce) dat.extend(to_4byte(self.asset_file_size)) if self.asset_file_size > 0: dat.extend(to_bigint(self.asset_file_digest)) dat.extend(to_2byte(self.asset_body_size)) if self.asset_body_size > 0: dat.extend(self.asset_body) return bytes(dat)
[docs] def deserialize(self, data): ptr = 0 try: ptr, self.asset_id = get_bigint(ptr, data) ptr, self.user_id = get_bigint(ptr, data) ptr, noncelen = get_n_byte_int(ptr, 2, data) ptr, self.nonce = get_n_bytes(ptr, noncelen, data) ptr, self.asset_file_size = get_n_byte_int(ptr, 4, data) if self.asset_file_size > 0: ptr, self.asset_file_digest = get_bigint(ptr, data) ptr, self.asset_body_size = get_n_byte_int(ptr, 2, data) if self.asset_body_size > 0: ptr, self.asset_body = get_n_bytes(ptr, self.asset_body_size, data) except: traceback.print_exc() return False return True
[docs]class BBcCrossRef: def __init__(self, asset_group_id=None, transaction_id=None): self.asset_group_id = asset_group_id self.transaction_id = transaction_id
[docs] def serialize(self): dat = bytearray(to_bigint(self.asset_group_id)) dat.extend(to_bigint(self.transaction_id)) return bytes(dat)
[docs] def deserialize(self, data): ptr = 0 try: ptr, self.asset_group_id = get_bigint(ptr, data) ptr, self.transaction_id = get_bigint(ptr, data) except: return False return True
[docs]class ServiceMessageType: REQUEST_SETUP_DOMAIN = 0 RESPONSE_SETUP_DOMAIN = 1 REQUEST_GET_PEERLIST = 2 RESPONSE_GET_PEERLIST = 3 REQUEST_SET_STATIC_NODE = 4 RESPONSE_SET_STATIC_NODE = 5 REQUEST_SETUP_ASSET_GROUP = 6 RESPONSE_SETUP_ASSET_GROUP = 7 REQUEST_GET_CONFIG = 8 RESPONSE_GET_CONFIG = 9 REQUEST_MANIP_LEDGER_SUBSYS = 10 RESPONSE_MANIP_LEDGER_SUBSYS = 11 DOMAIN_PING = 12 REQUEST_GET_DOMAINLIST = 13 RESPONSE_GET_DOMAINLIST = 14 REGISTER = 32 UNREGISTER = 33 MESSAGE = 34 REQUEST_GATHER_SIGNATURE = 35 RESPONSE_GATHER_SIGNATURE = 36 REQUEST_SIGNATURE = 37 RESPONSE_SIGNATURE = 38 REQUEST_INSERT = 39 RESPONSE_INSERT = 40 REQUEST_SEARCH_ASSET = 66 RESPONSE_SEARCH_ASSET = 67 REQUEST_SEARCH_TRANSACTION = 68 RESPONSE_SEARCH_TRANSACTION = 69 REQUEST_CROSS_REF = 70 RESPONSE_CROSS_REF = 71 REQUEST_REGISTER_HASH_IN_SUBSYS = 128 RESPONSE_REGISTER_HASH_IN_SUBSYS = 129 REQUEST_VERIFY_HASH_IN_SUBSYS = 130 RESPONSE_VERIFY_HASH_IN_SUBSYS = 131
[docs]def is_less_than(val_a, val_b): """ return True if val_a is less than val_b (evaluate as integer) :param val_a: :param val_b: :return: """ size = len(val_a) if size != len(val_b): return False for i in reversed(range(size)): if val_a[i] < val_b[i]: return True elif val_a[i] > val_b[i]: return False return False
[docs]class NodeInfo: """ node information entry (socket info) """ def __init__(self, node_id=domain_global_0, ipv4=None, ipv6=None, port=None): self.node_id = node_id if ipv4 is None or len(ipv4) == 0: self.ipv4 = "0.0.0.0" else: if isinstance(ipv4, bytes): self.ipv4 = ipv4.decode() else: self.ipv4 = ipv4 if ipv6 is None or len(ipv6) == 0: self.ipv6 = "::" else: if isinstance(ipv6, bytes): self.ipv6 = ipv6.decode() else: self.ipv6 = ipv6 self.port = port self.created_at = self.updated_at = time.time() self.is_alive = False self.disconnect_at = 0 def __lt__(self, other): if self.is_alive and other.is_alive: return is_less_than(self.node_id, other.node_id) elif self.is_alive and not other.is_alive: return True elif not self.is_alive and other.is_alive: return False else: return is_less_than(self.node_id, other.node_id) def __len__(self): return len(self.node_id)
[docs] def touch(self): self.updated_at = time.time() self.is_alive = True
[docs] def detect_disconnect(self): self.disconnect_at = time.time() self.is_alive = False
[docs] def update(self, ipv4=None, ipv6=None, port=None): if ipv4 is not None: self.ipv4 = ipv4 if ipv6 is not None: self.ipv6 = ipv6 if port is not None: self.port = port self.updated_at = time.time()
[docs] def get_nodeinfo(self): ipv4 = socket.inet_pton(socket.AF_INET, self.ipv4) ipv6 = socket.inet_pton(socket.AF_INET6, self.ipv6) return self.node_id, ipv4, ipv6, socket.htons(self.port).to_bytes(2, 'little')
[docs] def recover_nodeinfo(self, node_id, ipv4, ipv6, port): self.node_id = node_id self.ipv4 = socket.inet_ntop(socket.AF_INET, ipv4) self.ipv6 = socket.inet_ntop(socket.AF_INET6, ipv6) self.port = socket.ntohs(int.from_bytes(port, 'little')) self.touch()
[docs]class StorageType: NONE = 0 FILESYSTEM = 1
#HTTP_PUT = 2 #HTTP_POST = 3