123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324 |
- #ifndef __PRSONA_SERVER_HPP
- #define __PRSONA_SERVER_HPP
- #include <vector>
- #include "BGN.hpp"
- #include "Curvepoint.hpp"
- #include "Bipoint.hpp"
- #include "base.hpp"
- #include "EGCiphertext.hpp"
- #include "proof.hpp"
- class PrsonaServer : public PrsonaBase {
- public:
- // CONSTRUCTORS
- PrsonaServer(size_t numServers);
- PrsonaServer(size_t numServers, const BGN& other_bgn);
- // BASIC PUBLIC SYSTEM INFO GETTERS
- BGNPublicKey get_bgn_public_key() const;
- size_t get_num_clients() const;
- size_t get_num_servers() const;
- Twistpoint get_fresh_generator(std::vector<Proof>& pi) const;
-
- // FRESH GENERATOR CALCULATION
- Twistpoint add_curr_seed_to_generator(
- std::vector<Proof>& pi,
- const Twistpoint& currGenerator) const;
- Twistpoint add_next_seed_to_generator(
- std::vector<Proof>& pi,
- const Twistpoint& currGenerator) const;
- Twistpoint add_rand_seed_to_generator(
- std::vector<Proof>& pi,
- const Twistpoint& currGenerator) const;
- // ENCRYPTED DATA GETTERS
- std::vector<TwistBipoint> get_current_votes_by(
- Proof& pi, const Twistpoint& shortTermPublicKey) const;
- std::vector<std::vector<TwistBipoint>> get_all_current_votes(
- Proof& pi) const;
- EGCiphertext get_current_user_encrypted_tally(
- Proof& pi, const Twistpoint& shortTermPublicKey) const;
- CurveBipoint get_current_server_encrypted_tally(
- Proof& pi, const Twistpoint& shortTermPublicKey) const;
- std::vector<Twistpoint> get_current_pseudonyms(Proof& pi) const;
- std::vector<Twistpoint> get_current_pseudonyms() const;
- // PROOF COMMITMENT GETTERS
- Proof get_vote_row_commitment(const Twistpoint& request) const;
- Proof get_vote_matrix_commitment() const;
- Proof get_user_tally_commitment(const Twistpoint& request) const;
- Proof get_server_tally_commitment(const Twistpoint& request) const;
- Proof get_pseudonyms_commitment() const;
- void print_current_commitments() const;
- // CLIENT INTERACTIONS
- void add_new_client(
- std::vector<Proof>& proofOfValidAddition,
- const Proof& proofOfValidKey,
- const Twistpoint& shortTermPublicKey);
- bool receive_vote(
- const std::vector<Proof>& pi,
- const std::vector<TwistBipoint>& newVotes,
- const Twistpoint& shortTermPublicKey);
- void print_scores(const std::vector<CurveBipoint>& scores);
- // CONSTRUCTOR HELPERS
- const BGN& get_bgn_details() const;
- bool initialize_fresh_generator(
- const std::vector<Proof>& pi,
- const Twistpoint& firstGenerator);
- bool set_EG_blind_generator(
- const std::vector<Proof>& pi,
- const Twistpoint& currGenerator);
- // EPOCH ROUNDS
- void build_up_midway_pseudonyms(
- std::vector<std::vector<std::vector<Proof>>>& pi,
- std::vector<std::vector<std::vector<Twistpoint>>>& permutationCommits,
- std::vector<std::vector<std::vector<Twistpoint>>>& freshPseudonymCommits,
- std::vector<std::vector<std::vector<Twistpoint>>>& freshPseudonymSeedCommits,
- std::vector<std::vector<std::vector<CurveBipoint>>>& serverTallyCommits,
- std::vector<std::vector<std::vector<std::vector<TwistBipoint>>>>& partwayVoteMatrixCommits,
- std::vector<std::vector<std::vector<std::vector<TwistBipoint>>>>& finalVoteMatrixCommits,
- Twistpoint& nextGenerator);
- void break_down_midway_pseudonyms(
- const std::vector<Proof>& generatorProof,
- std::vector<std::vector<std::vector<Proof>>>& pi,
- std::vector<std::vector<std::vector<Twistpoint>>>& permutationCommits,
- std::vector<std::vector<std::vector<Twistpoint>>>& freshPseudonymCommits,
- std::vector<std::vector<std::vector<Twistpoint>>>& freshPseudonymSeedCommits,
- std::vector<std::vector<std::vector<CurveBipoint>>>& serverTallyCommits,
- std::vector<std::vector<std::vector<std::vector<TwistBipoint>>>>& partwayVoteMatrixCommits,
- std::vector<std::vector<std::vector<std::vector<TwistBipoint>>>>& finalVoteMatrixCommits,
- std::vector<std::vector<std::vector<Twistpoint>>>& userTallyMaskCommits,
- std::vector<std::vector<std::vector<Twistpoint>>>& userTallyMessageCommits,
- std::vector<std::vector<std::vector<Twistpoint>>>& userTallySeedCommits,
- const Twistpoint& nextGenerator);
- bool accept_epoch_updates(
- const std::vector<std::vector<Proof>>& pi,
- const std::vector<std::vector<Twistpoint>>& permutationCommits,
- const std::vector<std::vector<Twistpoint>>& freshPseudonymCommits,
- const std::vector<std::vector<Twistpoint>>& freshPseudonymSeedCommits,
- const std::vector<std::vector<CurveBipoint>>& serverTallyCommits,
- const std::vector<std::vector<std::vector<TwistBipoint>>>& partwayVoteMatrixCommits,
- const std::vector<std::vector<std::vector<TwistBipoint>>>& finalVoteMatrixCommits,
- const std::vector<std::vector<Twistpoint>>& userTallyMaskCommits,
- const std::vector<std::vector<Twistpoint>>& userTallyMessageCommits,
- const std::vector<std::vector<Twistpoint>>& userTallySeedCommits,
- const Twistpoint& nextGenerator,
- bool doUserTallies);
- // DATA MAINTENANCE
- void export_new_user_update(
- std::vector<CurveBipoint>& otherPreviousVoteTallies,
- std::vector<Twistpoint>& otherCurrentPseudonyms,
- std::vector<EGCiphertext>& otherCurrentUserEncryptedTallies,
- std::vector<std::vector<TwistBipoint>>& otherVoteMatrix
- ) const;
- bool import_new_user_update(
- const std::vector<Proof>& pi,
- const std::vector<CurveBipoint>& otherPreviousVoteTallies,
- const std::vector<Twistpoint>& otherCurrentPseudonyms,
- const std::vector<EGCiphertext>& otherCurrentUserEncryptedTallies,
- const std::vector<std::vector<TwistBipoint>>& otherVoteMatrix);
- // SCORE TALLYING
- std::vector<Scalar> tally_scores();
- Scalar get_max_possible_score();
- void receive_tallied_scores(
- const std::vector<EGCiphertext>& userTallyScores,
- const std::vector<CurveBipoint>& serverTallyScores);
- void encrypt(
- CurveBipoint& element,
- const Scalar& value);
- private:
- // constants for servers
- const size_t numServers;
- // Identical between all servers (but collaboratively constructed)
- BGN bgnSystem;
- // Private; different for each server
- Scalar currentSeed;
- Scalar nextSeed;
- // The actual data, which is collaboratively updated by all servers
- std::vector<Proof> currentGeneratorProof;
- Twistpoint currentFreshGenerator;
- std::vector<CurveBipoint> previousVoteTallies;
- std::vector<Twistpoint> currentPseudonyms;
- std::vector<EGCiphertext> currentUserEncryptedTallies;
- std::vector<std::vector<TwistBipoint>> voteMatrix;
- /**
- * NOTE: voteMatrix structure:
- * Each element represents a vote by <rowID> applied to <colID>.
- * The outer vector is a vector of rows and the inner vector is
- * a vector of encrypted votes.
- */
- // An imaginary class; it's just used right now to coordinate servers
- // in memory instead of via network action.
- friend class PrsonaServerEntity;
- // EPOCH HELPERS
- std::vector<std::vector<Proof>> epoch_calculations(
- std::vector<std::vector<Twistpoint>>& permutationCommits,
- std::vector<std::vector<Twistpoint>>& freshPseudonymCommits,
- std::vector<std::vector<Twistpoint>>& freshPseudonymSeedCommits,
- std::vector<std::vector<CurveBipoint>>& serverTallyCommits,
- std::vector<std::vector<std::vector<TwistBipoint>>>& partwayVoteMatrixCommits,
- std::vector<std::vector<std::vector<TwistBipoint>>>& finalVoteMatrixCommits,
- std::vector<std::vector<Twistpoint>>& userTallyMaskCommits,
- std::vector<std::vector<Twistpoint>>& userTallyMessageCommits,
- std::vector<std::vector<Twistpoint>>& userTallySeedCommits,
- const Scalar& power,
- const Twistpoint& nextGenerator,
- bool doUserTallies);
- std::vector<std::vector<Scalar>> generate_permutation_matrix(
- const Scalar& reorderSeed
- ) const;
- std::vector<std::vector<Twistpoint>> generate_commitment_matrix(
- const std::vector<std::vector<Scalar>>& permutations,
- std::vector<std::vector<Scalar>>& seeds
- ) const;
- std::vector<std::vector<Twistpoint>> generate_pseudonym_matrix(
- const std::vector<std::vector<Scalar>>& permutations,
- const Scalar& power,
- std::vector<std::vector<Scalar>>& seeds,
- std::vector<std::vector<Twistpoint>>& seedCommits
- ) const;
- std::vector<std::vector<CurveBipoint>> generate_server_tally_matrix(
- const std::vector<std::vector<Scalar>>& permutations,
- std::vector<std::vector<Scalar>>& seeds
- ) const;
- std::vector<std::vector<std::vector<TwistBipoint>>> generate_vote_tensor(
- const std::vector<std::vector<Scalar>>& permutations,
- const std::vector<std::vector<TwistBipoint>>& currVoteMatrix,
- std::vector<std::vector<std::vector<Scalar>>>& seeds,
- bool inverted
- ) const;
- std::vector<std::vector<TwistBipoint>> calculate_next_vote_matrix(
- const std::vector<std::vector<std::vector<TwistBipoint>>>& voteTensor
- ) const;
- void generate_vote_tensor_proofs(
- std::vector<std::vector<Proof>>& pi,
- const std::vector<std::vector<Scalar>>& permutations,
- const std::vector<std::vector<Scalar>>& permutationSeeds,
- const std::vector<std::vector<std::vector<Scalar>>>& matrixSeeds,
- const std::vector<std::vector<TwistBipoint>>& currMatrix,
- const std::vector<std::vector<Twistpoint>>& permutationCommits,
- const std::vector<std::vector<std::vector<TwistBipoint>>>& matrixCommits,
- bool inverted
- ) const;
- bool verify_vote_tensor_proofs(
- const std::vector<std::vector<Proof>>& pi,
- size_t start_offset,
- const std::vector<std::vector<TwistBipoint>>& currMatrix,
- const std::vector<std::vector<Twistpoint>>& permutationCommits,
- const std::vector<std::vector<std::vector<TwistBipoint>>>& matrixCommits,
- bool inverted
- ) const;
- void generate_user_tally_matrix(
- const std::vector<std::vector<Scalar>>& permutations,
- const Scalar& power,
- const Twistpoint& nextGenerator,
- const std::vector<Twistpoint>& currPseudonyms,
- std::vector<Twistpoint>& masks,
- std::vector<std::vector<Twistpoint>>& maskCommits,
- std::vector<Twistpoint>& messages,
- std::vector<std::vector<Twistpoint>>& messageCommits,
- std::vector<std::vector<Scalar>>& userTallySeeds,
- std::vector<std::vector<Twistpoint>>& userTallySeedCommits
- ) const;
- template <typename T>
- std::vector<std::vector<T>> generate_reordered_plus_power_matrix(
- const std::vector<std::vector<Scalar>>& permutations,
- const Scalar& power,
- const std::vector<T>& oldValues,
- std::vector<std::vector<Scalar>>& seeds,
- std::vector<std::vector<Twistpoint>>& seedCommits,
- const T& h
- ) const;
- template <typename T>
- std::vector<std::vector<T>> generate_reordered_matrix(
- const std::vector<std::vector<Scalar>>& permutations,
- const std::vector<T>& oldValues,
- std::vector<std::vector<Scalar>>& seeds,
- const T& h,
- bool cancelOut
- ) const;
- template <typename T>
- std::vector<std::vector<T>> transpose_matrix(
- const std::vector<std::vector<T>>& input
- ) const;
- std::vector<size_t> sort_data(
- const std::vector<Twistpoint>& inputs
- ) const;
- // A helper class for "ordering" data and for binary search
- struct SortingType {
- Twistpoint pseudonym;
- size_t index;
- bool operator<( const SortingType& rhs ) const
- { return pseudonym < rhs.pseudonym; }
- };
- template <typename T>
- T encrypt(
- const T& g,
- const T& h,
- const Scalar& plaintext,
- const Scalar& lambda
- ) const;
- bool update_data(
- const std::vector<std::vector<Twistpoint>>& freshPseudonymCommits,
- const std::vector<std::vector<CurveBipoint>>& serverTallyCommits,
- const std::vector<std::vector<std::vector<TwistBipoint>>>& voteMatrixCommits,
- const std::vector<std::vector<Twistpoint>>& userTallyMaskCommits,
- const std::vector<std::vector<Twistpoint>>& userTallyMessageCommits);
- bool pseudonyms_sorted(
- const std::vector<Twistpoint> newPseudonyms
- ) const;
- // DATA SAFEKEEPING
- std::vector<size_t> order_data();
-
- // BINARY SEARCH
- size_t binary_search(const Twistpoint& index) const;
- // VALID VOTE PROOFS
- bool verify_vote_proof(
- const std::vector<Proof>& pi,
- const std::vector<TwistBipoint>& oldVotes,
- const std::vector<TwistBipoint>& newVotes,
- const Twistpoint& shortTermPublicKey
- ) const;
- };
- #endif
|