init
This commit is contained in:
blue-lemon0104
446
src/crypto/cryptolib/libudf/decider/udfDispatch.cpp
Executable file
446
src/crypto/cryptolib/libudf/decider/udfDispatch.cpp
Executable file
@@ -0,0 +1,446 @@
|
||||
// 分流：是否是浮点数->TEE，纯软件
|
||||
// 不使用ECALL_BATCH,不进行TaskQueue任务批处理
|
||||
// 待处理：判断浮点数逻辑
|
||||
#include <memory>
|
||||
#include "Singleton.h"
|
||||
#include "Timer.h"
|
||||
#include "errmsgBridge.h"
|
||||
#include "TimeCounter.h"
|
||||
#include "LogBase.h"
|
||||
#include <iostream>
|
||||
// 新增
|
||||
#include <fstream>
|
||||
extern "C"
|
||||
{
|
||||
#include "Interface.h"
|
||||
#include "udfDispatch.h"
|
||||
}
|
||||
using namespace util;
|
||||
#ifdef USE_SGX
|
||||
#include "Enclave_u.h"
|
||||
#endif
|
||||
|
||||
#include "App.h"
|
||||
|
||||
#ifndef NDEBUG // DEBUG
|
||||
|
||||
#include <fstream>
|
||||
|
||||
extern "C"
|
||||
{
|
||||
#include <unistd.h>
|
||||
}
|
||||
using namespace std;
|
||||
|
||||
class CalledCounter {
|
||||
public:
|
||||
CalledCounter() {
|
||||
char buf[64];
|
||||
sprintf(buf, "/etc/encryptsql/UDFcalled_by%d.count", (int) getpid());
|
||||
|
||||
ofstream fout(buf);
|
||||
fout.close();
|
||||
}
|
||||
};
|
||||
|
||||
CalledCounter c;
|
||||
#endif
|
||||
|
||||
extern "C" int ore_compare(char *strctxt1, char *strctxt2);
|
||||
|
||||
constexpr char TERMCHAR = '^';
|
||||
// 新增
|
||||
char plaPath[] = "/etc/encryptsql/masterKey.plain";
|
||||
char encPath[] = "/etc/encryptsql/masterKey.encrypted";
|
||||
char dekPath[] = "/etc/encryptsql/dek.sealed";
|
||||
constexpr int keySize = 32;
|
||||
|
||||
#ifdef USE_SGX
|
||||
void encryptKey(sgx_enclave_id_t eid) {
|
||||
using namespace std;
|
||||
ifstream fin;
|
||||
fin.open(plaPath, ios::binary | ios::in);
|
||||
if(!fin.good()) {
|
||||
printf("can not open key file:%s\n", plaPath);
|
||||
exit(0);
|
||||
}
|
||||
char key[keySize];
|
||||
|
||||
fin.get(key, keySize + 1);
|
||||
fin.close();
|
||||
ecall_encryptKeyWithEnclave(eid, key, keySize, encPath);
|
||||
}
|
||||
#endif
|
||||
|
||||
char *saheAdd(char *cipher1, char *cipher2, char *aescipher1, char *aescipher2, bool isLeftFloat, bool isRightFloat) {
|
||||
try {
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
size_t csz2 = strlen(cipher2) + 1;
|
||||
Log("saheAdd: cipher1=%s", aescipher1, log::info);
|
||||
Log("saheAdd: cipher2=%s", aescipher2, log::info);
|
||||
Log("saheAdd: isLeftFloat=%d", isLeftFloat, log::info);
|
||||
Log("saheAdd: isRightFloat=%d", isRightFloat, log::info);
|
||||
if (!(cipher1[csz1 - 2] == TERMCHAR && cipher2[csz2 - 2] == TERMCHAR)) {
|
||||
Log("saheAdd: cannot find term char in cipher1(%s) or cipher2(%s)!", cipher1, cipher2, log::error);
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
bool isFloat = isLeftFloat || isRightFloat;
|
||||
#ifdef USE_SGX
|
||||
if (isFloat && ecall_sahe_add) {// 判断浮点数
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
sgx_enclave_id_t eid = enclave->getID();
|
||||
|
||||
// Log("saheAdd: Using enclave ID: %llu", eid, log::info);
|
||||
|
||||
size_t retsz = 128;
|
||||
ret = new char[retsz];
|
||||
if (strcmp(aescipher1, "null") == 0) {
|
||||
Log("saheAdd: client does not support SGX", log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
// 新增
|
||||
encryptKey(eid);
|
||||
// Log("saheAdd: Calling ecall_sahe_add", log::info);
|
||||
int isLeftFloatInt = isLeftFloat ? 1 : 0;
|
||||
int isRightFloatInt = isRightFloat ? 1 : 0;
|
||||
sgx_status_t status = ecall_sahe_add(eid, aescipher1, aescipher2, ret, retsz, isLeftFloatInt, isRightFloatInt);
|
||||
if (status != SGX_SUCCESS) {
|
||||
Log("saheAdd: ecall_sahe_add failed with status %d", status, log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
Log("saheAdd: ecall_sahe_add succeeded", log::info);
|
||||
} else {
|
||||
Log("saheAdd: Using software implementation", log::info);
|
||||
ret = SAHE_add_udf(cipher1, cipher2, csz1, csz2);
|
||||
}
|
||||
#else
|
||||
Log("saheAdd: Using software implementation", log::info);
|
||||
ret = SAHE_add_udf(cipher1, cipher2, csz1, csz2);
|
||||
#endif
|
||||
if (ret) {
|
||||
Log("saheAdd: Operation successful", log::info);
|
||||
} else {
|
||||
Log("saheAdd: Operation failed - null result", log::error);
|
||||
}
|
||||
return ret;
|
||||
} catch (const std::exception& e) {
|
||||
Log("saheAdd: Exception caught: %s", e.what(), log::error);
|
||||
return nullptr;
|
||||
} catch (...) {
|
||||
Log("saheAdd: Unknown exception caught", log::error);
|
||||
return nullptr;
|
||||
}
|
||||
}
|
||||
|
||||
char *saheSub(char *cipher1, char *cipher2, char *aescipher1, char *aescipher2, bool isLeftFloat, bool isRightFloat) {
|
||||
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
size_t csz2 = strlen(cipher2) + 1;
|
||||
Log("saheSub: input lengths - csz1=%zu, csz2=%zu", csz1, csz2, log::info);
|
||||
Log("saheSub: cipher1=%s", cipher1, log::info);
|
||||
Log("saheSub: cipher2=%s", cipher2, log::info);
|
||||
Log("saheSub: isLeftFloat=%d", isLeftFloat, log::info);
|
||||
Log("saheSub: isRightFloat=%d", isRightFloat, log::info);
|
||||
|
||||
if (!(cipher1[csz1 - 2] == TERMCHAR && cipher2[csz2 - 2] == TERMCHAR)) {
|
||||
Log("saheSub: cannot find term char in cipher1(%s) or cipher2(%s)!", cipher1, cipher2, log::error);
|
||||
abort();
|
||||
}
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
char *ret = nullptr;
|
||||
|
||||
Timer t = Timer();
|
||||
bool isFloat = isLeftFloat || isRightFloat;
|
||||
#ifdef USE_SGX
|
||||
if (isFloat && ecall_sahe_sub) {// 判断浮点数
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
sgx_enclave_id_t eid = enclave->getID();
|
||||
size_t retsz = 128;
|
||||
ret = new char[retsz];
|
||||
if (strcmp(aescipher1, "null") == 0) {
|
||||
Log("client does not support SGX while Server try to calculate AHE_SUB with SGX!\n", log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
encryptKey(eid);
|
||||
int isLeftFloatInt = isLeftFloat ? 1 : 0;
|
||||
int isRightFloatInt = isRightFloat ? 1 : 0;
|
||||
sgx_status_t status = ecall_sahe_sub(eid, aescipher1, aescipher2, ret, retsz, isLeftFloatInt, isRightFloatInt);
|
||||
if (status != SGX_SUCCESS) {
|
||||
Log("saheSub: ecall_sahe_sub failed with status %d", status, log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
Log("saheSub: ecall_sahe_sub succeeded", log::info);
|
||||
} else {
|
||||
Log("saheSub: Using software implementation", log::info);
|
||||
ret = SAHE_substract_udf(cipher1, cipher2, csz1, csz2);
|
||||
}
|
||||
#else
|
||||
Log("saheSub: Using software implementation", log::info);
|
||||
ret = SAHE_substract_udf(cipher1, cipher2, csz1, csz2);
|
||||
#endif
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *saheMul(char *cipher1, int m) {
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
if (strcmp(cipher1, "null") == 0) {
|
||||
#if !defined(USE_SGX) && !defined(SGX_ORE) // 前两个正常ORE为空，说明客户端有SGX_ORE，此时服务器没有SGX_ORE的话无法解密。
|
||||
Log("client use SGX to calculate MHE but server does not support SGX!\n", log::error);
|
||||
#endif
|
||||
}
|
||||
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
if (false && ecall_sahe_mul) { //判断浮点数
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
ret = new char[csz1];
|
||||
ecall_sahe_mul(enclave->getID(), cipher1, m, ret, csz1);
|
||||
} else
|
||||
ret = SAHE_multiply_udf(cipher1, m, csz1);
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *saheNeg(char *cipher1) {
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
if (false && ecall_sahe_neg) {//判断浮点数
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
ret = new char[csz1];
|
||||
ecall_sahe_neg(enclave->getID(), cipher1, ret, csz1);
|
||||
} else
|
||||
ret = SAHE_neggate_udf(cipher1, csz1);
|
||||
return ret;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
}
|
||||
|
||||
char *smheMul(char *cipher1, char *cipher2, char *aescipher1, char *aescipher2, bool isLeftFloat, bool isRightFloat) {
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
size_t csz2 = strlen(cipher2) + 1;
|
||||
Log("smheMul: input lengths - csz1=%zu, csz2=%zu", csz1, csz2, log::info);
|
||||
Log("smheMul: cipher1=%s", cipher1, log::info);
|
||||
Log("smheMul: cipher2=%s", cipher2, log::info);
|
||||
if (!(cipher1[csz1 - 2] == TERMCHAR && cipher2[csz2 - 2] == TERMCHAR)) {
|
||||
Log("smheMul: cannot find term char in cipher1(%s) or cipher2(%s)!", cipher1, cipher2, log::error);
|
||||
abort();
|
||||
}
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
bool HardWare = false;
|
||||
#ifdef SGX_MHE
|
||||
// 乘除用TEE
|
||||
HardWare = true;
|
||||
#endif
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
#ifdef USE_SGX
|
||||
if (HardWare && ecall_smhe_mul) {
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
sgx_enclave_id_t eid = enclave->getID();
|
||||
size_t retsz = 128;
|
||||
ret = new char[retsz];
|
||||
if (strcmp(aescipher1, "null") == 0) {
|
||||
Log("client does not support SGX while Server try to calculate MHE with SGX!\n", log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
encryptKey(eid);
|
||||
int isLeftFloatInt = isLeftFloat ? 1 : 0;
|
||||
int isRightFloatInt = isRightFloat ? 1 : 0;
|
||||
sgx_status_t status = ecall_smhe_mul(eid, aescipher1, aescipher2, ret, retsz, isLeftFloatInt, isRightFloatInt);
|
||||
if (status != SGX_SUCCESS) {
|
||||
Log("smheMul: ecall_smhe_mul failed with status %d", status, log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
Log("smheMul: ecall_smhe_mul succeeded", log::info);
|
||||
} else {
|
||||
// char *__ = new char[csz1];
|
||||
Log("smheMul: Using software implementation", log::info);
|
||||
ret = SMHE_MULTIPLY_udf(cipher1, cipher2, csz1, csz2);
|
||||
}
|
||||
#else
|
||||
Log("smheMul: Using software implementation", log::info);
|
||||
ret = SMHE_MULTIPLY_udf(cipher1, cipher2, csz1, csz2);
|
||||
#endif
|
||||
return ret;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
}
|
||||
|
||||
char *smheMulp(char *cipher1, int m) {
|
||||
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
if (false && ecall_smhe_mulp) {
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
ret = new char[csz1];
|
||||
|
||||
ecall_smhe_mulp(enclave->getID(), cipher1, m, ret, csz1);
|
||||
} else
|
||||
ret = SMHE_MULTIPLYPLAIN_udf(cipher1, m, csz1);
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *smheDiv(char *cipher1, char *cipher2, char *aescipher1, char *aescipher2, bool isLeftFloat, bool isRightFloat) {
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
size_t csz2 = strlen(cipher2) + 1;
|
||||
Log("smheDiv: input lengths - csz1=%zu, csz2=%zu", csz1, csz2, log::info);
|
||||
Log("smheDiv: cipher1=%s", cipher1, log::info);
|
||||
Log("smheDiv: cipher2=%s", cipher2, log::info);
|
||||
if (!(cipher1[csz1 - 2] == TERMCHAR && cipher2[csz2 - 2] == TERMCHAR)) {
|
||||
Log("smheDiv: cannot find term char in cipher1(%s) or cipher2(%s)!", cipher1, cipher2, log::error);
|
||||
// abort();
|
||||
}
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
bool HardWare = false;
|
||||
#ifdef SGX_MHE
|
||||
HardWare = true;
|
||||
#endif
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
#ifdef USE_SGX
|
||||
if (HardWare && ecall_smhe_div) {
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
sgx_enclave_id_t eid = enclave->getID();
|
||||
size_t retsz = 128;
|
||||
ret = new char[retsz];
|
||||
if (strcmp(aescipher1, "null") == 0) {
|
||||
Log("client does not support SGX while Server try to calculate MHE with SGX!\n", log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
encryptKey(eid);
|
||||
int isLeftFloatInt = isLeftFloat ? 1 : 0;
|
||||
int isRightFloatInt = isRightFloat ? 1 : 0;
|
||||
sgx_status_t status = ecall_smhe_div(eid, aescipher1, aescipher2, ret, retsz, isLeftFloatInt, isRightFloatInt);
|
||||
if (status != SGX_SUCCESS) {
|
||||
Log("smheDiv: ecall_smhe_div failed with status %d", status, log::error);
|
||||
delete[] ret;
|
||||
return nullptr;
|
||||
}
|
||||
Log("smheDiv: ecall_smhe_div succeeded", log::info);
|
||||
} else {
|
||||
Log("smheDiv: Using software implementation", log::info);
|
||||
ret = SMHE_DIVIDE_udf(cipher1, cipher2, csz1, csz2);
|
||||
}
|
||||
#else
|
||||
Log("smheDiv: Using software implementation", log::info);
|
||||
ret = SMHE_DIVIDE_udf(cipher1, cipher2, csz1, csz2);
|
||||
#endif
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *smhePow(char *cipher1, char *aescipher1, int m) {
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
if (false && ecall_smhe_pow) {
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
ret = new char[csz1 * 2];
|
||||
ecall_smhe_pow(enclave->getID(), aescipher1, m, ret, csz1);
|
||||
} else
|
||||
ret = SMHE_POW_udf(cipher1, m, csz1);
|
||||
return ret;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
}
|
||||
|
||||
char *smheInv(char *cipher1) {
|
||||
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
char *ret = nullptr;
|
||||
Timer t = Timer();
|
||||
if (false && ecall_smhe_inv) {
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
char *ret = new char[csz1];
|
||||
ecall_smhe_inv(enclave->getID(), cipher1, ret, csz1);
|
||||
} else
|
||||
ret = SMHE_INVERSE_udf(cipher1, csz1);
|
||||
return ret;
|
||||
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
}
|
||||
|
||||
char *saheAddp(char *cipher1, int m) {
|
||||
size_t sz1 = strlen(cipher1);
|
||||
return SAHE_addplaintext_udf(cipher1, m, sz1);
|
||||
}
|
||||
|
||||
string getOREType(int kind) {
|
||||
switch (kind) {
|
||||
case 1:
|
||||
return "LT";
|
||||
case 2:
|
||||
return "LE";
|
||||
case 3:
|
||||
return "GT";
|
||||
case 4:
|
||||
return "GE";
|
||||
case 5:
|
||||
return "EE";
|
||||
case 6:
|
||||
return "MAX";
|
||||
case 7:
|
||||
return "MIN";
|
||||
case 8:
|
||||
return "NE";
|
||||
return "UN";
|
||||
}
|
||||
}
|
||||
|
||||
int oreCompare(char *cipher1, char *cipher2, char *aescipher1, char *aescipher2, int kind) {
|
||||
size_t csz1 = strlen(cipher1) + 1;
|
||||
size_t csz2 = strlen(cipher2) + 1;
|
||||
|
||||
if (strcmp(cipher1, "null") == 0) {
|
||||
#if !defined(USE_SGX) && !defined(SGX_ORE) // 前两个正常ORE为空，说明客户端有SGX_ORE，此时服务器没有SGX_ORE的话无法解密。
|
||||
Log("client use SGX to calculate ORE but server does not support SGX!\n", log::error);
|
||||
#endif
|
||||
}
|
||||
|
||||
//!ORE
|
||||
int res = -2;
|
||||
bool hardWare = false;
|
||||
#ifdef SGX_ORE
|
||||
// ORE用TEE
|
||||
hardWare = true;
|
||||
#endif
|
||||
|
||||
int ret = -3;
|
||||
Timer t = Timer();
|
||||
if (hardWare && ecall_ore_compare) {
|
||||
if (strcmp(aescipher1, "null") == 0)
|
||||
Log("client does not support SGX while Server try to calculate ORE with SGX!\n", log::error);
|
||||
|
||||
EnclaveManager *enclave = EnclaveManager::getInstance();
|
||||
ecall_ore_compare(enclave->getID(), aescipher1, aescipher2, &ret);
|
||||
} else {
|
||||
ret = ore_compare(cipher1, cipher2);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
// int ore_com(char *cipher1, char *cipher2) {
|
||||
// int ret = -3;
|
||||
// ret = ore_compare(cipher1, cipher2);
|
||||
|
||||
// return ret;
|
||||
|
||||
// }
|
||||
Reference in New Issue
Block a user