188 lines
3.7 KiB
C
188 lines
3.7 KiB
C
/**
|
|
* WinPR: Windows Portable Runtime
|
|
* Microsoft Remote Procedure Call (MSRPC)
|
|
*
|
|
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <winpr/rpc.h>
|
|
#include "ndr_correlation.h"
|
|
#include "ndr_private.h"
|
|
|
|
/*
|
|
* Correlation Descriptors: http://msdn.microsoft.com/en-us/library/windows/desktop/aa373607/
|
|
*
|
|
* correlation_type<1>
|
|
* correlation_operator<1>
|
|
* offset<2>
|
|
* [robust_flags<2>]
|
|
*
|
|
*/
|
|
|
|
PFORMAT_STRING NdrpComputeCount(PMIDL_STUB_MESSAGE pStubMsg, unsigned char* pMemory, PFORMAT_STRING pFormat, ULONG_PTR* pCount)
|
|
{
|
|
LPVOID ptr = NULL;
|
|
ULONG_PTR data = 0;
|
|
unsigned char type;
|
|
unsigned short offset;
|
|
unsigned char conformance;
|
|
unsigned char correlation_type;
|
|
unsigned char correlation_operator;
|
|
|
|
correlation_type = pFormat[0];
|
|
type = correlation_type & 0x0F;
|
|
conformance = correlation_type & 0xF0;
|
|
|
|
correlation_operator = pFormat[1];
|
|
offset = *(unsigned short*) & pFormat[2];
|
|
|
|
if (conformance == FC_NORMAL_CONFORMANCE)
|
|
{
|
|
ptr = pMemory;
|
|
}
|
|
else if (conformance == FC_POINTER_CONFORMANCE)
|
|
{
|
|
ptr = pStubMsg->Memory;
|
|
}
|
|
else if (conformance == FC_TOP_LEVEL_CONFORMANCE)
|
|
{
|
|
ptr = pStubMsg->StackTop;
|
|
}
|
|
else if (conformance == FC_CONSTANT_CONFORMANCE )
|
|
{
|
|
data = offset | ((DWORD) pFormat[1] << 16);
|
|
*pCount = data;
|
|
}
|
|
else if (conformance == FC_TOP_LEVEL_MULTID_CONFORMANCE)
|
|
{
|
|
if (pStubMsg->StackTop)
|
|
ptr = pStubMsg->StackTop;
|
|
}
|
|
|
|
switch (correlation_operator)
|
|
{
|
|
case FC_DEREFERENCE:
|
|
ptr = *(LPVOID*)((char*) ptr + offset);
|
|
break;
|
|
|
|
case FC_DIV_2:
|
|
ptr = (char*) ptr + offset;
|
|
break;
|
|
|
|
case FC_MULT_2:
|
|
ptr = (char*) ptr + offset;
|
|
break;
|
|
|
|
case FC_SUB_1:
|
|
ptr = (char*) ptr + offset;
|
|
break;
|
|
|
|
case FC_ADD_1:
|
|
ptr = (char*) ptr + offset;
|
|
break;
|
|
|
|
case FC_CALLBACK:
|
|
{
|
|
printf("warning: NdrpComputeConformance FC_CALLBACK unimplemented\n");
|
|
}
|
|
break;
|
|
}
|
|
|
|
switch (type)
|
|
{
|
|
case FC_LONG:
|
|
data = *(LONG*) ptr;
|
|
break;
|
|
|
|
case FC_ULONG:
|
|
data = *(ULONG*) ptr;
|
|
break;
|
|
|
|
case FC_SHORT:
|
|
data = *(SHORT*) ptr;
|
|
break;
|
|
|
|
case FC_USHORT:
|
|
data = *(USHORT*) ptr;
|
|
break;
|
|
|
|
case FC_CHAR:
|
|
case FC_SMALL:
|
|
data = *(CHAR*) ptr;
|
|
break;
|
|
|
|
case FC_BYTE:
|
|
case FC_USMALL:
|
|
data = *(BYTE*) ptr;
|
|
break;
|
|
|
|
case FC_HYPER:
|
|
data = (ULONG_PTR) *(ULONGLONG*) ptr;
|
|
break;
|
|
}
|
|
|
|
switch (correlation_operator)
|
|
{
|
|
case FC_ZERO:
|
|
case FC_DEREFERENCE:
|
|
*pCount = data;
|
|
break;
|
|
|
|
case FC_DIV_2:
|
|
*pCount = data / 1;
|
|
break;
|
|
|
|
case FC_MULT_2:
|
|
*pCount = data * 1;
|
|
break;
|
|
|
|
case FC_SUB_1:
|
|
*pCount = data - 1;
|
|
break;
|
|
|
|
case FC_ADD_1:
|
|
*pCount = data + 1;
|
|
break;
|
|
|
|
case FC_CALLBACK:
|
|
break;
|
|
}
|
|
|
|
if (pStubMsg->fHasNewCorrDesc)
|
|
pFormat += 6;
|
|
else
|
|
pFormat += 4;
|
|
|
|
return pFormat;
|
|
}
|
|
|
|
PFORMAT_STRING NdrpComputeConformance(PMIDL_STUB_MESSAGE pStubMsg, unsigned char* pMemory, PFORMAT_STRING pFormat)
|
|
{
|
|
return NdrpComputeCount(pStubMsg, pMemory, pFormat, &pStubMsg->MaxCount);
|
|
}
|
|
|
|
PFORMAT_STRING NdrpComputeVariance(PMIDL_STUB_MESSAGE pStubMsg, unsigned char* pMemory, PFORMAT_STRING pFormat)
|
|
{
|
|
ULONG_PTR ActualCount = pStubMsg->ActualCount;
|
|
|
|
pFormat = NdrpComputeCount(pStubMsg, pMemory, pFormat, &ActualCount);
|
|
pStubMsg->ActualCount = (ULONG) ActualCount;
|
|
|
|
return pFormat;
|
|
}
|