coordinator.phase2_contribution_handler.Phase2ContributionHandler Class Reference

Inheritance diagram for coordinator.phase2_contribution_handler.Phase2ContributionHandler:

Collaboration diagram for coordinator.phase2_contribution_handler.Phase2ContributionHandler:

Public Member Functions
def	__init__ (self, Phase2ServerConfig phase2_config)
str	get_current_challenge_file (self, int contributor_idx)
bool	process_contribution (self, int contribution_idx, str file_name)
None	on_completed (self)

Public Attributes
	mpc

Detailed Description

Handler processing phase2 challenges and contributions.

Definition at line 65 of file phase2_contribution_handler.py.

Constructor & Destructor Documentation

__init__()

def coordinator.phase2_contribution_handler.Phase2ContributionHandler.__init__	(		self,
		Phase2ServerConfig	phase2_config
	)

Definition at line 70 of file phase2_contribution_handler.py.

    def __init__(self, phase2_config: Phase2ServerConfig):
        # Sanity check
        if not exists(CHALLENGE_0_FILE):
            raise Exception(f"no {CHALLENGE_0_FILE} found in server dir")
 
        # If there is no NEXT_CHALLENGE, there should also be no TRANSCRIPT
        if not exists(NEXT_CHALLENGE_FILE):
            if exists(TRANSCRIPT_FILE):
                raise Exception(f"unexpected {TRANSCRIPT_FILE} in server dir")
 
        self.mpc = MPCCommand(phase2_config.mpc_tool)
 

Member Function Documentation

get_current_challenge_file()

str coordinator.phase2_contribution_handler.Phase2ContributionHandler.get_current_challenge_file	(		self,
		int	contributor_idx
	)

Return the location of the current challenge to serve.

Reimplemented from coordinator.icontributionhandler.IContributionHandler.

Definition at line 82 of file phase2_contribution_handler.py.

    def get_current_challenge_file(self, contributor_idx: int) -> str:
        # If there is no NEXT_CHALLENGE_FILE, use CHALLENGE_0_FILE. (Note,
        # contributor_idx may be > 0, even if there is no NEXT_CHALLENGE_FILE.
        # The only condition related to contributor_idx is that, if
        # contributor_idx is 0, we MUST ONLY have the initial challenge.)
        have_next_challenge = exists(NEXT_CHALLENGE_FILE)
        if have_next_challenge:
            if contributor_idx == 0:
                raise Exception(
                    f"unexpected {NEXT_CHALLENGE_FILE} for 0-th contributor")
            return NEXT_CHALLENGE_FILE
        return CHALLENGE_0_FILE
 

on_completed()

None coordinator.phase2_contribution_handler.Phase2ContributionHandler.on_completed

(

self

)

All contributions have been received and the MPC is complete.

Reimplemented from coordinator.icontributionhandler.IContributionHandler.

Definition at line 115 of file phase2_contribution_handler.py.

    def on_completed(self) -> None:
        # Confirm that there has been at least one contribution, otherwise the
        # MPC is invalid.
        if not exists(NEXT_CHALLENGE_FILE):
            raise Exception("no contributions made")
 
        # Perform a validation of the full transcript
        mpc_valid = self.mpc.phase2_verify_transcript(
            CHALLENGE_0_FILE,
            NEXT_CHALLENGE_FILE,
            TRANSCRIPT_FILE)
        if not mpc_valid:
            raise Exception("error in MPC transcript")
 
        # If all is well, move the final challenge file
        rename(NEXT_CHALLENGE_FILE, FINAL_OUTPUT)
        rename(TRANSCRIPT_FILE, FINAL_TRANSCRIPT)
 
        # Notify that handler execution completed
        print("Phase 2 coordinator correctly executed.")
        print("(CTRL-C to stop the server)")

process_contribution()

bool coordinator.phase2_contribution_handler.Phase2ContributionHandler.process_contribution	(		self,
		int	contribution_idx,
		str	file_name
	)

Process the given uploaded file as a contribution. If any errors are
found, throw an exception with an appropriate message, or return false.

Reimplemented from coordinator.icontributionhandler.IContributionHandler.

Definition at line 95 of file phase2_contribution_handler.py.

    def process_contribution(
            self, contribution_idx: int, file_name: str) -> bool:
        orig_challenge = self.get_current_challenge_file(contribution_idx)
        contribution_valid = self.mpc.phase2_verify_contribution(
            orig_challenge=orig_challenge,
            response=file_name,
            out_new_challenge=NEW_CHALLENGE_FILE,
            transcript=TRANSCRIPT_FILE)
 
        if contribution_valid:
            if not exists(NEW_CHALLENGE_FILE):
                raise Exception("unknown error creating new challenge")
 
            # Contribution has been recorded in TRANSCRIPT_FILE. Replace
            # NEXT_CHALLENGE_FILE with NEW_CHALLENGE_FILE.
            rename(NEW_CHALLENGE_FILE, NEXT_CHALLENGE_FILE)
            return True
 
        return False
 

Here is the call graph for this function:

Member Data Documentation

mpc

coordinator.phase2_contribution_handler.Phase2ContributionHandler.mpc

Definition at line 80 of file phase2_contribution_handler.py.

---

The documentation for this class was generated from the following file:

• /home/runner/work/zeth/zeth/mpc/coordinator/phase2_contribution_handler.py