Skip to content

Commit

Permalink
util/mime: skip over any invalid char
Browse files Browse the repository at this point in the history
For certain edge case handling for spaces, spaces were handled
particularly in the remainder processing functions. Make sure that now
that as per RFC 2045, util-base64 would skip over any invalid char, the
edge cases in MIME processor also be handled the same way.

This completes the work done in e46b033.

Ticket 6135
Ticket 6207
  • Loading branch information
inashivb committed Jul 13, 2023
1 parent 8548159 commit cc3677d
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 3 deletions.
15 changes: 15 additions & 0 deletions src/util-base64.c
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,21 @@ static inline int GetBase64Value(uint8_t c)
return val;
}

/**
* \brief Checks if the given char in a byte array is Base64 alphabet
*
* \param Char that needs to be checked
*
* \return True if the char was Base64 alphabet, False otherwise
*/
bool IsBase64Alphabet(uint8_t encoded_byte)
{
if (GetBase64Value(encoded_byte) < 0 && encoded_byte != '=') {
return false;
}
return true;
}

/**
* \brief Decodes a 4-byte base64-encoded block into a 3-byte ascii-encoded block
*
Expand Down
1 change: 1 addition & 0 deletions src/util-base64.h
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,7 @@ typedef enum {
/* Function prototypes */
Base64Ecode DecodeBase64(uint8_t *dest, uint32_t dest_size, const uint8_t *src, uint32_t len,
uint32_t *consumed_bytes, uint32_t *decoded_bytes, Base64Mode mode);
bool IsBase64Alphabet(uint8_t encoded_byte);

#endif

Expand Down
7 changes: 4 additions & 3 deletions src/util-decode-mime.c
Original file line number Diff line number Diff line change
Expand Up @@ -1184,15 +1184,15 @@ static uint32_t ProcessBase64Remainder(

/* Strip spaces in remainder */
for (uint8_t i = 0; i < state->bvr_len; i++) {
if (state->bvremain[i] != ' ') {
if (IsBase64Alphabet(state->bvremain[i])) {
block[cnt++] = state->bvremain[i];
}
}

/* if we don't have 4 bytes see if we can fill it from `buf` */
if (buf && len > 0 && cnt != B64_BLOCK) {
for (uint32_t i = 0; i < len && cnt < B64_BLOCK; i++) {
if (buf[i] != ' ') {
if (IsBase64Alphabet(buf[i])) {
block[cnt++] = buf[i];
}
buf_consumed++;
Expand Down Expand Up @@ -1273,7 +1273,8 @@ static inline MimeDecRetCode ProcessBase64BodyLineCopyRemainder(
return MIME_DEC_ERR_DATA;

for (uint32_t i = offset; i < buf_len; i++) {
if (buf[i] != ' ') {
// Skip any characters outside of the base64 alphabet as per RFC 2045
if (IsBase64Alphabet(buf[i])) {
DEBUG_VALIDATE_BUG_ON(state->bvr_len >= B64_BLOCK);
if (state->bvr_len >= B64_BLOCK)
return MIME_DEC_ERR_DATA;
Expand Down

0 comments on commit cc3677d

Please sign in to comment.