driverinstall.cpp

来自「一个用VC开发的驱动程序安装卸载程序」· C++ 代码 · 共 119 行

// DriverInstall.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "DriverInstall.h"
#include "Install.h"

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

/////////////////////////////////////////////////////////////////////////////
// The one and only application object
CWinApp theApp;
int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
	int nRetCode = 0;
	CString sVal;
	// initialize MFC and print and error on failure
	if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
	{
		// TODO: change error code to suit your needs
		sVal.LoadString(IDS_MFCINIT);
		printf("%s\n", sVal);
		nRetCode = 1;
	}
	else
	{
		// TODO: code your application's behavior here.
		CInstall Install;
		if ( argc < 3 )
		{
			sVal.LoadString(IDS_NOTENLINE);
			printf("%s\n", sVal);
			return(1);
		}
		if ( argc > 3 )
		{
			sVal.LoadString(IDS_MANYLINE);
			printf("%s\n", sVal);
			return(1);
		}
		
		if ( !strcmp(argv[1], "TRUE") )//install
		{
			if ( !strcmp(argv[2], "2K") )//2K or XP
			{
				if (!Install.PFCInstall())
				{
					sVal.LoadString(IDS_PFCERR);
					printf("%s\n", sVal);
					return 1;
				}
				
				if (!Install.EWFInstall())
				{
					sVal.LoadString(IDS_EWFERR);
					printf("%s\n", sVal);
					return 1;
				}
				
				if (!Install.BootDiskInstall())
				{
					sVal.LoadString(IDS_BOOTDISKERR);
					printf("%s\n", sVal);
					return 1;
					
				}
				if (!Install.ModifyReg(SERVICE_BOOT_START))
				{
					sVal.LoadString(IDS_MODREGERR);
					printf("%s\n", sVal);
					return 1;
				}
				sVal.LoadString(IDS_INSTALLOK);
				printf("%s\n", sVal); 
			}
		}else//uninstall drivers
		{
			if ( !strcmp(argv[2], "2K") )//2K or XP
			{
				if (!Install.UnInstallBootDisk())
				{
					sVal.LoadString(IDS_UIBOOTDISKERR);
					printf("%s\n", sVal); 
					return 1;
				}
				if (!Install.UnInstallEWF())
				{
					sVal.LoadString(IDS_UIEWFERR);
					printf("%s\n", sVal); 
					return 1;
				}
				if (!Install.UnInstallPFC())
				{
					sVal.LoadString(IDS_UIPFCERR);
					printf("%s\n", sVal); 
					return 1;
				}
				if (!Install.ModifyReg(SERVICE_SYSTEM_START))
				{
					sVal.LoadString(IDS_MODREGERR);
					printf("%s\n", sVal);
					return 1;
				}
				sVal.LoadString(IDS_UIOK);
				printf("%s\n", sVal); 
			}
		}
	}
	return nRetCode;
}