/*****************************************************************************
 * PokerTH - The open source texas holdem engine                             *
 * Copyright (C) 2006-2011 Felix Hammer, Florian Thauer, Lothar May          *
 *                                                                           *
 * This program is free software: you can redistribute it and/or modify      *
 * it under the terms of the GNU Affero General Public License as            *
 * published by the Free Software Foundation, either version 3 of the        *
 * License, or (at your option) any later version.                           *
 *                                                                           *
 * This program is distributed in the hope that it will be useful,           *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of            *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the             *
 * GNU Affero General Public License for more details.                       *
 *                                                                           *
 * You should have received a copy of the GNU Affero General Public License  *
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.     *
 *****************************************************************************/
/* Helper class for crypt functions. */

#ifndef _CRYPTHELPER_H_
#define _CRYPTHELPER_H_

#include <string>
#include <vector>

#define MD5_DATA_SIZE		16
#define SHA1_DATA_SIZE		20

#define AES_BLOCK_SIZE		16
#define ADD_PADDING(x) ((((x) + 15) >> 4) << 4)

class HashBuf
{
public:
	virtual ~HashBuf();

	std::string ToString() const;
	bool FromString(const std::string &text);
	bool IsZero() const;

	bool operator==(const HashBuf &other) const;
	bool operator<(const HashBuf &other) const;

	virtual unsigned char *GetData() = 0;
	virtual const unsigned char *GetData() const = 0;
	virtual int GetDataSize() const = 0;
};

class MD5Buf : public HashBuf
{
public:
	MD5Buf();

	virtual unsigned char *GetData();
	virtual const unsigned char *GetData() const;
	virtual int GetDataSize() const;

private:
	unsigned char m_data[MD5_DATA_SIZE];
};

class SHA1Buf : public HashBuf
{
public:
	SHA1Buf();

	virtual unsigned char *GetData();
	virtual const unsigned char *GetData() const;
	virtual int GetDataSize() const;

	unsigned char m_data[SHA1_DATA_SIZE];
};

class CryptHelper
{
public:

	static bool MD5Sum(const std::string &fileName, MD5Buf &buf);
	static bool SHA1Hash(const unsigned char *data, unsigned dataSize, SHA1Buf &buf);
	static bool HMACSha1(const unsigned char *keyData, unsigned keySize, const unsigned char *plainData, unsigned plainSize, SHA1Buf &buf);
	static bool AES128Encrypt(const unsigned char *keyData, unsigned keySize, const std::string &plainStr, std::vector<unsigned char> &outCipher);
	static bool AES128Decrypt(const unsigned char *keyData, unsigned keySize, const unsigned char *cipher, unsigned cipherSize, std::string &outPlain);

private:
	static void BytesToKey(const unsigned char *keyData, unsigned keySize, unsigned char *key, unsigned char *iv);
};

#endif