金融证券系统

开发平台：
Visual C++

UpgradeDlg.cpp：源码内容
							
#include "stdafx.h"
#include "UpgradeDlg.h"
#define	ST_UPGRADE_PAUSED			1
#define	ST_UPGRADE_DLOK				2
#define	ST_UPGRADE_NOTFOUND			3
#define	ST_UPGRADE_CANNOTOPENFILE	4
HANDLE CUpgradeDlg::m_hEventKillUpgradeThread	= NULL;
HANDLE CUpgradeDlg::m_hEventUpgradeThreadKilled	= NULL;
HFILE  CUpgradeDlg::m_hFileUpgrade				= NULL;
BOOL CALLBACK UpgradeCallback(DWORD dwCode, DWORD dwProgress, LPCTSTR lpszMsg, void* cookie)
{
	HWND hMainWnd = (HWND)cookie;
	if (PROG_PROGRESS == dwCode)
	{
		::SendMessage(hMainWnd, WM_USER_UPGRADE_PROGRESS, dwProgress, (LPARAM)lpszMsg);
	}
	else
	{
		::SendMessage(hMainWnd, WM_USER_UPGRADE_MESSAGE, dwCode, (LPARAM)lpszMsg);
	}
	// User wants to quit program
	if (WAIT_OBJECT_0 == WaitForSingleObject(CUpgradeDlg::m_hEventKillUpgradeThread, 0))
	{
		::PostMessage(hMainWnd, WM_USER_UPGRADE_END, ST_UPGRADE_PAUSED, 0);
		SetEvent(CUpgradeDlg::m_hEventUpgradeThreadKilled);
		if (NULL != CUpgradeDlg::m_hFileUpgrade)
		{
			::CloseHandle((HANDLE)CUpgradeDlg::m_hFileUpgrade);
			CUpgradeDlg::m_hFileUpgrade	=	NULL;
		}
		AfxEndThread(0, TRUE);
		return FALSE;
	}
	return TRUE;
}
char szUpgradeFileName[] = "clking.exe";
UINT UpgradeMain(LPVOID pParam)
{	
	LPUPGRADE_INFO pInfo = (LPUPGRADE_INFO)pParam;
	PostMessage(pInfo->hMainWnd, WM_USER_UPGRADE_PROGRESS, 0, 0);
	CHttpClient	http;
	http.SetServer(AfxGetProfile().GetHTTPServerDomain(), AfxGetProfile().GetHTTPServerPort());
	http.SetAccessType(AfxGetQSProfile().GetAccessType(), AfxGetQSProfile().GetProxyType(),
		AfxGetQSProfile().GetProxyAddress(), AfxGetQSProfile().GetProxyPort(),
		AfxGetQSProfile().GetProxyUser(), AfxGetQSProfile().GetProxyPasswd());
	CFile file;
	CString	sFileName =	AfxGetProfile().GetWorkDirectory() + szUpgradeFileName;
	if (file.Open(sFileName, CFile::modeCreate | CFile::modeWrite | CFile::shareDenyWrite))
	{
		CUpgradeDlg::m_hFileUpgrade	= (HFILE)file.m_hFile;
		DWORD dwHttpCode = http.WGet(AfxGetProfile().GetUpgradeURL(), &file, UpgradeCallback, pInfo->hMainWnd);
		file.Close();
		CUpgradeDlg::m_hFileUpgrade	= NULL;
		if (HTTP_STATUS_OK == dwHttpCode)
		{
			::SendMessage(pInfo->hMainWnd, WM_USER_UPGRADE_END, ST_UPGRADE_DLOK, 0);
		}
		else
		{
			::SendMessage(pInfo->hMainWnd, WM_USER_UPGRADE_END, ST_UPGRADE_NOTFOUND, 0);
		}
	}
	else
	{
		::SendMessage(pInfo->hMainWnd, WM_USER_UPGRADE_END, ST_UPGRADE_CANNOTOPENFILE, 0);
		if (NULL != CUpgradeDlg::m_hFileUpgrade)
		{
			::CloseHandle((HANDLE)CUpgradeDlg::m_hFileUpgrade);
			CUpgradeDlg::m_hFileUpgrade	=	NULL;
		}
	}
	AfxEndThread(0, TRUE);
	return 0;
}
/////////////////////////////////////////////////////////////////////////////
// CUpgradeDlg dialog
BEGIN_MESSAGE_MAP(CUpgradeDlg, CDialog)
	//{{AFX_MSG_MAP(CUpgradeDlg)
	ON_BN_CLICKED(IDC_PROXY_SET, OnProxySet)
	ON_MESSAGE(WM_USER_UPGRADE_MESSAGE, OnUpgradeMessage)
	ON_MESSAGE(WM_USER_UPGRADE_PROGRESS, OnUpgradeProgress)
	ON_MESSAGE(WM_USER_UPGRADE_END, OnUpgradeEnd)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
CUpgradeDlg::CUpgradeDlg(CWnd* pParent /*=NULL*/) : CDialog(CUpgradeDlg::IDD, pParent)
{
	m_pUpgradeInfo	= NULL;
	m_bAutoRun		= FALSE;
}
CUpgradeDlg::~CUpgradeDlg()
{
	if (m_pUpgradeInfo)
	{
		delete	m_pUpgradeInfo;
		m_pUpgradeInfo = NULL;
	}
}
void CUpgradeDlg::SetAutoRun(BOOL bAutoRun)
{
	m_bAutoRun = bAutoRun;
}
void CUpgradeDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CUpgradeDlg)
	DDX_Control(pDX, IDOK, m_btnOK);
	DDX_Control(pDX, IDCANCEL, m_btnCancel);
	DDX_Control(pDX, IDC_STATIC_STATUS, m_staticStatus);
	DDX_Control(pDX, IDC_STATIC_PROGRESS, m_staticProgress);
	DDX_Control(pDX, IDC_PROXY_SET, m_btnProxySet);
	DDX_Control(pDX, IDC_EDIT_EXPLAIN, m_editExplain);
	DDX_Control(pDX, IDC_DL_PROGRESS, m_ctrlProgress);
	//}}AFX_DATA_MAP
}
/////////////////////////////////////////////////////////////////////////////
// CUpgradeDlg message handlers
BOOL CUpgradeDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	HRSRC hReSrc = NULL;
	HGLOBAL hRead = NULL;
	LPCTSTR lpText;
	HINSTANCE hModule =	AfxGetResourceHandle();
	hReSrc = ::FindResource(hModule, MAKEINTRESOURCE(IDR_TEXT_UPGRADEEXPLAIN), _T("Text"));
	if (hReSrc)
		hRead = ::LoadResource(hModule, hReSrc);
	if (hRead)
	{
		lpText = (LPCTSTR)::LockResource(hRead);
		if (lpText)
		{
			m_editExplain.SetWindowText(lpText);
		}
		::UnlockResource(hRead);
	}
	m_staticStatus.SetWindowPos(NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_HIDEWINDOW);
	m_ctrlProgress.SetWindowPos(NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_HIDEWINDOW);
	m_ctrlProgress.SetRange(0, STKLIB_MAX_PROGRESS);
	m_ctrlProgress.SetPos(0);
	if (m_bAutoRun)
		OnOK();
	return TRUE;
}
void CUpgradeDlg::OnOK() 
{
	UpdateData();
	// disable and enble buttons
	m_btnOK.EnableWindow(FALSE);
	m_btnProxySet.EnableWindow(FALSE);
	// show message
	m_staticStatus.SetWindowPos(NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_SHOWWINDOW);
	m_ctrlProgress.SetWindowPos(NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_SHOWWINDOW);
	CUpgradeDlg::m_hEventKillUpgradeThread = CreateEvent(NULL,FALSE,FALSE,NULL);
	CUpgradeDlg::m_hEventUpgradeThreadKilled = CreateEvent(NULL,FALSE,FALSE,NULL);
	if (NULL == m_pUpgradeInfo)
		m_pUpgradeInfo	= new UPGRADE_INFO;
	memset(m_pUpgradeInfo, 0, sizeof(UPGRADE_INFO));
	m_pUpgradeInfo->hMainWnd = GetSafeHwnd();
	AfxBeginThread(UpgradeMain, (LPVOID)m_pUpgradeInfo, THREAD_PRIORITY_NORMAL);
}
void CUpgradeDlg::OnCancel() 
{
	SetEvent(m_hEventKillUpgradeThread);
	// WaitForSingleObject(m_hEventUpgradeThreadKilled, INFINITE);
	CDialog::OnCancel();
}
void CUpgradeDlg::OnProxySet() 
{
	AfxDoProxySetting();
}
LRESULT CUpgradeDlg::OnUpgradeMessage(WPARAM wParam, LPARAM lParam)
{
	CString	strNewMsg;
	CString	strOldMsg;
	m_staticStatus.GetWindowText(strOldMsg);
	CString	strMsg = CNetDatabase::GetProgressMessage(wParam);
	if (strMsg.GetLength() > 0)
	{
		strNewMsg = strOldMsg;
		int	nIndex = strOldMsg.Find(" -- ");
		if (-1 != nIndex)
			strNewMsg = strOldMsg.Left(nIndex);
		strNewMsg += " -- " + strMsg;
	}
	if (strNewMsg.GetLength() > 0)
	{
		if (0 != strNewMsg.Compare(strOldMsg))
			m_staticStatus.SetWindowText(strNewMsg);
	}
	return 1;
}
LRESULT CUpgradeDlg::OnUpgradeProgress(WPARAM wParam, LPARAM lParam)
{
	// set message and enable buttons
	if (wParam < 0)
		wParam = 0;
	if (wParam > STKLIB_MAX_PROGRESS-1)
		wParam = STKLIB_MAX_PROGRESS-1;
	m_ctrlProgress.SetPos(wParam);
	CString	strProgressNew;
	strProgressNew.Format("%.2f%%", float(wParam)/100.);
	CString	strProgressOld;
	m_staticProgress.GetWindowText(strProgressOld);
	if (0 != strProgressNew.Compare(strProgressOld))
		m_staticProgress.SetWindowText(strProgressNew);
	return 1;
}
LRESULT CUpgradeDlg::OnUpgradeEnd(WPARAM wParam, LPARAM lParam)
{
	m_ctrlProgress.SetPos(STKLIB_MAX_PROGRESS);
	CString	strProgressNew;
	strProgressNew.Format("%.2f%%", 100.);
	m_staticProgress.SetWindowText(strProgressNew);
	// set message and enable buttons
	CString	strMsg;
	if (ST_UPGRADE_PAUSED == wParam)
		strMsg.LoadString(IDS_UPGRADE_PAUSED);
	else if (ST_UPGRADE_NOTFOUND == wParam)
		strMsg.LoadString(IDS_UPGRADE_NOTFOUND);
	else if (ST_UPGRADE_DLOK == wParam)
		strMsg.LoadString(IDS_UPGRADE_DLOK);
	else if (ST_UPGRADE_CANNOTOPENFILE == wParam)
		strMsg.LoadString(IDS_UPGRADE_CANNOTOPENFILE);
	else
		ASSERT(FALSE);
	m_staticStatus.SetWindowText(strMsg);
	if (ST_UPGRADE_DLOK == wParam)
	{
		// Upgrade Now
		AfxGetMainFrame()->PostMessage(WM_CLOSE);
		ShellExecute(NULL, "open", AfxGetProfile().GetWorkDirectory()+szUpgradeFileName,
					  NULL, AfxGetProfile().GetWorkDirectory(), SW_SHOW);
	}
	m_btnOK.EnableWindow(TRUE);
	m_btnCancel.EnableWindow(TRUE);
	m_btnProxySet.EnableWindow(TRUE);
	CloseHandle(CUpgradeDlg::m_hEventKillUpgradeThread);
	CloseHandle(CUpgradeDlg::m_hEventUpgradeThreadKilled);
	return 1;
}

						