lian/encryptsql
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复decimal解析错误问题，统一以%.15g输出

Browse Source
This commit is contained in:
blue-lemon0104
2026-04-17 13:05:30 +08:00
parent 675146e477
commit 47c20574f7
4 changed files with 116 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/encryptsql/decryptres.cpp
View File

@@ -106,9 +106,9 @@ void decryptResult(int numberAttr, int numTuples, pAttrDescs pattDescs, pTuples
void (*anyTypetoString)(const char *, char *const, void *) = NULL; // 把buffer转为相应类型的数据。
switch (ctype) {
// case TYPE_FLOAT:
// anyTypetoString = &FloatTypetoString;
// break;
case TYPE_FLOAT:
anyTypetoString = &DoubleTypetoString;
break;
case TYPE_STRING:
@@ -166,14 +166,11 @@ void decryptResult(int numberAttr, int numTuples, pAttrDescs pattDescs, pTuples
(char*)dek.c_str(), // 使用DEK
(char*)aesOut.get()
);
// 确保输出大小正确
if (outSize >= sizeof(double)) {
double d = *(double*)aesOut.get();
std::ostringstream ss;
ss << d;
strcpy(tuples[i][j].value, ss.str().c_str());
}
delete[] buff;
// 确保输出大小正确
if (outSize >= sizeof(double)) {
DoubleTypetoString((const char*)aesOut.get(), tuples[i][j].value, (void *) coltypename.get());
}
delete[] buff;
} else {
size_t buffSZ = tuples[i][j].len / 2 + 1; // 密文buffer长度， 应该大于hexstr长度的一半
size_t plainBuffSZ = buffSZ;
@@ -216,7 +213,7 @@ void decryptResult(int numberAttr, int numTuples, pAttrDescs pattDescs, pTuples
// plainBuff = encryptValue(t, pbuff, buffSZ, &plainBuffSZ, false);
if (anyTypetoString) {
//fixed by qxy for RND
if (type == CIPHER_RND) {
if (type == CIPHER_RND && ctype != TYPE_FLOAT) {
long l = *(long *)plainBuff;
std::ostringstream ss;
ss << l;

87
src/encryptsql/encryptstmt.cpp
View File

@@ -57,6 +57,24 @@ bool IsInt8ColumnType(const char* type_name)
return type.find("int8") != std::string::npos || type.find("bigint") != std::string::npos;
}
bool IsFloatColumnType(const char* type_name)
{
return type_name && unifyColumnType(type_name) == TYPE_FLOAT;
}
bool ShouldSetPeerFloat(const char* type_name, EncryptInfo* info)
{
if (!IsFloatColumnType(type_name))
return false;
if (!strcmp(type_name, "float"))
return true;
if (!info || !IsA((Node *) info->father, A_Expr))
return false;
const char* op = getAExprOp((A_Expr *) info->father);
return op && !strcmp(op, "=");
}
bool TryParseInt64Literal(const char* literal, int64_t* value)
{
if (!literal || !value)
@@ -224,15 +242,29 @@ static A_Const *encryptAConst(A_Const *aconst, T_Cipher encryptCipher, EncryptIn
}
plainText = (uint8_t *) tmpInt;
in_size = sizeof(int64_t);
} else if (encryptCipher == CIPHER_RND) {
*tmpInt = intVal(AConstValue);
// isFloat = true;
// *tmpInt *= Float_Scale;
} else if (encryptCipher == CIPHER_RND || encryptCipher == CIPHER_RNDSM4CK) {
if (isFloat || info->isPeerColFloat || IsA(AConstValue, Float)) {
isFloat = true;
if (IsA(AConstValue, Integer)) {
tmpDouble = static_cast<double>(intVal(AConstValue));
} else {
tmpDouble = atof(strVal(AConstValue));
}
plainText = (uint8_t *) &tmpDouble;
in_size = sizeof(double);
} else {
*tmpInt = intVal(AConstValue);
// isFloat = true;
// *tmpInt *= Float_Scale;
plainText = (uint8_t *) tmpInt;
in_size = sizeof(int64_t);
}
if (info->isPeerColFloat) {
info->isPeerColFloat = false;
}
if (info->isPeerColInt8) {
info->isPeerColInt8 = false;
}
plainText = (uint8_t *) tmpInt;
in_size = sizeof(int64_t);
} else if (IsA(AConstValue, Float)) //
{
isFloat = true;
@@ -257,17 +289,28 @@ static A_Const *encryptAConst(A_Const *aconst, T_Cipher encryptCipher, EncryptIn
plainText = (uint8_t *) &tmpDouble;
in_size = sizeof(double);
} else if (IsA(AConstValue, Integer)) {
*tmpInt = intVal(AConstValue);
// isFloat = true;
isFloat = false;
// *tmpInt *= Float_Scale;
if (isFloat || info->isPeerColFloat) {
isFloat = true;
tmpDouble = static_cast<double>(intVal(AConstValue));
plainText = (uint8_t *) &tmpDouble;
in_size = sizeof(double);
if (info->isPeerColFloat) {
info->isPeerColFloat = false;
}
} else {
*tmpInt = intVal(AConstValue);
// isFloat = true;
isFloat = false;
// *tmpInt *= Float_Scale;
plainText = (uint8_t *) tmpInt;
in_size = sizeof(int64_t);
}
if (info->isPeerColInt8) {
info->isPeerColInt8 = false;
}
plainText = (uint8_t *) tmpInt;
in_size = sizeof(int64_t);
} else {
*tmpInt = intVal(AConstValue);
tmpDouble = static_cast<double>(*tmpInt);
// if (info->isFloatorIntCol) { // 当前列是float或Int列
// isFloat = true;
// *tmpInt *= Float_Scale;
@@ -611,14 +654,14 @@ static ColumnRef *encryptColumnRef(ColumnRef *cref, T_Cipher encryptCipher, Encr
}
cipherColNameValue = makeString((char *) cipherColName);
newCref->fields = lappend(newCref->fields, cipherColNameValue);
p = lnext(p);
char typebuf[128];
if (IsA((Node *) info->father, A_Expr)) {
getColumnType(name, typebuf);
if (!strcmp(typebuf, "float")) {
info->isPeerColFloat = true;
}
}
p = lnext(p);
char typebuf[128];
if (IsA((Node *) info->father, A_Expr)) {
getColumnType(name, typebuf);
if (ShouldSetPeerFloat(typebuf, info)) {
info->isPeerColFloat = true;
}
}
}
return newCref;
}
@@ -1581,8 +1624,8 @@ static List *encryptValuesLists(List *valuesLists, List *cols, EncryptInfo *info
// info->isFloatorIntCol = true;
// } else {
// info->isFloatorIntCol = false;
// }
if (string(colInfo->type).find("float") != string::npos)
// }
if (IsFloatColumnType(colInfo->type))
{
info->isFloatCol = true;
}

9
src/utils/utils.cpp
View File

@@ -458,6 +458,15 @@ void FloatTypetoString(const char *buf, char *v,
}
}
void DoubleTypetoString(const char *buf, char *v, void *others)
{
double d = 0.0;
memcpy(&d, buf, sizeof(double));
sprintf(v, "%.15g", d);
if (!strcmp(v, "-0"))
strcpy(v, "0");
}
void FloatTypetoString2(const char *buf, char *v,
void *others) // TODO: // buf 不一定是 double的内存排布，还有可能是long， 因为词法分析不区分 (long)1, (double)1
{

2
src/utils/utils.h
View File

@@ -153,6 +153,8 @@ void IntTypetoString(const char *buf, char *const v, void *others);
void FloatTypetoString(const char *buf, char *const v, void *others);
void DoubleTypetoString(const char *buf, char *const v, void *others);
void FloatTypetoString2(const char *buf, char *const v, void *others);
void FloatTypetoString3(const char *buf, char *const v, void *others);