mii/FINAL
1
0
Fork 0
mirror of https://github.com/mii443/FINAL.git synced 2025-08-22 23:15:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Tests for LWEScheme::xor_gate

Browse Source
This commit is contained in:
hilder.vitor
2022-03-11 10:22:28 +01:00
parent b2aa2d2cda
commit f943091bf0
1 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
test.cpp
View File

@ -265,7 +265,7 @@ void test_sampler()
cout << "SAMPLER IS OK" << endl; cout << "SAMPLER IS OK" << endl;
} }
enum GateType {NAND, AND, OR}; enum GateType {NAND, AND, OR, XOR};
void test_ntruhe_gate_helper(int in1, int in2, const SchemeNTRU& s, GateType g) void test_ntruhe_gate_helper(int in1, int in2, const SchemeNTRU& s, GateType g)
{ {
@ -452,6 +452,18 @@ void test_lwehe_gate_helper(int in1, int in2, SchemeLWE& s, GateType g)
//cout << "OR output: " << output << endl; //cout << "OR output: " << output << endl;
assert(output == (in1 | in2)); assert(output == (in1 | in2));
} }
else if (g == XOR) {
auto start = clock();
s.xor_gate(ct_res, ct1, ct2);
// s.or_gate(ct_res, ct1, ct2);
avg_time += float(clock()-start)/CLOCKS_PER_SEC;
int output = s.decrypt(ct_res);
//cout << "OR output: " << output << endl;
assert(output == (in1 ^ in2));
}
} }
cout << "Avg. time: " << avg_time/100.0 << endl; cout << "Avg. time: " << avg_time/100.0 << endl;
} }
@ -493,10 +505,17 @@ void test_lwehe_or()
cout << "OR IS OK" << endl; cout << "OR IS OK" << endl;
} }
void test_lwehe_xor()
{
GateType g = XOR;
test_lwe_gate(g);
cout << "XOR IS OK" << endl;
}
void test_lwehe_gate_composition_helper(SchemeLWE& s, GateType g) void test_lwehe_gate_composition_helper(SchemeLWE& s, GateType g)
{ {
float avg_time = 0.0; float avg_time = 0.0;
int N_TESTS = 100; int N_TESTS = 120;
int in1, in2, exp_out; int in1, in2, exp_out;
in1 = binary_sampler(rand_engine); in1 = binary_sampler(rand_engine);
@ -514,7 +533,6 @@ void test_lwehe_gate_composition_helper(SchemeLWE& s, GateType g)
s.nand_gate(ct_res, ct_res, ct);// ct_res should encrypt NAND(exp_out, in2) s.nand_gate(ct_res, ct_res, ct);// ct_res should encrypt NAND(exp_out, in2)
avg_time += float(clock()-start)/CLOCKS_PER_SEC; avg_time += float(clock()-start)/CLOCKS_PER_SEC;
exp_out = !(exp_out & in2); // exp_out = NAND(exp_out, in2) exp_out = !(exp_out & in2); // exp_out = NAND(exp_out, in2)
//cout << "NAND output: " << output << endl; //cout << "NAND output: " << output << endl;
} }
else if (g == AND) { else if (g == AND) {
@ -531,6 +549,14 @@ void test_lwehe_gate_composition_helper(SchemeLWE& s, GateType g)
exp_out = (exp_out | in2); // exp_out = OR(exp_out, in2) exp_out = (exp_out | in2); // exp_out = OR(exp_out, in2)
//cout << "OR output: " << output << endl; //cout << "OR output: " << output << endl;
} }
else if (g == XOR) {
auto start = clock();
s.xor_gate(ct_res, ct_res, ct);
avg_time += float(clock()-start)/CLOCKS_PER_SEC;
exp_out = (exp_out ^ in2); // exp_out = XOR(exp_out, in2)
//cout << "XOR output: " << output << endl;
}
int output = s.decrypt(ct_res); int output = s.decrypt(ct_res);
assert(output == exp_out); assert(output == exp_out);
} }
@ -549,6 +575,9 @@ void test_lwehe_composition_of_gates()
test_lwehe_gate_composition_helper(s, OR); test_lwehe_gate_composition_helper(s, OR);
cout << "COMPOSING OR IS OK" << endl; cout << "COMPOSING OR IS OK" << endl;
test_lwehe_gate_composition_helper(s, XOR);
cout << "COMPOSING XOR IS OK" << endl;
} }
@ -573,6 +602,7 @@ int main()
test_lwehe_nand(); test_lwehe_nand();
test_lwehe_and(); test_lwehe_and();
test_lwehe_or(); test_lwehe_or();
test_lwehe_xor();
test_lwehe_composition_of_gates(); test_lwehe_composition_of_gates();
cout << "LWE tests PASSED" << endl; cout << "LWE tests PASSED" << endl;