|
|
//+----------------------------------------------------------------------------
//
// Copyright (C) 1992, Microsoft Corporation
//
// File: DfsMrshl.h
//
// Contents: Defines for Dfs marshalling routines
//
// Classes:
//
// Functions:
//
// History: March 29, 1994 Milans Created from Peterco's dfsrtl.h
//
//-----------------------------------------------------------------------------
#ifndef _DFSMRSHL_
#define _DFSMRSHL_
#ifdef __cplusplus
extern "C" {#endif
#include <stddef.h>
#include <guiddef.h>
//
// MARSHALLING AND UNMARSHALLING SUPPORT
//
#ifdef KERNEL_MODE
#define MarshalBufferAllocate(x) ExAllocatePoolWithTag(PagedPool, x, ' puM')
#define MarshalBufferFree(x) ExFreePool(x)
#else
#include <stdlib.h>
#define MarshalBufferAllocate(x) malloc(x)
#define MarshalBufferFree(x) free(x)
#endif // KERNEL_MODE
#define DfsAllocate MarshalBufferAllocate
#define DfsFree MarshalBufferFree
//
// Structure used when marshalling and unmarhalling
//
typedef struct _MARSHAL_BUFFER {
PUCHAR First; PUCHAR Current; PUCHAR Last;
} MARSHAL_BUFFER, *PMARSHAL_BUFFER;
typedef struct _MARSHAL_TYPE_INFO {
ULONG _type; // the type of item to be marshalled
ULONG _off; // offset of item (in the struct)
ULONG _cntsize; // size of counter for counted array
ULONG _cntoff; // else, offset count item (in the struct)
struct _MARSHAL_INFO * _subinfo;// if compound type, need info
} MARSHAL_TYPE_INFO, *PMARSHAL_TYPE_INFO;
typedef struct _MARSHAL_INFO {
ULONG _size; // size of item
ULONG _typecnt; // number of type infos
PMARSHAL_TYPE_INFO _typeInfo; // type infos
} MARSHAL_INFO, *PMARSHAL_INFO;
#define _mkMarshalInfo(s, i)\
{(ULONG)sizeof(s),(ULONG)(sizeof(i)/sizeof(MARSHAL_TYPE_INFO)),i}
#define MTYPE_BASE_TYPE (0x0000ffffL)
#define MTYPE_COMPOUND (0x00000001L)
#define MTYPE_GUID (0x00000002L)
#define MTYPE_STRING (0x00000003L)
#define MTYPE_UNICODE_STRING (0x00000004L)
#define MTYPE_ULONG (0x00000005L)
#define MTYPE_USHORT (0x00000006L)
#define MTYPE_PWSTR (0x00000007L)
#define MTYPE_UCHAR (0x00000008L)
#define MTYPE_CONFORMANT_CNT (0x00000009L)
#define MTYPE_INDIRECT (0x80000000L)
#define MTYPE_COMPLEX_TYPE (0x7fff0000L)
#define MTYPE_STATIC_ARRAY (0x00010000L)
#define MTYPE_COUNTED_ARRAY (0x00020000L)
#define _MCode_conformant(s,m,c)\
{MTYPE_CONFORMANT_CNT, sizeof(((s *) 0)->m[0]), sizeof(((s *) 0)->c), offsetof(s,c), 0L}
#define _MCode_Base(t,s,m,i)\
{t,offsetof(s,m),0L,0L,i}
#define _MCode_struct(s,m,i)\
_MCode_Base(MTYPE_COMPOUND,s,m,i)#define _MCode_guid(s,m)\
_MCode_Base(MTYPE_GUID,s,m,NULL)#define _MCode_str(s,m)\
_MCode_Base(MTYPE_STRING,s,m,NULL)#define _MCode_ustr(s,m)\
_MCode_Base(MTYPE_UNICODE_STRING,s,m,NULL)#define _MCode_pwstr(s,m)\
_MCode_Base(MTYPE_PWSTR,s,m,NULL)#define _MCode_ul(s,m)\
_MCode_Base(MTYPE_ULONG,s,m,NULL)#define _MCode_ush(s,m)\
_MCode_Base(MTYPE_USHORT,s,m,NULL)#define _MCode_uch(s,m)\
_MCode_Base(MTYPE_UCHAR,s,m,NULL)
#define _MCode_pstruct(s,m,i)\
_MCode_Base(MTYPE_COMPOUND|MTYPE_INDIRECT,s,m,i)#define _MCode_pguid(s,m)\
_MCode_Base(MTYPE_GUID|MTYPE_INDIRECT,s,m,NULL)#define _MCode_pstr(s,m)\
_MCode_Base(MTYPE_STRING|MTYPE_INDIRECT,s,m,NULL)#define _MCode_pustr(s,m)\
_MCode_Base(MTYPE_UNICODE_STRING|MTYPE_INDIRECT,s,m,NULL)#define _MCode_pul(s,m)\
_MCode_Base(MTYPE_ULONG|MTYPE_INDIRECT,s,m,NULL)#define _MCode_push(s,m)\
_MCode_Base(MTYPE_USHORT|MTYPE_INDIRECT,s,m,NULL)#define _MCode_puch(s,m)\
_MCode_Base(MTYPE_UCHAR|MTYPE_INDIRECT,s,m,NULL)
#define _MCode_aStatic(t,s,m,c,i)\
{t|MTYPE_STATIC_ARRAY,offsetof(s,m),0L,c,i}
#define _MCode_astruct(s,m,c,i)\
_MCode_aStatic(MTYPE_COMPOUND,s,m,c,i)#define _MCode_aguid(s,m,c)\
_MCode_aStatic(MTYPE_GUID,s,m,c,NULL)#define _MCode_astr(s,m,c)\
_MCode_aStatic(MTYPE_STRING,s,m,c,NULL)#define _MCode_austr(s,m,c)\
_MCode_aStatic(MTYPE_UNICODE_STRING,s,m,c,NULL)#define _MCode_aul(s,m,c)\
_MCode_aStatic(MTYPE_ULONG,s,m,c,NULL)#define _MCode_aush(s,m,c)\
_MCode_aStatic(MTYPE_USHORT,s,m,c,NULL)#define _MCode_auch(s,m,c)\
_MCode_aStatic(MTYPE_UCHAR,s,m,c,NULL)
#define _MCode_pastruct(s,m,c,i)\
_MCode_aStatic(MTYPE_COMPOUND|MTYPE_INDIRECT,s,m,c,i)#define _MCode_paguid(s,m,c)\
_MCode_aStatic(MTYPE_GUID|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_pastr(s,m,c)\
_MCode_aStatic(MTYPE_STRING|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_paustr(s,m,c)\
_MCode_aStatic(MTYPE_UNICODE_STRING|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_paul(s,m,c)\
_MCode_aStatic(MTYPE_ULONG|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_paush(s,m,c)\
_MCode_aStatic(MTYPE_USHORT|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_pauch(s,m,c)\
_MCode_aStatic(MTYPE_UCHAR|MTYPE_INDIRECT,s,m,c,NULL)
#define _MCode_aCounted(t,s,m,c,i) {\
t|MTYPE_COUNTED_ARRAY,\ offsetof(s,m),\ sizeof(((s *)0)->c),\ offsetof(s,c),\ i\ }
#define _MCode_castruct(s,m,c,i)\
_MCode_aCounted(MTYPE_COMPOUND,s,m,c,i)#define _MCode_caguid(s,m,c)\
_MCode_aCounted(MTYPE_GUID,s,m,c,NULL)#define _MCode_capwstr(s,m,c)\
_MCode_aCounted(MTYPE_PWSTR,s,m,c,NULL)#define _MCode_castr(s,m,c)\
_MCode_aCounted(MTYPE_STRING,s,m,c,NULL)#define _MCode_caustr(s,m,c)\
_MCode_aCounted(MTYPE_UNICODE_STRING,s,m,c,NULL)#define _MCode_caul(s,m,c)\
_MCode_aCounted(MTYPE_ULONG,s,m,c,NULL)#define _MCode_caush(s,m,c)\
_MCode_aCounted(MTYPE_USHORT,s,m,c,NULL)#define _MCode_cauch(s,m,c)\
_MCode_aCounted(MTYPE_UCHAR,s,m,c,NULL)
#define _MCode_pcastruct(s,m,c,i)\
_MCode_aCounted(MTYPE_COMPOUND|MTYPE_INDIRECT,s,m,c,i)#define _MCode_pcaguid(s,m,c)\
_MCode_aCounted(MTYPE_GUID|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_pcapwstr(s,m,c)\
_MCode_aCounted(MTYPE_PWSTR|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_pcastr(s,m,c)\
_MCode_aCounted(MTYPE_STRING|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_pcaustr(s,m,c)\
_MCode_aCounted(MTYPE_UNICODE_STRING|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_pcaul(s,m,c)\
_MCode_aCounted(MTYPE_ULONG|MTYPE_INDIRECT,s,m,c,NULL)#define _MCode_pcaush(s,m,c)\
_MCode_aCounted(MTYPE_USHORT|MTYPE_INDIRECT,s,m,c,NULL)
#define MarshalBufferInitialize( MarshalBuffer, BufferLength, Buffer ) {\
(MarshalBuffer)->First = (PUCHAR)(Buffer); \ (MarshalBuffer)->Current = (PUCHAR)(Buffer); \ (MarshalBuffer)->Last = &(MarshalBuffer)->Current[(BufferLength)]; \ }
//
// Defines and functions for unmarshalling base types.
// The BYTE masks are perfectly fine and they dont care if
// we are on working on LITTLE_ENDIAN or BIG_ENDIAN etc.
//
#define BYTE_0_MASK 0xFF
#define BYTE_0(Value) (UCHAR)( (Value) & BYTE_0_MASK)
#define BYTE_1(Value) (UCHAR)( ((Value) >> 8) & BYTE_0_MASK)
#define BYTE_2(Value) (UCHAR)( ((Value) >> 16) & BYTE_0_MASK)
#define BYTE_3(Value) (UCHAR)( ((Value) >> 24) & BYTE_0_MASK)
#define DfsRtlGetUshort(MarshalBuffer, pValue) ( \
((MarshalBuffer)->Current + 2 <= (MarshalBuffer)->Last) ? \ *(pValue) = (USHORT)((MarshalBuffer)->Current[0] ) | \ ((MarshalBuffer)->Current[1] << 8), \ (MarshalBuffer)->Current += 2, \ STATUS_SUCCESS \ : STATUS_DATA_ERROR \ )
#define DfsRtlPutUshort(MarshalBuffer, pValue) ( \
((MarshalBuffer)->Current + 2 <= (MarshalBuffer)->Last) ? \ (MarshalBuffer)->Current[0] = BYTE_0(*pValue), \ (MarshalBuffer)->Current[1] = BYTE_1(*pValue), \ (MarshalBuffer)->Current += 2, \ STATUS_SUCCESS \ : STATUS_BUFFER_TOO_SMALL \ )
#define DfsRtlGetUlong(MarshalBuffer, pValue) ( \
((MarshalBuffer)->Current + 4 <= (MarshalBuffer)->Last) ? \ *(pValue) = (ULONG) ((MarshalBuffer)->Current[0] ) | \ ((MarshalBuffer)->Current[1] << 8) | \ ((MarshalBuffer)->Current[2] << 16) | \ ((MarshalBuffer)->Current[3] << 24), \ (MarshalBuffer)->Current += 4, \ STATUS_SUCCESS \ : STATUS_DATA_ERROR \ )
#define DfsRtlPutUlong(MarshalBuffer, pValue) ( \
((MarshalBuffer)->Current + 4 <= (MarshalBuffer)->Last) ? \ (MarshalBuffer)->Current[0] = BYTE_0(*pValue), \ (MarshalBuffer)->Current[1] = BYTE_1(*pValue), \ (MarshalBuffer)->Current[2] = BYTE_2(*pValue), \ (MarshalBuffer)->Current[3] = BYTE_3(*pValue), \ (MarshalBuffer)->Current += 4, \ STATUS_SUCCESS \ : STATUS_BUFFER_TOO_SMALL \ )
#define DfsRtlGetGuid(MarshalBuffer, pValue) ( \
((MarshalBuffer)->Current + 16 <= (MarshalBuffer)->Last) ? \ (pValue)->Data1 = (ULONG) ((MarshalBuffer)->Current[0] ) | \ ((MarshalBuffer)->Current[1] << 8) | \ ((MarshalBuffer)->Current[2] << 16) | \ ((MarshalBuffer)->Current[3] << 24) , \ (pValue)->Data2 = (USHORT)((MarshalBuffer)->Current[4] ) | \ ((MarshalBuffer)->Current[5] << 8) , \ (pValue)->Data3 = (USHORT)((MarshalBuffer)->Current[6] ) | \ ((MarshalBuffer)->Current[7] << 8) , \ memcpy((pValue)->Data4, &(MarshalBuffer)->Current[8], 8), \ (MarshalBuffer)->Current += 16, \ STATUS_SUCCESS \ : STATUS_DATA_ERROR \ )
//
// This routine is being used for DFS_UPD_REFERRAL_BUFFER. These
// routines will continue to be used in the future as well since
// we want to keep the structure of DFS_UPD_REFERRAL_BUFFER well
// defined rather than using the Marshalling routines provided here.
//
#define _PutGuid(cp, pguid) \
cp[0] = BYTE_0((pguid)->Data1), \ cp[1] = BYTE_1((pguid)->Data1), \ cp[2] = BYTE_2((pguid)->Data1), \ cp[3] = BYTE_3((pguid)->Data1), \ cp[4] = BYTE_0((pguid)->Data2), \ cp[5] = BYTE_1((pguid)->Data2), \ cp[6] = BYTE_0((pguid)->Data3), \ cp[7] = BYTE_1((pguid)->Data3), \ memcpy(&cp[8], (pguid)->Data4, 8)
#define _PutULong(cp, ularg) \
cp[0] = BYTE_0(ularg), \ cp[1] = BYTE_1(ularg), \ cp[2] = BYTE_2(ularg), \ cp[3] = BYTE_3(ularg)
#define _GetULong(cp, ularg) \
ularg = (ULONG) (cp[0]) | \ (cp[1] << 8) | \ (cp[2] << 16) | \ (cp[3] << 24)
#define DfsRtlPutGuid(MarshalBuffer, pValue) ( \
((MarshalBuffer)->Current + 16 <= (MarshalBuffer)->Last) ? \ (MarshalBuffer)->Current[0] = BYTE_0((pValue)->Data1), \ (MarshalBuffer)->Current[1] = BYTE_1((pValue)->Data1), \ (MarshalBuffer)->Current[2] = BYTE_2((pValue)->Data1), \ (MarshalBuffer)->Current[3] = BYTE_3((pValue)->Data1), \ (MarshalBuffer)->Current[4] = BYTE_0((pValue)->Data2), \ (MarshalBuffer)->Current[5] = BYTE_1((pValue)->Data2), \ (MarshalBuffer)->Current[6] = BYTE_0((pValue)->Data3), \ (MarshalBuffer)->Current[7] = BYTE_1((pValue)->Data3), \ memcpy(&(MarshalBuffer)->Current[8], (pValue)->Data4, 8), \ (MarshalBuffer)->Current += 16, \ STATUS_SUCCESS \ : STATUS_BUFFER_TOO_SMALL \ )
#define DfsRtlSizeString(pString, pSize) ( \
((pString)->Length > 0) ? ( \ ((pString)->Buffer != NULL) ? \ (*(pSize)) += (2 + (pString)->Length), \ STATUS_SUCCESS \ : STATUS_DATA_ERROR \ ) \ : ((*(pSize)) += 2, \ STATUS_SUCCESS) \ )
#define DfsRtlSizepwString(pString, pSize) ( \
(*pString != NULL) ? \ (*(pSize)) += ((1 + wcslen(*pString))*sizeof(WCHAR)), \ STATUS_SUCCESS \ : ((*(pSize)) += 2, \ STATUS_SUCCESS) \ )
#define DfsRtlSizeUnicodeString(pUnicodeString, pSize) \
DfsRtlSizeString(pUnicodeString, pSize)
NTSTATUSDfsRtlGet( IN OUT PMARSHAL_BUFFER MarshalBuffer, IN PMARSHAL_INFO MarshalInfo, OUT PVOID Item);
NTSTATUSDfsRtlPut( IN OUT PMARSHAL_BUFFER MarshalBuffer, IN PMARSHAL_INFO MarshalInfo, OUT PVOID Item);
NTSTATUSDfsRtlSize( IN PMARSHAL_INFO MarshalInfo, IN PVOID Item, OUT PULONG Size);
VOIDDfsRtlUnwindGet( IN PMARSHAL_INFO MarshalInfo, IN PMARSHAL_TYPE_INFO LastTypeInfo, IN PVOID Item);
#ifdef __cplusplus
}#endif
#endif // _DFSMRSHL_
|
