#include <win32k.h>

Include dependency graph for useratom.c:

Functions
	DBG_DEFAULT_CHANNEL (UserMisc)

RTL_ATOM FASTCALL	IntAddAtom (LPWSTR AtomName)

ULONG FASTCALL	IntGetAtomName (RTL_ATOM nAtom, LPWSTR lpBuffer, ULONG cjBufSize)

RTL_ATOM FASTCALL	IntAddGlobalAtom (LPWSTR lpBuffer, BOOL PinAtom)

	_Success_ (return !=0)
	Returns the name of an atom.

Function Documentation

◆ _Success_()

_Success_ ( return ! = 0 )

Returns the name of an atom.

Parameters

atom	- The atom to be queried.
pustrName	- Pointer to an initialized UNICODE_STRING that receives the name of the atom. The function does not update the Length member. The string is always NULL-terminated.

Returns: The length of the name in characters, or 0 if the function fails.

Note: The function does not aquire any global lock, since synchronisation is handled by the RtlAtom function.

Definition at line 94 of file useratom.c.

{
    WCHAR awcBuffer[256];
    ULONG cjLength;
 
    /* Retrieve the atom name into a local buffer (max length is 255 chars) */
    cjLength = IntGetAtomName((RTL_ATOM)atom, awcBuffer, sizeof(awcBuffer));
    if (cjLength != 0)
    {
        _SEH2_TRY
        {
            /* Probe the unicode string and the buffer */
            ProbeForRead(pustrName, sizeof(*pustrName), 1);
            ProbeForWrite(pustrName->Buffer, pustrName->MaximumLength, 1);
 
            /* Check if we have enough space to write the NULL termination */
            if (pustrName->MaximumLength >= sizeof(UNICODE_NULL))
            {
                /* Limit the length to the buffer size */
                cjLength = min(pustrName->MaximumLength - sizeof(UNICODE_NULL),
                               cjLength);
 
                /* Copy the string and NULL terminate it */
                RtlCopyMemory(pustrName->Buffer, awcBuffer, cjLength);
                pustrName->Buffer[cjLength / sizeof(WCHAR)] = L'\0';
            }
            else
            {
                cjLength = 0;
            }
        }
        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
        {
            /* On exception, set last error and fail */
            SetLastNtError(_SEH2_GetExceptionCode());
            cjLength = 0;
        }
        _SEH2_END
    }
 
    /* Return the length in characters */
    return cjLength / sizeof(WCHAR);
}

◆ DBG_DEFAULT_CHANNEL()

DBG_DEFAULT_CHANNEL ( UserMisc )

◆ IntAddAtom()

RTL_ATOM FASTCALL IntAddAtom ( LPWSTR AtomName )

Definition at line 13 of file useratom.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
    PTHREADINFO pti;
    RTL_ATOM Atom;
 
    pti = PsGetCurrentThreadWin32Thread();
    if (pti->rpdesk == NULL)
    {
        SetLastNtError(Status);
        return (RTL_ATOM)0;
    }
 
    Status = RtlAddAtomToAtomTable(gAtomTable, AtomName, &Atom);
 
    if (!NT_SUCCESS(Status))
    {
        SetLastNtError(Status);
        return (RTL_ATOM)0;
    }
    return Atom;
}

Referenced by co_IntShellHookNotify(), and NtUserRegisterWindowMessage().

◆ IntAddGlobalAtom()

RTL_ATOM FASTCALL IntAddGlobalAtom	(	LPWSTR	lpBuffer,
		BOOL	PinAtom
	)

Definition at line 62 of file useratom.c.

{
    RTL_ATOM Atom;
    NTSTATUS Status = STATUS_SUCCESS;
 
    Status = RtlAddAtomToAtomTable(gAtomTable, lpBuffer, &Atom);
 
    if (!NT_SUCCESS(Status))
    {
        ERR("Error init Global Atom.\n");
        return 0;
    }
 
    if (Atom && PinAtom)
        RtlPinAtomInAtomTable(gAtomTable, Atom);
 
    return Atom;
}

Referenced by InitUserAtoms().

◆ IntGetAtomName()

ULONG FASTCALL IntGetAtomName	(	RTL_ATOM	nAtom,
		LPWSTR	lpBuffer,
		ULONG	cjBufSize
	)

Definition at line 37 of file useratom.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
    PTHREADINFO pti;
    ULONG Size = cjBufSize;
 
    pti = PsGetCurrentThreadWin32Thread();
    if (pti->rpdesk == NULL)
    {
        SetLastNtError(Status);
        return 0;
    }
 
    Status = RtlQueryAtomInAtomTable(gAtomTable, nAtom, NULL, NULL, lpBuffer, &Size);
 
    if (!NT_SUCCESS(Status))
    {
        SetLastNtError(Status);
        return 0;
    }
 
    return Size;
}