Module: wine Branch: master Commit: d5c0ccf93aed9f2d61734d2f8d3e2f5bee6744e7 URL: http://source.winehq.org/git/wine.git/?a=commit;h=d5c0ccf93aed9f2d61734d2f8d...
Author: Hans Leidekker hans@meelstraat.net Date: Fri Aug 15 14:41:36 2008 +0200
winhttp: Add a handle management implementation.
---
dlls/winhttp/Makefile.in | 4 +- dlls/winhttp/handle.c | 156 ++++++++++++++++++++++++++++++++++++++++ dlls/winhttp/winhttp_private.h | 68 +++++++++++++++++ 3 files changed, 227 insertions(+), 1 deletions(-)
diff --git a/dlls/winhttp/Makefile.in b/dlls/winhttp/Makefile.in index e69b137..bf26202 100644 --- a/dlls/winhttp/Makefile.in +++ b/dlls/winhttp/Makefile.in @@ -6,7 +6,9 @@ MODULE = winhttp.dll IMPORTLIB = winhttp IMPORTS = kernel32
-C_SRCS = main.c +C_SRCS = \ + handle.c \ + main.c
@MAKE_DLL_RULES@
diff --git a/dlls/winhttp/handle.c b/dlls/winhttp/handle.c new file mode 100644 index 0000000..8886788 --- /dev/null +++ b/dlls/winhttp/handle.c @@ -0,0 +1,156 @@ +/* + * Copyright 2008 Hans Leidekker for CodeWeavers + * + * Based on the handle implementation from wininet. + * + * 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 "config.h" +#include "wine/port.h" +#include "wine/debug.h" + +#include <stdarg.h> + +#include "windef.h" +#include "winbase.h" +#include "winnls.h" +#include "winhttp.h" + +#include "winhttp_private.h" + +WINE_DEFAULT_DEBUG_CHANNEL(winhttp); + +#define HANDLE_CHUNK_SIZE 0x10 + +static CRITICAL_SECTION handle_cs; +static CRITICAL_SECTION_DEBUG handle_cs_debug = +{ + 0, 0, &handle_cs, + { &handle_cs_debug.ProcessLocksList, &handle_cs_debug.ProcessLocksList }, + 0, 0, { (ULONG_PTR)(__FILE__ ": handle_cs") } +}; +static CRITICAL_SECTION handle_cs = { &handle_cs_debug, -1, 0, 0, 0, 0 }; + +static object_header_t **handles; +static ULONG_PTR next_handle; +static ULONG_PTR max_handles; + +object_header_t *addref_object( object_header_t *hdr ) +{ + ULONG refs = InterlockedIncrement( &hdr->refs ); + TRACE("%p -> refcount = %d\n", hdr, refs); + return hdr; +} + +object_header_t *grab_object( HINTERNET hinternet ) +{ + object_header_t *hdr = NULL; + ULONG_PTR handle = (ULONG_PTR)hinternet; + + EnterCriticalSection( &handle_cs ); + + if ((handle > 0) && (handle <= max_handles) && handles[handle - 1]) + hdr = addref_object( handles[handle - 1] ); + + LeaveCriticalSection( &handle_cs ); + + TRACE("handle 0x%lx -> %p\n", handle, hdr); + return hdr; +} + +void release_object( object_header_t *hdr ) +{ + ULONG refs = InterlockedDecrement( &hdr->refs ); + TRACE("object %p refcount = %d\n", hdr, refs); + if (!refs) + { + TRACE("destroying object %p\n", hdr); + if (hdr->type != WINHTTP_HANDLE_TYPE_SESSION) list_remove( &hdr->entry ); + hdr->vtbl->destroy( hdr ); + } +} + +HINTERNET alloc_handle( object_header_t *hdr ) +{ + object_header_t **p; + ULONG_PTR handle = 0, num; + + list_init( &hdr->children ); + + EnterCriticalSection( &handle_cs ); + if (!max_handles) + { + num = HANDLE_CHUNK_SIZE; + if (!(p = heap_alloc_zero( sizeof(ULONG_PTR) * num ))) goto end; + handles = p; + max_handles = num; + } + if (max_handles == next_handle) + { + num = max_handles + HANDLE_CHUNK_SIZE; + if (!(p = heap_realloc_zero( handles, sizeof(ULONG_PTR) * num ))) goto end; + handles = p; + max_handles = num; + } + handle = next_handle; + if (handles[handle]) ERR("handle isn't free but should be\n"); + + handles[handle] = addref_object( hdr ); + while (handles[next_handle] && (next_handle < max_handles)) next_handle++; + +end: + LeaveCriticalSection( &handle_cs ); + return hdr->handle = (HINTERNET)(handle + 1); +} + +BOOL free_handle( HINTERNET hinternet ) +{ + BOOL ret = FALSE; + ULONG_PTR handle = (ULONG_PTR)hinternet; + object_header_t *hdr = NULL, *child, *next; + + EnterCriticalSection( &handle_cs ); + + if ((handle > 0) && (handle <= max_handles)) + { + handle--; + if (handles[handle]) + { + hdr = handles[handle]; + TRACE("destroying handle 0x%lx for object %p\n", handle + 1, hdr); + handles[handle] = NULL; + ret = TRUE; + } + } + + LeaveCriticalSection( &handle_cs ); + + if (hdr) + { + LIST_FOR_EACH_ENTRY_SAFE( child, next, &hdr->children, object_header_t, entry ) + { + TRACE("freeing child handle %p for parent handle 0x%lx\n", child->handle, handle + 1); + free_handle( child->handle ); + } + release_object( hdr ); + } + + EnterCriticalSection( &handle_cs ); + if (next_handle > handle && !handles[handle]) next_handle = handle; + LeaveCriticalSection( &handle_cs ); + + return ret; +} diff --git a/dlls/winhttp/winhttp_private.h b/dlls/winhttp/winhttp_private.h new file mode 100644 index 0000000..7d06c87 --- /dev/null +++ b/dlls/winhttp/winhttp_private.h @@ -0,0 +1,68 @@ +/* + * Copyright 2008 Hans Leidekker for CodeWeavers + * + * 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 + */ + +#ifndef _WINE_WINHTTP_PRIVATE_H_ +#define _WINE_WINHTTP_PRIVATE_H_ + +#include "wine/list.h" + +typedef struct _object_header_t object_header_t; + +typedef struct +{ + void (*destroy)( object_header_t * ); + BOOL (*query_option)( object_header_t *, DWORD, void *, DWORD *, BOOL ); + BOOL (*set_option)( object_header_t *, DWORD, void *, DWORD ); +} object_vtbl_t; + +struct _object_header_t +{ + DWORD type; + HINTERNET handle; + const object_vtbl_t *vtbl; + DWORD flags; + DWORD error; + DWORD_PTR context; + LONG refs; + WINHTTP_STATUS_CALLBACK callback; + DWORD notify_mask; + struct list entry; + struct list children; +}; + +static inline void *heap_alloc( SIZE_T size ) +{ + return HeapAlloc( GetProcessHeap(), 0, size ); +} + +static inline void *heap_alloc_zero( SIZE_T size ) +{ + return HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ); +} + +static inline void *heap_realloc_zero( LPVOID mem, SIZE_T size ) +{ + return HeapReAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, mem, size ); +} + +static inline BOOL heap_free( LPVOID mem ) +{ + return HeapFree( GetProcessHeap(), 0, mem ); +} + +#endif /* _WINE_WINHTTP_PRIVATE_H_ */