mirror of
https://github.com/reactos/reactos.git
synced 2024-11-28 05:53:31 +08:00
145 lines
3.7 KiB
C
145 lines
3.7 KiB
C
/*
|
|
* 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;
|
|
}
|
|
|
|
/*!
|
|
* \brief Returns the name of an atom.
|
|
*
|
|
* \param atom - The atom to be queried.
|
|
* \param 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.
|
|
*
|
|
* \return 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.
|
|
*/
|
|
_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 */
|