darkfi/script/research/dpos/ouroboros/environment.py

import numpy as np
import math
import random
from ouroboros.logger import Logger
from ouroboros.consts import *
import time
'''
\class Z is the environment
'''
class Z(object):
    def __init__(self, stakeholdes, epoch_length, genesis_time=time.time()):
        self.genesis_time=genesis_time
        self.log = Logger(self, genesis_time)
        self.epoch_length=epoch_length
        self.stakeholders = np.array(stakeholdes)
        self.adversary_mask=np.array([True]*len(stakeholdes))
        self.current_epoch_leaders=[-1]*self.epoch_length
        self.current_slot=0
        self.log.info("Z initialized")
    
    def __repr__(self):
        buff= f"envirnment of {self.length} stakholders"
        for sh in self.stakeholders:
            buff+=str(sh)+"\n"
        return buff

    '''
        return genesis data of the current epoch
    '''
    def get_genesis_data(self):
        #TODO implement dynaming staking
        genesis_data = {STAKEHOLDERS: self.stakeholders,
        STAKEHOLDERS_DISTRIBUTIONS:[],
            SEED: ''}
        return genesis_data
    
    @property
    def current_leader_id(self):
        return self.current_slot%self.epoch_length
    @property
    def current_stakeholder(self):
        self.log.info(f"getting leader of id{self.current_leader_id} of size {len(self.stakeholders)}")
        return self.stakeholders[self.current_leader_id]

    @property
    def current_leader_vrf_pk(self):
        return self.stakeholders[self.current_leader_id].vrf_pk
    
    @property
    def current_leader_vrf_g(self):
        return self.stakeholders[self.current_leader_id].vrf_base

    #TODO complete
    def obfuscate_idx(self, i):
        return i

    #TODO complete
    def deobfuscate_idx(self, i):
        return i

    def corrupt(self, i):
        if i<0 or i>len(self.adversary_mask):
            return False
        self.adversary_mask[self.deobfuscate_idx(i)]=False
        return True
    
    '''
    return the length of all parties
    '''
    def __len__(self):
        return len(self.stakeholders)

    @property
    def length(self):
        return len(self.stakeholders)
    @property
    def honest(self):
        return len(self.stakeholders[self.adversary_mask])

    def select_epoch_leaders(self, sigmas, proofs):
        assert len(sigmas)==self.epoch_length and len(proofs)==self.epoch_length, self.log.error(f"size mismatch between sigmas: {len(sigmas)}, proofs: {len(proofs)}, and epoch_length: {self.epoch_length}")
        for i in range(self.epoch_length):
            self.log.info(f"current sigma of index {i} of total {len(sigmas)}, epoch_length: {self.epoch_length}")
            sigma = sigmas[i]
            assert sigma!=None, 'proof cant be None'
            def leader_selection_hash(sigma):
                Y = np.array(sigma)
                y_hypotenuse2 = math.ceil(np.sum(Y[1]**2+Y[2]**2))
                return y_hypotenuse2
            seed = leader_selection_hash(sigma)
            random.seed(seed)
            leader_idx=seed%self.length
            # only select an honest leaders
            while not self.adversary_mask[leader_idx]:
                leader_idx=random.randint(0,self.length)
            #TODO select the following leader for this epoch, note, 
            # under a single condition that no one is able to predict who is next
            self.current_epoch_leaders[i]=leader_idx
        return self.current_epoch_leaders

    def new_slot(self, slot, sigma, proof):
        self.current_slot=slot
        self.log.info(f"stakeholders: {self.stakeholders}")
        current_leader = self.stakeholders[self.current_leader_id]
        assert current_leader!=None, "current leader cant be None"
        if current_leader.is_leader:
            #pass leadership to the current slot leader from the epoch leader
            self.stakeholders[self.current_epoch_leaders[slot%self.epoch_length]].set_leader()
    
    def new_epoch(self, slot, sigmas, proofs):
        self.current_slot=slot
        #self.log.info(f"stakeholders: {self.stakeholders}")
        #current_leader = self.stakeholders[self.current_leader_id]
        #assert current_leader!=None, 'current leader cant be none'
        #assert(current_leader.is_leader)
        self.select_epoch_leaders(sigmas, proofs)

    def broadcast_block(self, signed_block):
        for stakeholder in self.stakeholders:
            if not stakeholder.is_leader:
                self.stakeholders.receive_block(signed_block)

    def start(self):
        for sh in self.stakeholders:
            sh(self)
        for sh in self.stakeholders:
            sh.start()

    def print_blockchain(self):
        bc = self.stakeholders[0].blockchain
        self.log.info(f"<blockchain>  {len(bc)} blocks: "+str(bc))