Polarssl Base64.c错误检查

2023年5月1日 256次阅读

我使用了polarssl的base64.c文件来解码一些数据.

int base64_decode( unsigned char *dst, size_t *dlen,
                   const unsigned char *src, size_t slen ){
size_t i, n=0;
uint32_t j, x;
unsigned char *p;

for( i = n = j = 0; i < slen; i++ )
{
    if( ( slen - i ) >= 2 &&
        src[i] == '\r' && src[i + 1] == '\n' )
        continue;

    if( src[i] == '\n' )
        continue;

    if( src[i] == '=' && ++j > 2 )
        return( POLARSSL_ERR_BASE64_INVALID_CHARACTER );

    if( src[i] > 127 || base64_dec_map[src[i]] == 127 )
        return( POLARSSL_ERR_BASE64_INVALID_CHARACTER );

    if( base64_dec_map[src[i]] < 64 && j != 0 )
        return( POLARSSL_ERR_BASE64_INVALID_CHARACTER );

    n++;
}
n = ((n * 6) + 7) >> 3;
if( dst == NULL || *dlen < n )
{
    *dlen = n;
    return( POLARSSL_ERR_BASE64_BUFFER_TOO_SMALL );
}
for( j = 3, n = x = 0, p = dst; i > 0; i--, src++ ){
    if( *src == '\r' || *src == '\n' )
        continue;

    j -= ( base64_dec_map[*src] == 64 );
    x  = (x << 6) | ( base64_dec_map[*src] & 0x3F );

    if( ++n == 4 )
    {
        n = 0;
        if( j > 0 ) *p++ = (unsigned char)( x >> 16 );
        if( j > 1 ) *p++ = (unsigned char)( x >>  8 );
        if( j > 2 ) *p++ = (unsigned char)( x       );
    }
}

*dlen = p - dst;

return( 0 );
}

当我的数据来到代码’n’的这部分时,它比’* dlen’更大;但是当我把这部分作为注释行时,功能也很好用.

if( dst == NULL || *dlen < n )
{
    *dlen = n;
    return( POLARSSL_ERR_BASE64_BUFFER_TOO_SMALL );
}

有没有可能这个代码部分是不必要的？我的意思是如果我从功能中删除这部分,我会错过业务逻辑中的细节吗？

编辑：

例如：

src：A7ViV8hpIon0lisFRCvQpw ==

格伦：18

dst：3 b5 62 57 c8 69 22 89 f4 96 2b 5 44 2b d0 a7 0 0

实际上’dst’中的最后两个零不应该是正确的结果,当我发送’dlen’为16并发表评论if if statement时,我得到正确的结果.我认为n计算不正确或我遗漏了一些东西.

最佳答案 你是对的！ base64_decode()中返回的长度(和长度检查)中存在错误.

之后丢失了一行：

n = ((n * 6) + 7) >> 3;

如果你把：

n -= j;

在此之后,它将按预期工作.

点赞