ReactOS 0.4.15-dev-8102-g108db8f
atom.c File Reference
#include <k32.h>
#include <debug.h>
Include dependency graph for atom.c:

Go to the source code of this file.

Macros

#define NDEBUG
 

Functions

PVOID WINAPI InternalInitAtomTable (VOID)
 
ATOM WINAPI InternalAddAtom (BOOLEAN Local, BOOLEAN Unicode, LPCSTR AtomName)
 
ATOM WINAPI InternalFindAtom (BOOLEAN Local, BOOLEAN Unicode, LPCSTR AtomName)
 
ATOM WINAPI InternalDeleteAtom (BOOLEAN Local, ATOM Atom)
 
UINT WINAPI InternalGetAtomName (BOOLEAN Local, BOOLEAN Unicode, ATOM Atom, LPSTR AtomName, DWORD Size)
 
ATOM WINAPI GlobalAddAtomA (LPCSTR lpString)
 
ATOM WINAPI GlobalAddAtomW (LPCWSTR lpString)
 
ATOM WINAPI GlobalDeleteAtom (ATOM nAtom)
 
ATOM WINAPI GlobalFindAtomA (LPCSTR lpString)
 
ATOM WINAPI GlobalFindAtomW (LPCWSTR lpString)
 
UINT WINAPI GlobalGetAtomNameA (ATOM nAtom, LPSTR lpBuffer, int nSize)
 
UINT WINAPI GlobalGetAtomNameW (ATOM nAtom, LPWSTR lpBuffer, int nSize)
 
BOOL WINAPI InitAtomTable (DWORD nSize)
 
ATOM WINAPI AddAtomA (LPCSTR lpString)
 
ATOM WINAPI AddAtomW (LPCWSTR lpString)
 
ATOM WINAPI DeleteAtom (ATOM nAtom)
 
ATOM WINAPI FindAtomA (LPCSTR lpString)
 
ATOM WINAPI FindAtomW (LPCWSTR lpString)
 
UINT WINAPI GetAtomNameA (ATOM nAtom, LPSTR lpBuffer, int nSize)
 
UINT WINAPI GetAtomNameW (ATOM nAtom, LPWSTR lpBuffer, int nSize)
 

Variables

PRTL_ATOM_TABLE BaseLocalAtomTable = NULL
 

Macro Definition Documentation

◆ NDEBUG

#define NDEBUG

Definition at line 12 of file atom.c.

Function Documentation

◆ AddAtomA()

ATOM WINAPI AddAtomA ( LPCSTR  lpString)

Definition at line 526 of file atom.c.

527{
528 return InternalAddAtom(TRUE, FALSE, lpString);
529}
#define TRUE
Definition: types.h:120
#define FALSE
Definition: types.h:117
ATOM WINAPI InternalAddAtom(BOOLEAN Local, BOOLEAN Unicode, LPCSTR AtomName)
Definition: atom.c:32

Referenced by test_local_add_atom(), test_local_error_handling(), test_local_get_atom_name(), and WDML_CreateString().

◆ AddAtomW()

ATOM WINAPI AddAtomW ( LPCWSTR  lpString)

Definition at line 536 of file atom.c.

537{
538 return InternalAddAtom(TRUE, TRUE, (LPSTR)lpString);
539}
char * LPSTR
Definition: xmlstorage.h:182

Referenced by test_local_add_atom(), test_local_error_handling(), test_local_get_atom_name(), UXTHEME_SetWindowProperty(), and WDML_CreateString().

◆ DeleteAtom()

ATOM WINAPI DeleteAtom ( ATOM  nAtom)

Definition at line 546 of file atom.c.

547{
548 return InternalDeleteAtom(TRUE, nAtom);
549}
ATOM WINAPI InternalDeleteAtom(BOOLEAN Local, ATOM Atom)
Definition: atom.c:245

Referenced by UXTHEME_SetWindowProperty(), and WDML_DecHSZ().

◆ FindAtomA()

ATOM WINAPI FindAtomA ( LPCSTR  lpString)

Definition at line 556 of file atom.c.

557{
558 return InternalFindAtom(TRUE, FALSE, lpString);
559}
ATOM WINAPI InternalFindAtom(BOOLEAN Local, BOOLEAN Unicode, LPCSTR AtomName)
Definition: atom.c:133

Referenced by test_DdeCreateStringHandleW(), test_local_add_atom(), test_local_error_handling(), and test_RegisterClipboardFormatA().

◆ FindAtomW()

ATOM WINAPI FindAtomW ( LPCWSTR  lpString)

Definition at line 566 of file atom.c.

567{
568 return InternalFindAtom(TRUE, TRUE, (LPSTR)lpString);
569
570}

Referenced by test_DdeCreateStringHandleW(), test_local_add_atom(), and test_local_error_handling().

◆ GetAtomNameA()

UINT WINAPI GetAtomNameA ( ATOM  nAtom,
LPSTR  lpBuffer,
int  nSize 
)

Definition at line 577 of file atom.c.

580{
582}
static TAGREF LPCWSTR LPDWORD LPVOID lpBuffer
Definition: db.cpp:175
UINT WINAPI InternalGetAtomName(BOOLEAN Local, BOOLEAN Unicode, ATOM Atom, LPSTR AtomName, DWORD Size)
Definition: atom.c:280
unsigned long DWORD
Definition: ntddk_ex.h:95
*nSize LPSTR _Inout_ LPDWORD nSize
Definition: winbase.h:2084

Referenced by test_local_get_atom_name(), test_RegisterClipboardFormatA(), and WDML_QueryString().

◆ GetAtomNameW()

UINT WINAPI GetAtomNameW ( ATOM  nAtom,
LPWSTR  lpBuffer,
int  nSize 
)

◆ GlobalAddAtomA()

ATOM WINAPI GlobalAddAtomA ( LPCSTR  lpString)

Definition at line 434 of file atom.c.

435{
436 return InternalAddAtom(FALSE, FALSE, lpString);
437}

Referenced by test_add_atom(), test_error_handling(), test_get_atom_name(), test_handles(), and test_msg_client().

◆ GlobalAddAtomW()

◆ GlobalDeleteAtom()

◆ GlobalFindAtomA()

ATOM WINAPI GlobalFindAtomA ( LPCSTR  lpString)

Definition at line 464 of file atom.c.

465{
466 return InternalFindAtom(FALSE, FALSE, lpString);
467}

Referenced by test_add_atom(), test_DdeCreateStringHandleW(), test_error_handling(), and test_RegisterClipboardFormatA().

◆ GlobalFindAtomW()

ATOM WINAPI GlobalFindAtomW ( LPCWSTR  lpString)

◆ GlobalGetAtomNameA()

UINT WINAPI GlobalGetAtomNameA ( ATOM  nAtom,
LPSTR  lpBuffer,
int  nSize 
)

◆ GlobalGetAtomNameW()

UINT WINAPI GlobalGetAtomNameW ( ATOM  nAtom,
LPWSTR  lpBuffer,
int  nSize 
)

Definition at line 496 of file atom.c.

499{
501 TRUE,
502 nAtom,
504 (DWORD)nSize);
505}

Referenced by EnumPropsExW(), EnumPropsW(), ObjectFromLresult(), test_get_atom_name(), and WDML_MakeHszFromAtom().

◆ InitAtomTable()

BOOL WINAPI InitAtomTable ( DWORD  nSize)

Definition at line 512 of file atom.c.

513{
514 /* Normalize size */
515 if (nSize < 4 || nSize > 511) nSize = 37;
516
517 DPRINT("Here\n");
519}
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
PRTL_ATOM_TABLE BaseLocalAtomTable
Definition: atom.c:17
NTSYSAPI NTSTATUS NTAPI RtlCreateAtomTable(_In_ ULONG TableSize, _Inout_ PRTL_ATOM_TABLE *AtomTable)
#define DPRINT
Definition: sndvol32.h:73

◆ InternalAddAtom()

ATOM WINAPI InternalAddAtom ( BOOLEAN  Local,
BOOLEAN  Unicode,
LPCSTR  AtomName 
)

Definition at line 32 of file atom.c.

35{
39 PUNICODE_STRING AtomNameString;
41
42 /* Check if it's an integer atom */
43 if ((ULONG_PTR)AtomName <= 0xFFFF)
44 {
45 /* Convert the name to an atom */
46 Atom = (ATOM)PtrToShort((PVOID)AtomName);
47 if (Atom >= MAXINTATOM)
48 {
49 /* Fail, atom number too large */
51 return INVALID_ATOM;
52 }
53
54 /* Return it */
55 return Atom;
56 }
57 else
58 {
59 /* Check if this is a unicode atom */
60 if (Unicode)
61 {
62 /* Use a unicode string */
63 AtomNameString = &UnicodeString;
64 RtlInitUnicodeString(AtomNameString, (LPWSTR)AtomName);
66 }
67 else
68 {
69 /* Use an ansi string */
70 RtlInitAnsiString(&AnsiString, AtomName );
71
72 /* Check if we can abuse the TEB */
73 if (AnsiString.MaximumLength > 260)
74 {
75 /* We can't, allocate a new string */
76 AtomNameString = &UnicodeString;
77 Status = RtlAnsiStringToUnicodeString(AtomNameString,
79 TRUE);
80 }
81 else
82 {
83 /* We can! Get the TEB String */
84 AtomNameString = &NtCurrentTeb()->StaticUnicodeString;
85
86 /* Convert it into the TEB */
87 Status = RtlAnsiStringToUnicodeString(AtomNameString,
89 FALSE);
90 }
91 }
92
93 /* Check for failure */
94 if (!NT_SUCCESS(Status))
95 {
97 return Atom;
98 }
99 }
100
101 /* Check if we're doing local add */
102 if (Local)
103 {
104 /* Do a local add */
106 AtomNameString->Buffer,
107 &Atom);
108 }
109 else
110 {
111 /* Do a global add */
112 Status = NtAddAtom(AtomNameString->Buffer,
113 AtomNameString->Length,
114 &Atom);
115 }
116
117 /* Check for failure */
119
120 /* Check if we were non-static ANSI */
121 if (!(Unicode) && (AtomNameString == &UnicodeString))
122 {
123 /* Free the allocated buffer */
124 RtlFreeUnicodeString(AtomNameString);
125 }
126
127 /* Return the atom */
128 return Atom;
129}
LONG NTSTATUS
Definition: precomp.h:26
#define PtrToShort(p)
Definition: basetsd.h:88
_Out_ RTL_ATOM * Atom
Definition: class.h:54
WORD ATOM
Definition: dimm.idl:113
PVOID WINAPI InternalInitAtomTable(VOID)
Definition: atom.c:23
@ AnsiString
Definition: dnslib.h:19
Status
Definition: gdiplustypes.h:25
#define NtCurrentTeb
NTSYSAPI NTSTATUS NTAPI RtlAddAtomToAtomTable(_In_ PRTL_ATOM_TABLE AtomTable, _In_ PWSTR AtomName, _Out_ PRTL_ATOM Atom)
NTSYSAPI NTSTATUS NTAPI RtlAnsiStringToUnicodeString(PUNICODE_STRING DestinationString, PANSI_STRING SourceString, BOOLEAN AllocateDestinationString)
NTSYSAPI VOID NTAPI RtlInitUnicodeString(PUNICODE_STRING DestinationString, PCWSTR SourceString)
NTSYSAPI VOID NTAPI RtlFreeUnicodeString(PUNICODE_STRING UnicodeString)
NTSYSAPI VOID NTAPI RtlInitAnsiString(PANSI_STRING DestinationString, PCSZ SourceString)
NTSTATUS NTAPI NtAddAtom(IN PWSTR AtomName, IN ULONG AtomNameLength, OUT PRTL_ATOM Atom)
Definition: atom.c:86
DWORD BaseSetLastNTError(IN NTSTATUS Status)
Definition: reactos.cpp:166
#define STATUS_SUCCESS
Definition: shellext.h:65
uint32_t ULONG_PTR
Definition: typedefs.h:65
#define STATUS_INVALID_PARAMETER
Definition: udferr_usr.h:135
#define INVALID_ATOM
Definition: winbase.h:450
#define MAXINTATOM
Definition: winbase.h:449
WCHAR * LPWSTR
Definition: xmlstorage.h:184

Referenced by AddAtomA(), AddAtomW(), GlobalAddAtomA(), and GlobalAddAtomW().

◆ InternalDeleteAtom()

ATOM WINAPI InternalDeleteAtom ( BOOLEAN  Local,
ATOM  Atom 
)

Definition at line 245 of file atom.c.

247{
249
250 /* Validate it */
251 if (Atom >= MAXINTATOM)
252 {
253 /* Check if it's a local delete */
254 if (Local)
255 {
256 /* Delete it locally */
258 }
259 else
260 {
261 /* Delete it globall */
263 }
264
265 /* Check for success */
266 if (!NT_SUCCESS(Status))
267 {
268 /* Fail */
270 return INVALID_ATOM;
271 }
272 }
273
274 /* Return failure */
275 return 0;
276}
NTSYSAPI NTSTATUS NTAPI RtlDeleteAtomFromAtomTable(_In_ PRTL_ATOM_TABLE AtomTable, _In_ RTL_ATOM Atom)
NTSTATUS NTAPI NtDeleteAtom(IN RTL_ATOM Atom)
Definition: atom.c:206

Referenced by DeleteAtom(), and GlobalDeleteAtom().

◆ InternalFindAtom()

ATOM WINAPI InternalFindAtom ( BOOLEAN  Local,
BOOLEAN  Unicode,
LPCSTR  AtomName 
)

Definition at line 133 of file atom.c.

136{
140 PUNICODE_STRING AtomNameString;
142
143 /* Check if it's an integer atom */
144 if ((ULONG_PTR)AtomName <= 0xFFFF)
145 {
146 /* Convert the name to an atom */
147 Atom = (ATOM)PtrToShort((PVOID)AtomName);
148 if (Atom >= MAXINTATOM)
149 {
150 /* Fail, atom number too large */
152 DPRINT1("Invalid atom\n");
153 }
154
155 /* Return it */
156 return Atom;
157 }
158 else
159 {
160 /* Check if this is a unicode atom */
161 if (Unicode)
162 {
163 /* Use a unicode string */
164 AtomNameString = &UnicodeString;
165 RtlInitUnicodeString(AtomNameString, (LPWSTR)AtomName);
167 }
168 else
169 {
170 /* Use an ansi string */
171 RtlInitAnsiString(&AnsiString, AtomName);
172
173 /* Check if we can abuse the TEB */
174 if (AnsiString.MaximumLength > 260)
175 {
176 /* We can't, allocate a new string */
177 AtomNameString = &UnicodeString;
178 Status = RtlAnsiStringToUnicodeString(AtomNameString,
179 &AnsiString,
180 TRUE);
181 }
182 else
183 {
184 /* We can! Get the TEB String */
185 AtomNameString = &NtCurrentTeb()->StaticUnicodeString;
186
187 /* Convert it into the TEB */
188 Status = RtlAnsiStringToUnicodeString(AtomNameString,
189 &AnsiString,
190 FALSE);
191 }
192 }
193
194 /* Check for failure */
195 if (!NT_SUCCESS(Status))
196 {
197 DPRINT1("Failed\n");
199 return Atom;
200 }
201 }
202
203 /* Check if we're doing local lookup */
204 if (Local)
205 {
206 /* Do a local lookup */
208 AtomNameString->Buffer,
209 &Atom);
210 }
211 else
212 {
213 /* Do a global search */
214 if (!AtomNameString->Length)
215 {
216 /* This is illegal in win32 */
217 DPRINT1("No name given\n");
219 }
220 else
221 {
222 /* Call the global function */
223 Status = NtFindAtom(AtomNameString->Buffer,
224 AtomNameString->Length,
225 &Atom);
226 }
227 }
228
229 /* Check for failure */
231
232 /* Check if we were non-static ANSI */
233 if (!(Unicode) && (AtomNameString == &UnicodeString))
234 {
235 /* Free the allocated buffer */
236 RtlFreeUnicodeString(AtomNameString);
237 }
238
239 /* Return the atom */
240 return Atom;
241}
#define DPRINT1
Definition: precomp.h:8
NTSYSAPI NTSTATUS NTAPI RtlLookupAtomInAtomTable(_In_ PRTL_ATOM_TABLE AtomTable, _In_ PWSTR AtomName, _Out_ PRTL_ATOM Atom)
NTSTATUS NTAPI NtFindAtom(IN PWSTR AtomName, IN ULONG AtomNameLength, OUT PRTL_ATOM Atom)
Definition: atom.c:243
#define STATUS_OBJECT_NAME_NOT_FOUND
Definition: udferr_usr.h:149

Referenced by FindAtomA(), FindAtomW(), GlobalFindAtomA(), and GlobalFindAtomW().

◆ InternalGetAtomName()

UINT WINAPI InternalGetAtomName ( BOOLEAN  Local,
BOOLEAN  Unicode,
ATOM  Atom,
LPSTR  AtomName,
DWORD  Size 
)

Definition at line 280 of file atom.c.

285{
287 DWORD RetVal = 0;
290 PVOID TempBuffer = NULL;
291 PWSTR AtomNameString;
292 ULONG AtomInfoLength;
293 ULONG AtomNameLength;
295
296 /* Normalize the size as not to overflow */
297 if (!Unicode && Size > 0x7000) Size = 0x7000;
298
299 /* Make sure it's valid too */
300 if (!Size)
301 {
303 return 0;
304 }
305 if (!Atom)
306 {
308 return 0;
309 }
310
311 /* Check if this is a global query */
312 if (Local)
313 {
314 /* Set the query length */
315 AtomNameLength = Size * sizeof(WCHAR);
316
317 /* If it's unicode, just keep the name */
318 if (Unicode)
319 {
320 AtomNameString = (PWSTR)AtomName;
321 }
322 else
323 {
324 /* Allocate memory for the ansi buffer */
325 TempBuffer = RtlAllocateHeap(RtlGetProcessHeap(),
326 0,
327 AtomNameLength);
328 AtomNameString = TempBuffer;
329 }
330
331 /* Query the name */
333 Atom,
334 NULL,
335 NULL,
336 AtomNameString,
337 &AtomNameLength);
338 }
339 else
340 {
341 /* We're going to do a global query, so allocate a buffer */
342 AtomInfoLength = sizeof(ATOM_BASIC_INFORMATION) +
343 (Size * sizeof(WCHAR));
344 AtomInfo = TempBuffer = RtlAllocateHeap(RtlGetProcessHeap(),
345 0,
346 AtomInfoLength);
347
348 if (!AtomInfo)
349 {
351 return 0;
352 }
353
354 /* Query the name */
357 AtomInfo,
358 AtomInfoLength,
359 &AtomInfoLength);
360 if (NT_SUCCESS(Status))
361 {
362 /* Success. Update the length and get the name */
363 AtomNameLength = (ULONG)AtomInfo->NameLength;
364 AtomNameString = AtomInfo->Name;
365 }
366 }
367
368 /* Check for global success */
369 if (NT_SUCCESS(Status))
370 {
371 /* Check if it was unicode */
372 if (Unicode)
373 {
374 /* We return the length in chars */
375 RetVal = AtomNameLength / sizeof(WCHAR);
376
377 /* Copy the memory if this was a global query */
378 if (AtomNameString != (PWSTR)AtomName)
379 {
380 RtlMoveMemory(AtomName, AtomNameString, AtomNameLength);
381 }
382
383 /* And null-terminate it if the buffer was too large */
384 if (RetVal < Size)
385 {
386 ((PWCHAR)AtomName)[RetVal] = UNICODE_NULL;
387 }
388 }
389 else
390 {
391 /* First create a unicode string with our data */
392 UnicodeString.Buffer = AtomNameString;
393 UnicodeString.Length = (USHORT)AtomNameLength;
394 UnicodeString.MaximumLength = (USHORT)(UnicodeString.Length +
395 sizeof(WCHAR));
396
397 /* Now prepare an ansi string for conversion */
398 AnsiString.Buffer = AtomName;
399 AnsiString.Length = 0;
400 AnsiString.MaximumLength = (USHORT)Size;
401
402 /* Convert it */
405 FALSE);
406
407 /* Return the length */
408 if (NT_SUCCESS(Status)) RetVal = AnsiString.Length;
409 }
410 }
411
412 /* Free the temporary buffer if we have one */
413 if (TempBuffer) RtlFreeHeap(RtlGetProcessHeap(), 0, TempBuffer);
414
415 /* Check for failure */
416 if (!NT_SUCCESS(Status))
417 {
418 /* Fail */
419 DPRINT("Failed: %lx\n", Status);
421 }
422
423 /* Return length */
424 return RetVal;
425}
PVOID NTAPI RtlAllocateHeap(IN PVOID HeapHandle, IN ULONG Flags, IN SIZE_T Size)
Definition: heap.c:590
BOOLEAN NTAPI RtlFreeHeap(IN PVOID HeapHandle, IN ULONG Flags, IN PVOID HeapBase)
Definition: heap.c:608
#define NULL
Definition: types.h:112
struct _ATOM_BASIC_INFORMATION ATOM_BASIC_INFORMATION
@ AtomBasicInformation
Definition: extypes.h:332
NTSYSAPI NTSTATUS NTAPI RtlQueryAtomInAtomTable(_In_ PRTL_ATOM_TABLE AtomTable, _In_ RTL_ATOM Atom, _Out_opt_ PULONG RefCount, _Out_opt_ PULONG PinCount, _Out_opt_z_bytecap_(*NameLength) PWSTR AtomName, _Inout_opt_ PULONG NameLength)
NTSYSAPI NTSTATUS NTAPI RtlUnicodeStringToAnsiString(PANSI_STRING DestinationString, PUNICODE_STRING SourceString, BOOLEAN AllocateDestinationString)
#define UNICODE_NULL
NTSTATUS NTAPI NtQueryInformationAtom(RTL_ATOM Atom, ATOM_INFORMATION_CLASS AtomInformationClass, PVOID AtomInformation, ULONG AtomInformationLength, PULONG ReturnLength)
Definition: atom.c:365
#define STATUS_NO_MEMORY
Definition: ntstatus.h:260
unsigned short USHORT
Definition: pedump.c:61
#define STATUS_BUFFER_OVERFLOW
Definition: shellext.h:66
uint16_t * PWSTR
Definition: typedefs.h:56
#define RtlMoveMemory(Destination, Source, Length)
Definition: typedefs.h:264
uint16_t * PWCHAR
Definition: typedefs.h:56
uint32_t ULONG
Definition: typedefs.h:59
_Must_inspect_result_ _In_ WDFDEVICE _In_ PWDF_DEVICE_PROPERTY_DATA _In_ DEVPROPTYPE _In_ ULONG Size
Definition: wdfdevice.h:4533
__wchar_t WCHAR
Definition: xmlstorage.h:180

Referenced by GetAtomNameA(), GetAtomNameW(), GlobalGetAtomNameA(), and GlobalGetAtomNameW().

◆ InternalInitAtomTable()

PVOID WINAPI InternalInitAtomTable ( VOID  )

Definition at line 23 of file atom.c.

24{
25 /* Create or return the local table */
27 return BaseLocalAtomTable;
28}

Referenced by InternalAddAtom(), InternalDeleteAtom(), InternalFindAtom(), and InternalGetAtomName().

Variable Documentation

◆ BaseLocalAtomTable

PRTL_ATOM_TABLE BaseLocalAtomTable = NULL

Definition at line 17 of file atom.c.

Referenced by InitAtomTable(), and InternalInitAtomTable().