Hans Leidekker : winhttp: Add a handle management implementation.
Alexandre Julliard
julliard at winehq.org
Tue Aug 19 08:46:57 CDT 2008
Module: wine
Branch: master
Commit: d5c0ccf93aed9f2d61734d2f8d3e2f5bee6744e7
URL: http://source.winehq.org/git/wine.git/?a=commit;h=d5c0ccf93aed9f2d61734d2f8d3e2f5bee6744e7
Author: Hans Leidekker <hans at 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_ */
More information about the wine-cvs
mailing list