X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=PyORAM%2Fexamples%2Fpath_oram_ram.py;fp=PyORAM%2Fexamples%2Fpath_oram_ram.py;h=d7b99b0c0100f8d75bf5e31e90badf029ffe122c;hb=94177d594aa9ce08f05c7e7b44e5ca90c63a8cf6;hp=0000000000000000000000000000000000000000;hpb=0ff7f37c96339ab80c68452113b94e1074284bde;p=iotcloud.git

diff --git a/PyORAM/examples/path_oram_ram.py b/PyORAM/examples/path_oram_ram.py
new file mode 100644
index 0000000..d7b99b0
--- /dev/null
+++ b/PyORAM/examples/path_oram_ram.py
@@ -0,0 +1,91 @@
+#
+# This example measures the performance of Path ORAM
+# when storage is accessed through RAM.
+#
+
+import os
+import random
+import time
+
+import pyoram
+from pyoram.util.misc import MemorySize
+from pyoram.oblivious_storage.tree.path_oram import \
+    PathORAM
+from pyoram.storage.block_storage_ram import \
+    BlockStorageRAM
+
+import tqdm
+
+pyoram.config.SHOW_PROGRESS_BAR = True
+
+# Set the storage location and size
+storage_name = "heap.bin"
+# 4KB block size
+block_size = 4000
+# one block per bucket in the
+# storage heap of height 8
+block_count = 2**(8+1)-1
+
+def main():
+
+    print("Storage Name: %s" % (storage_name))
+    print("Block Count: %s" % (block_count))
+    print("Block Size: %s" % (MemorySize(block_size)))
+    print("Total Memory: %s"
+          % (MemorySize(block_size*block_count)))
+    print("Actual Storage Required: %s"
+          % (MemorySize(
+              PathORAM.compute_storage_size(
+                  block_size,
+                  block_count,
+                  storage_type='ram'))))
+    print("")
+
+    print("Setting Up Path ORAM Storage")
+    setup_start = time.time()
+    with PathORAM.setup(storage_name, # RAM storage ignores this argument
+                        block_size,
+                        block_count,
+                        storage_type='ram',
+                        ignore_existing=True) as f:
+        print("Total Setup Time: %2.f s"
+              % (time.time()-setup_start))
+        print("Current Stash Size: %s"
+              % len(f.stash))
+        print("Total Data Transmission: %s"
+              % (MemorySize(f.bytes_sent + f.bytes_received)))
+        print("")
+
+    # This must be done after closing the file to ensure the lock flag
+    # is set to False in the saved data. The tofile method only exists
+    # on BlockStorageRAM
+    f.raw_storage.tofile(storage_name)
+
+    # We close the device and reopen it after
+    # setup to reset the bytes sent and bytes
+    # received stats.
+    with PathORAM(BlockStorageRAM.fromfile(storage_name),
+                  f.stash,
+                  f.position_map,
+                  key=f.key) as f:
+
+        test_count = 100
+        start_time = time.time()
+        for t in tqdm.tqdm(list(range(test_count)),
+                           desc="Running I/O Performance Test"):
+            f.read_block(random.randint(0,f.block_count-1))
+        stop_time = time.time()
+        print("Current Stash Size: %s"
+              % len(f.stash))
+        print("Access Block Avg. Data Transmitted: %s (%.3fx)"
+              % (MemorySize((f.bytes_sent + f.bytes_received)/float(test_count)),
+                 (f.bytes_sent + f.bytes_received)/float(test_count)/float(block_size)))
+        print("Access Block Avg. Latency: %.2f ms"
+              % ((stop_time-start_time)/float(test_count)*1000))
+        print("")
+
+    # cleanup because this is a test example
+    os.remove(storage_name)
+
+if __name__ == "__main__":
+    main()                                             # pragma: no cover