|
|
#include "allinc.h"
DWORD RtmEventCallback ( IN RTM_ENTITY_HANDLE hRtmHandle, IN RTM_EVENT_TYPE retEvent, IN PVOID pContext1, IN PVOID pContext2 )
/*++
Routine Description:
This callback is given by RTM when we have changed dests to process. We just queue a work item to process changed destinations.
Arguments:
hRtmHandle - Handle that we got during registration retEvent - Event type - we only handle events of type "more changes available" for now pContext1 - Notification handle on which changes are available
pContext2 - Context supplied during notification registration time Return Value:
Status of the operation. --*/
{ DWORD dwResult; // Only "change notifications available" is supported
if (retEvent != RTM_CHANGE_NOTIFICATION) { return ERROR_NOT_SUPPORTED; }
return ((HANDLE) pContext1) == g_hNotification ? ProcessChanges(g_hNotification) : ProcessDefaultRouteChanges( g_hDefaultRouteNotification ); }
DWORD WINAPI ProcessChanges ( IN HANDLE hNotifyHandle )
/*++
Routine Description:
Upon learning that we have changed destinations to process, this function gets called. We retrieve all destinations to process and take appropriate action.
Arguments:
hRtmHandle - RTM registration handle hNotifyHandle - Handle correponding to the change notification that is being signalled Return Value:
Status of the operation. --*/
{ PRTM_DEST_INFO pDestInfo; PRTM_ROUTE_INFO pRouteInfo; DWORD dwDests; DWORD dwResult; BOOL bMark = FALSE;
TraceEnter("ProcessChanges");
pRouteInfo = HeapAlloc( IPRouterHeap, 0, RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) );
if (pRouteInfo == NULL) { Trace1( ERR, "ProcessChanges : error allocating %d bytes for " "route info", RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) );
TraceLeave("ProcessChanges");
return ERROR_NOT_ENOUGH_MEMORY; } pDestInfo = HeapAlloc( IPRouterHeap, 0, RTM_SIZE_OF_DEST_INFO(g_rtmProfile.NumberOfViews) );
if (pDestInfo == NULL) { Trace1( ERR, "ProcessChanges : error allocating %d bytes for " "dest. info", RTM_SIZE_OF_DEST_INFO(g_rtmProfile.NumberOfViews) );
HeapFree(IPRouterHeap, 0, pRouteInfo); TraceLeave("ProcessChanges");
return ERROR_NOT_ENOUGH_MEMORY; }
dwDests = 1; // Get each changed dest from the table
do { RtmGetChangedDests(g_hLocalRoute, hNotifyHandle, &dwDests, pDestInfo); if (dwDests < 1) { break; }
//
// For default routes, mark the route so that future changes
// are managed by ProcessDefaultRouteChanges.
//
// We need to do this here so that default routes added by
// routing protocols RIP/OSPF are marked for change notification
// These default routes are added by entities other than
// RouterManager. Default routes added by RM i.e STATIC,
// AUTO-STATIC and NETMGMT default routes are already marked for
// change notification when they are added by RM.
//
// By marking routing protocol default routes here we make sure
// that all default routes are subsequently handled by marked changed
// mechanism (ProcessDefaultRouteChanges).
//
if (pDestInfo->DestAddress.NumBits is 0) { TraceRoute2( ROUTE, "Checking dest %d.%d.%d.%d/%d is marked", PRINT_IPADDR(*(ULONG *)pDestInfo->DestAddress.AddrBits), PRINT_IPADDR(pDestInfo->DestAddress.NumBits) );
dwResult = RtmIsMarkedForChangeNotification( g_hNetMgmtRoute, g_hDefaultRouteNotification, pDestInfo->DestHandle, &bMark );
if (dwResult is NO_ERROR) { if (bMark) { //
// default route is already marked, nothing further
// to do here. This default route change will be
// handled by ProcessDefaultRouteChanges
//
TraceRoute0( ROUTE, "ProcessChanges : Route 0/0 is already marked" );
RtmReleaseChangedDests(g_hLocalRoute, hNotifyHandle, dwDests, pDestInfo);
continue; }
//
// Default route is not marked, mark it
//
dwResult = RtmMarkDestForChangeNotification( g_hNetMgmtRoute, g_hDefaultRouteNotification, pDestInfo->DestHandle, TRUE );
if (dwResult isnot NO_ERROR) { //
// Failed to mark 0/0 route. The consequence is that
// only best route changes are processed. We will
// have to live with the fact that we cannot
// install multiple NETMGMT default routes since
// this is performed by the mark dest. change
// processing (in ProcessDefaultRouteChanges)
//
Trace1( ERR, "ProcessChanges: error %d marking default route", dwResult ); } }
else { //
// Failed to check is 0/0 destination has been
// marked for change notification
// - Refer previous comment
//
Trace1( ERR, "ProcessChanges: error %d checking if default route " "marked", dwResult ); } }
// Check if we have a route in Unicast view
if (pDestInfo->BelongsToViews & RTM_VIEW_MASK_UCAST) { // This is either a new or update route
// Update the same route in KM Frwder
ASSERT(pDestInfo->ViewInfo[0].ViewId is RTM_VIEW_ID_UCAST);
dwResult = RtmGetRouteInfo(g_hLocalRoute, pDestInfo->ViewInfo[0].Route, pRouteInfo, NULL);
// An error mean route just got deleted
// Ignore this change as it is obsolete
if (dwResult is NO_ERROR) { ChangeRouteWithForwarder(&pDestInfo->DestAddress, pRouteInfo, TRUE, TRUE);
RtmReleaseRouteInfo(g_hLocalRoute, pRouteInfo); } } else { // The last UCAST route has been deleted
// Delete the same route from KM Frwder
ChangeRouteWithForwarder(&pDestInfo->DestAddress, NULL, FALSE, TRUE); }
RtmReleaseChangedDests(g_hLocalRoute, hNotifyHandle, dwDests, pDestInfo); } while (TRUE);
HeapFree(IPRouterHeap, 0, pRouteInfo);
HeapFree(IPRouterHeap, 0, pDestInfo); TraceLeave("ProcessChanges");
return NO_ERROR; }
DWORD WINAPI ProcessDefaultRouteChanges( IN HANDLE hNotifyHandle )
/*++
Routine Description:
This function is invoked in response to changes to the default route. If the best default route is owned by protocol PROTO_IP_NETMGMT enumerate all PROTO_IP_NETMGMT routes for default route 0/0 and set them as one multihop route to the forwarder Arguments:
hRtmHandle - RTM registration handle hNotifyHandle - Handle correponding to the change notification that is being signalled
Return Value:
NO_ERROR - Success
System error code - Otherwise --*/ { PRTM_DEST_INFO pDestInfo; PRTM_ROUTE_INFO pRouteInfo; DWORD dwDests; DWORD dwResult;
TraceEnter("ProcessDefaultRouteChanges"); pRouteInfo = HeapAlloc( IPRouterHeap, 0, RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) ); if (pRouteInfo == NULL) { Trace1( ERR, "ProcessDefaultRouteChanges : error allocating %d bytes for " "route info", RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) );
TraceLeave("ProcessDefaultRouteChanges");
return ERROR_NOT_ENOUGH_MEMORY; } pDestInfo = HeapAlloc( IPRouterHeap, 0, RTM_SIZE_OF_DEST_INFO(g_rtmProfile.NumberOfViews) );
if (pDestInfo == NULL) { Trace1( ERR, "ProcessDefaultRouteChanges : error allocating %d bytes for " "dest. info", RTM_SIZE_OF_DEST_INFO(g_rtmProfile.NumberOfViews) );
HeapFree(IPRouterHeap, 0, pRouteInfo); TraceLeave("ProcessDefaultRouteChanges");
return ERROR_NOT_ENOUGH_MEMORY; }
do { //
// retreive changed dests
//
dwDests = 1;
dwResult = RtmGetChangedDests( g_hNetMgmtRoute, hNotifyHandle, &dwDests, pDestInfo );
if ((dwResult isnot NO_ERROR) and (dwResult isnot ERROR_NO_MORE_ITEMS)) { Trace1( ERR, "ProcessDefaultRouteChanges: error %d retrieving changed dests", dwResult );
break; }
if (dwDests < 1) { //
// no more dests to enumerate
//
break; }
do { //
// Make sure this the default route 0/0. This functions
// only processes default route changes.
//
if ((pDestInfo->DestAddress.NumBits isnot 0) or (*((ULONG *)pDestInfo->DestAddress.AddrBits) isnot 0)) { Trace2( ERR, "ProcessDefaultRouteChanges: Not default route %d.%d.%d.%d/%d", PRINT_IPADDR(*((ULONG *)pDestInfo->DestAddress.AddrBits)), pDestInfo->DestAddress.NumBits );
break; }
//
// If all routes to 0/0 have been deleted,
// delete it from the forwarder too.
//
if (!(pDestInfo->BelongsToViews & RTM_VIEW_MASK_UCAST)) { dwResult = ChangeRouteWithForwarder( &(pDestInfo->DestAddress), NULL, FALSE, TRUE );
break; }
//
// A route to 0/0 was added/updated
//
if (pDestInfo->ViewInfo[0].Owner isnot g_hNetMgmtRoute) { //
// Default route is not owned by PROTO_IP_NETMGT
// Add only the best route to forwarder
//
TraceRoute1( ROUTE, "ProcessDefaultRouteChanges: Adding non-NetMgmt" " route to forwarder, owner RTM handle 0x%x", pDestInfo->ViewInfo[0].Owner );
dwResult = RtmGetRouteInfo( g_hNetMgmtRoute, pDestInfo->ViewInfo[0].Route, pRouteInfo, NULL );
if (dwResult is NO_ERROR) { ChangeRouteWithForwarder( &pDestInfo->DestAddress, pRouteInfo, TRUE, TRUE );
dwResult = RtmReleaseRouteInfo( g_hNetMgmtRoute, pRouteInfo ); if (dwResult isnot NO_ERROR) { Trace1( ERR, "ProcessDefaultRouteChanges: Failed " "to release route info", dwResult ); } } break; } //
// Default route owned by PROTO_IP_NETMGMT
//
//
// First delete existing 0/0 from the TCP/IP forwarder
//
dwResult = ChangeRouteWithForwarder( &(pDestInfo->DestAddress), NULL, FALSE, TRUE );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "ProcessDefaultRouteChanges: error %d deleting " "old NetMgmt default routes from forwarder", dwResult ); // break;
}
//
// Second add all NETMGMT 0/0 to the TCP/IP forwarder
//
AddNetmgmtDefaultRoutesToForwarder(pDestInfo); } while( FALSE );
//
// release handles to changed destinations
//
dwResult = RtmReleaseChangedDests( g_hNetMgmtRoute, hNotifyHandle, dwDests, pDestInfo );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "ProcessDefaultRouteChanges: error %d releasing dests ", dwResult ); }
} while ( TRUE );
HeapFree(IPRouterHeap, 0, pRouteInfo);
HeapFree(IPRouterHeap, 0, pDestInfo); TraceLeave("ProcessDefaultRouteChanges");
return dwResult; }
DWORD WINAPI AddNetmgmtDefaultRoutesToForwarder( PRTM_DEST_INFO pDestInfo ) /*++
Routine Description:
This routine enumerates the routes to 0/0 added by protocol PROTO_IP_NETMGT and adds them to the forwarder. This routine is invoked in response to any change to the default route If the best default route is owned by PROTO_IP_NETMGMT, all PROTO_IP_NETMGMT default routes are added to the TCP/IP forwarder.
This is required since the TCP/IP stack does dead gateway detection and that required multiple default routes if present to be installed in the stack.
An implicit assumption here is that PROTO_IP_NETMGMT routes alone merit this treatment. In case of static or other protocol generated 0/0 routes, only the best route is added to the stack. It is assumed that in the later case(s) the administrator (for static routes) or the protocol has a better idea of routing and so the dead gateway detection is suppressed in the stack by the addition of the best route to 0/0 alone.
Arguments:
pDestInfo - RTM destination info structure of 0/0 route
Return Value :
NO_ERROR - Sucess
Win32 error code - Otherwise
--*/ { DWORD dwResult, dwNumHandles = 0, i; BOOL bRelEnum = FALSE, bRelRoutes = FALSE; PRTM_ROUTE_INFO pRouteInfo; PRTM_ROUTE_HANDLE pHandles; RTM_ENUM_HANDLE hRouteEnum;
dwNumHandles = pDestInfo->ViewInfo[0].NumRoutes; pHandles = HeapAlloc( IPRouterHeap, 0, dwNumHandles * sizeof(RTM_ROUTE_HANDLE) );
if (pHandles == NULL) { Trace1( ERR, "AddNetmgmtDefaultRoutesToForwarder: error allocating %d bytes" "for route handles", dwNumHandles * sizeof(RTM_ROUTE_HANDLE) ); return ERROR_NOT_ENOUGH_MEMORY; }
pRouteInfo = HeapAlloc( IPRouterHeap, 0, RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) ); if (pRouteInfo == NULL) { Trace1( ERR, "AddNetmgmtDefaultRoutesToForwarder: error allocating %d bytes" "for route info", RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) ); HeapFree(IPRouterHeap, 0, pHandles); return ERROR_NOT_ENOUGH_MEMORY; }
do { //
// Enumerate and add all NETMGMT routes to the forwarder
//
dwResult = RtmCreateRouteEnum( g_hNetMgmtRoute, pDestInfo->DestHandle, RTM_VIEW_MASK_UCAST, RTM_ENUM_OWN_ROUTES, NULL, 0, NULL, 0, &hRouteEnum );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "AddNetmgmtDefaultRoutesToForwarder: error %d creating route " "enumeration", dwResult );
break; }
bRelEnum = TRUE; dwResult = RtmGetEnumRoutes( g_hNetMgmtRoute, hRouteEnum, &dwNumHandles, pHandles );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "ProcessDefaultRouteChanges:error %d enumerating " "routes", dwResult );
break; }
bRelRoutes = TRUE;
//
// Change route with the forwarder
//
for (i = 0; i < dwNumHandles; i++) { dwResult = RtmGetRouteInfo( g_hNetMgmtRoute, pHandles[i], pRouteInfo, NULL );
if (dwResult is NO_ERROR) { ChangeRouteWithForwarder( &(pDestInfo->DestAddress), pRouteInfo, TRUE, FALSE );
dwResult = RtmReleaseRouteInfo( g_hNetMgmtRoute, pRouteInfo );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "ProcessDefaultRouteChanges: error %d releasing " "route info ", dwResult ); } } else { Trace2( ERR, "ProcessDefaultRouteChanges: error %d getting route " "info for route %d", dwResult, i ); } } } while( FALSE );
//
// Release handles
//
if (bRelRoutes) { Trace0(ROUTE, "Releasing routes to 0/0");
dwResult = RtmReleaseRoutes( g_hNetMgmtRoute, dwNumHandles, pHandles );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "ProcessDefaultRouteChanges: error %d deleting enum " "handle", dwResult ); } }
if (bRelEnum) { Trace0(ROUTE, "Releasing route enum for 0/0"); dwResult = RtmDeleteEnumHandle( g_hNetMgmtRoute, hRouteEnum );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "ProcessDefaultRouteChanges: error %d deleting enum " "handle", dwResult ); } }
HeapFree(IPRouterHeap, 0, pHandles); HeapFree(IPRouterHeap, 0, pRouteInfo);
return dwResult; }
DWORD AddRtmRoute ( IN HANDLE hRtmHandle, IN PINTERFACE_ROUTE_INFO pRtInfo, IN DWORD dwRouteFlags, IN DWORD dwNextHopMask, IN DWORD dwTimeToLive, OUT HANDLE *phRtmRoute )
/*++
Routine Description:
Adds a route to RTM with the specified route information.
Arguments:
hRtmHandle - RTM registration handle used in RTM calls
pRtInfo -
dwNextHopMask -
dwTimeToLive - Time for which the route is kept in RTM before being deleted (value is seconds).
Return Value:
Status of the operation.
--*/
{ PRTM_NET_ADDRESS pDestAddr; PRTM_ROUTE_INFO pRouteInfo; RTM_NEXTHOP_INFO rniInfo; DWORD dwFlags; DWORD dwResult; HANDLE hNextHopHandle; PADAPTER_INFO pBinding;
// Initialize output before caling ops
if (ARGUMENT_PRESENT(phRtmRoute)) { *phRtmRoute = NULL; } pDestAddr = HeapAlloc( IPRouterHeap, 0, sizeof(RTM_NET_ADDRESS) );
if (pDestAddr == NULL) { Trace1( ERR, "AddRtmRoute : error allocating %d bytes" "for dest. address", RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) ); return ERROR_NOT_ENOUGH_MEMORY; }
pRouteInfo = HeapAlloc( IPRouterHeap, 0, RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) );
if (pRouteInfo == NULL) { Trace1( ERR, "AddRtmRoute : error allocating %d bytes" "for route info", RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) ); HeapFree(IPRouterHeap, 0, pDestAddr); return ERROR_NOT_ENOUGH_MEMORY; }
//
// Add a next hop if not already present
//
RTM_IPV4_MAKE_NET_ADDRESS(&rniInfo.NextHopAddress, pRtInfo->dwRtInfoNextHop, 32); rniInfo.InterfaceIndex = pRtInfo->dwRtInfoIfIndex; rniInfo.EntitySpecificInfo = (PVOID) (ULONG_PTR)dwNextHopMask; rniInfo.Flags = 0; rniInfo.RemoteNextHop = NULL;
hNextHopHandle = NULL; dwResult = RtmAddNextHop(hRtmHandle, &rniInfo, &hNextHopHandle, &dwFlags);
if (dwResult is NO_ERROR) { TraceRoute2( ROUTE, "Route to %d.%d.%d.%d/%d.%d.%d.%d", PRINT_IPADDR(pRtInfo->dwRtInfoDest), PRINT_IPADDR(pRtInfo->dwRtInfoMask) ); TraceRoute4( ROUTE, "Next Hop %d.%d.%d.%d/%d.%d.%d.%d, If 0x%x, handle is 0x%x", PRINT_IPADDR(pRtInfo->dwRtInfoNextHop), PRINT_IPADDR(dwNextHopMask), pRtInfo->dwRtInfoIfIndex, hNextHopHandle );
dwResult = ConvertRouteInfoToRtm(hRtmHandle, pRtInfo, hNextHopHandle, dwRouteFlags, pDestAddr, pRouteInfo); if (dwResult is NO_ERROR) { //
// If we are adding a non-dod route we should
// adjust the state of the route to match
// that of the interface it is being added on
//
if ((hRtmHandle == g_hNonDodRoute) || (hRtmHandle == g_hNetMgmtRoute)) { //
// Find the binding given the interface id
//
ENTER_READER(BINDING_LIST);
pBinding = GetInterfaceBinding(pRtInfo->dwRtInfoIfIndex);
if ((!pBinding) || (!pBinding->bBound)) { // Interface has been deleted meanwhile
// or is not bound at this point - quit
EXIT_LOCK(BINDING_LIST); return ERROR_INVALID_PARAMETER; } } //
// Convert TimeToLive from secs to millisecs
//
if (dwTimeToLive != INFINITE) { if (dwTimeToLive < (INFINITE / 1000)) { dwTimeToLive *= 1000; } else { dwTimeToLive = INFINITE; } }
dwFlags = 0;
//
// Add the new route using the RTMv2 API call
//
dwResult = RtmAddRouteToDest(hRtmHandle, phRtmRoute, pDestAddr, pRouteInfo, dwTimeToLive, NULL, 0, NULL, &dwFlags);
if ((hRtmHandle == g_hNonDodRoute) || (hRtmHandle == g_hNetMgmtRoute)) { EXIT_LOCK(BINDING_LIST); }
//
// check if route is 0/0 and route protocol is
// PROTO_IP_NETMGMT. If so mark for change notification
//
if ((pRtInfo->dwRtInfoDest is 0) and (pRtInfo->dwRtInfoMask is 0)) { RTM_DEST_INFO rdi; BOOL bMark; BOOL bRelDest = FALSE; do { TraceRoute2( ROUTE, "Checking dest %d.%d.%d.%d/%d for mark", PRINT_IPADDR(*(ULONG *)pDestAddr->AddrBits), PRINT_IPADDR(pDestAddr->NumBits) );
dwResult = RtmGetExactMatchDestination( g_hNetMgmtRoute, pDestAddr, RTM_THIS_PROTOCOL, RTM_VIEW_MASK_UCAST, &rdi );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "AddRtmRoute: error %d failed to get " "destination 0/0 for change notification", dwResult );
break; }
bRelDest = TRUE;
dwResult = RtmIsMarkedForChangeNotification( g_hNetMgmtRoute, g_hDefaultRouteNotification, rdi.DestHandle, &bMark );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "AddRtmRoute: error %d failed to check " "destination 0/0 for change notification", dwResult );
break; }
if (!bMark) { dwResult = RtmMarkDestForChangeNotification( g_hNetMgmtRoute, g_hDefaultRouteNotification, rdi.DestHandle, TRUE ); if (dwResult isnot NO_ERROR) { Trace1( ERR, "AddRtmRoute: error %d failed to nark " "destination 0/0 for change notification", dwResult ); break; } //
// Add route once more, to force marked dest
// change notifications to be issued for this
// change
//
dwFlags = 0; dwResult = RtmAddRouteToDest( hRtmHandle, phRtmRoute, pDestAddr, pRouteInfo, dwTimeToLive, NULL, 0, NULL, &dwFlags );
if (dwResult isnot NO_ERROR) { Trace1( ERR, "AddRtmRoute: error %d added route after " "marking destination", dwResult );
break; }
TraceRoute2( ROUTE, "Marked dest %d.%d.%d.%d/%d and added", PRINT_IPADDR(*(ULONG *)pDestAddr->AddrBits), PRINT_IPADDR(pDestAddr->NumBits) );
} } while (FALSE);
if (bRelDest) { RtmReleaseDestInfo( g_hNetMgmtRoute, &rdi ); } } } // Release the next hop handle obtained above
RtmReleaseNextHops(hRtmHandle, 1, &hNextHopHandle); }
HeapFree(IPRouterHeap, 0, pDestAddr); HeapFree(IPRouterHeap, 0, pRouteInfo); return dwResult; }
DWORD DeleteRtmRoute ( IN HANDLE hRtmHandle, IN PINTERFACE_ROUTE_INFO pRtInfo )
/*++
Routine Description:
Deletes an RTM route with the specified route information.
Arguments:
hRtmHandle - RTM registration handle used in RTM calls
pRtInfo -
Return Value:
Status of the operation.
--*/
{ PRTM_NET_ADDRESS pDestAddr; PRTM_ROUTE_INFO pRouteInfo; RTM_NEXTHOP_INFO rniInfo; DWORD dwFlags; DWORD dwResult; HANDLE hRouteHandle; HANDLE hNextHopHandle; pRouteInfo = HeapAlloc( IPRouterHeap, 0, RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) );
if (pRouteInfo == NULL) { Trace1( ERR, "DeleteRtmRoute : error allocating %d bytes" "for route info", RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) ); return ERROR_NOT_ENOUGH_MEMORY; }
pDestAddr = HeapAlloc( IPRouterHeap, 0, sizeof(RTM_NET_ADDRESS) );
if (pDestAddr == NULL) { Trace1( ERR, "AddRtmRoute : error allocating %d bytes" "for dest. address", RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute) ); HeapFree(IPRouterHeap, 0, pRouteInfo); return ERROR_NOT_ENOUGH_MEMORY; }
//
// Obtain a handle to the next hop in the route
//
RTM_IPV4_MAKE_NET_ADDRESS(&rniInfo.NextHopAddress, pRtInfo->dwRtInfoNextHop, 32); rniInfo.InterfaceIndex = pRtInfo->dwRtInfoIfIndex;
rniInfo.NextHopOwner = hRtmHandle; dwResult = RtmFindNextHop(hRtmHandle, &rniInfo, &hNextHopHandle, NULL);
if (dwResult isnot NO_ERROR) { HeapFree(IPRouterHeap, 0, pDestAddr); HeapFree(IPRouterHeap, 0, pRouteInfo); return dwResult; }
//
// We can get this route by matching the route's
// net addr, its owner and neighbour learnt from
//
ConvertRouteInfoToRtm(hRtmHandle, pRtInfo, hNextHopHandle, 0, pDestAddr, pRouteInfo);
dwResult = RtmGetExactMatchRoute(hRtmHandle, pDestAddr, RTM_MATCH_OWNER | RTM_MATCH_NEIGHBOUR, pRouteInfo, 0, 0, &hRouteHandle); if (dwResult is NO_ERROR) { //
// Delete the route found above using the handle
//
dwResult = RtmDeleteRouteToDest(hRtmHandle, hRouteHandle, &dwFlags);
if (dwResult isnot NO_ERROR) { // If delete successful, deref is automatic
RtmReleaseRoutes(hRtmHandle, 1, &hRouteHandle); }
// Release the route information obtained above
RtmReleaseRouteInfo(hRtmHandle, pRouteInfo); }
// Release the next hop handle obtained above
RtmReleaseNextHops(hRtmHandle, 1, &hNextHopHandle); HeapFree(IPRouterHeap, 0, pDestAddr); HeapFree(IPRouterHeap, 0, pRouteInfo);
return dwResult; }
DWORD ConvertRouteInfoToRtm( IN HANDLE hRtmHandle, IN PINTERFACE_ROUTE_INFO pRtInfo, IN HANDLE hNextHopHandle, IN DWORD dwRouteFlags, OUT PRTM_NET_ADDRESS pDestAddr, OUT PRTM_ROUTE_INFO pRouteInfo ) { DWORD dwAddrLen; // Fill the destination addr structure
RTM_IPV4_LEN_FROM_MASK(dwAddrLen, pRtInfo->dwRtInfoMask);
RTM_IPV4_MAKE_NET_ADDRESS(pDestAddr, pRtInfo->dwRtInfoDest, dwAddrLen);
// Fill in the route information now
ZeroMemory(pRouteInfo, sizeof(RTM_ROUTE_INFO));
pRouteInfo->RouteOwner = hRtmHandle; pRouteInfo->Neighbour = hNextHopHandle; pRouteInfo->PrefInfo.Metric = pRtInfo->dwRtInfoMetric1; pRouteInfo->PrefInfo.Preference = pRtInfo->dwRtInfoPreference; pRouteInfo->BelongsToViews = pRtInfo->dwRtInfoViewSet;
//
// BUG BUG BUG BUG :
// This is broken for future references
//
if(g_pLoopbackInterfaceCb && pRtInfo->dwRtInfoIfIndex is g_pLoopbackInterfaceCb->dwIfIndex) { pRouteInfo->BelongsToViews &= ~RTM_VIEW_MASK_MCAST; }
pRouteInfo->NextHopsList.NumNextHops = 1; pRouteInfo->NextHopsList.NextHops[0] = hNextHopHandle;
// an unsigned integer is converted to a shorter
// unsigned integer by truncating the high-order bits!
pRouteInfo->Flags1 = (UCHAR) dwRouteFlags; pRouteInfo->Flags = (USHORT) (dwRouteFlags >> 16); // Get the preference for this route
return ValidateRouteForProtocol(pRtInfo->dwRtInfoProto, pRouteInfo, pDestAddr);
// The following information is lost
//
// dwForwardMetric2,3
// dwForwardPolicy
// dwForwardType
// dwForwardAge
// dwForwardNextHopAS
}
VOID ConvertRtmToRouteInfo ( IN DWORD ownerProtocol, IN PRTM_NET_ADDRESS pDestAddr, IN PRTM_ROUTE_INFO pRouteInfo, IN PRTM_NEXTHOP_INFO pNextHop, OUT PINTERFACE_ROUTE_INFO pRtInfo ) { pRtInfo->dwRtInfoDest = *(ULONG *)pDestAddr->AddrBits; pRtInfo->dwRtInfoMask = RTM_IPV4_MASK_FROM_LEN(pDestAddr->NumBits);
pRtInfo->dwRtInfoIfIndex = pNextHop->InterfaceIndex; pRtInfo->dwRtInfoNextHop = *(ULONG *)pNextHop->NextHopAddress.AddrBits;
pRtInfo->dwRtInfoProto = ownerProtocol;
pRtInfo->dwRtInfoMetric1 = pRtInfo->dwRtInfoMetric2 = pRtInfo->dwRtInfoMetric3 = pRouteInfo->PrefInfo.Metric; pRtInfo->dwRtInfoPreference = pRouteInfo->PrefInfo.Preference; pRtInfo->dwRtInfoViewSet = pRouteInfo->BelongsToViews; pRtInfo->dwRtInfoPolicy = 0; pRtInfo->dwRtInfoType = 0; pRtInfo->dwRtInfoAge = 0; pRtInfo->dwRtInfoNextHopAS = 0;
return; }
PINTERFACE_ROUTE_INFO ConvertMibRouteToRouteInfo( IN PMIB_IPFORWARDROW pMibRow ) { PINTERFACE_ROUTE_INFO pRouteInfo = (PINTERFACE_ROUTE_INFO)pMibRow;
pMibRow->dwForwardMetric2 = 0; pMibRow->dwForwardMetric3 = 0; pMibRow->dwForwardMetric4 = 0; pMibRow->dwForwardMetric5 = 0;
// Make sure Metric1 isn't 0
if (pRouteInfo->dwRtInfoMetric1 is 0) { pRouteInfo->dwRtInfoMetric1 = 1; }
// By default put it in both views
pRouteInfo->dwRtInfoViewSet = RTM_VIEW_MASK_UCAST | RTM_VIEW_MASK_MCAST;
return pRouteInfo; }
PMIB_IPFORWARDROW ConvertRouteInfoToMibRoute( IN PINTERFACE_ROUTE_INFO pRouteInfo ) { PMIB_IPFORWARDROW pMibRoute = (PMIB_IPFORWARDROW) pRouteInfo;
pMibRoute->dwForwardMetric2 = pMibRoute->dwForwardMetric3 = pMibRoute->dwForwardMetric4 = pMibRoute->dwForwardMetric5 = IRE_METRIC_UNUSED;
pMibRoute->dwForwardAge = INFINITE; pMibRoute->dwForwardPolicy = 0; pMibRoute->dwForwardNextHopAS = 0; pMibRoute->dwForwardType = IRE_TYPE_INDIRECT;
return pMibRoute; }
VOID ConvertRouteNotifyOutputToRouteInfo( IN PIPRouteNotifyOutput pirno, OUT PINTERFACE_ROUTE_INFO pRtInfo ) {
ZeroMemory(pRtInfo, sizeof(INTERFACE_ROUTE_INFO));
pRtInfo->dwRtInfoDest = pirno->irno_dest; pRtInfo->dwRtInfoMask = pirno->irno_mask; pRtInfo->dwRtInfoIfIndex = pirno->irno_ifindex; pRtInfo->dwRtInfoNextHop = pirno->irno_nexthop;
pRtInfo->dwRtInfoProto = pirno->irno_proto;
pRtInfo->dwRtInfoMetric1 = pRtInfo->dwRtInfoMetric2 = pRtInfo->dwRtInfoMetric3 = pirno->irno_metric;
pRtInfo->dwRtInfoPreference = ComputeRouteMetric(pirno->irno_proto);
pRtInfo->dwRtInfoViewSet = RTM_VIEW_MASK_UCAST | RTM_VIEW_MASK_MCAST;
pRtInfo->dwRtInfoType = (pirno->irno_proto == PROTO_IP_LOCAL) ? MIB_IPROUTE_TYPE_DIRECT : 0;
pRtInfo->dwRtInfoAge = INFINITE; pRtInfo->dwRtInfoNextHopAS = 0; pRtInfo->dwRtInfoPolicy = 0;
return; }
DWORD BlockConvertRoutesToStatic ( IN HANDLE hRtmHandle, IN DWORD dwIfIndex, IN DWORD dwProtocolId ) { HANDLE hRtmEnum; RTM_ENTITY_INFO reiInfo; RTM_NET_ADDRESS rnaDest; PRTM_ROUTE_INFO pRouteInfo1; PRTM_ROUTE_INFO pRouteInfo2; RTM_NEXTHOP_INFO nhiInfo; RTM_NEXTHOP_HANDLE hNextHop; PHANDLE hRoutes; DWORD dwHandles; DWORD dwFlags; DWORD dwNumBytes; DWORD i, j, k; BOOL fDeleted; DWORD dwResult;
hRoutes = HeapAlloc( IPRouterHeap, 0, g_rtmProfile.MaxHandlesInEnum * sizeof(HANDLE) );
if (hRoutes == NULL) { Trace1( ERR, "BlockConvertRoutesToStatic : error allocating %d bytes" "for route handes", g_rtmProfile.MaxHandlesInEnum * sizeof(HANDLE) ); SetLastError(ERROR_NOT_ENOUGH_MEMORY); return ERROR_NOT_ENOUGH_MEMORY; }
dwNumBytes = RTM_SIZE_OF_ROUTE_INFO(g_rtmProfile.MaxNextHopsInRoute); pRouteInfo1 = HeapAlloc( IPRouterHeap, 0, dwNumBytes );
if (pRouteInfo1 == NULL) { Trace1( ERR, "BlockConvertRoutesToStatic : error allocating %d bytes" "for route info", dwNumBytes ); HeapFree(IPRouterHeap, 0, hRoutes); SetLastError(ERROR_NOT_ENOUGH_MEMORY); return ERROR_NOT_ENOUGH_MEMORY; }
pRouteInfo2 = HeapAlloc( IPRouterHeap, 0, dwNumBytes );
if (pRouteInfo2 == NULL) { Trace1( ERR, "BlockConvertRoutesToStatic : error allocating %d bytes" "for route info", dwNumBytes ); HeapFree(IPRouterHeap, 0, hRoutes); HeapFree(IPRouterHeap, 0, pRouteInfo1); SetLastError(ERROR_NOT_ENOUGH_MEMORY);
return ERROR_NOT_ENOUGH_MEMORY; }
//
// Enum all routes on the interface that we need
//
dwResult = RtmCreateRouteEnum(hRtmHandle, NULL, RTM_VIEW_MASK_ANY, RTM_ENUM_ALL_ROUTES, NULL, RTM_MATCH_INTERFACE, NULL, dwIfIndex, &hRtmEnum);
if(dwResult isnot NO_ERROR) { Trace2(ERR, "BlockConvertRoutesToStatic: Error %d creating handle for %d\n", dwResult, hRtmHandle); HeapFree(IPRouterHeap, 0, hRoutes); HeapFree(IPRouterHeap, 0, pRouteInfo1);
HeapFree(IPRouterHeap, 0, pRouteInfo2);
return dwResult; }
do { dwHandles = g_rtmProfile.MaxHandlesInEnum; dwResult = RtmGetEnumRoutes(hRtmHandle, hRtmEnum, &dwHandles, hRoutes);
for (i = 0; i < dwHandles; i++) { fDeleted = FALSE; // Get the route info from the handle
if (RtmGetRouteInfo(hRtmHandle, hRoutes[i], pRouteInfo1, &rnaDest) is NO_ERROR) { // Does this match the routing protocol we want ?
if ((RtmGetEntityInfo(hRtmHandle, pRouteInfo1->RouteOwner, &reiInfo) is NO_ERROR) && (reiInfo.EntityId.EntityProtocolId is dwProtocolId)) { //
// Add new static route with same information
//
CopyMemory(pRouteInfo2, pRouteInfo1, sizeof(RTM_ROUTE_INFO));
// Adjust the preference to confirm to protocol
pRouteInfo2->PrefInfo.Preference = ComputeRouteMetric(PROTO_IP_NT_AUTOSTATIC);
// Adjust the neighbour to corr to new protocol
if (pRouteInfo1->Neighbour) { // In case we cant get convert the neighbour
pRouteInfo2->Neighbour = NULL; if (RtmGetNextHopInfo(hRtmHandle, pRouteInfo1->Neighbour, &nhiInfo) is NO_ERROR) { // Add the same neigbour using new protocol
hNextHop = NULL;
if (RtmAddNextHop(hRtmHandle, &nhiInfo, &hNextHop, &dwFlags) is NO_ERROR) { pRouteInfo2->Neighbour = hNextHop; }
RtmReleaseNextHopInfo(hRtmHandle, &nhiInfo); } }
// Adjust the next hops to corr to new protocol
for (j = k = 0; j < pRouteInfo1->NextHopsList.NumNextHops; j++) { if (RtmGetNextHopInfo(hRtmHandle, pRouteInfo1->NextHopsList.NextHops[j], &nhiInfo) is NO_ERROR) { // Add the same nexthop using new protocol
hNextHop = NULL;
if (RtmAddNextHop(hRtmHandle, &nhiInfo, &hNextHop, &dwFlags) is NO_ERROR) { pRouteInfo2->NextHopsList.NextHops[k++] = hNextHop; }
RtmReleaseNextHopInfo(hRtmHandle, &nhiInfo); } }
pRouteInfo2->NextHopsList.NumNextHops = (USHORT) k;
// Add the new route with the next hop information
if (k > 0) { dwFlags = 0; if (RtmAddRouteToDest(hRtmHandle, NULL, &rnaDest, pRouteInfo2, INFINITE, NULL, 0, NULL, &dwFlags) is NO_ERROR) { // Route add is successful - delete old route
if (RtmDeleteRouteToDest(pRouteInfo1->RouteOwner, hRoutes[i], &dwFlags) is NO_ERROR) { fDeleted = TRUE; } }
RtmReleaseNextHops(hRtmHandle, k, pRouteInfo2->NextHopsList.NextHops); } }
RtmReleaseRouteInfo(hRtmHandle, pRouteInfo1); }
if (!fDeleted) { RtmReleaseRoutes(hRtmHandle, 1, &hRoutes[i]); } } } while (dwResult is NO_ERROR);
RtmDeleteEnumHandle(hRtmHandle, hRtmEnum);
HeapFree(IPRouterHeap, 0, hRoutes); HeapFree(IPRouterHeap, 0, pRouteInfo1);
HeapFree(IPRouterHeap, 0, pRouteInfo2);
return NO_ERROR; }
DWORD DeleteRtmRoutes ( IN HANDLE hRtmHandle, IN DWORD dwIfIndex, IN BOOL fDeleteAll ) { HANDLE hRtmEnum; PHANDLE hRoutes; DWORD dwHandles; DWORD dwFlags; DWORD i; DWORD dwResult;
hRoutes = HeapAlloc( IPRouterHeap, 0, g_rtmProfile.MaxHandlesInEnum * sizeof(HANDLE) );
if (hRoutes == NULL) { Trace1(ERR, "DeleteRtmRoutes: Error allocating %d bytes", g_rtmProfile.MaxHandlesInEnum * sizeof(HANDLE)); return ERROR_NOT_ENOUGH_MEMORY; }
dwFlags = fDeleteAll ? 0: RTM_MATCH_INTERFACE; dwResult = RtmCreateRouteEnum(hRtmHandle, NULL, RTM_VIEW_MASK_ANY, RTM_ENUM_OWN_ROUTES, NULL, dwFlags, NULL, dwIfIndex, &hRtmEnum);
if(dwResult isnot NO_ERROR) { Trace2(ERR, "DeleteRtmRoutes: Error %d creating handle for %d\n", dwResult, hRtmHandle); HeapFree(IPRouterHeap, 0, hRoutes); return dwResult; }
do { dwHandles = g_rtmProfile.MaxHandlesInEnum; dwResult = RtmGetEnumRoutes(hRtmHandle, hRtmEnum, &dwHandles, hRoutes);
for (i = 0; i < dwHandles; i++) { if (RtmDeleteRouteToDest(hRtmHandle, hRoutes[i], &dwFlags) isnot NO_ERROR) { // If delete is successful, this is automatic
RtmReleaseRoutes(hRtmHandle, 1, &hRoutes[i]); } } } while (dwResult is NO_ERROR);
RtmDeleteEnumHandle(hRtmHandle, hRtmEnum);
HeapFree(IPRouterHeap, 0, hRoutes); return NO_ERROR; }
DWORD DeleteRtmNexthops ( IN HANDLE hRtmHandle, IN DWORD dwIfIndex, IN BOOL fDeleteAll ) { PRTM_NEXTHOP_INFO pNexthop; PHANDLE hNexthops; HANDLE hRtmEnum; DWORD dwHandles; DWORD i; DWORD dwResult;
hNexthops = HeapAlloc( IPRouterHeap, 0, g_rtmProfile.MaxHandlesInEnum * sizeof(HANDLE) );
if (hNexthops == NULL) { Trace1(ERR, "DeleteRtmNextHops: Error allocating %d bytes", g_rtmProfile.MaxHandlesInEnum * sizeof(HANDLE)); return ERROR_NOT_ENOUGH_MEMORY; }
dwResult = RtmCreateNextHopEnum(hRtmHandle, 0, NULL, &hRtmEnum);
if(dwResult isnot NO_ERROR) { Trace2(ERR, "DeleteAllNexthops: Error %d creating handle for %d\n", dwResult, hRtmHandle); HeapFree(IPRouterHeap, 0, hNexthops); return dwResult; }
do { dwHandles = g_rtmProfile.MaxHandlesInEnum; dwResult = RtmGetEnumNextHops(hRtmHandle, hRtmEnum, &dwHandles, hNexthops);
for (i = 0; i < dwHandles; i++) { if (!fDeleteAll) { //
// Make sure that the interface matches
//
if ((RtmGetNextHopPointer(hRtmHandle, hNexthops[i], &pNexthop) isnot NO_ERROR) || (pNexthop->InterfaceIndex != dwIfIndex)) { RtmReleaseNextHops(hRtmHandle, 1, &hNexthops[i]); continue; } }
// We need to delete this next hop here
if (RtmDeleteNextHop(hRtmHandle, hNexthops[i], NULL) isnot NO_ERROR) { // If delete is successful, this is automatic
RtmReleaseNextHops(hRtmHandle, 1, &hNexthops[i]); } } } while (dwResult is NO_ERROR);
RtmDeleteEnumHandle(hRtmHandle, hRtmEnum);
HeapFree(IPRouterHeap, 0, hNexthops);
return NO_ERROR; }
|