/**

 * WinPR: Windows Portable Runtime

 * Network Data Representation (NDR)

 *

 * 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.

 */

#ifndef WINPR_RPC_NDR_H

#define WINPR_RPC_NDR_H

#include <winpr/rpc.h>

#include <winpr/wtypes.h>

#ifndef _WIN32

#define __RPC_WIN32__ 1

#define TARGET_IS_NT50_OR_LATER 1

typedef union _CLIENT_CALL_RETURN {

	void* Pointer;

	LONG_PTR Simple;

} CLIENT_CALL_RETURN;

typedef struct _RPC_VERSION

{

	unsigned short MajorVersion;

	unsigned short MinorVersion;

} RPC_VERSION;

typedef struct _RPC_SYNTAX_IDENTIFIER

{

	GUID SyntaxGUID;

	RPC_VERSION SyntaxVersion;

} RPC_SYNTAX_IDENTIFIER, PRPC_SYNTAX_IDENTIFIER;

#define RPC_MGR_EPV void

typedef struct _RPC_MESSAGE

{

	RPC_BINDING_HANDLE Handle;

	ULONG DataRepresentation;

	void* Buffer;

	unsigned int BufferLength;

	unsigned int ProcNum;

	PRPC_SYNTAX_IDENTIFIER TransferSyntax;

	void* RpcInterfaceInformation;

	void* ReservedForRuntime;

	RPC_MGR_EPV* ManagerEpv;

	void* ImportContext;

	ULONG RpcFlags;

} RPC_MESSAGE, *PRPC_MESSAGE;

typedef void (*RPC_DISPATCH_FUNCTION)(PRPC_MESSAGE Message);

typedef struct

{

	unsigned int DispatchTableCount;

	RPC_DISPATCH_FUNCTION* DispatchTable;

	LONG_PTR Reserved;

} RPC_DISPATCH_TABLE, *PRPC_DISPATCH_TABLE;

typedef struct _RPC_PROTSEQ_ENDPOINT

{

	unsigned char* RpcProtocolSequence;

	unsigned char* Endpoint;

} RPC_PROTSEQ_ENDPOINT, *PRPC_PROTSEQ_ENDPOINT;

typedef struct _RPC_SERVER_INTERFACE

{

	unsigned int Length;

	RPC_SYNTAX_IDENTIFIER InterfaceId;

	RPC_SYNTAX_IDENTIFIER TransferSyntax;

	PRPC_DISPATCH_TABLE DispatchTable;

	unsigned int RpcProtseqEndpointCount;

	PRPC_PROTSEQ_ENDPOINT RpcProtseqEndpoint;

	RPC_MGR_EPV* DefaultManagerEpv;

	void const* InterpreterInfo;

	unsigned int Flags;

} RPC_SERVER_INTERFACE, *PRPC_SERVER_INTERFACE;

typedef struct _RPC_CLIENT_INTERFACE

{

	unsigned int Length;

	RPC_SYNTAX_IDENTIFIER InterfaceId;

	RPC_SYNTAX_IDENTIFIER TransferSyntax;

	PRPC_DISPATCH_TABLE DispatchTable;

	unsigned int RpcProtseqEndpointCount;

	PRPC_PROTSEQ_ENDPOINT RpcProtseqEndpoint;

	ULONG_PTR Reserved;

	void const* InterpreterInfo;

	unsigned int Flags;

} RPC_CLIENT_INTERFACE, *PRPC_CLIENT_INTERFACE;

typedef void* (*GENERIC_BINDING_ROUTINE)(void*);

typedef void (*GENERIC_UNBIND_ROUTINE)(void*, unsigned char*);

typedef struct _GENERIC_BINDING_ROUTINE_PAIR

{

	GENERIC_BINDING_ROUTINE pfnBind;

	GENERIC_UNBIND_ROUTINE pfnUnbind;

} GENERIC_BINDING_ROUTINE_PAIR, *PGENERIC_BINDING_ROUTINE_PAIR;

typedef struct __GENERIC_BINDING_INFO

{

	void* pObj;

	unsigned int Size;

	GENERIC_BINDING_ROUTINE pfnBind;

	GENERIC_UNBIND_ROUTINE pfnUnbind;

} GENERIC_BINDING_INFO, *PGENERIC_BINDING_INFO;

typedef void (*NDR_RUNDOWN)(void* context);

typedef void (*NDR_NOTIFY_ROUTINE)(void);

typedef const unsigned char* PFORMAT_STRING;

typedef struct _MIDL_STUB_DESC MIDL_STUB_DESC;

typedef MIDL_STUB_DESC* PMIDL_STUB_DESC;

typedef struct _MIDL_STUB_MESSAGE

{

	PRPC_MESSAGE RpcMsg;

	unsigned char* Buffer;

	unsigned char* BufferStart;

	unsigned char* BufferEnd;

	unsigned char* BufferMark;

	ULONG BufferLength;

	ULONG MemorySize;

	unsigned char* Memory;

	int IsClient;

	int ReuseBuffer;

	struct NDR_ALLOC_ALL_NODES_CONTEXT* pAllocAllNodesContext;

	struct NDR_POINTER_QUEUE_STATE* pPointerQueueState;

	int IgnoreEmbeddedPointers;

	unsigned char* PointerBufferMark;

	unsigned char fBufferValid;

	unsigned char uFlags;

	unsigned short Unused2;

	ULONG_PTR MaxCount;

	ULONG Offset;

	ULONG ActualCount;

	void* (*pfnAllocate)(size_t);

	void (*pfnFree)(void*);

	unsigned char* StackTop;

	unsigned char* pPresentedType;

	unsigned char* pTransmitType;

	handle_t SavedHandle;

	const struct _MIDL_STUB_DESC* StubDesc;

	struct _FULL_PTR_XLAT_TABLES* FullPtrXlatTables;

	ULONG FullPtrRefId;

	ULONG PointerLength;

	int fInDontFree : 1;

	int fDontCallFreeInst : 1;

	int fInOnlyParam : 1;

	int fHasReturn : 1;

	int fHasExtensions : 1;

	int fHasNewCorrDesc : 1;

	int fUnused : 10;

	int fUnused2 : 16;

	ULONG dwDestContext;

	void* pvDestContext;

	// NDR_SCONTEXT* SavedContextHandles;

	long ParamNumber;

	struct IRpcChannelBuffer* pRpcChannelBuffer;

	// PARRAY_INFO pArrayInfo;

	ULONG* SizePtrCountArray;

	ULONG* SizePtrOffsetArray;

	ULONG* SizePtrLengthArray;

	void* pArgQueue;

	ULONG dwStubPhase;

	void* LowStackMark;

	// PNDR_ASYNC_MESSAGE pAsyncMsg;

	// PNDR_CORRELATION_INFO pCorrInfo;

	unsigned char* pCorrMemory;

	void* pMemoryList;

	// CS_STUB_INFO* pCSInfo;

	unsigned char* ConformanceMark;

	unsigned char* VarianceMark;

	void* BackingStoreLowMark;

	INT_PTR Unused;

	struct _NDR_PROC_CONTEXT* pContext;

	INT_PTR Reserved51_1;

	INT_PTR Reserved51_2;

	INT_PTR Reserved51_3;

	INT_PTR Reserved51_4;

	INT_PTR Reserved51_5;

} MIDL_STUB_MESSAGE, *PMIDL_STUB_MESSAGE;

typedef void (*EXPR_EVAL)(struct _MIDL_STUB_MESSAGE*);

typedef void (*XMIT_HELPER_ROUTINE)(PMIDL_STUB_MESSAGE);

typedef struct _XMIT_ROUTINE_QUINTUPLE

{

	XMIT_HELPER_ROUTINE pfnTranslateToXmit;

	XMIT_HELPER_ROUTINE pfnTranslateFromXmit;

	XMIT_HELPER_ROUTINE pfnFreeXmit;

	XMIT_HELPER_ROUTINE pfnFreeInst;

} XMIT_ROUTINE_QUINTUPLE, *PXMIT_ROUTINE_QUINTUPLE;

typedef ULONG (*USER_MARSHAL_SIZING_ROUTINE)(ULONG*, ULONG, void*);

typedef unsigned char* (*USER_MARSHAL_MARSHALLING_ROUTINE)(ULONG*, unsigned char*, void*);

typedef unsigned char* (*USER_MARSHAL_UNMARSHALLING_ROUTINE)(ULONG*, unsigned char*, void*);

typedef void (*USER_MARSHAL_FREEING_ROUTINE)(ULONG*, void*);

typedef struct _USER_MARSHAL_ROUTINE_QUADRUPLE

{

	USER_MARSHAL_SIZING_ROUTINE pfnBufferSize;

	USER_MARSHAL_MARSHALLING_ROUTINE pfnMarshall;

	USER_MARSHAL_UNMARSHALLING_ROUTINE pfnUnmarshall;

	USER_MARSHAL_FREEING_ROUTINE pfnFree;

} USER_MARSHAL_ROUTINE_QUADRUPLE;

typedef struct _MALLOC_FREE_STRUCT

{

	void* (*pfnAllocate)(size_t);

	void (*pfnFree)(void*);

} MALLOC_FREE_STRUCT;

typedef struct _COMM_FAULT_OFFSETS

{

	short CommOffset;

	short FaultOffset;

} COMM_FAULT_OFFSETS;

typedef void* NDR_CS_ROUTINES;

typedef void* NDR_EXPR_DESC;

struct _MIDL_STUB_DESC

{

	void* RpcInterfaceInformation;

	void* (*pfnAllocate)(size_t);

	void (*pfnFree)(void*);

	union {

		handle_t* pAutoHandle;

		handle_t* pPrimitiveHandle;

		PGENERIC_BINDING_INFO pGenericBindingInfo;

	} IMPLICIT_HANDLE_INFO;

	const NDR_RUNDOWN* apfnNdrRundownRoutines;

	const GENERIC_BINDING_ROUTINE_PAIR* aGenericBindingRoutinePairs;

	const EXPR_EVAL* apfnExprEval;

	const XMIT_ROUTINE_QUINTUPLE* aXmitQuintuple;

	const unsigned char* pFormatTypes;

	int fCheckBounds;

	ULONG Version;

	MALLOC_FREE_STRUCT* pMallocFreeStruct;

	long MIDLVersion;

	const COMM_FAULT_OFFSETS* CommFaultOffsets;

	const USER_MARSHAL_ROUTINE_QUADRUPLE* aUserMarshalQuadruple;

	const NDR_NOTIFY_ROUTINE* NotifyRoutineTable;

	ULONG_PTR mFlags;

	const NDR_CS_ROUTINES* CsRoutineTables;

	void* ProxyServerInfo;

	const NDR_EXPR_DESC* pExprInfo;

};

typedef struct

{

	unsigned char FullPtrUsed : 1;

	unsigned char RpcSsAllocUsed : 1;

	unsigned char ObjectProc : 1;

	unsigned char HasRpcFlags : 1;

	unsigned char IgnoreObjectException : 1;

	unsigned char HasCommOrFault : 1;

	unsigned char UseNewInitRoutines : 1;

	unsigned char Unused : 1;

} INTERPRETER_FLAGS, *PINTERPRETER_FLAGS;

typedef struct

{

	unsigned short MustSize : 1;

	unsigned short MustFree : 1;

	unsigned short IsPipe : 1;

	unsigned short IsIn : 1;

	unsigned short IsOut : 1;

	unsigned short IsReturn : 1;

	unsigned short IsBasetype : 1;

	unsigned short IsByValue : 1;

	unsigned short IsSimpleRef : 1;

	unsigned short IsDontCallFreeInst : 1;

	unsigned short SaveForAsyncFinish : 1;

	unsigned short Unused : 2;

	unsigned short ServerAllocSize : 3;

} PARAM_ATTRIBUTES, *PPARAM_ATTRIBUTES;

typedef struct

{

	unsigned char ServerMustSize : 1;

	unsigned char ClientMustSize : 1;

	unsigned char HasReturn : 1;

	unsigned char HasPipes : 1;

	unsigned char Unused : 1;

	unsigned char HasAsyncUuid : 1;

	unsigned char HasExtensions : 1;

	unsigned char HasAsyncHandle : 1;

} INTERPRETER_OPT_FLAGS, *PINTERPRETER_OPT_FLAGS;

typedef struct

{

	unsigned char HasNewCorrDesc : 1;

	unsigned char ClientCorrCheck : 1;

	unsigned char ServerCorrCheck : 1;

	unsigned char HasNotify : 1;

	unsigned char HasNotify2 : 1;

	unsigned char Unused : 3;

} INTERPRETER_OPT_FLAGS2, *PINTERPRETER_OPT_FLAGS2;

typedef struct _NDR_CORRELATION_FLAGS

{

	unsigned char Early : 1;

	unsigned char Split : 1;

	unsigned char IsIidIs : 1;

	unsigned char DontCheck : 1;

	unsigned char Unused : 4;

} NDR_CORRELATION_FLAGS;

#define FC_ALLOCATE_ALL_NODES 0x01

#define FC_DONT_FREE 0x02

#define FC_ALLOCED_ON_STACK 0x03

#define FC_SIMPLE_POINTER 0x04

#define FC_POINTER_DEREF 0x05

#define HANDLE_PARAM_IS_VIA_PTR 0x80

#define HANDLE_PARAM_IS_IN 0x40

#define HANDLE_PARAM_IS_OUT 0x20

#define HANDLE_PARAM_IS_RETURN 0x21

#define NDR_STRICT_CONTEXT_HANDLE 0x08

#define NDR_CONTEXT_HANDLE_NO_SERIALIZE 0x04

#define NDR_CONTEXT_HANDLE_SERIALIZE 0x02

#define NDR_CONTEXT_HANDLE_CANNOT_BE_NULL 0x01

typedef struct

{

	PARAM_ATTRIBUTES Attributes;

	unsigned short StackOffset;

	union {

		unsigned char FormatChar;

		unsigned short Offset;

	} Type;

} NDR_PARAM;

typedef struct

{

	unsigned char Size;

	INTERPRETER_OPT_FLAGS2 Flags2;

	unsigned short ClientCorrHint;

	unsigned short ServerCorrHint;

	unsigned short NotifyIndex;

} NDR_PROC_HEADER_EXTS;

typedef struct _NDR_PROC_HEADER

{

	unsigned char HandleType;

	INTERPRETER_FLAGS OldOiFlags;

	unsigned short RpcFlagsLow;

	unsigned short RpcFlagsHi;

	unsigned short ProcNum;

	unsigned short StackSize;

} NDR_PROC_HEADER, *PNDR_PROC_HEADER;

typedef struct _NDR_OI2_PROC_HEADER

{

	unsigned short ClientBufferSize;

	unsigned short ServerBufferSize;

	INTERPRETER_OPT_FLAGS Oi2Flags;

	unsigned char NumberParams;

} NDR_OI2_PROC_HEADER, *PNDR_OI2_PROC_HEADER;

typedef enum _NDR_PHASE

{

	NDR_PHASE_SIZE,

	NDR_PHASE_MARSHALL,

	NDR_PHASE_UNMARSHALL,

	NDR_PHASE_FREE

} NDR_PHASE;

#define FC_NORMAL_CONFORMANCE 0x00

#define FC_POINTER_CONFORMANCE 0x10

#define FC_TOP_LEVEL_CONFORMANCE 0x20

#define FC_CONSTANT_CONFORMANCE 0x40

#define FC_TOP_LEVEL_MULTID_CONFORMANCE 0x80

/* Type Format Strings: http://msdn.microsoft.com/en-us/library/windows/desktop/aa379093/ */

#define FC_ZERO 0x00

#define FC_BYTE 0x01

#define FC_CHAR 0x02

#define FC_SMALL 0x03

#define FC_USMALL 0x04

#define FC_WCHAR 0x05

#define FC_SHORT 0x06

#define FC_USHORT 0x07

#define FC_LONG 0x08

#define FC_ULONG 0x09

#define FC_FLOAT 0x0A

#define FC_HYPER 0x0B

#define FC_DOUBLE 0x0C

#define FC_ENUM16 0x0D

#define FC_ENUM32 0x0E

#define FC_IGNORE 0x0F

#define FC_ERROR_STATUS_T 0x10

#define FC_RP 0x11

#define FC_UP 0x12

#define FC_OP 0x13

#define FC_FP 0x14

#define FC_STRUCT 0x15

#define FC_PSTRUCT 0x16

#define FC_CSTRUCT 0x17

#define FC_CPSTRUCT 0x18

#define FC_CVSTRUCT 0x19

#define FC_BOGUS_STRUCT 0x1A

#define FC_CARRAY 0x1B

#define FC_CVARRAY 0x1C

#define FC_SMFARRAY 0x1D

#define FC_LGFARRAY 0x1E

#define FC_SMVARRAY 0x1F

#define FC_LGVARRAY 0x20

#define FC_BOGUS_ARRAY 0x21

#define FC_C_CSTRING 0x22

#define FC_C_BSTRING 0x23

#define FC_C_SSTRING 0x24

#define FC_C_WSTRING 0x25

#define FC_CSTRING 0x26

#define FC_BSTRING 0x27

#define FC_SSTRING 0x28

#define FC_WSTRING 0x29

#define FC_ENCAPSULATED_UNION 0x2A

#define FC_NON_ENCAPSULATED_UNION 0x2B

#define FC_BYTE_COUNT_POINTER 0x2C

#define FC_TRANSMIT_AS 0x2D

#define FC_REPRESENT_AS 0x2E

#define FC_IP 0x2F

#define FC_BIND_CONTEXT 0x30

#define FC_BIND_GENERIC 0x31

#define FC_BIND_PRIMITIVE 0x32

#define FC_AUTO_HANDLE 0x33

#define FC_CALLBACK_HANDLE 0x34

#define FC_UNUSED1 0x35

#define FC_POINTER 0x36

#define FC_ALIGNM2 0x37

#define FC_ALIGNM4 0x38

#define FC_ALIGNM8 0x39

#define FC_UNUSED2 0x3A

#define FC_UNUSED3 0x3B

#define FC_UNUSED4 0x3C

#define FC_STRUCTPAD1 0x3D

#define FC_STRUCTPAD2 0x3E

#define FC_STRUCTPAD3 0x3F

#define FC_STRUCTPAD4 0x40

#define FC_STRUCTPAD5 0x41

#define FC_STRUCTPAD6 0x42

#define FC_STRUCTPAD7 0x43

#define FC_STRING_SIZED 0x44

#define FC_UNUSED5 0x45

#define FC_NO_REPEAT 0x46

#define FC_FIXED_REPEAT 0x47

#define FC_VARIABLE_REPEAT 0x48

#define FC_FIXED_OFFSET 0x49

#define FC_VARIABLE_OFFSET 0x4A

#define FC_PP 0x4B

#define FC_EMBEDDED_COMPLEX 0x4C

#define FC_IN_PARAM 0x4D

#define FC_IN_PARAM_BASETYPE 0x4E

#define FC_IN_PARAM_NO_FREE_INST 0x4F

#define FC_IN_OUT_PARAM 0x50

#define FC_OUT_PARAM 0x51

#define FC_RETURN_PARAM 0x52

#define FC_RETURN_PARAM_BASETYPE 0x53

#define FC_DEREFERENCE 0x54

#define FC_DIV_2 0x55

#define FC_MULT_2 0x56

#define FC_ADD_1 0x57

#define FC_SUB_1 0x58

#define FC_CALLBACK 0x59

#define FC_CONSTANT_IID 0x5A

#define FC_END 0x5B

#define FC_PAD 0x5C

#define FC_SPLIT_DEREFERENCE 0x74

#define FC_SPLIT_DIV_2 0x75

#define FC_SPLIT_MULT_2 0x76

#define FC_SPLIT_ADD_1 0x77

#define FC_SPLIT_SUB_1 0x78

#define FC_SPLIT_CALLBACK 0x79

#define FC_HARD_STRUCT 0xB1

#define FC_TRANSMIT_AS_PTR 0xB2

#define FC_REPRESENT_AS_PTR 0xB3

#define FC_USER_MARSHAL 0xB4

#define FC_PIPE 0xB5

#define FC_BLKHOLE 0xB6

#define FC_RANGE 0xB7

#define FC_INT3264 0xB8

#define FC_UINT3264 0xB9

#define FC_END_OF_UNIVERSE 0xBA

#define NdrFcShort(s) (byte)(s & 0xFF), (byte)(s >> 8)

#define NdrFcLong(s) \

	(byte)(s & 0xFF), (byte)((s & 0x0000FF00) >> 8), (byte)((s & 0x00FF0000) >> 16), (byte)(s >> 24)

typedef void (*NDR_TYPE_SIZE_ROUTINE)(PMIDL_STUB_MESSAGE pStubMsg, unsigned char* pMemory,

                                      PFORMAT_STRING pFormat);

typedef void (*NDR_TYPE_MARSHALL_ROUTINE)(PMIDL_STUB_MESSAGE pStubMsg, unsigned char* pMemory,

                                          unsigned char FormatChar);

typedef void (*NDR_TYPE_UNMARSHALL_ROUTINE)(PMIDL_STUB_MESSAGE pStubMsg, unsigned char* pMemory,

                                            unsigned char FormatChar);

typedef void (*NDR_TYPE_FREE_ROUTINE)(PMIDL_STUB_MESSAGE pStubMsg, unsigned char* pMemory,

                                      PFORMAT_STRING pFormat);

#ifdef __cplusplus

extern "C"

{

#endif

	WINPR_API CLIENT_CALL_RETURN NdrClientCall2(PMIDL_STUB_DESC pStubDescriptor,

	                                            PFORMAT_STRING pFormat, ...);

#ifdef __cplusplus

}

#endif

#endif

#endif /* WINPR_RPC_NDR_H */