TCP/IP网络协议仿真教学系统（通用版）课件

/* * 文件: rtable.cpp * 描述： 利用IP助手管理路由表示例程序，制台应用程序 * 时间： 2006.3.1 */ //说明:ip助手管理路由表 #include "stdafx.h" #include <WinSock2.h> #include <Windows.h> #include <IPHlpApi.h> #include <iostream> #include <iomanip> #include <conio.h> #pragma comment( lib, "Iphlpapi.lib" ) #pragma comment( lib, "Ws2_32.lib" ) using namespace std ; //************************************ //void showInformation( void ) ; void Usage( void ); //************************************* void showRouteTable( void ) ; void deleteRouteTableItem( _TCHAR* index ) ; void addRouteTableItem( _TCHAR* tcDest , _TCHAR* tcMask , _TCHAR* tcGateway , _TCHAR* tcInter , _TCHAR* tcMetric ) ; int _tmain( int argc, _TCHAR* argv[ ] ) { WSAData wsadata ; if ( WSAStartup( MAKEWORD( 1, 1 ) , &wsadata ) ) { cout << "socket error!" << endl ; return 0 ; } if ( argc == 2 || argc == 3 || argc == 7 ) { for( int i = 1 ; i < argc ; i++ ) { if ( argv[ i ][ 0 ] == '-' || argv[i][0] == '/' ) { switch( argv[ i ][ 1 ] ) { case 's': showRouteTable( ) ; break ; case 'd': deleteRouteTableItem( &argv[ ++i ][ 0 ] ) ; break ; case 'a': if ( argc == 7 ) { addRouteTableItem( &argv[ i+1 ][ 0 ] , &argv[ i+2 ][ 0 ] , &argv[ i+3 ][ 0 ] , &argv[ i+4 ][ 0 ] , &argv[ i+5 ][ 0 ] ) ; i += 5 ; } else { cout << "command error!" << endl ; Usage( ) ; return 0 ; } break ; case 'h': case '?': Usage( ) ; break ; default: cout << "command error!" << endl ; Usage( ) ; return 0 ; } } else { cout << "command error!" << endl ; Usage( ) ; return 0 ; } } } else { cout << "command error!" << endl ; Usage( ) ; return 0 ; } WSACleanup( ) ; return 0; } void Usage( void ) //********************************** { cout << "对本机的路由表进行显示,删除和添加条目操作." << endl ; cout << "rtable的命令格式是:" << endl ; cout << endl ; cout << "rtable -h" << endl ; cout << "rtable -s" << endl ; cout << "rtable -d index" << endl ; cout << "rtable -a Dest Mask Gateway Inter Metric" << endl ; cout << endl ; cout << setw( 5 ) << " -h" << "\t\t\t\t\t显示帮助." << endl; cout << setw( 5 ) << " -s" << "\t\t\t\t\t显示路由表条目." << endl; cout << setw( 5 ) << " -d" << " index" << "\t\t\t\t删除路由表条目." << endl; cout << setw( 5 ) << " -a" << " Dest Mask Gateway Inter Metric" << "\t添加路由表条目." << endl ; cout << endl; cout << "例如:" << endl ; cout << " > rtable -h" << "\t\t\t\t\t\t\t" << "...显示帮助" << endl; cout << " > rtable -s" << "\t\t\t\t\t\t\t" << "...显示路由表" << endl; cout << " > rtable -d 2" << "\t\t\t\t\t\t" << "...删除第2条目" << endl; cout << " > rtable -a 192.168.0.12 255.255.255.255 172.16.0.4 1 1" << "\t...添加条目" << endl ; } void showRouteTable( void ) { //Ip地址表： PMIB_IPADDRTABLE pIpAddrTable; pIpAddrTable = (PMIB_IPADDRTABLE) GlobalAlloc( GPTR , sizeof(MIB_IPADDRTABLE) ); DWORD dwIpAddrLen = sizeof(MIB_IPADDRTABLE); if( GetIpAddrTable( pIpAddrTable , &dwIpAddrLen , TRUE ) == ERROR_INSUFFICIENT_BUFFER ) { GlobalFree( pIpAddrTable ) ; pIpAddrTable = ( PMIB_IPADDRTABLE ) GlobalAlloc( GPTR , dwIpAddrLen ); } if ( GetIpAddrTable( pIpAddrTable , &dwIpAddrLen , TRUE ) != NO_ERROR ) { cout << "show error!" << endl; GlobalFree( pIpAddrTable ); return ; } PMIB_IPFORWARDTABLE pIpForwardTable ; pIpForwardTable = ( PMIB_IPFORWARDTABLE ) GlobalAlloc( GPTR , sizeof( MIB_IPFORWARDTABLE ) ) ; DWORD dwSize = sizeof( MIB_IPFORWARDTABLE ) ; if( GetIpForwardTable( pIpForwardTable , &dwSize , TRUE ) == ERROR_INSUFFICIENT_BUFFER ) { GlobalFree( pIpForwardTable ) ; pIpForwardTable = ( PMIB_IPFORWARDTABLE ) GlobalAlloc( GPTR , dwSize ) ; } string strIp ,strMask , strGateway , strInterface , strMetric ; if( GetIpForwardTable( pIpForwardTable , &dwSize , TRUE ) == NO_ERROR ) { cout << "\n" << setw(2) << "no" << setw(16) << "Destination" << setw(16) << "mask" << setw(16) << "gateway" << setw(16) << "interface" << setw(8) << "metric" << "\n" ; for( int i = 0 ; i < pIpForwardTable->dwNumEntries ; i++ ) { in_addr ip ; ip.S_un.S_addr = pIpForwardTable->table[ i ].dwForwardDest ; strIp = inet_ntoa( ip ) ; in_addr Maskip ; Maskip.S_un.S_addr = pIpForwardTable->table[ i ].dwForwardMask ; strMask = inet_ntoa( Maskip ) ; in_addr gatewayip ; gatewayip.S_un.S_addr = pIpForwardTable->table[ i ].dwForwardNextHop ; strGateway = inet_ntoa( gatewayip ) ; for ( int j = 0 ; j < pIpAddrTable->dwNumEntries ; j++ ) { if ( pIpForwardTable->table[ i ].dwForwardIfIndex == pIpAddrTable->table[ j ].dwIndex ) { if ( pIpAddrTable->table[ j ].dwAddr != 0 ) { in_addr interfaceip ; interfaceip.S_un.S_addr = pIpAddrTable->table[ j ].dwAddr; strInterface = inet_ntoa( interfaceip ) ; } else { char tmp[ 20 ] ; ltoa( pIpForwardTable->table[i].dwForwardIfIndex , tmp , 20 ) ; strInterface = tmp ; } } } cout << setw( 2 ) << i << setw( 16 ) << strIp.c_str( ) << setw( 16 ) << strMask.c_str( ) << setw( 16 ) << strGateway.c_str( ) << setw( 16 ) << strInterface.c_str( ) << setw( 8 ) << pIpForwardTable->table[i].dwForwardMetric1 << "\n" ; } } else { cout << "show error!" << endl ; GlobalFree( pIpForwardTable ) ; GlobalFree( pIpAddrTable ) ; return ; } } void deleteRouteTableItem( _TCHAR* index ) { int nindex = atoi( index ) ; PMIB_IPFORWARDTABLE pIpForwardTable ; DWORD dwSize = 0 ; DWORD dwRetVal = 0 ; pIpForwardTable = (MIB_IPFORWARDTABLE*) GlobalAlloc( GPTR , sizeof(MIB_IPFORWARDTABLE) ) ; if ( GetIpForwardTable( pIpForwardTable, &dwSize, 0 ) == ERROR_INSUFFICIENT_BUFFER ) { GlobalFree( pIpForwardTable ); pIpForwardTable = (MIB_IPFORWARDTABLE*) GlobalAlloc( GPTR , dwSize) ; } if ((dwRetVal = GetIpForwardTable(pIpForwardTable, &dwSize, 0)) == NO_ERROR) { if ( nindex < 0 || nindex >= pIpForwardTable->dwNumEntries ) { cout << "\nIndex error!\n" ; GlobalFree( pIpForwardTable ) ; return ; } if ( DeleteIpForwardEntry( &pIpForwardTable->table[nindex]) != NO_ERROR) cout << "\nDelete route table item fail!\n" ; } GlobalFree( pIpForwardTable ) ; } void addRouteTableItem( _TCHAR* tcDest , _TCHAR* tcMask , _TCHAR* tcGateway , _TCHAR* tcInter , _TCHAR* tcMetric ) { PMIB_IPFORWARDROW pRoute ; DWORD nSize = sizeof(MIB_IPFORWARDROW) ; pRoute = (PMIB_IPFORWARDROW)GlobalAlloc( GPTR , nSize ) ; pRoute->dwForwardDest = inet_addr(tcDest) ; pRoute->dwForwardIfIndex = atol(tcInter) ; pRoute->dwForwardMask = inet_addr(tcMask) ; pRoute->dwForwardMetric1 = atol(tcMetric) ; pRoute->dwForwardNextHop = inet_addr(tcGateway) ; pRoute->dwForwardAge = 10000 ; pRoute->dwForwardMetric2 = 1 ; pRoute->dwForwardMetric3 = 1 ; pRoute->dwForwardMetric4 = 5 ; pRoute->dwForwardMetric5 = 3 ; pRoute->dwForwardNextHopAS = 0 ; pRoute->dwForwardPolicy = 0 ; pRoute->dwForwardProto = 3 ; pRoute->dwForwardType = 0 ; if ( CreateIpForwardEntry( pRoute ) != NO_E