kernel32: Fix the 'MAP_FOLDCZONE' ok() test.
[wine] / dlls / kernel32 / atom16.c
1 /*
2  * Atom table functions - 16 bit variant
3  *
4  * Copyright 1993, 1994, 1995 Alexandre Julliard
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 /*
22  * Warning: The code assumes that LocalAlloc() returns a block aligned
23  * on a 4-bytes boundary (because of the shifting done in
24  * HANDLETOATOM).  If this is not the case, the allocation code will
25  * have to be changed.
26  */
27
28 #include "config.h"
29 #include "wine/port.h"
30
31 #include <stdlib.h>
32 #include <stdarg.h>
33 #include <stdio.h>
34 #include <string.h>
35 #include <ctype.h>
36
37 #include "windef.h"
38 #include "winbase.h"
39 #include "winerror.h"
40 #include "winternl.h"
41
42 #include "wine/unicode.h"
43 #include "wine/winbase16.h"
44 #include "kernel_private.h"
45 #include "kernel16_private.h"
46
47 #include "wine/debug.h"
48
49 WINE_DEFAULT_DEBUG_CHANNEL(atom);
50
51 #define DEFAULT_ATOMTABLE_SIZE    37
52 #define MAX_ATOM_LEN              255
53
54 #define ATOMTOHANDLE(atom)        ((HANDLE16)(atom) << 2)
55 #define HANDLETOATOM(handle)      ((ATOM)(0xc000 | ((handle) >> 2)))
56
57 typedef struct
58 {
59     HANDLE16    next;
60     WORD        refCount;
61     BYTE        length;
62     CHAR        str[1];
63 } ATOMENTRY;
64
65 typedef struct
66 {
67     WORD        size;
68     HANDLE16    entries[1];
69 } ATOMTABLE;
70
71
72 /***********************************************************************
73  *           ATOM_GetTable
74  *
75  * Return a pointer to the atom table of a given segment, creating
76  * it if necessary.
77  *
78  * RETURNS
79  *      Pointer to table: Success
80  *      NULL: Failure
81  */
82 static ATOMTABLE *ATOM_GetTable( BOOL create  /* [in] Create */ )
83 {
84     INSTANCEDATA *ptr = MapSL( MAKESEGPTR( CURRENT_DS, 0 ) );
85     if (ptr->atomtable)
86     {
87         ATOMTABLE *table = (ATOMTABLE *)((char *)ptr + ptr->atomtable);
88         if (table->size) return table;
89     }
90     if (!create) return NULL;
91     if (!InitAtomTable16( 0 )) return NULL;
92     /* Reload ptr in case it moved in linear memory */
93     ptr = MapSL( MAKESEGPTR( CURRENT_DS, 0 ) );
94     return (ATOMTABLE *)((char *)ptr + ptr->atomtable);
95 }
96
97
98 /***********************************************************************
99  *           ATOM_Hash
100  * RETURNS
101  *      The hash value for the input string
102  */
103 static WORD ATOM_Hash(
104             WORD entries, /* [in] Total number of entries */
105             LPCSTR str,   /* [in] Pointer to string to hash */
106             WORD len      /* [in] Length of string */
107 ) {
108     WORD i, hash = 0;
109
110     TRACE("%x, %s, %x\n", entries, str, len);
111
112     for (i = 0; i < len; i++) hash ^= toupper(str[i]) + i;
113     return hash % entries;
114 }
115
116
117 /***********************************************************************
118  *           ATOM_IsIntAtomA
119  */
120 static BOOL ATOM_IsIntAtomA(LPCSTR atomstr,WORD *atomid)
121 {
122     UINT atom = 0;
123     if (!HIWORD(atomstr)) atom = LOWORD(atomstr);
124     else
125     {
126         if (*atomstr++ != '#') return FALSE;
127         while (*atomstr >= '0' && *atomstr <= '9')
128         {
129             atom = atom * 10 + *atomstr - '0';
130             atomstr++;
131         }
132         if (*atomstr) return FALSE;
133     }
134     if (atom >= MAXINTATOM)
135     {
136         SetLastError( ERROR_INVALID_PARAMETER );
137         atom = 0;
138     }
139     *atomid = atom;
140     return TRUE;
141 }
142
143
144 /***********************************************************************
145  *           ATOM_MakePtr
146  *
147  * Make an ATOMENTRY pointer from a handle (obtained from GetAtomHandle()).
148  */
149 static inline ATOMENTRY *ATOM_MakePtr( HANDLE16 handle /* [in] Handle */ )
150 {
151     return MapSL( MAKESEGPTR( CURRENT_DS, handle ) );
152 }
153
154
155 /***********************************************************************
156  *           InitAtomTable   (KERNEL.68)
157  */
158 WORD WINAPI InitAtomTable16( WORD entries )
159 {
160     int i;
161     HANDLE16 handle;
162     ATOMTABLE *table;
163
164       /* Allocate the table */
165
166     if (!entries) entries = DEFAULT_ATOMTABLE_SIZE;  /* sanity check */
167     handle = LocalAlloc16( LMEM_FIXED, sizeof(ATOMTABLE) + (entries-1) * sizeof(HANDLE16) );
168     if (!handle) return 0;
169     table = MapSL( MAKESEGPTR( CURRENT_DS, handle ) );
170     table->size = entries;
171     for (i = 0; i < entries; i++) table->entries[i] = 0;
172
173       /* Store a pointer to the table in the instance data */
174
175     ((INSTANCEDATA *)MapSL( MAKESEGPTR( CURRENT_DS, 0 )))->atomtable = handle;
176     return handle;
177 }
178
179 /***********************************************************************
180  *           GetAtomHandle   (KERNEL.73)
181  */
182 HANDLE16 WINAPI GetAtomHandle16( ATOM atom )
183 {
184     if (atom < MAXINTATOM) return 0;
185     return ATOMTOHANDLE( atom );
186 }
187
188
189 /***********************************************************************
190  *           AddAtom   (KERNEL.70)
191  *
192  * Windows DWORD aligns the atom entry size.
193  * The remaining unused string space created by the alignment
194  * gets padded with '\0's in a certain way to ensure
195  * that at least one trailing '\0' remains.
196  *
197  * RETURNS
198  *      Atom: Success
199  *      0: Failure
200  */
201 ATOM WINAPI AddAtom16( LPCSTR str )
202 {
203     char buffer[MAX_ATOM_LEN+1];
204     WORD hash;
205     HANDLE16 entry;
206     ATOMENTRY * entryPtr;
207     ATOMTABLE * table;
208     int len, ae_len;
209     WORD iatom;
210
211     if (ATOM_IsIntAtomA( str, &iatom )) return iatom;
212
213     TRACE("%s\n",debugstr_a(str));
214
215     if (!(table = ATOM_GetTable( TRUE ))) return 0;
216
217     /* Make a copy of the string to be sure it doesn't move in linear memory. */
218     lstrcpynA( buffer, str, sizeof(buffer) );
219
220     len = strlen( buffer );
221     hash = ATOM_Hash( table->size, buffer, len );
222     entry = table->entries[hash];
223     while (entry)
224     {
225         entryPtr = ATOM_MakePtr( entry );
226         if ((entryPtr->length == len) &&
227             (!strncasecmp( entryPtr->str, buffer, len )))
228         {
229             entryPtr->refCount++;
230             TRACE("-- existing 0x%x\n", entry);
231             return HANDLETOATOM( entry );
232         }
233         entry = entryPtr->next;
234     }
235
236     ae_len = (sizeof(ATOMENTRY)+len+3) & ~3;
237     entry = LocalAlloc16( LMEM_FIXED, ae_len );
238     if (!entry) return 0;
239     /* Reload the table ptr in case it moved in linear memory */
240     table = ATOM_GetTable( FALSE );
241     entryPtr = ATOM_MakePtr( entry );
242     entryPtr->next = table->entries[hash];
243     entryPtr->refCount = 1;
244     entryPtr->length = len;
245     memcpy( entryPtr->str, buffer, len);
246     /* Some applications _need_ the '\0' padding provided by memset */
247     /* Note that 1 byte of the str is accounted for in the ATOMENTRY struct */
248     memset( entryPtr->str+len, 0, ae_len - sizeof(ATOMENTRY) - (len - 1));
249     table->entries[hash] = entry;
250     TRACE("-- new 0x%x\n", entry);
251     return HANDLETOATOM( entry );
252 }
253
254
255 /***********************************************************************
256  *           DeleteAtom   (KERNEL.71)
257  */
258 ATOM WINAPI DeleteAtom16( ATOM atom )
259 {
260     ATOMENTRY * entryPtr;
261     ATOMTABLE * table;
262     HANDLE16 entry, *prevEntry;
263     WORD hash;
264
265     if (atom < MAXINTATOM) return 0;  /* Integer atom */
266
267     TRACE("0x%x\n",atom);
268
269     if (!(table = ATOM_GetTable( FALSE ))) return 0;
270     entry = ATOMTOHANDLE( atom );
271     entryPtr = ATOM_MakePtr( entry );
272
273     /* Find previous atom */
274     hash = ATOM_Hash( table->size, entryPtr->str, entryPtr->length );
275     prevEntry = &table->entries[hash];
276     while (*prevEntry && *prevEntry != entry)
277     {
278         ATOMENTRY * prevEntryPtr = ATOM_MakePtr( *prevEntry );
279         prevEntry = &prevEntryPtr->next;
280     }
281     if (!*prevEntry) return atom;
282
283     /* Delete atom */
284     if (--entryPtr->refCount == 0)
285     {
286         *prevEntry = entryPtr->next;
287         LocalFree16( entry );
288     }
289     return 0;
290 }
291
292
293 /***********************************************************************
294  *           FindAtom   (KERNEL.69)
295  */
296 ATOM WINAPI FindAtom16( LPCSTR str )
297 {
298     ATOMTABLE * table;
299     WORD hash,iatom;
300     HANDLE16 entry;
301     int len;
302
303     TRACE("%s\n",debugstr_a(str));
304
305     if (ATOM_IsIntAtomA( str, &iatom )) return iatom;
306     if ((len = strlen( str )) > 255) len = 255;
307     if (!(table = ATOM_GetTable( FALSE ))) return 0;
308     hash = ATOM_Hash( table->size, str, len );
309     entry = table->entries[hash];
310     while (entry)
311     {
312         ATOMENTRY * entryPtr = ATOM_MakePtr( entry );
313         if ((entryPtr->length == len) &&
314             (!strncasecmp( entryPtr->str, str, len )))
315         {
316             TRACE("-- found %x\n", entry);
317             return HANDLETOATOM( entry );
318         }
319         entry = entryPtr->next;
320     }
321     TRACE("-- not found\n");
322     return 0;
323 }
324
325
326 /***********************************************************************
327  *           GetAtomName   (KERNEL.72)
328  */
329 UINT16 WINAPI GetAtomName16( ATOM atom, LPSTR buffer, INT16 count )
330 {
331     ATOMENTRY * entryPtr;
332     HANDLE16 entry;
333     char * strPtr;
334     INT len;
335     char text[8];
336
337     TRACE("%x\n",atom);
338
339     if (!count) return 0;
340     if (atom < MAXINTATOM)
341     {
342         sprintf( text, "#%d", atom );
343         len = strlen(text);
344         strPtr = text;
345     }
346     else
347     {
348         if (!ATOM_GetTable( FALSE )) return 0;
349         entry = ATOMTOHANDLE( atom );
350         entryPtr = ATOM_MakePtr( entry );
351         len = entryPtr->length;
352         strPtr = entryPtr->str;
353     }
354     if (len >= count) len = count-1;
355     memcpy( buffer, strPtr, len );
356     buffer[len] = '\0';
357     return len;
358 }