#include "StdAfx.h"
#include "FileUploadControl.h"
#include "ProcessController.h"
//#include "AOIDefinition.h"

#define		FTPUPLOADER_NAME		_T("FTPUploader.exe")
#define		COPYMANAGER_NAME		_T("CopyManager.exe")
#define		FTP_HOME_PATH_LTPS		_T("ltps/")
#define		FTP_HOME_PATH_LTPS01	_T("ltps01/")
#define		COPYDATA_FTP_ADD_FILE	27				// "AOIDefinition.h"

CFileUploadControl::CFileUploadControl(void)
{
	InitializeCriticalSection(&m_csImgUpload);
}

CFileUploadControl::~CFileUploadControl(void)
{
	DeleteCriticalSection(&m_csImgUpload);
}

void CFileUploadControl::RestartFTPUploader(CString strFilePath)
{
	KillProcess(FTPUPLOADER_NAME);
	Sleep(100);
	CString strPath;
	strPath.Format(_T("%s\\%s"),strFilePath,FTPUPLOADER_NAME);

	int nSize = 0;
	char* strTemp = NULL;
#ifdef _UNICODE
	nSize = WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, NULL, 0, NULL,NULL); 
	strTemp = new char[nSize]; 
	WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, strTemp, nSize, NULL, NULL); 
#else
	nSize = strPath.GetLength();
	strTemp = new char[nSize+1]; 
	memcpy(strTemp, strPath.GetBuffer(), sizeof(char)*nSize);
	strTemp[nSize] = NULL;
#endif

	WinExec(strTemp, SW_HIDE);

	if(strTemp)
	{
		delete[] strTemp;
		strTemp = NULL;
	}
}

void CFileUploadControl::RestartCopyManager(CString strFilePath)
{
	KillProcess(COPYMANAGER_NAME);
	Sleep(100);
	CString strPath;
	strPath.Format(_T("%s\\%s"),strFilePath,COPYMANAGER_NAME);
	
	int nSize = 0;
	char* strTemp = NULL;
#ifdef _UNICODE
	nSize = WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, NULL, 0, NULL,NULL); 
	strTemp = new char[nSize]; 
	WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, strTemp, nSize, NULL, NULL); 
#else
	nSize = strPath.GetLength();
	strTemp = new char[nSize+1]; 
	memcpy(strTemp, strPath.GetBuffer(), sizeof(char)*nSize);
	strTemp[nSize] = NULL;
#endif

	WinExec(strTemp, SW_HIDE);

	if(strTemp)
	{
		delete[] strTemp;
		strTemp = NULL;
	}
}

void CFileUploadControl::StartFTPUploader(CString strFilePath)
{
	CString strPath;
	strPath.Format(_T("%s\\%s"),strFilePath,FTPUPLOADER_NAME);
	
	int nSize = 0;
	char* strTemp = NULL;
#ifdef _UNICODE
	nSize = WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, NULL, 0, NULL,NULL); 
	strTemp = new char[nSize]; 
	WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, strTemp, nSize, NULL, NULL); 
#else
	nSize = strPath.GetLength();
	strTemp = new char[nSize+1]; 
	memcpy(strTemp, strPath.GetBuffer(), sizeof(char)*nSize);
	strTemp[nSize] = NULL;
#endif
	
	WinExec(strTemp, SW_HIDE);

	if(strTemp)
	{
		delete[] strTemp;
		strTemp = NULL;
	}
}

void CFileUploadControl::StartCopyManager(CString strFilePath)
{
	CString strPath;
	strPath.Format(_T("%s\\%s"),strFilePath,COPYMANAGER_NAME);

	int nSize = 0;
	char* strTemp = NULL;
#ifdef _UNICODE
	nSize = WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, NULL, 0, NULL,NULL); 
	strTemp = new char[nSize]; 
	WideCharToMultiByte(CP_ACP, 0, strPath.GetBuffer(), -1, strTemp, nSize, NULL, NULL); 
#else
	nSize = strPath.GetLength();
	strTemp = new char[nSize+1]; 
	memcpy(strTemp, strPath.GetBuffer(), sizeof(char)*nSize);
	strTemp[nSize] = NULL;
#endif

	WinExec(strTemp, SW_HIDE);

	if(strTemp)
	{
		delete[] strTemp;
		strTemp = NULL;
	}
}

BOOL CFileUploadControl::KillProcess(CString strProcess)
{
	const int MaxNum = 100;
	int count = 0;
	PROCESSLIST ProcessList[MaxNum];
	CProcessController		ProcessController;

	count = ProcessController.GetProcessList(ProcessList, MaxNum);

	for (int i=0; i<count; i++)
	{
		if (ProcessList[i].ProcessName.CompareNoCase(strProcess)==0)
		{
			HANDLE hToken;
			TOKEN_PRIVILEGES tkp;

			// Get a token for this process.
			if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
				return FALSE;

			// Get the LUID for the shutdown privilege.
			LookupPrivilegeValue(NULL, SE_SECURITY_NAME, &tkp.Privileges[0].Luid);

			tkp.PrivilegeCount = 1;  // one privilege to set
			tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

			// Get the shutdown privilege for this process.
			AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);

			if (GetLastError() != ERROR_SUCCESS)
				return FALSE;

			ProcessController.TerminateProcessByName(ProcessList[i].ProcessName);
		}
	}

	return TRUE;
}

// BOOL CFileUploadControl::MakeResultInfoByCopyManager(CString strRawFolderPath, CString strHostFolderPath, BOOL bFTPUpload)
// {
// 	HWND hWnd;
// 	BOOL bRet = FALSE;
// 
// 	COPYDATASTRUCT copydata = {0,0,0};
// 
// 	CCommendParam *pParam = NULL;
// 
// 	//FTPUpLoad�̸� ��ȯ
// 	if( bFTPUpload )
// 		return FALSE;
// 
// 	hWnd = ::FindWindow(NULL, _T("CopyManager"));
// 
// 	pParam = new CCommendParam();
// 
// 	pParam->m_eCommend = eMakeResultInfo;
// 
// 	sprintf_s(pParam->m_strRawFileFolder,sizeof(char)*255,"%s",(LPSTR)(LPCTSTR)strRawFolderPath);
// 	sprintf_s(pParam->m_strImageFileFolder,sizeof(char)*255,"%s",(LPSTR)(LPCTSTR)strHostFolderPath);
// 
// 	copydata.dwData = eMakeResultInfo;
// 	copydata.cbData = sizeof(CCommendParam);
// 	copydata.lpData = pParam;
// 
// 	DWORD dwReturn = 0;
// 	DWORD dwError = 0;
// 
// 	if (NULL == hWnd)
// 	{
// 		g_pLog->DisplayMessage(_T("Find CopyManager Program : Fail.."));
// 		StartCopyManager(_T("C:\\AOIServer"));
// 	}
// 	else
// 	{
// 		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 5000, (PDWORD_PTR)(dwReturn) ) == TRUE )
// 		{
// 			g_pLog->DisplayMessage(_T("Send Data to CopyManager : Success."));
// 			bRet = TRUE;
// 		}
// 		else
// 		{
// 			g_pLog->DisplayMessage(_T("Send Data to CopyManager : Fail."));
// 			bRet = FALSE;
// 		}
// 	}
// 
// 	g_pLog->DisplayMessage(_T("Raw[%s]"), pParam->m_strRawFileFolder);
// 	g_pLog->DisplayMessage(_T("Image[%s]"),  pParam->m_strImageFileFolder);
// 
// 	delete pParam;
// 
// 	return TRUE;
// }

// BOOL CFileUploadControl::UploadFileByFTPUploader(CCommendParam &Param)
// {
// 	HWND hWnd;
// 	BOOL bRet = FALSE;
// 
// 	COPYDATASTRUCT copydata = {0,0,0};
// 
// 	hWnd = ::FindWindow(NULL, _T("CopyManager"));
// 
// 	copydata.dwData = eFileCopy;
// 	copydata.cbData = sizeof(CCommendParam);
// 	copydata.lpData = &Param;
// 
// 	DWORD dwReturn = 0;
// 	DWORD dwError = 0;
// 
// 	if (NULL == hWnd)
// 	{
// 		g_pLog->DisplayMessage(_T("Find CopyManager Program : Fail."));
// 		StartCopyManager(_T("C:\\AOIServer"));
// 	}
// 	else
// 	{
// 		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 60000, (PDWORD_PTR)(dwReturn) ) == TRUE )
// 		{
// 			g_pLog->DisplayMessage(_T("Send Data to CopyManager : Success."));
// 			bRet = TRUE;
// 		}
// 		else
// 		{
// 			g_pLog->DisplayMessage(_T("Send Data to CopyManager : Fail."));
// 			RestartCopyManager(_T("C:\\AOIServer")); //A2E FTPUploader
// 			bRet = FALSE;
// 		}
// 	}
// 
// 	g_pLog->DisplayMessage(_T("Local[%s]"), Param.m_strOriginPath);
// 	g_pLog->DisplayMessage(_T("Host[%s]"),  Param.m_strTargetPath);
// 
// 	return bRet;
// }

BOOL CFileUploadControl::UploadFileByFTPUploader(CString strLocalFilePath, CString strHostSubFolder, BOOL bDelete)
{
	HWND hWnd;
	BOOL bRet = FALSE;

	COPYDATASTRUCT copydata = {0,0,0};

	CFTPParamCopyData Param;

	hWnd = ::FindWindow(NULL, _T("FTPUploader"));

	Param.bUpload=TRUE;
	Param.bDelete = bDelete;
	Param.bMakeResultFile = FALSE;

#ifdef _UNICODE
	WideCharToMultiByte(CP_ACP, 0, strLocalFilePath, -1, Param.strLocalFilePath, strLocalFilePath.GetLength(), NULL, NULL); 
	WideCharToMultiByte(CP_ACP, 0, strHostSubFolder, -1, Param.strHostSubFolder, strHostSubFolder.GetLength(), NULL, NULL); 
#else
	sprintf_s(Param.strLocalFilePath,sizeof(char)*255,"%s",strLocalFilePath);
	sprintf_s(Param.strHostSubFolder,sizeof(char)*255,"%s",strHostSubFolder);
#endif

	copydata.dwData = COPYDATA_FTP_ADD_FILE;
	copydata.cbData = sizeof(CFTPParamCopyData);
	copydata.lpData = &Param;

	DWORD dwReturn = 0;
	DWORD dwError = 0;

	if (NULL == hWnd)
	{
		g_pLog->DisplayMessage(_T("Find FTPUploader Program : Fail."));

		StartFTPUploader(_T("D:\\DIT_Review"));  //210805
	}
	else
	{
		//sprintf_s(Param.strHostSubFolder,sizeof(char)*255,"%s%s",FTP_HOME_PATH_LTPS01, (LPSTR)(LPCTSTR)strHostSubFolder);
		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 5000, (PDWORD_PTR)(dwReturn) ) == TRUE )
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Success."));
			bRet = TRUE;
		}
		else
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Fail."));
			RestartFTPUploader(_T("D:\\DIT_Review")); //A2E FTPUploader //210805
			bRet = FALSE;
		}
	}

	g_pLog->DisplayMessage(_T("Local[%s]"), Param.strLocalFilePath);
	g_pLog->DisplayMessage(_T("Host[%s]"),  Param.strHostSubFolder);


	return bRet;
}

BOOL CFileUploadControl::DownloadFileByFTPUploader(CString strHostSubFolder, CString strLocalFilePath)
{
	HWND hWnd = ::FindWindow(NULL, _T("FTPUploader"));
	BOOL bRet = FALSE;

	if (NULL == hWnd)
	{
		g_pLog->DisplayMessage(_T("Find FTPUploader Program : Fail."));
		StartFTPUploader(_T("C:\\AOIServer"));
	}
	else
	{
		COPYDATASTRUCT copydata = {0,0,0};

		CFTPParamCopyData Param;

		Param.bUpload=FALSE;
		Param.bDelete = FALSE;

		sprintf_s(Param.strLocalFilePath,sizeof(char)*255,"%s",strLocalFilePath);
		sprintf_s(Param.strHostSubFolder,sizeof(char)*255,"%s",strHostSubFolder);
		sprintf_s(Param.strFindFileName,sizeof(char)*255,"%s","");

		copydata.dwData = COPYDATA_FTP_ADD_FILE;
		copydata.cbData = sizeof(CFTPParamCopyData);
		copydata.lpData = &Param;

		DWORD dwReturn = 0;
		DWORD dwError = 0;
		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 5000, (PDWORD_PTR)(dwReturn) ) == TRUE )
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Success."));
			bRet = TRUE;
		}
		else
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Fail."));
			bRet = FALSE;
		}

		g_pLog->DisplayMessage(_T("Local[%s]"), Param.strLocalFilePath);
		g_pLog->DisplayMessage(_T("Host[%s]"), Param.strHostSubFolder);
	}

	return bRet;
}

BOOL CFileUploadControl::FindDownloadFileByFTPUploader(CString strHostSubFolder, CString strLocalFilePath, CString strFileName)
{
	HWND hWnd = ::FindWindow(NULL, _T("FTPUploader"));
	BOOL bRet = FALSE;

	if (NULL == hWnd)
	{
		g_pLog->DisplayMessage(_T("Find FTPUploader Program : Fail."));
		StartFTPUploader(_T("C:\\AOIServer"));
	}
	else
	{
		COPYDATASTRUCT copydata = {0,0,0};

		CFTPParamCopyData Param;

		Param.bUpload=FALSE;
		Param.bDelete = FALSE;

		sprintf_s(Param.strLocalFilePath,sizeof(char)*255,"%s",strLocalFilePath);
		sprintf_s(Param.strHostSubFolder,sizeof(char)*255,"%s",strHostSubFolder);
		sprintf_s(Param.strFindFileName,sizeof(char)*255,"%s",strFileName);

		copydata.dwData = COPYDATA_FTP_ADD_FILE;
		copydata.cbData = sizeof(CFTPParamCopyData);
		copydata.lpData = &Param;

		DWORD dwReturn = 0;
		DWORD dwError = 0;
		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 5000, (PDWORD_PTR)(dwReturn) ) == TRUE )
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Success."));
			bRet = TRUE;
		}
		else
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Fail."));
			bRet = FALSE;
		}

		g_pLog->DisplayMessage(_T("Local[%s]"), Param.strLocalFilePath);
		g_pLog->DisplayMessage(_T("Host[%s]"), Param.strHostSubFolder);
		g_pLog->DisplayMessage(_T("FindFile[%s]"), Param.strFindFileName);
	}

	return bRet;
}

// BOOL CFileUploadControl::DownloadStack(CString strHostSubFolder, CString strLocalFilePath, CString strFileName)
// {
// 	HWND hWnd = ::FindWindow(NULL, _T("FTPUploader"));
// 	BOOL bRet = FALSE;
// 
// 	if (NULL == hWnd)
// 	{
// 		g_pLog->DisplayMessage(TRUE, _T("Find to FTPUploader Program : Fail."));
// 		StartFTPUploader(_T("C:\\AOIServer"));
// 	}
// 	else
// 	{
// 		COPYDATASTRUCT copydata = {0,0,0};
// 
// 		CFTPParamCopyData Param;
// 
// 		Param.bUpload=FALSE;
// 		Param.bDelete=TRUE;
// 
// 		sprintf_s(Param.strLocalFilePath,sizeof(char)*255,_T("%s"),(LPSTR)(LPCTSTR)strLocalFilePath);
// 		sprintf_s(Param.strHostSubFolder,sizeof(char)*255,_T("%s"),(LPSTR)(LPCTSTR)strHostSubFolder);
// 		sprintf_s(Param.strFindFileName,sizeof(char)*255,_T("%s"),(LPSTR)(LPCTSTR)strFileName);
// 		copydata.dwData = COPYDATA_FTP_ADD_FILE;
// 		copydata.cbData = sizeof(CFTPParamCopyData);
// 		copydata.lpData = &Param;
// 
// 		DWORD dwReturn = 0;
// 		DWORD dwError = 0;
// 		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 5000, (PDWORD_PTR)(dwReturn) ) == TRUE )
// 		{
// 			g_pLog->DisplayMessage(TRUE, _T("Send Data to FTPUploader : Success."));
// 			bRet = TRUE;
// 		}
// 		else
// 		{
// 			g_pLog->DisplayMessage(TRUE, _T("Send Data to FTPUploader : Fail."));
// 			bRet = FALSE;
// 		}
// 
// 		g_pLog->DisplayMessage(TRUE, _T("Local[%s]"), Param.strLocalFilePath);
// 		g_pLog->DisplayMessage(TRUE, _T("Host[%s]"), Param.strHostSubFolder);
// 		g_pLog->DisplayMessage(TRUE, _T("GlassID[%s]"), Param.strFindFileName);
// 	}
// 
// 	return bRet;
// }

// BOOL CFileUploadControl::UploadImageByFTPUploader(CCommendParam &Param)
// {
// 	HWND hWnd;
// 	BOOL bRet = FALSE;
// 
// 	COPYDATASTRUCT copydata = {0,0,0};
// 
// 	hWnd = ::FindWindow(NULL, _T("CopyManager"));
// 
// 	copydata.dwData = eDirectoryCopy;
// 	copydata.cbData = sizeof(CCommendParam);
// 	copydata.lpData = &Param;
// 
// 	DWORD dwReturn = 0;
// 	DWORD dwError = 0;
// 
// 	if (NULL == hWnd)
// 	{
// 		g_pLog->DisplayMessage(TRUE, _T("Find CopyManager Program : Fail."));
// 		StartCopyManager(_T("C:\\AOIServer"));
// 	}
// 	else
// 	{
// 		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 60000, (PDWORD_PTR)(dwReturn) ) == TRUE )
// 		{
// 			g_pLog->DisplayMessage(TRUE, _T("Send Data to CopyManager : Success."));
// 			bRet = TRUE;
// 		}
// 		else
// 		{
// 			g_pLog->DisplayMessage(TRUE, _T("Send Data to CopyManager : Fail."));
// 			bRet = FALSE;
// 		}
// 	}
// 
// 	g_pLog->DisplayMessage(TRUE, "Local[%s]", Param.m_strOriginPath);
// 	g_pLog->DisplayMessage(TRUE, "Host[%s]",  Param.m_strTargetPath);
// 
// 	return bRet;
// }

BOOL CFileUploadControl::UploadImageByFTPUploader(CString strLocalFolderPath, CString strHostSubFolder, CString strFileFormat, BOOL bDelete/*=FALSE*/)
{
	HWND hWnd;
	BOOL bRet = FALSE;

	COPYDATASTRUCT copydata = {0,0,0};

	CFTPParamCopyData FTPParam;

	hWnd = ::FindWindow(NULL, _T("FTPUploader"));

	FTPParam.bUpload=TRUE;
	FTPParam.bDelete = bDelete;
	FTPParam.bMakeResultFile = TRUE;

#ifdef _UNICODE
	WideCharToMultiByte(CP_ACP, 0, strLocalFolderPath, -1, FTPParam.strFolderPath, strLocalFolderPath.GetLength(), NULL, NULL); 
	WideCharToMultiByte(CP_ACP, 0, strHostSubFolder, -1, FTPParam.strHostSubFolder, strHostSubFolder.GetLength(), NULL, NULL); 
	WideCharToMultiByte(CP_ACP, 0, strFileFormat, -1, FTPParam.strFolderFormat, strFileFormat.GetLength(), NULL, NULL); 
#else
	sprintf_s(FTPParam.strFolderPath,sizeof(char)*255,"%s",strLocalFolderPath);
	sprintf_s(FTPParam.strHostSubFolder,sizeof(char)*255,"%s",strHostSubFolder);
	sprintf_s(FTPParam.strFolderFormat,sizeof(char)*255,"%s",strFileFormat);
#endif

	copydata.dwData = COPYDATA_FTP_ADD_FILE;
	copydata.cbData = sizeof(CFTPParamCopyData);
	copydata.lpData = &FTPParam;

	DWORD dwReturn = 0;
	DWORD dwError = 0;

	if (NULL == hWnd)
	{
		g_pLog->DisplayMessage(_T("Find FTPUploader Program : Fail."));
		StartFTPUploader(_T("C:\\DIT_Review"));
	}
	else
	{
		//sprintf_s(Param.strHostSubFolder,sizeof(char)*255,"%s%s",FTP_HOME_PATH_LTPS01, (LPSTR)(LPCTSTR)strHostSubFolder);
		if( SendMessageTimeout(hWnd, WM_COPYDATA, NULL, (LPARAM)&copydata, SMTO_NORMAL, 5000, (PDWORD_PTR)(dwReturn) ) == TRUE )
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Success."));
			bRet = TRUE;
		}
		else
		{
			g_pLog->DisplayMessage(_T("Send Data to FTPUploader : Fail."));
			bRet = FALSE;
		}
	}

	g_pLog->DisplayMessage(_T("Local[%s]"), FTPParam.strFolderPath);
	g_pLog->DisplayMessage(_T("Host[%s]"),  FTPParam.strHostSubFolder);

	return bRet;
}