liuye / etradeclient · Files

// AutoUpdate.cpp : 定义应用程序的类行为。
//

#include "stdafx.h"
#include "AutoUpdate.h"
#include "AutoUpdateDlg.h"
#include <sstream>
#include "dbghelp.h"
#include <boost/algorithm/string.hpp>
#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/json_parser.hpp>
#include "url_config.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#pragma comment(lib, "dbghelp.lib")

//异常处理 add by liuye 2016.9.23
LONG WINAPI ExceptionHandler(PEXCEPTION_POINTERS ExceptionPointer)
{
	AfxMessageBox(_T("升级程序发生异常,请联系工作人员!"));
	// Open the file

	CTime tmCurTime;
	tmCurTime = CTime::GetCurrentTime();
	CString sData = tmCurTime.Format(_T("%y%m%d"));
	CString sDumpName = _T("AutoUpdate_") + sData + ".dmp";

	HANDLE hFile = CreateFile(sDumpName, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	if ((hFile != NULL) && (hFile != INVALID_HANDLE_VALUE))
	{
		// Create the minidump
		MINIDUMP_EXCEPTION_INFORMATION stuDump;
		stuDump.ThreadId = GetCurrentThreadId();
		stuDump.ExceptionPointers = ExceptionPointer;
		stuDump.ClientPointers = FALSE;
		MINIDUMP_TYPE emDumpType = MiniDumpNormal;
		BOOL bCreateDump = MiniDumpWriteDump(GetCurrentProcess(), GetCurrentProcessId(), hFile, emDumpType, (ExceptionPointer != 0) ? &stuDump : 0, 0, 0);

		if (!bCreateDump)
			_tprintf(_T("MiniDumpWriteDump failed. Error: %u \n"), GetLastError());
		else
			_tprintf(_T("Minidump created.\n"));

		// Close the file
		CloseHandle(hFile);

		CWinApp* pApp = AfxGetApp();
		SendMessage(pApp->m_pMainWnd->m_hWnd, WM_CRASH_CLOSE, NULL, NULL);
	}
	else
	{
		_tprintf(_T("CreateFile failed. Error: %u \n"), GetLastError());
	}
	return EXCEPTION_EXECUTE_HANDLER;
}


// CAUAutoUpdateApp

BEGIN_MESSAGE_MAP(CAUAutoUpdateApp, CWinApp)
	ON_COMMAND(ID_HELP, &CWinApp::OnHelp)
END_MESSAGE_MAP()


// CAUAutoUpdateApp 构造

CAUAutoUpdateApp::CAUAutoUpdateApp()
{
	// TODO:  在此处添加构造代码， 
	// 将所有重要的初始化放置在 InitInstance 中
}


// 唯一的一个 CAUAutoUpdateApp 对象

CAUAutoUpdateApp theApp;


// CAUAutoUpdateApp 初始化

BOOL CAUAutoUpdateApp::InitInstance()
{
	CWinApp::InitInstance();

	SetUnhandledExceptionFilter(ExceptionHandler);

	// 创建 shell 管理器，以防对话框包含
	// 任何 shell 树视图控件或 shell 列表视图控件。
	CShellManager *pShellManager = new CShellManager;

	// 激活“Windows Native”视觉管理器，以便在 MFC 控件中启用主题
	CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));

	// 标准初始化
	// 如果未使用这些功能并希望减小
	// 最终可执行文件的大小，则应移除下列
	// 不需要的特定初始化例程
	// 更改用于存储设置的注册表项
	// TODO:  应适当修改该字符串，
	// 例如修改为公司或组织名
	SetRegistryKey(_T("应用程序向导生成的本地应用程序"));

	CAUAutoUpdateDlg dlgUpdate;
	do 
	{
		CString sClientType;

		if (AfxGetApp()->m_lpCmdLine)
		{
			
			CString sClientInfo(AfxGetApp()->m_lpCmdLine);
		
			int iName = sClientInfo.Find(":");
			dlgUpdate.m_pDataManage->m_stuLocalConfigData.sAppName = sClientInfo.Left(iName);
			CString sJsonInfo = sClientInfo.Right(sClientInfo.GetLength() - iName - 1);
			
			std::stringstream ss;
			ss << sJsonInfo;
			namespace PT = boost::property_tree;
			
			try //Parse the configuration file
			{
				PT::ptree ptree;
				PT::read_json(ss, ptree);

				dlgUpdate.m_pDataManage->m_stuLocalConfigData.sVersion = ptree.get<std::string>("version").c_str();
				sClientType = ptree.get<std::string>("type").c_str();
			}
			catch (...) // Catch the exception in order for logging.
			{
				Log(CString(L"解析命令行参数时出错！请确认存放的json数据格式正确！"));
				break; 
			}
		}

		dlgUpdate.m_pDataManage->m_stuLocalConfigData.sAppDir = GetModulePath();
		dlgUpdate.m_pDataManage->m_stuLocalConfigData.sSelfName = AfxGetAppName() + CString(".exe");
		dlgUpdate.m_pDataManage->m_stuLocalConfigData.sTempDir = dlgUpdate.m_pDataManage->m_stuLocalConfigData.sAppDir + CString("temp");


		if (sClientType == "card")
		{
			dlgUpdate.m_pDataManage->m_ClientType = CRAD_CLIENT;
		}
		else if (sClientType == "shop")
		{
			dlgUpdate.m_pDataManage->m_ClientType = SHOP_CLIENT;
		}
		else if (sClientType == "central")
		{
			dlgUpdate.m_pDataManage->m_ClientType = CENTRAL_CLIENT;
		}

		if (dlgUpdate.m_pDataManage->m_stuLocalConfigData.sAppName.IsEmpty())
		{
			AfxMessageBox("获取程序版本信息失败！");
			Log("获取程序版本信息失败！");
			break;
		}

		Log("检查更新。");
		if (dlgUpdate.m_pDataManage->CheckVersion())
		{
			Log("已经是最新版本。");
			break;
		}

		if (IDCANCEL == AfxMessageBox(("检查到有新版本，是否升级？"), MB_OKCANCEL))
		{
			Log("暂不升级。");
			break;
		}
		
		Log("确认升级。");
		m_pMainWnd = &dlgUpdate;
		INT_PTR nResponse = dlgUpdate.DoModal();
		if (nResponse == IDOK)
		{
			// TODO:  在此放置处理何时用
			//  “确定”来关闭对话框的代码
		}
		else if (nResponse == IDCANCEL)
		{
			// TODO:  在此放置处理何时用
			//  “取消”来关闭对话框的代码
		}
		else if (nResponse == -1)
		{
			TRACE(traceAppMsg, 0, "警告: 对话框创建失败，应用程序将意外终止。\n");
			TRACE(traceAppMsg, 0, "警告: 如果您在对话框上使用 MFC 控件，则无法 #define _AFX_NO_MFC_CONTROLS_IN_DIALOGS。\n");
		}
		
	} while (0);

	CAUDataManage::Instantialize()->DestoryInstance();

	// 删除上面创建的 shell 管理器。
	if (pShellManager != NULL)
	{
		delete pShellManager;
	}

	// 由于对话框已关闭，所以将返回 FALSE 以便退出应用程序，
	//  而不是启动应用程序的消息泵。
	return FALSE;
}