database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[fix](encrypt) wrong mode arg of encrypt and decrypt function make BE crash (#40726) (#40868)

Browse Source 
pick #40726 to branch-2.1
This commit is contained in:
camby
2024-09-15 21:31:00 +08:00
committed by GitHub
parent 0d3b9c3f18
commit f3b1f1c19b
3 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
be/src/vec/functions/function_encryption.cpp
View File

@ -244,8 +244,9 @@ struct EncryptionAndDecryptTwoImpl {
if (mode_arg.size != 0) {
if (!aes_mode_map.contains(mode_str)) {
all_insert_null = true;
} else {
encryption_mode = aes_mode_map.at(mode_str);
}
encryption_mode = aes_mode_map.at(mode_str);
}
const ColumnString::Offsets* offsets_column = &column->get_offsets();
const ColumnString::Chars* chars_column = &column->get_chars();
@ -371,13 +372,15 @@ struct EncryptionAndDecryptMultiImpl {
if constexpr (is_sm_mode) {
if (sm4_mode_map.count(mode_str) == 0) {
all_insert_null = true;
} else {
encryption_mode = sm4_mode_map.at(mode_str);
}
encryption_mode = sm4_mode_map.at(mode_str);
} else {
if (aes_mode_map.count(mode_str) == 0) {
all_insert_null = true;
} else {
encryption_mode = aes_mode_map.at(mode_str);
}
encryption_mode = aes_mode_map.at(mode_str);
}
}