ptopgen.in 4.89 KB
Newer Older
Leigh B. Stoller's avatar
Leigh B. Stoller committed
1
#!/usr/bin/perl -w
2
3
use English;
use Getopt::Std;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
4

5
6
7
8
9
# Settings - Change these to your particular network.

# Switch interface speed - In Mbps
$switch_speed = 100;

10
11
12
13
14
15
16
sub usage()
{
    print STDOUT "Usage: ptopgen [-s switch]\n";
    exit(-1);
}
my  $optlist = "s:";

17
18
######################################################################

19
my $TBROOT = "@prefix@";
20
use lib '@prefix@/lib';
21
require exitonwarn;
22
use libdb;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
23

24
%switches=();
25
%used_switches=();
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
my $switchtouse;

#
# Parse command arguments. Once we return from getopts, all that should be
# left are the required arguments.
#
%options = ();
if (! getopts($optlist, \%options)) {
    usage();
}
if (@ARGV) {
    usage();
}
if (defined($options{"s"})) {
    $switchtouse = $options{"s"};
}
42

43
# Read class/type maps
44
45
my $result = DBQueryFatal("select class,type,delay_capacity from node_types");
while (($class,$type,$delaycapacity) = $result->fetchrow_array) {
46
    $classes{$type} = $class;
47
    $nodetypes{$type} = $delaycapacity;
48
}
49
$result->finish;
50
51
52
53

# switches can't delay
$nodetypes{"switch"} = 0;

54
# Print switches
55
56
57
58
59
60
61
62
63
64
65
66
67
if (defined($switchtouse)) {
    print "node $switchtouse switch:1\n";
    $switches{$switchtouse} = 1;
}
else {
    $result =
	DBQueryFatal("select node_id from nodes where role = \"testswitch\"");

    while (($switch) = $result->fetchrow_array) {
	print "node $switch switch:1\n";
	$switches{$switch} = 1;
    }
    $result->finish;
68
}
69

Leigh B. Stoller's avatar
Leigh B. Stoller committed
70
# Find available nodes
71
$result = DBQueryFatal("select a.node_id,a.type from" .
Leigh B. Stoller's avatar
Leigh B. Stoller committed
72
73
		     " nodes as a left join reserved as b" .
		     " on a.node_id=b.node_id" .
74
		     " where b.node_id is null and a.role = \"testnode\"");
75
while (($node,$type) = $result->fetchrow_array) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
76
77
78
79
80
81
82
83
84
    # Shark hack
    if (($shelf,$number) = ($node =~ /^sh(\d+)-(\d+)/)) {
	if ($number == 1) {
	    $nodes{"sh$shelf"} = "shark-shelf";
	}
    } else {
	$nodes{$node} = $type;
    }
}
85
$result->finish;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
86
87


88
$sharklinks = "";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
89
90
91
92
foreach $node (keys(%nodes)) {
    if ($node =~ /^sh/) {
	print "node $node shark-shelf:1\n";
    } else {
93
94
95
96
	$text = "node $node $nodes{$node}:1 $classes{$nodes{$node}}:1";
	$delay_capacity = $nodetypes{$nodes{$node}};
	if ($delay_capacity > 0) {
	    $text .= " delay:$delay_capacity";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
97
98
99
100
101
	}
	print "$text\n";
    }
}

102
# Read interfaces
103
$result = DBQueryFatal("SELECT node_id,card,port,iface,interface_type" .
104
		     " from interfaces");
105
while (($node,$card,$port,$iface,$type) = $result->fetchrow_array) {
106
107
108
109
110
    $interfacemap{"$node:$card:$port"} = $iface;
    if ($type ne "") {
	$interfacetypes{"$node:$card:$port"} = $type;
    }
}
111
$result->finish;
112
113

# Read interface types
114
115
$result = DBQueryFatal("SELECT type,max_speed from interface_types");
while (($type,$speed) = $result->fetchrow_array) {
116
117
    $interfacespeeds{$type} = $speed;
}
118
$result->finish;
119

120
$result = DBQueryFatal("SELECT node_id1,card1,port1,node_id2,card2,port2" .
121
 		    " from wires where type=\"Node\" or type=\"Trunk\"");
122
while (($node1,$card1,$port1,$node2,$card2,$port2) = 
123
       $result->fetchrow_array) {
124
    if ((defined($nodes{$node1}) || defined($switches{$node1})) && 
125
	(defined($nodes{$node2}) || defined($switches{$node2}))) {
126
127
128
129
130
131
132
133
134
	$iface1 = get_iface($node1,$card1,$port1);
	$iface2 = get_iface($node2,$card2,$port2);
	$iface1bw = get_ifacebw($node1,$card1,$port1);
	$iface2bw = get_ifacebw($node2,$card2,$port2);
	if ($iface1bw < $iface2bw) {
	    $bw = $iface1bw;
	} else {
	    $bw = $iface2bw;
	}
135
	if (defined($switches{$node2})) {
136
137
	    $used_switches{$node2} = 1;
	}
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
	if (defined($switches{$node1})) {
	    $used_switches{$node1} = 1;
	}
	if (defined($switches{$node1}) && defined($switches{$node2})) {
	    # interswitch link
	    if (defined($interconnects{"$node1:$node2"})) {
		$interconnects{"$node1:$node2"} += $bw;
	    } else {
		$interconnects{"$node1:$node2"} = $bw;
	    }
	} else {
	    # !!! - Here we use our knowledge that in the wires table links
	    # to the switch always come as node2.
	    print "link link-$node1:$iface1-$node2:$iface2 $node1:$node1/$iface1" .
		" $node2:$iface2 $bw 1\n";
	}
    } 
155
}
156
$result->finish;
157
158
159
160
foreach $interconnect (keys(%interconnects)) {
    ($src,$dst) = split(":",$interconnect);
    print "link link-$interconnect $src $dst $interconnects{$interconnect} 1\n";
}
161
162
163

print $sharklinks;

Leigh B. Stoller's avatar
Leigh B. Stoller committed
164
# Add a bunch of LANs
165
166
167
168
169
170
foreach $switch (keys(%used_switches)) {
    for ($i = 0;$i <= 20;$i++) {
	print "node lan-$switch-$i lan:1\n";
	print "link link-lan-$switch-$i lan-$switch-${i}:lan-$switch-$i" .
	    " $switch 100000 1000\n";
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
171
}
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198


# Print out links
sub get_iface {
    ($node,$card,$port) = @_;
    if (defined($interfacemap{"$node:$card:$port"})) {
	return $interfacemap{"$node:$card:$port"};
    } else {
	# shark hack
	if ($node =~ /^sh/) {
	    return "eth0";
	} else {
	    return "(null)";
	}
	# end shark hack 
    }
};

# Find the bandwidth for an interface
sub get_ifacebw {
    ($node,$card,$port) = @_;
    if (defined($interfacetypes{"$node:$card:$port"})) {
	return $interfacespeeds{$interfacetypes{"$node:$card:$port"}};
    } else {
	return $switch_speed;
    }
}