Shell编程

开发平台：
Visual C++

dllreg_job.cpp：源码内容
							#include "dllreg_xhdr.h"
#include "dllregshex.h"
#include "dllreg_util.h"
#include "resource.h"
#include <stdio.h>
extern HINSTANCE g_hmodThisDll;	
STDMETHODIMP 
CShellExt::_GetFullFileName()
{
	HRESULT hr = S_FALSE;
	//
	//	IEnumFORMATETC. Needed for format enumeration.
	//
	IEnumFORMATETC *pefEtc = 0;
	hr = m_pDataObj->EnumFormatEtc(DATADIR_GET, &pefEtc);
	if(SUCCEEDED(hr))
	{
		hr = pefEtc->Reset();	//	Reset enumeration.
		if(SUCCEEDED(hr))
		{
			//
			//	FORMATETC. Needed for get data about object.
			//
			FORMATETC fEtc;
			ULONG ulFetched = 0L;
			while(TRUE)
			{
				hr = pefEtc->Next(1, &fEtc, &ulFetched);
				if(FAILED(hr) || (ulFetched <= 0))
					break;
				//
				//	'Arm' format and 'launch' to obtain STGMEDIUM...
				//
				fEtc.cfFormat	= CF_HDROP;
				fEtc.dwAspect	= DVASPECT_CONTENT;
				fEtc.lindex		= -1;
				fEtc.ptd		= NULL;
				fEtc.tymed		= TYMED_HGLOBAL;
				//
				//	IDataObject : GetData. Returned as TYMED_HGLOBAL.
				//
				STGMEDIUM stgM;
				hr = m_pDataObj->GetData(&fEtc, &stgM);
				if(SUCCEEDED(hr))
				{
					if(stgM.tymed == TYMED_HGLOBAL)
					{
						m_szFileUserClickedOn[0] = '';
						if(DragQueryFile((HDROP)stgM.hGlobal, (UINT)(-1), NULL, 0) == 1)	
						{
							//	one file only; eliminate this and improve code to allow multiple-DLL file (un)registering
							DragQueryFile((HDROP)stgM.hGlobal, 0, m_szFileUserClickedOn, _MAX_PATH + 1);
						}
					}
				}
			}
		}
	}
	if(pefEtc)
		pefEtc->Release();
	return hr;
}
STDMETHODIMP			
CShellExt::_DoRegisterJob(HWND hParent, LPCSTR pszWorkingDir, LPCSTR pszCmd, LPCSTR pszParam, int iShowCmd, 
	const int cintOpCode)
{
	_UNUSED_PARAMETER(iShowCmd);
	_UNUSED_PARAMETER(pszParam);
	_UNUSED_PARAMETER(pszCmd);
	HRESULT hr = S_FALSE;
	hr = _GetFullFileName();
	if(SUCCEEDED(hr))
	{
		char lpszFmtText[512], lpszAsk[1024], lpszRelName[2 * (_MAX_PATH + 1)];
		//	format string for user confirmation
		LoadString(g_hmodThisDll, IDS_Q_INSTALL, lpszFmtText, 512);
		//	relative filename; pointer arithmetic ("C:\TEMP\file.dll" - "C:\TEMP\" = "file.dll"
		strcpy(lpszRelName, m_szFileUserClickedOn + ((strlen(pszWorkingDir) + 1) * sizeof(char)));
		//	compose message
		sprintf(lpszAsk, lpszFmtText, lpszRelName);
		if(MessageBox(0, lpszAsk, 0, MB_YESNO | MB_ICONQUESTION) == IDYES)
		{
			char pszCmdLine[_MAX_PATH + 1];	//	command line for regsvr32.exe
			switch(cintOpCode)
			{
				case REGDLL_INSTALL:
					sprintf(pszCmdLine, "regsvr32.exe "%s"", m_szFileUserClickedOn);
					break;
				case REGDLL_UNINSTALL:
					sprintf(pszCmdLine, "regsvr32.exe /u "%s"", m_szFileUserClickedOn);
					break;
				default:
					break;
			}
			STARTUPINFO si;
			PROCESS_INFORMATION pi;
			ZeroMemory(&si, sizeof(si));
			si.cb = sizeof(STARTUPINFO);
			si.dwFlags = STARTF_USESHOWWINDOW;
			si.wShowWindow = SW_SHOW;
			ZeroMemory(&pi, sizeof(pi));
			if(CreateProcess(0, pszCmdLine, 0, 0, FALSE, CREATE_DEFAULT_ERROR_MODE | NORMAL_PRIORITY_CLASS, 0, 0, &si, &pi))
			{
				CloseHandle(pi.hProcess);
				CloseHandle(pi.hThread);
			}
			else
				MessageBox(0, "Unable to run regsvr32.exe process.", DLLREGUNREGNAME, MB_OK);
		}
		else
			MessageBox(hParent, "User abort!", DLLREGUNREGNAME, MB_OK);
	}
	return NOERROR;
}
STDMETHODIMP 
CShellExt::DoInstall(HWND hParent, LPCSTR pszWorkingDir, LPCSTR pszCmd, LPCSTR pszParam, int iShowCmd)
{
	return _DoRegisterJob(hParent, pszWorkingDir, pszCmd, pszParam, iShowCmd, REGDLL_INSTALL);
}
STDMETHODIMP 
CShellExt::DoUninstall(HWND hParent, LPCSTR pszWorkingDir, LPCSTR pszCmd, LPCSTR pszParam, int iShowCmd)
{
	return _DoRegisterJob(hParent, pszWorkingDir, pszCmd, pszParam, iShowCmd, REGDLL_UNINSTALL);
}
STDMETHODIMP 
CShellExt::DoAbout(HWND hParent, LPCSTR pszWorkingDir, LPCSTR pszCmd, LPCSTR pszParam, int iShowCmd)
{
	_UNUSED_PARAMETER(iShowCmd);
	_UNUSED_PARAMETER(pszParam);
	_UNUSED_PARAMETER(pszCmd);
	_UNUSED_PARAMETER(pszWorkingDir);
	char szAboutText[1024];
	LoadString(g_hmodThisDll, IDS_ABOUT, szAboutText, 1024);
	MessageBox(hParent, szAboutText, DLLREGUNREGNAME, MB_OK | MB_ICONINFORMATION);
	return NOERROR;
}

						