threshold-multiparty-psi/main.cpp at master · jellevos/threshold-multiparty-psi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#include <iostream>
#include <vector>
#include "psi_protocols.h"
#include "benchmarking.h"
// #include "NTL/BasicThreadPool.h"


// TODO: Allow variable set sizes?
int main() {
    // SetNumThreads(8);

    Keys keys;
    key_gen(&keys, 1024, 2, 3);

    std::vector<long> client1_set({1, 3, 5, 7});
    std::vector<long> client2_set({2, 3, 4, 5});
    std::vector<long> server_set({6, 5, 2, 1});

    std::cout << "Computing the set intersection between multiple parties using a (2, 3)-encryption of 1024 bits."
              << std::endl;

    auto start = std::chrono::high_resolution_clock::now();
    std::vector<long> result = multiparty_psi(std::vector({client1_set, client2_set}), server_set,
                                              2,
                                              16, 4,
                                              keys);
    auto stop = std::chrono::high_resolution_clock::now();

    std::cout << "The resulting set intersection was: { ";
    for (long element : result) {
        std::cout << element << " ";
    }
    std::cout << "}." << std::endl;
    std::cout << "Took: " << std::chrono::duration_cast<std::chrono::milliseconds>(stop - start).count() << " ms."
        << std::endl << std::endl;


    std::cout << "Computing the threshold set intersection between multiple parties using a (2, 3)-encryption of 1024 bits."
              << std::endl;

    start = std::chrono::high_resolution_clock::now();
    result = threshold_multiparty_psi(std::vector({client1_set, client2_set}), server_set,
                                                            2,
                                                            16, 4,
                                                            1,
                                                            keys);
    stop = std::chrono::high_resolution_clock::now();


    std::cout << "The resulting threshold set intersection was: { ";
    for (long element : result) {
        std::cout << element << " ";
    }
    std::cout << "}." << std::endl << std::endl;
    std::cout << "Took: " << std::chrono::duration_cast<std::chrono::milliseconds>(stop - start).count() << " ms."
              << std::endl << std::endl;

    std::cout << "Run benchmarks for MPSI (y/n): ";

    if (std::cin.get() == 'y') {
        std::cout << "Running benchmarks (without simulated delays) using a 1024-bit key:" << std::endl;
        benchmark(std::vector<long>({10, 20, 30, 40, 50, 60, 70, 80, 90, 100}), std::vector<long>({4, 6, 7}));
    }

    std::cin.ignore( std::numeric_limits<std::streamsize>::max(), '\n');

    std::cout << "Run benchmarks for T-MPSI (y/n): ";

    if (std::cin.get() == 'y') {
        std::cout << "Running benchmarks (without simulated delays) using a 1024-bit key:" << std::endl;
        threshold_benchmark(std::vector<long>({5, 15, 25, 35}), std::vector<long>({2, 4, 5}));
    }

    return 0;
}