run-experiment.sh 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. #!/bin/bash
  2. # Get the IP addresses
  3. p0addr=$(docker inspect --format='{{ .NetworkSettings.IPAddress }}' duoram_p0)
  4. p1addr=$(docker inspect --format='{{ .NetworkSettings.IPAddress }}' duoram_p1)
  5. p2addr=$(docker inspect --format='{{ .NetworkSettings.IPAddress }}' duoram_p2)
  6. # The Duoram mode and size
  7. # mode is read, write, readwrite, or init
  8. duoramtype=2P
  9. phase=online
  10. mode=read
  11. size=20
  12. iters=128
  13. modeval=0
  14. ramsize=0
  15. if [ "$1" != "" ]; then
  16. mode="$1"
  17. fi
  18. if [ "$2" != "" ]; then
  19. size="$2"
  20. fi
  21. if [ "$3" != "" ]; then
  22. iters="$3"
  23. fi
  24. if [ "$4" != "" ]; then
  25. phase="$4"
  26. fi
  27. if [ "$5" != "" ]; then
  28. duoramtype="$5"
  29. fi
  30. if [ "$6" != "" ]; then
  31. ramsize="$6"
  32. fi
  33. if [ "$mode" = "read" ]; then
  34. modeval="0"
  35. fi
  36. if [ "$mode" = "write" ]; then
  37. modeval="0"
  38. fi
  39. if [ "$mode" = "readwrite" ]; then
  40. modeval="1"
  41. fi
  42. items=$((1<<size))
  43. echo ===== Running duoram $mode $size $iters
  44. date "+===== Start %s %F %T"
  45. if [ "$4" = "online" -a "$5" = "3P" ]; then
  46. # Run, saving the output
  47. savefilep2=$$.p2.out
  48. savefilep0=$$.p1.out
  49. savefilep1=$$.p0.out
  50. docker exec -w /root/duoram/duoram-online duoram_p2 bash -c "$DUORAM_NUMA_P2 stdbuf -o 0 ./p2 $p0addr $p1addr $size $iters $iters $iters 1 > $savefilep2 2>&1" &
  51. sleep 4
  52. docker exec -w /root/duoram/duoram-online duoram_p1 bash -c "$DUORAM_NUMA_P1 stdbuf -o 0 ./duoram1 $p0addr $p2addr $size $iters $iters $iters 1 > $savefilep1 2>&1" &
  53. sleep 2
  54. docker exec -w /root/duoram/duoram-online duoram_p0 bash -c "$DUORAM_NUMA_P0 stdbuf -o 0 ./duoram0 $p1addr $p2addr $size $iters $iters $iters 1 > $savefilep0 2>&1" &
  55. wait
  56. echo ===== P0 output
  57. docker exec -w /root/duoram/duoram-online duoram_p0 cat $savefilep0
  58. docker exec -w /root/duoram/duoram-online duoram_p0 rm -f $savefilep0
  59. echo ===== P1 output
  60. docker exec -w /root/duoram/duoram-online duoram_p1 cat $savefilep1
  61. docker exec -w /root/duoram/duoram-online duoram_p1 rm -f $savefilep1
  62. date "+===== End %s %F %T"
  63. fi
  64. if [ "$4" = "preproc" -a "$5" = "3P" ]; then
  65. # Run, saving the output
  66. savefilep2=$$.p2.out
  67. savefilep0=$$.p1.out
  68. savefilep1=$$.p0.out
  69. docker exec -w /root/duoram/preprocessing duoram_p2 bash -c "$DUORAM_NUMA_P2 stdbuf -o 0 ./p2preprocessing $p0addr $p1addr $iters $size $modeval $ramsize > $savefilep2 2>&1" &
  70. sleep 4
  71. docker exec -w /root/duoram/preprocessing duoram_p1 bash -c "$DUORAM_NUMA_P1 stdbuf -o 0 ./preprocessing1 $p0addr $p2addr $iters $size $modeval $ramsize > $savefilep1 2>&1" &
  72. sleep 2
  73. docker exec -w /root/duoram/preprocessing duoram_p0 bash -c "$DUORAM_NUMA_P0 stdbuf -o 0 ./preprocessing0 $p1addr $p2addr $iters $size $modeval $ramsize > $savefilep0 2>&1" &
  74. wait
  75. echo ===== P0 output
  76. docker exec -w /root/duoram/preprocessing duoram_p0 cat $savefilep0
  77. docker exec -w /root/duoram/preprocessing duoram_p0 rm -f $savefilep0
  78. echo ===== P1 output
  79. docker exec -w /root/duoram/preprocessing duoram_p1 cat $savefilep1
  80. docker exec -w /root/duoram/preprocessing duoram_p1 rm -f $savefilep1
  81. date "+===== End %s %F %T"
  82. fi
  83. if [ "$1" = "read" -a "$5" = "2P" ]; then
  84. # Run, saving the output
  85. savefilep0=$$.p1.out
  86. savefilep1=$$.p0.out
  87. docker exec -w /root/duoram/cpir-read/cxx duoram_p1 bash -c "$DUORAM_NUMA_P1 stdbuf -o 0 ./spir_test1 $p0addr $size $iters $iters $iters > $savefilep1 2>&1" &
  88. echo "P0 launched"
  89. sleep 4
  90. docker exec -w /root/duoram/cpir-read/cxx duoram_p0 bash -c "$DUORAM_NUMA_P0 stdbuf -o 0 ./spir_test0 $p1addr $size $iters $iters $iters > $savefilep0 2>&1" &
  91. echo "P1 launched"
  92. wait
  93. echo ===== P0 output
  94. docker exec -w /root/duoram/cpir-read/cxx duoram_p0 cat $savefilep0
  95. docker exec -w /root/duoram/cpir-read/cxx duoram_p0 rm -f $savefilep0
  96. echo ===== P1 output
  97. docker exec -w /root/duoram/cpir-read/cxx duoram_p1 cat $savefilep1
  98. docker exec -w /root/duoram/cpir-read/cxx duoram_p1 rm -f $savefilep1
  99. date "+===== End %s %F %T"
  100. fi
  101. if [ "$1" = "write" -a "$5" = "2P" -a "$4" = "preproc" ]; then
  102. # Run, saving the output
  103. savefilep0=$$.p1.out
  104. savefilep1=$$.p0.out
  105. docker exec -w /root/duoram/2p-preprocessing duoram_p1 ./OT $p0addr $p1addr 1 128 &
  106. echo "P1 launched"
  107. sleep 1
  108. docker exec -w /root/duoram/2p-preprocessing duoram_p0 ./OT $p0addr $p1addr 0 128 &
  109. sleep 2
  110. wait
  111. docker exec -w /root/duoram/2p-preprocessing duoram_p1 bash -c "$DUORAM_NUMA_P1 stdbuf -o 0 ./preprocessing1 $p0addr $size $iters > $savefilep1 2>&1" &
  112. sleep 4
  113. docker exec -w /root/duoram/2p-preprocessing duoram_p0 bash -c "$DUORAM_NUMA_P0 stdbuf -o 0 ./preprocessing0 $p1addr $size $iters > $savefilep0 2>&1" &
  114. wait
  115. echo ===== P0 output
  116. docker exec -w /root/duoram/2p-preprocessing duoram_p0 cat $savefilep0
  117. docker exec -w /root/duoram/2p-preprocessing duoram_p0 rm -f $savefilep0
  118. echo ===== P1 output
  119. docker exec -w /root/duoram/2p-preprocessing duoram_p1 cat $savefilep1
  120. docker exec -w /root/duoram/2p-preprocessing duoram_p1 rm -f $savefilep1
  121. date "+===== End %s %F %T"
  122. fi
  123. if [ "$1" = "write" -a "$5" = "2P" -a "$4" = "online" ]; then
  124. # Run, saving the output
  125. savefilep0=$$.p1.out
  126. savefilep1=$$.p0.out
  127. docker exec -w /root/duoram/duoram-online duoram_p1 bash -c "$DUORAM_NUMA_P1 stdbuf -o 0 ./2Pduoram1 $p0addr $p2addr $size $iters $iters $iters 1 > $savefilep1 2>&1" &
  128. sleep 4
  129. docker exec -w /root/duoram/duoram-online duoram_p0 bash -c "$DUORAM_NUMA_P0 stdbuf -o 0 ./2Pduoram0 $p1addr $p2addr $size $iters $iters $iters 1 > $savefilep0 2>&1" &
  130. wait
  131. echo ===== P0 output
  132. docker exec -w /root/duoram/duoram-online duoram_p0 cat $savefilep0
  133. docker exec -w /root/duoram/duoram-online duoram_p0 rm -f $savefilep0
  134. echo ===== P1 output
  135. docker exec -w /root/duoram/duoram-online duoram_p1 cat $savefilep1
  136. docker exec -w /root/duoram/duoram-online duoram_p1 rm -f $savefilep1
  137. date "+===== End %s %F %T"
  138. fi