#include "stdafx.h"
#include "desc.h"
#include "packet.h"

typedef std::vector<std::pair<std::string, BYTE* > > PanamaVectorType;
static PanamaVectorType s_panamaVector;

size_t PanamaLoad()
{
	FILE* fp;

	fp = fopen("panama/panama.lst", "r");

	if (!fp)
		return 0;

	char buf[256];

	while (fgets(buf, 256, fp))
	{
		char szPackName[256];
		char szIVFileName[512];
		char szIVFileNameConfig[256];

		sscanf(buf, " %s %s ", szPackName, szIVFileNameConfig);
		snprintf(szIVFileName, sizeof(szIVFileName), "panama/%s", szIVFileNameConfig);

		FILE * fpIV = fopen(szIVFileName, "rb");

		if (!fpIV)
		{
			sys_err("cannot open iv file %s", szIVFileName);
			continue;
		}

		BYTE abIV[32];

		if (32 != fread(abIV, sizeof(BYTE), 32, fpIV))
			sys_err("IV file format error! %s", szIVFileName);
		else
		{
			char szHex[64 + 1];

			for (int i = 0; i < 32; ++i)
				snprintf(szHex + i * 2, sizeof(szHex) - i * 2, "%02x", abIV[i]);

			sys_log(0, "PANAMA: %s %s", szPackName, szHex);

			s_panamaVector.push_back(std::make_pair(szPackName, M2_NEW BYTE[32]));
			memcpy(s_panamaVector[s_panamaVector.size() - 1].second, abIV, 32);
		}

		fclose(fpIV);
	}

	fclose(fp);
	return s_panamaVector.size();
}

void PanamaDestroy()
{
	PanamaVectorType::iterator it = s_panamaVector.begin();

	while (it != s_panamaVector.end()) {
		M2_DELETE_ARRAY(it->second);
		++it;
	}
}

void SendPanamaList(LPDESC d)
{
	TPacketGCPanamaPack pack;

	pack.bHeader = HEADER_GC_PANAMA_PACK;

	PanamaVectorType::iterator it = s_panamaVector.begin();

	while (it != s_panamaVector.end())
	{
		strlcpy(pack.szPackName, it->first.c_str(), sizeof(pack.szPackName));
		memcpy(pack.abIV, it->second, sizeof(pack.abIV));


		DWORD* ivs = (DWORD*)pack.abIV;
		for (int i = 0; i != 32 / sizeof(DWORD); i++)
		{
			ivs[i] ^= d->GetPanamaKey() + i * 16777619;	// ´õºí¿öµå´ÜÀ§·Î º¯ÇüµÈ Æijª¸¶ Å°¸¦ XOR ÇØÁØ´Ù
		}
		++it;

		d->Packet(&pack, sizeof(pack));
	}
}