123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211 |
- package protocols;
- import communication.Communication;
- import crypto.Crypto;
- import exceptions.NoSuchPartyException;
- import oram.Forest;
- import oram.Metadata;
- import struct.OutFF;
- import struct.Party;
- import subprotocols.Shift;
- import util.M;
- import util.P;
- import util.Util;
- public class FlipFlag extends Protocol {
- int pid = P.FF;
- public FlipFlag(Communication con1, Communication con2) {
- super(con1, con2);
- online_band = all.online_band[pid];
- offline_band = all.offline_band[pid];
- timer = all.timer[pid];
- }
- // TODO: remove loop around setXor: use Util.setXor(byte[][], byte[][])
- public OutFF runE(byte[][] fb_DE, byte[][] fb_CE, int i2) {
- timer.start(M.offline_comp);
- int n = fb_DE.length;
- OutFF outff = new OutFF();
- outff.fb_DE = new byte[n][1];
- for (int i = 0; i < n; i++) {
- Crypto.sr_DE.nextBytes(outff.fb_DE[i]);
- }
- timer.stop(M.offline_comp);
- // ----------------------------------------- //
- timer.start(M.online_comp);
- byte[][] a2 = new byte[n][1];
- Shift shift = new Shift(con1, con2);
- byte[][] m2 = shift.runE(a2, n - i2);
- for (int i = 0; i < n; i++)
- Util.setXor(m2[i], outff.fb_DE[i]);
- timer.start(M.online_write);
- con2.write(online_band, m2);
- timer.stop(M.online_write);
- timer.start(M.online_read);
- byte[][] m1 = con2.readDoubleByteArrayAndDec();
- timer.stop(M.online_read);
- outff.fb_CE = Util.xor(m1, m2);
- Util.setXor(outff.fb_CE, fb_CE);
- Util.setXor(outff.fb_DE, fb_DE);
- timer.stop(M.online_comp);
- return outff;
- }
- public OutFF runD(byte[][] fb_DE, byte[][] fb_CD, int i2) {
- timer.start(M.offline_comp);
- int n = fb_DE.length;
- OutFF outff = new OutFF();
- outff.fb_CD = new byte[n][1];
- outff.fb_DE = new byte[n][1];
- for (int i = 0; i < n; i++) {
- Crypto.sr_CD.nextBytes(outff.fb_CD[i]);
- Crypto.sr_DE.nextBytes(outff.fb_DE[i]);
- }
- timer.stop(M.offline_comp);
- // ----------------------------------------- //
- timer.start(M.online_comp);
- Shift shift = new Shift(con1, con2);
- shift.runD(n - i2, n, 1);
- Util.setXor(outff.fb_CD, fb_CD);
- Util.setXor(outff.fb_DE, fb_DE);
- timer.stop(M.online_comp);
- return outff;
- }
- public OutFF runC(byte[][] fb_CD, byte[][] fb_CE, int i1) {
- timer.start(M.offline_comp);
- int n = fb_CD.length;
- OutFF outff = new OutFF();
- outff.fb_CD = new byte[n][1];
- for (int i = 0; i < n; i++) {
- Crypto.sr_CD.nextBytes(outff.fb_CD[i]);
- }
- timer.stop(M.offline_comp);
- // ----------------------------------------- //
- timer.start(M.online_comp);
- byte[][] a1 = new byte[n][1];
- a1[i1][0] = 1;
- Shift shift = new Shift(con1, con2);
- byte[][] m1 = shift.runC(a1);
- for (int i = 0; i < n; i++)
- Util.setXor(m1[i], outff.fb_CD[i]);
- timer.start(M.online_write);
- con1.write(online_band, m1);
- timer.stop(M.online_write);
- timer.start(M.online_read);
- byte[][] m2 = con1.readDoubleByteArrayAndDec();
- timer.stop(M.online_read);
- outff.fb_CE = Util.xor(m1, m2);
- Util.setXor(outff.fb_CE, fb_CE);
- Util.setXor(outff.fb_CD, fb_CD);
- timer.stop(M.online_comp);
- return outff;
- }
- @Override
- public void run(Party party, Metadata md, Forest[] forest) {
- for (int j = 0; j < 100; j++) {
- int n = 100;
- int l = 1;
- byte[][] fb_CD = new byte[n][l];
- byte[][] fb_CE = new byte[n][l];
- byte[][] fb_DE = new byte[n][l];
- byte[][] fb = new byte[n][1];
- for (int i = 0; i < n; i++) {
- Crypto.sr.nextBytes(fb_CD[i]);
- Crypto.sr.nextBytes(fb_CE[i]);
- Crypto.sr.nextBytes(fb_DE[i]);
- fb[i][0] = (byte) (fb_CD[i][0] ^ fb_DE[i][0] ^ fb_CE[i][0]);
- }
- int i1 = Crypto.sr.nextInt(n);
- int i2 = Crypto.sr.nextInt(n);
- int i = (i1 + i2) % n;
- OutFF outff = new OutFF();
- if (party == Party.Eddie) {
- con1.write(fb_CD);
- con1.write(fb_DE);
- con1.write(i2);
- con2.write(fb_CE);
- con2.write(fb_CD);
- con2.write(i1);
- outff = this.runE(fb_DE, fb_CE, i2);
- outff.fb_CD = con1.readDoubleByteArray();
- byte[][] fbp = Util.xor(Util.xor(outff.fb_CD, outff.fb_CE), outff.fb_DE);
- fbp[i][0] = (byte) (fbp[i][0] ^ 1);
- boolean fail = false;
- for (int k = 0; k < n; k++) {
- if ((fb[k][0] & 1) != (fbp[k][0] & 1)) {
- System.err.println(j + ": FlipFlag test failed");
- fail = true;
- break;
- }
- }
- if (!fail)
- System.out.println(j + ": FlipFlag test passed");
- } else if (party == Party.Debbie) {
- fb_CD = con1.readDoubleByteArray();
- fb_DE = con1.readDoubleByteArray();
- i2 = con1.readInt();
- outff = this.runD(fb_DE, fb_CD, i2);
- con1.write(outff.fb_CD);
- } else if (party == Party.Charlie) {
- fb_CE = con1.readDoubleByteArray();
- fb_CD = con1.readDoubleByteArray();
- i1 = con1.readInt();
- outff = this.runC(fb_CD, fb_CE, i1);
- } else {
- throw new NoSuchPartyException(party + "");
- }
- }
- }
- @Override
- public void run(Party party, Metadata md, Forest forest) {
- }
- }
|