/*******************************************************************************
 * File Name         : HIDO_Base64.c
 * Description       :
 * Created on        : 2018Äê6ÔÂ1ÈÕ
 * Author            : www.hido-studio.com
 *******************************************************************************/

/*******************************************************************************
 *                              Include Files                                  *
 *******************************************************************************/
#include "HIDO_Base64.h"
#include "stdlib.h"
#include "string.h"

/*******************************************************************************
 *                                  Macro                                      *
 *******************************************************************************/

/*******************************************************************************
 *                             Type Definition                                 *
 *******************************************************************************/

/*******************************************************************************
 *                             Local Variable                                  *
 *******************************************************************************/

/*******************************************************************************
 *                        Local Function Declaration                           *
 *******************************************************************************/

/*******************************************************************************
 *                             Local Function                                  *
 *******************************************************************************/

/*******************************************************************************
 *                             Global Function                                 *
 *******************************************************************************/

/*******************************************************************************
 * Function Name     : HIDO_Base64Encode
 * Description       : 
 * Input             : 
 * Output            : 
 * Return            : 
 * Author            : www.hido-studio.com
 * Modified Date:    : 2018Äê6ÔÂ1ÈÕ
 *******************************************************************************/
HIDO_INT32 HIDO_Base64Encode(HIDO_UINT8 *_pcSrc, HIDO_UINT32 _u32SrcLen, HIDO_CHAR *_pcDst, HIDO_UINT32 _u32DstSize)
{
    HIDO_UINT32 u32DstLen = 0;
    HIDO_UINT32 i = 0;
    HIDO_UINT32 j = 0;
    HIDO_CHAR *pcBase64EncodeTable = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";

    if(0 == _u32SrcLen)
    {
        return HIDO_ERR;
    }

    if ((_u32SrcLen % 3) == 0)
    {
        u32DstLen = (_u32SrcLen / 3) * 4;
    }
    else
    {
        u32DstLen = ((_u32SrcLen / 3) + 1) * 4;
    }

    if (_u32DstSize < (u32DstLen + 1))
    {
        return HIDO_ERR;
    }

    _pcDst[u32DstLen] = '\0';
    for (i = 0, j = 0; i < u32DstLen - 2; j += 3, i += 4)
    {
        _pcDst[i] = pcBase64EncodeTable[_pcSrc[j] >> 2];
        _pcDst[i + 1] = pcBase64EncodeTable[(_pcSrc[j] & 0x3) << 4 | (_pcSrc[j + 1] >> 4)];
        _pcDst[i + 2] = pcBase64EncodeTable[(_pcSrc[j + 1] & 0xf) << 2 | (_pcSrc[j + 2] >> 6)];
        _pcDst[i + 3] = pcBase64EncodeTable[_pcSrc[j + 2] & 0x3f];
    }

    switch (_u32SrcLen % 3)
    {
        case 1:
            _pcDst[i - 2] = '=';
            _pcDst[i - 1] = '=';
            break;
        case 2:
            _pcDst[i - 1] = '=';
            break;
    }

    return HIDO_OK;
}

/*******************************************************************************
 * Function Name     : HIDO_Base64Encode
 * Description       :
 * Input             :
 * Output            :
 * Return            :
 * Author            : www.hido-studio.com
 * Modified Date:    : 2018Äê6ÔÂ1ÈÕ
 *******************************************************************************/
HIDO_INT32 HIDO_Base64Decode(HIDO_CHAR *_pcSrc, HIDO_UINT8 *_pu8Dst, HIDO_UINT32 _u32DstSize, HIDO_UINT32 *_pu32DstLen)
{
    const HIDO_UINT8 au8Base64DecodeTable[] =
    {
            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
            0, 0, 0, 0, 0, 62, 0, 0, 0, 63, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4,
            5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 0, 0, 0, 0, 0, 0, 26, 27, 28,
            29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51
    };

    HIDO_UINT32 u32DstLen = 0;
    HIDO_UINT32 u32SrcLen = 0;
    HIDO_UINT32 i = 0;
    HIDO_UINT32 j = 0;

    u32SrcLen = strlen(_pcSrc);
    if (u32SrcLen < 4)
    {
        return HIDO_ERR;
    }

    for (i = 0; _pcSrc[i] != '\0'; i++)
    {
        if(!((_pcSrc[i] >= 'A' && _pcSrc[i] <= 'Z') || (_pcSrc[i] >= 'a' && _pcSrc[i] <= 'z')
                || (_pcSrc[i] >= '0' && _pcSrc[i] <= '9') || _pcSrc[i] == '=' || _pcSrc[i] == '+' || _pcSrc[i] == '/'))
        {
            return HIDO_ERR;
        }
    }

    if (strstr(_pcSrc, "==") != HIDO_NULL)
    {
        u32DstLen = (u32SrcLen / 4) * 3 - 2;
    }
    else if (strstr(_pcSrc, "="))
    {
        u32DstLen = (u32SrcLen / 4) * 3 - 1;
    }
    else
    {
        u32DstLen = (u32SrcLen / 4) * 3;
    }

    for (i = 0, j = 0; i < u32SrcLen - 2; j += 3, i += 4)
    {
        _pu8Dst[j] = (((HIDO_UINT8) au8Base64DecodeTable[_pcSrc[i]]) << 2) | (((HIDO_UINT8) au8Base64DecodeTable[_pcSrc[i + 1]]) >> 4);
        _pu8Dst[j + 1] = (((HIDO_UINT8) au8Base64DecodeTable[_pcSrc[i + 1]]) << 4) | (((HIDO_UINT8) au8Base64DecodeTable[_pcSrc[i + 2]]) >> 2);
        _pu8Dst[j + 2] = (((HIDO_UINT8) au8Base64DecodeTable[_pcSrc[i + 2]]) << 6) | ((HIDO_UINT8) au8Base64DecodeTable[_pcSrc[i + 3]]);
    }

    if (_pu32DstLen != HIDO_NULL)
    {
        *_pu32DstLen = u32DstLen;
    }

    return HIDO_OK;
}