#include <iostream>
#include <fstream>
#include "Blowfish.h"

using namespace std;

//Function to convert unsigned char to string of length 2
void Char2Hex(const unsigned char ch, char* szHex)
{
	unsigned char byte[2];
	byte[0] = ch/16;
	byte[1] = ch%16;
	for(int i=0; i<2; i++)
	{
		if(byte[i] >= 0 && byte[i] <= 9)
			szHex[i] = '0' + byte[i];
		else
			szHex[i] = 'A' + byte[i] - 10;
	}
	szHex[2] = 0;
}

//Function to convert string of length 2 to unsigned char
void Hex2Char(const char* szHex, unsigned char& rch)
{
	rch = 0;
	for(int i=0; i<2; i++)
	{
		if(*(szHex + i) >='0' && *(szHex + i) <= '9')
			rch = (rch << 4) + (*(szHex + i) - '0');
		else if(*(szHex + i) >='A' && *(szHex + i) <= 'F')
			rch = (rch << 4) + (*(szHex + i) - 'A' + 10);
		else
			break;
	}
}    

//Function to convert string of unsigned chars to string of chars
void CharStr2HexStr(const unsigned char* pucCharStr, char* pszHexStr, int iSize)
{
	int i;
	char szHex[3];
	pszHexStr[0] = 0;
	for(i=0; i<iSize; i++)
	{
		Char2Hex(pucCharStr[i], szHex);
		strcat(pszHexStr, szHex);
	}
}

//Function to convert string of chars to string of unsigned chars
void HexStr2CharStr(const char* pszHexStr, unsigned char* pucCharStr, int iSize)
{
	int i;
	unsigned char ch;
	for(i=0; i<iSize; i++)
	{
		Hex2Char(pszHexStr+2*i, ch);
		pucCharStr[i] = ch;
	}
}

void main()
{
	//TESTING Blowfish
	//ecb test data (taken from the DES validation tests)
	char szData[34][2][17] = {
		{"0000000000000000", "0000000000000000"},
		{"FFFFFFFFFFFFFFFF", "FFFFFFFFFFFFFFFF"},
		{"3000000000000000", "1000000000000001"},
		{"1111111111111111", "1111111111111111"},
		{"0123456789ABCDEF", "1111111111111111"},
		{"1111111111111111", "0123456789ABCDEF"},
		{"0000000000000000", "0000000000000000"},
		{"FEDCBA9876543210", "0123456789ABCDEF"},
		{"7CA110454A1A6E57", "01A1D6D039776742"},
		{"0131D9619DC1376E", "5CD54CA83DEF57DA"},
		{"07A1133E4A0B2686", "0248D43806F67172"},
		{"3849674C2602319E", "51454B582DDF440A"},
		{"04B915BA43FEB5B6", "42FD443059577FA2"},
		{"0113B970FD34F2CE", "059B5E0851CF143A"},
		{"0170F175468FB5E6", "0756D8E0774761D2"},
		{"43297FAD38E373FE", "762514B829BF486A"},
		{"07A7137045DA2A16", "3BDD119049372802"},
		{"04689104C2FD3B2F", "26955F6835AF609A"},
		{"37D06BB516CB7546", "164D5E404F275232"},
		{"1F08260D1AC2465E", "6B056E18759F5CCA"},
		{"584023641ABA6176", "004BD6EF09176062"},
		{"025816164629B007", "480D39006EE762F2"},
		{"49793EBC79B3258F", "437540C8698F3CFA"},
		{"4FB05E1515AB73A7", "072D43A077075292"},
		{"49E95D6D4CA229BF", "02FE55778117F12A"},
		{"018310DC409B26D6", "1D9D5C5018F728C2"},
		{"1C587F1C13924FEF", "305532286D6F295A"},
		{"0101010101010101", "0123456789ABCDEF"},
		{"1F1F1F1F0E0E0E0E", "0123456789ABCDEF"},
		{"E0FEE0FEF1FEF1FE", "0123456789ABCDEF"},
		{"0000000000000000", "FFFFFFFFFFFFFFFF"},
		{"FFFFFFFFFFFFFFFF", "0000000000000000"},
		{"0123456789ABCDEF", "0000000000000000"},
		{"FEDCBA9876543210", "FFFFFFFFFFFFFFFF"}
	};
	unsigned char aucKey[8];
	unsigned char aucPlainText[8];
	char szHex[17];
	szHex[16] = 0;
	unsigned char aucCipherText[8];
	ofstream out("out.txt", ios::trunc);
	for(int i=0; i<34; i++)
	{
		strcpy(szHex, szData[i][0]);
		out << szHex << " ";
		HexStr2CharStr(szHex, aucKey, 8);
		strcpy(szHex, szData[i][1]);
		out << szHex << " ";
		HexStr2CharStr(szHex, aucPlainText, 8);
		CBlowFish oBlowFish(aucKey, 8);
		oBlowFish.Encrypt(aucPlainText, aucCipherText, 8);
		CharStr2HexStr(aucCipherText, szHex, 8);
		out << szHex << endl;
	}
	out.close();

/*
//OK, the same results
Key              Text             Cipher
0000000000000000 0000000000000000 4EF997456198DD78
FFFFFFFFFFFFFFFF FFFFFFFFFFFFFFFF 51866FD5B85ECB8A
3000000000000000 1000000000000001 0781261C9136315C ???
1111111111111111 1111111111111111 2466DD878B963C9D
0123456789ABCDEF 1111111111111111 61F9C3802281B096
1111111111111111 0123456789ABCDEF 7D0CC630AFDA1EC7
0000000000000000 0000000000000000 4EF997456198DD78
FEDCBA9876543210 0123456789ABCDEF 0ACEAB0FC6A0A28D
7CA110454A1A6E57 01A1D6D039776742 59C68245EB05282B
0131D9619DC1376E 5CD54CA83DEF57DA B1B8CC0B250F09A0
07A1133E4A0B2686 0248D43806F67172 1730E5778BEA1DA4
3849674C2602319E 51454B582DDF440A A25E7856CF2651EB
04B915BA43FEB5B6 42FD443059577FA2 353882B109CE8F1A
0113B970FD34F2CE 059B5E0851CF143A 48F4D0884C379918
0170F175468FB5E6 0756D8E0774761D2 432193B78951FC98
43297FAD38E373FE 762514B829BF486A 13F04154D69D1AE5
07A7137045DA2A16 3BDD119049372802 2EEDDA93FFD39C79
04689104C2FD3B2F 26955F6835AF609A D887E0393C2DA6E3
37D06BB516CB7546 164D5E404F275232 5F99D04F5B163969
1F08260D1AC2465E 6B056E18759F5CCA 4A057A3B24D3977B
584023641ABA6176 004BD6EF09176062 452031C1E4FADA8E
025816164629B007 480D39006EE762F2 7555AE39F59B87BD
49793EBC79B3258F 437540C8698F3CFA 53C55F9CB49FC019
4FB05E1515AB73A7 072D43A077075292 7A8E7BFA937E89A3
49E95D6D4CA229BF 02FE55778117F12A CF9C5D7A4986ADB5
018310DC409B26D6 1D9D5C5018F728C2 D1ABB290658BC778
1C587F1C13924FEF 305532286D6F295A 55CB3774D13EF201
0101010101010101 0123456789ABCDEF FA34EC4847B268B2
1F1F1F1F0E0E0E0E 0123456789ABCDEF A790795108EA3CAE
E0FEE0FEF1FEF1FE 0123456789ABCDEF C39E072D9FAC631D
0000000000000000 FFFFFFFFFFFFFFFF 014933E0CDAFF6E4
FFFFFFFFFFFFFFFF 0000000000000000 F21E9A77B71C49BC
0123456789ABCDEF 0000000000000000 245946885754369A
FEDCBA9876543210 FFFFFFFFFFFFFFFF 6B5C5A9C5D9E0A5A
*/

/*
	//set_key test data
	char* szKey[] =
	{
		"F0", "F0E1", "F0E1D2", "F0E1D2C3", "F0E1D2C3B4", "F0E1D2C3B4A5",
		"F0E1D2C3B4A596", "F0E1D2C3B4A59687", "F0E1D2C3B4A5968778", "F0E1D2C3B4A596877869",
		"F0E1D2C3B4A5968778695A", "F0E1D2C3B4A5968778695A4B", "F0E1D2C3B4A5968778695A4B3C",
		"F0E1D2C3B4A5968778695A4B3C2D", "F0E1D2C3B4A5968778695A4B3C2D1E", "F0E1D2C3B4A5968778695A4B3C2D1E0F",
		"F0E1D2C3B4A5968778695A4B3C2D1E0F00", "F0E1D2C3B4A5968778695A4B3C2D1E0F0011",
		"F0E1D2C3B4A5968778695A4B3C2D1E0F001122", "F0E1D2C3B4A5968778695A4B3C2D1E0F00112233",
		"F0E1D2C3B4A5968778695A4B3C2D1E0F0011223344", "F0E1D2C3B4A5968778695A4B3C2D1E0F001122334455",
		"F0E1D2C3B4A5968778695A4B3C2D1E0F00112233445566", "F0E1D2C3B4A5968778695A4B3C2D1E0F0011223344556677"
	};
	unsigned char aucKey[24];
	unsigned char aucPlainText[8];
	char szHex[49];
	szHex[16] = 0;
	unsigned char aucCipherText[8];
	ofstream out("out.txt", ios::trunc);
	int iKey;
	for(int i=0; i<24; i++)
	{
		iKey = i+1;
		strcpy(szHex, szKey[i]);
		out << szHex << " ";
		HexStr2CharStr(szHex, aucKey, iKey);
		strcpy(szHex, "FEDCBA9876543210");
		out << szHex << " ";
		HexStr2CharStr(szHex, aucPlainText, 8);
		CBlowFish oBlowFish(aucKey, iKey);
		oBlowFish.Encrypt(aucPlainText, aucCipherText, 8);
		CharStr2HexStr(aucCipherText, szHex, 8);
		out << szHex << endl;
	}
	out.close();
*/

//OK, the same results
/*
Key                                                 Text             Cipher
F0													FEDCBA9876543210 F9AD597C49DB005E
F0E1												FEDCBA9876543210 E91D21C1D961A6D6
F0E1D2												FEDCBA9876543210 E9C2B70A1BC65CF3
F0E1D2C3											FEDCBA9876543210 BE1E639408640F05
F0E1D2C3B4											FEDCBA9876543210 B39E44481BDB1E6E
F0E1D2C3B4A5										FEDCBA9876543210 9457AA83B1928C0D
F0E1D2C3B4A596										FEDCBA9876543210 8BB77032F960629D
F0E1D2C3B4A59687 									FEDCBA9876543210 E87A244E2CC85E82
F0E1D2C3B4A5968778									FEDCBA9876543210 15750E7A4F4EC577
F0E1D2C3B4A596877869								FEDCBA9876543210 122BA70B3AB64AE0
F0E1D2C3B4A5968778695A								FEDCBA9876543210 3A833C9AFFC537F6
F0E1D2C3B4A5968778695A4B							FEDCBA9876543210 9409DA87A90F6BF2
F0E1D2C3B4A5968778695A4B3C							FEDCBA9876543210 884F80625060B8B4
F0E1D2C3B4A5968778695A4B3C2D						FEDCBA9876543210 1F85031C19E11968
F0E1D2C3B4A5968778695A4B3C2D1E						FEDCBA9876543210 79D9373A714CA34F
F0E1D2C3B4A5968778695A4B3C2D1E0F					FEDCBA9876543210 93142887EE3BE15C
F0E1D2C3B4A5968778695A4B3C2D1E0F00					FEDCBA9876543210 03429E838CE2D14B
F0E1D2C3B4A5968778695A4B3C2D1E0F0011				FEDCBA9876543210 A4299E27469FF67B
F0E1D2C3B4A5968778695A4B3C2D1E0F001122				FEDCBA9876543210 AFD5AED1C1BC96A8
F0E1D2C3B4A5968778695A4B3C2D1E0F00112233			FEDCBA9876543210 10851C0E3858DA9F
F0E1D2C3B4A5968778695A4B3C2D1E0F0011223344			FEDCBA9876543210 E6F51ED79B9DB21F
F0E1D2C3B4A5968778695A4B3C2D1E0F001122334455 		FEDCBA9876543210 64A6E14AFD36B46F
F0E1D2C3B4A5968778695A4B3C2D1E0F00112233445566		FEDCBA9876543210 80C7D7D45A5479AD
F0E1D2C3B4A5968778695A4B3C2D1E0F0011223344556677	FEDCBA9876543210 05044B62FA52D080
*/

}