windows-server-2003/admin/snapin/eventlog/exestub/main.cxx

//+--------------------------------------------------------------------------
//
//  Microsoft Windows
//  Copyright (C) Microsoft Corporation, 1999.
//
//  File:       main.cxx
//
//  Contents:   Stub executable that launches Event Viewer Snapin.
//
//  History:    01-14-1999   DavidMun   Created
//
//---------------------------------------------------------------------------

#include "headers.hxx"
#pragma hdrstop
#include "strsafe.h" // 553383-2002/04/30-JonN full path for mmc.exe

#define THIS_FILE                   L"main.cxx"
#define MAX_START_WAIT              500
const WCHAR c_wzBaseCommand[]       = L"mmc.exe\" /s \"";
const WCHAR c_wzEventMsc[]          = L"eventvwr.msc";
const WCHAR c_wzComputerSwitch[]    = L" /COMPUTER=";

//+--------------------------------------------------------------------------
//
//  Function:   NextSpace
//
//  Synopsis:   Return a pointer to the first space character found,
//              starting at [pwz], or to the end of the string, whichever
//              is first.
//
//  History:    01-14-1999   DavidMun   Created
//
//---------------------------------------------------------------------------

PWSTR
NextSpace(
    PWSTR pwz)
{
    while (*pwz && *pwz != L' ')
    {
        pwz++;
    }
    return pwz;
}




//+--------------------------------------------------------------------------
//
//  Function:   main
//
//  Synopsis:   Launch the event viewer snapin.
//
//  Returns:    0
//
//  History:    01-14-1999   DavidMun   Created
//              04-30-2002   JonN       full path for mmc.exe
//
//---------------------------------------------------------------------------

int _cdecl
main(int argc, char * argv[])
{
    // 553383-2002/04/30-JonN full path for mmc.exe
    WCHAR wzSysDir[MAX_PATH+1];
    ZeroMemory( wzSysDir, sizeof(wzSysDir) );
    UINT cchSysDir = GetSystemDirectory(wzSysDir, ARRAYLEN(wzSysDir)-1);
    if (!cchSysDir || cchSysDir >= ARRAYLEN(wzSysDir)-1)
    {
        PopupMessage(IDS_NO_SYSTEMDIR);
        return 0;
    }
    if (wzSysDir[cchSysDir - 1] != L'\\')
        wzSysDir[cchSysDir++] = L'\\';

    //
    // Build the path to the eventvwr.msc file, which is installed
    // in system32.
    //
    WCHAR wzMscPath[MAX_PATH];
    ZeroMemory( wzMscPath, sizeof(wzMscPath) );
    (void) StringCchCopy(wzMscPath, ARRAYLEN(wzMscPath), wzSysDir);
    (void) StringCchCat (wzMscPath, ARRAYLEN(wzMscPath), c_wzEventMsc);

    //
    // Verify that the .msc file exists, because mmc doesn't give a good
    // error message if it doesn't.
    //
    HANDLE hMscFile;
    hMscFile = CreateFile(wzMscPath,
                          0, // no access, just checking for existence
                          FILE_SHARE_READ | FILE_SHARE_WRITE,
                          NULL,
                          OPEN_EXISTING,
                          FILE_ATTRIBUTE_NORMAL,
                          NULL);
    if (hMscFile == INVALID_HANDLE_VALUE)
    {
        PopupMessage(IDS_EVENTVWR_MSC_NOT_FOUND, wzMscPath);
        return 0;
    }
    CloseHandle(hMscFile);

    //
    // Process the command line, ignoring /L and /H switches and treating
    // anything else as a server name.
    //

    PWSTR pwzArgs = GetCommandLine();

    //
    // Skip this executable's name
    //

    pwzArgs = NextSpace(pwzArgs);

    //
    // Advance until the server name or end of string, ignoring switches
    //

    while (*pwzArgs)
    {
        if (*pwzArgs == L' ')
        {
            pwzArgs++;
        }
        else if (*pwzArgs == L'-' || *pwzArgs == L'/')
        {
            pwzArgs = NextSpace(pwzArgs);
        }
        else
        {
            PWSTR pwzFileEnd = NextSpace(pwzArgs);

            *pwzFileEnd = L'\0';
            break;
        }
    }

    //
    // pwzArgs either points to a server name or end of string.
    // Build up a commandline for the CreateProcess call.

    WCHAR wzCommandLine[  1
                        + ARRAYLEN(wzSysDir) - 1
                        + ARRAYLEN(c_wzBaseCommand) - 1
                        + 1
                        + ARRAYLEN(wzMscPath) - 1
                        + 1
                        + ARRAYLEN(c_wzComputerSwitch) - 1
                        + MAX_PATH // pwzArgs
                        + 1 // NULL-termination
                       ];
    ZeroMemory( wzCommandLine, sizeof(wzCommandLine) );

    (void) StringCchCopy(wzCommandLine, ARRAYLEN(wzCommandLine), L"\"");
    (void) StringCchCat (wzCommandLine, ARRAYLEN(wzCommandLine), wzSysDir);
    (void) StringCchCat (wzCommandLine, ARRAYLEN(wzCommandLine), c_wzBaseCommand);

    //
    // Add eventvwr.msc to command line.
    //
    (void) StringCchCat (wzCommandLine, ARRAYLEN(wzCommandLine), wzMscPath);
    (void) StringCchCat (wzCommandLine, ARRAYLEN(wzCommandLine), L"\"");

    //
    // Append the server name, if supplied
    //

    if (*pwzArgs)
    {
        (void) StringCchCat(wzCommandLine, ARRAYLEN(wzCommandLine), c_wzComputerSwitch);
        (void) StringCchCat(wzCommandLine, ARRAYLEN(wzCommandLine), pwzArgs);
    }

    //
    // Try to launch mmc
    //

    BOOL fOk;
    STARTUPINFO si;
    PROCESS_INFORMATION pi;

    ZeroMemory(&si, sizeof si);
    si.cb = sizeof si;

    si.dwFlags = STARTF_USESHOWWINDOW;
    si.wShowWindow = SW_SHOWDEFAULT;

    ZeroMemory(&pi, sizeof pi);

    fOk = CreateProcess(NULL,
                        wzCommandLine,
                        NULL,
                        NULL,
                        FALSE,
                        NORMAL_PRIORITY_CLASS,
                        NULL,
                        NULL,
                        &si,
                        &pi);

    if (fOk)
    {
        WaitForInputIdle(pi.hProcess, MAX_START_WAIT);
        CloseHandle(pi.hThread);
        CloseHandle(pi.hProcess);
    }
    else
    {
        HRESULT hr = HRESULT_FROM_WIN32(GetLastError());
        PopupMessageAndCode(THIS_FILE, __LINE__, hr, IDS_MMC_LAUNCH_FAILED);
    }

    return 0;
}