init.c File Reference

#include "winsrv.h"
#include <debug.h>

Include dependency graph for init.c:

Go to the source code of this file.

Macros
#define	NDEBUG
#define	DEFAULT_AUTO_END_TASKS   FALSE
#define	DEFAULT_HUNG_APP_TIMEOUT   5000
#define	DEFAULT_WAIT_TO_KILL_APP_TIMEOUT   20000
#define	DEFAULT_PROCESS_TERMINATE_TIMEOUT   90000

Functions
static ULONG	GetRegIntFromID (IN HANDLE KeyHandle, IN PWCHAR ValueName, IN ULONG DefaultValue)
VOID FASTCALL	GetTimeouts (IN PSHUTDOWN_SETTINGS ShutdownSettings)
BOOL WINAPI	DllMain (IN HINSTANCE hInstanceDll, IN DWORD dwReason, IN LPVOID lpReserved)

Variables
SHUTDOWN_SETTINGS	ShutdownSettings

Macro Definition Documentation

DEFAULT_AUTO_END_TASKS

#define DEFAULT_AUTO_END_TASKS   FALSE

Definition at line 19 of file init.c.

DEFAULT_HUNG_APP_TIMEOUT

#define DEFAULT_HUNG_APP_TIMEOUT   5000

Definition at line 20 of file init.c.

DEFAULT_PROCESS_TERMINATE_TIMEOUT

#define DEFAULT_PROCESS_TERMINATE_TIMEOUT   90000

Definition at line 22 of file init.c.

DEFAULT_WAIT_TO_KILL_APP_TIMEOUT

#define DEFAULT_WAIT_TO_KILL_APP_TIMEOUT   20000

Definition at line 21 of file init.c.

NDEBUG

#define NDEBUG

Definition at line 14 of file init.c.

Function Documentation

DllMain()

BOOL WINAPI DllMain	(	IN HINSTANCE	hInstanceDll,
		IN DWORD	dwReason,
		IN LPVOID	lpReserved
	)

@DllMain

Core routine of the Utility Manager's library.

Parameters

[in]	hDllInstance	The entry point instance of the library.
[in]	fdwReason	The reason argument to indicate the motive DllMain is being called.
[in]	lpvReserved	Reserved.

Returns

Returns TRUE when main call initialization has succeeded, FALSE otherwise.

Definition at line 172 of file init.c.

{
    UNREFERENCED_PARAMETER(hInstanceDll);
    UNREFERENCED_PARAMETER(dwReason);
    UNREFERENCED_PARAMETER(lpReserved);

    return TRUE;
}

GetRegIntFromID()

static ULONG GetRegIntFromID ( IN HANDLE  KeyHandle, IN PWCHAR  ValueName, IN ULONG  DefaultValue  )

static

Definition at line 36 of file init.c.

{
    UNICODE_STRING ValueString;
    ULONG Length;
    UCHAR Buffer[sizeof(KEY_VALUE_PARTIAL_INFORMATION) + 32 * sizeof(WCHAR)];
    PKEY_VALUE_PARTIAL_INFORMATION PartialInfo = (PVOID)Buffer;
    NTSTATUS Status;
    ULONG Value;

    /* Open the key */
    RtlInitUnicodeString(&ValueString, ValueName);
    Status = NtQueryValueKey(KeyHandle,
                             &ValueString,
                             KeyValuePartialInformation,
                             PartialInfo,
                             sizeof(Buffer),
                             &Length);
    if (NT_SUCCESS(Status))
    {
        if (PartialInfo->Type == REG_SZ)
        {
            /* Convert to integer */
            RtlInitUnicodeString(&ValueString, (PWCHAR)PartialInfo->Data);
            Status = RtlUnicodeStringToInteger(&ValueString, 10, &Value);
        }
        else if (PartialInfo->Type == REG_DWORD)
        {
            /* Directly retrieve the data */
            Value = *(PULONG)PartialInfo->Data;
            Status = STATUS_SUCCESS;
        }
        else
        {
            DPRINT1("Unexpected registry type %d for setting %S\n", PartialInfo->Type, ValueName);
            Status = STATUS_UNSUCCESSFUL;
        }
    }

    if (!NT_SUCCESS(Status))
    {
        /* Use default value instead */
        Value = DefaultValue;
    }

    /* Return the value */
    return Value;
}

Referenced by GetTimeouts().

GetTimeouts()

VOID FASTCALL GetTimeouts

(

IN PSHUTDOWN_SETTINGS

ShutdownSettings

)

Definition at line 87 of file init.c.

{
    NTSTATUS Status;
    UNICODE_STRING RegistryString;
    OBJECT_ATTRIBUTES ObjectAttributes;
    HANDLE CurrentUserKeyHandle;
    HANDLE KeyHandle;

    /* Initialize with defaults first */
    ShutdownSettings->AutoEndTasks         = DEFAULT_AUTO_END_TASKS;
    ShutdownSettings->HungAppTimeout       = DEFAULT_HUNG_APP_TIMEOUT;
    ShutdownSettings->WaitToKillAppTimeout = DEFAULT_WAIT_TO_KILL_APP_TIMEOUT;
    ShutdownSettings->WaitToKillServiceTimeout = ShutdownSettings->WaitToKillAppTimeout;
    ShutdownSettings->ProcessTerminateTimeout  = DEFAULT_PROCESS_TERMINATE_TIMEOUT;

    /* Open the per-user desktop key */
    Status = RtlOpenCurrentUser(MAXIMUM_ALLOWED, &CurrentUserKeyHandle);
    if (NT_SUCCESS(Status))
    {
        RtlInitUnicodeString(&RegistryString,
                             L"Control Panel\\Desktop");
        InitializeObjectAttributes(&ObjectAttributes,
                                   &RegistryString,
                                   OBJ_CASE_INSENSITIVE,
                                   CurrentUserKeyHandle,
                                   NULL);
        Status = NtOpenKey(&KeyHandle, KEY_READ, &ObjectAttributes);
        if (NT_SUCCESS(Status))
        {
            /* Read timeouts */
            ShutdownSettings->HungAppTimeout = GetRegIntFromID(KeyHandle,
                                                              L"HungAppTimeout",
                                                              DEFAULT_HUNG_APP_TIMEOUT);
            ShutdownSettings->WaitToKillAppTimeout = GetRegIntFromID(KeyHandle,
                                                                    L"WaitToKillAppTimeout",
                                                                    DEFAULT_WAIT_TO_KILL_APP_TIMEOUT);
            ShutdownSettings->AutoEndTasks = GetRegIntFromID(KeyHandle,
                                                            L"AutoEndTasks",
                                                            DEFAULT_AUTO_END_TASKS);
            /* Done */
            NtClose(KeyHandle);
        }

        /* Done */
        NtClose(CurrentUserKeyHandle);
    }

    /* Now open the control key */
    RtlInitUnicodeString(&RegistryString,
                         L"\\Registry\\Machine\\System\\CurrentControlSet\\Control");
    InitializeObjectAttributes(&ObjectAttributes,
                               &RegistryString,
                               OBJ_CASE_INSENSITIVE,
                               NULL, NULL);
    Status = NtOpenKey(&KeyHandle, KEY_READ, &ObjectAttributes);
    if (NT_SUCCESS(Status))
    {
        /* Read the services timeout */
        ShutdownSettings->WaitToKillServiceTimeout = GetRegIntFromID(KeyHandle,
                                                                    L"WaitToKillServiceTimeout",
                                                                    ShutdownSettings->WaitToKillAppTimeout);

        /*
         * Retrieve the process terminate timeout.
         * See ftp://ftp.microsoft.com/MISC1/BUSSYS/WINNT/KB/Q234/6/06.TXT
         * and https://web.archive.org/web/20050216235758/http://support.microsoft.com/kb/234606/EN-US/
         * for more details.
         *
         * NOTE: Unused at the moment...
         */
        ShutdownSettings->ProcessTerminateTimeout = GetRegIntFromID(KeyHandle,
                                                                   L"ProcessTerminateTimeout",
                                                                   DEFAULT_PROCESS_TERMINATE_TIMEOUT);
        if (ShutdownSettings->ProcessTerminateTimeout < DEFAULT_HUNG_APP_TIMEOUT)
            ShutdownSettings->ProcessTerminateTimeout = DEFAULT_HUNG_APP_TIMEOUT;

        /* Done */
        NtClose(KeyHandle);
    }
}

Referenced by CSR_API().

Variable Documentation

ShutdownSettings

SHUTDOWN_SETTINGS ShutdownSettings

Initial value:

=
{
    DEFAULT_AUTO_END_TASKS,
    DEFAULT_HUNG_APP_TIMEOUT,
    DEFAULT_WAIT_TO_KILL_APP_TIMEOUT,
    DEFAULT_WAIT_TO_KILL_APP_TIMEOUT,
    DEFAULT_PROCESS_TERMINATE_TIMEOUT
}

Definition at line 24 of file init.c.

Referenced by CSR_API(), GetTimeouts(), NonConsoleProcessShutdown(), NotifyConsoleProcessForShutdown(), NotifyUserProcessForShutdown(), and UserClientShutdown().