diff options
Diffstat (limited to 'fcp/secagg/client/secagg_client_r1_share_keys_input_set_state.h')
-rw-r--r-- | fcp/secagg/client/secagg_client_r1_share_keys_input_set_state.h | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/fcp/secagg/client/secagg_client_r1_share_keys_input_set_state.h b/fcp/secagg/client/secagg_client_r1_share_keys_input_set_state.h new file mode 100644 index 0000000..ebaf6c4 --- /dev/null +++ b/fcp/secagg/client/secagg_client_r1_share_keys_input_set_state.h @@ -0,0 +1,87 @@ +/* + * Copyright 2018 Google LLC + * + * 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. + */ + +#ifndef FCP_SECAGG_CLIENT_SECAGG_CLIENT_R1_SHARE_KEYS_INPUT_SET_STATE_H_ +#define FCP_SECAGG_CLIENT_SECAGG_CLIENT_R1_SHARE_KEYS_INPUT_SET_STATE_H_ + +#include <cstdint> +#include <memory> +#include <string> +#include <vector> + +#include "fcp/base/monitoring.h" +#include "fcp/secagg/client/secagg_client_r1_share_keys_base_state.h" +#include "fcp/secagg/client/secagg_client_state.h" +#include "fcp/secagg/client/send_to_server_interface.h" +#include "fcp/secagg/client/state_transition_listener_interface.h" +#include "fcp/secagg/shared/aes_prng_factory.h" +#include "fcp/secagg/shared/ecdh_key_agreement.h" +#include "fcp/secagg/shared/input_vector_specification.h" +#include "fcp/secagg/shared/prng.h" +#include "fcp/secagg/shared/secagg_messages.pb.h" + +namespace fcp { +namespace secagg { + +// This class represents the client's Round 1: Share Keys state with the input +// already set. This state should transition to the Round 2: Masked Input +// Collection (Input Set) state, but can also transition directly to the +// Completed or Aborted states. + +class SecAggClientR1ShareKeysInputSetState + : public SecAggClientR1ShareKeysBaseState { + public: + SecAggClientR1ShareKeysInputSetState( + uint32_t max_neighbors_expected, + uint32_t minimum_surviving_neighbors_for_reconstruction, + std::unique_ptr<EcdhKeyAgreement> enc_key_agreement, + std::unique_ptr<SecAggVectorMap> input_map, + std::unique_ptr<std::vector<InputVectorSpecification> > + input_vector_specs, + std::unique_ptr<SecurePrng> prng, + std::unique_ptr<EcdhKeyAgreement> prng_key_agreement, + std::unique_ptr<SendToServerInterface> sender, + std::unique_ptr<StateTransitionListenerInterface> transition_listener, + std::unique_ptr<AesPrngFactory> prng_factory, + AsyncAbort* async_abort = nullptr); + + ~SecAggClientR1ShareKeysInputSetState() override = default; + + StatusOr<std::unique_ptr<SecAggClientState> > HandleMessage( + const ServerToClientWrapperMessage& message) override; + + // Returns the name of this state, "R1_SHARE_KEYS_INPUT_SET". + std::string StateName() const override; + + private: + friend class SecAggClientR1ShareKeysInputSetStateTest_ShareKeysRequestIsHandledCorrectlyWithDeadClient_Test; // NOLINT + + const uint32_t max_neighbors_expected_; + const uint32_t minimum_surviving_neighbors_for_reconstruction_; + std::unique_ptr<EcdhKeyAgreement> enc_key_agreement_; + std::unique_ptr<SecAggVectorMap> input_map_; + std::unique_ptr<std::vector<InputVectorSpecification> > input_vector_specs_; + std::unique_ptr<SecurePrng> prng_; + std::unique_ptr<EcdhKeyAgreement> prng_key_agreement_; + std::unique_ptr<AesPrngFactory> prng_factory_; + std::vector<ShamirShare> self_prng_key_shares_; + std::vector<ShamirShare> pairwise_prng_key_shares_; +}; + +} // namespace secagg +} // namespace fcp + +#endif // FCP_SECAGG_CLIENT_SECAGG_CLIENT_R1_SHARE_KEYS_INPUT_SET_STATE_H_ |