/*
 * Copyright (c) 2010 SURFnet bv
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 THE AUTHOR 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.
 */

/*****************************************************************************
 ByteStringTests.cpp

 Contains test cases to test the ByteString class
 *****************************************************************************/

#include <stdlib.h>
#include <stdio.h>
#include <cppunit/extensions/HelperMacros.h>
#include <string>
#include "ByteStringTests.h"
#include "ByteString.h"

CPPUNIT_TEST_SUITE_REGISTRATION(ByteStringTests);

void ByteStringTests::setUp()
{
}

void ByteStringTests::tearDown()
{
	fflush(stdout);
}

void ByteStringTests::testIntegrity()
{
	unsigned char testData[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
	                             0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };

	ByteString b(testData, sizeof(testData));

	// Test if the right data is returned
	CPPUNIT_ASSERT(memcmp(testData, b.byte_str(), sizeof(testData)) == 0);

	// Test size
	CPPUNIT_ASSERT(b.size() == sizeof(testData));

	// Test the copy constructor
	ByteString b2(b);

	// Test using comparison operator
	CPPUNIT_ASSERT(b == b2);

	// Test using memcmp
	CPPUNIT_ASSERT(memcmp(b.byte_str(), b2.byte_str(), b.size()) == 0);

	// Modify the copied version and test again
	b2[1] = 0x20;

	// Test using comparison operator
	CPPUNIT_ASSERT(b != b2);

	// Test using memcmp directly
	CPPUNIT_ASSERT(memcmp(b.byte_str(), b2.byte_str(), b.size()) != 0);

	// Verify that b was not affected
	CPPUNIT_ASSERT(memcmp(b.byte_str(), testData, sizeof(testData)) == 0);

	// Modify the source data and check if the array operator has functioned correctly
	testData[1] = 0x20;

	// Test if the right data is in b2
	CPPUNIT_ASSERT(memcmp(b2.byte_str(), testData, sizeof(testData)) == 0);
}

void ByteStringTests::testAppend()
{
	unsigned char testData[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
	                             0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };

	ByteString b;
	ByteString b1(testData, sizeof(testData));

	// Test that b is empty and b1 is not
	CPPUNIT_ASSERT((b.size() == 0) && (b1.size() == sizeof(testData)));

	// Append 1 byte to b
	b += 0x01;

	// Check the contents of b
	CPPUNIT_ASSERT(b.size() == 1);
	CPPUNIT_ASSERT(b[0] == 0x01);

	// Append another byte to b
	b += 0x02;

	// Check the contents of b
	CPPUNIT_ASSERT(b.size() == 2);
	CPPUNIT_ASSERT((b[0] == 0x01) && (b[1] == 0x02));

	// Append b1 to b
	b += b1;

	// Check the contents of b
	CPPUNIT_ASSERT(b.size() == 2 + sizeof(testData));
	CPPUNIT_ASSERT((b[0] == 0x01) && (b[1] == 0x02));
	CPPUNIT_ASSERT(memcmp(&b[2], testData, sizeof(testData)) == 0);

	// Append b to b
	b += b;

	// Check the contents of b
	CPPUNIT_ASSERT(b.size() == 2 * (2 + sizeof(testData)));
	CPPUNIT_ASSERT((b[0] == 0x01) && (b[1] == 0x02) && 
	               (b[(2 + sizeof(testData)) + 0] == 0x01) &&
		       (b[(2 + sizeof(testData)) + 1] == 0x02));
	CPPUNIT_ASSERT((memcmp(&b[2], testData, sizeof(testData)) == 0) &&
	               (memcmp(&b[2 + 2 + sizeof(testData)], testData, sizeof(testData)) == 0));
}

void ByteStringTests::testSubstr()
{
	unsigned char testData[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
	                             0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };

	ByteString b;
	ByteString b1(testData, sizeof(testData));

	// Take a substring
	b = b1.substr(8, 4);
	
	// Check b
	CPPUNIT_ASSERT(b.size() == 4);
	CPPUNIT_ASSERT(memcmp(b.byte_str(), &testData[8], 4) == 0);

	// Take another substring
	b = b1.substr(8);

	// Check b
	CPPUNIT_ASSERT(b.size() == 8);
	CPPUNIT_ASSERT(memcmp(b.byte_str(), &testData[8], 8) == 0);

	// Two substrings added should yield the original string
	b = b1.substr(0, 8) + b1.substr(8);

	// Check b
	CPPUNIT_ASSERT(b.size() == sizeof(testData));
	CPPUNIT_ASSERT(memcmp(b.byte_str(), testData, sizeof(testData)) == 0);
}

void ByteStringTests::testFromHexStr()
{
	unsigned char testData[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
	                             0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };

	ByteString b("0102030405060708090a0b0c0d0e0f10");
	ByteString b1("0102030405060708090A0B0C0D0E0F10");

	CPPUNIT_ASSERT(memcmp(b.byte_str(), testData, sizeof(testData)) == 0);
	CPPUNIT_ASSERT(memcmp(b1.byte_str(), testData, sizeof(testData)) == 0);
}

void ByteStringTests::testXOR()
{
	unsigned char left[]	= { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 };
	unsigned char right[]	= { 0x80, 0x70, 0x60, 0x50, 0x40, 0x30, 0x20, 0x10 };
	unsigned char xorred[]	= { 0x81, 0x72, 0x63, 0x54, 0x45, 0x36, 0x27, 0x18 };

	ByteString l(left, 8);
	ByteString r(right, 8);
	ByteString x(xorred, 8);
	ByteString xed;

	xed = l ^ r;

	CPPUNIT_ASSERT(xed == x);

	ByteString l1(left, 8);
	ByteString r1(right, 8);
	
	l1 ^= r1;

	CPPUNIT_ASSERT(l1 == x);

	l1 ^= l;

	CPPUNIT_ASSERT(l1 == r);

	ByteString l_(left, 7);

	xed = l_ ^ r;

	CPPUNIT_ASSERT((xed.size() == 7) && (xed == x.substr(0, 7)));

	ByteString r_(right, 7);

	xed = l ^ r_;
	
	CPPUNIT_ASSERT((xed.size() == 7) && (xed == x.substr(0, 7)));

	ByteString l1_(left, 8);

	l1_ ^= r_;

	CPPUNIT_ASSERT((l1.size() == 8) && (l1_.substr(0, 7) == x.substr(0,7)) && (l1_[7] == l[7]));

	ByteString l1__(left, 7);

	l1__ ^= r;

	CPPUNIT_ASSERT((l1__ == x.substr(0,7)) && (l1__.size() == 7));
}

void ByteStringTests::testToHexStr()
{
	ByteString b("0102030405060708090A0B0C0D0E0F");
	ByteString b1("DEADBEEF");
	ByteString b2("deadC0FFEE");

	std::string s = b.hex_str();
	std::string s1 = b1.hex_str();
	std::string s2 = b2.hex_str();

	CPPUNIT_ASSERT(s.compare("0102030405060708090A0B0C0D0E0F") == 0);
	CPPUNIT_ASSERT(s1.compare("DEADBEEF") == 0);
	CPPUNIT_ASSERT(s2.compare("DEADC0FFEE") == 0);
}

void ByteStringTests::testLongValues()
{
	unsigned long ul1 = 0x00112233;
	unsigned long ul2 = 0x10203040;
	unsigned long ul3 = 0xF0E0D0C0;

	ByteString b1(ul1);
	ByteString b2(ul2);
	ByteString b3(ul3);

	CPPUNIT_ASSERT(b1 == ByteString("0000000000112233"));
	CPPUNIT_ASSERT(b2 == ByteString("0000000010203040"));
	CPPUNIT_ASSERT(b3 == ByteString("00000000F0E0D0C0"));

	CPPUNIT_ASSERT(b1.long_val() == ul1);
	CPPUNIT_ASSERT(b2.long_val() == ul2);
	CPPUNIT_ASSERT(b3.long_val() == ul3);

	ByteString concat = b1 + b2 + b3;

	CPPUNIT_ASSERT(concat == ByteString("0000000000112233000000001020304000000000F0E0D0C0"));

	unsigned long ulr1 = concat.firstLong();

	CPPUNIT_ASSERT(ulr1 == ul1);
	CPPUNIT_ASSERT(concat == ByteString("000000001020304000000000F0E0D0C0"));

	unsigned long ulr2 = concat.firstLong();

	CPPUNIT_ASSERT(ulr2 == ul2);
	CPPUNIT_ASSERT(concat == ByteString("00000000F0E0D0C0"));

	unsigned long ulr3 = concat.firstLong();

	CPPUNIT_ASSERT(ulr3 == ul3);
	CPPUNIT_ASSERT(concat.size() == 0);

	ByteString b4("ABCDEF");

	CPPUNIT_ASSERT(b4.long_val() == 0xABCDEF);
	CPPUNIT_ASSERT(b4.size() == 3);
	CPPUNIT_ASSERT(b4.firstLong() == 0xABCDEF);
	CPPUNIT_ASSERT(b4.size() == 0);
}

void ByteStringTests::testSplitting()
{
	ByteString b("AABBCCDDEEFF112233445566");
	
	ByteString b1 = b.split(6);

	CPPUNIT_ASSERT(b == ByteString("112233445566"));
	CPPUNIT_ASSERT(b1 == ByteString("AABBCCDDEEFF"));

	ByteString b2 = b1.split(8);

	CPPUNIT_ASSERT(b2 == ByteString("AABBCCDDEEFF"));
	CPPUNIT_ASSERT(b1.size() == 0);
}

void ByteStringTests::testBits()
{
	ByteString b1("0");
	ByteString b2("08");
	ByteString b3("00FFFFF");
	ByteString b4("123456");

	CPPUNIT_ASSERT(b1.bits() == 0);
	CPPUNIT_ASSERT(b2.bits() == 4);
	CPPUNIT_ASSERT(b3.bits() == 20);
	CPPUNIT_ASSERT(b4.bits() == 21);
}

void ByteStringTests::testSerialising()
{
	ByteString b1("AA11AA11AA11AA11AA11AA11AA11");
	ByteString b2("BB22BB22BB22BB22BB22BB22");
	ByteString b3("CC33CC33CC33CC33CC33CC33CC33CC33");

	ByteString s1 = b1.serialise();

	CPPUNIT_ASSERT(s1.size() == b1.size() + 8);

	ByteString d1 = ByteString::chainDeserialise(s1);

	CPPUNIT_ASSERT(s1.size() == 0);
	CPPUNIT_ASSERT(d1 == b1);

	ByteString s2 = b3.serialise() + b2.serialise() + b1.serialise();

	CPPUNIT_ASSERT(s2.size() == b1.size() + b2.size() + b3.size() + (3*8));

	d1 = ByteString::chainDeserialise(s2);
	
	CPPUNIT_ASSERT(d1.size() == b3.size());
	CPPUNIT_ASSERT(s2.size() == b1.size() + b2.size() + (2*8));

	ByteString d2 = ByteString::chainDeserialise(s2);

	CPPUNIT_ASSERT(d2.size() == b2.size());
	CPPUNIT_ASSERT(s2.size() == b1.size() + 8);

	ByteString d3 = ByteString::chainDeserialise(s2);

	CPPUNIT_ASSERT(d3.size() == b1.size());
	CPPUNIT_ASSERT(s2.size() == 0);

	CPPUNIT_ASSERT(d1 == b3);
	CPPUNIT_ASSERT(d2 == b2);
	CPPUNIT_ASSERT(d3 == b1);
}