340 lines
No EOL
9.1 KiB
C++
340 lines
No EOL
9.1 KiB
C++
// AutoUpdateDlg.cpp : implementation file
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "DnAutoUpdate.h"
|
|
#include "DnAutoUpdateDlg.h"
|
|
#include "CHttpDownload.h"
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#endif
|
|
|
|
|
|
CString g_strHttpPath;
|
|
CString g_strLocalPath;
|
|
vector<int> g_vecUpdateVersion;
|
|
CString g_strCurrentDownloadFileName;
|
|
int g_nCurrentDownloadCount;
|
|
|
|
CHttpDownload g_HttpDownload;
|
|
CDnAutoUpdateDlg* g_pMainDialog;
|
|
|
|
// Download Thread
|
|
UINT threadDownloadProc( LPVOID pParam )
|
|
{
|
|
HINTERNET hInternetSession;
|
|
// ¼¼¼Ç ¿±â
|
|
hInternetSession = InternetOpen( NULL, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0 );
|
|
|
|
if( hInternetSession != NULL )
|
|
{
|
|
int nRtn = CHttpDownload::HTTPDOWNLOAD_ERROR;
|
|
int nVersionCount = static_cast<int>( g_vecUpdateVersion.size() );
|
|
for( int i=0; i<nVersionCount; i++ )
|
|
{
|
|
g_HttpDownload.Reset();
|
|
|
|
CString strVersion;
|
|
strVersion.Format( _T( "%08d" ), g_vecUpdateVersion[i] );
|
|
|
|
CString strUrl;
|
|
strUrl += g_strHttpPath;
|
|
strUrl += strVersion;
|
|
strUrl += _T("/");
|
|
|
|
CString strPatchFileUrl, strPatchFileName;
|
|
strPatchFileUrl.Format( _T( "%sPatch%08d.pak" ), strUrl.GetBuffer(), g_vecUpdateVersion[i] );
|
|
strPatchFileName.Format( _T( "%sPatch%08d.pak.tmp" ), g_strLocalPath, g_vecUpdateVersion[i] );
|
|
|
|
g_nCurrentDownloadCount = i + 1;
|
|
g_strCurrentDownloadFileName = strPatchFileName;
|
|
|
|
for( int i=0; i<RETRY_MAX_COUNT; i++ )
|
|
{
|
|
nRtn = g_HttpDownload.DownloadFile( hInternetSession, strPatchFileUrl, strPatchFileName );
|
|
if( nRtn == CHttpDownload::HTTPDOWNLOAD_SUCCESS )
|
|
break;
|
|
}
|
|
|
|
if( nRtn == CHttpDownload::HTTPDOWNLOAD_ERROR || nRtn == CHttpDownload::HTTPDOWNLOAD_TERMINATE )
|
|
break;
|
|
}
|
|
|
|
InternetCloseHandle( hInternetSession );
|
|
if( g_pMainDialog )
|
|
g_pMainDialog->DownloadComplete( nRtn );
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
// CDnAutoUpdateDlg dialog
|
|
|
|
CDnAutoUpdateDlg::CDnAutoUpdateDlg(CWnd* pParent /*=NULL*/)
|
|
: CDialog(CDnAutoUpdateDlg::IDD, pParent), m_bShowWindow( FALSE ), m_bShowTrayTooltip( FALSE ), m_pThread( NULL ), m_bDownloadEnd( FALSE ), m_dwExitTick( 0 )
|
|
{
|
|
m_hIcon = AfxGetApp()->LoadIcon( IDR_MAINFRAME );
|
|
g_pMainDialog = this;
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
|
|
DDX_Control( pDX, IDC_PROGRESS1, m_progressCtrl );
|
|
}
|
|
|
|
|
|
BEGIN_MESSAGE_MAP(CDnAutoUpdateDlg, CDialog)
|
|
ON_WM_PAINT()
|
|
ON_WM_QUERYDRAGICON()
|
|
ON_WM_SYSCOMMAND()
|
|
ON_WM_DESTROY()
|
|
ON_WM_TIMER()
|
|
ON_WM_CLOSE()
|
|
ON_WM_WINDOWPOSCHANGING()
|
|
ON_MESSAGE( WM_NOTIFYICON_MSG, OnTrayNotifyMsg )
|
|
ON_COMMAND( ID_AUTOUPDATE_EXIT, OnExit )
|
|
ON_COMMAND( ID_AUTOUPDATE_PAUSE, &CDnAutoUpdateDlg::OnBnClickedButtonPause )
|
|
ON_COMMAND( ID_AUTOUPDATE_RESUME, &CDnAutoUpdateDlg::OnBnClickedButtonResume )
|
|
//}}AFX_MSG_MAP
|
|
ON_BN_CLICKED( IDC_BUTTON_PAUSE, &CDnAutoUpdateDlg::OnBnClickedButtonPause )
|
|
ON_BN_CLICKED( IDC_BUTTON_RESUME, &CDnAutoUpdateDlg::OnBnClickedButtonResume )
|
|
ON_BN_CLICKED( IDC_BUTTON_EXIT, &CDnAutoUpdateDlg::OnBnClickedButtonExit )
|
|
END_MESSAGE_MAP()
|
|
|
|
void CDnAutoUpdateDlg::OnDestroy()
|
|
{
|
|
// Thread Terminate
|
|
if( m_pThread )
|
|
{
|
|
g_HttpDownload.SetExit( TRUE );
|
|
::WaitForSingleObject( m_pThread->m_hThread, INFINITE );
|
|
}
|
|
|
|
Shell_NotifyIcon( NIM_DELETE, &m_stNID ); // Æ®·¹ÀÌ ÇØÁ¦
|
|
|
|
CDialog::OnDestroy();
|
|
}
|
|
// CDnAutoUpdateDlg message handlers
|
|
|
|
void CDnAutoUpdateDlg::OnClose()
|
|
{
|
|
ShowWindowEx( SW_HIDE );
|
|
}
|
|
|
|
BOOL CDnAutoUpdateDlg::OnInitDialog()
|
|
{
|
|
CDialog::OnInitDialog();
|
|
|
|
// Set the icon for this dialog. The framework does this automatically
|
|
// when the application's main window is not a dialog
|
|
SetIcon(m_hIcon, TRUE); // Set big icon
|
|
SetIcon(m_hIcon, FALSE); // Set small icon
|
|
|
|
// Æ®·¹ÀÌ Ç³¼±¸» Ç¥½Ã
|
|
m_stNID.cbSize = sizeof( NOTIFYICONDATA );
|
|
m_stNID.hWnd = this->GetSafeHwnd();;
|
|
m_stNID.dwInfoFlags = NIIF_USER; //dz¼±¿¡ ¶ã ¾ÆÀÌÄÜÀ» »ç¿ëÀÚ ÁöÁ¤ ¾ÆÀÌÄÜÀ¸·Î
|
|
m_stNID.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP | NIF_INFO;
|
|
m_stNID.uID = IDR_MAINFRAME;
|
|
m_stNID.uCallbackMessage = WM_NOTIFYICON_MSG;
|
|
m_stNID.hIcon = LoadIcon( AfxGetInstanceHandle(), MAKEINTRESOURCE (IDR_MAINFRAME) );
|
|
m_stNID.uTimeout = 4000;
|
|
|
|
CString strInfoTitle(_T("DragonNest"));
|
|
CString strInfo(_T("AutoUpdate Start!"));
|
|
lstrcpyn( m_stNID.szInfoTitle, (LPCTSTR)strInfoTitle, sizeof( m_stNID.szInfoTitle ) );
|
|
lstrcpyn( m_stNID.szInfo, (LPCTSTR)strInfo, sizeof( m_stNID.szTip ) );
|
|
|
|
Shell_NotifyIcon( NIM_ADD, &m_stNID );
|
|
|
|
m_TrayMenu.LoadMenu(IDR_MENU1);
|
|
|
|
// ´Ù¿î·Îµå ¾²·¹µå »ý¼º
|
|
m_pThread = AfxBeginThread( threadDownloadProc , this );
|
|
|
|
SetTimer( 1, 500, 0 );
|
|
|
|
return TRUE; // return TRUE unless you set the focus to a control
|
|
}
|
|
|
|
// If you add a minimize button to your dialog, you will need the code below
|
|
// to draw the icon. For MFC applications using the document/view model,
|
|
// this is automatically done for you by the framework.
|
|
|
|
void CDnAutoUpdateDlg::OnPaint()
|
|
{
|
|
if (IsIconic())
|
|
{
|
|
CPaintDC dc(this); // device context for painting
|
|
|
|
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
|
|
|
|
// Center icon in client rectangle
|
|
int cxIcon = GetSystemMetrics(SM_CXICON);
|
|
int cyIcon = GetSystemMetrics(SM_CYICON);
|
|
CRect rect;
|
|
GetClientRect(&rect);
|
|
int x = (rect.Width() - cxIcon + 1) / 2;
|
|
int y = (rect.Height() - cyIcon + 1) / 2;
|
|
|
|
// Draw the icon
|
|
dc.DrawIcon(x, y, m_hIcon);
|
|
}
|
|
else
|
|
{
|
|
CDialog::OnPaint();
|
|
}
|
|
}
|
|
|
|
// The system calls this function to obtain the cursor to display while the user drags
|
|
// the minimized window.
|
|
HCURSOR CDnAutoUpdateDlg::OnQueryDragIcon()
|
|
{
|
|
return static_cast<HCURSOR>(m_hIcon);
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::OnExit()
|
|
{
|
|
CDialog::OnOK();
|
|
}
|
|
|
|
LRESULT CDnAutoUpdateDlg::OnTrayNotifyMsg( WPARAM wParam, LPARAM lParam )
|
|
{
|
|
UINT uMsg = (UINT) lParam;
|
|
switch( uMsg )
|
|
{
|
|
case WM_LBUTTONDBLCLK:
|
|
ShowWindowEx( SW_SHOW );
|
|
break;
|
|
case WM_RBUTTONUP:
|
|
{
|
|
CPoint pt;
|
|
GetCursorPos( &pt );
|
|
m_TrayMenu.GetSubMenu( 0 )->TrackPopupMenu( TPM_RIGHTALIGN | TPM_LEFTBUTTON ,pt.x, pt.y, this );
|
|
}
|
|
break;
|
|
case WM_MOUSEMOVE:
|
|
{
|
|
if( !m_bShowTrayTooltip )
|
|
{
|
|
Shell_NotifyIcon( NIM_DELETE, &m_stNID );
|
|
m_stNID.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
|
|
Shell_NotifyIcon( NIM_ADD, &m_stNID );
|
|
m_bShowTrayTooltip = TRUE;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::OnWindowPosChanging( WINDOWPOS FAR* lpwndpos )
|
|
{
|
|
CDialog::OnWindowPosChanging(lpwndpos);
|
|
|
|
if( m_bShowWindow )
|
|
lpwndpos->flags |= SWP_SHOWWINDOW;
|
|
else
|
|
lpwndpos->flags &= ~SWP_SHOWWINDOW;
|
|
}
|
|
|
|
BOOL CDnAutoUpdateDlg::ShowWindowEx( int nCmdShow )
|
|
{
|
|
m_bShowWindow = ( nCmdShow == SW_SHOW );
|
|
return (GetSafeHwnd()) ? ShowWindow( nCmdShow ) : TRUE;
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::OnSysCommand( UINT nID, LPARAM lParam )
|
|
{
|
|
if( (nID & 0xFFF0) == SC_MINIMIZE ) // ÃÖ¼ÒÈ ¹öÆ°Àº °É·¯¼ Hide 󸮸¸ ÇÑ´Ù.
|
|
ShowWindowEx( SW_HIDE );
|
|
else
|
|
CDialog::OnSysCommand( nID, lParam );
|
|
}
|
|
void CDnAutoUpdateDlg::OnBnClickedButtonPause()
|
|
{
|
|
if( m_pThread )
|
|
SuspendThread( m_pThread->m_hThread );
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::OnBnClickedButtonResume()
|
|
{
|
|
if( m_pThread )
|
|
ResumeThread( m_pThread->m_hThread );
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::OnBnClickedButtonExit()
|
|
{
|
|
CDialog::OnOK();
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::OnTimer( UINT_PTR nIDEvent )
|
|
{
|
|
CDialog::OnTimer(nIDEvent);
|
|
|
|
m_progressCtrl.SetRange32( 0, g_HttpDownload.GetTotalFileSize() );
|
|
m_progressCtrl.SetPos( g_HttpDownload.GetTotalDownloadBytes() );
|
|
SetDownloadInfo( g_HttpDownload.GetDownloadRate(), g_HttpDownload.GetDownloadPercent() );
|
|
|
|
if( m_bDownloadEnd )
|
|
{
|
|
if( GetTickCount() - m_dwExitTick > 5000 )
|
|
CDialog::OnOK();
|
|
}
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::SetDownloadInfo( float fRate, float fPercent )
|
|
{
|
|
CString strState;
|
|
if( fPercent < 10.0f )
|
|
strState.Format( L"Download Patch : %1.0f%%\n[%4.2f KB/s]", fPercent, fRate );
|
|
else if( fPercent < 100.0f )
|
|
strState.Format( L"Download Patch : %2.0f%%\n[%4.2f KB/s]", fPercent, fRate );
|
|
else
|
|
strState.Format( L"Download Patch : %3.0f%%\n[%4.2f KB/s]", fPercent, fRate );
|
|
|
|
GetDlgItem( IDC_STATIC_STATE )->SetWindowText( strState );
|
|
GetDlgItem( IDC_STATIC_FILENAME )->SetWindowText( g_strCurrentDownloadFileName );
|
|
strState.Format( L"%d / %d", g_nCurrentDownloadCount, g_vecUpdateVersion.size() );
|
|
GetDlgItem( IDC_STATIC_DOWNLOAD_COUNT )->SetWindowText( strState );
|
|
|
|
if( m_bShowTrayTooltip )
|
|
{
|
|
m_stNID.dwInfoFlags = NIIF_NONE;
|
|
m_stNID.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
|
|
m_stNID.uTimeout = 0;
|
|
lstrcpyn( m_stNID.szTip, (LPCTSTR)strState, sizeof( m_stNID.szTip ) );
|
|
DWORD dwMessage = NIM_MODIFY;
|
|
Shell_NotifyIcon( dwMessage, &m_stNID );
|
|
}
|
|
}
|
|
|
|
void CDnAutoUpdateDlg::DownloadComplete( int nRtn )
|
|
{
|
|
Shell_NotifyIcon( NIM_DELETE, &m_stNID );
|
|
|
|
m_stNID.dwInfoFlags = NIIF_USER; //dz¼±¿¡ ¶ã ¾ÆÀÌÄÜÀ» »ç¿ëÀÚ ÁöÁ¤ ¾ÆÀÌÄÜÀ¸·Î
|
|
m_stNID.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP | NIF_INFO;
|
|
m_stNID.hIcon = LoadIcon( AfxGetInstanceHandle(), MAKEINTRESOURCE (IDR_MAINFRAME) );
|
|
m_stNID.uTimeout = 4000;
|
|
|
|
CString strInfoTitle( _T("DragonNest") );
|
|
CString strInfo;
|
|
if( nRtn == CHttpDownload::HTTPDOWNLOAD_SUCCESS || nRtn == CHttpDownload::HTTPDOWNLOAD_ALREADY_DOWNLOAD )
|
|
strInfo = _T("AutoUpdate Complete!");
|
|
else
|
|
strInfo = _T("AutoUpdate Failed!");
|
|
|
|
lstrcpyn( m_stNID.szInfoTitle, (LPCTSTR)strInfoTitle, sizeof( m_stNID.szInfoTitle ) );
|
|
lstrcpyn( m_stNID.szInfo, (LPCTSTR)strInfo, sizeof( m_stNID.szTip ) );
|
|
|
|
Shell_NotifyIcon( NIM_ADD, &m_stNID );
|
|
m_bShowTrayTooltip = FALSE;
|
|
m_bDownloadEnd = TRUE;
|
|
m_dwExitTick = GetTickCount();
|
|
} |