useratom.c

Go to the documentation of this file.

/*
 * COPYRIGHT:        See COPYING in the top level directory
 * PROJECT:          ReactOS kernel
 * PURPOSE:          User Atom helper routines
 * FILE:             win32ss/user/ntuser/useratom.c
 * PROGRAMER:        Filip Navara <xnavara@volny.cz>
 */
 
#include <win32k.h>
DBG_DEFAULT_CHANNEL(UserMisc);
 
RTL_ATOM FASTCALL
IntAddAtom(LPWSTR AtomName)
{
    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;
}
 
ULONG FASTCALL
IntGetAtomName(RTL_ATOM nAtom, LPWSTR lpBuffer, ULONG cjBufSize)
{
    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;
}
 
RTL_ATOM FASTCALL
IntAddGlobalAtom(LPWSTR lpBuffer, BOOL PinAtom)
{
    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;
}
 
_Success_(return != 0)
_At_(pustrName->Buffer, _Out_z_bytecap_post_bytecount_(pustrName->MaximumLength, return * 2 + 2))
ULONG
APIENTRY
NtUserGetAtomName(
    _In_ ATOM atom,
    _Inout_ PUNICODE_STRING pustrName)
{
    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);
}
 
/* EOF */