60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from pydoc import plain
|
||
|
|
import sys
|
||
|
|
import multiprocessing
|
||
|
|
import numpy as np
|
||
|
|
import tensorflow as tf
|
||
|
|
import time
|
||
|
|
import random
|
||
|
|
|
||
|
|
def loadCipherTable(id, saveTableName, cipherTableShape):
|
||
|
|
'''
|
||
|
|
This module read and reconstruct the ciphertext table from bytesbuffer
|
||
|
|
'''
|
||
|
|
# filePath = "cache/P{}_".format(id) + saveTableName + "_ciphertext.sdata"
|
||
|
|
filePath = "server/P{}/P{}_".format(id,id) + saveTableName + ".sdata"
|
||
|
|
fd = open(filePath, "rb")
|
||
|
|
bytesbuffer = fd.read()
|
||
|
|
fd.close()
|
||
|
|
'''
|
||
|
|
The reconstruct parameters is equal to saveCipherTable.py
|
||
|
|
'''
|
||
|
|
cipherTable = np.frombuffer(bytesbuffer, dtype = "S16", count = cipherTableShape[0]*cipherTableShape[1], offset = 0)
|
||
|
|
cipherTable.resize((cipherTableShape[0], cipherTableShape[1]))
|
||
|
|
return cipherTable
|
||
|
|
|
||
|
|
def plaintextReveal(id, saveTableName, cipherTableShape):
|
||
|
|
'''
|
||
|
|
This module aimd to verify the correctness for loadCipherTable module
|
||
|
|
'''
|
||
|
|
sys.argv.extend(["--node_id", "P{}".format(id)])
|
||
|
|
import latticex.rosetta as rtt
|
||
|
|
rtt.activate("SecureNN")
|
||
|
|
|
||
|
|
res = loadCipherTable(id, saveTableName, cipherTableShape)
|
||
|
|
sess = tf.Session()
|
||
|
|
sess.run(tf.compat.v1.global_variables_initializer())
|
||
|
|
# Take a glance at the ciphertext
|
||
|
|
ciphertext = tf.reshape(res, res.shape)
|
||
|
|
cipher_result = sess.run(ciphertext)
|
||
|
|
# print('From ID:{} local ciphertext result:\n'.format(id), cipher_result)
|
||
|
|
|
||
|
|
a_and_c_can_get_plain = 0b101
|
||
|
|
print('From ID:{} plaintext reveal result:\n'.format(id), sess.run(rtt.SecureReveal(cipher_result, a_and_c_can_get_plain)))
|
||
|
|
|
||
|
|
|
||
|
|
# p0 = multiprocessing.Process(target = plaintextReveal, args = (0, "toyExampleTable", [1000, 6]))
|
||
|
|
# p1 = multiprocessing.Process(target = plaintextReveal, args = (1, "toyExampleTable", [1000, 6]))
|
||
|
|
# p2 = multiprocessing.Process(target = plaintextReveal, args = (2, "toyExampleTable", [1000, 6]))
|
||
|
|
|
||
|
|
# p0.daemon = True
|
||
|
|
# p0.start()
|
||
|
|
# p1.daemon = True
|
||
|
|
# p1.start()
|
||
|
|
# p2.daemon = True
|
||
|
|
# p2.start()
|
||
|
|
|
||
|
|
# p0.join()
|
||
|
|
# p1.join()
|
||
|
|
# p2.join()
|