init

Author: the-redhood

client.py

@@ -0,0 +1,80 @@
+from secretsharing import SecretSharer as Shamir
+import numpy as np
+import json
+import requests
+import time
+from random import randint
+
+#m1_raw = raw_input("User message:")
+m1_raw = "It is a secret message"
+
+u_length = 3
+v_length = 3
+location_U = randint(0, u_length-1)
+location_V = randint(0, v_length-1)	
+
+def parser(u1,a,b):
+    k = []
+
+    for i in range(len(u1)):
+        shares = Shamir.split_secret(u1[i],a,b)
+        tmp = []
+        for i in shares:
+            tmp.append(int(i[2:],16))
+        k.append(tmp)
+
+    k = np.transpose(np.array(k)).tolist()
+    return k
+
+# url = 'http://localhost:5000/server_0_uploader'
+url = 'http://localhost:5000/batch_predict'
+
+server_number = 3
+m1 = m1_raw.encode('hex')
+# m2 = 'let me die'.encode('hex')
+
+u1 = []
+v1 = []
+v1_square = []
+for i in range(u_length):
+	if i == location_U:
+		u1.append('1')
+	else:
+		u1.append('0')
+
+for i in range(v_length):
+	if i == location_V:
+		v1.append(m1)
+		v1_square.append(hex(int(m1,16) ** 2)[2:-1])
+	else:
+		v1.append('0')
+		v1_square.append('0')
+
+# u2 = ['0','1','0','0']
+# v2 = [m2,'0','0','0']
+
+start_time = time.time()
+u1_split = parser(u1,2,server_number)
+v1_split = parser(v1,2,server_number)
+v1_square = parser(v1_square,2,server_number)
+# h,i,j = parser(u2,2,server_number)
+# k,l,m = parser(v2,2,server_number)
+print("--- %s seconds ---" % (time.time() - start_time))
+#print u1_split
+#print v1_split
+
+
+# payload = {	'u_0' : u1_split[0],
+# 			'u_1' : u1_split[1],
+# 			'u_2' : u1_split[2],
+# 			'v_0' : v1_split[0],
+# 			'v_1' : v1_split[1],
+# 			'v_2' : v1_split[2],
+# 			'v1_square_0' : v1_square[0],
+# 			'v1_square_1' : v1_square[1],
+# 			'v1_square_2' : v1_square[2],
+# 		}
+payload = {	'u_0' : 0,
+			'u_1' : 1,
+		}
+r = requests.post(url, json=payload)

collision.py

@@ -0,0 +1,92 @@
+from secretsharing import SecretSharer as Shamir
+import numpy as np
+from decimal import *
+
+server_number = 3
+
+def parser(u1,a,b):
+	k = []
+
+	for i in range(len(u1)):
+		shares = Shamir.split_secret(u1[i],a,b)
+		tmp = []
+		for i in shares:
+			tmp.append(int(i[2:],16))
+		k.append(tmp)
+
+	k = np.transpose(np.array(k))
+	return k
+
+def recover(s):
+	db = []
+
+	for i in range(np.shape(s)[1]):
+		for j in range(np.shape(s)[2]):
+			tmp1 = []
+			for k in range(np.shape(s)[0]):
+				tmp1.append(str(k+1)+"-"+str(hex(s[k][i][j]))[2:-1])
+			answer = Shamir.recover_secret(tmp1)
+			if len(answer) % 2 == 1:
+				answer = '0' + answer
+			db.append(answer.decode('hex'))
+	return db
+
+
+
+m1 = 'the first secret message'.encode('hex')
+m2 = 'the second secret message'.encode('hex')
+
+u1 = ['1']
+v1 = [m1]
+v1_square = [hex(int(m1,16) ** 2)[2:-1]]
+
+u2 = ['1']
+v2 = [m2]
+v2_square = [hex(int(m2,16) ** 2)[2:-1]]
+
+#for i in range(server_number):
+#	s = u_share[i].reshape([-1,1]).dot(v_share[i].reshape([1,-1]))
+
+a,b,c = parser(u1,2,server_number)
+d,e,f = parser(v1,2,server_number)
+d_1,e_1,f_1 = parser(v1_square,2,server_number)
+h,i,j = parser(u2,2,server_number)
+k,l,m = parser(v2,2,server_number)
+k_1,l_1,m_1 = parser(v2_square,2,server_number)
+
+s1 = a.reshape([-1,1]).dot(d.reshape([1,-1]))
+s2 = b.reshape([-1,1]).dot(e.reshape([1,-1]))
+s3 = c.reshape([-1,1]).dot(f.reshape([1,-1]))
+
+s1_1 = a.reshape([-1,1]).dot(d_1.reshape([1,-1]))
+s2_1 = b.reshape([-1,1]).dot(e_1.reshape([1,-1]))
+s3_1 = c.reshape([-1,1]).dot(f_1.reshape([1,-1]))
+
+s4 = h.reshape([-1,1]).dot(k.reshape([1,-1]))
+s5 = i.reshape([-1,1]).dot(l.reshape([1,-1]))
+s6 = j.reshape([-1,1]).dot(m.reshape([1,-1]))
+
+s4_1 = h.reshape([-1,1]).dot(k_1.reshape([1,-1]))
+s5_1 = i.reshape([-1,1]).dot(l_1.reshape([1,-1]))
+s6_1 = j.reshape([-1,1]).dot(m_1.reshape([1,-1]))
+
+s1 += s4
+s2 += s5
+s3 += s6
+
+s1_1 += s4_1
+s2_1 += s5_1
+s3_1 += s6_1
+
+s = np.array([s1,s2,s3])
+s_1 = np.array([s1_1,s2_1,s3_1])
+
+msg_sum = recover(s)[0].encode('hex')
+square_sum = recover(s_1)[0].encode('hex')
+
+getcontext().prec = 100
+msg_minus = long(Decimal(int(square_sum,16) * 2-int(msg_sum,16) ** 2).sqrt())
+
+print hex((-msg_minus + int(msg_sum,16))/2)[2:-1].decode('hex')
+print hex((msg_minus + int(msg_sum,16))/2)[2:-1].decode('hex')
+

offline_message.py

@@ -0,0 +1,117 @@
+from flask import Flask, render_template, request
+import requests
+import json
+import os
+from secretsharing import SecretSharer as Shamir
+import numpy as np
+import time
+from werkzeug.contrib.cache import SimpleCache
+import atexit
+from apscheduler.schedulers.background import BackgroundScheduler
+from apscheduler.triggers.interval import IntervalTrigger
+import logging
+logging.basicConfig()
+
+cache = SimpleCache()
+app = Flask(__name__)
+
+scheduler = BackgroundScheduler()
+scheduler.start()
+# log = logging.getLogger('apscheduler.executors.default')
+# log.setLevel(logging.INFO)  # DEBUG
+
+# fmt = logging.Formatter('%(levelname)s:%(name)s:%(message)s')
+# h = logging.StreamHandler()
+# h.setFormatter(fmt)
+# log.addHandler(h)
+
+
+@app.route('/')
+def hello_world():
+    return 'Welcome to test'
+
+#request function that receive clients' vectors with jason form
+@app.route('/server_0_uploader', methods = ['POST'])
+def upload_request_():
+	if request.method == 'POST':
+		request_json  = request.get_json()
+		#print request_json
+		u_0 = np.array(request_json.get('u_0'))
+		u_1 = np.array(request_json.get('u_1'))
+		u_2 = np.array(request_json.get('u_2'))
+		v_0 = np.array(request_json.get('v_0'))
+		v_1 = np.array(request_json.get('v_1'))
+		v_2 = np.array(request_json.get('v_2'))
+		# v1_square_0 = np.array(request_json.get('v_0_square'))
+		# v1_square_1 = np.array(request_json.get('v_1_square'))
+		# v1_square_2 = np.array(request_json.get('v_2_square'))
+		# use reshape to perform 1-d array transpose
+		u_0_v_0 = u_0.reshape([-1,1]).dot(v_0.reshape([1,-1]))
+		u_1_v_1 = u_1.reshape([-1,1]).dot(v_1.reshape([1,-1]))
+		u_2_v_2 = u_2.reshape([-1,1]).dot(v_2.reshape([1,-1]))
+
+		# u_0_v_0_square = u_0.reshape([-1,1]).dot(v1_square_0.reshape([1,-1]))
+		# u_1_v_1_square = u_1.reshape([-1,1]).dot(v1_square_1.reshape([1,-1]))
+		# u_2_v_2_square = u_2.reshape([-1,1]).dot(v1_square_2.reshape([1,-1]))
+
+		batch_list = np.array([u_0_v_0,u_1_v_1,u_2_v_2])
+		# batch_square_list = [u_0_v_0_square,u_1_v_1_square,u_2_v_2_square]
+		# batch_list = np.array([batch_list, batch_square_list])
+		batch_result_save(batch_list)
+
+		return 'message sent'
+
+#batch function that used to save client's results
+def batch_result_save(batch_list):
+	batch_result = get_batch_result()
+	if (type(batch_result) is not np.ndarray):
+		set_batch_result(batch_list)
+	else:
+		for i in range(0,3):
+			batch_result[i] = batch_result[i] + batch_list[i]
+		set_batch_result(batch_result)
+
+#used to print result for simple demo, will be replaced if database is implemented
+def batch_result():
+	batch_result = get_batch_result()
+	if (type(batch_result) is not np.ndarray): 
+		print "message is empty"
+	else:
+		print recover(batch_result)
+		set_batch_result(None)
+
+
+#recover function perform database reconstruction
+def recover(s):
+	db = []
+
+	for i in range(np.shape(s)[1]):
+		for j in range(np.shape(s)[2]):
+			tmp1 = []
+			for k in range(np.shape(s)[0]):
+				tmp1.append(str(k+1)+"-"+str(hex(s[k][i][j]))[2:-1])
+			answer = Shamir.recover_secret(tmp1)
+			if len(answer) % 2 == 1:
+				answer = '0' + answer
+			db.append(answer.decode('hex'))
+	return db
+
+def get_batch_result():
+	return cache.get('batch_result')
+
+def set_batch_result(batch_input):
+	cache.set('batch_result', batch_input, timeout=5 * 60)
+
+scheduler.add_job(
+    func=batch_result,
+    trigger=IntervalTrigger(seconds=8),
+    id='batch_result',
+    name='Print datebase with 5s batch',
+    replace_existing=True)
+# Shut down the scheduler when exiting the app
+atexit.register(lambda: scheduler.shutdown())
+
+if __name__ == '__main__':
+	app.run(host='0.0.0.0', port=80, debug=True)
+
+