add things to comm

Author: herocharge

src/client.py

@@ -2,6 +2,8 @@
 from comm import Comm
 import tenseal as ts
 import cv2
+import dill
+import matplotlib.pyplot as plt
 
 def init_enc():
     ## Encryption Parameters
@@ -35,6 +37,21 @@ def encrypt_image(context, img):
         enc_img.append(enc_row)
     return enc_img
 
+def encrypt_pyramid(context, img):
+    enc_pyramid = []
+    for octave in range(len(img)):
+        enc_octave = []
+        for size in range(len(img[0])):
+            enc_img = []
+            for row in range(len(img[0][0])):
+                enc_row = []
+                for col in range(len(img[0][0][0])):
+                    enc_row.append(ts.ckks_vector(context, img[octave][size][row][col]))
+                enc_img.append(enc_row)
+            enc_octave.append(enc_img)
+        enc_pyramid.append(enc_octave)
+    return enc_pyramid
+
 def load_image(context, img):
     enc_img = []
     for row in range(len(img)):
@@ -44,32 +61,138 @@ def load_image(context, img):
         enc_img.append(enc_row)
     return enc_img
 
+def serialize_image(context, img):
+    enc_img = []
+    for row in range(len(img)):
+        enc_row = []
+        for col in range(len(img[0])):
+            enc_row.append((img[row][col].serialize()))
+        enc_img.append(enc_row)
+    return enc_img
+
+def load_pyramid(context, img):
+    enc_pyramid = []
+    for octave in range(len(img)):
+        enc_octave = []
+        for size in range(len(img[0])):
+            enc_img = []
+            for row in range(len(img[0][0])):
+                enc_row = []
+                for col in range(len(img[0][0][0])):
+                    enc_row.append(ts.ckks_vector_from(context, img[octave][size][row][col]))
+                enc_img.append(enc_row)
+            enc_octave.append(enc_img)
+        enc_pyramid.append(enc_octave)
+    return enc_pyramid
+
+def serialize_pyramid(context, img):
+    enc_pyramid = []
+    for octave in range(len(img)):
+        enc_octave = []
+        for size in range(len(img[0])):
+            enc_img = []
+            for row in range(len(img[0][0])):
+                enc_row = []
+                for col in range(len(img[0][0][0])):
+                    enc_row.append((img[octave][size][row][col].serialize()))
+                enc_img.append(enc_row)
+            enc_octave.append(enc_img)
+        enc_pyramid.append(enc_octave)
+    return enc_pyramid
+
+def show_pyramid(img, secret_key):
+    for octave in range(len(img)):
+        for size in range(len(img[0])):
+            base_img = ([[int(col.decrypt(secret_key)[0]) for col in row] for row in img[0][0]])
+            plt.imshow(base_img)
+            plt.show()
+
 
 def main():
     comm = Comm()
     comm.start_client()
     context, secret_key = init_enc()
 
+    def cmp(x, a, b):
+        x = dill.loads(x)
+        a = dill.loads(a)
+        b = dill.loads(b)
+        if isinstance(x, ts.CKKSVector):
+            x = x.decrypt(secret_key)[0]
+        if isinstance(a, ts.CKKSVector):
+            a = a.decrypt(secret_key)[0]
+        if isinstance(b, ts.CKKSVector):
+            b = b.decrypt(secret_key)[0]
+        return dill.dumps(ts.ckks_vector(context=context, vector=[int(a < x < b)]))
+
+    def refresh(value):
+        value = dill.loads(value)
+        decrypted_value = value.decrypt(secret_key)[0]
+        return dill.dumps(ts.ckks_vector(context=context, vector=[decrypted_value]))
+
+
     img = cv2.imread('../assets/fly-20x20.jpg')
-    for row in img:
-        for col in img:
-            print(col)
+    sigma=1.6
+    num_intervals=3
+    assumed_blur=0.5
+    image_border_width=5
 
     comm.call_api(b'init_context', context.serialize())
 
     enc_img = encrypt_image(context, img)
 
+    # Generate Base Image
     comm.call_api(b'generate_base_image')
 
     print("Sending image...")
     comm.send_img(enc_img)
     print("Image sent...")
 
-    res_img = load_image(context, comm.recv_img())
 
-    for row in res_img:
-        for col in row:
-            print(col.decrypt(secret_key))
+    comm.check_interactive(cmp=cmp, refresh=refresh)
+
+    base_img_enc = load_image(context, comm.recv_img())
+
+    base_img = ([[int(col.decrypt(secret_key)[0]) for col in row] for row in base_img_enc])
+    plt.imshow(base_img)
+    plt.show()
+
+    # Get number of octaves
+    comm.call_api(b'compute_num_octaves')
+
+    comm.send_bytes(dill.dumps((len(base_img), len(base_img[0]))))
+
+    comm.check_interactive(cmp=cmp, refresh=refresh)
+
+    num_octaves = dill.loads(comm.receive_bytes())  
+
+    print("Number of octaves: ", num_octaves)  
+
+    # Get gaussian kernels
+    comm.call_api(b'generate_gaussian_kernels')
+
+    comm.send_bytes(dill.dumps(sigma))
+    comm.send_bytes(dill.dumps(num_intervals))
+
+    comm.check_interactive(cmp=cmp, refresh=refresh)
+
+    gaussian_kernel_sizes = dill.loads(comm.receive_bytes())  
+    
+    print("Gaussian kernel sizes: ", (gaussian_kernel_sizes))  
+
+    # Get Gaussian pyramid
+    comm.call_api(b'generate_gaussian_images')
+
+    comm.send_bytes(dill.dumps(serialize_image(context, base_img_enc)))
+    comm.send_bytes(dill.dumps(num_octaves))
+    comm.send_bytes(dill.dumps(gaussian_kernel_sizes))
+
+    comm.check_interactive(cmp=cmp, refresh=refresh)
+
+    gaussian_pyramid_enc = load_pyramid(context, dill.loads(comm.receive_bytes()))  
+    
+    # print("Gaussian kernel sizes: ", (gaussian_kernel_sizes)) 
+    show_pyramid(gaussian_pyramid_enc, secret_key)
 
     comm.call_api(b'exit')
 

src/comm.py

@@ -98,4 +98,24 @@ def recv_img(self):
             for _ in range(n_cols):
                 recv_row.append(self.receive_bytes())
             img.append(recv_row)
-        return img
+        return img
+    
+    
+    def check_interactive(self, cmp=None, refresh=None):
+        while True:
+            api_name = self.receive_bytes().decode()
+            # print(api_name)
+            if api_name == 'end_interaction':
+                break
+            elif api_name == 'cmp':
+                x = self.receive_bytes().decode()
+                a = self.receive_bytes().decode()
+                b = self.receive_bytes().decode()
+                self.send_bytes(cmp(x, a, b))
+            elif api_name == 'refresh':
+                x = self.receive_bytes().decode()
+                self.send_bytes(refresh(x))
+        
+
+    def end_interactive(self):
+        self.send_bytes(b'end_interaction')

src/secsift/grep

@@ -0,0 +1,86 @@
+Traceback (most recent call last):
+  File "defer.py", line 130, in <module>
+    serialized_num = dill.dumps(d)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 288, in dumps
+    dump(obj, file, protocol, byref, fmode, recurse, **kwds)#, strictio)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 260, in dump
+    Pickler(file, protocol, **_kwds).dump(obj)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 428, in dump
+    StockPickler.dump(self, obj)
+  File "/usr/lib/python3.8/pickle.py", line 487, in dump
+    self.save(obj)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 603, in save
+    self.save_reduce(obj=obj, *rv)
+  File "/usr/lib/python3.8/pickle.py", line 717, in save_reduce
+    save(state)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 560, in save
+    f(self, obj)  # Call unbound method with explicit self
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 1233, in save_module_dict
+    StockPickler.save_dict(pickler, obj)
+  File "/usr/lib/python3.8/pickle.py", line 971, in save_dict
+    self._batch_setitems(obj.items())
+  File "/usr/lib/python3.8/pickle.py", line 997, in _batch_setitems
+    save(v)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 560, in save
+    f(self, obj)  # Call unbound method with explicit self
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 2013, in save_function
+    _save_with_postproc(pickler, (_create_function, (
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 1133, in _save_with_postproc
+    pickler.save_reduce(*reduction)
+  File "/usr/lib/python3.8/pickle.py", line 692, in save_reduce
+    save(args)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 560, in save
+    f(self, obj)  # Call unbound method with explicit self
+  File "/usr/lib/python3.8/pickle.py", line 886, in save_tuple
+    save(element)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 603, in save
+    self.save_reduce(obj=obj, *rv)
+  File "/usr/lib/python3.8/pickle.py", line 717, in save_reduce
+    save(state)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 560, in save
+    f(self, obj)  # Call unbound method with explicit self
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 1233, in save_module_dict
+    StockPickler.save_dict(pickler, obj)
+  File "/usr/lib/python3.8/pickle.py", line 971, in save_dict
+    self._batch_setitems(obj.items())
+  File "/usr/lib/python3.8/pickle.py", line 997, in _batch_setitems
+    save(v)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 560, in save
+    f(self, obj)  # Call unbound method with explicit self
+  File "/usr/lib/python3.8/pickle.py", line 886, in save_tuple
+    save(element)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 603, in save
+    self.save_reduce(obj=obj, *rv)
+  File "/usr/lib/python3.8/pickle.py", line 717, in save_reduce
+    save(state)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 560, in save
+    f(self, obj)  # Call unbound method with explicit self
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 1233, in save_module_dict
+    StockPickler.save_dict(pickler, obj)
+  File "/usr/lib/python3.8/pickle.py", line 971, in save_dict
+    self._batch_setitems(obj.items())
+  File "/usr/lib/python3.8/pickle.py", line 1002, in _batch_setitems
+    save(v)
+  File "/home/bhargav/.local/lib/python3.8/site-packages/dill/_dill.py", line 422, in save
+    StockPickler.save(self, obj, save_persistent_id)
+  File "/usr/lib/python3.8/pickle.py", line 578, in save
+    rv = reduce(self.proto)
+TypeError: cannot pickle '_tenseal_cpp.CKKSVector' object

src/secsift/secPysift.py

@@ -1,10 +1,10 @@
-from primitives import *
-from primitives import *
+from secsift.primitives import *
+# from primitives import *
 from numpy import sqrt, log
 from tqdm import tqdm
 import numpy as np
 
-# DEBUG=1
+DEBUG=0
 def secGenerateBaseImage(image, sigma, assumed_blur, kernel_size=3):
     """
         Only image is encrypted

src/secsift/test.ipynb

@@ -5511,7 +5511,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "precog_know_inf",
+   "display_name": "3.9.7",
    "language": "python",
    "name": "python3"
   },
@@ -5525,7 +5525,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.7"
+   "version": "3.9.7"
   }
  },
  "nbformat": 4,