#include "encode.h"

#define PADDING '='

unsigned char b64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" \
                      "abcdefghijklmnopqrstuvwxyz" \
                      "0123456789" \
                      "+/";

unsigned char b64u[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" \
                       "abcdefghijklmnopqrstuvwxyz" \
                       "0123456789" \
                       "-_";

int
encode(unsigned char *op, int olen, unsigned char *sp, int url)
{
    int n;
    unsigned char *sb, *enc;

    sb = sp;
    enc = (!url) ? b64 : b64u;
    switch (olen) {
        case 3:
            *sp++ = enc[(*op & ~3) >> 2];
            n = (*op++ & 3) << 4;
            *sp++ = enc[n + ((*op & ~15) >> 4)];
            n = (*op++ & 15) << 2;
            *sp++ = enc[n + ((*op & 192) >> 6)];
            *sp = enc[*op & ~192];
            break;
        case 2:
            *sp++ = enc[(*op & ~3) >> 2];
            n = (*op++ & 3) << 4;
            *sp++ = enc[n + ((*op & ~15) >> 4)];
            *sp++ = enc[(*op & 15) << 2];
            *sp = PADDING;
            break;
        case 1:
            *sp++ = enc[(*op & ~3) >> 2];
            *sp++ = enc[(*op & 3) << 4];
            *sp++ = PADDING;
            *sp = PADDING;
            break;
    }

    return sp-sb+1;
}

int
decode(unsigned char *sp, int slen, unsigned char *op, int url)
{
    int n, b, atob(int c, int url);
    unsigned char *ob;

    ob = op;
    switch (slen) {
        case 4:
            n = atob(*sp++, url) << 2;
            b = atob(*sp++, url);
            *op++ = n + ((b & ~15) >> 4);

            n = (b & 15) << 4;
            b = atob(*sp++, url);
            *op++ = n + ((b & ~3) >> 2);

            *op = ((b & 3) << 6) + atob(*sp, url);
            break;
        case 3:
            n = atob(*sp++, url) << 2;
            b = atob(*sp++, url);
            *op++ = n + ((b & ~15) >> 4);

            n = (b & 15) << 4;
            *op = n + ((atob(*sp, url) & ~3) >> 2);
            break;
        case 2:
            n = atob(*sp++, url) << 2;
            *op = n + ((atob(*sp, url) & ~15) >> 4);
            break;
    }

    return op-ob+1;
}

int atob(int c, int url)
{
    if (c >= 'A' && c <= 'Z')
        c -= 'A';
    else if (c >= 'a' && c <= 'z')
        c = c - 'a' + 26;
    else if (c >= '0' && c <= '9')
        c = c - '0' + 26 * 2;
    else if (c == ((!url) ? '+' : '-'))
        c = 62;
    else
        c = 63;

    return c;
}