windows-server-2003/net/rras/netsh/ip/ipmon/routing.c

/*++

Copyright (c) 1998  Microsoft Corporation

Module Name:

    routing\netsh\ip\ipmon\routing.c

Abstract:

    Routing Command dispatcher.

Revision History:

    Dave Thaler              3/03/99  Created

--*/

#include "precomp.h"

const GUID g_RoutingGuid = ROUTING_GUID;
const GUID g_NetshGuid   = NETSH_ROOT_GUID;
PWCHAR g_pwszRouter      = NULL;

HRESULT RouterReset(LPCTSTR pszMachineName);

NS_CONTEXT_DUMP_FN     RoutingDump;
NS_CONTEXT_CONNECT_FN  RoutingConnect;

FN_HANDLE_CMD HandleRoutingReset;

DWORD                ParentVersion;
BOOL                 g_bRoutingDirty = FALSE;

CMD_ENTRY g_RoutingCmds[] =
{
    CREATE_CMD_ENTRY(IP_RESET, HandleRoutingReset),
};

ULONG g_ulRoutingNumTopCmds = sizeof(g_RoutingCmds)/sizeof(CMD_ENTRY);

DWORD
WINAPI
RoutingStartHelper(
    IN CONST GUID *pguidParent,
    IN DWORD       dwVersion
    )
{
    DWORD dwErr;
    NS_CONTEXT_ATTRIBUTES attMyAttributes;

    ParentVersion         = dwVersion;

    ZeroMemory(&attMyAttributes, sizeof(attMyAttributes));

    attMyAttributes.pwszContext   = L"routing";
    attMyAttributes.guidHelper    = g_RoutingGuid;
    attMyAttributes.dwVersion     = 1;
    attMyAttributes.dwFlags       = 0;
    attMyAttributes.ulNumTopCmds  = g_ulRoutingNumTopCmds;
    attMyAttributes.pTopCmds      = (CMD_ENTRY (*)[])&g_RoutingCmds;
    attMyAttributes.ulNumGroups   = 0; 
    attMyAttributes.pCmdGroups    = NULL;
    attMyAttributes.pfnCommitFn   = NULL; // RoutingCommit;
    attMyAttributes.pfnDumpFn     = RoutingDump;
    attMyAttributes.pfnConnectFn  = RoutingConnect;

    dwErr = RegisterContext( &attMyAttributes );

    return dwErr;
}

DWORD
HandleRoutingReset(
    IN      LPCWSTR   pwszMachine,
    IN OUT  LPWSTR   *ppwcArguments,
    IN      DWORD     dwCurrentIndex,
    IN      DWORD     dwArgCount,
    IN      DWORD     dwFlags,
    IN      LPCVOID   pvData,
    OUT     BOOL     *pbDone
    )
{
    // Call KennT's "router upgrade" function
    RouterReset( g_pwszRouter );

    return NO_ERROR;
}

DWORD
ConnectToRouter(
    IN  LPCWSTR  pwszRouter
    )
{
    DWORD    rc;

    if (g_pwszRouter != pwszRouter)
    {
        if (g_hMprConfig)
        {
            MprConfigServerDisconnect(g_hMprConfig);
            g_hMprConfig = NULL;
        }

        if (g_hMprAdmin)
        {
            MprAdminServerDisconnect(g_hMprAdmin);
            g_hMprAdmin = NULL;
        }

        if (g_hMIBServer)
        {
            MprAdminMIBServerDisconnect(g_hMIBServer);
            g_hMIBServer = NULL;
        }

        if (g_pwszRouter)
        {
            FREE(g_pwszRouter);
            g_pwszRouter = NULL;
        }

        if (pwszRouter)
        {
            g_pwszRouter = MALLOC((wcslen(pwszRouter)+1)*sizeof(WCHAR));
            if (g_pwszRouter)
            {
                wcscpy(g_pwszRouter, pwszRouter);
            }
            else
            {
                return ERROR_NOT_ENOUGH_MEMORY;
            }
        }
    }

    if (!g_hMprConfig)
    {
        //
        // first time connecting to router config
        //

        rc = MprConfigServerConnect ((LPWSTR)pwszRouter,
                                     &g_hMprConfig);

        if (rc isnot NO_ERROR)
        {
            //
            // cannot connect to router config.
            //
            return ERROR_CONNECT_REMOTE_CONFIG;
        }
    }

    //
    // Check to see if router is running. If so, get the handles
    //

    do
    {
        if (MprAdminIsServiceRunning((LPWSTR)pwszRouter))
        {
            if (MprAdminServerConnect((LPWSTR)pwszRouter, &g_hMprAdmin) == NO_ERROR)
            {
                DEBUG("Got admin handle");
                if (MprAdminMIBServerConnect ((LPWSTR)pwszRouter,
                                              &g_hMIBServer) == NO_ERROR)
                {
                    DEBUG("Got server handle");
                    break;
                }
                else
                {
                    MprAdminServerDisconnect (g_hMprAdmin);
                }
            }
        }
        g_hMprAdmin = g_hMIBServer = NULL;

    } while (FALSE);

    return NO_ERROR;
}

DWORD WINAPI
RoutingConnect(
    IN  LPCWSTR  pwszRouter
    )
{
    // If context info is dirty, reregister it
    if (g_bRoutingDirty)
    {
        RoutingStartHelper(NULL, ParentVersion);
    }

    return ConnectToRouter(pwszRouter);
}

DWORD
RoutingDump(
    IN  LPCWSTR     pwszRouter,
    IN  WCHAR     **ppwcArguments,
    IN  DWORD       dwArgCount,
    IN  PVOID       pvData
    )
{    
    DWORD dwErr;

    dwErr = ConnectToRouter(pwszRouter);

    // Dump routing information
    DisplayMessage( g_hModule, DMP_ROUTING_HEADER_COMMENTS);
    DisplayMessageT(DMP_ROUTING_HEADER);
    DisplayMessageT(DMP_POPD);

    return NO_ERROR;
}