github/emp-ag2pc
1
1
Fork 0
mirror of https://github.com/emp-toolkit/emp-ag2pc.git synced 2026-01-09 07:18:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

allow alice getting output

Browse Source
This commit is contained in:
Xiao Wang
2020-09-20 16:22:08 -05:00
parent 45396efe12
commit 65a788d09d
2 changed files with 47 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
emp-ag2pc/2pc.h
View File

@@ -271,7 +271,7 @@ class C2PC { public:
io->flush();
}
void online (bool * input, bool * output) {
void online (bool * input, bool * output, bool alice_output = false) {
uint8_t * mask_input = new uint8_t[cf->num_wire];
memset(mask_input, 0, cf->num_wire);
block tmp;
@@ -357,6 +357,50 @@ class C2PC { public:
output[i] = logic_xor(output[i], getLSB(mac[cf->num_wire - cf->n3 + i]));
}
delete[] o;
if(alice_output) {
send_partial_block<T, SSP>(io, mac+cf->num_wire - cf->n3, cf->n3);
send_partial_block<T, SSP>(io, labels+cf->num_wire - cf->n3, cf->n3);
io->send_data(mask_input + cf->num_wire - cf->n3, cf->n3);
io->flush();
}
} else {//ALICE
if(alice_output) {
block * tmp_mac = new block[cf->n3];
block * tmp_label = new block[cf->n3];
bool * tmp_mask_input = new bool[cf->n3];
recv_partial_block<T, SSP>(io, tmp_mac, cf->n3);
recv_partial_block<T, SSP>(io, tmp_label, cf->n3);
io->recv_data(tmp_mask_input, cf->n3);
io->flush();
for(int i = 0; i < cf->n3; ++i) {
block tmp = tmp_mac[i];
tmp = tmp & MASK;
block ttt = key[cf->num_wire - cf-> n3 + i] ^ fpre->Delta;
ttt = ttt & MASK;
key[cf->num_wire - cf-> n3 + i] = key[cf->num_wire - cf-> n3 + i] & MASK;
if(cmpBlock(&tmp, &key[cf->num_wire - cf-> n3 + i], 1))
output[i] = false;
else if(cmpBlock(&tmp, &ttt, 1))
output[i] = true;
else cout <<"no match output label!"<<endl;
block mask_label = tmp_label[i];
if(tmp_mask_input[i])
mask_label = mask_label ^ fpre->Delta;
mask_label = mask_label & MASK;
block masked_labels = labels[cf->num_wire - cf-> n3 + i] & MASK;
if(!cmpBlock(&mask_label, &masked_labels, 1))
cout <<"no match output label2!"<<endl;
output[i] = logic_xor(output[i], tmp_mask_input[i]);
output[i] = logic_xor(output[i], getLSB(mac[cf->num_wire - cf->n3 + i]));
}
delete[] tmp_mac;
delete[] tmp_label;
delete[] tmp_mask_input;
}
}
delete[] mask_input;
}

4
test/single_execution.h
View File

@@ -59,9 +59,9 @@ void test(int party, T* io, string name, string check_output = "") {
memset(in, false, max(cf.n1, cf.n2));
memset(out, false, cf.n3);
t1 = clock_start();
twopc.online(in, out);
twopc.online(in, out, true);
cout << "online:\t"<<party<<"\t"<<time_from(t1)<<endl;
if(party == BOB and check_output.size() > 0){
if(check_output.size() > 0){
string res = "";
for(int i = 0; i < cf.n3; ++i)
res += (out[i]?"1":"0");