#include "commonheaders.h"

int hLangpack = 0;

BOOL APIENTRY DllMain(HINSTANCE hInst, DWORD dwReason, LPVOID)
{
	g_hInst = hInst;
	if (dwReason == DLL_PROCESS_ATTACH) {
		INITCOMMONCONTROLSEX icce = { sizeof(icce), ICC_LISTVIEW_CLASSES | ICC_TAB_CLASSES };
		InitCommonControlsEx(&icce);
	}
	return TRUE;
}

extern "C" __declspec(dllexport) PLUGININFOEX* MirandaPluginInfoEx(DWORD mirandaVersion)
{
	return &pluginInfoEx;
}

HANDLE AddMenuItem(LPCSTR name,int pos,HICON hicon,LPCSTR service,int flags=0,WPARAM wParam=0)
{
	CLISTMENUITEM mi = { sizeof(mi) };
	mi.flags = flags | CMIF_HIDDEN;
	mi.position = pos;
	mi.hIcon = hicon;
	mi.pszName = (char*)name;
	mi.pszPopupName = (char*)-1;
	mi.pszService = (char*)service;
	return Menu_AddContactMenuItem(&mi);
}

HANDLE AddSubItem(HANDLE rootid,LPCSTR name,int pos,int poppos,LPCSTR service,WPARAM wParam=0)
{
	CLISTMENUITEM mi = { sizeof(mi) };
	mi.flags = CMIF_CHILDPOPUP | CMIF_HIDDEN;
	mi.position = pos;
	mi.popupPosition = poppos;
	mi.hIcon = NULL;
	mi.pszName = (char*)name;
	mi.pszPopupName = (char*)rootid;
	mi.pszService = (char*)service;
	return Menu_AddContactMenuItem(&mi);
}

extern "C" __declspec(dllexport) int __cdecl Load(void)
{
	mir_getLP(&pluginInfoEx);

	DisableThreadLibraryCalls(g_hInst);
	InitializeCriticalSection(&localQueueMutex);

	char temp[MAX_PATH];
	GetTempPath(sizeof(temp),temp);
	GetLongPathName(temp,TEMP,sizeof(TEMP));
	TEMP_SIZE = (int)strlen(TEMP);
	if (TEMP[TEMP_SIZE-1] == '\\') {
		TEMP_SIZE--;
		TEMP[TEMP_SIZE]='\0';
	}

	// check for support TrueColor Icons
	BOOL bIsComCtl6 = FALSE;
	HMODULE hComCtlDll = LoadLibrary(_T("comctl32.dll"));
	if (hComCtlDll) {
		typedef HRESULT (CALLBACK *PFNDLLGETVERSION)(DLLVERSIONINFO*);
		PFNDLLGETVERSION pfnDllGetVersion = (PFNDLLGETVERSION) GetProcAddress(hComCtlDll,"DllGetVersion");
		if (pfnDllGetVersion) {
			DLLVERSIONINFO dvi;
			memset(&dvi,0,sizeof(dvi));
			dvi.cbSize = sizeof(dvi);
			HRESULT hRes = (*pfnDllGetVersion)(&dvi);
			if (SUCCEEDED(hRes) && dvi.dwMajorVersion >= 6) {
				bIsComCtl6 = TRUE;
			}
		}
		FreeLibrary(hComCtlDll);
	}
	if (bIsComCtl6)	iBmpDepth = ILC_COLOR32 | ILC_MASK;  // 32-bit images are supported
	else		iBmpDepth = ILC_COLOR24 | ILC_MASK;

	iCoreVersion = CallService(MS_SYSTEM_GETVERSION,0,0);

	// load crypo++ dll
	if (!loadlib()) {
		msgbox1(0,sim107,MODULENAME,MB_OK|MB_ICONSTOP);
		return 1;
	}

	load_rtfconv();

	// register plugin module
	PROTOCOLDESCRIPTOR pd = { sizeof(pd) };
	pd.szName = (char*)MODULENAME;
	pd.type = PROTOTYPE_ENCRYPTION;
	CallService(MS_PROTO_REGISTERMODULE, 0, (LPARAM)&pd);

	// hook events
	HookEvent(ME_SYSTEM_MODULESLOADED, onModulesLoaded);
	HookEvent(ME_SYSTEM_OKTOEXIT, onSystemOKToExit);
	HookEvent(ME_SYSTEM_MODULELOAD, ModuleLoad);
	HookEvent(ME_SYSTEM_MODULEUNLOAD, ModuleLoad);

	g_hEvent[0] = CreateHookableEvent(MODULENAME"/Disabled");
	g_hEvent[1] = CreateHookableEvent(MODULENAME"/Established");

	CreateServiceFunction(MODULENAME"/IsContactSecured",Service_IsContactSecured);
	CreateServiceFunction(MODULENAME"/SIM_EST",Service_CreateIM);
	CreateServiceFunction(MODULENAME"/SIM_DIS",Service_DisableIM);
	CreateServiceFunction(MODULENAME"/SIM_ST_DIS",Service_StatusDis);
	CreateServiceFunction(MODULENAME"/SIM_ST_ENA",Service_StatusEna);
	CreateServiceFunction(MODULENAME"/SIM_ST_TRY",Service_StatusTry);
	CreateServiceFunction(MODULENAME"/PGP_SET",Service_PGPsetKey);
	CreateServiceFunction(MODULENAME"/PGP_DEL",Service_PGPdelKey);
	CreateServiceFunction(MODULENAME"/GPG_SET",Service_GPGsetKey);
	CreateServiceFunction(MODULENAME"/GPG_DEL",Service_GPGdelKey);
	CreateServiceFunction(MODULENAME"/MODE_NAT",Service_ModeNative);
	CreateServiceFunction(MODULENAME"/MODE_PGP",Service_ModePGP);
	CreateServiceFunction(MODULENAME"/MODE_GPG",Service_ModeGPG);
	CreateServiceFunction(MODULENAME"/MODE_RSA",Service_ModeRSAAES);
	return 0;
}

extern "C" __declspec(dllexport) int __cdecl Unload()
{
	DeleteCriticalSection(&localQueueMutex);
	return 0;
}

int ModuleLoad(WPARAM wParam, LPARAM lParam)
{
	bPopupExists = ServiceExists(MS_POPUP_ADDPOPUP);
	bMetaContacts = ServiceExists(MS_MC_GETMETACONTACT);
	return 0;
}

int onModulesLoaded(WPARAM wParam, LPARAM lParam)
{
	InitNetlib();

	g_hFolders = FoldersRegisterCustomPathT(LPGEN("SecureIM"), LPGEN("Icons"), _T(MIRANDA_PATH"\\icons"));

	InitIcons();
	GetFlags();
	ModuleLoad(0, 0);

	// RSA/AES
	Sent_NetLog("rsa_init");
	rsa_init(&exp, &imp);

	DBVARIANT dbv;
	dbv.type = DBVT_BLOB;

	if (DBGetContactSetting(0,MODULENAME,"rsa_priv",&dbv) == 0) {
		exp->rsa_set_keypair(CPP_MODE_RSA_4096,dbv.pbVal,dbv.cpbVal);
		db_free(&dbv);
		rsa_4096=1;
	}
	else if (DBGetContactSetting(0,MODULENAME,"rsa_priv_4096",&dbv) == 0) {
		exp->rsa_set_keypair(CPP_MODE_RSA_4096|CPP_MODE_RSA_BER,dbv.pbVal,dbv.cpbVal);
		db_free(&dbv);

		char priv_key[4096]; int priv_len;
		char pub_key[4096]; int pub_len;

		DBCONTACTWRITESETTING cws;
		cws.szModule = MODULENAME;
		cws.value.type = DBVT_BLOB;

		exp->rsa_get_keypair(CPP_MODE_RSA_4096,(PBYTE)&priv_key,&priv_len,(PBYTE)&pub_key,&pub_len);

		cws.szSetting = "rsa_priv";
		cws.value.pbVal = (PBYTE)&priv_key;
		cws.value.cpbVal = priv_len;
		CallService(MS_DB_CONTACT_WRITESETTING, 0, (LPARAM)&cws);

		cws.szSetting = "rsa_pub";
		cws.value.pbVal = (PBYTE)&pub_key;
		cws.value.cpbVal = pub_len;
		CallService(MS_DB_CONTACT_WRITESETTING, 0, (LPARAM)&cws);

		db_unset(0, MODULENAME, "rsa_priv_2048");
		db_unset(0, MODULENAME, "rsa_pub_2048");
		rsa_4096 = 1;
	}

	if (!rsa_4096)
		mir_forkthread(sttGenerateRSA, NULL);

	exp->rsa_set_timeout( db_get_w(0,MODULENAME,"ket",10));

	Sent_NetLog("pgp_init");

	bPGP = db_get_b(0, MODULENAME, "pgp", 0);
	if (bPGP) { //PGP
		bPGPloaded = pgp_init();
		bUseKeyrings = db_get_b(0,MODULENAME,"ukr",1);
		LPSTR priv = myDBGetStringDecode(0,MODULENAME,"pgpPrivKey");
		if (priv) {
			bPGPprivkey = true;
			if (bPGPloaded)
				pgp_set_priv_key(priv);
			mir_free(priv);
		}// if (priv)
		if (bPGPloaded && bUseKeyrings) {
			char PubRingPath[MAX_PATH], SecRingPath[MAX_PATH];
			PubRingPath[0]='\0'; SecRingPath[0]='\0';
			if (pgp_get_version()<0x02000000) { // 6xx
				bPGPkeyrings = pgp_open_keyrings(PubRingPath,SecRingPath);
			}
			else {
				LPSTR tmp = db_get_sa(0,MODULENAME,"pgpPubRing");
				if (tmp) {
					strncpy(PubRingPath,tmp,sizeof(PubRingPath));
					mir_free(tmp);
				}
				if (tmp = db_get_sa(0,MODULENAME,"pgpSecRing")) {
					strncpy(SecRingPath,tmp,sizeof(SecRingPath));
					mir_free(tmp);
				}
				if (PubRingPath[0] && SecRingPath[0]) {
					bPGPkeyrings = pgp_open_keyrings(PubRingPath,SecRingPath);
					if (bPGPkeyrings) {
						db_set_s(0,MODULENAME,"pgpPubRing",PubRingPath);
						db_set_s(0,MODULENAME,"pgpSecRing",SecRingPath);
					}
					else {
						db_unset(0, MODULENAME, "pgpPubRing");
						db_unset(0, MODULENAME, "pgpSecRing");
					}
				}
			}
		}// if (bPGPloaded && bUseKeyrings)
	}// if (bPGP)

	Sent_NetLog("gpg_init");

	bGPG = db_get_b(0, MODULENAME, "gpg", 0);
	if (bGPG) { //GPG
		bGPGloaded = gpg_init();

		char gpgexec[MAX_PATH], gpghome[MAX_PATH];
		gpgexec[0]='\0'; gpghome[0]='\0';

		LPSTR tmp = db_get_sa(0,MODULENAME,"gpgExec");
		if (tmp) {
			strncpy(gpgexec,tmp,sizeof(gpgexec));
			mir_free(tmp);
		}
		if (tmp = db_get_sa(0,MODULENAME,"gpgHome")) {
			strncpy(gpghome,tmp,sizeof(gpghome));
			mir_free(tmp);
		}

		if (db_get_b(0, MODULENAME, "gpgLogFlag",0)) {
			if (tmp = db_get_sa(0,MODULENAME,"gpgLog")) {
				gpg_set_log(tmp);
				mir_free(tmp);
			}
		}

		if (db_get_b(0, MODULENAME, "gpgTmpFlag",0)) {
			if (tmp = db_get_sa(0,MODULENAME,"gpgTmp")) {
				gpg_set_tmp(tmp);
				mir_free(tmp);
			}
		}

		bGPGkeyrings = gpg_open_keyrings(gpgexec,gpghome);
		if (bGPGkeyrings) {
			db_set_s(0,MODULENAME,"gpgExec",gpgexec);
			db_set_s(0,MODULENAME,"gpgHome",gpghome);
		}
		else {
			db_unset(0, MODULENAME, "gpgExec");
			db_unset(0, MODULENAME, "gpgHome");
		}

		bSavePass = db_get_b(0,MODULENAME,"gpgSaveFlag",0);
		if (bSavePass) {
			if (tmp = db_get_sa(0,MODULENAME,"gpgSave")) {
				gpg_set_passphrases(tmp);
				mir_free(tmp);
			}
		}
	}

	Sent_NetLog("loadContactList");
	loadContactList();

	// add new skin sound
	SkinAddNewSound("IncomingSecureMessage",LPGEN("Incoming Secure Message"),"Sounds\\iSecureMessage.wav");
	SkinAddNewSound("OutgoingSecureMessage",LPGEN("Outgoing Secure Message"),"Sounds\\oSecureMessage.wav");

	// init extra icons
	for (int i=0; i < SIZEOF(g_IEC); i++)
		g_IEC[i] = (HANDLE)-1;

	HookEvent(ME_CLIST_PREBUILDCONTACTMENU, onRebuildContactMenu);

	g_hCLIcon = ExtraIcon_Register(MODULENAME, LPGEN("SecureIM status"), "sim_cm_est", onExtraImageListRebuilding, onExtraImageApplying);

	// hook init options
	HookEvent(ME_OPT_INITIALISE, onRegisterOptions);
	if (bPopupExists)
		HookEvent(ME_OPT_INITIALISE, onRegisterPopOptions);
	HookEvent(ME_PROTO_ACK, onProtoAck);
	HookEvent(ME_DB_CONTACT_SETTINGCHANGED, onContactSettingChanged);
	HookEvent(ME_DB_CONTACT_ADDED, onContactAdded);
	HookEvent(ME_DB_CONTACT_DELETED, onContactDeleted);

	// hook message transport
	CreateProtoServiceFunction(MODULENAME, PSR_MESSAGE, onRecvMsg);
	CreateProtoServiceFunction(MODULENAME, PSS_MESSAGE, onSendMsg);
	CreateProtoServiceFunction(MODULENAME, PSS_MESSAGE"W", onSendMsgW);
	CreateProtoServiceFunction(MODULENAME, PSS_FILE, onSendFile);

	// create a menu item for creating a secure im connection to the user.
	g_hMenu[0] = AddMenuItem(sim301, 110000, g_hICO[ICO_CM_EST], MODULENAME"/SIM_EST", CMIF_NOTOFFLINE);
	g_hMenu[1] = AddMenuItem(sim302, 110001, g_hICO[ICO_CM_DIS], MODULENAME"/SIM_DIS", CMIF_NOTOFFLINE);

	if (ServiceExists(MS_CLIST_MENUBUILDSUBGROUP)) {
		g_hMenu[2] = AddMenuItem(sim312[0], 110002, NULL, NULL, CMIF_ROOTPOPUP);
		g_hMenu[3] = AddSubItem(g_hMenu[2], sim232[0], 110003, 110002, MODULENAME"/SIM_ST_DIS");
		g_hMenu[4] = AddSubItem(g_hMenu[2], sim232[1], 110004, 110002, MODULENAME"/SIM_ST_ENA");
		g_hMenu[5] = AddSubItem(g_hMenu[2], sim232[2], 110005, 110002, MODULENAME"/SIM_ST_TRY");
	}
	else {
		g_hMenu[2] = 0;
		g_hMenu[3] = AddMenuItem(sim232[0], 110003, NULL, MODULENAME"/SIM_ST_DIS");
		g_hMenu[4] = AddMenuItem(sim232[1], 110004, NULL, MODULENAME"/SIM_ST_ENA");
		g_hMenu[5] = AddMenuItem(sim232[2], 110005, NULL, MODULENAME"/SIM_ST_TRY");
	}

	if (bPGPloaded) {
		g_hMenu[6] = AddMenuItem(sim306, 110006, mode2icon(MODE_PGP|SECURED,2), MODULENAME"/PGP_SET", 0);
		g_hMenu[7] = AddMenuItem(sim307, 110007, mode2icon(MODE_PGP,2), MODULENAME"/PGP_DEL", 0);
	}

	if (bGPGloaded) {
		g_hMenu[8] = AddMenuItem(sim308, 110008, mode2icon(MODE_GPG|SECURED,2), MODULENAME"/GPG_SET", 0);
		g_hMenu[9] = AddMenuItem(sim309, 110009, mode2icon(MODE_GPG,2), MODULENAME"/GPG_DEL", 0);
	}

	if (ServiceExists(MS_CLIST_MENUBUILDSUBGROUP)) {
		g_hMenu[10] = AddMenuItem(sim311[0],110010,NULL,NULL,CMIF_ROOTPOPUP);
		g_hMenu[11] = AddSubItem(g_hMenu[10],sim231[0],110011,110010,MODULENAME"/MODE_NAT");
		g_hMenu[12] = AddSubItem(g_hMenu[10],sim231[1],110012,110010,MODULENAME"/MODE_PGP");
		g_hMenu[13] = AddSubItem(g_hMenu[10],sim231[2],110013,110010,MODULENAME"/MODE_GPG");
		g_hMenu[14] = AddSubItem(g_hMenu[10],sim231[3],110014,110010,MODULENAME"/MODE_RSA");
	}
	else {
		g_hMenu[10] = 0;
		g_hMenu[11] = AddMenuItem(sim231[0],110011,NULL,MODULENAME"/MODE_NAT");
		g_hMenu[12] = AddMenuItem(sim231[1],110012,NULL,MODULENAME"/MODE_PGP");
		g_hMenu[13] = AddMenuItem(sim231[2],110013,NULL,MODULENAME"/MODE_GPG");
		g_hMenu[14] = AddMenuItem(sim231[3],110014,NULL,MODULENAME"/MODE_RSA");
	}

	// add icon to srmm status icons
	if (ServiceExists(MS_MSG_ADDICON)) {
		StatusIconData sid = { sizeof(sid) };
		sid.szModule = (char*)MODULENAME;
		sid.flags = MBF_DISABLED|MBF_HIDDEN;
		// Native
		sid.dwId = MODE_NATIVE;
		sid.hIcon = mode2icon(MODE_NATIVE|SECURED,3);
		sid.hIconDisabled = mode2icon(MODE_NATIVE,3);
		sid.szTooltip = Translate("SecureIM [Native]");
		CallService(MS_MSG_ADDICON, 0, (LPARAM)&sid);
		// PGP
		sid.dwId = MODE_PGP;
		sid.hIcon = mode2icon(MODE_PGP|SECURED,3);
		sid.hIconDisabled = mode2icon(MODE_PGP,3);
		sid.szTooltip = Translate("SecureIM [PGP]");
		CallService(MS_MSG_ADDICON, 0, (LPARAM)&sid);
		// GPG
		sid.dwId = MODE_GPG;
		sid.hIcon = mode2icon(MODE_GPG|SECURED,3);
		sid.hIconDisabled = mode2icon(MODE_GPG,3);
		sid.szTooltip = Translate("SecureIM [GPG]");
		CallService(MS_MSG_ADDICON, 0, (LPARAM)&sid);
		// RSAAES
		sid.dwId = MODE_RSAAES;
		sid.hIcon = mode2icon(MODE_RSAAES|SECURED,3);
		sid.hIconDisabled = mode2icon(MODE_RSAAES,3);
		sid.szTooltip = Translate("SecureIM [RSA/AES]");
		CallService(MS_MSG_ADDICON, 0, (LPARAM)&sid);

		// hook the window events so that we can can change the status of the icon
		HookEvent(ME_MSG_WINDOWEVENT, onWindowEvent);
		HookEvent(ME_MSG_ICONPRESSED, onIconPressed);
	}

	return 0;
}

int onSystemOKToExit(WPARAM wParam, LPARAM lParam)
{
	if (bSavePass) {
		LPSTR tmp = gpg_get_passphrases();
		db_set_s(0,MODULENAME,"gpgSave",tmp);
		LocalFree(tmp);
	}
	else db_unset(0,MODULENAME,"gpgSave");

	if (bPGPloaded) pgp_done();
	if (bGPGloaded) gpg_done();
	rsa_done();

	DestroyHookableEvent(g_hEvent[0]);
	DestroyHookableEvent(g_hEvent[1]);

	freeContactList();
	free_rtfconv();

	DeinitNetlib();
	return 0;
}

// EOF