#include "StdAfx.h"
#include "resource.h"
#include <afxdllx.h>
#include "KTMenuCtrlBar.h"
#include "CheckLock.h"
#include "Cmdpalette.h"

#include "uiiado.h"
#include<Iphlpapi.h> 
#include "afxinet.h"
#include "CoreTools.h"
#include "CmdQueryDlg.h"
#include "initools.h"
#include "ShortcutKeyMappingDlg.h"
#pragma comment(lib, "IPHLPAPI.lib")

HINSTANCE _hdllInstance = NULL;
CCheckLock g_lockChk;

#define g_pszLicenseFileName _T("licenseTime.dat")
#define g_pszMacDbFileName _T("cityStandard.db")


#ifdef NEWUI
CmdPaletteSet g_wndMenuBar;
#else
KTMenuCtrlBar g_wndMenuBar;
#endif

void InitApplication();
void UnloadApplication();

AC_IMPLEMENT_EXTENSION_MODULE(KTMainDLL);

#define TIMER_CUIX 1
void CALLBACK TimerProc(HWND hWnd, UINT nMsg, UINT_PTR nTimerid, DWORD dwTime)
{
	if (nTimerid == TIMER_CUIX)
	{
		void OnMenu();
		OnMenu();
		::KillTimer(hWnd, TIMER_CUIX);
	}
}

extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID /*lpReserved*/)
{
	if (dwReason == DLL_PROCESS_ATTACH)
	{
        _hdllInstance = hInstance;
		KTMainDLL.AttachInstance(hInstance);
		InitAcUiDLL();
	}
	else if (dwReason == DLL_PROCESS_DETACH)
	{
		KTMainDLL.DetachInstance();
	}

	return TRUE;
}

bool SpecialUserPermission()
{
	return true;
}

typedef struct LicenseTime
{
	int nYear;
	int nMonth;
	int nDay;
	int nHour;
	int nMinute;
}LICENSETIME;

void ReadLicenseTime(CTime& tStop)
{
	TCHAR szfileName[MAX_PATH],szFullPath[MAX_PATH];
	GetModuleFileName(_hdllInstance,szfileName,MAX_PATH);

	TCHAR	drive[_MAX_DRIVE] = _T(""),
		dir[_MAX_DIR] = _T(""),
		fname[_MAX_FNAME] = _T(""), 
		ext[_MAX_EXT] = _T("");

	//e:\,program files\autodesk\,myarx,exe
	_tsplitpath(szfileName, drive, dir, fname, ext);
	_tcscpy(szFullPath, drive);
	_tcscat(szFullPath, dir);

	CString sFullPath = szFullPath;
	sFullPath.TrimRight(_T('\\'));

	//上一级目录
	int nPos = sFullPath.ReverseFind(_T('\\'));
	CString sPath = sFullPath.Mid(0,nPos + 1);

	sFullPath = sPath + _T("sys\\");
	sFullPath += _T("licenseTime.dat");

	CFileFind find;
	if (find.FindFile(sFullPath))
	{
		CFile file;
		if (file.Open(sFullPath,CFile::modeRead))
		{
			char szHead[37];
			//strcpy(szHead,"kingtouch license file 1.0");
			file.Read((void*)szHead,sizeof(szHead));

			UINT nSize = 1	// 加密信息(4位随机数+4位加密方法)
				+ sizeof(LicenseTime);

			byte* pData = new byte[nSize];
			ASSERT(NULL != pData);

			file.Read((void*)pData, nSize);
			byte* pBuffer = pData + 1;

			LicenseTime lt;
			Adesk::UInt32 n;
			switch (pData[0] & 0x0F)
			{
			case 0x0a:	// 加密方法0x0a解密:将每个字节高低4位颠倒,然后和加密信息异或
				{
					for (n = 1; n < nSize; n++)
					{
						byte chLow = (pData[n] & 0x0F);
						byte chHigh = (pData[n] & 0xF0);
						byte chNew = ((chLow << 4) | (chHigh >> 4));
						pData[n] = (chNew ^ pData[0]);
					}

					memcpy(&lt, pBuffer, sizeof(LicenseTime));
					CTime tTmp(lt.nYear,lt.nMonth,lt.nDay,lt.nHour,lt.nMinute,0);
					tStop = tTmp;

					//acutPrintf(_T("\n%d %d %d %d %d"),lt.nYear,lt.nMonth,lt.nDay,lt.nHour,lt.nMinute);
				}
			default:
				break;
			}

			file.Close();
		}
	}
}

void GetMacAdr(CStringList& arMacs)
{
	PIP_ADAPTER_INFO pAdapterInfo;  
	PIP_ADAPTER_INFO pAdapter = NULL;  
	DWORD dwRetVal = 0;  
	ULONG ulOutBufLen;  
	pAdapterInfo=(PIP_ADAPTER_INFO)malloc(sizeof(IP_ADAPTER_INFO));  
	ulOutBufLen = sizeof(IP_ADAPTER_INFO);  

	// 第一次调用GetAdapterInfo获取ulOutBufLen大小  
	if (::GetAdaptersInfo( pAdapterInfo, &ulOutBufLen) == ERROR_BUFFER_OVERFLOW)  
	{  
		free(pAdapterInfo);  
		pAdapterInfo = (IP_ADAPTER_INFO *) malloc (ulOutBufLen);   
	}  

	CString sText;
	if ((dwRetVal = ::GetAdaptersInfo( pAdapterInfo, &ulOutBufLen)) == NO_ERROR)
	{  
		pAdapter = pAdapterInfo;  
		while (pAdapter)   
		{  
			sText = pAdapter->Description;
			sText.MakeLower();
			//acutPrintf(_T("\nName=%s"),sText);
			if (sText.Find(_T("virtualbox")) == -1 &&
				sText.Find(_T("vmware")) == -1)//不考虑虚拟机器
			{
				sText.Format(_T("%02x@%02x@%02x@%02x@%02x@%02x"),
					pAdapter->Address[0],  
					pAdapter->Address[1],  
					pAdapter->Address[2],  
					pAdapter->Address[3],  
					pAdapter->Address[4],  
					pAdapter->Address[5]);

				arMacs.AddTail(sText);
			}
			pAdapter = pAdapter->Next;  
		}  
	}
}

//禁止软件变为绿色版的！！！
int DbSetMacOrHasThis()
{
	return 1395; //不限制
#ifdef BAL_REMOTE //集团版不检测绿色
	return 1395;
#endif

#ifdef _DEBUG
	return 1395;
#endif

	TCHAR szfileName[MAX_PATH],szFullPath[MAX_PATH];
	GetModuleFileName(_hdllInstance,szfileName,MAX_PATH);

	TCHAR	drive[_MAX_DRIVE] = _T(""),
		dir[_MAX_DIR] = _T(""),
		fname[_MAX_FNAME] = _T(""), 
		ext[_MAX_EXT] = _T("");

	//e:\,program files\autodesk\,myarx,exe
	_tsplitpath(szfileName, drive, dir, fname, ext);
	_tcscpy(szFullPath, drive);
	_tcscat(szFullPath, dir);

	CString sFullPath = szFullPath;
	sFullPath.TrimRight(_T('\\'));

	//上一级目录
	int nPos = sFullPath.ReverseFind(_T('\\'));
	CString sPath = sFullPath.Mid(0,nPos + 1);

	sFullPath = sPath + _T("sys\\");
	sFullPath += g_pszMacDbFileName;

	::CoInitialize(NULL);
	CUIISQLiteServer sqldb;
	if (!sqldb.InitADOServer(sFullPath))
	{
		return -1;
	}

	CStringList listMac;
	GetMacAdr(listMac);

	CString sSQL,sMac,sText;
	sSQL = _T("select cptpos from aec_position");

	CStringArray arVals;
	sqldb.GetSelData(sSQL,arVals);

	if (arVals.IsEmpty())
	{
		for (POSITION ps = listMac.GetHeadPosition();ps != NULL;)
		{
			sMac = listMac.GetNext(ps);
			sSQL.Format(_T("insert into aec_position (cptpos) values ('%s')"),sMac);
			if ( !sqldb.ExecuteSQL(sSQL) )
			{
				return -19;
			}
		}
		return 1395;
	}
	else
	{
		for (int i = 0;i < arVals.GetSize();i++)
		{
			sText = arVals.GetAt(i);
			if ( listMac.Find(sText) != NULL)
			{
				return 2396;
			}
		}
	}
	return 19;
}

static bool g_bHasDog = true;
extern "C" AcRx::AppRetCode
acrxEntryPoint(AcRx::AppMsgCode msg, void* pkt)
{
	switch (msg) 
	{
	case AcRx::kLoadDwgMsg:
	{
		// 保证每次打开文档的时候，命令行会打开
		if (::IsWindow(g_wndMenuBar.m_hWnd) && !g_wndMenuBar.IsVisible())
			g_wndMenuBar.m_pDockSite->ShowControlBar(&g_wndMenuBar, TRUE, FALSE);
	}
	case AcRx::kInitAppMsg:
		{
			acrxDynamicLinker->unlockApplication(pkt);
			acrxDynamicLinker->registerAppMDIAware(pkt);

			int nDbSetMacOrHasThis = DbSetMacOrHasThis();

			//if (!g_lockChk.CheckLock())
			if (false)
			{
				CTime tStop(2020,5,1,2,2,2);
				ReadLicenseTime(tStop);

				CTime tCur = CTime::GetCurrentTime();
				if (tCur < tStop)
				{
					if (nDbSetMacOrHasThis > 1394)
					{
						InitApplication();
					}
				}
				else
				{
					CString sTip;
					sTip.LoadString(IDS_LOCK_CHKERROR);
					acedAlert(sTip);
					g_bHasDog = false;
					return AcRx::kRetError;
				}
			}
			else
			{
				if (nDbSetMacOrHasThis > 1394)
				{
					InitApplication();
				}
				else
				{
					AfxMessageBox(_T("检查授权失败!"));
				}
			}
			break;
		}
	case AcRx::kUnloadAppMsg:
	{
		g_lockChk.SendLogoutMessage();
		if (g_bHasDog)
			UnloadApplication();
		else
			return AcRx::kRetError;
	}
		break;
	}
	return AcRx::kRetOK;
}

void OnMain()
{

	// 在这里检查一下 "GP" 命令有没有注册，如果没有，退出
	AcEdCommandStruc st;
	Adesk::Boolean hasGP = acedCmdLookup(_T("GP"), Adesk::kFalse, &st);
	if (!hasGP)
		return;

	if (::IsWindow(g_wndMenuBar.m_hWnd))
	{
		if (!g_wndMenuBar.IsVisible())
			g_wndMenuBar.m_pDockSite->ShowControlBar(&g_wndMenuBar, TRUE, FALSE);
		CString	ss;
		g_wndMenuBar.GetWindowText(ss);
	}
	else
	{
		std::map<CString, CString> mm;
		IniTools::getSectionContent(FilePath::GetSysDir() + _T("\\product.ini")
			, _T("Product")
			, mm);
		CString sNameOfPalette = _T("上华软件");
		if (mm.find(_T("CName")) != mm.end() && mm.find(_T("Version")) != mm.end())
		{
			sNameOfPalette = mm[_T("CName")] + _T(" ") + mm[_T("Version")];
		}
		#ifdef NEWUI
		g_wndMenuBar.CreatePaletteSet(sNameOfPalette);
#else
		g_wndMenuBar.CreateMenuBar(sNameOfPalette);
		g_wndMenuBar.SetName(sNameOfPalette);
#endif

		if (::IsWindow(g_wndMenuBar.m_hWnd))
		{
			g_wndMenuBar.m_pDockSite->ShowControlBar(&g_wndMenuBar, TRUE, FALSE);
			g_wndMenuBar.RestoreControlBar(AFX_IDW_DOCKBAR_RIGHT);
		}
	}
}

void OnMenu()
{
	CString sys(FilePath::GetSysDir()), strIni(_T("")), strCuix(_T(""));
	strIni = sys + _T("\\product.ini");
	CString product(_T(""));
	if (_taccess(strIni, 0x0) == 0x0)
	{
		TCHAR buf[MAX_PATH + 1] = { 0, };
		::GetPrivateProfileString(_T("Product"), _T("ProfilesName"), _T(""), buf, MAX_PATH, strIni);
		product = buf;
	}

	if (product.Find(_T("PP3D")) != -1 || product.Find(_T("ENVI")) != -1)
	{
#ifndef ZWCAD
		strCuix = sys + _T("\\shs-ew.cuix");
		if (!acedIsMenuGroupLoaded(_T("shs-ew")))
			SHads_command(RTSTR, _T(".MENULOAD"), RTSTR, strCuix, RTNONE);
		else
			SHads_command(RTSTR, _T(".MENUUNLOAD"), RTSTR, _T("shs-ew"), RTNONE);
#else
		strCuix = sys + _T("\\shs-zw.cuix");
		if (!acedIsMenuGroupLoaded(_T("shs-zw")))
			SHads_command(RTSTR, _T(".MENULOAD"), RTSTR, strCuix, RTNONE);
		else
			SHads_command(RTSTR, _T(".MENUUNLOAD"), RTSTR, _T("shs-zw"), RTNONE);
#endif
	}
}

void CMDSearch()
{
	CAcModuleResourceOverride myResources;
	if (!::IsWindow(g_CmdQueryDlg::instance().m_hWnd))
	{
		g_CmdQueryDlg::instance().Create(CmdQueryDlg::IDD, acedGetAcadFrame());
		g_CmdQueryDlg::instance().CenterWindow();
	}

	if (!g_CmdQueryDlg::instance().IsWindowVisible())
		g_CmdQueryDlg::instance().ShowWindow(SW_SHOW);

	g_CmdQueryDlg::instance().initShow();
}

void KJJ()
{
	CShortcutKeyMappingDlg::start();
}

// commands, reactors...
void InitApplication()
{
	CoreTools::AddCommand(_T("DHMENU"), _T("GP"), _T("GP"), ACRX_CMD_TRANSPARENT | ACRX_CMD_USEPICKSET, OnMain);
	CoreTools::AddCommand(_T("DHMENU"), _T("GPM"), _T("GPM"), ACRX_CMD_TRANSPARENT | ACRX_CMD_USEPICKSET, OnMenu);
	CoreTools::AddCommand(_T("DHMENU"), _T("CMDS"), _T("CMDS"), ACRX_CMD_TRANSPARENT | ACRX_CMD_USEPICKSET, CMDSearch);
	CoreTools::AddCommand(_T("DHMENU"), _T("CO_KJJ"), _T("CO_KJJ"), ACRX_CMD_TRANSPARENT | ACRX_CMD_USEPICKSET, KJJ);
	OnMain();
	::SetTimer(acedGetAcadFrame()->GetSafeHwnd(), TIMER_CUIX, 3000, TimerProc);
}

void UnloadApplication()
{
	acedRegCmds->removeGroup(_T("DHMENU"));
}