ntlib.c

从内核直接获取NT操作系统的信息。

// __________________________________________________________
//
//                          ntlib.c
//              Windows NT Utility Library V1.00
//                11-15-1998 Sven B. Schreiber
//                       [e-mail]
// __________________________________________________________

#define  _NTLIB_DLL_
#include "NtLib.h"

// =================================================================
// DISCLAIMER
// =================================================================

/*

This software is provided "as is" and any express or implied
warranties, including, but not limited to, the implied warranties of
merchantibility and fitness for a particular purpose are disclaimed.
In no event shall the author Sven B. Schreiber be liable for any
direct, indirect, incidental, special, exemplary, or consequential
damages (including, but not limited to, procurement of substitute
goods or services; loss of use, data, or profits; or business
interruption) however caused and on any theory of liability,
whether in contract, strict liability, or tort (including negligence
or otherwise) arising in any way out of the use of this software,
even if advised of the possibility of such damage.

*/

// =================================================================
// REVISION HISTORY
// =================================================================

/*

11-15-1998 V1.00 Original version (SBS).

*/

// =================================================================
// GLOBAL VARIABLES
// =================================================================

SYSTEM_BASIC_INFORMATION gsbi;

// =================================================================
// GLOBAL STRINGS
// =================================================================

TBYTE atNull [] = T("");

// =================================================================
// LOOKUP TABLES
// =================================================================

NTL_LOOKUP aObjectTypes [] =
  {
  OB_TYPE_TYPE,                  T("Type"),
  OB_TYPE_DIRECTORY,             T("Directory"),
  OB_TYPE_SYMBOLIC_LINK,         T("SymbolicLink"),
  OB_TYPE_TOKEN,                 T("Token"),
  OB_TYPE_PROCESS,               T("Process"),
  OB_TYPE_THREAD,                T("Thread"),
  OB_TYPE_EVENT,                 T("Event"),
  OB_TYPE_EVENT_PAIR,            T("EventPair"),
  OB_TYPE_MUTANT,                T("Mutant"),
  OB_TYPE_SEMAPHORE,             T("Semaphore"),
  OB_TYPE_TIMER,                 T("Timer"),
  OB_TYPE_PROFILE,               T("Profile"),
  OB_TYPE_WINDOW_STATION,        T("WindowStation"),
  OB_TYPE_DESKTOP,               T("Desktop"),
  OB_TYPE_SECTION,               T("Section"),
  OB_TYPE_KEY,                   T("Key"),
  OB_TYPE_PORT,                  T("Port"),
  OB_TYPE_ADAPTER,               T("Adapter"),
  OB_TYPE_CONTROLLER,            T("Controller"),
  OB_TYPE_DEVICE,                T("Device"),
  OB_TYPE_DRIVER,                T("Driver"),
  OB_TYPE_IO_COMPLETION,         T("IoCompletion"),
  OB_TYPE_FILE,                  T("File"),
  };

#define N_OBJECT_TYPES (sizeof (aObjectTypes) / NTL_LOOKUP_)

// -----------------------------------------------------------------

NTL_LOOKUP aPoolTypes [] =
  {
  NonPagedPool,                  T("NonPagedPool"),
  PagedPool,                     T("PagedPool"),
  NonPagedPoolMustSucceed,       T("NonPagedPoolMustSucceed"),
  DontUseThisType,               T("DontUseThisType"),
  NonPagedPoolCacheAligned,      T("NonPagedPoolCacheAligned"),
  PagedPoolCacheAligned,         T("PagedPoolCacheAligned"),
  NonPagedPoolCacheAlignedMustS, T("NonPagedPoolCacheAlignedMustS"),
  };

#define N_POOL_TYPES (sizeof (aPoolTypes) / NTL_LOOKUP_)

// =================================================================
// USER INTERFACE
// =================================================================

int WINAPI
NtlBox (HWND   hWnd,
        UINT   uType,
        PTBYTE ptCaption,
        PTBYTE ptFormat,
        ...)
    {
    TBYTE atBuffer [NTL_WVSPRINTF];
    PVOID pArgs = &ptFormat + 1;

    wvsprintf (atBuffer, ptFormat, pArgs);
    return MessageBox (hWnd, atBuffer, ptCaption, uType);
    }

// =================================================================
// POINTER RETRIEVAL
// =================================================================

PVOID WINAPI
NtlPointerFs (DWORD dOffset)
    {
    DWORD dBackup;
    PVOID pData = NULL;

    __try
        {
        __asm
            {
            mov     dBackup, eax
            mov     eax, dOffset
            and     eax, 0xfffffffc
            mov     eax, fs:[eax]
            mov     pData, eax
            mov     eax, dBackup
            }
        }
    __except (EXCEPTION_EXECUTE_HANDLER)
        {
        pData = NULL;
        }
    return pData;
    }

// -----------------------------------------------------------------

PTEB WINAPI
NtlPointerTeb (void)
    {
    return NtCurrentTeb ();
    }

// -----------------------------------------------------------------

PPEB WINAPI
NtlPointerPeb (void)
    {
    PTEB pTeb = NtlPointerTeb ();

    return (pTeb != NULL ? pTeb->pPeb : NULL);
    }

// =================================================================
// DEVICE MANAGEMENT
// =================================================================

BOOL WINAPI
NtlDeviceClose (PHANDLE phDevice)
    {
    BOOL fOk = FALSE;

    if (*phDevice != INVALID_HANDLE_VALUE)
        {
        CloseHandle (*phDevice);
        *phDevice = INVALID_HANDLE_VALUE;

        fOk = TRUE;
        }
    return fOk;
    }

// -----------------------------------------------------------------

BOOL WINAPI
NtlDeviceOpen (PHANDLE phDevice,
               PTBYTE  atPath)
    {
    *phDevice = CreateFile (atPath, GENERIC_READ,
                            FILE_SHARE_READ | FILE_SHARE_WRITE,
                            NULL, OPEN_EXISTING,
                            FILE_ATTRIBUTE_NORMAL, NULL);

    return (*phDevice != INVALID_HANDLE_VALUE);
    }

// -----------------------------------------------------------------

DWORD WINAPI
NtlDeviceStop (PTBYTE atName,
               DWORD  dStatus)
    {
    SC_HANDLE      hControl, hService;
    SERVICE_STATUS ss;

    if ((dStatus & (NTL_STATUS_CREATED | NTL_STATUS_STARTED)) &&
        ((hControl = OpenSCManager (NULL, NULL,
                                    SC_MANAGER_ALL_ACCESS))
         != NULL))
        {
        if ((hService = OpenService (hControl, atName,
                                     SERVICE_ALL_ACCESS))
            != NULL)
            {
            if (dStatus & NTL_STATUS_STARTED)
                {
                ControlService (hService, SERVICE_CONTROL_STOP,
                                &ss);
                }
            if (dStatus & NTL_STATUS_CREATED)
                {
                DeleteService (hService);
                }
            CloseServiceHandle (hService);
            }
        CloseServiceHandle (hControl);
        }
    return 0;
    }

// -----------------------------------------------------------------

DWORD WINAPI
NtlDeviceStart (PTBYTE atName,
                PTBYTE atDisplay,
                PTBYTE atFile)
    {
    SC_HANDLE hControl, hService;
    DWORD     dStatus = 0;

    if ((hControl = OpenSCManager (NULL, NULL,
                                   SC_MANAGER_ALL_ACCESS))
        != NULL)
        {
        if ((hService = OpenService (hControl,
                                     atName,
                                     SERVICE_ALL_ACCESS))
            == NULL)
            {
            if ((hService = CreateService (hControl,
                                           atName, atDisplay,
                                           SERVICE_ALL_ACCESS,
                                           SERVICE_KERNEL_DRIVER,
                                           SERVICE_DEMAND_START,
                                           SERVICE_ERROR_NORMAL,
                                           atFile, NULL, NULL, NULL,
                                           NULL, NULL))
                != NULL)
                {
                dStatus |= NTL_STATUS_CREATED;
                }
            }
        if (hService != NULL)
            {
            if (StartService (hService, 0, NULL))
                {
                dStatus |= (NTL_STATUS_STARTED |
                            NTL_STATUS_RUNNING);
                }
            else
                {
                if (GetLastError ()
                    == ERROR_SERVICE_ALREADY_RUNNING)
                    {
                    dStatus |= NTL_STATUS_RUNNING;
                    }
                }
            CloseServiceHandle (hService);
            }
        CloseServiceHandle (hControl);
        }
    return dStatus;
    }

// -----------------------------------------------------------------

DWORD WINAPI
NtlDeviceCloseEx (PHANDLE phDevice,
                  PTBYTE  atName,
                  DWORD   dStatus)
    {
    if (dStatus & NTL_STATUS_OPENED)
        {
        NtlDeviceClose (phDevice);
        }
    return NtlDeviceStop (atName, dStatus);
    }

// -----------------------------------------------------------------

DWORD WINAPI
NtlDeviceOpenEx (PHANDLE phDevice,
                 PTBYTE  atName,
                 PTBYTE  atDisplay,
                 PTBYTE  atFile,
                 PTBYTE  atPath)
    {
    DWORD dStatus = 0;

    if (NtlDeviceOpen (phDevice, atPath))
        {
        dStatus |= NTL_STATUS_OPENED;
        }
    else
        {
        dStatus |= NtlDeviceStart (atName, atDisplay, atFile);

        if ((dStatus & NTL_STATUS_RUNNING) &&
            NtlDeviceOpen (phDevice, atPath))
            {
            dStatus |= NTL_STATUS_OPENED;
            }
        else
            {
            dStatus = NtlDeviceCloseEx (phDevice, atName, dStatus);
            }
        }
    return dStatus;
    }

// =================================================================
// COMMAND LINE PARSER
// =================================================================

PNTL_CMD_OPTION WINAPI
NtlCommandEnumerate (PNTL_CMD_OPTION pncoTable,
                     PTBYTE          ptName,
                     PDWORD          pdIndex)
    {
    DWORD           i, n;
    int             iSize;
    PNTL_CMD_OPTION pnco = NULL;

    for (n = 0; pncoTable [n].ptName != NULL; n++);

    for (i = *pdIndex; i < n; i++)
        {
        iSize = lstrlen (ptName);

        if ((pncoTable [i].Handler != NULL)
            &&
            (iSize <= lstrlen (pncoTable [i].ptName))
            &&
            (CompareString (LOCALE_USER_DEFAULT, NORM_IGNORECASE,
                            ptName, iSize,
                            pncoTable [i].ptName, iSize)
             == CSTR_EQUAL))
            {
            pnco = pncoTable + i;
            break;
            }
        }
    *pdIndex = (pnco != NULL ? ++i : 0);
    return pnco;
    }

// -----------------------------------------------------------------

PNTL_CMD_OPTION WINAPI
NtlCommandOption (PNTL_CMD_OPTION pncoTable,
                  PTBYTE          ptName,
                  PBOOL           pfAmbiguous)
    {
    int             iSize;
    PNTL_CMD_OPTION pnco, pncoNext;

    *pfAmbiguous = FALSE;
    pnco         = NULL;

    for (pncoNext = pncoTable; pncoNext->ptName != NULL; pncoNext++)
        {
        iSize = lstrlen (ptName);

        if ((pncoNext->Handler != NULL)
            &&
            (iSize <= lstrlen (pncoNext->ptName))
            &&
            (CompareString (LOCALE_USER_DEFAULT, NORM_IGNORECASE,
                            ptName, iSize,
                            pncoNext->ptName, iSize)
             == CSTR_EQUAL))
            {
            if (pnco == NULL)
                {
                pnco = pncoNext;
                }
            else
                {
                *pfAmbiguous = TRUE;
                pnco         = NULL;
                break;
                }
            }
        }
    return ((pnco != NULL) && (pnco->Handler != NULL) ? pnco
                                                      : NULL);
    }

// -----------------------------------------------------------------

void WINAPI
NtlCommandFlag (PNTL_CMD_FLAG pncfTable,
                PTBYTE        ptName,
                PDWORD        pdFlags)
    {
    DWORD i;

    for (i = 0; pncfTable [i].ptName != NULL; i++)
        {
        if (!lstrcmpi (ptName, pncfTable [i].ptName))
            {
            *pdFlags |= pncfTable [i].dMask;
            break;
            }
        }
    return;
    }

// -----------------------------------------------------------------

void WINAPI
NtlCommandParameter (PTBYTE             ptValue,
                     PNTL_CMD_PARAMETER pncp)
    {
    DWORD i, n;

    for (n = i = 0; ptValue [i]; i++)
        {
        if ((ptValue [i] < '0') || (ptValue [i] > '9')) break;

        n *= 10;
        n += (ptValue [i] - '0');
        }
    pncp->ptValue  = ptValue;
    pncp->dValue   = n;
    pncp->fNumeric = !ptValue [i];
    return;
    }