testBase64.cpp

Go to the documentation of this file.

/*
 * testBase64.cpp
 *
 * Copyright (C) 2007,2011  Thomas A. Vaughan
 * All rights reserved.
 *
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of the <organization> nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THOMAS A. VAUGHAN ''AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THOMAS A. VAUGHAN BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Quick test for base 64 encoding/decoding.
 */

// includes --------------------------------------------------------------------
#include "wave-crypto/wave-crypto.h"// always include our own header file first!

#include <iostream>

#include "common/wave_ex.h"
#include "perf/perf.h"



struct test_t {
        const char *    input;
        const char *    expected;
};


static const test_t s_tests[] = {
        // input                        // expected
        { "test",                       "AAAFc2V0AAB0" },
        { "y",                          "AAACAAB5" },
        { "",                           "AAABAAAA" },
        { "This is a much longer string!  Yes, verily.",
           "AAAsaWhUaSBzYSBzdW0gIGhjbm9scmVndHMgbmlyICFnZVkgICxzcmV2eWxpAAAu" },
        { "00000000123456789abcdefghijklmnopqrstruvwxyz",
           "AAAtMDAwMDAwMTAwNDMyNzY1YTk4ZGNiZ2ZlamlobWxrcG9uc3JxdXJ0eHd2AHp5"},

        // must be last!
        { NULL, NULL }
};



////////////////////////////////////////////////////////////////////////////////
//
//      static helper methods
//
////////////////////////////////////////////////////////////////////////////////

static void
doTest
(
IN const char * data,
IN const char * expected
)
{
        ASSERT(data, "null");
        ASSERT(expected, "null");

        long bytes = strlen(data) + 1;          // include null terminator
        DPRINTF("'%s': %ld bytes", data, bytes);
        std::string enc = crypto::encodeBase64((byte_t *) data, bytes, true);

        DPRINTF("  base64: %s", enc.c_str());

        crypto::vec_byte_t out;
        crypto::decodeBase64(enc.c_str(), out);

        DPRINTF("  Decoding found %d bytes", (int) out.size());

        const char * back = (const char *)&out[0];
        DPRINTF("  Decoded: '%s'", back);

        if (bytes != (long) out.size()) {
                WAVE_EX(wex);
                wex << "Received an improper count on decoding!";
        }
        if (strcmp(data, back)) {
                WAVE_EX(wex);
                wex << "Decoding didn't return what we encoded!";
        }
        if (strcmp(enc.c_str(), expected)) {
                WAVE_EX(wex);
                wex << "Encoding didn't produce what we expected!\n";
                wex << "Expected: " << expected << "\n";
                wex << "Got: " << enc;
        }
}



static void
doDwordTest
(
IN dword_t dw
)
{
        int bytes = sizeof(dword_t);
        std::string enc = crypto::encodeBase64((byte_t *) &dw, bytes, false);

        DPRINTF("dword 0x%08x --> '%s'", dw, enc.c_str());
}



////////////////////////////////////////////////////////////////////////////////
//
//      entry point
//
////////////////////////////////////////////////////////////////////////////////

int
main
(
IN int argc,
IN const char * argv[]
)
{
        int retval = 0;
        try {
                for (const test_t * p = s_tests; p->input; ++p) {
                        doTest(p->input, p->expected);
                }

                // attempt to encode some integers
                doDwordTest(0x00000000);
                doDwordTest(0x0000000F);
                doDwordTest(0x000000FF);
                doDwordTest(0x00000FFF);
                doDwordTest(0x0000FFFF);
                doDwordTest(0x000FFFFF);
                doDwordTest(0x00FFFFFF);
                doDwordTest(0x0FFFFFFF);
                doDwordTest(0xFFFFFFFF);
                doDwordTest(0xFFFFFFFE);
                doDwordTest(0xFFFFFFFA);
                doDwordTest(0xFFFFFFF0);
                doDwordTest(0x70000000);
                doDwordTest(0x07000000);
                doDwordTest(0x00000001);
        } catch (std::exception& e) {
                DPRINTF("Exception: %s", e.what());
                retval = 1;
        } catch (...) {
                DPRINTF("Unknown exception!");
                retval = 2;
        }
        perf::dumpTimingSummary(std::cerr);

        return retval;
}