|
|
#include "precomp.h"
#pragma hdrstop
#define DEFAULT_VIEW_MASK RTM_VIEW_MASK_UCAST|RTM_VIEW_MASK_MCAST // both
#define DEFAULT_VIEW_ID RTM_VIEW_ID_UCAST
#define DEFAULT_ADDR 0 // 0.0.0.0
#define DEFAULT_MASK 0 // 0.0.0.0
#define DEFAULT_PROTO RTM_BEST_PROTOCOL
DWORDHandleIpShowRtmDestinations( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone ){ TAG_TYPE pttTags[] = {{TOKEN_OPERATOR, FALSE,FALSE}, {TOKEN_DEST, FALSE,FALSE}, {TOKEN_MASK, FALSE,FALSE}, {TOKEN_VIEW, FALSE,FALSE}, {TOKEN_PROTOCOL, FALSE,FALSE}}; DWORD pdwTagType[sizeof(pttTags)/sizeof(TAG_TYPE)]; DWORD dwErr, i, dwOperator = DEST_LONGER; DWORD dwViewMask = DEFAULT_VIEW_MASK; DWORD dwProtocol = DEFAULT_PROTO; IPV4_ADDRESS ipMask = DEFAULT_MASK; IPV4_ADDRESS ipAddress = DEFAULT_ADDR; MIB_OPAQUE_QUERY QueryBuff[3]; // more than enough
MIB_OPAQUE_QUERY *pQuery = QueryBuff; DWORD dwOutEntrySize, dwCount; PMIB_OPAQUE_INFO pRpcInfo; PMIB_IPDESTTABLE lprpcTable; WCHAR wcszBuffer[80], wcszNHop[80]; WCHAR wcszName[MAX_INTERFACE_NAME_LEN+1]; WCHAR wszViews[3];
//
// We can show non persistent info only if router is running
//
CHECK_ROUTER_RUNNING();
// Do generic processing
dwErr = PreHandleCommand( ppwcArguments, dwCurrentIndex, dwArgCount, pttTags, sizeof(pttTags)/sizeof(TAG_TYPE), 0, sizeof(pttTags)/sizeof(TAG_TYPE), pdwTagType );
if (dwErr) { return dwErr; }
dwArgCount -= dwCurrentIndex;
for (i=0; i<dwArgCount; i++) { switch(pdwTagType[i]) { case 0: // OPERATOR
{ TOKEN_VALUE rgEnums[] ={{ TOKEN_VALUE_MATCHING, DEST_MATCHING}, { TOKEN_VALUE_LONGER, DEST_LONGER }, { TOKEN_VALUE_SHORTER, DEST_SHORTER}};
dwErr = MatchEnumTag( g_hModule, ppwcArguments[i + dwCurrentIndex], sizeof(rgEnums)/sizeof(TOKEN_VALUE), rgEnums, &dwOperator);
if (dwErr isnot NO_ERROR) { DispTokenErrMsg( g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex] );
return ERROR_INVALID_PARAMETER; }
break; }
case 1: // ADDR
{ dwErr = GetIpPrefix( ppwcArguments[i+dwCurrentIndex], &ipAddress, &ipMask );
if (dwErr is ERROR_INVALID_PARAMETER) { DisplayMessage( g_hModule, MSG_IP_BAD_IP_ADDR, ppwcArguments[i + dwCurrentIndex]);
DispTokenErrMsg(g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex]); i = dwArgCount; break; } break; }
case 2: // MASK
{ dwErr = GetIpMask( ppwcArguments[i+dwCurrentIndex], &ipMask );
if (dwErr is ERROR_INVALID_PARAMETER) { DisplayMessage( g_hModule, MSG_IP_BAD_IP_ADDR, ppwcArguments[i + dwCurrentIndex]);
DispTokenErrMsg(g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex]); i = dwArgCount; break; } break; }
case 3: // VIEW
{ TOKEN_VALUE rgMaskEnums[] = { { TOKEN_VALUE_UNICAST, RTM_VIEW_MASK_UCAST }, { TOKEN_VALUE_MULTICAST, RTM_VIEW_MASK_MCAST }, { TOKEN_VALUE_BOTH, RTM_VIEW_MASK_UCAST |RTM_VIEW_MASK_MCAST } };
dwErr = MatchEnumTag( g_hModule, ppwcArguments[i + dwCurrentIndex], sizeof(rgMaskEnums)/sizeof(TOKEN_VALUE), rgMaskEnums, &dwViewMask);
if (dwErr isnot NO_ERROR) { DispTokenErrMsg( g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex] );
i = dwArgCount; return ERROR_INVALID_PARAMETER; }
break; }
case 4: // PROTO
{ dwProtocol = MatchRoutingProtoTag(ppwcArguments[i + dwCurrentIndex]); break; } } }
if (dwErr isnot NO_ERROR) { return dwErr; }
// Compose MIB query
pQuery->dwVarId = dwOperator; pQuery->rgdwVarIndex[0] = ipAddress; pQuery->rgdwVarIndex[1] = ipMask; pQuery->rgdwVarIndex[2] = dwViewMask; pQuery->rgdwVarIndex[3] = dwProtocol;
dwErr = MibGet( PID_IP, IPRTRMGR_PID, (PVOID) pQuery, sizeof(MIB_OPAQUE_QUERY) + 3*sizeof(DWORD), (PVOID *) &pRpcInfo, &dwOutEntrySize );
if ( dwErr isnot NO_ERROR ) { DisplayMessage(g_hModule, MSG_IP_DIM_ERROR, dwErr ); return dwErr; }
if ( pRpcInfo is NULL ) { DisplayMessage(g_hModule, MSG_IP_NO_ENTRIES ); return dwErr; }
// Display info
lprpcTable = (PMIB_IPDESTTABLE)(pRpcInfo->rgbyData); dwCount = lprpcTable->dwNumEntries;
DisplayMessage( g_hModule, MSG_RTR_ROUTE_HDR );
for (i=0; i<dwCount; i++) { DWORD BufLen = sizeof(wcszName); MakePrefixStringW( wcszBuffer, lprpcTable->table[i].dwForwardDest, lprpcTable->table[i].dwForwardMask );
if (IpmontrGetFriendlyNameFromIfIndex( g_hMIBServer, lprpcTable->table[i].dwForwardIfIndex, wcszName, BufLen ) != NO_ERROR) { //
// If we do not have a name for this index, display index
//
swprintf( wcszName, L"0x%x", lprpcTable->table[i].dwForwardIfIndex ); }
MakeAddressStringW( wcszNHop, lprpcTable->table[i].dwForwardNextHop );
wszViews[0] = (lprpcTable->table[i].dwForwardViewSet & RTM_VIEW_MASK_UCAST)? 'U':' '; wszViews[1] = (lprpcTable->table[i].dwForwardViewSet & RTM_VIEW_MASK_MCAST)? 'M':' '; wszViews[2] = '\0';
DisplayMessage( g_hModule, MSG_RTR_ROUTE_INFO, wcszBuffer, GetProtoProtoString( PROTO_TYPE_UCAST, 0, lprpcTable->table[i].dwForwardProto ), lprpcTable->table[i].dwForwardPreference, lprpcTable->table[i].dwForwardMetric1, wcszNHop, wszViews, wcszName ); }
MprAdminMIBBufferFree( (PVOID) pRpcInfo );
return dwErr;}
DWORDHandleIpShowRtmRoutes( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone ){ TAG_TYPE pttTags[] = {{TOKEN_OPERATOR, FALSE,FALSE}, {TOKEN_DEST, FALSE,FALSE}, {TOKEN_MASK, FALSE,FALSE}, {TOKEN_VIEW, FALSE,FALSE}, {TOKEN_PROTOCOL, FALSE,FALSE}}; DWORD pdwTagType[sizeof(pttTags)/sizeof(TAG_TYPE)]; DWORD dwErr, i, dwOperator = ROUTE_LONGER; DWORD dwViewMask = DEFAULT_VIEW_MASK; DWORD dwProtocol = DEFAULT_PROTO; IPV4_ADDRESS ipMask = DEFAULT_MASK; IPV4_ADDRESS ipAddress = DEFAULT_ADDR; MIB_OPAQUE_QUERY QueryBuff[3]; // more than enough
MIB_OPAQUE_QUERY *pQuery = QueryBuff; DWORD dwOutEntrySize, dwCount; PMIB_OPAQUE_INFO pRpcInfo; PMIB_IPDESTTABLE lprpcTable; WCHAR wcszBuffer[80], wcszNHop[80]; WCHAR wcszName[MAX_INTERFACE_NAME_LEN+1]; WCHAR wszViews[3];
//
// We can show non persistent info only if router is running
//
CHECK_ROUTER_RUNNING();
// Do generic processing
dwErr = PreHandleCommand( ppwcArguments, dwCurrentIndex, dwArgCount, pttTags, sizeof(pttTags)/sizeof(TAG_TYPE), 0, sizeof(pttTags)/sizeof(TAG_TYPE), pdwTagType );
if (dwErr) { return dwErr; }
dwArgCount -= dwCurrentIndex;
for (i=0; i<dwArgCount; i++) { switch(pdwTagType[i]) { case 0: // OPERATOR
{ TOKEN_VALUE rgEnums[] ={{ TOKEN_VALUE_MATCHING,ROUTE_MATCHING}, { TOKEN_VALUE_LONGER, ROUTE_LONGER }, { TOKEN_VALUE_SHORTER, ROUTE_SHORTER}};
dwErr = MatchEnumTag( g_hModule, ppwcArguments[i + dwCurrentIndex], sizeof(rgEnums)/sizeof(TOKEN_VALUE), rgEnums, &dwOperator);
if (dwErr isnot NO_ERROR) { DispTokenErrMsg( g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex] );
return ERROR_INVALID_PARAMETER; }
break; }
case 1: // ADDR
{ dwErr = GetIpPrefix( ppwcArguments[i+dwCurrentIndex], &ipAddress, &ipMask );
if (dwErr is ERROR_INVALID_PARAMETER) { DisplayMessage( g_hModule, MSG_IP_BAD_IP_ADDR, ppwcArguments[i + dwCurrentIndex]);
DispTokenErrMsg(g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex]); i = dwArgCount; break; } break; }
case 2: // MASK
{ dwErr = GetIpMask( ppwcArguments[i+dwCurrentIndex], &ipMask );
if (dwErr is ERROR_INVALID_PARAMETER) { DisplayMessage( g_hModule, MSG_IP_BAD_IP_ADDR, ppwcArguments[i + dwCurrentIndex]);
DispTokenErrMsg(g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex]); i = dwArgCount; break; } break; }
case 3: // VIEW
{ TOKEN_VALUE rgMaskEnums[] = { { TOKEN_VALUE_UNICAST, RTM_VIEW_MASK_UCAST }, { TOKEN_VALUE_MULTICAST, RTM_VIEW_MASK_MCAST }, { TOKEN_VALUE_BOTH, RTM_VIEW_MASK_UCAST |RTM_VIEW_MASK_MCAST } };
dwErr = MatchEnumTag( g_hModule, ppwcArguments[i + dwCurrentIndex], sizeof(rgMaskEnums)/sizeof(TOKEN_VALUE), rgMaskEnums, &dwViewMask);
if (dwErr isnot NO_ERROR) { DispTokenErrMsg( g_hModule, MSG_IP_BAD_OPTION_VALUE, pttTags[pdwTagType[i]].pwszTag, ppwcArguments[i + dwCurrentIndex] );
i = dwArgCount; return ERROR_INVALID_PARAMETER; }
break; }
case 4: // PROTO
{ dwProtocol = MatchRoutingProtoTag(ppwcArguments[i + dwCurrentIndex]);
break; } } }
if (dwErr isnot NO_ERROR) { return dwErr; }
// Compose MIB query
pQuery->dwVarId = dwOperator; pQuery->rgdwVarIndex[0] = ipAddress; pQuery->rgdwVarIndex[1] = ipMask; pQuery->rgdwVarIndex[2] = dwViewMask; pQuery->rgdwVarIndex[3] = dwProtocol;
dwErr = MibGet( PID_IP, IPRTRMGR_PID, (PVOID) pQuery, sizeof(MIB_OPAQUE_QUERY) + 3*sizeof(DWORD), (PVOID *) &pRpcInfo, &dwOutEntrySize );
if ( dwErr isnot NO_ERROR ) { DisplayMessage(g_hModule, MSG_IP_DIM_ERROR, dwErr ); return dwErr; }
if ( pRpcInfo is NULL ) { DisplayMessage(g_hModule, MSG_IP_NO_ENTRIES ); return dwErr; }
// Display info
lprpcTable = (PMIB_IPDESTTABLE)(pRpcInfo->rgbyData); dwCount = lprpcTable->dwNumEntries;
DisplayMessage( g_hModule, MSG_RTR_ROUTE_HDR );
for (i=0; i<dwCount; i++) { DWORD BufLen = sizeof(wcszName); MakePrefixStringW( wcszBuffer, lprpcTable->table[i].dwForwardDest, lprpcTable->table[i].dwForwardMask );
if (IpmontrGetFriendlyNameFromIfIndex( g_hMIBServer, lprpcTable->table[i].dwForwardIfIndex, wcszName, BufLen ) != NO_ERROR) { //
// If we do not have a name for this index, display index
//
swprintf( wcszName, L"0x%x", lprpcTable->table[i].dwForwardIfIndex ); }
MakeAddressStringW( wcszNHop, lprpcTable->table[i].dwForwardNextHop );
wszViews[0] = (lprpcTable->table[i].dwForwardViewSet & RTM_VIEW_MASK_UCAST)? 'U':' '; wszViews[1] = (lprpcTable->table[i].dwForwardViewSet & RTM_VIEW_MASK_MCAST)? 'M':' '; wszViews[2] = '\0';
DisplayMessage( g_hModule, MSG_RTR_ROUTE_INFO, wcszBuffer, GetProtoProtoString( PROTO_TYPE_UCAST, 0, lprpcTable->table[i].dwForwardProto ), lprpcTable->table[i].dwForwardPreference, lprpcTable->table[i].dwForwardMetric1, wcszNHop, wszViews, wcszName ); }
MprAdminMIBBufferFree( (PVOID) pRpcInfo );
return dwErr;}
|
