#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <ntddser.h>

#include <windows.h>

#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include <malloc.h>

#include <rasman.h>
#include <dim.h>
#include <routprot.h>
#include <ipxrtdef.h>
#include <utils.h>

// [pmay] this will no longer be neccessary when the ipx router
// is converted to use MprInfo api's.
typedef RTR_INFO_BLOCK_HEADER IPX_INFO_BLOCK_HEADER, *PIPX_INFO_BLOCK_HEADER;


DIM_ROUTER_INTERFACE dimif;

LPVOID
GetInfoEntry(PIPX_INFO_BLOCK_HEADER	InterfaceInfop,
	     ULONG			InfoEntryType);

VOID
TestMib(PDIM_ROUTER_INTERFACE dimifp);

VOID
SaveConfiguration(PDIM_ROUTER_INTERFACE dimifp);

VOID
RestoreConfiguration(PDIM_ROUTER_INTERFACE dimifp);

VOID
RouterStopped(DWORD	protid,
	      DWORD	err)
{
    printf("main: RouterStopped: protid 0x%x err 0x%x\n", protid, err);
}

DWORD
SaveInterfaceInfo(
                IN      HANDLE          hDIMInterface, 
                IN      DWORD           dwProtocolId,
                IN      LPVOID          pInterfaceInfo,
		IN	DWORD		cbInterfaceInfoSize)
{
    printf("Main: SaveInterfaceInfo: entered for Dim if handle %d\n", hDIMInterface);

    return NO_ERROR;
}


HANDLE	    RmEvent;

//*** IPX Router Global Info ***

struct _GlobalInfo {

    IPX_INFO_BLOCK_HEADER	GI_Header;
    IPX_TOC_ENTRY		GI_RipTocEntry;
    IPX_TOC_ENTRY		GI_SapTocEntry;
    IPX_GLOBAL_INFO		GI_IpxGlobalInfo;
    WCHAR			SapName[MAX_DLL_NAME];
    RIP_GLOBAL_INFO		GI_RipGlobalInfo;
    RIP_ROUTE_FILTER_INFO	GI_RipRouteFilter[15];
    SAP_GLOBAL_INFO		GI_SapGlobalInfo;

    } gi;

struct _GlobalInfo  getgi;

PWCHAR	IpxRipNamep = L"IPXRIP";
PWCHAR	IpxSapNamep = L"IPXSAP";

struct _DBG_IF {

    IPX_INFO_BLOCK_HEADER	header;
    IPX_TOC_ENTRY		toc[6];
    IPX_IF_INFO 		ipxifinfo;
    RIP_IF_INFO 		ripifinfo;
    SAP_IF_INFO 		sapifinfo;
    IPXWAN_IF_INFO		ipxwanifinfo;
    IPX_ADAPTER_INFO		adapterinfo;
    IPX_STATIC_ROUTE_INFO	routeinfo[10];
    IPX_STATIC_SERVICE_INFO	srvinfo[10];
    } dbgif;

LPVOID	     dbgifgetp;

WCHAR		MainAdapterName[48];
WCHAR		MainInterfaceName[48];
ULONG		MainInterfaceType = ROUTER_IF_TYPE_DEDICATED;

HANDLE		MainInterfaceHandle;


#define ipxtoc			  dbgif.header.TocEntry[0]
#define riptoc			  dbgif.toc[0]
#define saptoc			  dbgif.toc[1]
#define ipxwantoc		  dbgif.toc[2]
#define adaptertoc		  dbgif.toc[3]
#define routetoc		  dbgif.toc[4]
#define srvtoc			  dbgif.toc[5]

VOID
MainAddInterface(VOID)
{
    PIPX_IF_INFO	ipxinfop;
    PRIP_IF_INFO	ripifinfop;
    PSAP_IF_INFO	sapifinfop;
    PIPXWAN_IF_INFO	ipxwaninfop;
    PIPX_ADAPTER_INFO	adapterinfop;
    PIPX_STATIC_ROUTE_INFO   routeinfop;
    PIPX_STATIC_SERVICE_INFO srvinfop;
    int			i,j;
    BOOL		str = FALSE; // static routes info
    DWORD		rc, update;

    memset(&dbgif, 0, sizeof(dbgif));

    printf("Enter the interface name:");
    scanf("%S", &MainInterfaceName);

    printf("Enter the DIM Interface Handle:");
    scanf("%d", &i);

    printf("Enter interface type (0,1 - wan client, 2- wan router, 3 - lan, 4 - internal):");
    scanf("%d", &MainInterfaceType);

    switch(MainInterfaceType) {

	case 2:

	str = TRUE;
	break;

	case 3:

	printf("Enter LAN Adapter name:");
	scanf("%S", &MainAdapterName);
	break;

	default:

	break;
    }

    printf("Enter update mode on this interface: 1-standard, 2-none, 3-autostatic:");
    scanf("%d", &update);

    dbgif.header.Version = IPX_ROUTER_VERSION_1;
    dbgif.header.Size = sizeof(dbgif);
    if (str)
	dbgif.header.TocEntriesCount = 7;
    else
	dbgif.header.TocEntriesCount = 6;

    ipxtoc.InfoType = IPX_INTERFACE_INFO_TYPE;
    ipxtoc.InfoSize = sizeof(IPX_IF_INFO);
    ipxtoc.Count = 1;
    ipxtoc.Offset = (ULONG)((PUCHAR)&dbgif.ipxifinfo - (PUCHAR)&dbgif);

    riptoc.InfoType = IPX_PROTOCOL_RIP;
    riptoc.InfoSize = sizeof(RIP_IF_INFO);
    riptoc.Count = 1;
    riptoc.Offset = ipxtoc.Offset + sizeof(IPX_IF_INFO);

    saptoc.InfoType = IPX_PROTOCOL_SAP;
    saptoc.InfoSize = sizeof(SAP_IF_INFO);
    saptoc.Count = 1;
    saptoc.Offset = riptoc.Offset + sizeof(RIP_IF_INFO);

    ipxwantoc.InfoType = IPXWAN_INTERFACE_INFO_TYPE;
    ipxwantoc.InfoSize = sizeof(IPXWAN_IF_INFO);
    ipxwantoc.Count = 1;
    ipxwantoc.Offset = saptoc.Offset + sizeof(SAP_IF_INFO);

    adaptertoc.InfoType = IPX_ADAPTER_INFO_TYPE;
    adaptertoc.InfoSize = sizeof(IPX_ADAPTER_INFO);
    adaptertoc.Count = 1;
    adaptertoc.Offset = ipxwantoc.Offset + ipxwantoc.InfoSize;

    if (str) {

	routetoc.InfoType = IPX_STATIC_ROUTE_INFO_TYPE;
	routetoc.InfoSize = sizeof(IPX_STATIC_ROUTE_INFO);
	routetoc.Count = 3;
	routetoc.Offset = adaptertoc.Offset + adaptertoc.InfoSize;

	srvtoc.InfoType = IPX_STATIC_SERVICE_INFO_TYPE;
	srvtoc.InfoSize = sizeof(IPX_STATIC_SERVICE_INFO);
	srvtoc.Count = 3;
	srvtoc.Offset = (ULONG)((PUCHAR)&dbgif.srvinfo - (PUCHAR)&dbgif);
    }

    ipxinfop = (PIPX_IF_INFO)((PUCHAR)&dbgif + ipxtoc.Offset);
    ipxinfop->AdminState = ADMIN_STATE_ENABLED;
    ipxinfop->NetbiosAccept = ADMIN_STATE_ENABLED;
    ipxinfop->NetbiosDeliver = ADMIN_STATE_DISABLED;

    adapterinfop = (PIPX_ADAPTER_INFO)((PUCHAR)&dbgif + adaptertoc.Offset);
    adapterinfop->PacketType = 2;
    wcscpy(adapterinfop->AdapterName, MainAdapterName);

    ripifinfop = &dbgif.ripifinfo;

    ripifinfop->AdminState = ADMIN_STATE_ENABLED;
    ripifinfop->UpdateMode = update;
    ripifinfop->PacketType = IPX_STANDARD_PACKET_TYPE;
    ripifinfop->Supply = ADMIN_STATE_ENABLED;
    ripifinfop->Listen = ADMIN_STATE_ENABLED;
    ripifinfop->EnableGlobalFiltering = ADMIN_STATE_DISABLED;

    sapifinfop = &dbgif.sapifinfo;

    sapifinfop->AdminState = ADMIN_STATE_ENABLED;
    sapifinfop->UpdateMode = update;
    sapifinfop->PacketType = IPX_STANDARD_PACKET_TYPE;
    sapifinfop->Supply = ADMIN_STATE_ENABLED;
    sapifinfop->Listen = ADMIN_STATE_ENABLED;
    sapifinfop->EnableGlobalFiltering = ADMIN_STATE_DISABLED;

    ipxwaninfop = &dbgif.ipxwanifinfo;
    ipxwaninfop->AdminState = ADMIN_STATE_ENABLED;

    if (str) {

	routeinfop = dbgif.routeinfo;
	srvinfop = dbgif.srvinfo;

	for(j=0; j <3; j++, routeinfop++, srvinfop++)
	{
	    memset(routeinfop->Network, 0, 4);
	    routeinfop->Network[3] = i * 0x10 + j;
	    routeinfop->HopCount = 1;
	    routeinfop->TickCount = 1;
	    memset(routeinfop->NextHopMacAddress, i * 0x10 + j, 6);

	    srvinfop->Type = 4;
	    strcpy(srvinfop->Name, "TEST_STATIC_SERVER00");
	    srvinfop->Name[strlen(srvinfop->Name) - 2] = i + '0';
	    srvinfop->Name[strlen(srvinfop->Name) - 1] = j + '0';
	    memcpy(srvinfop->Network, routeinfop->Network, 4);
	    memset(srvinfop->Node, i * 0x10 + j, 6);
	    memset(srvinfop->Socket, j, 2);
	    srvinfop->HopCount = 2;
	}
    }

    rc = (*dimif.AddInterface)(
			       MainInterfaceName,
			       &dbgif,
			       NULL,
			       NULL,
			       MainInterfaceType,
			       TRUE,
			       (HANDLE)i,
			       &MainInterfaceHandle);

    printf("main: AddInterface returned %d and IPX Interface Index  = %d\n",
	    rc,	MainInterfaceHandle);

}

VOID
MainDeleteInterface(VOID)
{
    ULONG ii;

    printf("Enter IPX Interface Index:");
    scanf("%d", &ii);
    (*dimif.DeleteInterface)((HANDLE)ii);
}

VOID
MainGetInterface(VOID)
{
    ULONG ii;
    DWORD ifinfosize;
    DWORD infiltinfosize, outfiltinfosize;
    DWORD rc;

    printf("Enter IPX Interface Index:");
    scanf("%d", &ii);

    rc = (*dimif.GetInterfaceInfo)((HANDLE)ii,
				NULL,
				&ifinfosize,
				NULL,
				&infiltinfosize,
				NULL,
				&outfiltinfosize);

    if(rc != ERROR_INSUFFICIENT_BUFFER) {

	printf("main: GetInterfaceInfo failed with rc=%d\n", rc);
	return;
    }

    dbgifgetp = GlobalAlloc(GPTR, ifinfosize);

    rc = (*dimif.GetInterfaceInfo)((HANDLE)ii,
				dbgifgetp,
				&ifinfosize,
				NULL,
				&infiltinfosize,
				NULL,
				&outfiltinfosize);

    if(rc != NO_ERROR) {

	printf("main: GetInterfaceInfo failed on second call with rc=%d\n", rc);
	return;
    }
}

VOID
MainSetInterface(VOID)
{
    PIPX_IF_INFO	ipxinfop;
    PRIP_IF_INFO	ripinfop;
    PSAP_IF_INFO	sapinfop;
    DWORD		rc;
    ULONG		IfIndex;
    ULONG		AdminState, RipAdminState, RipFilter, Update;
    ULONG		ifinfosize, infiltinfosize, outfiltinfosize;
    PIPX_INFO_BLOCK_HEADER	InterfaceInfop;

    printf("Enter IPX Interface Index:");
    scanf("%d", &IfIndex);

    printf("Enter the new IPX Admin State: 1- ENABLED 2- DISABLED:");
    scanf("%d", &AdminState);

    printf("Enter the new RIP Admin State: 1- ENABLED 2- DISABLED:");
    scanf("%d", &RipAdminState);

    printf("Enter Update Mode on this interface: 1- STANDARD, 2- NONE, 3- AUTO-STATIC:");
    scanf("%d", &Update);

    printf("Enter the RIP global filter action on this if: 1- ENABLED, 2- DISABLED:");
    scanf("%d", &RipFilter);

    rc = (*dimif.GetInterfaceInfo)((HANDLE)IfIndex,
				NULL,
				&ifinfosize,
				NULL,
				&infiltinfosize,
				NULL,
				&outfiltinfosize);

    if(rc != ERROR_INSUFFICIENT_BUFFER) {

	printf("main: GetInterfaceInfo failed with rc=%d\n", rc);
	return;
    }

    InterfaceInfop = GlobalAlloc(GPTR, ifinfosize);

    rc = (*dimif.GetInterfaceInfo)((HANDLE)IfIndex,
				InterfaceInfop,
				&ifinfosize,
				NULL,
				&infiltinfosize,
				NULL,
				&outfiltinfosize);

    if(rc != NO_ERROR) {

	printf("main: GetInterfaceInfo failed on second call with rc=%d\n", rc);
	return;
    }

    ipxinfop = GetInfoEntry(InterfaceInfop, IPX_INTERFACE_INFO_TYPE);
    ipxinfop->AdminState = AdminState;

    ripinfop = GetInfoEntry(InterfaceInfop, IPX_PROTOCOL_RIP);
    ripinfop->AdminState = RipAdminState;
    ripinfop->UpdateMode = Update;
    ripinfop->EnableGlobalFiltering = RipFilter;

    sapinfop = GetInfoEntry(InterfaceInfop, IPX_PROTOCOL_SAP);
    sapinfop->UpdateMode = Update;


    rc = (*dimif.SetInterfaceInfo)(
			       (HANDLE)IfIndex,
			       InterfaceInfop,
			       NULL,
			       NULL);

    printf("main: SetInterfaceInfo returned %d \n", rc);
}

VOID
MainUpdate(VOID)
{
    ULONG	    IfIndex;
    DWORD	    UpdateResult;
    DWORD	    rc;

    printf("Enter interface index:");
    scanf("%d", &IfIndex);

    RmEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    rc = (*dimif.UpdateRoutes)((HANDLE)IfIndex, RmEvent);

    if(rc != NO_ERROR) {

	printf("UpdateRoutes failed with rc = %d\n", rc);
	return;
    }

    printf("Main: Waiting for the update result ... (for 30 sec)\n");

    rc = WaitForSingleObject(RmEvent, 30000);

    if(rc == WAIT_TIMEOUT) {

	printf("Main: Update wait failed with timeout\n");
	return;
    }

    if((rc = (*dimif.GetUpdateRoutesResult)((HANDLE)IfIndex, &UpdateResult)) != NO_ERROR) {

	printf("Cannot get update result rc = %d\n", rc);
    }
    else
    {
	printf("UpdateResult = %d\n", UpdateResult);
    }

    CloseHandle(RmEvent);
}

VOID
SetRipFilters(VOID)
{
    ULONG			 filtcount, i, netnumber, rc;
    PRIP_ROUTE_FILTER_INFO	 rfip;

    printf("Enter RIP global filtering action: 1-advertise, 2-suppress:");
    scanf("%d", &gi.GI_RipGlobalInfo.RouteFilterAction);

    printf("Enter the number of filters (up to 16):");
    scanf("%d", &filtcount);

    for(i=0, rfip = &gi.GI_RipGlobalInfo.RouteFilter[0];
	i<filtcount;
	i++, rfip++)
    {
	printf("Enter net nr for filter # %d:", i);
	scanf("%x", &netnumber);

	PUTULONG2LONG(rfip->Network, netnumber);
    }

    gi.GI_RipGlobalInfo.RouteFiltersCount = filtcount;

    rc = (*dimif.SetGlobalInfo)(&gi);

    printf("SetGlobalInfo returned %d\n", rc);
}

VOID _cdecl
main(
    IN WORD argc,
    IN LPSTR argv[]
    )

{
    DWORD	rc;
    int 	i;
    PIPX_INFO_BLOCK_HEADER	   ph;
    PIPX_TOC_ENTRY		   tocep;
    PIPX_GLOBAL_INFO		   ipxgp;


    dimif.RouterStopped = RouterStopped;
    dimif.SaveInterfaceInfo = SaveInterfaceInfo;

    ph = &gi.GI_Header;
    ph->Version = 1;
    ph->Size = sizeof(gi);
    ph->TocEntriesCount = 3;

    tocep = ph->TocEntry;
    tocep->InfoType = IPX_GLOBAL_INFO_TYPE;
    tocep->InfoSize = sizeof(IPX_GLOBAL_INFO);
    tocep->Count = 1;
    tocep->Offset = (ULONG)((PUCHAR)&gi.GI_IpxGlobalInfo - (PUCHAR)&gi);

    tocep++;

    tocep->InfoType = IPX_PROTOCOL_RIP;
    tocep->InfoSize = sizeof(RIP_GLOBAL_INFO);
    tocep->Count = 1;
    tocep->Offset = (ULONG)((PUCHAR)&gi.GI_RipGlobalInfo - (PUCHAR)&gi);

    tocep++;

    tocep->InfoType = IPX_PROTOCOL_SAP;
    tocep->InfoSize = sizeof(SAP_GLOBAL_INFO);
    tocep->Count = 1;
    tocep->Offset = (ULONG)((PUCHAR)&gi.GI_SapGlobalInfo - (PUCHAR)&gi);

    ipxgp = &gi.GI_IpxGlobalInfo;
    ipxgp->NumRoutingProtocols = 2;
    wcscpy(ipxgp->DllName, IpxRipNamep);
    wcscpy(gi.SapName, IpxSapNamep);

    for(;;) {

	printf("IPX Router Test Menu:\n");
	printf("1. Start Router\n");
	printf("2. Stop Router\n");
	printf("3. Add Interface\n");
	printf("4. Delete Interface\n");
	printf("5. Get Interface\n");
	printf("6. Set Interface\n");
	printf("7. Update\n");
	printf("8. Save Configuration\n");
	printf("9. Restore Configuration\n");
	printf("10. Rip filters test\n");
	printf("20. MIB Tests\n");
	printf("99. Exit\n");
	printf("Enter your option:");

	scanf("%d", &i);

	switch(i) {

	    case 1:

		rc = StartRouter(&dimif, FALSE, &gi);

		printf("main: StartRouter returned rc=%d\n", rc);
		break;

	    case 2:

		(*dimif.StopRouter)();
		printf("main: StopRouter \n");
		break;

	    case 3:

		MainAddInterface();
		break;

	    case 4:

		MainDeleteInterface();
		break;

	    case 5:

		MainGetInterface();
		break;

	    case 6:

		MainSetInterface();
		break;

	    case 7:

		MainUpdate();
		break;

	    case 8:

		SaveConfiguration(&dimif);
		break;

	    case 9:

		RestoreConfiguration(&dimif);
		break;

	    case 10:

		SetRipFilters();
		break;

	    case 20:

		TestMib(&dimif);
		break;

	    case 99:

		printf("exit\n");
		goto Exit;

	    default:

		break;
	}
    }

Exit:

    ExitProcess(0);
}

/*++

Function:	GetTocEntry
Descr:		Returns a pointer to the specified table of contents entry
		in the interface info block.

--*/



PIPX_TOC_ENTRY
GetTocEntry(PIPX_INFO_BLOCK_HEADER	InterfaceInfop,
	    ULONG			InfoEntryType)
{
    PIPX_TOC_ENTRY	tocep;
    UINT		i;

    for(i=0, tocep = InterfaceInfop->TocEntry;
	i<InterfaceInfop->TocEntriesCount;
	i++, tocep++) {

	if(tocep->InfoType == InfoEntryType) {

	    return tocep;
	}
    }

    return NULL;
}

/*++

Function:	GetInfoEntry
Descr:		Returns a pointer to the specified info entry in the interface
		control block. If more then one entries, returns a pointer to
		the first one.

--*/

LPVOID
GetInfoEntry(PIPX_INFO_BLOCK_HEADER	InterfaceInfop,
	     ULONG			InfoEntryType)
{
    PIPX_TOC_ENTRY	tocep;

    if(tocep = GetTocEntry(InterfaceInfop, InfoEntryType)) {

	return((LPVOID)((PUCHAR)InterfaceInfop + tocep->Offset));
    }
    else
    {
	return NULL;
    }
}