DotNet4应用程序打包工具->升级版【二】安装工具分析

在本系列的上一篇中，

我们给出了这个工具的具体的思路。

得到了很多朋友的反馈！

综合朋友的意见，

在没有改变工具原理的基础上

我对这个程序做了升级

如下图：

如你所见，

现在这个打包工具可以打包dotNet2.0  3.5  4

乃至所有在注册表中添加过注册表项的应用程序

下面我们就开始分析安装工具（也就是上面你看到的那个图片）

--------------------------

入口函数：

int WinMain(HINSTANCE hInstance,HINSTANCE hPreInstance,LPSTR lpCmdLine,int nCmdShow)
{	
	DialogBox(hInstance,MAKEINTRESOURCE(MainWinDL),0,DlgProc);
	return 0;
}

好吧，入口函数很简单，只是创建了一个窗体，并注册了窗口过程函数

-----------------------------

窗口过程

//窗口过程
BOOL CALLBACK DlgProc (HWND hDlg, UINT message,WPARAM wParam, LPARAM lParam)
{
    switch (message)
    {
		case WM_INITDIALOG :
			OnInitDlg(hDlg);
			return TRUE ;
		case WM_COMMAND :
			switch (LOWORD(wParam))
			{
				case IDC_STATIC_Name:
					ShellExecute(hDlg,"open","http://www.cnblogs.com/liulun",NULL,NULL,SW_SHOWNORMAL); 
					break;
				case IDC_BUTTON1:	
					GetFile(hDlg,IDC_EDIT1);
					break;
				case IDC_BUTTON3:
					GetFile(hDlg,IDC_EDIT3);
					break;
				case IDC_RADIO1:
					CheckRadio(hDlg,IDC_RADIO1,"SOFTWARE\\Microsoft\\NET Framework Setup\\NDP\\v2.0.50727");
					break;
				case IDC_RADIO2:
					CheckRadio(hDlg,IDC_RADIO2,"SOFTWARE\\Microsoft\\NET Framework Setup\\NDP\\v3.5");
					break;
				case IDC_RADIO3:
					CheckRadio(hDlg,IDC_RADIO3,"SOFTWARE\\Microsoft\\NET Framework Setup\\NDP\\v4");
					break;
				case IDC_RADIO_ELSE:
					CheckRadio(hDlg,IDC_RADIO_ELSE,"");
					break;
				case IDOK:
					ReleaseTar(hDlg);
					ReplaceICO(hDlg);
					BagTar(hDlg,IDC_EDIT3);
					BagTar(hDlg,IDC_EDIT1);
					BagStr(hDlg);
					Alert("打包成功");
					break;
				case IDCANCEL:
					EndDialog (hDlg, 0) ;
					return TRUE ;
			}
		break;
    }
    return FALSE ;
}

在这个过程函数里

接收到的每个消息都执行了一个或几个函数

那么，我们就一个函数一个函数的讲

-------------------------------------------------------

窗口初始化消息里

我们默认选中了dotNet4的单选按钮

void OnInitDlg(HWND hwnd)
{
	HWND cld = ::GetDlgItem(hwnd,IDC_RADIO3);
	::SendMessage(cld,BM_SETCHECK,1,0);
	::SetDlgItemText(hwnd,IDC_EDIT2,"SOFTWARE\\Microsoft\\NET Framework Setup\\NDP\\v4");
	LastCheckRdioId = IDC_RADIO3;
}

---------------------------------------------------------

四个单选按钮的单击事件

设置了文本框的内容，

并记录了当前选中的是哪个单选按钮

void CheckRadio(HWND hwnd,int rdioID,LPCSTR val)
{
	if(rdioID == LastCheckRdioId)
	{
		return;
	}
	::SetDlgItemText(hwnd,IDC_EDIT2,val);
	LastCheckRdioId = rdioID;
}

------------------------------------------------------------

选择文件

把选中的文件路径赋值给相应的文本框

//得到文件
void GetFile(HWND hwnd,int EDITId)
{
	char szFile[MAX_PATH] = {0};
	OPENFILENAME ofn;
	memset(&ofn, 0, sizeof(OPENFILENAME));   
	ofn.lStructSize = sizeof(OPENFILENAME);     
	ofn.lpstrFile = szFile;   
	ofn.nMaxFile = MAX_PATH;   
	ofn.lpstrFilter = "应用程序 (.exe)\0*.exe\0\0";
	ofn.lpstrDefExt = "exe";   
	ofn.lpstrTitle = "选择exe文件";
	ofn.nFilterIndex = 1;   
	ofn.lpstrFileTitle = NULL;   
	ofn.nMaxFileTitle = 0;   
	ofn.lpstrInitialDir = NULL;
	if(GetOpenFileName(&ofn))
	{
		SetDlgItemText(hwnd,EDITId,szFile);
	}
}

--------------------------------------------------------------------

从资源中读取宿主程序，并按指定的文件名，释放到当前目录下

//释放资源
int ReleaseTar(HWND hwnd)
{	
	::GetDlgItemText(hwnd,IDC_EDIT1,szFilePath,MAX_PATH);
	::strcat(szFilePath,".bag.exe");

	HMODULE hInstance = ::GetModuleHandle(NULL);
    HRSRC hResID = ::FindResource(hInstance,(LPCSTR)IDR_BIN1,"bin"); 
    HGLOBAL hRes = ::LoadResource(hInstance,hResID);
    LPVOID pRes = ::LockResource(hRes);
    DWORD dwResSize = ::SizeofResource(hInstance,hResID);
	if(!dwResSize)
	{
		return 0;
	}
    HANDLE hResFile = CreateFile(szFilePath,GENERIC_WRITE,0,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);    
    DWORD dwWritten = 0;
    WriteFile(hResFile,pRes,dwResSize,&dwWritten,NULL);
    CloseHandle(hResFile);
    if(dwResSize == dwWritten);
	{
		return 1;
	}
	return 0;
}

---------------------------------------------------------------------------

替换宿主程序的ICO图标资源

这里需要重点说明一下：

要想更新一个应用程序的资源

必须先知道这个资源的ID

GetIcoIndex函数的工作就是获取资源ID的

因为一般的应用程序图标资源都会有两个

所以获取了两个图表资源的ID

其他的WINAPI就不多解释了~~

int GetIcoIndex(HMODULE hExe,int index[])
{
	HRSRC hRes;
	int iLoop;
	int i = 0;
	for(iLoop = 1;;iLoop++)
    {
        hRes = FindResource(hExe, MAKEINTRESOURCE(iLoop), RT_ICON); 
		if (NULL == hRes)
		{
			if(iLoop == 60)
			{
				break;
			}
			continue ; 
		}
        else
		{
			index[i]  = iLoop;
			i +=1;
			if(i == 2)
			{
				break;
			}
		}
    }
	return 1;
}
int ReplaceICO(HWND hwnd)
{
	HMODULE hSrcExe,hDestExe;
	HANDLE hUpdateRes;
	HRSRC hRes;
	HRSRC hResLoad;
	char *lpResLock;
	int result;
	char szFile[MAX_PATH+1] = {0};
	int hSrcIndex[2] = {0};
	int hDestIndex[2] = {0};
	::GetDlgItemText(hwnd,IDC_EDIT1,szFile,MAX_PATH);
	hSrcExe = LoadLibrary(szFile);
	hDestExe = LoadLibrary(szFilePath);
	GetIcoIndex(hSrcExe,hSrcIndex);
	GetIcoIndex(hDestExe,hDestIndex);
	for(int i=0;i<2;i++)
	{
		hRes = FindResource(hSrcExe, MAKEINTRESOURCE(hSrcIndex[i]), RT_ICON); 
		hResLoad=(HRSRC)LoadResource(hSrcExe,hRes);
		lpResLock=(char*)LockResource(hResLoad);
		FreeLibrary(hDestExe);
		hUpdateRes=BeginUpdateResource(szFilePath,FALSE);
		result=UpdateResource(hUpdateRes,RT_ICON,MAKEINTRESOURCE(hDestIndex[i]),MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),lpResLock,SizeofResource(hSrcExe,hRes));
		EndUpdateResource(hUpdateRes, FALSE);
	}
	FreeLibrary(hSrcExe);
	return result;
}

---------------------------------------

为宿主程序增加目标程序资源和dotNet安装包资源

增加的资源也是需要标明ID的

因为宿主程序会根据约定好的ID来得到这些资源

EditId参数就是这些资源的ID

int BagTar(HWND hwnd,int EditId)
{
	HANDLE hFile;
	DWORD dwFileSize,dwBytesRead;
	LPBYTE lpBuffer;
	char szFile[MAX_PATH+1] = {0};
	::GetDlgItemText(hwnd,EditId,szFile,MAX_PATH);
	hFile = CreateFile(szFile,GENERIC_READ,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
	dwFileSize = GetFileSize(hFile, NULL);
	lpBuffer = new BYTE[dwFileSize];
	ReadFile(hFile, lpBuffer, dwFileSize, &dwBytesRead, NULL);
	HANDLE hResource = BeginUpdateResource(szFilePath, FALSE);
	UpdateResource(hResource,RT_RCDATA,MAKEINTRESOURCE(EditId),MAKELANGID(LANG_NEUTRAL,SUBLANG_DEFAULT),(LPVOID)lpBuffer,dwFileSize);	
	EndUpdateResource(hResource, FALSE);	
	delete [] lpBuffer; 
	CloseHandle(hFile);
	return 1;
}

--------------------------------------------------------

把注册表项的路径也当作资源打包进宿主程序

我们约定这个资源的ID为1039

int BagStr(HWND hwnd)
{
	char szFile[MAX_PATH+1] = {0};
	::GetDlgItemText(hwnd,IDC_EDIT2,szFile,MAX_PATH);
	HANDLE hUpdateRes=BeginUpdateResource(szFilePath,FALSE);
	int result=UpdateResource(hUpdateRes,RT_RCDATA,MAKEINTRESOURCE(1039),MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),(LPVOID)szFile,::strlen(szFile));
	EndUpdateResource(hUpdateRes, FALSE);
	return 0;
}

-------------------------------------------------------------

其他的一些代码如下

#include <Windows.h>
#include <ShlObj.h>
#include "resource.h"

TCHAR szFilePath[MAX_PATH + 1];

int LastCheckRdioId;
//提示
void Alert(LPCSTR msg)
{
	MessageBox(NULL,msg,"系统提示",MB_OK);	
}

---------------------------------------------------------------

后记：

没有写容错的代码~

也没有遵循命名规范~

大家见谅~

请各位推荐我的文章

因为你们的支持才是我的动力->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->->

此工具编译后的可执行文件下载地址：BagDotNet.zip

（因为不在需要把dotNet4安装程序打包进来，所以只有几十K了！多轻便啊！）