mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 22:28:36 +00:00
273 lines
7.8 KiB
C
273 lines
7.8 KiB
C
/* Poor man's route
|
|
*
|
|
* Supported commands:
|
|
*
|
|
* "print"
|
|
* "add" target ["mask" mask] gw ["metric" metric]
|
|
* "delete" target gw
|
|
*
|
|
* Goals:
|
|
*
|
|
* Flexible, simple
|
|
*/
|
|
|
|
#define WIN32_NO_STATUS
|
|
#include <stdarg.h>
|
|
#include <windef.h>
|
|
#include <winbase.h>
|
|
#include <stdio.h>
|
|
#include <malloc.h>
|
|
#define _INC_WINDOWS
|
|
#include <winsock2.h>
|
|
#include <iphlpapi.h>
|
|
#include <tchar.h>
|
|
|
|
#define IPBUF 17
|
|
#define IN_ADDR_OF(x) *((struct in_addr *)&(x))
|
|
|
|
static int Usage()
|
|
{
|
|
_ftprintf( stderr,
|
|
_T("route usage:\n")
|
|
_T("route print\n")
|
|
_T(" prints the route table\n")
|
|
_T("route add <target> [mask <mask>] <gw> [metric <m>]\n")
|
|
_T(" adds a route\n")
|
|
_T("route delete <target> <gw>\n")
|
|
_T(" deletes a route\n") );
|
|
return 1;
|
|
}
|
|
|
|
static int PrintRoutes()
|
|
{
|
|
PMIB_IPFORWARDTABLE IpForwardTable = NULL;
|
|
PIP_ADAPTER_INFO pAdapterInfo = NULL;
|
|
ULONG Size = 0;
|
|
DWORD Error = 0;
|
|
ULONG adaptOutBufLen = sizeof(IP_ADAPTER_INFO);
|
|
TCHAR DefGate[16];
|
|
TCHAR Destination[IPBUF], Gateway[IPBUF], Netmask[IPBUF];
|
|
unsigned int i;
|
|
|
|
/* set required buffer size */
|
|
pAdapterInfo = (IP_ADAPTER_INFO *) malloc( adaptOutBufLen );
|
|
if (pAdapterInfo == NULL)
|
|
{
|
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto Error;
|
|
}
|
|
if (GetAdaptersInfo( pAdapterInfo, &adaptOutBufLen) == ERROR_BUFFER_OVERFLOW)
|
|
{
|
|
free (pAdapterInfo);
|
|
pAdapterInfo = (IP_ADAPTER_INFO *) malloc (adaptOutBufLen);
|
|
if (pAdapterInfo == NULL)
|
|
{
|
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto Error;
|
|
}
|
|
}
|
|
|
|
if( (GetIpForwardTable( NULL, &Size, TRUE )) == ERROR_INSUFFICIENT_BUFFER )
|
|
{
|
|
if (!(IpForwardTable = malloc( Size )))
|
|
{
|
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto Error;
|
|
}
|
|
}
|
|
|
|
if (((Error = GetAdaptersInfo(pAdapterInfo, &adaptOutBufLen)) == NO_ERROR) &&
|
|
((Error = GetIpForwardTable(IpForwardTable, &Size, TRUE)) == NO_ERROR))
|
|
{
|
|
_stprintf(DefGate,
|
|
#ifdef UNICODE
|
|
_T("%hs"),
|
|
#else
|
|
_T("%s"),
|
|
#endif
|
|
pAdapterInfo->GatewayList.IpAddress.String);
|
|
_tprintf(_T("===========================================================================\n"));
|
|
_tprintf(_T("Interface List\n"));
|
|
/* FIXME - sort by the index! */
|
|
while (pAdapterInfo)
|
|
{
|
|
#ifdef UNICODE
|
|
_tprintf(_T("0x%lu ........................... %hs\n"),
|
|
#else
|
|
_tprintf(_T("0x%lu ........................... %s\n"),
|
|
#endif
|
|
pAdapterInfo->Index, pAdapterInfo->Description);
|
|
pAdapterInfo = pAdapterInfo->Next;
|
|
}
|
|
_tprintf(_T("===========================================================================\n"));
|
|
|
|
_tprintf(_T("===========================================================================\n"));
|
|
_tprintf(_T("Active Routes:\n"));
|
|
_tprintf( _T("%-27s%-17s%-14s%-11s%-10s\n"),
|
|
_T("Network Destination"),
|
|
_T("Netmask"),
|
|
_T("Gateway"),
|
|
_T("Interface"),
|
|
_T("Metric") );
|
|
for( i = 0; i < IpForwardTable->dwNumEntries; i++ )
|
|
{
|
|
_stprintf( Destination,
|
|
#ifdef UNICODE
|
|
_T("%hs"),
|
|
#else
|
|
_T("%s"),
|
|
#endif
|
|
inet_ntoa( IN_ADDR_OF(IpForwardTable->table[i].dwForwardDest) ) );
|
|
_stprintf( Netmask,
|
|
#ifdef UNICODE
|
|
_T("%hs"),
|
|
#else
|
|
_T("%s"),
|
|
#endif
|
|
inet_ntoa( IN_ADDR_OF(IpForwardTable->table[i].dwForwardMask) ) );
|
|
_stprintf( Gateway,
|
|
#ifdef UNICODE
|
|
_T("%hs"),
|
|
#else
|
|
_T("%s"),
|
|
#endif
|
|
inet_ntoa( IN_ADDR_OF(IpForwardTable->table[i].dwForwardNextHop) ) );
|
|
|
|
_tprintf( _T("%17s%17s%17s%16ld%9ld\n"),
|
|
Destination,
|
|
Netmask,
|
|
Gateway,
|
|
IpForwardTable->table[i].dwForwardIfIndex,
|
|
IpForwardTable->table[i].dwForwardMetric1 );
|
|
}
|
|
_tprintf(_T("Default Gateway:%18s\n"), DefGate);
|
|
_tprintf(_T("===========================================================================\n"));
|
|
_tprintf(_T("Persistent Routes:\n"));
|
|
|
|
free(IpForwardTable);
|
|
free(pAdapterInfo);
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
Error:
|
|
if (pAdapterInfo) free(pAdapterInfo);
|
|
if (IpForwardTable) free(IpForwardTable);
|
|
_ftprintf( stderr, _T("Route enumerate failed\n") );
|
|
return Error;
|
|
}
|
|
}
|
|
|
|
static int convert_add_cmd_line( PMIB_IPFORWARDROW RowToAdd,
|
|
int argc, TCHAR **argv ) {
|
|
int i;
|
|
#ifdef UNICODE
|
|
char addr[16];
|
|
#endif
|
|
|
|
if( argc > 1 )
|
|
{
|
|
#ifdef UNICODE
|
|
sprintf( addr, "%ls", argv[0] );
|
|
RowToAdd->dwForwardDest = inet_addr( addr );
|
|
#else
|
|
RowToAdd->dwForwardDest = inet_addr( argv[0] );
|
|
#endif
|
|
}
|
|
else
|
|
return FALSE;
|
|
for( i = 1; i < argc; i++ )
|
|
{
|
|
if( !_tcscmp( argv[i], _T("mask") ) )
|
|
{
|
|
i++; if( i >= argc ) return FALSE;
|
|
#ifdef UNICODE
|
|
sprintf( addr, "%ls", argv[i] );
|
|
RowToAdd->dwForwardMask = inet_addr( addr );
|
|
#else
|
|
RowToAdd->dwForwardMask = inet_addr( argv[i] );
|
|
#endif
|
|
}
|
|
else if( !_tcscmp( argv[i], _T("metric") ) )
|
|
{
|
|
i++;
|
|
if( i >= argc )
|
|
return FALSE;
|
|
RowToAdd->dwForwardMetric1 = _ttoi( argv[i] );
|
|
}
|
|
else
|
|
{
|
|
#ifdef UNICODE
|
|
sprintf( addr, "%ls", argv[i] );
|
|
RowToAdd->dwForwardNextHop = inet_addr( addr );
|
|
#else
|
|
RowToAdd->dwForwardNextHop = inet_addr( argv[i] );
|
|
#endif
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static int add_route( int argc, TCHAR **argv ) {
|
|
MIB_IPFORWARDROW RowToAdd = { 0 };
|
|
DWORD Error;
|
|
|
|
if( argc < 2 || !convert_add_cmd_line( &RowToAdd, argc, argv ) )
|
|
{
|
|
_ftprintf( stderr,
|
|
_T("route add usage:\n")
|
|
_T("route add <target> [mask <mask>] <gw> [metric <m>]\n")
|
|
_T(" Adds a route to the IP route table.\n")
|
|
_T(" <target> is the network or host to add a route to.\n")
|
|
_T(" <mask> is the netmask to use (autodetected if unspecified)\n")
|
|
_T(" <gw> is the gateway to use to access the network\n")
|
|
_T(" <m> is the metric to use (lower is preferred)\n") );
|
|
return 1;
|
|
}
|
|
|
|
if( (Error = CreateIpForwardEntry( &RowToAdd )) == ERROR_SUCCESS )
|
|
return 0;
|
|
|
|
_ftprintf( stderr, _T("Route addition failed\n") );
|
|
return Error;
|
|
}
|
|
|
|
static int del_route( int argc, TCHAR **argv )
|
|
{
|
|
MIB_IPFORWARDROW RowToDel = { 0 };
|
|
DWORD Error;
|
|
|
|
if( argc < 2 || !convert_add_cmd_line( &RowToDel, argc, argv ) )
|
|
{
|
|
_ftprintf( stderr,
|
|
_T("route delete usage:\n")
|
|
_T("route delete <target> <gw>\n")
|
|
_T(" Removes a route from the IP route table.\n")
|
|
_T(" <target> is the network or host to add a route to.\n")
|
|
_T(" <gw> is the gateway to remove the route from.\n") );
|
|
return 1;
|
|
}
|
|
|
|
if( (Error = DeleteIpForwardEntry( &RowToDel )) == ERROR_SUCCESS )
|
|
return 0;
|
|
|
|
_ftprintf( stderr, _T("Route addition failed\n") );
|
|
return Error;
|
|
}
|
|
|
|
int _tmain( int argc, TCHAR **argv )
|
|
{
|
|
if( argc < 2 )
|
|
return Usage();
|
|
else if ( !_tcscmp( argv[1], _T("print") ) )
|
|
return PrintRoutes();
|
|
else if( !_tcscmp( argv[1], _T("add") ) )
|
|
return add_route( argc-2, argv+2 );
|
|
else if( !_tcscmp( argv[1], _T("delete") ) )
|
|
return del_route( argc-2, argv+2 );
|
|
else
|
|
return Usage();
|
|
}
|