tqcq/sled
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix base64 crash
All checks were successful
linux-x64-gcc / linux-gcc (Debug) (push) Successful in 1m42s
Details
linux-x64-gcc / linux-gcc (Release) (push) Successful in 1m39s
Details
linux-mips64-gcc / linux-gcc-mips64el (push) Successful in 3m0s
Details

Browse Source
This commit is contained in:
tqcq 2024-03-29 17:03:45 +08:00
parent 69d6b80638
commit f3555ec520
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/strings/base64.cc
View File

@ -40,10 +40,11 @@ Base64::Encode(const uint8_t *ptr, size_t len)
{ {
// std::stringstream ss; // std::stringstream ss;
std::string result(EncodedLength(len), 0); auto encoded_length = EncodedLength(len);
std::string result(encoded_length, 0);
int write_idx = 0; int write_idx = 0;
int value = 0; int value = 0;
int value_bits = 0; int value_bits = 0;
while (len > 0) { while (len > 0) {
value = (value << 8) + *ptr; value = (value << 8) + *ptr;
@ -67,7 +68,7 @@ Base64::Encode(const uint8_t *ptr, size_t len)
// ss << kBase64Chars[((value << 8) >> (value_bits + 2)) & 0x3F]; // ss << kBase64Chars[((value << 8) >> (value_bits + 2)) & 0x3F];
} }
// while (ss.str().size() % 4) { ss << '='; } // while (ss.str().size() % 4) { ss << '='; }
while (write_idx % 4) { result[write_idx++] = '='; } while (write_idx < encoded_length) { result[write_idx++] = '='; }
// return ss.str(); // return ss.str();
return std::move(result); return std::move(result);
@ -84,9 +85,9 @@ Base64::Decode(const uint8_t *ptr, size_t len)
int write_idx = 0; int write_idx = 0;
std::string data(DecodedLength((char *) ptr, len), 0); std::string data(DecodedLength((char *) ptr, len), 0);
// std::stringstream ss; // std::stringstream ss;
int value = 0; int value = 0;
int value_bits = 0; int value_bits = 0;
int index = 0; int index = 0;
for (int i = 0; i < len; i++) { for (int i = 0; i < len; i++) {
char c = ptr[i]; char c = ptr[i];
if (-1 != kInvBase64Chars[c]) { if (-1 != kInvBase64Chars[c]) {