clientobj.c

Go to the documentation of this file.

/*
 * COPYRIGHT:       See COPYING in the top level directory
 * PROJECT:         ReactOS WIN32 subsystem
 * PURPOSE:         Support functions for GDI client objects
 * PROGRAMMER:      Timo Kreuzer (timo.kreuzer@reactos.org)
 */
 
#include <precomp.h>
 
CRITICAL_SECTION gcsClientObjLinks;
ULONG gcClientObj;
 
typedef struct _CLIENTOBJLINK
{
    struct _CLIENTOBJLINK *pcolNext;
    HGDIOBJ hobj;
    PVOID pvObj;
} CLIENTOBJLINK, *PCLIENTOBJLINK;
 
PCLIENTOBJLINK gapcolHashTable[127];
 
BOOL
WINAPI
GdiCreateClientObjLink(
    _In_ HGDIOBJ hobj,
    _In_ PVOID pvObject)
{
    PCLIENTOBJLINK pcol;
    ULONG iHashIndex;
 
    /* Allocate a link structure */
    pcol = HeapAlloc(GetProcessHeap(), 0, sizeof(*pcol));
    if (pcol == NULL)
    {
        return FALSE;
    }
 
    /* Setup the link structure */
    pcol->hobj = hobj;
    pcol->pvObj = pvObject;
 
    /* Calculate the hash index */
    iHashIndex = (ULONG_PTR)hobj % _countof(gapcolHashTable);
 
    /* Enter the critical section */
    EnterCriticalSection(&gcsClientObjLinks);
 
    /* Insert the link structure */
    pcol->pcolNext = gapcolHashTable[iHashIndex];
    gapcolHashTable[iHashIndex] = pcol;
    gcClientObj++;
 
    /* Leave the critical section */
    LeaveCriticalSection(&gcsClientObjLinks);
 
    return TRUE;
}
 
PVOID
WINAPI
GdiGetClientObjLink(
    _In_ HGDIOBJ hobj)
{
    ULONG iHashIndex;
    PCLIENTOBJLINK pcol;
    PVOID pvObject = NULL;
 
    /* Calculate the hash index */
    iHashIndex = (ULONG_PTR)hobj % _countof(gapcolHashTable);
 
    /* Enter the critical section */
    EnterCriticalSection(&gcsClientObjLinks);
 
    /* Loop the link entries in this hash bucket */
    pcol = gapcolHashTable[iHashIndex];
    while (pcol != NULL)
    {
        /* Check if this is the object we want */
        if (pcol->hobj == hobj)
        {
            /* Get the object pointer and bail out */
            pvObject = pcol->pvObj;
            break;
        }
 
        /* Go to the next entry */
        pcol = pcol->pcolNext;
    }
 
    /* Leave the critical section */
    LeaveCriticalSection(&gcsClientObjLinks);
 
    return pvObject;
}
 
PVOID
WINAPI
GdiRemoveClientObjLink(
    _In_ HGDIOBJ hobj)
{
    PCLIENTOBJLINK pcol, *ppcol;
    ULONG iHashIndex;
    PVOID pvObject = NULL;
 
    /* Calculate the hash index */
    iHashIndex = (ULONG_PTR)hobj % _countof(gapcolHashTable);
 
    /* Enter the critical section */
    EnterCriticalSection(&gcsClientObjLinks);
 
    /* Loop the link entries in this hash bucket */
    ppcol = &gapcolHashTable[iHashIndex];
    while (*ppcol != NULL)
    {
        /* Get the current client object link */
        pcol = *ppcol;
 
        /* Check if this is the one we want */
        if (pcol->hobj == hobj)
        {
            /* Update the link pointer, removing this link */
            *ppcol = pcol->pcolNext;
            gcClientObj--;
 
            /* Get the object pointer */
            pvObject = pcol->pvObj;
 
            /* Free the link structure */
            HeapFree(GetProcessHeap(), 0, pcol);
 
            /* We're done */
            break;
        }
 
        /* Go to the next link pointer */
        ppcol = &(pcol->pcolNext);
    }
 
    /* Leave the critical section */
    LeaveCriticalSection(&gcsClientObjLinks);
 
    /* Return the object pointer, or NULL if we did not find it */
    return pvObject;
}
 
HGDIOBJ
WINAPI
GdiCreateClientObj(
    _In_ PVOID pvObject,
    _In_ GDILOOBJTYPE eObjType)
{
    HGDIOBJ hobj;
 
    /* Call win32k to create a client object handle */
    hobj = NtGdiCreateClientObj(eObjType);
    if (hobj == NULL)
    {
        return NULL;
    }
 
    /* Create the client object link */
    if (!GdiCreateClientObjLink(hobj, pvObject))
    {
        NtGdiDeleteClientObj(hobj);
        return NULL;
    }
 
    return hobj;
}
 
PVOID
WINAPI
GdiDeleteClientObj(
    _In_ HGDIOBJ hobj)
{
    PVOID pvObject;
 
    /* Remove the client object link */
    pvObject = GdiRemoveClientObjLink(hobj);
    if (pvObject == NULL)
    {
        return NULL;
    }
 
    /* Call win32k to delete the handle */
    if (!NtGdiDeleteClientObj(hobj))
    {
        ASSERT(FALSE);
    }
 
    return pvObject;
}