Windows核心编程学习三：利用专有命名空间实现单一实例

注：源码为学习《Windows核心编程》的一些尝试，非原创。若能有助于一二访客，幸甚。

1.基本框架

/*******************************************************************************
 * File:	ErrorShow.cpp
 * Time:	2013-04-16
 * 描述：	修改试验原书同名程序
 *******************************************************************************/

#include <Windows.h>
#include <Windowsx.h>
#include <tchar.h>
#include "resource.h"

// Always compiler using Unicode.
#ifndef UNICODE
	#define UNICODE
#endif


#define chHANDLE_DLGMSG(hWnd, message, fn)                 \
   case (message): return (SetDlgMsgResult(hWnd, uMsg,     \
      HANDLE_##message((hWnd), (wParam), (lParam), (fn))))


BOOL Dlg_OnInitDialog(HWND hwnd, HWND hwndFocus, LPARAM lParam)
{
	return TRUE;
}


void Dlg_OnCommand(HWND hwnd, int id, HWND hwndCtl, UINT codeNotify)
{
	switch (id) {
		case IDOK:
		case IDCANCEL:
			EndDialog(hwnd, id);
			break;
	}
}


INT_PTR WINAPI Dlg_Proc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	switch (uMsg) {
		chHANDLE_DLGMSG(hwnd, WM_COMMAND,	 Dlg_OnCommand);
		chHANDLE_DLGMSG(hwnd, WM_INITDIALOG, Dlg_OnInitDialog);
	}

	return FALSE;
}

int APIENTRY _tWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPTSTR lpCmdLine, int nCmdShow)
{
	UNREFERENCED_PARAMETER(hPrevInstance);
	UNREFERENCED_PARAMETER(lpCmdLine);

	DialogBox(hInstance, MAKEINTRESOURCE(IDD_SINGLETON), NULL, Dlg_Proc);

	return 0;
}

2.使用可变参数

// Main dialog
HWND	g_hDlg;


// 添加一个字符串到编辑框
void AddText(PCTSTR pszFormat, ...)
{
	/* VA_LIST的用法：
	 *（1）首先在函数里定义一具VA_LIST型的变量，这个变量是指向参数的指针；
	 *（2）然后用VA_START宏初始化变量刚定义的VA_LIST变量；
	 *（3）然后用VA_ARG返回可变的参数，VA_ARG的第二个参数是你要返回的参数的
	 *		类型（如果函数有多个可变参数的，依次调用VA_ARG获取各个参数）；
	 *（4）最后用VA_END宏结束可变参数的获取。 */
	va_list argList;

	va_start(argList, pszFormat);

	TCHAR sz[20 * 1024];

	Edit_GetText(GetDlgItem(g_hDlg, IDC_EDIT_DETAILS), sz, _countof(sz));
	_vstprintf_s(_tcschr(sz, TEXT('\0')), _countof(sz) - _tcslen(sz), pszFormat, argList);
	Edit_SetText(GetDlgItem(g_hDlg, IDC_EDIT_DETAILS), sz);

	va_end(argList);
}


BOOL Dlg_OnInitDialog(HWND hwnd, HWND hwndFocus, LPARAM lParam)
{
	g_hDlg = hwnd;

	AddText(TEXT("a~z: \r\n"));
	for (char c = 'a'; c <= 'z'; c++)
		AddText(TEXT("%c "), c);

	return TRUE;
}

3.利用专有命名空间实现单实例应用程序

void CheckInstances()
{
	// 创建边界描述符
	g_hBoundary = CreateBoundaryDescriptor(g_szBoundary, 0);

	// 创建一个对应于本地管理员组的安全描述符SID
	BYTE localAdminSID[SECURITY_MAX_SID_SIZE];
	PSID pLocalAdminSID = &localAdminSID;
	DWORD cbSID = sizeof(localAdminSID);
		
	/*
		The CreateWellKnownSid function creates a SID for predefined aliases.
		BOOL WINAPI CreateWellKnownSid(
		  __in       WELL_KNOWN_SID_TYPE WellKnownSidType,
		  __in_opt   PSID DomainSid,
		  __out_opt  PSID pSid,
		  __inout    DWORD* cbSid
		);

		Parameters
		WellKnownSidType 
		Member of the WELL_KNOWN_SID_TYPE enumeration that specifies what the SID will identify.
		DomainSid 
		A pointer to a SID that identifies the domain control to use when creating the SID. 
		Pass NULL to use the local computer.
		pSid 
		A pointer to memory where CreateWellKnownSid will store the new SID.
		cbSid 
		A pointer to a DWORD that contains the number of bytes available at pSid. 
		The CreateWellKnownSid function stores the number of bytes actually used at this location.
	*/
	if (!CreateWellKnownSid(WinBuiltinAdministratorsSid, NULL, pLocalAdminSID, &cbSID))
	{
		AddText(TEXT("添加安全描述符到边界描述符失败: %u\r\n"), GetLastError());
		return;
	}

	// 将本地管理员组的安全描述符与边界描述符关联起来
	// 只有管理员身份运行的应用程序能获得该命名空间下的内核对象
	if (!AddSIDToBoundaryDescriptor(&g_hBoundary, pLocalAdminSID))
	{
		AddText(TEXT("添加安全描述符到边界描述符失败: %u\r\n"), GetLastError());
		return;
	}

	// 为本地管理员创建命名空间
	SECURITY_ATTRIBUTES sa;
	sa.nLength = sizeof(sa);
	sa.bInheritHandle = FALSE;
	if (!ConvertStringSecurityDescriptorToSecurityDescriptor(TEXT("D:(A;;GA;;;BA)"), 
		SDDL_REVISION_1, &sa.lpSecurityDescriptor, NULL))
	{
		AddText(TEXT("安全描述符创建失败: %u\r\n"),  GetLastError());
		return;
	}

	g_hNamespace = CreatePrivateNamespace(&sa, g_hBoundary, g_szNamespace);

	LocalFree(sa.lpSecurityDescriptor);

	// 检查私有命名空间创建是否成功
	DWORD dwLastError = GetLastError();

	// 创建失败
	if (g_hNamespace == NULL)
	{
		// 如果被拒绝访问，则直接返回
		// 这段代码必须在本地管理员账户运行
		if (dwLastError == ERROR_ACCESS_DENIED)
		{
			AddText(TEXT("创建命名空间时访问被拒绝.\r\n"));
			AddText(TEXT("	必须以管理员身份运行。\r\n\r\n"));
			return;
		}
		else
		{
			// 如果在该命名空间另一个实例已经被创建
			if (dwLastError == ERROR_ALREADY_EXISTS)
			{
				AddText(TEXT("创建私有命名空间失败: %u\r\n"), dwLastError);
				g_hNamespace = OpenPrivateNamespace(g_hBoundary, g_szNamespace);
				if (g_hNamespace == NULL)
				{
					AddText(TEXT("	并且打开私有命名空间失败： %u\r\n"), dwLastError);
					return;
				}
				else
				{
					g_bNamespaceOpened = TRUE;
					AddText(TEXT("	但是打开私有命名空间成功\r\n\r\n"));
				}
			}
			else
			{
				AddText(TEXT("发生了未知的错误: %u\r\n\r\n"), dwLastError);
				return;
			}
		}
	}

	// 尝试创建命名互斥量对象
	TCHAR szMutexName[64];
	StringCchPrintf(szMutexName, _countof(szMutexName), TEXT("%s\\%s"),
		g_szNamespace, TEXT("单一实例"));

	g_hSingleton = CreateMutex(NULL, FALSE, szMutexName);
	if (GetLastError() == ERROR_ALREADY_EXISTS)
	{
		AddText(TEXT("单一实例应用程序另一个实例已经运行:\r\n"));
		AddText(TEXT("--> 不能访问应用程序功能.\r\n"));
	}
	else
	{
		AddText(TEXT("单一实例应用程序的第一个实例\r\n"));
		AddText(TEXT("--> 现在访问应用程序功能\r\n"));
	}
}