test-suite-mixmode.tcl

ns gpsr路由协议 在ns２平台下实现的 对大家很有好处

# This test suite is for validating the session level simulation support
# in ns.
#
# To run all tests:  test-all-mixmode
#
# To run individual tests:
# ns test-suite-mixmode.tcl Mixmode1
# ns test-suite-mixmode.tcl Mixmode2
# ...
#
# To view a list of available tests to run with this script:
# ns test-suite-mixmode.tcl
#

Class TestSuite

TestSuite instproc init {} {
	$self instvar ns_ net_ defNet_ test_ topo_ node_ testName_
	set ns_ [new SessionSim]
	#$ns_ use-scheduler List
        $ns_ multicast

	#$ns_ namtrace-all [open all.tr w]
	puts "tracing"
	if {$net_ == ""} {
		set net_ $defNet_
	}
	if ![Topology/$defNet_ info subclass Topology/$net_] {
		global argv0
		puts "$argv0: cannot run test $test_ over topology $net_"
		exit 1
	}

	set topo_ [new Topology/$net_ $ns_]
	foreach i [$topo_ array names node_] {
		# This would be cool, but lets try to be compatible
		# with test-suite.tcl as far as possible.
		#
		# $self instvar $i
		# set $i [$topo_ node? $i]
		#
		set node_($i) [$topo_ node? $i]
	}

	if {$net_ == $defNet_} {
		set testName_ "$test_"
	} else {
		set testName_ "$test_:$net_"
	}
}

TestSuite instproc finish args {
	$self instvar ns_
	
	$ns_ flush-trace
	puts "\t#pkt\t#pkt"
	puts "rcvr\trcvd\tlost"
	set i 0
	foreach index $args {
		puts "$i\t[$index set npkts_]\t[$index set nlost_]"
		incr i
	}
#	exec awk -f ../nam-demo/nstonam.awk all.tr > [append file \.tr]
#	puts "running nam ..."
#	exec nam $file &
	exit 0
}

proc usage {} {
	global argv0
	puts stderr "usage: ns $argv0 <tests> \[<topologies>\]"
	puts stderr "Valid tests are:\t[get-subclasses TestSuite Test/]"
	puts stderr "Valid Topologies are:\t[get-subclasses SkelTopology Topology/]"
	exit 1
}

proc isProc? {cls prc} {
	if [catch "Object info subclass $cls/$prc" r] {
		global argv0
		puts stderr "$argv0: no such $cls: $prc"
		usage
	}
}

proc get-subclasses {cls pfx} {
	set ret ""
	set l [string length $pfx]

	set c $cls
	while {[llength $c] > 0} {
		set t [lindex $c 0]
		set c [lrange $c 1 end]
		if [string match ${pfx}* $t] {
			lappend ret [string range $t $l end]
		}
		eval lappend c [$t info subclass]
	}
	set ret
}

TestSuite proc runTest {} {
	global argc argv

	switch $argc {
		1 {
			set test $argv
			isProc? Test $test

			set topo ""
		}
		2 {
			set test [lindex $argv 0]
			isProc? Test $test

			set topo [lindex $argv 1]
			isProc? Topology $topo
		}
		default {
			usage
		}
	}
	set t [new Test/$test $topo]
	$t run
}

# Skeleton topology base class
Class SkelTopology

SkelTopology instproc init {} {
    $self next
}

SkelTopology instproc node? n {
    $self instvar node_
    if [info exists node_($n)] {
	set ret $node_($n)
    } else {
	set ret ""
    }
    set ret
}

SkelTopology instproc add-fallback-links {ns nodelist bw delay qtype args} {
   $self instvar node_
    set n1 [lindex $nodelist 0]
    foreach n2 [lrange $nodelist 1 end] {
	if ![info exists node_($n2)] {
	    set node_($n2) [$ns node]
	}
	$ns duplex-link $node_($n1) $node_($n2) $bw $delay $qtype
	foreach opt $args {
	    set cmd [lindex $opt 0]
	    set val [lindex $opt 1]
	    if {[llength $opt] > 2} {
		set x1 [lindex $opt 2]
		set x2 [lindex $opt 3]
	    } else {
		set x1 $n1
		set x2 $n2
	    }
	    $ns $cmd $node_($x1) $node_($x2) $val
	    $ns $cmd $node_($x2) $node_($x1) $val
	}
	set n1 $n2
    }
}


Class NodeTopology/4nodes -superclass SkelTopology


NodeTopology/4nodes instproc init ns {
    $self next

    $self instvar node_
    set node_(n0) [$ns node]
    set node_(n1) [$ns node]
    set node_(n2) [$ns node]
    set node_(n3) [$ns node]
}


Class Topology/net4 -superclass NodeTopology/4nodes

# Create a simple four node topology:
#
#	              n3
#	             / 
#       1.5Mb,10ms  / 1.5Mb,10ms                              
#    n0 --------- n1
#                  \  1.5Mb,10ms
#	            \ 
#	             n2
#

Topology/net4 instproc init ns {
    $self next $ns
    $self instvar node_
    $ns duplex-link $node_(n0) $node_(n1) 1.5Mb 10ms DropTail
    $ns duplex-link $node_(n1) $node_(n2) 1.5Mb 10ms DropTail
    $ns duplex-link $node_(n1) $node_(n3) 1.5Mb 10ms DropTail
    if {[$class info instprocs config] != ""} {
	$self config $ns
    }
}

Class NodeTopology/6nodes -superclass SkelTopology

NodeTopology/6nodes instproc init ns {
    $self next

    $self instvar node_
    set node_(n0) [$ns node]
    set node_(n1) [$ns node]
    set node_(n2) [$ns node]
    set node_(n3) [$ns node]
    set node_(n4) [$ns node]
    set node_(n5) [$ns node]
}


Class Topology/net6 -superclass NodeTopology/6nodes

#
# Create a simple six node topology:
#
#                  n0
#                 /  \                    
#               n1    n2
#              /  \  /  \
#             n3   n4   n5
#
# All links are of 1.5Mbps bandwidth with 10ms latency
#

Topology/net6 instproc init ns {
    $self next $ns
    $self instvar node_
    $ns duplex-link $node_(n0) $node_(n1) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n0) $node_(n2) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n1) $node_(n3) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n1) $node_(n4) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n2) $node_(n4) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n2) $node_(n5) 1.5Mb 10ms DropTail 
    if {[$class info instprocs config] != ""} {
	$self config $ns
    }
}

Class NodeTopology/10nodes -superclass SkelTopology
NodeTopology/10nodes instproc init ns {
    $self next

    $self instvar node_
    set node_(n0) [$ns node]
    set node_(n1) [$ns node]
    set node_(n2) [$ns node]
    set node_(n3) [$ns node]
    set node_(n4) [$ns node]
    set node_(n5) [$ns node]
    set node_(n6) [$ns node]
    set node_(n7) [$ns node]
    set node_(n8) [$ns node]
    set node_(n9) [$ns node]
}


Class Topology/net10 -superclass NodeTopology/10nodes

#
# Create a simple ten node topology:
#
#                  n0
#               +---+-+
#              n1     n2
#            +--+-+   |
#           n3    n4  n5
#          +-+-+  |   |
#         n6  n7  n8  n9
#
#
# All links are of 1.5Mbps bandwidth with 10ms latency
#

Topology/net10 instproc init ns {
    $self next $ns
    $self instvar node_

    $ns duplex-link $node_(n0) $node_(n1) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n0) $node_(n2) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n1) $node_(n3) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n1) $node_(n4) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n2) $node_(n5) 1.5Mb 10ms DropTail 
    $ns duplex-link $node_(n3) $node_(n6) 1.5Mb 10ms DropTail
    $ns duplex-link $node_(n3) $node_(n7) 1.5Mb 10ms DropTail
    $ns duplex-link $node_(n4) $node_(n8) 1.5Mb 10ms DropTail
    $ns duplex-link $node_(n5) $node_(n9) 1.5Mb 10ms DropTail
    if {[$class info instprocs config] != ""} {
	$self config $ns
    }
}

# Definition of test-suite tests

# Testing group join for mix-mode sim in a simple topology
Class Test/Mixmode1 -superclass TestSuite
Test/Mixmode1 instproc init net {
	$self instvar defNet_ test_ net_
	set defNet_	net4
	set test_	Mixmode1
	set net_	$net
	$self next
}

Test/Mixmode1 instproc run {} {
	$self instvar ns_ node_ testName_

	$ns_ detailed-duplex-link $node_(n0) $node_(n1)

        set mproto CtrMcast
        set mrthandle [$ns_ mrtproto $mproto {}]
        if {$mrthandle != ""} {
	    $ns_ at 0.01 "$mrthandle switch-treetype 0x8001"
        }

	set udp0 [new Agent/UDP]
	$ns_ attach-agent $node_(n2) $udp0
	$udp0 set dst_ 0x8001
	set cbr0 [new Application/Traffic/CBR]
	$cbr0 attach-agent $udp0
	$ns_ create-session $node_(n2) $udp0	
	
	set rcvr0 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n0) $rcvr0
	set rcvr1 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n1) $rcvr1
	set rcvr2 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n2) $rcvr2
	set rcvr3 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n3) $rcvr3

	$ns_ at 0.3 "$cbr0 start"
	$ns_ at 0.3 "$node_(n1) join-group $rcvr1 0x8001"
	$ns_ at 0.3 "$node_(n0) join-group $rcvr0 0x8001"
	$ns_ at 0.3 "$node_(n3) join-group $rcvr3 0x8001"
	$ns_ at 0.3 "$node_(n2) join-group $rcvr2 0x8001"
	
	$ns_ at 1.1 "$self finish [list $rcvr0 $rcvr1 $rcvr2 $rcvr3]"
	$ns_ run
}

Class Test/Mixmode2 -superclass TestSuite
Test/Mixmode2 instproc init net {
	$self instvar defNet_ test_ net_
	set defNet_	net4
	set test_	Mixmode2
	set net_	$net
	$self next
}

Test/Mixmode2 instproc run {} {
	$self instvar ns_ node_ testName_

	$ns_ detailed-duplex-link $node_(n1) $node_(n2)

        set mproto CtrMcast
        set mrthandle [$ns_ mrtproto $mproto {}]
        if {$mrthandle != ""} {
	    $ns_ at 0.01 "$mrthandle switch-treetype 0x8001"
        }

	set udp0 [new Agent/UDP]
	$ns_ attach-agent $node_(n2) $udp0
	$udp0 set dst_ 0x8001
	set cbr0 [new Application/Traffic/CBR]
	$cbr0 attach-agent $udp0
	$ns_ create-session $node_(n2) $udp0	
	
	set rcvr0 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n0) $rcvr0
	set rcvr1 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n1) $rcvr1
	set rcvr2 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n2) $rcvr2
	set rcvr3 [new Agent/LossMonitor]
	$ns_ attach-agent $node_(n3) $rcvr3

	$ns_ at 0.3 "$cbr0 start"
	$ns_ at 0.3 "$node_(n1) join-group $rcvr1 0x8001"
	$ns_ at 0.3 "$node_(n0) join-group $rcvr0 0x8001"
	$ns_ at 0.3 "$node_(n3) join-group $rcvr3 0x8001"
	$ns_ at 0.3 "$node_(n2) join-group $rcvr2 0x8001"
	
	$ns_ at 1.1 "$self finish [list $rcvr0 $rcvr1 $rcvr2 $rcvr3]"
	$ns_ run
}

Class Test/Mixmode3 -superclass TestSuite
Test/Mixmode3 instproc init net {
	$self instvar defNet_ test_ net_
	set defNet_	net4
	set test_	Mixmode3
	set net_	$net
	$self next
}

Test/Mixmode3 instproc run {} {
	$self instvar ns_ node_ testName_

	$ns_ detailed-duplex-link $node_(n1) $node_(n3)

        set mproto CtrMcast
        set mrthandle [$ns_ mrtproto $mproto {}]
        if {$mrthandle != ""} {
	    $ns_ at 0.01 "$mrthandle switch-treetype 0x8001"
        }