// Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. // // 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. #pragma once #include #include #include #include "core/paddlefl_mpc/mpc_protocol/abstract_context.h" #include "core/paddlefl_mpc/mpc_protocol/abstract_network.h" #include "prng_utils.h" namespace aby3 { using AbstractNetwork = paddle::mpc::AbstractNetwork; using AbstractContext = paddle::mpc::AbstractContext; class PrivCContext : public AbstractContext { public: PrivCContext(size_t party, std::shared_ptr network, const block &seed = g_zero_block) { init(party, network, g_zero_block, seed); } PrivCContext(const PrivCContext &other) = delete; PrivCContext &operator=(const PrivCContext &other) = delete; void init(size_t party, std::shared_ptr network, block seed, block seed2) override { set_party(party); set_network(network); set_num_party(2); if (psi::equals(seed2, psi::g_zero_block)) { seed2 = psi::block_from_dev_urandom(); } // seed2 is private set_random_seed(seed2, 2); } }; } // namespace aby3