Commit e8f37c7a authored by Owen Rudge's avatar Owen Rudge Committed by Alexandre Julliard

wsdapi: Implement WSD LinkedMemory functions.

parent 8d6cf800
...@@ -4,4 +4,5 @@ IMPORTS = kernel32 ...@@ -4,4 +4,5 @@ IMPORTS = kernel32
C_SRCS = \ C_SRCS = \
discovery.c \ discovery.c \
main.c main.c \
memory.c
/*
* Web Services on Devices
*
* Copyright 2017 Owen Rudge 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
*/
#include <stdarg.h>
#define COBJMACROS
#include "windef.h"
#include "winbase.h"
#include "wine/debug.h"
#include "wine/list.h"
#include "wsdapi.h"
WINE_DEFAULT_DEBUG_CHANNEL(wsdapi);
#define MEMORY_ALLOCATION_MAGIC 0xB10C5EED
#define ALIGNMENT (2 * sizeof(void*))
#define ROUND_TO_ALIGNMENT(size) (((size) + ALIGNMENT - 1) & ~(ALIGNMENT - 1))
#define MEMORY_ALLOCATION_SIZE ROUND_TO_ALIGNMENT(sizeof(struct memory_allocation))
struct memory_allocation
{
int magic;
struct list entry;
struct list children;
};
static struct memory_allocation *find_allocation(void *ptr)
{
struct memory_allocation *allocation;
if (ptr == NULL)
{
return NULL;
}
allocation = (struct memory_allocation *)((char *)ptr - MEMORY_ALLOCATION_SIZE);
if (allocation->magic != MEMORY_ALLOCATION_MAGIC)
{
return NULL;
}
return allocation;
}
static void free_allocation(struct memory_allocation *item)
{
struct memory_allocation *child, *cursor;
LIST_FOR_EACH_ENTRY_SAFE(child, cursor, &item->children, struct memory_allocation, entry)
{
free_allocation(child);
}
list_remove(&item->entry);
item->magic = 0;
HeapFree(GetProcessHeap(), 0, item);
}
void * WINAPI WSDAllocateLinkedMemory(void *pParent, SIZE_T cbSize)
{
struct memory_allocation *allocation, *parent;
void *ptr;
TRACE("(%p, %lu)\n", pParent, cbSize);
ptr = HeapAlloc(GetProcessHeap(), 0, MEMORY_ALLOCATION_SIZE + cbSize);
if (ptr == NULL)
{
return NULL;
}
allocation = ptr;
allocation->magic = MEMORY_ALLOCATION_MAGIC;
list_init(&allocation->children);
/* See if we have a parent */
parent = find_allocation(pParent);
if (parent != NULL)
{
list_add_tail(&parent->children, &allocation->entry);
}
else
{
list_init(&allocation->entry);
}
return (char *)ptr + MEMORY_ALLOCATION_SIZE;
}
void WINAPI WSDAttachLinkedMemory(void *pParent, void *pChild)
{
struct memory_allocation *parent, *child;
TRACE("(%p, %p)\n", pParent, pChild);
child = find_allocation(pChild);
parent = find_allocation(pParent);
TRACE("child: %p, parent: %p\n", child, parent);
if ((child == NULL) || (parent == NULL))
{
return;
}
list_remove(&child->entry);
list_add_tail(&parent->children, &child->entry);
}
void WINAPI WSDDetachLinkedMemory(void *pVoid)
{
struct memory_allocation *allocation;
TRACE("(%p)\n", pVoid);
allocation = find_allocation(pVoid);
if (allocation == NULL)
{
TRACE("Memory allocation not found\n");
return;
}
list_remove(&allocation->entry);
}
void WINAPI WSDFreeLinkedMemory(void *pVoid)
{
struct memory_allocation *allocation;
TRACE("(%p)\n", pVoid);
allocation = find_allocation(pVoid);
if (allocation == NULL)
{
TRACE("Memory allocation not found\n");
return;
}
free_allocation(allocation);
}
...@@ -2,4 +2,5 @@ TESTDLL = wsdapi.dll ...@@ -2,4 +2,5 @@ TESTDLL = wsdapi.dll
IMPORTS = wsdapi ole32 user32 IMPORTS = wsdapi ole32 user32
C_SRCS = \ C_SRCS = \
discovery.c discovery.c \
memory.c
/*
* Web Services on Devices
* Memory tests
*
* Copyright 2017 Owen Rudge 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
*/
#define COBJMACROS
#include <windows.h>
#include "wine/test.h"
#include "wsdapi.h"
#define ALIGNMENT (2 * sizeof(void*))
#define NUM_ITERATIONS 100
static void AllocateLinkedMemory_tests(void)
{
void *alloc1;
void *alloc2;
void *alloc_ptrs[NUM_ITERATIONS];
int i;
/* Test a parentless allocation */
alloc1 = WSDAllocateLinkedMemory(NULL, 1024);
ok(alloc1 != NULL, "WSDAllocateLinkedMemory with null parent failed\n");
WSDFreeLinkedMemory(alloc1);
/* Test a parented allocation */
alloc1 = WSDAllocateLinkedMemory(NULL, 1024);
ok(alloc1 != NULL, "WSDAllocateLinkedMemory with null parent failed\n");
alloc2 = WSDAllocateLinkedMemory(alloc1, 1024);
ok(alloc2 != NULL, "WSDAllocateLinkedMemory with parent failed\n");
WSDFreeLinkedMemory(alloc1);
/* Try allocating against the freed parent */
alloc2 = WSDAllocateLinkedMemory(alloc1, 1024);
ok(alloc2 != NULL, "WSDAllocateLinkedMemory with parent failed\n");
/* Test that the returned pointers are aligned as expected */
for (i = 0; i < NUM_ITERATIONS; i++)
{
alloc_ptrs[i] = WSDAllocateLinkedMemory(NULL, i);
ok(((UINT_PTR)alloc_ptrs[i] % ALIGNMENT) == 0, "WSDAllocateLinkedMemory returned pointer not aligned (size = %d).\n", i);
}
for (i = 0; i < NUM_ITERATIONS; i++)
{
WSDFreeLinkedMemory(alloc_ptrs[i]);
}
}
START_TEST(memory)
{
AllocateLinkedMemory_tests();
}
...@@ -4,8 +4,8 @@ ...@@ -4,8 +4,8 @@
@ stub WSDNotifyNetworkChange @ stub WSDNotifyNetworkChange
@ stub WSDRemoveFirewallCheck @ stub WSDRemoveFirewallCheck
@ stub WSDXMLCompareNames @ stub WSDXMLCompareNames
@ stub WSDAllocateLinkedMemory @ stdcall WSDAllocateLinkedMemory(ptr long)
@ stub WSDAttachLinkedMemory @ stdcall WSDAttachLinkedMemory(ptr ptr)
@ stub WSDCompareEndpoints @ stub WSDCompareEndpoints
@ stub WSDCopyEndpoint @ stub WSDCopyEndpoint
@ stub WSDCreateDeviceHost2 @ stub WSDCreateDeviceHost2
...@@ -26,8 +26,8 @@ ...@@ -26,8 +26,8 @@
@ stub WSDCreateUdpAddress @ stub WSDCreateUdpAddress
@ stub WSDCreateUdpMessageParameters @ stub WSDCreateUdpMessageParameters
@ stub WSDCreateUdpTransport @ stub WSDCreateUdpTransport
@ stub WSDDetachLinkedMemory @ stdcall WSDDetachLinkedMemory(ptr)
@ stub WSDFreeLinkedMemory @ stdcall WSDFreeLinkedMemory(ptr)
@ stub WSDGenerateFault @ stub WSDGenerateFault
@ stub WSDGenerateFaultEx @ stub WSDGenerateFaultEx
@ stub WSDGenerateRandomDelay @ stub WSDGenerateRandomDelay
......
...@@ -693,6 +693,7 @@ HEADER_SRCS = \ ...@@ -693,6 +693,7 @@ HEADER_SRCS = \
ws2tcpip.h \ ws2tcpip.h \
wsdapi.h \ wsdapi.h \
wsdtypes.h \ wsdtypes.h \
wsdutil.h \
wsdxmldom.h \ wsdxmldom.h \
wshisotp.h \ wshisotp.h \
wsipx.h \ wsipx.h \
......
...@@ -24,5 +24,6 @@ ...@@ -24,5 +24,6 @@
#include <wsdxmldom.h> #include <wsdxmldom.h>
#include <wsdtypes.h> #include <wsdtypes.h>
#include <wsddisco.h> #include <wsddisco.h>
#include <wsdutil.h>
#endif #endif
/*
* Copyright 2017 Owen Rudge 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 WSDUTIL_H
#define WSDUTIL_H
void * WINAPI WSDAllocateLinkedMemory(void *pParent, SIZE_T cbSize);
void WINAPI WSDAttachLinkedMemory(void *pParent, void *pChild);
void WINAPI WSDDetachLinkedMemory(void *pVoid);
void WINAPI WSDFreeLinkedMemory(void *pVoid);
#endif
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment