#include <stdlib.h>
#include <stdio.h>
#include <ctype.h>
#include "ctx.h"

#define TYPE_FORMAT_STRING_SIZE   23
#define PROC_FORMAT_STRING_SIZE   21
#define TRANSMIT_AS_TABLE_SIZE    0
#define WIRE_MARSHAL_TABLE_SIZE   0

typedef struct _MIDL_TYPE_FORMAT_STRING
    {
    short          Pad;
    unsigned char  Format[ TYPE_FORMAT_STRING_SIZE ];
    } MIDL_TYPE_FORMAT_STRING;

typedef struct _MIDL_PROC_FORMAT_STRING
    {
    short          Pad;
    unsigned char  Format[ PROC_FORMAT_STRING_SIZE ];
} MIDL_PROC_FORMAT_STRING;

extern const MIDL_STUB_DESC hello_StubDesc;
extern const MIDL_PROC_FORMAT_STRING __MIDL_ProcFormatString;
//extern const MIDL_TYPE_FORMAT_STRING __MIDL_TypeFormatString;*/

/*****************************************************************
 * Modified from midl-generated stubs                            *
 *****************************************************************/


void m_CtxOpen(
    /* [out] */ PCTXTYPE __RPC_FAR *pphContext,
    /* [in] */ long Value)
{
    RPC_BINDING_HANDLE _Handle	=	0;

    RPC_MESSAGE _RpcMessage;

    MIDL_STUB_MESSAGE _StubMsg;

	char *ctx, *buf;
	int i;

	printf("\n*******************************************************************\n");
	printf("**** CtxOpen()                                                  ***\n");
	printf("*******************************************************************\n\n");

    if(!pphContext)
        {
        RpcRaiseException(RPC_X_NULL_REF_POINTER);
        }
    RpcTryFinally
        {
        NdrClientInitializeNew(
                          ( PRPC_MESSAGE  )&_RpcMessage,
                          ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                          ( PMIDL_STUB_DESC  )&hello_StubDesc,
                          0);


        _Handle = hBinding;


        _StubMsg.BufferLength = 4U;
        NdrGetBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg, _StubMsg.BufferLength, _Handle );

        *(( long __RPC_FAR * )_StubMsg.Buffer)++ = Value;

        NdrSendReceive( (PMIDL_STUB_MESSAGE) &_StubMsg, (unsigned char __RPC_FAR *) _StubMsg.Buffer );

        if ( (_RpcMessage.DataRepresentation & 0X0000FFFFUL) != NDR_LOCAL_DATA_REPRESENTATION )
            NdrConvert( (PMIDL_STUB_MESSAGE) &_StubMsg, (PFORMAT_STRING) &__MIDL_ProcFormatString.Format[0] );


        printf("Before NdrClientContextUnmarshall: Buflen=%d\nBuffer: ",  _StubMsg.RpcMsg->BufferLength);
        for(buf = _StubMsg.Buffer, i = 0; i < _StubMsg.RpcMsg->BufferLength; i++)
        	printf("0x%x, ", buf[i] & 0x0FF);
        printf("\n\n");

        *pphContext = (void *)0;
        NdrClientContextUnmarshall(
                              ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                              ( NDR_CCONTEXT __RPC_FAR * )pphContext,
                              _Handle);

        ctx = (char*)*pphContext;
        printf("\nNdrClientContextUnmarshall returned: handle=0x%p\n", ctx);
        printf("00: 0x%x <- obviously pointer to binding handle copyed from _Handle\n", *((int*)ctx));
        ctx+=4;
        printf("04: 0x%x <- unknown field\n", *((int*)ctx));
        printf("08: ");

        for(ctx+=4, i = 0; i < 20; i++)
        	printf("0x%x,", *(ctx+i) & 0x0FF); printf(" <- ndr 20 bytes\n\n");

        printf("Buflen=%d, Buffer: ", _StubMsg.BufferLength);
        for(buf = _StubMsg.BufferStart; buf < _StubMsg.BufferEnd; buf++)
        	printf("0x%x,", *buf & 0x0FF);
        printf("\n");


        }
    RpcFinally
        {
        NdrFreeBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg );

        }
    RpcEndFinally

}

void m_CtxOpen2(
    /* [out] */ PCTXTYPE __RPC_FAR *pphContext,
    /* [in] */ long Value)
{

    RPC_BINDING_HANDLE _Handle	=	0;

    RPC_MESSAGE _RpcMessage;

    MIDL_STUB_MESSAGE _StubMsg;

    char buf[255];

        NdrClientInitializeNew(
                          ( PRPC_MESSAGE  )&_RpcMessage,
                          ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                          ( PMIDL_STUB_DESC  )&hello_StubDesc,
                          0);


        _Handle = hBinding;


        _StubMsg.BufferLength = 4U;
        NdrGetBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg, _StubMsg.BufferLength, _Handle );

        *(( long __RPC_FAR * )_StubMsg.Buffer)++ = Value;

        NdrSendReceive( (PMIDL_STUB_MESSAGE) &_StubMsg, (unsigned char __RPC_FAR *) _StubMsg.Buffer );

        if ( (_RpcMessage.DataRepresentation & 0X0000FFFFUL) != NDR_LOCAL_DATA_REPRESENTATION )
            NdrConvert( (PMIDL_STUB_MESSAGE) &_StubMsg, (PFORMAT_STRING) &__MIDL_ProcFormatString.Format[0] );

        *pphContext = (void *)0;

        RpcTryExcept
        {
        NdrClientContextUnmarshall(
                              ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                              NULL,
                              _Handle);
        }
        RpcExcept(1)
        {
        	printf("NdrClientContextUnmarshall reported exception = %d\n", RpcExceptionCode());
        }
        RpcEndExcept



        RpcTryExcept
        {
        NdrClientContextUnmarshall(
                              ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                              (NDR_CCONTEXT __RPC_FAR * )pphContext,
                              NULL);
        }
        RpcExcept(1)
        {
        	printf("NdrClientContextUnmarshall reported exception = %d\n", RpcExceptionCode());
        }
        RpcEndExcept


        RpcTryExcept
        {
        NdrClientContextMarshall(
                            ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                            NULL,
                            1);
        }
        RpcExcept(1)
        {
        	printf("NdrClientContextMarshall reported exception = %d\n", RpcExceptionCode());
        }
        RpcEndExcept

        RpcTryExcept
        {
		NDRCContextUnmarshall( NULL, _Handle, buf, _RpcMessage.DataRepresentation  );
        }
        RpcExcept(1)
        {
        	printf("NDRCContextUnmarshall reported exception = %d\n", RpcExceptionCode());
        }
        RpcEndExcept


        NdrFreeBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg );


}

void m_CtxHello(
    /* [in] */ PCTXTYPE phContext)
{

    RPC_BINDING_HANDLE _Handle	=	0;

    RPC_MESSAGE _RpcMessage;

    MIDL_STUB_MESSAGE _StubMsg;

    char *buf;
    int i;

	printf("\n*******************************************************************\n");
	printf("**** CtxHello()                                                 ***\n");
	printf("*******************************************************************\n\n");

    RpcTryFinally
        {
        NdrClientInitializeNew(
                          ( PRPC_MESSAGE  )&_RpcMessage,
                          ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                          ( PMIDL_STUB_DESC  )&hello_StubDesc,
                          1);


        if(phContext != 0)
            {
            _Handle = NDRCContextBinding(( NDR_CCONTEXT  )phContext);;

            }
        else
            {
            RpcRaiseException(RPC_X_SS_IN_NULL_CONTEXT);
            }

        _StubMsg.BufferLength = 20U;
        NdrGetBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg, _StubMsg.BufferLength, _Handle );

        NdrClientContextMarshall(
                            ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                            ( NDR_CCONTEXT  )phContext,
                            1);
        printf("After NdrClientContextMarshall: Buflen=%d\nBuffer: ", _StubMsg.BufferLength );
        for(buf = _StubMsg.Buffer, i = 0; i < _StubMsg.BufferLength; i++)
        	printf("0x%x, ", buf[i] & 0x0FF);
        printf("\n\n");

        NdrSendReceive( (PMIDL_STUB_MESSAGE) &_StubMsg, (unsigned char __RPC_FAR *) _StubMsg.Buffer );

        }
    RpcFinally
        {
        NdrFreeBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg );

        }
    RpcEndFinally

}


void m_CtxClose(
    /* [out][in] */ PCTXTYPE __RPC_FAR *pphContext)
{

    RPC_BINDING_HANDLE _Handle	=	0;

    RPC_MESSAGE _RpcMessage;

    MIDL_STUB_MESSAGE _StubMsg;
    char *buf;
    int i;

	printf("\n*******************************************************************\n");
	printf("**** CtxClose()                                                 ***\n");
	printf("*******************************************************************\n\n");

    if(!pphContext)
        {
        RpcRaiseException(RPC_X_NULL_REF_POINTER);
        }
    RpcTryFinally
        {
        NdrClientInitializeNew(
                          ( PRPC_MESSAGE  )&_RpcMessage,
                          ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                          ( PMIDL_STUB_DESC  )&hello_StubDesc,
                          2);


        if(*pphContext != 0)
            {
            _Handle = NDRCContextBinding(( NDR_CCONTEXT  )*pphContext);;

            }

        _StubMsg.BufferLength = 20U;
        NdrGetBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg, _StubMsg.BufferLength, _Handle );

        NdrClientContextMarshall(
                            ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                            ( NDR_CCONTEXT  )*pphContext,
                            0);

        NdrSendReceive( (PMIDL_STUB_MESSAGE) &_StubMsg, (unsigned char __RPC_FAR *) _StubMsg.Buffer );

        if ( (_RpcMessage.DataRepresentation & 0X0000FFFFUL) != NDR_LOCAL_DATA_REPRESENTATION )
            NdrConvert( (PMIDL_STUB_MESSAGE) &_StubMsg, (PFORMAT_STRING) &__MIDL_ProcFormatString.Format[14] );


      printf("Before NdrClientContextUnmarshall: Buflen=%d\nBuffer: ",  _StubMsg.BufferLength );
        for(buf = _StubMsg.Buffer, i = 0; i < _StubMsg.BufferLength; i++)
        	printf("0x%x, ", buf[i] & 0x0FF);
        printf("\n\n");

        NdrClientContextUnmarshall(
                              ( PMIDL_STUB_MESSAGE  )&_StubMsg,
                              ( NDR_CCONTEXT __RPC_FAR * )pphContext,
                              _Handle);


		printf("\nNdrClientContextUnmarshall returned: handle=0x%p\n", *pphContext);


        }
    RpcFinally
        {
        NdrFreeBuffer( (PMIDL_STUB_MESSAGE) &_StubMsg );

        }
    RpcEndFinally

}

int interactive = 0;
void interact()
{
	if(interactive)
	{
		printf("\nPress any key to continue...");
		getch();
		printf("\n\n");
	}
}

void main(int argc, char **argv)
{
	RPC_STATUS status;
	unsigned long ulCode;
	PCTXTYPE hContext;
	char *pszStringBinding = NULL;
	RPC_BINDING_HANDLE Handle =	0;
	char buffer[255];

	int test_num = 0;
	int test_value = 31337;

	if(argc<2)
	{
		printf("USAGE: client.exe <test_number> [test_value] [interactive]\n"
			"Available tests:\n"
			"0. General test\n"
			"1. NULL pointer test\n"
			"2. Context rundown routine");
		return;
	}

	test_num = atoi(argv[1]);
	if(argc>2) test_value = atoi(argv[2]);
	if(argc>3) interactive = 1;

	status = RpcStringBindingCompose(NULL,
		"ncacn_np",
		NULL,
		"\\pipe\\hello",
		NULL,
		&pszStringBinding);

	if (status)
	{
		printf("RpcStringBindingCompose %x\n", status);
		exit(status);
	}

	status = RpcBindingFromStringBinding(pszStringBinding, &hBinding);

	if (status)
	{
		printf("RpcBindingFromStringBinding %x\n", status);
		exit(status);
	}

	RpcStringFree(&pszStringBinding);

	switch(test_num)
	{
	case 0:
		m_CtxOpen(&hContext, test_value);
		RpcBindingFree(&hBinding);
		m_CtxHello(hContext);
		interact();
		m_CtxClose(&hContext);
		break;
	case 1:
		/////////////////////////////////////////////////////////////////////////////////////////
		RpcTryExcept
		{
			Handle = NDRCContextBinding(NULL);
			printf("NDRCContextBinding(NULL) returned %p\n", Handle);
		}
		RpcExcept(1)
		{
			 printf("NDRCContextBinding(NULL) reported exception = %d\n", RpcExceptionCode());
		}
		RpcEndExcept

		m_CtxOpen2(&hContext, test_value);

		/////////////////////////////////////////////////////////////////////////////////////////
		RpcTryExcept
		{
			NDRCContextMarshall(NULL, &buffer);
			printf("NDRCContextMarshall(NULL) returned %p\n", Handle);
		}
		RpcExcept(1)
		{
			 printf("NDRCContextMarshall(NULL) reported exception = %d\n", RpcExceptionCode());
		}
		RpcEndExcept
		/////////////////////////////////////////////////////////////////////////////////////////
		break;
	case 2:
		CtxOpen(&hContext, test_value);
		interact();
		ExitProcess(0);
		break;
	default:
		printf("Unknown test %d\n", test_num);
	}

}


void __RPC_FAR * __RPC_USER midl_user_allocate(size_t len)
{
	return(malloc(len));
}

void __RPC_USER midl_user_free(void __RPC_FAR * ptr)
{
	free(ptr);
}