standalone_sha3.mpc

from Compiler.script_utils import output_utils

from Compiler.script_utils.data import data


from Compiler import ml
from Compiler import library

from Compiler.script_utils.audit import rand_smoothing

from Compiler.script_utils import config, timers, input_consistency

from Compiler.script_utils.consistency_cerebro import compute_commitment


class Sha3StandaloneConfig(config.BaseAuditModel):

    compute_input: bool = True # whether to compute input or output

program.options_from_args()
cfg = config.from_program_args(program.args, Sha3StandaloneConfig)


program.use_trunc_pr = cfg.trunc_pr
sfix.round_nearest = cfg.round_nearest
ml.set_n_threads(cfg.n_threads)

train_dataset, _, _ = data._load_dataset_args(cfg.dataset)
n_players = len(train_dataset)

if cfg.compute_input:
    sha_fun = input_consistency.compute_sha3_inner(cfg.sha3_approx_factor)

    library.start_timer(timer_id=timers.TIMER_INPUT_CONSISTENCY_CHECK)

    for player_id in range(0, n_players):
        objects = input_consistency.read_input_format_from_file(player_id)
        print("Player", player_id, "has", len(objects), "objects")
        for object in objects:
            len_items = sum([item["length"] for item in object["items"]])
            print("Computing commitment of length ", len_items)
            print_ln("Computing commitment for player %s with size %s", player_id, len_items)
            input_comm = Array(len_items, sint)
            # Should load input from secret shares for correctness,
            # we dont to make it faster (since we would have already loaded the data).
            library.stop_timer(timer_id=timers.TIMER_INPUT_CONSISTENCY_CHECK)
            input_comm.assign_all(2)
            library.start_timer(timer_id=timers.TIMER_INPUT_CONSISTENCY_CHECK)

            sha_fun(input_comm, None, cfg.n_threads)

    library.stop_timer(timer_id=timers.TIMER_INPUT_CONSISTENCY_CHECK)
else:
    sha_fun = input_consistency.compute_sha3_inner(cfg.sha3_approx_factor,
                                                   timer_bit_decompose=timers.TIMER_OUTPUT_CONSISTENCY_SHA_BIT_DECOMPOSE,
                                                   timer_hash_variable=timers.TIMER_OUTPUT_CONSISTENCY_SHA_HASH_VARIABLE)

    library.start_timer(timer_id=timers.TIMER_OUTPUT_COMMIT)
    objects = input_consistency.read_output_format_from_file()

    for object in objects:
        print_ln("Object type %s of length %s", object["object_type"], object["length"])
        len_items = object["length"]
        input_comm = Array(len_items, sint)
        # Should load input from secret shares for correctness,
        # we dont to make it faster (since we would have already loaded the data).
        library.stop_timer(timer_id=timers.TIMER_OUTPUT_COMMIT)
        input_comm.assign_all(2)
        library.start_timer(timer_id=timers.TIMER_OUTPUT_COMMIT)

        sha_fun(input_comm, None, cfg.n_threads)

    library.stop_timer(timer_id=timers.TIMER_OUTPUT_COMMIT)

print_ln("Done computing commitments!")