303 lines
8.6 KiB
C
303 lines
8.6 KiB
C
/*
|
|
* Atom table functions
|
|
*
|
|
* Copyright 1993, 1994, 1995 Alexandre Julliard
|
|
* Copyright 2004,2005 Eric Pouech
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
|
|
#include "ntstatus.h"
|
|
#define WIN32_NO_STATUS
|
|
#include "windef.h"
|
|
|
|
#include "wine/server.h"
|
|
#include "ntdll_misc.h"
|
|
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(atom);
|
|
|
|
#define MAX_ATOM_LEN 255
|
|
#define IS_INTATOM(x) (((ULONG_PTR)(x) >> 16) == 0)
|
|
|
|
/******************************************************************
|
|
* is_integral_atom
|
|
* Returns STATUS_SUCCESS if integral atom and 'pAtom' is filled
|
|
* STATUS_INVALID_PARAMETER if 'atomstr' is too long
|
|
* STATUS_MORE_ENTRIES otherwise
|
|
*/
|
|
static NTSTATUS is_integral_atom( LPCWSTR atomstr, size_t len, RTL_ATOM* pAtom )
|
|
{
|
|
RTL_ATOM atom;
|
|
|
|
if (!IS_INTATOM( atomstr ))
|
|
{
|
|
const WCHAR* ptr = atomstr;
|
|
if (!len) return STATUS_OBJECT_NAME_INVALID;
|
|
|
|
if (*ptr++ == '#')
|
|
{
|
|
atom = 0;
|
|
while (ptr < atomstr + len && *ptr >= '0' && *ptr <= '9')
|
|
{
|
|
atom = atom * 10 + *ptr++ - '0';
|
|
}
|
|
if (ptr > atomstr + 1 && ptr == atomstr + len) goto done;
|
|
}
|
|
if (len > MAX_ATOM_LEN) return STATUS_INVALID_PARAMETER;
|
|
return STATUS_MORE_ENTRIES;
|
|
}
|
|
else atom = LOWORD( atomstr );
|
|
done:
|
|
if (!atom || atom >= MAXINTATOM) return STATUS_INVALID_PARAMETER;
|
|
*pAtom = atom;
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlDeleteAtomFromAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlDeleteAtomFromAtomTable( RTL_ATOM_TABLE table, RTL_ATOM atom )
|
|
{
|
|
NTSTATUS status;
|
|
|
|
TRACE( "%p %x\n", table, atom );
|
|
if (!table) status = STATUS_INVALID_PARAMETER;
|
|
else
|
|
{
|
|
SERVER_START_REQ( delete_atom )
|
|
{
|
|
req->atom = atom;
|
|
req->table = wine_server_obj_handle( table );
|
|
status = wine_server_call( req );
|
|
}
|
|
SERVER_END_REQ;
|
|
}
|
|
return status;
|
|
}
|
|
|
|
/******************************************************************
|
|
* integral_atom_name (internal)
|
|
*
|
|
* Helper for fetching integral (local/global) atoms names.
|
|
*/
|
|
static ULONG integral_atom_name(WCHAR* buffer, ULONG len, RTL_ATOM atom)
|
|
{
|
|
static const WCHAR fmt[] = {'#','%','u',0};
|
|
WCHAR tmp[16];
|
|
int ret;
|
|
|
|
ret = swprintf( tmp, ARRAY_SIZE(tmp), fmt, atom );
|
|
if (!len) return ret * sizeof(WCHAR);
|
|
if (len <= ret) ret = len - 1;
|
|
memcpy( buffer, tmp, ret * sizeof(WCHAR) );
|
|
buffer[ret] = 0;
|
|
return ret * sizeof(WCHAR);
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlQueryAtomInAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlQueryAtomInAtomTable( RTL_ATOM_TABLE table, RTL_ATOM atom, ULONG* ref,
|
|
ULONG* pin, WCHAR* name, ULONG* len )
|
|
{
|
|
NTSTATUS status = STATUS_SUCCESS;
|
|
DWORD wlen = 0;
|
|
|
|
if (!table) status = STATUS_INVALID_PARAMETER;
|
|
else if (atom < MAXINTATOM)
|
|
{
|
|
if (!atom) return STATUS_INVALID_PARAMETER;
|
|
if (len) wlen = integral_atom_name( name, *len, atom);
|
|
if (ref) *ref = 1;
|
|
if (pin) *pin = 1;
|
|
}
|
|
else
|
|
{
|
|
SERVER_START_REQ( get_atom_information )
|
|
{
|
|
req->atom = atom;
|
|
req->table = wine_server_obj_handle( table );
|
|
if (len && *len && name)
|
|
wine_server_set_reply( req, name, *len );
|
|
status = wine_server_call( req );
|
|
if (status == STATUS_SUCCESS)
|
|
{
|
|
wlen = reply->total;
|
|
if (ref) *ref = reply->count;
|
|
if (pin) *pin = reply->pinned;
|
|
}
|
|
}
|
|
SERVER_END_REQ;
|
|
}
|
|
if (status == STATUS_SUCCESS && len)
|
|
{
|
|
if (*len)
|
|
{
|
|
wlen = min( *len - sizeof(WCHAR), wlen );
|
|
if (name) name[wlen / sizeof(WCHAR)] = 0;
|
|
}
|
|
else status = STATUS_BUFFER_TOO_SMALL;
|
|
*len = wlen;
|
|
}
|
|
|
|
TRACE( "%p %x -> %s (%x)\n",
|
|
table, atom, len ? debugstr_wn(name, wlen / sizeof(WCHAR)) : "(null)", status );
|
|
return status;
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlCreateAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlCreateAtomTable( ULONG size, RTL_ATOM_TABLE* table )
|
|
{
|
|
NTSTATUS status;
|
|
|
|
if (*table)
|
|
{
|
|
if (size) status = STATUS_INVALID_PARAMETER;
|
|
else status = STATUS_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
SERVER_START_REQ( init_atom_table )
|
|
{
|
|
req->entries = size;
|
|
status = wine_server_call( req );
|
|
*table = wine_server_ptr_handle( reply->table );
|
|
}
|
|
SERVER_END_REQ;
|
|
}
|
|
return status;
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlDestroyAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlDestroyAtomTable( RTL_ATOM_TABLE table )
|
|
{
|
|
if (!table) return STATUS_INVALID_PARAMETER;
|
|
return NtClose( table );
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlAddAtomToAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlAddAtomToAtomTable( RTL_ATOM_TABLE table, const WCHAR* name, RTL_ATOM* atom )
|
|
{
|
|
NTSTATUS status;
|
|
|
|
if (!table) status = STATUS_INVALID_PARAMETER;
|
|
else
|
|
{
|
|
size_t len = IS_INTATOM(name) ? 0 : wcslen(name);
|
|
status = is_integral_atom( name, len, atom );
|
|
if (status == STATUS_MORE_ENTRIES)
|
|
{
|
|
SERVER_START_REQ( add_atom )
|
|
{
|
|
wine_server_add_data( req, name, len * sizeof(WCHAR) );
|
|
req->table = wine_server_obj_handle( table );
|
|
status = wine_server_call( req );
|
|
*atom = reply->atom;
|
|
}
|
|
SERVER_END_REQ;
|
|
}
|
|
}
|
|
TRACE( "%p %s -> %x\n",
|
|
table, debugstr_w(name), status == STATUS_SUCCESS ? *atom : 0 );
|
|
|
|
return status;
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlLookupAtomInAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlLookupAtomInAtomTable( RTL_ATOM_TABLE table, const WCHAR* name, RTL_ATOM* atom )
|
|
{
|
|
NTSTATUS status;
|
|
|
|
if (!table) status = STATUS_INVALID_PARAMETER;
|
|
else
|
|
{
|
|
size_t len = IS_INTATOM(name) ? 0 : wcslen(name);
|
|
status = is_integral_atom( name, len, atom );
|
|
if (status == STATUS_MORE_ENTRIES)
|
|
{
|
|
SERVER_START_REQ( find_atom )
|
|
{
|
|
wine_server_add_data( req, name, len * sizeof(WCHAR) );
|
|
req->table = wine_server_obj_handle( table );
|
|
status = wine_server_call( req );
|
|
*atom = reply->atom;
|
|
}
|
|
SERVER_END_REQ;
|
|
}
|
|
}
|
|
TRACE( "%p %s -> %x\n",
|
|
table, debugstr_w(name), status == STATUS_SUCCESS ? *atom : 0 );
|
|
return status;
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlEmptyAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlEmptyAtomTable( RTL_ATOM_TABLE table, BOOLEAN delete_pinned )
|
|
{
|
|
NTSTATUS status;
|
|
|
|
if (!table) status = STATUS_INVALID_PARAMETER;
|
|
else
|
|
{
|
|
SERVER_START_REQ( empty_atom_table )
|
|
{
|
|
req->table = wine_server_obj_handle( table );
|
|
req->if_pinned = delete_pinned;
|
|
status = wine_server_call( req );
|
|
}
|
|
SERVER_END_REQ;
|
|
}
|
|
return status;
|
|
}
|
|
|
|
/******************************************************************
|
|
* RtlPinAtomInAtomTable (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI RtlPinAtomInAtomTable( RTL_ATOM_TABLE table, RTL_ATOM atom )
|
|
{
|
|
NTSTATUS status;
|
|
|
|
if (!table) return STATUS_INVALID_PARAMETER;
|
|
if (atom < MAXINTATOM) return STATUS_SUCCESS;
|
|
|
|
SERVER_START_REQ( set_atom_information )
|
|
{
|
|
req->table = wine_server_obj_handle( table );
|
|
req->atom = atom;
|
|
req->pinned = TRUE;
|
|
status = wine_server_call( req );
|
|
}
|
|
SERVER_END_REQ;
|
|
|
|
return status;
|
|
}
|