Merge pull request #4 from wfus/homo

Homo

Author: raylin1000

homo/Makefile

@@ -7,7 +7,7 @@ LDFLAGS=-g
 
 .PHONY : all clean
 
-all : raymond dct encode test
+all : raymond dct encode test sender
 
 
 raymond : raymond.cpp
@@ -25,5 +25,10 @@ test: test.cpp
 	@-mkdir -p $(dir $@)
 	$(CXX) $(CXXFLAGS) $(LDFLAGS) test.cpp $(addprefix -I,$(INCLUDE_DIR)) $(addprefix -L,$(LIB_DIR)) -lseal -o $(BIN_DIR)/test
 
+sender: test.cpp
+	@-mkdir -p $(dir $@)
+	$(CXX) $(CXXFLAGS) $(LDFLAGS) sender.cpp $(addprefix -I,$(INCLUDE_DIR)) $(addprefix -L,$(LIB_DIR)) -lseal -o $(BIN_DIR)/sender
+
+
 clean :
 	@-rm -f ../bin/* 

homo/fhe_image.h

@@ -16,6 +16,7 @@
 
 
 #include "seal/seal.h"
+
 using namespace seal;
 
 
@@ -36,6 +37,14 @@ std::vector<std::vector<double>> split_image_eight_block(std::vector<double> im,
     return lst;
 }
 
+void print_image(uint8_t *im, int w, int h) {
+    std::cout << "Printing Image dim: (" << w << "," << h << ")" << std::endl;
+    for (int i = 0; i < w*h; i++) {
+        printf("%.2x ", im[i]);
+        if ((i + 1) % w == 0) std::cout << std::endl;
+    }
+} 
+
 void print_image(std::vector<double> &im, int w, int h) {
     std::cout << "Printing Image dim: (" << w << "," << h << ")" << std::endl;
     for (int i = 0; i < im.size(); i++) {
@@ -74,6 +83,24 @@ std::vector<double> read_image(std::string fname) {
     return im; 
 }
 
+std::vector<double> read_image(std::string fname, int* w, int *h) {
+
+    std::vector<double> im; 
+    std::ifstream myfile;
+        
+    myfile.open(fname.c_str());
+    myfile >> *w;
+    myfile >> *h;
+    std::cout << "Read in " << fname << "with dimensions: " << w << " x " << h << std::endl;
+
+    float tmp;
+    for (int i = 0; i < (*w) * (*h); i++) {
+        myfile >> tmp;
+        im.push_back(tmp);
+    }   
+    return im; 
+}
+
 
 /* Recieves a 8x8 box of pixels, in ciphertext form. 
  * Data should be laid out in a 64 element vector with 
@@ -280,6 +307,21 @@ void dct_blocks(std::vector<std::vector<double>> &blocks) {
     for (int a = 0; a < blocks.size(); a++) {
         dct(&blocks[a][0]);
     }
+}
+
+void print_parameters(const SEALContext &context) {
+    std::cout << "/ Encryption parameters:" << std::endl;
+    std::cout << "| poly_modulus: " << context.poly_modulus().to_string() << std::endl;
+
+    /*
+    Print the size of the true (product) coefficient modulus
+    */
+    std::cout << "| coeff_modulus size: " 
+        << context.total_coeff_modulus().significant_bit_count() << " bits" << std::endl;
+
+    std::cout << "| plain_modulus: " << context.plain_modulus().value() << std::endl;
+    std::cout << "\\ noise_standard_deviation: " << context.noise_standard_deviation() << std::endl;
+    std::cout << std::endl;
 } 
 
 #endif

homo/raymond.cpp

@@ -25,22 +25,6 @@ const std::vector<double> S_STD_LUM_QUANT = { 16,11,12,14,12,10,16,14,13,14,18,1
 const std::vector<double> S_STD_CROMA_QUANT = { 17,18,18,24,21,24,47,26,26,47,99,66,56,66,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99 };
 
 void raymond_average();
-std::vector<double> read_image(std::string fname);
-
-void print_parameters(const SEALContext &context) {
-    std::cout << "/ Encryption parameters:" << std::endl;
-    std::cout << "| poly_modulus: " << context.poly_modulus().to_string() << std::endl;
-
-    /*
-    Print the size of the true (product) coefficient modulus
-    */
-    std::cout << "| coeff_modulus size: " 
-        << context.total_coeff_modulus().significant_bit_count() << " bits" << std::endl;
-
-    std::cout << "| plain_modulus: " << context.plain_modulus().value() << std::endl;
-    std::cout << "\\ noise_standard_deviation: " << context.noise_standard_deviation() << std::endl;
-    std::cout << std::endl;
-}
 
 
 int main()
@@ -184,4 +168,4 @@ void raymond_average() {
     }
 
     return;
-}
+}

homo/sender.cpp

@@ -0,0 +1,70 @@
+#include "seal/seal.h"
+#include "fhe_image.h"
+#include "jpge.h"
+#include "stb_image.c"
+
+
+using namespace seal;
+
+auto start = std::chrono::steady_clock::now();
+const bool VERBOSE = true;
+
+std::vector<double> read_image(std::string fname);
+
+int main()
+{
+    const char* test_filename = "../image/kung.jpg";
+    const int requested_composition = 3;
+    int width = 0, height = 0, actual_composition = 0;
+    uint8_t *image_data = stbi_load(test_filename, &width, &height, &actual_composition, requested_composition);
+    std::cout << width << " x " << height << std::endl;
+    print_image(image_data,  width, height);
+
+
+    // Encryption Parameters
+    EncryptionParameters params;
+    params.set_poly_modulus("1x^32768 + 1");
+    params.set_coeff_modulus(coeff_modulus_128(2048));
+    params.set_plain_modulus(1 << 14);
+    SEALContext context(params);
+    print_parameters(context);
+
+
+    // Generate keys
+    start = std::chrono::steady_clock::now(); 
+    KeyGenerator keygen(context);
+    auto public_key = keygen.public_key();
+    auto secret_key = keygen.secret_key();
+    auto diff = std::chrono::steady_clock::now() - start; 
+    std::cout << "KeyGen: ";
+    std::cout << chrono::duration<double, milli>(diff).count() << " ms" << std::endl;
+    
+    // Encrytor and decryptor setup
+    Encryptor encryptor(context, public_key);
+    Evaluator evaluator(context);
+    Decryptor decryptor(context, secret_key);
+
+    // Base + Number of coefficients used for encoding past the decimal point (both pos and neg)
+    // Example: if poly_base = 11, and N_FRACTIONAL_COEFFS=3, then we will have 
+    // a1 * 11^-1 + a2 * 11^-2 + a3 * 11^-3
+    const int POLY_BASE = 11;
+    const int N_FRACTIONAL_COEFFS = 3;  
+    const int N_NUMBER_COEFFS = 10;
+
+    FractionalEncoder encoder(context.plain_modulus(), context.poly_modulus(), N_NUMBER_COEFFS, N_FRACTIONAL_COEFFS, POLY_BASE);
+    
+    std::ofstream myfile;
+    myfile.open("../image/nothingpersonnel.txt");
+    start = std::chrono::steady_clock::now(); 
+    for (int i = 0; i < width * height; i++) {
+        Ciphertext c;
+        double conv = (double)(image_data[i]);
+        std::cout<< conv <<std::endl;
+        encryptor.encrypt(encoder.encode(conv), c);
+        c.save(myfile);
+    }
+
+
+    myfile.close();
+    return 0;
+}