192 lines
3.3 KiB
C++
192 lines
3.3 KiB
C++
/*
|
|
* PolRef.cpp
|
|
*
|
|
* Author: BreenH
|
|
*
|
|
* Policy reference list code.
|
|
*/
|
|
|
|
/*
|
|
* Includes
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
#include "lscore.h"
|
|
#include "session.h"
|
|
#include "policy.h"
|
|
#include "pollist.h"
|
|
|
|
/*
|
|
* Globals
|
|
*/
|
|
|
|
LIST_ENTRY g_PolicyList;
|
|
ULONG g_cPolicies;
|
|
|
|
/*
|
|
* Internal Function Prototypes
|
|
*/
|
|
|
|
LPLCPOLICYREF
|
|
FindPolicyListEntry(
|
|
ULONG ulPolicyId
|
|
);
|
|
|
|
/*
|
|
* Function Implementations
|
|
*/
|
|
|
|
NTSTATUS
|
|
PolicyListAdd(
|
|
CPolicy *pPolicy
|
|
)
|
|
{
|
|
LPLCPOLICYREF lpPolicyRef;
|
|
|
|
ASSERT(pPolicy != NULL);
|
|
|
|
lpPolicyRef = (LPLCPOLICYREF)LocalAlloc(LPTR, sizeof(LCPOLICYREF));
|
|
|
|
if (lpPolicyRef != NULL)
|
|
{
|
|
lpPolicyRef->pPolicy = pPolicy;
|
|
|
|
g_cPolicies++;
|
|
InsertTailList(&g_PolicyList, &(lpPolicyRef->ListEntry));
|
|
return(STATUS_SUCCESS);
|
|
}
|
|
else
|
|
{
|
|
return(STATUS_NO_MEMORY);
|
|
}
|
|
}
|
|
|
|
VOID
|
|
PolicyListDelete(
|
|
ULONG ulPolicyId
|
|
)
|
|
{
|
|
LPLCPOLICYREF lpPolicyRef;
|
|
|
|
lpPolicyRef = FindPolicyListEntry(ulPolicyId);
|
|
|
|
if (lpPolicyRef != NULL)
|
|
{
|
|
g_cPolicies--;
|
|
RemoveEntryList(&(lpPolicyRef->ListEntry));
|
|
LocalFree(lpPolicyRef);
|
|
}
|
|
}
|
|
|
|
NTSTATUS
|
|
PolicyListEnumerateIds(
|
|
PULONG *ppulPolicyIds,
|
|
PULONG pcPolicies
|
|
)
|
|
{
|
|
NTSTATUS Status;
|
|
|
|
*ppulPolicyIds = (PULONG)MIDL_user_allocate(g_cPolicies * sizeof(ULONG));
|
|
|
|
if (*ppulPolicyIds != NULL)
|
|
{
|
|
LPLCPOLICYREF pTemp;
|
|
ULONG i;
|
|
|
|
pTemp = CONTAINING_RECORD(g_PolicyList.Flink, LCPOLICYREF, ListEntry);
|
|
|
|
for (i = 0; i < g_cPolicies; i++)
|
|
{
|
|
(*ppulPolicyIds)[i] = pTemp->pPolicy->GetId();
|
|
|
|
pTemp = CONTAINING_RECORD(pTemp->ListEntry.Flink, LCPOLICYREF, ListEntry);
|
|
}
|
|
|
|
*pcPolicies = g_cPolicies;
|
|
Status = STATUS_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
*pcPolicies = 0;
|
|
Status = STATUS_NO_MEMORY;
|
|
}
|
|
|
|
return(Status);
|
|
}
|
|
|
|
CPolicy *
|
|
PolicyListFindById(
|
|
ULONG ulPolicyId
|
|
)
|
|
{
|
|
LPLCPOLICYREF lpPolicyRef;
|
|
|
|
lpPolicyRef = FindPolicyListEntry(ulPolicyId);
|
|
|
|
return(lpPolicyRef != NULL ? lpPolicyRef->pPolicy : NULL);
|
|
}
|
|
|
|
NTSTATUS
|
|
PolicyListInitialize(
|
|
VOID
|
|
)
|
|
{
|
|
g_cPolicies = 0;
|
|
InitializeListHead(&g_PolicyList);
|
|
|
|
return(STATUS_SUCCESS);
|
|
}
|
|
|
|
CPolicy *
|
|
PolicyListPop(
|
|
VOID
|
|
)
|
|
{
|
|
CPolicy *pPolicy;
|
|
PLIST_ENTRY pTemp;
|
|
LPLCPOLICYREF lpCurrentRef;
|
|
|
|
if (!IsListEmpty(&g_PolicyList))
|
|
{
|
|
pTemp = RemoveHeadList(&g_PolicyList);
|
|
|
|
lpCurrentRef = CONTAINING_RECORD(pTemp, LCPOLICYREF, ListEntry);
|
|
pPolicy = lpCurrentRef->pPolicy;
|
|
LocalFree(lpCurrentRef);
|
|
}
|
|
else
|
|
{
|
|
pPolicy = NULL;
|
|
}
|
|
|
|
return(pPolicy);
|
|
}
|
|
|
|
/*
|
|
* Internal Function Implementations
|
|
*/
|
|
|
|
LPLCPOLICYREF
|
|
FindPolicyListEntry(
|
|
ULONG ulPolicyId
|
|
)
|
|
{
|
|
PLIST_ENTRY pNext;
|
|
LPLCPOLICYREF lpPolicyRef, lpCurrentRef;
|
|
|
|
lpPolicyRef = NULL;
|
|
|
|
for (pNext = g_PolicyList.Flink; pNext != &g_PolicyList; pNext = pNext->Flink)
|
|
{
|
|
lpCurrentRef = CONTAINING_RECORD(pNext, LCPOLICYREF, ListEntry);
|
|
|
|
if (lpCurrentRef->pPolicy->GetId() == ulPolicyId)
|
|
{
|
|
lpPolicyRef = lpCurrentRef;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return(lpPolicyRef);
|
|
}
|