123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416 |
- package protocols;
- import java.math.BigInteger;
- import java.util.Arrays;
- import communication.Communication;
- import crypto.Crypto;
- import exceptions.NoSuchPartyException;
- import oram.Forest;
- import oram.Global;
- import oram.Metadata;
- import oram.Tree;
- import oram.Tuple;
- import struct.OutFF;
- import struct.OutPIRAccess;
- import struct.OutULiT;
- import struct.Party;
- import struct.TwoThreeXorByte;
- import struct.TwoThreeXorInt;
- import util.Bandwidth;
- import util.M;
- import util.P;
- import util.StopWatch;
- import util.Util;
- // TODO: really FlipFlag on path, and update path in Eviction
- // TODO: fix simulation
- public class PIRRetrieve extends Protocol {
- Communication[] cons1;
- Communication[] cons2;
- int pid = P.RTV;
- public PIRRetrieve(Communication con1, Communication con2) {
- super(con1, con2);
- online_band = all.online_band[pid];
- offline_band = all.offline_band[pid];
- timer = all.timer[pid];
- }
- public void setCons(Communication[] a, Communication[] b) {
- cons1 = a;
- cons2 = b;
- }
- public OutPIRAccess runE(Metadata md, Tree tree_DE, Tree tree_CE, byte[] Li, TwoThreeXorByte L, TwoThreeXorByte N,
- TwoThreeXorInt dN) {
- timer.start(M.online_comp);
- int treeIndex = tree_DE.getTreeIndex();
- boolean isLastTree = treeIndex == md.getNumTrees() - 1;
- boolean isFirstTree = treeIndex == 0;
- PIRAccess piracc = new PIRAccess(con1, con2);
- OutPIRAccess outpiracc = piracc.runE(md, tree_DE, tree_CE, Li, L, N, dN);
- OutULiT T = new OutULiT();
- if (!isLastTree) {
- TwoThreeXorByte Lp = new TwoThreeXorByte(md.getLBytesOfTree(treeIndex));
- TwoThreeXorByte Lpi = new TwoThreeXorByte(md.getLBytesOfTree(treeIndex + 1));
- ULiT ulit = new ULiT(con1, con2, Crypto.sr_DE, Crypto.sr_CE);
- T = ulit.runE(outpiracc.X, N, dN, Lp, Lpi, outpiracc.nextL, md.getTwoTauPow());
- } else {
- T.DE = outpiracc.pathTuples_DE[0];
- T.CE = outpiracc.pathTuples_CE[0];
- }
- int pathTuples = outpiracc.pathTuples_CE.length;
- if (!isFirstTree) {
- byte[][] fb_DE = new byte[pathTuples][];
- byte[][] fb_CE = new byte[pathTuples][];
- for (int i = 0; i < pathTuples; i++) {
- fb_DE[i] = outpiracc.pathTuples_DE[i].getF();
- fb_CE[i] = outpiracc.pathTuples_CE[i].getF();
- }
- FlipFlag ff = new FlipFlag(con1, con2);
- OutFF outff = ff.runE(fb_DE, fb_CE, outpiracc.j.s_DE);
- for (int i = 0; i < pathTuples; i++) {
- // outpiracc.pathTuples_DE[i].setF(outff.fb_DE[i]);
- // outpiracc.pathTuples_CE[i].setF(outff.fb_CE[i]);
- }
- }
- int stashSize = tree_DE.getStashSize();
- int[] tupleParam = new int[] { treeIndex == 0 ? 0 : 1, md.getNBytesOfTree(treeIndex),
- md.getLBytesOfTree(treeIndex), md.getABytesOfTree(treeIndex) };
- Tuple[] path = new Tuple[pathTuples];
- for (int i = 0; i < pathTuples; i++) {
- path[i] = outpiracc.pathTuples_DE[i].xor(outpiracc.pathTuples_CE[i]);
- }
- Tuple[] R = Arrays.copyOfRange(path, 0, stashSize);
- UpdateRoot updateroot = new UpdateRoot(con1, con2);
- R = updateroot.runE(isFirstTree, stashSize, md.getLBitsOfTree(treeIndex), tupleParam, Li, R, T.DE.xor(T.CE));
- System.arraycopy(R, 0, path, 0, R.length);
- Eviction eviction = new Eviction(con1, con2);
- eviction.runE(isFirstTree, tupleParam, Li, path, tree_DE);
- // simulation of Reshare
- timer.start(M.online_write);
- con2.write(online_band, path);
- timer.stop(M.online_write);
- timer.start(M.online_read);
- con2.readTupleArrayAndDec();
- timer.stop(M.online_read);
- // second eviction sim
- for (int i = 0; i < pathTuples; i++) {
- path[i] = outpiracc.pathTuples_DE[i].xor(outpiracc.pathTuples_CE[i]);
- }
- R = Arrays.copyOfRange(path, 0, stashSize);
- R = updateroot.runE(isFirstTree, stashSize, md.getLBitsOfTree(treeIndex), tupleParam, Li, R, T.DE.xor(T.CE));
- System.arraycopy(R, 0, path, 0, R.length);
- eviction.runE(isFirstTree, tupleParam, Li, path, tree_DE);
- // simulation of Reshare
- timer.start(M.online_write);
- con2.write(online_band, path);
- timer.stop(M.online_write);
- timer.start(M.online_read);
- con2.readTupleArrayAndDec();
- timer.stop(M.online_read);
- timer.stop(M.online_comp);
- return outpiracc;
- }
- public OutPIRAccess runD(Metadata md, Tree tree_DE, Tree tree_CD, byte[] Li, TwoThreeXorByte L, TwoThreeXorByte N,
- TwoThreeXorInt dN) {
- timer.start(M.online_comp);
- int treeIndex = tree_DE.getTreeIndex();
- boolean isLastTree = treeIndex == md.getNumTrees() - 1;
- boolean isFirstTree = treeIndex == 0;
- PIRAccess piracc = new PIRAccess(con1, con2);
- OutPIRAccess outpiracc = piracc.runD(md, tree_DE, tree_CD, Li, L, N, dN);
- OutULiT T = new OutULiT();
- if (!isLastTree) {
- TwoThreeXorByte Lp = new TwoThreeXorByte(md.getLBytesOfTree(treeIndex));
- TwoThreeXorByte Lpi = new TwoThreeXorByte(md.getLBytesOfTree(treeIndex + 1));
- ULiT ulit = new ULiT(con1, con2, Crypto.sr_DE, Crypto.sr_CD);
- T = ulit.runD(outpiracc.X, N, dN, Lp, Lpi, outpiracc.nextL, md.getTwoTauPow());
- } else {
- T.CD = outpiracc.pathTuples_CD[0];
- T.DE = outpiracc.pathTuples_DE[0];
- }
- int pathTuples = outpiracc.pathTuples_CD.length;
- if (!isFirstTree) {
- byte[][] fb_DE = new byte[pathTuples][];
- byte[][] fb_CD = new byte[pathTuples][];
- for (int i = 0; i < pathTuples; i++) {
- fb_DE[i] = outpiracc.pathTuples_DE[i].getF();
- fb_CD[i] = outpiracc.pathTuples_CD[i].getF();
- }
- FlipFlag ff = new FlipFlag(con1, con2);
- OutFF outff = ff.runD(fb_DE, fb_CD, outpiracc.j.s_DE);
- for (int i = 0; i < pathTuples; i++) {
- // outpiracc.pathTuples_DE[i].setF(outff.fb_DE[i]);
- // outpiracc.pathTuples_CD[i].setF(outff.fb_CD[i]);
- }
- }
- int stashSize = tree_DE.getStashSize();
- int[] tupleParam = new int[] { treeIndex == 0 ? 0 : 1, md.getNBytesOfTree(treeIndex),
- md.getLBytesOfTree(treeIndex), md.getABytesOfTree(treeIndex) };
- UpdateRoot updateroot = new UpdateRoot(con1, con2);
- updateroot.runD(isFirstTree, stashSize, md.getLBitsOfTree(treeIndex), tupleParam, Li, tree_DE.getW());
- Eviction eviction = new Eviction(con1, con2);
- eviction.runD(isFirstTree, tupleParam, Li, tree_DE);
- // second eviction sim
- updateroot.runD(isFirstTree, stashSize, md.getLBitsOfTree(treeIndex), tupleParam, Li, tree_DE.getW());
- eviction.runD(isFirstTree, tupleParam, Li, tree_DE);
- timer.stop(M.online_comp);
- return outpiracc;
- }
- public OutPIRAccess runC(Metadata md, Tree tree_CD, Tree tree_CE, byte[] Li, TwoThreeXorByte L, TwoThreeXorByte N,
- TwoThreeXorInt dN) {
- timer.start(M.online_comp);
- int treeIndex = tree_CE.getTreeIndex();
- boolean isLastTree = treeIndex == md.getNumTrees() - 1;
- boolean isFirstTree = treeIndex == 0;
- PIRAccess piracc = new PIRAccess(con1, con2);
- OutPIRAccess outpiracc = piracc.runC(md, tree_CD, tree_CE, Li, L, N, dN);
- OutULiT T = new OutULiT();
- if (!isLastTree) {
- TwoThreeXorByte Lp = new TwoThreeXorByte(md.getLBytesOfTree(treeIndex));
- TwoThreeXorByte Lpi = new TwoThreeXorByte(md.getLBytesOfTree(treeIndex + 1));
- ULiT ulit = new ULiT(con1, con2, Crypto.sr_CE, Crypto.sr_CD);
- T = ulit.runC(outpiracc.X, N, dN, Lp, Lpi, outpiracc.nextL, md.getTwoTauPow());
- } else {
- T.CD = outpiracc.pathTuples_CD[0];
- T.CE = outpiracc.pathTuples_CE[0];
- }
- int pathTuples = outpiracc.pathTuples_CD.length;
- if (!isFirstTree) {
- byte[][] fb_CE = new byte[pathTuples][];
- byte[][] fb_CD = new byte[pathTuples][];
- for (int i = 0; i < pathTuples; i++) {
- fb_CE[i] = outpiracc.pathTuples_CE[i].getF();
- fb_CD[i] = outpiracc.pathTuples_CD[i].getF();
- }
- FlipFlag ff = new FlipFlag(con1, con2);
- OutFF outff = ff.runC(fb_CD, fb_CE, outpiracc.j.t_C);
- for (int i = 0; i < pathTuples; i++) {
- // outpiracc.pathTuples_CD[i].setF(outff.fb_CD[i]);
- // outpiracc.pathTuples_CE[i].setF(outff.fb_CE[i]);
- }
- }
- int stashSize = tree_CE.getStashSize();
- int[] tupleParam = new int[] { treeIndex == 0 ? 0 : 1, md.getNBytesOfTree(treeIndex),
- md.getLBytesOfTree(treeIndex), md.getABytesOfTree(treeIndex) };
- Tuple[] path = outpiracc.pathTuples_CD;
- Tuple[] R = Arrays.copyOfRange(path, 0, stashSize);
- UpdateRoot updateroot = new UpdateRoot(con1, con2);
- R = updateroot.runC(isFirstTree, tupleParam, R, T.CD);
- System.arraycopy(R, 0, path, 0, R.length);
- Eviction eviction = new Eviction(con1, con2);
- eviction.runC(isFirstTree, tupleParam, path, tree_CD.getD(), stashSize, tree_CD.getW());
- // simulation of Reshare
- timer.start(M.online_write);
- con1.write(online_band, path);
- timer.stop(M.online_write);
- timer.start(M.online_read);
- con1.readTupleArrayAndDec();
- timer.stop(M.online_read);
- // second eviction sim
- R = Arrays.copyOfRange(path, 0, stashSize);
- R = updateroot.runC(isFirstTree, tupleParam, R, T.CD);
- System.arraycopy(R, 0, path, 0, R.length);
- eviction.runC(isFirstTree, tupleParam, path, tree_CD.getD(), stashSize, tree_CD.getW());
- // simulation of Reshare
- timer.start(M.online_write);
- con1.write(online_band, path);
- timer.stop(M.online_write);
- timer.start(M.online_read);
- con1.readTupleArrayAndDec();
- timer.stop(M.online_read);
- timer.stop(M.online_comp);
- return outpiracc;
- }
- @Override
- public void run(Party party, Metadata md, Forest[] forest) {
- StopWatch ete = new StopWatch("ETE");
- Tree tree_CD = null;
- Tree tree_DE = null;
- Tree tree_CE = null;
- int iterations = md.getIters()+2;
- int reset = 2;
- for (int test = 0; test < iterations; test++) {
- if (test == reset) {
- all.resetTime();
- ete.reset();
- }
- if (test == 1) {
- Global.bandSwitch = false;
- }
- for (int treeIndex = 0; treeIndex < md.getNumTrees(); treeIndex++) {
- if (party == Party.Eddie) {
- tree_DE = forest[0].getTree(treeIndex);
- tree_CE = forest[1].getTree(treeIndex);
- } else if (party == Party.Debbie) {
- tree_DE = forest[0].getTree(treeIndex);
- tree_CD = forest[1].getTree(treeIndex);
- } else if (party == Party.Charlie) {
- tree_CE = forest[0].getTree(treeIndex);
- tree_CD = forest[1].getTree(treeIndex);
- } else {
- throw new NoSuchPartyException(party + "");
- }
- int Llen = md.getLBytesOfTree(treeIndex);
- int Nlen = md.getNBytesOfTree(treeIndex);
- TwoThreeXorInt dN = new TwoThreeXorInt();
- TwoThreeXorByte N = new TwoThreeXorByte();
- N.CD = new byte[Nlen];
- N.DE = new byte[Nlen];
- N.CE = new byte[Nlen];
- TwoThreeXorByte L = new TwoThreeXorByte();
- L.CD = new byte[Llen];
- L.DE = new byte[Llen];
- L.CE = new byte[Llen];
- byte[] Li = new byte[Llen];
- if (party == Party.Eddie) {
- ete.start();
- OutPIRAccess out = this.runE(md, tree_DE, tree_CE, Li, L, N, dN);
- ete.stop();
- out.j.t_D = con1.readInt();
- out.j.t_C = con2.readInt();
- out.X.CD = con1.read();
- int pathTuples = out.pathTuples_CE.length;
- int index = (out.j.t_D + out.j.s_CE) % pathTuples;
- byte[] X = Util.xor(Util.xor(out.X.DE, out.X.CE), out.X.CD);
- boolean fail = false;
- if (index != 0) {
- System.err.println(test + " " + treeIndex + ": PIRAcc test failed on KSearch index");
- fail = true;
- }
- if (new BigInteger(1, X).intValue() != 0) {
- System.err.println(test + " " + treeIndex + ": PIRAcc test failed on 3ShiftPIR X");
- fail = true;
- }
- if (treeIndex < md.getNumTrees() - 1 && new BigInteger(1, out.Lip1).intValue() != 0) {
- System.err.println(test + " " + treeIndex + ": PIRAcc test failed on 3ShiftXorPIR Lip1");
- fail = true;
- }
- if (!fail)
- System.out.println(test + " " + treeIndex + ": PIRAcc test passed");
- } else if (party == Party.Debbie) {
- ete.start();
- OutPIRAccess out = this.runD(md, tree_DE, tree_CD, Li, L, N, dN);
- ete.stop();
- con1.write(out.j.t_D);
- con1.write(out.X.CD);
- } else if (party == Party.Charlie) {
- ete.start();
- OutPIRAccess out = this.runC(md, tree_CD, tree_CE, Li, L, N, dN);
- ete.stop();
- con1.write(out.j.t_C);
- } else {
- throw new NoSuchPartyException(party + "");
- }
- }
- }
- System.out.println();
- Bandwidth band_on = new Bandwidth("Total Online Band");
- Bandwidth band_off = new Bandwidth("Total Offline Band");
- for (int i = 0; i < P.size; i++) {
- System.out.println(all.offline_band[i].toString());
- System.out.println(all.online_band[i].toString());
- band_on.add(all.online_band[i]);
- band_off.add(all.offline_band[i]);
- }
- System.out.println(band_off.toString());
- System.out.println(band_on.toString());
- System.out.println();
- StopWatch time_on = new StopWatch("Total Online Time");
- StopWatch time_off = new StopWatch("Total Offline Time");
- for (int i = 0; i < P.size; i++) {
- for (int j = 0; j < 3; j++)
- time_on.add(all.timer[i].watches[j]);
- for (int j = 3; j < M.size; j++)
- time_off.add(all.timer[i].watches[j]);
- all.timer[i].noPrePrint();
- }
- System.out.println(time_off.toMS());
- System.out.println(time_on.toMS());
- System.out.println();
- System.out.println(ete.toMS());
- System.out.println();
- sanityCheck();
- }
- @Override
- public void run(Party party, Metadata md, Forest forest) {
- }
- }
|