Homec4science
Diffusion SecGene-i2b2 (master) Browse

SecGene/crypto/edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge.cpp
2d922360738bmaster

Last Change
Raw File

edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge.cpp
View Options

#include "edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge.h"
#include <nfl.hpp>
// Polynomial degree
#define POLY_DEGREE 2048
// Parameter initialization
// Brought from Test_poly_encrypt.cpp
using poly_t = nfl::poly_from_modulus<uint64_t, POLY_DEGREE, 80>;
jclass java_util_List;
jmethodID java_util_List_get;
jmethodID java_util_List_set;
bool initialized = false;
void init(JNIEnv *env) {
initialized = true;
java_util_List = env->FindClass("java/util/List");
java_util_List_set = env->GetMethodID(java_util_List, "set", "(ILjava/lang/Object;)Ljava/lang/Object;");
java_util_List_get = env->GetMethodID(java_util_List, "get", "(I)Ljava/lang/Object;");
}
JNIEXPORT void JNICALL Java_edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge_multiply___3J_3J(JNIEnv *env, jobject job, jlongArray p1_array, jlongArray p2_array) {
jlong* p1_elements;
jlong* p2_elements;
poly_t p1;
poly_t p2;
p2_elements = env->GetLongArrayElements(p2_array, NULL);
for (size_t i=0; i<POLY_DEGREE; i++) {
(p2)(0,i)=(uint64_t) p2_elements[i];
}
env->ReleaseLongArrayElements(p2_array, p2_elements, JNI_ABORT);
p1_elements = env->GetLongArrayElements(p1_array, NULL);
for (size_t i=0; i<POLY_DEGREE; i++) {
(p1)(0,i)=(uint64_t) p1_elements[i];
}
poly_t product = p1 * p2;
product.invntt_pow_invphi();
for (size_t i=0; i<POLY_DEGREE; i++) {
p1_elements[i] = (jlong) product(0,i);
}
env->ReleaseLongArrayElements(p1_array, p1_elements, 0);
}
JNIEXPORT void JNICALL Java_edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge_multiply__Ljava_util_List_2_3J(JNIEnv *env, jobject job, jobject p1_object, jlongArray p2_array) {
jlong* p2_elements;
poly_t p1;
poly_t p2;
if (!initialized)
init(env);
p2_elements = env->GetLongArrayElements(p2_array, NULL);
for (size_t i=0; i<POLY_DEGREE; i++) {
p1(0,i) = (uint64_t) env->CallObjectMethod(p1_object, java_util_List_get, i);
std::cout<< p1(0,i) << std::endl;
p2(0,i) = (uint64_t) p2_elements[i];
}
env->ReleaseLongArrayElements(p2_array, p2_elements, JNI_ABORT);
poly_t product = p1 * p2;
product.invntt_pow_invphi();
for (size_t i=0; i<POLY_DEGREE; i++) {
env->CallObjectMethod(p1_object, java_util_List_set, i, (jlong*) &product(0,i));
}
}
JNIEXPORT jlongArray JNICALL Java_edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge_getModuliFromNative(JNIEnv *env, jobject job) {
jlongArray moduli_array;
jlong* moduli_elements;
moduli_array = env->NewLongArray(poly_t::nmoduli);
moduli_elements = env->GetLongArrayElements(moduli_array, NULL);
for (size_t i=0; i<poly_t::nmoduli; i++) {
moduli_elements[i] = (jlong) poly_t::get_modulus(i);
}
env->ReleaseLongArrayElements(moduli_array, moduli_elements, 0);
return moduli_array;
}
JNIEXPORT jint JNICALL Java_edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge_getNbModuliFromNative(JNIEnv *env, jobject job) {
return (jint) poly_t::nmoduli;
}
JNIEXPORT jint JNICALL Java_edu_harvard_i2b2_gen_util_PolynomialMultiplicationBridge_getPolyDegreeFromNative(JNIEnv *env, jobject job) {
return (jint) poly_t::degree;
}
c4science · Help