第3章 内核对象（2）

#include <stdio.h>
#include <windows.h>

int main()
{

    //利用GUID生成唯一的锁名，防止内核对象重名
    HANDLE h = CreateMutex(NULL, FALSE,
            TEXT("{349210D3-EF54-4EC9-8313-9F47435D785D}"));

    if (GetLastError()== ERROR_ALREADY_EXISTS)
    {
        CloseHandle(h);
        return 0;
    }

    printf("单例程序正在运行中...\n");
    //为了演示，这里暂停一下cmd输出窗口
    system("pause");

    CloseHandle(h);
    return 0;

}

/*****************************************************************************************
Module:Singleton.cpp
Notices:Copyright(c) 2008 Jeffery Richter & Christophe Nasarre
*****************************************************************************************/

#include "..\..\CommonFiles\CmnHdr.h"
#include <tchar.h>
#include <strsafe.h>
#include <sddl.h>
#include "resource.h"
/////////////////////////////////////////////////////////////////////////////////////////

//主对话框
HWND g_hDlg;

//互斥对象、边界描述符和命名空间——用来检测前一个实例是否正在运行
HANDLE g_hSingleton = NULL;
HANDLE g_hBoundary = NULL;
HANDLE g_hNameSpace = NULL;

//跟踪命令空间是否被创建或被打开
BOOL g_bNamesapceOpened = FALSE;

//专有命名空间和边界描述符的名称
PCTSTR g_szBoundary = TEXT("3-Boundary");
PCTSTR g_szNameSpace = TEXT("3-Namespace");

#define DETAILS_CTRL  GetDlgItem(g_hDlg,IDC_EDIT_DETAILS)

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

    Edit_GetText(DETAILS_CTRL, sz, _countof(sz));
    //strchr函数原型：extern char *strchr(const char *s,char c);查找字符串s中首次出现字符c的位置
    //strchr在字符串str中查找字符ch第一次出现的位置，找到后返回一个指向该位置的指针。如果该字符不
    //存在于字符串中，则返回一个NULL指针
    //本例先找到编辑框字符串最后的位置，用strchr函数查找，然后用_vstprintfs_s格式化缓冲区
    _vstprintf_s(_tcschr(sz, TEXT(‘\0‘)), _countof(sz) - _tcslen(sz), pszFormat, argList);

    Edit_SetText(DETAILS_CTRL, sz);
    va_end(argList);
}

/////////////////////////////////////////////////////////////////////////////////////////
void CheckInstance()
{
    //创建边界描述符
    //参数Name：边界描述符名称；Flags：保留字，没用，填0
    //返回值：虽为句柄，但实际上是个指针，不能CloseHandle，要用DeleteBoundaryDescriptor删除
    g_hBoundary = CreateBoundaryDescriptor(g_szBoundary, 0);

    //创建与本地管理员组相关的用户SID
    BYTE LocalAdminSID[SECURITY_MAX_SID_SIZE];
    PSID pLocalAdminSID = &LocalAdminSID;
    DWORD cbSID = sizeof(LocalAdminSID);

    //CreateWellKnownSid——创建与本地管理员组相关的用户SID
    //第1个参数： WinBuiltinAdministratorsSid表示管理员账户组，WinWorldSid：表示所有账户
    //第2个参数：指向创建了SID的域的指针，为NULL时表示使用本地计算机
    //第3个参数：指向要返回的SID的存储地址，为传出的参数
    //第4个参数：指向要返回的SID的存储地址，为传出的参数
    if (!CreateWellKnownSid(   
                         WinBuiltinAdministratorsSid,NULL,pLocalAdminSID,&cbSID ))
    {
        AddText(TEXT("添加安全描述符到边界描述符失败：%u\r\n"), GetLastError());
        return;
    }

    //将SID与边界描述符关联起来AddSIDToBoundaryDescriptor，用来决定谁能进入边界并创建命名空间
    //只有管理员身份运行的应用程序能获得该命名空间中的内核对象
    if (!AddSIDToBoundaryDescriptor(&g_hBoundary, pLocalAdminSID)){
        AddText(TEXT("添加安全描述符到边界描述符失败：%u\r\n"), GetLastError());
        return;
    }

    //创建本地管理员专有的命名空间
    SECURITY_ATTRIBUTES sa;
    sa.nLength = sizeof(sa);
    sa.bInheritHandle = FALSE;
    //以下这个函数在sddl.h文件中，用来将一个字符串格式 安全描述符 转换为一个有效的、 功能
    //的安全描述符。
    //第1个参数：指向一个空结尾的字符串包含要转换的字符串格式安全描述符的指针
    //           "D:"表示DACL，
    //           (A;;GA;;;BA)表示允许完全控制管理员帐号(Allow full control to Adminstrators)
    /*
         // Define the SDDL for the DACL. This example sets
         // the following access:
         //     Built-in guests are denied all access.
         //     Anonymous logon is denied all access.
         //     Authenticated users are allowed read/write/execute access.
         //     Administrators are allowed full control.
         // Modify these values as needed to generate the proper
         // DACL for your application.
         TCHAR * szSD = TEXT("D:")       // Discretionary ACL
            TEXT("(D;OICI;GA;;BG)")      // Deny access to built-in guests
            TEXT("(D;OICI;GA;;;AN)")     // Deny access to anonymous logon
            TEXT("(A;OICI;GRGWGX;;;AU)") // Allow read/write/execute to authenticated users
            TEXT("(A;OICI;GA;;;BA)");    // Allow full control to administrators
    */
    //第2个参数：指定 StringSecurityDescriptor 字符串的修订级别。 当前，此值必须 SDDL_REVISION_1
    //第3个参数：指向一个变量，接收转换后的安全描述符的指针的指针。要释放返回的缓冲区，调用 LocalFree 函数
    //第4个参数：指向一个变量，接收该的大小以字节为单位的转换后的安全描述符的指针。此参数可以是NULL
    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"), GetLastError());
                    return;
                }else{
                    g_bNamesapceOpened = 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("Singleton"));

    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"));
    }
}

/////////////////////////////////////////////////////////////////////////////////////////
void Dlg_OnCommand(HWND hwnd, int id, HWND hwndCtl, UINT codeNotify)
{
    switch (id)
    {
    case IDOK:
    case IDCANCEL:
        //用户单击退出按钮或按ESC键
        EndDialog(hwnd, id);
        break;
    }
}

BOOL Dlg_OnInitDialog(HWND hwnd, HWND hwndFocus, LPARAM lParam)
{
    chSETDLGICONS(hwnd, IDD_SINGLETON);

    g_hDlg = hwnd;
    
    //检测是否有另一个实例正在运行
    CheckInstance();
    return TRUE;
}

/////////////////////////////////////////////////////////////////////////////////////////
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);
}

//{{NO_DEPENDENCIES}}
// Microsoft Visual C++ 生成的包含文件。
// 供 Singleton2.rc 使用
//
#define IDD_SINGLETON                   101
#define IDI_SINGLETON                   102
#define IDC_EDIT_DETAILS                1001

// Next default values for new objects
// 
#ifdef APSTUDIO_INVOKED
#ifndef APSTUDIO_READONLY_SYMBOLS
#define _APS_NEXT_RESOURCE_VALUE        103
#define _APS_NEXT_COMMAND_VALUE         40001
#define _APS_NEXT_CONTROL_VALUE         1002
#define _APS_NEXT_SYMED_VALUE           101
#endif
#endif

// Microsoft Visual C++ generated resource script.
//
#include "resource.h"

#define APSTUDIO_READONLY_SYMBOLS
/////////////////////////////////////////////////////////////////////////////
//
// Generated from the TEXTINCLUDE 2 resource.
//
#include "winres.h"

/////////////////////////////////////////////////////////////////////////////
#undef APSTUDIO_READONLY_SYMBOLS

/////////////////////////////////////////////////////////////////////////////
// 中文(简体，中国) resources

#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_CHS)
LANGUAGE LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED

#ifdef APSTUDIO_INVOKED
/////////////////////////////////////////////////////////////////////////////
//
// TEXTINCLUDE
//

1 TEXTINCLUDE 
BEGIN
    "resource.h\0"
END

2 TEXTINCLUDE 
BEGIN
    "#include ""winres.h""\r\n"
    "\0"
END

3 TEXTINCLUDE 
BEGIN
    "\r\n"
    "\0"
END

#endif    // APSTUDIO_INVOKED


/////////////////////////////////////////////////////////////////////////////
//
// Dialog
//

IDD_SINGLETON DIALOGEX 0, 0, 251, 161
STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | DS_CENTER | WS_MINIMIZEBOX | WS_VISIBLE | WS_CAPTION | WS_SYSMENU
EXSTYLE WS_EX_APPWINDOW
CAPTION "单一实例"
FONT 8, "MS Shell Dlg", 400, 0, 0x1
BEGIN
    DEFPUSHBUTTON   "退出",IDOK,197,136,40,14
    EDITTEXT        IDC_EDIT_DETAILS,13,13,225,117,ES_MULTILINE | ES_AUTOVSCROLL | ES_AUTOHSCROLL | ES_READONLY
END


/////////////////////////////////////////////////////////////////////////////
//
// DESIGNINFO
//

#ifdef APSTUDIO_INVOKED
GUIDELINES DESIGNINFO
BEGIN
    IDD_SINGLETON, DIALOG
    BEGIN
        LEFTMARGIN, 7
        RIGHTMARGIN, 244
        TOPMARGIN, 7
        BOTTOMMARGIN, 154
    END
END
#endif    // APSTUDIO_INVOKED


/////////////////////////////////////////////////////////////////////////////
//
// Icon
//

// Icon with lowest ID value placed first to ensure application icon
// remains consistent on all systems.
IDI_SINGLETON           ICON                    "Singleton.ico"
#endif    // 中文(简体，中国) resources
/////////////////////////////////////////////////////////////////////////////



#ifndef APSTUDIO_INVOKED
/////////////////////////////////////////////////////////////////////////////
//
// Generated from the TEXTINCLUDE 3 resource.
//


/////////////////////////////////////////////////////////////////////////////
#endif    // not APSTUDIO_INVOKED