update.cpp

实现程序自动更新

// Update.cpp : Defines the class behaviors for the application.
//

#include "stdafx.h"
#include "Update.h"
#include "UpdateDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CUpdateApp

BEGIN_MESSAGE_MAP(CUpdateApp, CWinApp)
	//{{AFX_MSG_MAP(CUpdateApp)
		// NOTE - the ClassWizard will add and remove mapping macros here.
		//    DO NOT EDIT what you see in these blocks of generated code!
	//}}AFX_MSG
	ON_COMMAND(ID_HELP, CWinApp::OnHelp)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CUpdateApp construction

CUpdateApp::CUpdateApp()
{
	// TODO: add construction code here,
	// Place all significant initialization in InitInstance
}

/////////////////////////////////////////////////////////////////////////////
// The one and only CUpdateApp object

CUpdateApp theApp;

/////////////////////////////////////////////////////////////////////////////
// CUpdateApp initialization

BOOL CUpdateApp::InitInstance()
{
	if (!AfxSocketInit())
	{
		AfxMessageBox(IDP_SOCKETS_INIT_FAILED);
		return FALSE;
	}
	AfxEnableControlContainer();
	// Standard initialization
	// If you are not using these features and wish to reduce the size
	//  of your final executable, you should remove from the following
	//  the specific initialization routines you do not need.

	CStringArray arr;
	GetCmdLinePara(arr);//获得命令行
	if(arr.GetSize()<3)
	{
		AfxMessageBox("用法:update.exe 程序名 版本 版本文件URL\nupdate.exe VolleyMail 3.0 http://www.extice.com/update/update.ini",MB_ICONSTOP);
		return FALSE;
	}

#ifdef _AFXDLL
	Enable3dControls();			// Call this when using MFC in a shared DLL
#else
	Enable3dControlsStatic();	// Call this when linking to MFC statically
#endif

	DWORD dwType;
	CUpdateDlg dlg;
	dlg.m_strSoft=arr.GetAt(0);
	dlg.m_strVersion=arr.GetAt(1);
	AfxParseURL(arr.GetAt(2),dwType,dlg.m_strServer,dlg.m_strIniPath,dlg.m_dwPort);
	if(dwType!=AFX_INET_SERVICE_HTTP)
	{
		AfxMessageBox("版本文件URL应一http://开头!",MB_ICONSTOP);
		return FALSE;
	}
	m_pMainWnd = &dlg;
	int nResponse = dlg.DoModal();
	if (nResponse == IDOK)
	{
	}
	else if (nResponse == IDCANCEL)
	{
	}

	// Since the dialog has been closed, return FALSE so that we exit the
	//  application, rather than start the application's message pump.
	return FALSE;
}

//字符串分解
void CUpdateApp::GetCmdLinePara(CStringArray &paraArr)
{
	paraArr.RemoveAll();
	CString strLine=::AfxGetApp()->m_lpCmdLine;
	if(strLine.IsEmpty())
		return;

	int nLength=strLine.GetLength();
	char *buf=new char[nLength+1];
	strcpy(buf,strLine);
	char *p=buf;
	for(int i=0;i<128;i++)
	{
		if(buf[i]==0x20)//空格
		{
			buf[i]=0x00;
			paraArr.Add(p);
			i++;
			p=buf+i;
		}
		if(buf[i]==0x00)
		{
			paraArr.Add(p);
			break;
		}
	}
	delete buf;
}