240 lines
6.6 KiB
C++
240 lines
6.6 KiB
C++
#include "stdafx.h"
|
|
#include "drsr_alloc.h"
|
|
|
|
template<>
|
|
midl_ptr<DRS_EXTENSIONS_INT> make_midl_ptr()
|
|
{
|
|
auto extensions = midl_ptr<DRS_EXTENSIONS_INT>((DRS_EXTENSIONS_INT*)midl_user_allocate(sizeof(DRS_EXTENSIONS_INT)));
|
|
// size excluding the actual count
|
|
extensions->cb = sizeof(DRS_EXTENSIONS_INT) - sizeof(DWORD);
|
|
return extensions;
|
|
}
|
|
|
|
template<>
|
|
midl_ptr<DSNAME> make_midl_ptr(ULONG numChars)
|
|
{
|
|
ULONG totalSize = sizeof(DSNAME) + numChars * sizeof(WCHAR);
|
|
auto dsName = midl_ptr<DSNAME>((DSNAME*)midl_user_allocate(totalSize));
|
|
dsName->NameLen = numChars;
|
|
dsName->structLen = totalSize;
|
|
return dsName;
|
|
}
|
|
|
|
template<>
|
|
midl_ptr<UPTODATE_VECTOR_V1_EXT> make_midl_ptr(ULONG numCursors)
|
|
{
|
|
size_t totalSize = sizeof(UPTODATE_VECTOR_V1_EXT) + numCursors * sizeof(UPTODATE_CURSOR_V1);
|
|
auto vector = midl_ptr<UPTODATE_VECTOR_V1_EXT>((UPTODATE_VECTOR_V1_EXT*)midl_user_allocate(totalSize));
|
|
vector->cNumCursors = numCursors;
|
|
vector->dwVersion = 1;
|
|
return vector;
|
|
}
|
|
|
|
template<>
|
|
midl_ptr<PARTIAL_ATTR_VECTOR_V1_EXT> make_midl_ptr(ULONG numAttributes)
|
|
{
|
|
auto totalSize = sizeof(PARTIAL_ATTR_VECTOR_V1_EXT) + numAttributes * sizeof(ATTRTYP);
|
|
auto pas = midl_ptr<PARTIAL_ATTR_VECTOR_V1_EXT>((PARTIAL_ATTR_VECTOR_V1_EXT*)midl_user_allocate(totalSize));
|
|
pas->cAttrs = numAttributes;
|
|
pas->dwVersion = 1;
|
|
return pas;
|
|
}
|
|
|
|
template<>
|
|
midl_ptr<DRS_MSG_CRACKREQ_V1> make_midl_ptr(ULONG numNames)
|
|
{
|
|
auto request = midl_ptr<DRS_MSG_CRACKREQ_V1>((DRS_MSG_CRACKREQ_V1*)midl_user_allocate(sizeof(DRS_MSG_CRACKREQ_V1)));
|
|
request->cNames = numNames;
|
|
request->rpNames = (WCHAR**)midl_user_allocate(numNames * sizeof(WCHAR*));
|
|
return request;
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_GETCHGREQ_V5>::operator()(DRS_MSG_GETCHGREQ_V5* request) const
|
|
{
|
|
if (request != nullptr)
|
|
{
|
|
// Perform deep free
|
|
midl_user_free(request->pNC);
|
|
midl_user_free(request->pUpToDateVecDestV1);
|
|
|
|
// Now free the encapsulating object:
|
|
midl_user_free(request);
|
|
}
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_GETCHGREQ_V8>::operator()(DRS_MSG_GETCHGREQ_V8* request) const
|
|
{
|
|
if (request != nullptr)
|
|
{
|
|
// Perform deep free
|
|
midl_user_free(request->pPartialAttrSet);
|
|
midl_user_free(request->pPartialAttrSetEx);
|
|
// TODO: Free all PrefixTableDest items separately?
|
|
midl_user_free(request->PrefixTableDest.pPrefixEntry);
|
|
|
|
// The DRS_MSG_GETCHGREQ_V8 message is a superset of DRS_MSG_GETCHGREQ_V5.
|
|
auto requestV5 = midl_ptr<DRS_MSG_GETCHGREQ_V5>((DRS_MSG_GETCHGREQ_V5*)request);
|
|
}
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_GETCHGREQ_V10>::operator()(DRS_MSG_GETCHGREQ_V10* request) const
|
|
{
|
|
if (request != nullptr)
|
|
{
|
|
// The DRS_MSG_GETCHGREQ_V10 message is a superset of DRS_MSG_GETCHGREQ_V8.
|
|
auto requestV8 = midl_ptr<DRS_MSG_GETCHGREQ_V8>((DRS_MSG_GETCHGREQ_V8*)request);
|
|
}
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_CRACKREQ_V1>::operator()(DRS_MSG_CRACKREQ_V1* request) const
|
|
{
|
|
if (request == nullptr)
|
|
{
|
|
return;
|
|
}
|
|
midl_user_free(request->rpNames);
|
|
midl_user_free(request);
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_CRACKREPLY_V1>::operator()(DRS_MSG_CRACKREPLY_V1* reply) const
|
|
{
|
|
if (reply == nullptr)
|
|
{
|
|
return;
|
|
}
|
|
if (reply->pResult != nullptr)
|
|
{
|
|
// Perform deep free
|
|
auto numItems = reply->pResult->cItems;
|
|
for (DWORD i = 0; i < numItems; i++)
|
|
{
|
|
auto currentItem = reply->pResult->rItems[i];
|
|
midl_user_free(currentItem.pDomain);
|
|
midl_user_free(currentItem.pName);
|
|
}
|
|
midl_user_free(reply->pResult->rItems);
|
|
midl_user_free(reply->pResult);
|
|
}
|
|
// Free the wrapping object itself
|
|
midl_user_free(reply);
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_GETCHGREPLY_V1>::operator()(DRS_MSG_GETCHGREPLY_V1* reply) const
|
|
{
|
|
if (reply == nullptr)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Perform deep free
|
|
midl_user_free(reply->pNC);
|
|
midl_user_free(reply->pUpToDateVecSrcV1);
|
|
|
|
// Free the prefix table:
|
|
int numPrefixes = reply->PrefixTableSrc.PrefixCount;
|
|
for (int i = 0; i < numPrefixes; i++)
|
|
{
|
|
midl_user_free(reply->PrefixTableSrc.pPrefixEntry[i].prefix.elements);
|
|
}
|
|
midl_user_free(reply->PrefixTableSrc.pPrefixEntry);
|
|
|
|
// Free the replicated objects:
|
|
auto currentObject = reply->pObjects;
|
|
while (currentObject != nullptr)
|
|
{
|
|
midl_user_free(currentObject->Entinf.pName);
|
|
midl_user_free(currentObject->pMetaDataExt);
|
|
midl_user_free(currentObject->pParentGuid);
|
|
|
|
// Free all object attributes:
|
|
ULONG numAttributes = currentObject->Entinf.AttrBlock.attrCount;
|
|
for (ULONG i = 0; i < numAttributes; i++)
|
|
{
|
|
auto currentAttribute = currentObject->Entinf.AttrBlock.pAttr[i];
|
|
// Free all attribute values:
|
|
ULONG numValues = currentAttribute.AttrVal.valCount;
|
|
for (ULONG j = 0; j < numValues; j++)
|
|
{
|
|
auto currentValue = currentAttribute.AttrVal.pAVal[j];
|
|
midl_user_free(currentValue.pVal);
|
|
}
|
|
// Free the value list:
|
|
midl_user_free(currentAttribute.AttrVal.pAVal);
|
|
}
|
|
// Free the attribute list:
|
|
midl_user_free(currentObject->Entinf.AttrBlock.pAttr);
|
|
|
|
// Free the current object and go to next:
|
|
auto nextObject = currentObject->pNextEntInf;
|
|
midl_user_free(currentObject);
|
|
currentObject = nextObject;
|
|
}
|
|
|
|
// Finally, free the encapsulating object:
|
|
midl_user_free(reply);
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_GETCHGREPLY_V6>::operator()(DRS_MSG_GETCHGREPLY_V6* reply) const
|
|
{
|
|
if (reply != nullptr)
|
|
{
|
|
// Free the linked values (REPLVALINF_V1):
|
|
for (DWORD i = 0; i < reply->cNumValues; i++)
|
|
{
|
|
auto currentValue = reply->rgValues[i];
|
|
midl_user_free(currentValue.pObject);
|
|
midl_user_free(currentValue.Aval.pVal);
|
|
}
|
|
|
|
// Free the encapsulating array. It does not matter whether it is of type REPLVALINF_V1 or REPLVALINF_V3.
|
|
midl_user_free(reply->rgValues);
|
|
|
|
// The DRS_MSG_GETCHGREPLY_V6 message is a superset of DRS_MSG_GETCHGREPLY_V1.
|
|
auto replyV1 = midl_ptr<DRS_MSG_GETCHGREPLY_V1>((DRS_MSG_GETCHGREPLY_V1*)reply);
|
|
}
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_GETCHGREPLY_V9>::operator()(DRS_MSG_GETCHGREPLY_V9* reply) const
|
|
{
|
|
if (reply != nullptr)
|
|
{
|
|
// Free the linked values (REPLVALINF_V3):
|
|
for (DWORD i = 0; i < reply->cNumValues; i++)
|
|
{
|
|
auto currentValue = reply->rgValues[i];
|
|
midl_user_free(currentValue.pObject);
|
|
midl_user_free(currentValue.Aval.pVal);
|
|
}
|
|
|
|
/* The DRS_MSG_GETCHGREPLY_V6 deleter should not go through these values,
|
|
because it would interpret them as REPLVALINF_V1 instead of REPLVALINF_V3.
|
|
*/
|
|
reply->cNumValues = 0;
|
|
|
|
// The DRS_MSG_GETCHGREPLY_V9 message is a superset of DRS_MSG_GETCHGREPLY_V6.
|
|
auto replyV6 = midl_ptr<DRS_MSG_GETCHGREPLY_V6>((DRS_MSG_GETCHGREPLY_V6*)reply);
|
|
}
|
|
}
|
|
|
|
template<>
|
|
void midl_delete<DRS_MSG_GETREPLINFO_REQ_V1>::operator()(DRS_MSG_GETREPLINFO_REQ_V1* request) const
|
|
{
|
|
if (request == nullptr)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Free the DN string
|
|
midl_user_free(request->pszObjectDN);
|
|
|
|
// Free the encapsulating object:
|
|
midl_user_free(request);
|
|
} |