123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656 |
- #include <iostream>
- #include <sstream>
- #include <cstring>
- #include <cstdio>
- #include "networkServer.hpp"
- void obtain_update_locks(
- std::unique_lock<std::mutex> &lck,
- const std::vector<std::string>& serverIPs,
- const std::string& selfIP,
- struct synchronization_tool *sync)
- {
- size_t i = 0;
- while (i < serverIPs.size())
- {
- if (serverIPs[i] == selfIP)
- {
- lck.lock();
- i++;
- continue;
- }
- struct mg_connection *conn =
- mg_connect_websocket_client(
- serverIPs[i].c_str(),
- PRSONA_PORT,
- USE_SSL,
- NULL,
- 0,
- UPDATE_LOCK_URI,
- "null",
- synchro_websocket_data_handler,
- synchro_websocket_close_handler,
- (void *) sync);
- if (!conn)
- {
- std::cerr << "Couldn't get server " << i << "'s lock" << std::endl;
- continue;
- }
- unique_lock<mutex> lck(sync->mtx);
- sync->val = 0;
- sync->val2 = 0;
- mg_websocket_client_write(
- conn,
- MG_WEBSOCKET_OPCODE_DATACOMPLETE,
- "",
- 0);
- while (!sync->val2)
- sync->cv.wait(lck);
- mg_close_connection(conn);
- if (sync->val)
- i++;
- }
- }
- void release_update_locks(
- std::unique_lock<std::mutex> &lck,
- const std::vector<std::string>& serverIPs,
- const std::string& selfIP,
- struct synchronization_tool *sync)
- {
- ssize_t i = serverIPs.size() - 1;
- while (i >= 0)
- {
- if (serverIPs[i] == selfIP)
- {
- lck.unlock();
- i--;
- continue;
- }
- struct mg_connection *conn =
- mg_connect_websocket_client(
- serverIPs[i].c_str(),
- PRSONA_PORT,
- USE_SSL,
- NULL,
- 0,
- UPDATE_UNLOCK_URI,
- "null",
- synchro_websocket_data_handler,
- synchro_websocket_close_handler,
- (void *) sync);
- if (!conn)
- {
- std::cerr << "Couldn't get server " << i << "'s lock" << std::endl;
- continue;
- }
- unique_lock<mutex> lck(sync->mtx);
- sync->val = 0;
- sync->val2 = 0;
- mg_websocket_client_write(
- conn,
- MG_WEBSOCKET_OPCODE_DATACOMPLETE,
- "",
- 0);
- while (!sync->val2)
- sync->cv.wait(lck);
- mg_close_connection(conn);
- if (sync->val)
- i--;
- }
- }
- std::string make_epoch_initiator_string(
- const std::vector<Proof>& generatorProof,
- const Twistpoint& nextGenerator)
- {
- std::stringstream buffer;
-
- BinarySizeT sizeOfVector(generatorProof.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << generatorProof[i];
-
- buffer << nextGenerator;
- return buffer.str();
- }
- void read_epoch_initiator_string(
- const char *filename,
- std::vector<Proof>& generatorProof,
- Twistpoint& nextGenerator)
- {
- ifstream file(filename);
- BinarySizeT sizeOfVector;
- generatorProof.clear();
- file >> sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- Proof currProof;
- file >> currProof;
- generatorProof.push_back;
- }
-
- file >> nextGenerator;
- }
- std::string make_epoch_update_string(
- 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)
- {
- std::stringstream buffer;
-
- BinarySizeT sizeOfVectorI, sizeOfVectorJ;
- sizeOfVectorI.set(pi.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- sizeOfVectorJ.set(pi[i].size());
- buffer << sizeOfVectorJ;
- for (size_t j = 0; j < sizeOfVectorJ.val(); j++)
- buffer << pi[i][j];
- }
- sizeOfVectorI.set(permutationCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- buffer << permutationCommits[i][j];
- sizeOfVectorI.set(freshPseudonymCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- buffer << freshPseudonymCommits[i][j];
- sizeOfVectorI.set(freshPseudonymSeedCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- buffer << freshPseudonymSeedCommits[i][j];
- sizeOfVectorI.set(serverTallyCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- buffer << serverTallyCommits[i][j];
- sizeOfVectorI.set(partwayVoteMatrixCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- for (size_t k = 0; k < sizeOfVectorI.val(); k++)
- buffer << partwayVoteMatrixCommits[i][j][k];
- sizeOfVectorI.set(finalVoteMatrixCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- for (size_t k = 0; k < sizeOfVectorI.val(); k++)
- buffer << finalVoteMatrixCommits[i][j][k];
- sizeOfVectorI.set(userTallyMaskCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- buffer << userTallyMaskCommits[i][j];
- sizeOfVectorI.set(userTallyMessageCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- buffer << userTallyMessageCommits[i][j];
- sizeOfVectorI.set(userTallySeedCommits.size());
- buffer << sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- buffer << userTallySeedCommits[i][j];
- buffer << nextGenerator;
- BinaryBool flag(doUserTallies);
- buffer << flag;
- return buffer.str();
- }
- bool read_epoch_update_string(
- const char *filename,
- std::vector<std::vector<Proof>>& pi,
- 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,
- Twistpoint& nextGenerator)
- {
- ifstream file(filename);
- BinarySizeT sizeOfVectorI, sizeOfVectorJ;
-
- pi.clear();
- permutationCommits.clear();
- freshPseudonymCommits.clear();
- freshPseudonymSeedCommits.clear();
- serverTallyCommits.clear();
- partwayVoteMatrixCommits.clear();
- finalVoteMatrixCommits.clear();
- userTallyMaskCommits.clear();
- userTallyMessageCommits.clear();
- userTallySeedCommits.clear();
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<Proof> currRow;
- file >> sizeOfVectorJ;
- for (size_t j = 0; j < sizeOfVectorJ.val(); j++)
- {
- Proof currProof;
- file >> currProof;
- currRow.push_back(currProof);
- }
- pi.push_back(currRow);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<Twistpoint> currRow;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- Twistpoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- permutationCommits.push_back(currRow);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<Twistpoint> currRow;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- Twistpoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- freshPseudonymCommits.push_back(currRow);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<Twistpoint> currRow;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- Twistpoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- freshPseudonymSeedCommits.push_back(currRow);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<CurveBipoint> currRow;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- CurveBipoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- serverTallyCommits.push_back(currRow);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<std::vector<TwistBipoint>> currMatrix;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- std::vector<TwistBipoint> currRow;
- for (size_t k = 0; k < sizeOfVectorI.val(); k++)
- {
- TwistBipoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- currMatrix.push_back(currRow);
- }
- partwayVoteMatrixCommits.push_back(currMatrix);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<std::vector<TwistBipoint>> currMatrix;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- std::vector<TwistBipoint> currRow;
- for (size_t k = 0; k < sizeOfVectorI.val(); k++)
- {
- TwistBipoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- currMatrix.push_back(currRow);
- }
- finalVoteMatrixCommits.push_back(currMatrix);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<Twistpoint> currRow;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- Twistpoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- userTallyMaskCommits.push_back(currRow);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<Twistpoint> currRow;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- Twistpoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- userTallyMessageCommits.push_back(currRow);
- }
- file >> sizeOfVectorI;
- for (size_t i = 0; i < sizeOfVectorI.val(); i++)
- {
- std::vector<Twistpoint> currRow;
- for (size_t j = 0; j < sizeOfVectorI.val(); j++)
- {
- Twistpoint currCommit;
- file >> currCommit;
- currRow.push_back(currCommit);
- }
- userTallySeedCommits.push_back(currRow);
- }
- file >> nextGenerator;
-
- BinaryBool doUserTallies;
- file >> doUserTallies;
- return doUserTallies.val();
- }
- void distribute_epoch_updates(
- const std::string& recipient,
- const std::string& data,
- const struct synchronization_tool* sync)
- {
- bool flag = false;
- while (!flag)
- {
- struct mg_connection *conn =
- mg_connect_websocket_client(
- serverIPs[i].c_str(),
- PRSONA_PORT,
- USE_SSL,
- NULL,
- 0,
- ACCEPT_EPOCH_UPDATES_URI,
- "null",
- synchro_websocket_data_handler,
- empty_websocket_close_handler,
- (void *) sync);
- if (!conn)
- {
- std::cerr << "Trouble giving epoch updates to server at " << recipient << std::endl;
- continue;
- }
-
- mg_websocket_client_write(
- conn,
- MG_WEBSOCKET_OPCODE_BINARY,
- data.c_str(),
- data.length());
- mg_websocket_client_write(
- conn,
- MG_WEBSOCKET_OPCODE_DATACOMPLETE,
- "",
- 0);
- mg_close_connection(conn);
- flag = true;
- }
- }
- /*
- * CONSTRUCTORS
- */
- PrsonaServerWebSocketHandler::PrsonaServerWebSocketHandler(
- const PrsonaServer *prsonaServer,
- const std::mutex *updateMtx,
- const std::atomic<size_t> *epochNum,
- const std::vector<std::string> &serverIPs,
- const std::string &selfIP)
- : prsonaServer(prsonaServer), updateMtx(updateMtx), epochNum(epochNum),
- serverIPs(serverIPs), selfIP(selfIP)
- { /* */ }
- virtual bool PrsonaServerWebSocketHandler::handleConnection(
- CivetServer *server,
- const struct mg_connection *conn)
- {
- const struct mg_request_info *info = mg_get_request_info(conn);
-
- bool flag = info->query_string && info->query_string[0] >= PRSONA_ADD_CLIENT && info->query_string[0] <= PRSONA_RECEIVE_PARTIAL_DECRYPTION;
- flag = flag || (info->query_string && info->query_string[0] == PRSONA_GET_FRESH_GENERATOR);
- flag = flag || (info->query_string && info->query_string[0] == PRSONA_GET_EG_BLIND_GENERATOR);
- return flag;
- }
- virtual void PrsonaServerWebSocketHandler::handleReadyState(
- CivetServer *server,
- struct mg_connection *conn)
- {
- const struct mg_request_info *info = mg_get_request_info(conn);
- switch (info->query_string[0])
- {
- case PRSONA_ADD_CLIENT:
- case PRSONA_RECEIVE_VOTE:
- case PRSONA_GET_VOTES_BY:
- case PRSONA_GET_USER_TALLY:
- case PRSONA_GET_SERVER_TALLY:
- case PRSONA_GET_VOTE_ROW_COMMITMENT:
- case PRSONA_GET_USER_TALLY_COMMITMENT:
- case PRSONA_GET_SERVER_TALLY_COMMITMENT:
- case PRSONA_ADD_CURR_SEED_TO_GENERATOR:
- case PRSONA_SET_FRESH_GENERATOR:
- case PRSONA_ADD_RAND_SEED_TO_GENERATOR:
- case PRSONA_SET_EG_BLIND_GENERATOR:
- case PRSONA_EPOCH_UPDATE:
- case PRSONA_NEW_USER_UPDATE:
- case PRSONA_RECEIVE_PARTIAL_DECRYPTION:
- set_temp_filename(conn);
- break;
- default:
- mg_set_user_connection_data(conn, NULL);
- break;
- }
- }
- virtual bool PrsonaServerWebSocketHandler::handleData(
- CivetServer *server,
- struct mg_connection *conn,
- int bits,
- char *data,
- size_t data_len)
- {
- char *filename = (char *) mg_get_user_connection_data(conn);
- if ((bits & 0xf) == MG_WEBSOCKET_OPCODE_DATACOMPLETE)
- {
- generate_response(conn, filename);
- return false;
- }
- if ((bits & 0xf) != MG_WEBSOCKET_OPCODE_BINARY && (bits & 0xf) != MG_WEBSOCKET_OPCODE_CONTINUATION)
- {
- std::cerr << "Unknown opcode: failing." << std::endl;
- return false;
- }
- FILE *currFile = fopen(filename, "ab");
- fwrite(data, sizeof(char), data_len, currFile);
- fclose(currFile);
- return true;
- }
- void PrsonaServerWebSocketHandler::generate_response(
- struct mg_connection *conn,
- char *filename)
- {
- const struct mg_request_info *info = mg_get_request_info(conn);
- switch (info->query_string[0])
- {
- case PRSONA_ADD_CLIENT:
- add_new_client(conn, filename);
- break;
- case PRSONA_RECEIVE_VOTE:
- receive_vote(filename);
- break;
- case PRSONA_GET_BGN_PUBKEY:
- get_bgn_public_key(conn);
- break;
- case PRSONA_GET_NUM_CLIENTS:
- get_num_clients(conn);
- break;
- case PRSONA_GET_NUM_SERVERS:
- get_num_servers(conn);
- break;
- case PRSONA_GET_VOTES_BY:
- get_current_votes_by(conn, filename);
- break;
- case PRSONA_GET_ALL_VOTES:
- get_all_current_votes(conn);
- break;
- case PRSONA_GET_USER_TALLY:
- get_current_user_encrypted_tally(conn, filename);
- break;
- case PRSONA_GET_SERVER_TALLY:
- get_current_server_encrypted_tally(conn, filename);
- break;
- case PRSONA_GET_PSEUDONYMS:
- get_current_pseudonyms(conn);
- break;
- case PRSONA_GET_VOTE_ROW_COMMITMENT:
- get_vote_row_commitment(conn, filename);
- break;
- case PRSONA_GET_VOTE_MATRIX_COMMITMENT:
- get_vote_matrix_commitment(conn);
- break;
- case PRSONA_GET_USER_TALLY_COMMITMENT:
- get_user_tally_commitment(conn, filename);
- break;
- case PRSONA_GET_SERVER_TALLY_COMMITMENT:
- get_server_tally_commitment(conn, filename);
- break;
- case PRSONA_GET_PSEUDONYMS_COMMITMENT:
- get_pseudonyms_commitment(conn);
- break;
- case PRSONA_GET_BGN_DETAILS:
- get_bgn_details(conn);
- break;
- case PRSONA_ADD_CURR_SEED_TO_GENERATOR:
- add_seed_to_generator(conn, filename, true);
- break;
- case PRSONA_SET_FRESH_GENERATOR:
- set_generator(filename, true);
- break;
- case PRSONA_ADD_RAND_SEED_TO_GENERATOR:
- add_seed_to_generator(conn, filename, false);
- break;
- case PRSONA_SET_EG_BLIND_GENERATOR:
- set_generator(filename, false);
- break;
- case PRSONA_EPOCH_BUILD_UP:
- build_up_midway_pseudonyms(conn);
- break;
- case PRSONA_EPOCH_BREAK_DOWN:
- break_down_midway_pseudonyms(conn);
- break;
- case PRSONA_EPOCH_UPDATE:
- accept_epoch_updates(filename);
- break;
- case PRSONA_NEW_USER_UPDATE:
- import_new_user_update(filename);
- break;
- case PRSONA_GET_PARTIAL_DECRYPTION:
- get_partial_decryption(conn, filename);
- break;
- case PRSONA_RECEIVE_PARTIAL_DECRYPTION:
- receive_tallied_scores(conn, filename);
- break;
- case PRSONA_GET_FRESH_GENERATOR:
- get_generator(conn, true);
- break;
- case PRSONA_GET_EG_BLIND_GENERATOR:
- get_generator(conn, false);
- break;
- default:
- break;
- }
- }
- virtual void PrsonaServerWebSocketHandler::handleClose(
- CivetServer *server,
- const struct mg_connection *conn)
- {
- char *filename = (char *) mg_get_user_connection_data(conn);
- if (!filename)
- return;
- remove(filename);
- delete filename;
- }
- /*
- * BASIC PUBLIC SYSTEM INFO GETTERS
- */
- void PrsonaServerWebSocketHandler::get_bgn_public_key(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- BGNPublicKey pubKey = prsonaServer->get_bgn_public_key();
- buffer << pubKey;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_num_clients(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- BinarySizeT numClients(prsonaServer->get_num_clients());
- buffer << numClients;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_num_servers(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- BinarySizeT numServers(prsonaServer->get_num_servers());
- buffer << numServers;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- /*
- * ENCRYPTED DATA GETTERS
- */
- void PrsonaServerWebSocketHandler::get_current_votes_by(
- struct mg_connection *conn, const char *filename) const
- {
- ifstream file(filename);
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- std::stringstream buffer;
- std::string data;
- Proof pi;
- std::vector<TwistBipoint> votes =
- prsonaServer->get_current_votes_by(pi, shortTermPublicKey);
-
- buffer << pi;
-
- BinarySizeT sizeOfVector(votes.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << votes[i];
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_all_current_votes(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- Proof pi;
- std::vector<std::vector<TwistBipoint>> votes =
- prsonaServer->get_all_current_votes(pi);
-
- buffer << pi;
-
- BinarySizeT sizeOfVector(votes.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- for (size_t j = 0; j < sizeOfVector.val(); j++)
- buffer << votes[i][j];
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_current_user_encrypted_tally(
- struct mg_connection *conn, const char *filename) const
- {
- ifstream file(filename);
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- std::stringstream buffer;
- std::string data;
- Proof pi;
- EGCiphertext tally =
- prsonaServer->get_current_user_encrypted_tally(pi, shortTermPublicKey);
-
- buffer << pi;
- buffer << tally;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_current_server_encrypted_tally(
- struct mg_connection *conn, const char *filename) const
- {
- ifstream file(filename);
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- std::stringstream buffer;
- std::string data;
- Proof pi;
- CurveBipoint tally =
- prsonaServer->get_current_server_encrypted_tally(pi, shortTermPublicKey);
-
- buffer << pi;
- buffer << tally;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_current_pseudonyms(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- Proof pi;
- std::vector<Twistpoint> pseudonyms =
- prsonaServer->get_current_pseudonyms(pi);
-
- buffer << pi;
- BinarySizeT sizeOfVector(pseudonyms.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << pseudonyms[i];
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- /*
- * PROOF COMMITMENT GETTERS
- */
- void PrsonaServerWebSocketHandler::get_vote_row_commitment(
- struct mg_connection *conn, const char *filename) const
- {
- ifstream file(filename);
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- std::stringstream buffer;
- std::string data;
- Proof pi =
- prsonaServer->get_vote_row_commitment(shortTermPublicKey);
-
- buffer << pi;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_vote_matrix_commitment(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- Proof pi =
- prsonaServer->get_vote_matrix_commitment();
-
- buffer << pi;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_user_tally_commitment(
- struct mg_connection *conn, const char *filename) const
- {
- ifstream file(filename);
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- std::stringstream buffer;
- std::string data;
- Proof pi =
- prsonaServer->get_user_tally_commitment(shortTermPublicKey);
-
- buffer << pi;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_server_tally_commitment(
- struct mg_connection *conn, const char *filename) const
- {
- ifstream file(filename);
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- std::stringstream buffer;
- std::string data;
- Proof pi =
- prsonaServer->get_server_tally_commitment(shortTermPublicKey);
-
- buffer << pi;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::get_pseudonyms_commitment(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- Proof pi =
- prsonaServer->get_pseudonyms_commitment();
-
- buffer << pi;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::distribute_new_user_updates(
- std::vector<Proof> proofOfValidAddition,
- std::vector<CurveBipoint> previousVoteTallies,
- std::vector<Twistpoint> currentPseudonyms,
- std::vector<EGCiphertext> currentUserEncryptedTallies,
- std::vector<std::vector<TwistBipoint>> voteMatrix) const
- {
- std::stringstream buffer;
- std::string data;
- BinarySizeT sizeOfVector;
- sizeOfVector.set(proofOfValidAddition.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << proofOfValidAddition[i];
- sizeOfVector.set(previousVoteTallies.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << previousVoteTallies[i];
- sizeOfVector.set(currentPseudonyms.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << currentPseudonyms[i];
- sizeOfVector.set(currentUserEncryptedTallies.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << currentUserEncryptedTallies[i];
- sizeOfVector.set(voteMatrix.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- for (size_t j = 0; j < sizeOfVector.val(); j++)
- buffer << voteMatrix[i][j];
- data = buffer.str();
- size_t i = 0;
- while (i < serverIPs.size())
- {
- if (serverIPs[i] == selfIP)
- continue;
- struct mg_connection *conn =
- mg_connect_websocket_client(
- serverIPs[i].c_str(),
- PRSONA_PORT,
- USE_SSL,
- NULL,
- 0,
- GIVE_NEW_USER_URI,
- "null",
- synchro_websocket_data_handler,
- synchro_websocket_close_handler,
- (void *) distributeSync);
- if (!conn)
- {
- std::cerr << "Couldn't give server " << i << " new user" << std::endl;
- continue;
- }
- unique_lock<mutex> syncLock(distributeSync->mtx);
- distributeSync->val = 0;
- distributeSync->val2 = 0;
- mg_websocket_client_write(
- conn,
- MG_WEBSOCKET_OPCODE_BINARY,
- data.c_str(),
- data.length());
- while (!distributeSync->val2)
- distributeSync->cv.wait(syncLock);
- mg_close_connection(conn);
- if (distributeSync->val)
- i++;
- }
- }
- void PrsonaServerWebSocketHandler::distribute_new_vote(
- std::vector<Proof> pi;
- std::vector<TwistBipoint> newVotes,
- Twistpoint shortTermPublicKey) const
- {
- std::stringstream buffer;
- std::string data;
- BinarySizeT sizeOfVector;
- sizeOfVector.set(pi.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << pi[i];
- sizeOfVector.set(newVotes.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << newVotes[i];
- buffer << shortTermPublicKey;
- BinaryBool flag(false);
- buffer << flag;
- data = buffer.str();
- size_t i = 0;
- while (i < serverIPs.size())
- {
- if (serverIPs[i] == selfIP)
- continue;
- struct mg_connection *conn =
- mg_connect_websocket_client(
- serverIPs[i].c_str(),
- PRSONA_PORT,
- USE_SSL,
- NULL,
- 0,
- GIVE_NEW_VOTE_URI,
- "null",
- synchro_websocket_data_handler,
- synchro_websocket_close_handler,
- (void *) distributeSync);
- if (!conn)
- {
- std::cerr << "Couldn't give server " << i << " new user" << std::endl;
- continue;
- }
- unique_lock<mutex> syncLock(distributeSync->mtx);
- distributeSync->val = 0;
- distributeSync->val2 = 0;
- mg_websocket_client_write(
- conn,
- MG_WEBSOCKET_OPCODE_BINARY,
- data.c_str(),
- data.length());
- while (!distributeSync->val2)
- distributeSync->cv.wait(syncLock);
- mg_close_connection(conn);
- if (distributeSync->val)
- i++;
- }
- }
- /*
- * CLIENT INTERACTIONS
- */
- void PrsonaServerWebSocketHandler::add_new_client(
- struct mg_connection *conn, const char *filename)
- {
- ifstream file(filename);
- Proof proofOfValidKey;
- file >> proofOfValidKey;
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- std::unique_lock<std::mutex> lck(*updateMtx, std::defer_lock);
- obtain_update_locks(
- lck,
- serverIPs,
- selfIP,
- &updateSync);
- std::vector<Proof> proofOfValidAddition;
- prsonaServer->add_new_client(
- proofOfValidAddition,
- proofOfValidKey,
- shortTermPublicKey);
- std::vector<CurveBipoint> previousVoteTallies;
- std::vector<Twistpoint> currentPseudonyms;
- std::vector<EGCiphertext> currentUserEncryptedTallies;
- std::vector<std::vector<TwistBipoint>> voteMatrix;
- prsonaServer->export_new_user_update(
- previousVoteTallies,
- currentPseudonyms,
- currentUserEncryptedTallies,
- voteMatrix);
- distribute_new_user_updates(
- proofOfValidAddition,
- previousVoteTallies,
- currentPseudonyms,
- currentUserEncryptedTallies,
- voteMatrix);
- release_update_locks(
- lck,
- serverIPs,
- selfIP,
- &updateSync);
- BinarySizeT sizeOfVector(proofOfValidAddition.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << proofOfValidAddition[i];
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::receive_vote(
- const char *filename)
- {
- ifstream file(filename);
- BinarySizeT sizeOfVector;
- file >> sizeOfVector;
- std::vector<Proof> pi;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- Proof currProof;
- file >> currProof;
- pi.push_back(currProof);
- }
- file >> sizeOfVector;
- std::vector<TwistBipoint> newVotes;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- TwistBipoint currVote;
- file >> currVote;
- newVotes.push_back(currVote);
- }
- Twistpoint shortTermPublicKey;
- file >> shortTermPublicKey;
- BinaryBool shouldDeal;
- file >> shouldDeal;
- std::unique_lock<std::mutex> lck(*updateMtx, std::defer_lock);
- if (shouldDeal.val())
- {
- obtain_update_locks(
- lck,
- serverIPs,
- selfIP,
- &distributeSync);
- }
- prsonaServer->receive_vote(
- pi,
- newVotes,
- shortTermPublicKey);
- if (shouldDeal.val())
- {
- distribute_new_vote(pi, newVotes, shortTermPublicKey);
- release_update_locks(
- lck,
- serverIPs,
- selfIP,
- &distributeSync);
- }
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_DATACOMPLETE, "", 0);
- }
- /*
- * CONSTRUCTOR HELPERS
- */
- void PrsonaServerWebSocketHandler::get_bgn_details(
- struct mg_connection *conn) const
- {
- std::stringstream buffer;
- std::string data;
- const BGN& sharedBGN = prsonaServer->get_bgn_details();
- buffer << sharedBGN;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::add_seed_to_generator(
- struct mg_connection *conn, const char *filename, bool fresh) const
- {
- ifstream file(filename);
- Twistpoint currGenerator;
- file >> currGenerator;
- std::stringstream buffer;
- std::string data;
- std::vector<Proof> pi;
- if (fresh)
- {
- currGenerator =
- prsonaServer->add_curr_seed_to_generator(pi, currGenerator);
- }
- else
- {
- currGenerator =
- prsonaServer->add_rand_seed_to_generator(pi, currGenerator);
- }
-
- buffer << pi[0];
- buffer << currGenerator;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- }
- void PrsonaServerWebSocketHandler::set_generator(
- const char *filename, bool fresh)
- {
- ifstream file(filename);
- BinarySizeT sizeOfVector;
- file >> sizeOfVector;
- std::vector<Proof> pi;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- Proof currProof;
- file >> currProof;
- pi.push_back(currProof);
- }
- Twistpoint newGenerator;
- file >> newGenerator;
- if (fresh)
- prsonaServer->initialize_fresh_generator(pi, newGenerator);
- else
- prsonaServer->set_EG_blind_generator(pi, newGenerator);
- }
- /*
- * EPOCH ROUNDS
- */
- void PrsonaServerWebSocketHandler::build_up_midway_pseudonyms(
- struct mg_connection *conn, const char *filename)
- {
- std::vector<std::vector<Proof>> generatorProofHolder;
- std::vector<Proof> generatorProof;
- Twistpoint nextGenerator;
- read_epoch_initiator_string(
- filename,
- generatorProof,
- nextGenerator);
- generatorProofHolder.push_back(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;
- pi.push_back(generatorProofHolder);
- prsonaServer->build_up_midway_pseudonyms(
- pi,
- permutationCommits,
- freshPseudonymCommits,
- freshPseudonymSeedCommits,
- serverTallyCommits,
- partwayVoteMatrixCommits,
- finalVoteMatrixCommits,
- nextGenerator);
- std::vector<std::vector<Twistpoint>> userTallyMaskCommits, userTallyMessageCommits, userTallySeedCommits;
- string data =
- make_epoch_update_string(
- pi[1],
- permutationCommits[0],
- freshPseudonymCommits[0],
- freshPseudonymSeedCommits[0],
- serverTallyCommits[0],
- partwayVoteMatrixCommits[0],
- finalVoteMatrixCommits[0],
- userTallyMaskCommits,
- userTallyMessageCommits,
- userTallySeedCommits,
- nextGenerator,
- false);
- struct synchronization_tool epochSync;
- epochSync->val = 1;
- for (size_t i = 0; i < serverIPs.size(); i++)
- {
- if (serverIPs[i] == selfIP)
- continue;
- distribute_epoch_updates(
- serverIPs[i],
- data,
- &epochSync);
- }
- unique_lock<mutex> lck(epochSync);
- while (epochSync.val < serverIPs.size())
- epochSync.cv.wait(lck);
-
- data = make_epoch_initiator_string(
- pi[0][0],
- nextGenerator);
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_DATACOMPLETE, "", 0);
- }
- void PrsonaServerWebSocketHandler::break_down_midway_pseudonyms(
- struct mg_connection *conn, const char *filename)
- {
- std::vector<Proof> generatorProof;
- Twistpoint nextGenerator;
- read_epoch_initiator_string(
- filename,
- generatorProof,
- nextGenerator);
- 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;
- prsonaServer->break_down_midway_pseudonyms(
- generatorProof,
- pi,
- permutationCommits,
- freshPseudonymCommits,
- freshPseudonymSeedCommits,
- serverTallyCommits,
- partwayVoteMatrixCommits,
- finalVoteMatrixCommits,
- userTallyMaskCommits,
- userTallyMessageCommits,
- userTallySeedCommits,
- nextGenerator);
- string data =
- make_epoch_update_string(
- pi[0],
- permutationCommits[0],
- freshPseudonymCommits[0],
- freshPseudonymSeedCommits[0],
- serverTallyCommits[0],
- partwayVoteMatrixCommits[0],
- finalVoteMatrixCommits[0],
- userTallyMaskCommits[0],
- userTallyMessageCommits[0],
- userTallySeedCommits[0],
- nextGenerator,
- true);
- struct synchronization_tool epochSync;
- epochSync.val = 1;
- for (size_t i = 0; i < serverIPs.size(); i++)
- {
- if (serverIPs[i] == selfIP)
- continue;
- distribute_epoch_updates(
- serverIPs[i],
- data,
- &epochSync);
- }
- unique_lock<mutex> lck(epochSync.mtx);
- while (epochSync.val < serverIPs.size())
- epochSync.cv.wait(lck);
- epochNum->fetch_add(1);
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_DATACOMPLETE, "", 0);
- }
- void PrsonaServerWebSocketHandler::accept_epoch_updates(
- struct mg_connection *conn, const char *filename)
- {
- std::vector<std::vector<Proof>> pi;
- 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;
- Twistpoint nextGenerator;
- bool doUserTallies =
- read_epoch_update_string(
- filename,
- pi,
- permutationCommits,
- freshPseudonymCommits,
- freshPseudonymSeedCommits,
- serverTallyCommits,
- partwayVoteMatrixCommits,
- finalVoteMatrixCommits,
- userTallyMaskCommits,
- userTallyMessageCommits,
- userTallySeedCommits,
- nextGenerator);
- prsonaServer->accept_epoch_updates(
- pi,
- permutationCommits,
- freshPseudonymCommits,
- freshPseudonymSeedCommits,
- serverTallyCommits,
- partwayVoteMatrixCommits,
- finalVoteMatrixCommits,
- userTallyMaskCommits,
- userTallyMessageCommits,
- userTallySeedCommits,
- nextGenerator,
- doUserTallies);
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_DATACOMPLETE, "", 0);
- }
- /*
- * DATA MAINTENANCE
- */
- void PrsonaServerWebSocketHandler::import_new_user_update(
- const char *filename)
- {
- std::vector<Proof> proofOfValidAddition;
- std::vector<CurveBipoint> previousVoteTallies;
- std::vector<Twistpoint> currentPseudonyms;
- std::vector<EGCiphertext> currentUserEncryptedTallies;
- std::vector<std::vector<TwistBipoint>> voteMatrix;
- ifstream file(filename);
- BinarySizeT sizeOfVector;
-
- file >> sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- Proof currProof;
- file >> currProof;
- proofOfValidAddition.push_back(currProof);
- }
- file >> sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- CurveBipoint currTally;
- file >> currTally;
- previousVoteTallies.push_back(currTally);
- }
- file >> sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- Twistpoint currNym;
- file >> currNym;
- currentPseudonyms.push_back(currNym);
- }
- file >> sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- EGCiphertext currTally;
- file >> currTally;
- currentUserEncryptedTallies.push_back(currTally);
- }
- file >> sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- std::vector<TwistBipoint> currRow;
- for (size_t j = 0; j < sizeOfVector.val(); j++)
- {
- TwistBipoint currVote;
- file >> currVote;
- currRow.push_back(currVote);
- }
- voteMatrix.push_back(currRow);
- }
- prsonaServer->import_new_user_update(
- proofOfValidAddition,
- previousVoteTallies,
- currentPseudonyms,
- currentUserEncryptedTallies,
- voteMatrix);
- }
- void PrsonaServerWebSocketHandler::get_partial_decryption(
- struct mg_connection *conn)
- {
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_DATACOMPLETE, "", 0);
- }
- void PrsonaServerWebSocketHandler::receive_tallied_scores(
- struct mg_connection *conn, const char *filename)
- {
- ifstream file(filename);
- std::vector<EGCiphertext> userScores;
- std::vector<CurveBipoint> serverScores;
- BinarySizeT sizeOfVector;
- file >> sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- EGCiphertext currScore;
- file >> currScore;
- userScores.push_back(currScore);
- }
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- {
- CurveBipoint currScore;
- file >> currScore;
- serverScores.push_back(currScore);
- }
- prsonaServer->receive_tallied_scores(userScores, serverScores);
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_DATACOMPLETE, "", 0);
- }
- void PrsonaServerWebSocketHandler::get_generator(
- struct mg_connection *conn, bool fresh)
- {
- Twistpoint generator;
- std::vector<Proof> pi;
- if (fresh)
- generator = prsonaServer->get_fresh_generator(pi);
- else
- generator = prsonaServer->get_blinding_generator(pi);
- std::stringstream buffer;
- std::string data;
- BinarySizeT sizeOfVector(pi.size());
- buffer << sizeOfVector;
- for (size_t i = 0; i < sizeOfVector.val(); i++)
- buffer << pi[i];
- buffer << generator;
- data = buffer.str();
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_BINARY, data.c_str(), data.length());
- mg_websocket_write(conn, MG_WEBSOCKET_OPCODE_DATACOMPLETE, "", 0);
- }
|