node.tcl 4.11 KB
Newer Older
Leigh B. Stoller's avatar
Leigh B. Stoller committed
1 2 3 4 5 6 7 8 9 10 11 12
######################################################################
# node.tcl
#
# This defines the Node class.  Instances of this class are created by
# the 'node' method of Simulator.  A Node is connected to a number of
# LanLinks.  Each such connection is associated with a virtual port,
# an integer.  Each virtual port also has an IP address.  Virtual
# ports start at 0 and go up continuously.  Besides the port
# information each node also has a variety of strings.  These strings
# are set by tb-* commands and dumped to the DB but are otherwise
# uninterpreted.
######################################################################
Mac Newbold's avatar
Mac Newbold committed
13

Leigh B. Stoller's avatar
Leigh B. Stoller committed
14
Class Node -superclass NSObject
Mac Newbold's avatar
Mac Newbold committed
15

Leigh B. Stoller's avatar
Leigh B. Stoller committed
16 17
Node instproc init {s} {
    $self set sim $s
Mac Newbold's avatar
Mac Newbold committed
18

Leigh B. Stoller's avatar
Leigh B. Stoller committed
19 20 21 22 23 24 25 26 27 28
    # portlist is a list of connections for the node.  It is sorted
    # by portnumber.  I.e. the ith element of portlist is the connection
    # on port i.
    $self set portlist {}

    # iplist, like portlist, is supported by portnumber.  An entry of
    # {} indicates an unassigned IP address for that port.
    $self set iplist {}

    # The type of the node.
29
    $self set type "pc" 
Leigh B. Stoller's avatar
Leigh B. Stoller committed
30 31 32 33 34 35 36 37 38 39 40 41

    # If osid remains blank when updatedb is called it is changed
    # to the default OS based on it's type (taken from node_types
    # table).
    $self set osid ""

    # These are just various strings that we pass through to the DB.
    $self set cmdline ""
    $self set rpms ""
    $self set startup ""
    $self set deltas ""
    $self set tarfiles ""
42
    $self set failureaction "fatal"
43
    $self set routertype "none"
Leigh B. Stoller's avatar
Leigh B. Stoller committed
44 45 46 47 48 49 50
}

# The following procs support renaming (see README)
Node instproc rename {old new} {
    $self instvar portlist
    foreach object $portlist {
	$object rename_node $old $new
Mac Newbold's avatar
Mac Newbold committed
51
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
52 53 54 55 56 57 58 59 60 61
    [$self set sim] rename_node $old $new
}

Node instproc rename_lanlink {old new} {
    $self instvar portlist
    set newportlist {}
    foreach node $portlist {
	if {$node == $old} {
	    lappend newportlist $new
	} else {
62
	    lappend newportlist $node
Leigh B. Stoller's avatar
Leigh B. Stoller committed
63
	}
Mac Newbold's avatar
Mac Newbold committed
64
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
65
    set portlist $newportlist
Mac Newbold's avatar
Mac Newbold committed
66 67
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
68 69 70 71 72 73 74 75 76 77 78 79
# updatedb DB
# This adds a row to the virt_nodes table corresponding to this node.
Node instproc updatedb {DB} {
    $self instvar portlist
    $self instvar type
    $self instvar osid
    $self instvar cmdline
    $self instvar rpms
    $self instvar startup
    $self instvar deltas
    $self instvar iplist
    $self instvar tarfiles
80
    $self instvar failureaction
81
    $self instvar routertype
Leigh B. Stoller's avatar
Leigh B. Stoller committed
82 83
    var_import ::GLOBALS::pid
    var_import ::GLOBALS::eid
Mac Newbold's avatar
Mac Newbold committed
84

Leigh B. Stoller's avatar
Leigh B. Stoller committed
85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
    # If we haven't specified a osid so far then we should fill it
    # with the id from the node_types table now.
    if {$osid == {}} {
	sql query $DB "select osid from node_types where type = \"$type\""
	set osid [sql fetchrow $DB]
	sql endquery $DB
    }

    # We need to generate the IP column from our iplist.
    set ipraw {}
    set i 0
    foreach ip $iplist {
	lappend ipraw $i:$ip
	incr i
    }
Mac Newbold's avatar
Mac Newbold committed
100

Leigh B. Stoller's avatar
Leigh B. Stoller committed
101
    # Update the DB
102
    sql exec $DB "insert into virt_nodes (pid,eid,vname,type,ips,osid,cmd_line,rpms,deltas,startupcmd,tarfiles,failureaction,routertype) values (\"$pid\",\"$eid\",\"$self\",\"$type\",\"$ipraw\",\"$osid\",\"$cmdline\",\"$rpms\",\"$deltas\",\"$startup\",\"$tarfiles\",\"$failureaction\",\"$routertype\")";
Mac Newbold's avatar
Mac Newbold committed
103
}
104

Leigh B. Stoller's avatar
Leigh B. Stoller committed
105 106 107 108 109 110 111 112 113 114
# add_lanlink lanlink
# This creates a new virtual port and connects the specified LanLink to it.
# The port number is returned.
Node instproc add_lanlink {lanlink} {
    $self instvar portlist
    $self instvar iplist
    lappend portlist $lanlink
    lappend iplist ""
    return [expr [llength $portlist] - 1]
}
115

Leigh B. Stoller's avatar
Leigh B. Stoller committed
116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
# ip port
# ip port ip
# In the first form this returns the IP address associated with the port.
# In the second from this sets the IP address of a port.
Node instproc ip {port args} {
    $self instvar iplist
    $self instvar sim
    if {$args == {}} {
	return [lindex $iplist $port]
    } else {
	set ip [lindex $args 0]
	set subnet [join [lrange [split $ip .] 0 2] .]
	set iplist [lreplace $iplist $port $port $ip]
	$sim use_subnet $subnet
    }    
131 132
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
133 134 135 136 137 138
# find_port lanlink
# This takes a lanlink and returns the port it is connected to or 
# -1 if there is no connection.
Node instproc find_port {lanlink} {
    return [lsearch [$self set portlist] $lanlink]
}