C++实现base64编码(1)

下面的代码是php里面的base64编码逻辑，确实比我之前的要美观很多，我只是简单的用C++的类进行了一下封装，删除了一些没用的逻辑，基本上还是原来PHP的代码：

#include <iostream>
#include <cstring>
#include <cctype>
using namespace std;

class Base64{
private:
    static const char base64_pad = '=';

public:
    unsigned char * Encode(const unsigned char *str, int length, int &ret_length);
    unsigned char * Decode(const unsigned char *str, int length, int &ret_length);
};

unsigned char * Base64::Encode(const unsigned char *str, int length, int &ret_length) /* {{{ */
{
    /* {{{ base64 tables */
    const char base64_table[] = {
        'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
        'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
        'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
        'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
        '', '', '', '', '', '', '', '', '', '', '+', '/', '\0'
    };
    const unsigned char *current = str;
    unsigned char *p;
    unsigned char * result;

    result = new unsigned char[(length+)/*];
    p = result;

    if (length < ) {
        ret_length = ;
        return NULL;
    }

    while (length > ) { /* keep going until we have less than 24 bits */
        *p++ = base64_table[current[] >> ];
        *p++= base64_table[((current[] & 0x03) << ) + (current[] >> )];
        *p++= base64_table[((current[] & 0x0f) << ) + (current[] >> )];
        *p++= base64_table[current[] & 0x3f];

        current += ;
        length -= ; /* we just handle 3 octets of data */
    }

    /* now deal with the tail end of things */
    if (length != ) {
        *p++= base64_table[current[] >> ];
        if (length > ) {
            *p++= base64_table[((current[] & 0x03) << ) + (current[] >> )];
            *p++= base64_table[(current[] & 0x0f) << ];
            *p++= base64_pad;
        } else {
            *p++ = base64_table[(current[] & 0x03) << ];
            *p++ = base64_pad;
            *p++ = base64_pad;
        }
    }
    ret_length = (int)(p - result);
    *p = '\0';
    return result;
}

unsigned char * Base64::Decode(const unsigned char *str, int length, int &ret_length) /* {{{ */
{
    const short base64_reverse_table[] = {
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, , -, -, -, ,
        , , , , , , , , , , -, -, -, -, -, -,
        -,  ,  ,  ,  ,  ,  ,  ,  ,  ,  , , , , , ,
        , , , , , , , , , , , -, -, -, -, -,
        -, , , , , , , , , , , , , , , ,
        , , , , , , , , , , , -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -,
        -, -, -, -, -, -, -, -, -, -, -, -, -, -, -, -
    };
    const unsigned char *current = str;
    int ch, i = , j = , k;
    /* this sucks for threaded environments */

    unsigned char * result;

    result = new unsigned char[length];

    /* run through the whole string, converting as we go */
    while ((ch = *current++) != '\0' && length-- > ) {
        if (ch == base64_pad) {
            if (*current != '=' && (i % ) == ) {
                return NULL;
            }
            continue;
        }
        ch = base64_reverse_table[ch];
        if (ch <  || ch == -) { /* a space or some other separator character, we simply skip over */
            continue;
        } else if (ch == -) {
            return NULL;
        }

        switch(i % ) {
        case :
            result[j] = ch << ;
            break;
        case :
            result[j++] |= ch >> ;
            result[j] = (ch & 0x0f) << ;
            break;
        case :
            result[j++] |= ch >>;
            result[j] = (ch & 0x03) << ;
            break;
        case :
            result[j++] |= ch;
            break;
        }
        i++;
        cout << "result == " << result << endl;
    }

    k = j;
    /* mop things up if we ended on a boundary */
    if (ch == base64_pad) {
        switch(i % ) {
        case :
            return NULL;
        case :
            k++;
        case :
            result[k] = ;
        }
    }
    if(ret_length) {
        ret_length = j;
    }
    result[j] = '\0';
    return result;
}
int main()
{
    unsigned char str[] = "     aHR0cDo     vL2ltZy52LmNtY20uY29tLzdkNjZkNy0wYzJkLTVhNTgtYTA3Mi03MWY4MjhiOTRjYmNfY3JvcF8yMTZ{4MTUwLmpwZw=   ======";
    unsigned char *normal,*encoded;
    int i,len = ,ret_len=;

    cout << "original string : " << str << endl;
    len = sizeof(str)-;
    Base64 * base64 = new Base64();
    cout << "str = " << str << endl;
    normal = base64->Decode(str,len,ret_len);
    cout << "base64 decode : " << normal <<endl;
    delete [] encoded;
    delete [] normal;
    return ;
}

上面的代码对php源码中的逻辑做了优化，删除了decode方法中判断结尾的“=”号时多余的逻辑，以免干扰视线。具体删除的代码参照php源码中ext/standard/base64.c

上一篇php实现base64的代码也进行了调整，提高了容错。