AutoUpdateDlg.cpp

// AutoUpdateDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "AutoUpdate.h"
#include "AutoUpdateDlg.h"
#include ".\autoupdatedlg.h"
#include <tlhelp32.h>
#include <afxsock.h>		// MFC socket extensions
#include <afxinet.h>

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CAutoUpdateDlg 对话框
CAutoUpdateDlg::CAutoUpdateDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CAutoUpdateDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	m_dwDownloadedSize = 0;
	m_dwTotalSize = 0;
	m_bFileOK = FALSE;
}

void CAutoUpdateDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_PROGRESS1, m_ctlProcess);
	DDX_Control(pDX, IDC_PROCESSVALUE, m_ctPercentValue);
}

BEGIN_MESSAGE_MAP(CAutoUpdateDlg, CDialog)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDCANCEL, OnBnClickedCancel)
	ON_MESSAGE(WM_POSMESSAGE,OnPositionMessage)
	ON_MESSAGE(WM_ENDMESSAGE,OnEndMessage)
END_MESSAGE_MAP()


// CAutoUpdateDlg 消息处理程序

BOOL CAutoUpdateDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码
	
	DWORD dwThreadID;
	HANDLE hThread = ::CreateThread(0,0,DoRun,this,0,&dwThreadID);
	CloseHandle(hThread);
	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CAutoUpdateDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		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;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CAutoUpdateDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

// CMsgDialog 消息处理程序

LRESULT CAutoUpdateDlg::OnPositionMessage(WPARAM wParam,LPARAM lParam)
{
	int iPer = 0;
	if(m_dwTotalSize != 0) iPer = (m_dwDownloadedSize*100)/m_dwTotalSize;
	CString csPercent; csPercent.Format("%d",iPer); csPercent = csPercent+"%";
	m_ctPercentValue.SetWindowText(csPercent);
	m_ctlProcess.SetRange(0,100);
	m_ctlProcess.SetPos(iPer);
	return 1;
}

LRESULT CAutoUpdateDlg::OnEndMessage(WPARAM wParam,LPARAM lParam)
{
	OnCancel();
	return 0;
}

void CAutoUpdateDlg::OnBnClickedCancel()
{
	// TODO: 在此添加控件通知处理程序代码
	OnCancel();
}

 DWORD WINAPI CAutoUpdateDlg::DoRun(LPVOID lParam)
{
	CAutoUpdateDlg * pThis = (CAutoUpdateDlg *)lParam;
	pThis->DoDownLoadThread();
	return TRUE;
}

CString GetModuleDirectory()
{
	CString strFilePath;
	GetModuleFileName(AfxGetInstanceHandle(),strFilePath.GetBuffer(MAX_PATH),MAX_PATH);
	strFilePath.ReleaseBuffer();
	int iStart = strFilePath.ReverseFind('\\');
	strFilePath = strFilePath.Mid(0,iStart + 1);
	return strFilePath;	
}

DWORD CAutoUpdateDlg::DoDownLoadThread(void)
{
	CString csCurDir = GetModuleDirectory()+"YLBSetup.exe";
	if(DownloadFile(m_csDownloadURL,csCurDir))
	{
		m_bFileOK = TRUE;
	}
	return PostMessage(WM_ENDMESSAGE,0,0);
}

#define BLOCKSIZE 1024
BOOL CAutoUpdateDlg::DownloadFile(LPCSTR lpURL,LPCSTR lpDestFile)
{
	CFile cUdpFile; 
	if(!cUdpFile.Open(lpDestFile,CFile::modeCreate|CFile::modeWrite|CFile::typeBinary|CFile::shareDenyWrite))
		return FALSE;
	BOOL bResult = FALSE;
	CInternetSession sessionDownload;
	try
	{
		CHttpFile* pFile = (CHttpFile*)sessionDownload.OpenURL("http://download.microsoft.com/download/B/A/D/BAD203B3-EF77-469E-9483-E0329590BBC3/wpilauncher_n.exe",1,INTERNET_FLAG_TRANSFER_BINARY|INTERNET_FLAG_RELOAD|INTERNET_FLAG_DONT_CACHE);
		CString   query = "";
		pFile->QueryInfo(HTTP_QUERY_CONTENT_LENGTH,query);
		long file_len=atol(query);
		m_dwTotalSize = file_len;
		PostMessage(WM_POSMESSAGE,0,0);
		DWORD dwStatus;
		if (pFile->QueryInfoStatusCode(dwStatus))
		{
			if (dwStatus == 200)
			{
				pFile->SetReadBufferSize(10240);
				if (TRUE)
				{
					DWORD dwLen = 0;
					char buf[BLOCKSIZE];
					while (TRUE)
					{
						DWORD dwTemp = pFile->Read(buf,BLOCKSIZE);
						if (dwTemp)
						{
							cUdpFile.Write(buf,dwTemp);
						}
						m_dwDownloadedSize += dwTemp;
						PostMessage(WM_POSMESSAGE,0,0);
						if (dwTemp < BLOCKSIZE)
						{
							break;
						}
					}
				}
			}
		}
		pFile->Close();
		bResult = TRUE;
	}
	catch(CInternetException* pException)
	{
		pException->Delete();
		return bResult;
	}
	sessionDownload.Close();
	cUdpFile.Close();
	return bResult;
}

BOOL CAutoUpdateDlg::idle(void) 
{
	::Sleep(1000);
	return TRUE;
}