xentaur

annotate xentaur.py @ 63:07c6777758dc

* path to shapes
* dynamips icon
author igor
date Fri Feb 29 07:22:46 2008 +0200 (2008-02-29)
parents e838c1223b89
children cf823d27b029
rev   line source
igor@0 1 #!/usr/bin/python
igor@60 2 # vim: set fileencoding=utf-8 :
igor@0 3
igor@2 4 import sys,os,time
igor@35 5
igor@35 6 xentaur_path=os.environ['HOME']+"/xentaur"
igor@35 7
igor@0 8 sys.path.append('/etc/xen')
igor@35 9 sys.path.append(xentaur_path)
igor@63 10 sys.path.append('.')
igor@11 11
igor@56 12 #network='snrs_ipsec_rsa_1'
igor@56 13 node_object={}
igor@56 14 link_object={}
igor@56 15 bridge_object={}
igor@56 16
igor@63 17 network='openvpnbridge'
igor@63 18 domain='debian1'
igor@62 19
igor@61 20 #network='snrs'
igor@61 21 #domain='dyn1'
igor@35 22 from xendomain import *
igor@35 23
igor@29 24 bridges_turned_down=[]
igor@29 25
igor@0 26 from IPython.Shell import IPShellEmbed
igor@0 27
igor@2 28
igor@2 29 screenrc=os.environ['HOME']+"/.screenrc_xentaur"
igor@63 30 path_shapes='/xen/xentaur/shapes'
igor@2 31
igor@10 32 def run(program, *args):
igor@10 33 pid = os.fork()
igor@10 34 if not pid:
igor@10 35 os.execvp(program, (program,) + args)
igor@10 36 return os.wait()[0]
igor@10 37
igor@10 38 def run_command(line):
igor@10 39 #cmds=line.split()
igor@10 40 #run(cmds[0],*cmds[1:])
igor@10 41 run("/bin/sh", "-c", line)
igor@10 42
igor@23 43 def run_command_return_stdout(line):
igor@23 44 p = os.popen(line)
igor@23 45 output = p.read()
igor@23 46 p.close()
igor@23 47 return output
igor@23 48
igor@49 49 ################################################################################
igor@49 50 #Xentaur command-line commands
igor@49 51
igor@49 52 ## Start
igor@49 53
igor@49 54 def start_bridges():
igor@38 55 unbound_bridges=set(bridges)-set(real_bridges)
igor@56 56 script=""
igor@56 57 script="\n".join(map(lambda x: "sudo brctl show | awk '{print $1}' | grep -qx "+x+" || sudo brctl addbr "+x, unbound_bridges))
igor@56 58 script+="\n"+"\n".join(map(lambda x: "sudo brctl stp "+x+" off", unbound_bridges))
igor@56 59 script+="\n"+"\n".join(map(lambda x: "sudo ip link set "+x+" up", unbound_bridges))
igor@0 60
igor@0 61 print """#!/bin/sh
igor@0 62 # create unbound bridges
igor@56 63 %s
igor@56 64 """ % (script)
igor@0 65
igor@49 66 def start_domain(domain):
igor@49 67 print "sudo xm create "+xentaur_path+"/xendomain.py "+" domain="+domain+" network="+network+" && sleep 1 && sudo xm sched-credit -d $(sudo xm list | grep "+domain+" | awk '{print $2}') -c 10 && sleep 1"
igor@0 68
igor@49 69 def start_domains(doms=domains):
igor@49 70 for domain in doms:
igor@38 71 if not domain in real_nodes:
igor@49 72 start_domain(domain)
igor@0 73
igor@49 74 def start_all():
igor@49 75 graph()
igor@49 76 screen()
igor@49 77 start_bridges()
igor@49 78 start_domains()
igor@49 79
igor@49 80 ## Stop
igor@49 81
igor@49 82 def stop_domain(domain,wait=0):
igor@49 83 if wait:
igor@49 84 print "sudo xm shutdown -w "+domain
igor@49 85 else:
igor@49 86 print "sudo xm shutdown "+domain
igor@49 87
igor@49 88 def stop_domains(doms=domains, wait=0):
igor@49 89 for domain in doms:
igor@38 90 if not domain in real_nodes:
igor@49 91 stop_domain(domain,wait)
igor@0 92
igor@49 93 def stop_bridges():
igor@49 94 ###FIXME###
igor@49 95 return 0
igor@49 96
igor@49 97 def stop_all(wait=0):
igor@49 98 stop_domains(domains, wait)
igor@49 99 stop_bridges()
igor@49 100
igor@49 101 def restart_all():
igor@49 102 stop_all(1)
igor@49 103 start_all()
igor@49 104
igor@56 105 ####################################################
igor@56 106
igor@56 107 def create_objects():
igor@56 108 create_node_objects()
igor@56 109 create_bridge_objects()
igor@56 110 create_link_objects()
igor@56 111
igor@56 112 def create_node_objects():
igor@56 113 for dom in domains:
igor@56 114 node_object[dom]=Node(dom)
igor@56 115
igor@56 116 def create_bridge_objects():
igor@56 117 for bridge in bridges:
igor@56 118 bridge_object[bridge]=Bridge(bridge)
igor@56 119
igor@56 120 def create_link_objects():
igor@56 121
igor@56 122 for node, bridges_raw in vbridges_table.iteritems():
igor@56 123 interface=0
igor@56 124 j=0
igor@56 125 for this_bridge in bridges_raw:
igor@56 126 int_label=""
igor@56 127 if this_bridge.find(':') != -1:
igor@56 128 res = this_bridge.split(':')
igor@56 129 this_bridge= res[0]
igor@56 130 bridges_raw[j] = this_bridge
igor@56 131 int_label = res[1]
igor@56 132 if not [ node, bridges_raw.index(this_bridge), this_bridge ] in temporary_links:
igor@56 133 name="%s %s %s" % (node,interface,this_bridge)
igor@56 134 link_object[name]=Link(name,node,interface,this_bridge,int_label)
igor@56 135 interface+=1
igor@56 136 vbridges_table[node]=bridges_raw
igor@56 137
igor@56 138 for node, bridges_raw in bridge_bridge_table.iteritems():
igor@56 139 interface=0
igor@56 140 j=0
igor@56 141 for this_bridge in bridges_raw:
igor@56 142 int_label=""
igor@56 143 if this_bridge.find(':') != -1:
igor@56 144 res = this_bridge.split(':')
igor@63 145 this_bridge= res[2]
igor@56 146 bridges_raw[j] = this_bridge
igor@56 147 int_label = res[1]
igor@56 148 if not [ node, bridges_raw.index(this_bridge), this_bridge ] in temporary_links:
igor@56 149 name="%s %s %s" % (node,interface,this_bridge)
igor@56 150 link_object[name]=Link(name,node,interface,this_bridge,int_label)
igor@56 151 interface+=1
igor@56 152 bridge_bridge_table[node]=bridges_raw
igor@56 153
igor@56 154 for node,interface,bridge in temporary_links:
igor@56 155 name="%s %s %s" % (node,interface,bridge)
igor@56 156 link_object[name]=Link(name,node,interface,this_bridge)
igor@56 157
igor@56 158 for node,interface,bridge in broken_links:
igor@56 159 name="%s %s %s" % (node,interface,bridge)
igor@56 160 link_object[name]=Link(name,node,interface,this_bridge)
igor@56 161
igor@56 162
igor@56 163 ####################################################
igor@56 164
igor@49 165 def screen():
igor@60 166 wait_seconds=1
igor@0 167 screens=[]
igor@0 168 for domain in domains:
igor@56 169 screens.append("screen -t %s %s sh -c 'while true; do %s ; echo Retrying in %s secods...; sleep %s ; clear; done'" %
igor@60 170 (domain,domains.index(domain)+1,node_object[domain].console_string(),wait_seconds,wait_seconds))
igor@0 171 screenlist="\n".join(screens)
igor@0 172
igor@49 173 hardstatus='hardstatus string "%{rk}Xentaur%{bk}@%H %{gk}%c %{yk}%d.%m %{wk}%?%-Lw%?%{bw}%n*%f%t%?(%u)%?%{wk}%?%+Lw%?"'
igor@10 174
igor@11 175 f=open(screenrc, "w");
igor@10 176 f.write("""
igor@2 177 hardstatus on
igor@2 178 hardstatus alwayslastline
igor@49 179 %s
igor@2 180
igor@60 181 screen -t console 0 sh -c 'while true; do cd %s; ./xentaur.py shell ; echo Retrying in %s secods...; sleep %s ; clear; done'
igor@60 182 #screen -t xentaur - sh -c 'while true; do bash ; echo Retrying in %s secods...; sleep %s ; clear; done'
igor@10 183 %s
igor@60 184 """ % (hardstatus,xentaur_path,wait_seconds,wait_seconds,wait_seconds,wait_seconds,screenlist))
igor@10 185 f.close()
igor@49 186 print "# GNU Screen config file is written to: %s" % screenrc
igor@0 187
igor@0 188 def graph():
igor@0 189 nodelist=""
igor@0 190 bridgelist=""
igor@0 191 linklist=""
igor@0 192 physicallist=""
igor@0 193 networklist=""
igor@0 194
igor@56 195 nodelist=";\n ".join(map(lambda node: node_object[node].graphviz_string(),nodes))
igor@0 196 if nodelist: nodelist += ";"
igor@56 197 bridgelist=";\n ".join(map(lambda bridge: bridge_object[bridge].graphviz_string(),bridges))
igor@0 198 if bridgelist: bridgelist += ";"
igor@56 199 linklist=";\n ".join(map(lambda link: link_object[link].graphviz_string(),link_object.keys()))
igor@56 200 if linklist: linklist += ";"
igor@0 201
igor@46 202 f = open(network+".dot", "w");
igor@10 203 f.write ("""
igor@0 204 graph G {
igor@0 205 edge [len=1.25];
igor@0 206 splines=true;
igor@0 207 // nodes
igor@56 208 %s
igor@0 209
igor@0 210 // bridges
igor@56 211 %s
igor@0 212
igor@0 213 // physical
igor@0 214 node [shape=rectangle,color=blue];
igor@56 215 %s
igor@0 216
igor@0 217 // networks (not bridges, not physical)
igor@0 218 node [shape=rectangle,color=green];
igor@56 219 %s
igor@0 220
igor@0 221 // links (between nodes and bridges)
igor@56 222 %s
igor@0 223
igor@0 224 };
igor@56 225 """ % (nodelist, bridgelist, physicallist, networklist, linklist))
igor@10 226 f.close()
igor@46 227 run_command("neato -Tpng -o %s.png %s.dot "%(network,network))
igor@46 228 run_command("neato -Tjpg -o %s.jpg %s.dot "%(network,network))
igor@46 229 run_command("neato -Tsvg -o %s.svg %s.dot "%(network,network))
igor@56 230 run_command("neato -Tcmapx -o %s.cmapx -NURL=http://google.com %s.dot "%(network,network))
igor@49 231 print "# Network map is written to files: %s.{png,svg,jpg,dot}" % network
igor@0 232
igor@27 233 def autoredraw():
igor@27 234 graph()
igor@27 235
igor@0 236 def shell():
nata@31 237 autoredraw()
igor@0 238 ipshell = IPShellEmbed()
igor@0 239 ipshell()
igor@0 240
igor@60 241 def version():
igor@61 242 print """
igor@61 243 Xentaur 0.1-PRE
igor@61 244
igor@61 245 ,--,
igor@61 246 _ ___/ /\\|
igor@61 247 ,;`( )__, ) ~
igor@61 248 // .// '--;
igor@61 249 ' / \ |
igor@61 250
igor@61 251 """
igor@61 252 # print "Xentaur 0.1-PRE"
igor@61 253 # print "(Godzilla-mutant) _"
igor@61 254 # print " / * \\"
igor@61 255 # print " / .-"
igor@61 256 # print " / |"
igor@61 257 # print " | \\ \\\\ \\"
igor@61 258 # print " _ -------| \\ \\\\ \\"
igor@61 259 # print " / / \\_\\ -"
igor@61 260 # print "/ |\\ | |"
igor@61 261 # print "| | \\ .-----. | \\ |"
igor@61 262 # print " | / \\ \\ \\ \\"
igor@61 263 # print " \\/|.\\ \\ \\ \\ \\"
igor@61 264 # print " \\| - . \\_\\ \\_\\"
igor@61 265 # print "-----------------------------------------------"
igor@60 266
igor@60 267
igor@60 268 def info():
igor@60 269 version()
igor@60 270
igor@60 271 print "Network name: ", network
igor@60 272 print "-----------------------------------------------"
igor@60 273 print
igor@60 274 print "Nodes: ", len(domains)
igor@60 275 print " * virtual nodes: ", len(domains)-len(real_nodes)
igor@60 276 print " * real nodes:", len(real_nodes)
igor@60 277 print
igor@60 278 print "Bridges:", len(bridges)
igor@60 279 print " * virtual bridges:", len(bridges)-len(real_bridges)-len(cross_bridges)
igor@60 280 print " * real switches:", len(real_bridges)
igor@60 281 print " * direct links:", len(cross_bridges)
igor@60 282
igor@0 283 def show_usage():
igor@0 284 print """Usage:
igor@49 285 xentaur <command> [<argument>]
igor@49 286
igor@49 287 Commands:
igor@49 288 start-all -- start bridges and domains
igor@49 289 start-domains -- start domains only
igor@49 290 start-bridges -- start bridges only
igor@49 291 stop-all -- stop bridges and domains
igor@49 292 stop-domains -- stop domains only
igor@49 293 stop-bridges -- stop bridges only (domains have to be stopped already)
igor@49 294 restart-all -- restart bridges and domains
igor@49 295
igor@49 296 start <domain> -- start the <domain>
igor@49 297 stop <domain> -- stop the <domain>
igor@49 298
igor@49 299 graph -- generate network scheme (result is in <network>.{png,jpg,svg})
igor@49 300 screen -- generate GNU Screen config file (~/.screenrc_xentaur)
igor@49 301 shell -- run Xentaur shell
igor@49 302
igor@0 303 """
igor@0 304
igor@33 305 def save():
igor@33 306 print "network =", xen_config_name
igor@33 307 print "domains =", domains
igor@33 308 print "domain_types =", domain_types
igor@33 309 print "bridges =", bridges
igor@33 310 print "vbridges_table =", vbridges_table
igor@33 311 print "hidden_bridges =", hidden_bridges
igor@33 312 print "broken_links =", broken_links
igor@33 313 print "temporary_links =", temporary_links
igor@33 314 print "bridges_turned_down =", bridges_turned_down
igor@33 315
igor@33 316 #-----------------------------------------------------------------------
igor@33 317 # CLASSES
igor@33 318
igor@33 319 class Bridge:
igor@33 320 def __init__ (self,name):
igor@33 321 self.name=name
igor@33 322 def up(self):
igor@33 323 bridge_up(self.name)
igor@33 324 def down(self):
igor@33 325 bridge_down(self.name)
igor@33 326 def show(self):
igor@33 327 show_bridge(self.name)
igor@33 328 def dump_start(self,filter=""):
igor@33 329 dump_start(self.name,filter)
igor@33 330
igor@56 331 def is_hidden(self):
igor@56 332 return self.name in hidden_bridges
igor@56 333 def is_real(self):
igor@56 334 return self.name in real_bridges
igor@56 335 def is_turned_down(self):
igor@56 336 return self.name in bridges_turned_down
igor@56 337 def is_cross(self):
igor@56 338 return self.name in cross_bridges
igor@33 339
igor@56 340 def graphviz_string(self):
igor@56 341 if self.is_hidden():
igor@63 342 return "//"
igor@56 343 elif self.is_cross():
igor@56 344 return "%s [shape=circle,height=0.03,color=black,fillcolor=black,style=filled,label=\"\"]" % (self.name)
igor@56 345 elif self.is_real():
igor@63 346 return "%s [color=white,shape=none,shapefile=\"%s/all/real_switch.png\"]" % (self.name, path_shapes)
igor@56 347 elif self.is_turned_down():
igor@63 348 return "%s [color=white,shape=none,shapefile=\"%s/all/switch_turned_down.png\"]" % (self.name, path_shapes)
igor@56 349 else:
igor@63 350 return "%s [color=white,shape=none,shapefile=\"%s/all/switch.png\"]" % (self.name, path_shapes)
igor@56 351
igor@56 352
igor@56 353 class Node:
igor@38 354 def __init__ (self,name):
igor@38 355 self.name=name
igor@56 356 self.type=domain_types[domains.index(name)]
igor@38 357 def start(self):
igor@38 358 return ""
igor@38 359 def stop(self):
igor@38 360 return ""
igor@38 361 def start_commandline(self):
igor@38 362 return ""
igor@38 363 def get_domain_id(self):
igor@38 364 return get_domain_id(self.name)
igor@56 365 def graphviz_string(self):
igor@63 366 return self.name+" [color=white,shape=plaintext,label=\" "+self.name+"\",shapefile=\""+path_shapes+"/all/"+\
igor@56 367 domain_types[domains.index(self.name)]+".png\",fontcolor=black,fontsize=16,target=\"http://google.com\"]"
igor@56 368 def console_string(self):
igor@62 369 if self.type in [ 'quagga', 'xenomips', 'freebsd', 'linux' ]:
igor@56 370 return "sudo xm console "+self.name
igor@56 371 elif self.name in real_bridges or self.name in real_nodes:
igor@56 372 return "echo Press enter to connect; read line; "+connection_table[self.name]
igor@56 373
igor@56 374
igor@56 375 class Link:
igor@56 376 def __init__ (self,name,node,interface,bridge,label=""):
igor@56 377 self.name=name
igor@56 378 self.node=node
igor@56 379 self.interface=interface
igor@56 380 self.bridge=bridge
igor@56 381 self.label=label
igor@56 382
igor@56 383 def is_temporary(self):
igor@56 384 return [self.node,self.interface,self.bridge] in temporary_links
igor@56 385
igor@56 386 def is_broken(self):
igor@56 387 return ([self.node,self.interface,self.bridge] in broken_links)
igor@56 388
igor@63 389 def is_hidden(self):
igor@63 390 return self.bridge in hidden_bridges
igor@63 391
igor@56 392 def graphviz_string(self):
igor@63 393 if self.is_hidden():
igor@63 394 return "//"
igor@56 395 if self.is_temporary():
igor@56 396 return self.node+" -- "+self.bridge+" [taillabel=\"fa"+str(self.interface)+"/0\",color=blue,len=10,w=5,weight=5]"
igor@56 397 if self.is_broken():
igor@56 398 return self.node+" -- "+self.bridge+" [taillabel=\"fa"+str(self.interface)+"/0\",style=dashed]"
igor@56 399
igor@56 400 ip="\\n.%s.%s" % (bridges.index(self.bridge)+1, domains.index(self.node)+1)
igor@56 401 if domain_types[domains.index(self.node)] == 'xenomips':
igor@56 402 int_name="fa"+str(self.interface)+"/0"
igor@56 403 else:
igor@56 404 int_name="eth"+str(self.interface)
igor@56 405 if self.label != "":
igor@56 406 int_name = self.label
igor@56 407 return self.node+" -- "+self.bridge+" [taillabel=\""+int_name+ip+"\",fontsize=14,fontname=fixed]"
igor@56 408
igor@38 409
igor@23 410 #-----------------------------------------------------------------------
igor@23 411 # DOMAINS
igor@23 412
igor@23 413 def get_domain_id(domain):
igor@23 414 return run_command_return_stdout("sudo xm list | awk '{if ($1 == \"'%s'\") print $2}'" % domain).rstrip("\n")
igor@23 415
igor@22 416
igor@22 417 #-----------------------------------------------------------------------
igor@22 418 # BRIDGES and IFACES
igor@22 419
igor@22 420 def bridge_down(bridge):
igor@22 421 """
igor@22 422 Turn the bridge <bridge> down
igor@22 423 """
igor@38 424 if bridge in real_bridges:
igor@38 425 print "Bridge %s is a real bridge" % (bridge)
igor@38 426 return -1
igor@29 427 if bridge in bridges_turned_down:
igor@29 428 print "Bridge %s is turned down already" % (bridge)
igor@29 429 else:
igor@29 430 bridges_turned_down.append(bridge)
igor@29 431 run_command("sudo ip link set %s down" % bridge)
igor@29 432 autoredraw()
igor@22 433
igor@22 434 def bridge_up(bridge):
igor@22 435 """
igor@22 436 Turn the bridge <bridge> up
igor@22 437 """
igor@38 438 if bridge in real_bridges:
igor@38 439 print "Bridge %s is a real bridge" % (bridge)
igor@38 440 return -1
igor@29 441 if not (bridge in bridges_turned_down):
igor@29 442 print "Bridge %s is turned up already" % (bridge)
igor@29 443 else:
igor@29 444 bridges_turned_down.remove(bridge)
igor@29 445 run_command("sudo ip link set %s up" % bridge)
igor@29 446 autoredraw()
igor@22 447
igor@22 448 def show_bridge(bridge):
igor@22 449 """
igor@22 450 Show the state of the bridge <bridge>
igor@22 451 """
igor@38 452 if bridge in real_bridges:
igor@38 453 print "Bridge %s is a real bridge" % (bridge)
igor@38 454 return -1
igor@22 455 run_command("sudo ip link show %s" % bridge)
igor@22 456
igor@23 457
igor@23 458 def int_disconnect(domain, int_number):
igor@23 459 """
igor@23 460 Disconnect the interface with the number <int_number>
igor@23 461 of the domain <domain> from the bridge to which
igor@23 462 it is connected
igor@23 463 """
igor@23 464 dom_id=get_domain_id(domain)
igor@23 465 bridge=vbridges_table[domain][int_number]
igor@23 466 if not bridge:
igor@23 467 print "Interface %s of the %s domain is not connected" % (int_number, domain)
igor@23 468 return 1
igor@23 469 run_command("sudo brctl delif %s vif%s.%s" % (bridge, dom_id, int_number))
igor@23 470 vbridges_table[domain][int_number]=''
igor@28 471 if [ domain, int_number, bridge ] in temporary_links:
igor@28 472 temporary_links.remove([ domain, int_number, bridge ])
igor@27 473 else:
igor@28 474 broken_links.append([ domain, int_number, bridge ])
igor@27 475 autoredraw()
igor@23 476
igor@23 477 def int_connect(domain, int_number, bridge):
igor@23 478 """
igor@23 479 Connect the interface with the number <int_number>
igor@24 480 of the domain <domain> to the bridge <bridge>
igor@23 481 """
igor@38 482 if bridge in real_bridges:
igor@38 483 print "Bridge %s is a real bridge" % (bridge)
igor@38 484 return -1
igor@38 485
igor@23 486 dom_id=get_domain_id(domain)
igor@23 487 if vbridges_table[domain][int_number]:
igor@23 488 print "Interface %s of the %s domain is connected already to the %s bridge" % (int_number, domain, vbridges_table[domain][int_number])
igor@23 489 return 1
igor@23 490 run_command("sudo brctl addif %s vif%s.%s" % (bridge, dom_id, int_number))
igor@23 491 vbridges_table[domain][int_number]=bridge
igor@28 492 if [ domain, int_number, bridge ] in broken_links:
igor@28 493 broken_links.remove([ domain, int_number, bridge ])
igor@27 494 else:
igor@28 495 temporary_links.append([ domain, int_number, bridge ])
igor@27 496 autoredraw()
igor@23 497
igor@24 498 def int_reconnect(domain, int_number, bridge):
igor@24 499 """
igor@24 500 Reconnect the interface with the number <int_number>
igor@24 501 of the domain <domain> from the bridge to which
igor@24 502 it is connected to the bridge <bridge>
igor@24 503 """
igor@38 504 if bridge in real_bridges:
igor@38 505 print "Bridge %s is a real bridge" % (bridge)
igor@38 506 return -1
igor@38 507
igor@24 508 int_disconnect(domain, int_number)
igor@24 509 int_connect(domain, int_number, bridge)
igor@24 510
igor@24 511 def show_int(domain, int_number):
igor@25 512 """
igor@25 513 Show information about the interface <int_nuber>
igor@25 514 of the domain <domain>
igor@25 515 """
igor@26 516 return vbridges_table[domain][int_number]
igor@24 517
igor@28 518
igor@28 519 def dump_start(bridge, filter=""):
igor@38 520 if bridge in real_bridges:
igor@38 521 print "Bridge %s is a real bridge" % (bridge)
igor@38 522 return -1
igor@32 523 try:
igor@32 524 print "Writing dump... (press Ctrl-C to stop)"
igor@32 525 run_command("sudo tcpdump -w xentaur.dump -i %s %s > /dev/null 2>&1 " % (bridge,filter))
igor@32 526 except:
igor@32 527 print "Done.\n Dump is written to xentaur.dump"
igor@28 528 return 0
igor@28 529
igor@28 530 def dump_stop():
igor@28 531 return 0
igor@33 532
igor@33 533
igor@33 534 #-----------------------------------------------------------------------
igor@33 535 # CONFIGURATION TEMPLATES
igor@33 536
igor@33 537
igor@33 538 def configure_ip_addresses(doms=domains):
igor@40 539
igor@40 540 cisco_set_ip_on_int="""
igor@40 541 \n\n\n
igor@40 542 int fa%s/0
igor@40 543 no ip address
igor@40 544 ip address %s 255.255.255.0
igor@40 545 no shutdown
igor@40 546 exit
igor@40 547 """
igor@40 548
igor@40 549 quagga_set_ip_on_int="""
igor@40 550 int eth%s
igor@40 551 no ip address
igor@40 552 ip address %s/24
igor@40 553 no shutdown
igor@40 554 exit
igor@40 555 """
igor@40 556
igor@63 557 linux_set_ip_on_int="""
igor@63 558 ifconfig eth%s %s netmask 255.255.255.0
igor@63 559 """
igor@63 560
igor@40 561 for dom in doms:
igor@40 562 i=domains.index(dom)+1
igor@40 563 if domain_types[domains.index(dom)] == 'quagga':
igor@40 564 command = quagga_set_ip_on_int
igor@40 565 write_to(i,"\nconf t\n")
igor@40 566 j=0
igor@40 567 for br in vbridges_table[dom]:
igor@40 568 write_to(i,command % (j, "192.168.%s.%s"%(bridges.index(br)+1,i)))
igor@40 569 j+=1
igor@40 570 write_to(i,"\nend\n")
igor@63 571 elif domain_types[domains.index(dom)] == 'linux':
igor@63 572 command = linux_set_ip_on_int
igor@63 573 j=0
igor@63 574 for br in vbridges_table[dom]:
igor@63 575 #write_to(i, command % (j, "192.168.%s.%s"%(bridges.index(br)+1,i)) )
igor@63 576 print i, command % (j, "192.168.%s.%s"%(bridges.index(br)+1,i))
igor@63 577 j+=1
igor@40 578 else:
igor@40 579 command = cisco_set_ip_on_int
igor@40 580 write_to(i,"\nena\nconf t\n")
igor@40 581 j=0
igor@40 582 for br in vbridges_table[dom]:
igor@40 583 write_to(i,command % (j, "192.168.%s.%s"%(bridges.index(br)+1,i)))
igor@40 584 j+=1
igor@40 585 write_to(i,"\nend\n")
igor@40 586 return 0
igor@40 587
igor@40 588 def configure_no_ip_addresses(doms=domains):
igor@40 589
igor@40 590 cisco_set_ip_on_int="""
igor@40 591 \n\n\n
igor@40 592 int fa%s/0
igor@40 593 no ip address %s 255.255.255.0
igor@40 594 exit
igor@40 595 """
igor@40 596
igor@40 597 quagga_set_ip_on_int="""
igor@40 598 int eth%s
igor@40 599 no ip address %s/24
igor@40 600 exit
igor@40 601 """
igor@40 602
igor@40 603 for dom in doms:
igor@40 604 i=domains.index(dom)+1
igor@40 605 if domain_types[domains.index(dom)] == 'quagga':
igor@40 606 command = quagga_set_ip_on_int
igor@40 607 write_to(i,"\nconf t\n")
igor@40 608 j=0
igor@40 609 for br in vbridges_table[dom]:
igor@40 610 write_to(i,command % (j, "192.168.%s.%s"%(bridges.index(br)+1,i)))
igor@40 611 j+=1
igor@40 612 write_to(i,"\nend\n")
igor@40 613 else:
igor@40 614 command = cisco_set_ip_on_int
igor@40 615 write_to(i,"\nena\nconf t\n")
igor@40 616 j=0
igor@40 617 for br in vbridges_table[dom]:
igor@40 618 write_to(i,command % (j, "192.168.%s.%s"%(bridges.index(br)+1,i)))
igor@40 619 j+=1
igor@40 620 write_to(i,"\nend\n")
igor@33 621 return 0
igor@33 622
igor@33 623 def configure_ospf(doms=domains):
igor@40 624 for dom in doms:
igor@40 625 if domain_types[domains.index(dom)] == 'quagga':
igor@40 626 write_to(dom,"\n\nconf t\nrouter ospf\nnetwork 192.168.0.0/16 area 0\nend\n")
igor@40 627 else:
igor@40 628 write_to(dom,"\n\nena\nconf t\nrouter ospf 1\nnetwork 192.168.0.0 0.0.255.255 area 0\nend\n")
igor@33 629 return 0
igor@33 630
igor@49 631 def configure_hostname(doms=domains):
igor@49 632 for dom in doms:
igor@49 633 if domain_types[domains.index(dom)] == 'quagga':
igor@49 634 write_to(dom,"\n\nconf t\nhostname %s\nend\n" % dom)
igor@49 635 else:
igor@49 636 write_to(dom,"\n\nena\nconf t\nhostname %s\nend\n" % dom)
igor@49 637 return 0
igor@49 638
igor@49 639 def configure_logging_synchronous(doms=domains):
igor@49 640 for dom in domains:
igor@49 641 if domain_types[domains.index(dom)] == 'quagga':
igor@49 642 0
igor@49 643 else:
igor@49 644 write_to(dom,"\n\nena\nconf t\nline console 0\nlogging synchronous\nend\n")
igor@49 645 return 0
igor@49 646
igor@51 647 def configure_exec_timeout_0(doms=domains):
igor@51 648 for dom in domains:
igor@51 649 if domain_types[domains.index(dom)] == 'quagga':
igor@51 650 0
igor@51 651 else:
igor@52 652 write_to(dom,"\n\nena\nconf t\nline console 0\nexec-timeout 0\nend\n")
igor@51 653 return 0
igor@51 654
igor@55 655 def configure_no_cdp_log_mismatch_duplex(doms=domains):
igor@55 656 for dom in filter_by_type(domains,'xenomips'):
igor@55 657 write_to(dom,"\n\nena\nconf t\nno cdp log mismatch duplex\nend\n")
igor@55 658
igor@33 659 def configure_save(doms=domains):
igor@33 660 write_to(doms,"\nwr\n")
igor@33 661
igor@49 662 def configure_root(doms=domains):
igor@49 663 write_to(doms,"root\n")
igor@49 664
igor@0 665 #-----------------------------------------------------------------------
igor@0 666
igor@0 667
igor@0 668 def add_domain(name,type):
igor@0 669 domains.append(name)
igor@0 670 domain_types.append(type)
igor@0 671
igor@0 672 def brake_link(domain,bridge):
igor@0 673 broken_links.append([domain,bridge])
igor@0 674
igor@4 675 wt_timeout=0.5
igor@8 676 def write_to(screen,string,return_to_screen=""):
igor@2 677 """
igor@2 678 write_to(screen,string):
igor@2 679
igor@8 680 Type *string* to the specified screen(s).
igor@8 681 Screen may be specified with the number *screen*,
igor@8 682 with array of numbers,
igor@8 683 with array of names.
igor@2 684
igor@2 685 """
igor@5 686 screen_numbers=[] # number of the screens to write to
igor@5 687 if type(screen) == list:
igor@5 688 screen_numbers=map(lambda x: domains.index(x)+1, screen)
igor@5 689 elif type(screen) == int:
igor@5 690 screen_numbers=[screen]
igor@5 691 else:
igor@5 692 screen_numbers=[domains.index(screen)+1]
igor@5 693
igor@5 694 for screen_number in screen_numbers:
igor@5 695 run_command("screen -X select "+str(screen_number))
igor@5 696 time.sleep(wt_timeout)
igor@5 697 for line in string.splitlines():
igor@5 698 f=open('/tmp/xentaurbuf', 'w')
igor@5 699 f.write(line+"\n")
igor@5 700 f.close()
igor@5 701 run_command("screen -X readreg p /tmp/xentaurbuf")
igor@5 702 time.sleep(wt_timeout)
igor@5 703 run_command("nohup screen -X paste p >& /dev/null")
igor@5 704 time.sleep(wt_timeout)
igor@5 705
igor@8 706 if return_to_screen != "":
igor@8 707 run_command("screen -X select %s" % (return_to_screen))
igor@8 708 time.sleep(wt_timeout)
igor@0 709
igor@49 710 def filter_by_type(doms,type):
igor@49 711 """
igor@49 712 filter_by_type(doms,type)
igor@49 713
igor@49 714 Return only domains of *doms* that have specified *type*
igor@49 715 """
igor@49 716 return filter(lambda x: domain_types[domains.index(x)]==type,domains)
igor@49 717
igor@0 718 #-----------------------------------------------------------------------
igor@0 719
igor@5 720 cisco_fa01_up="""
igor@5 721 ena
igor@5 722 conf t
igor@5 723 int fa0/0
igor@49 724 duplex half
igor@5 725 no shutdown
igor@5 726 exit
igor@5 727 int fa1/0
igor@49 728 duplex half
igor@5 729 no shutdown
igor@5 730 exit
igor@5 731 exit
igor@5 732 exit
igor@5 733 """
igor@5 734
igor@5 735 cisco_set_ip_on_int="""
igor@5 736 interface fa%s/0
igor@5 737 no ip address
igor@5 738 ip address %s 255.255.255.0
igor@5 739 exit
igor@5 740 """
igor@5 741
igor@0 742 nodes=domains
igor@0 743
igor@56 744 create_objects()
igor@56 745
igor@56 746
igor@49 747 if len(sys.argv) == 2:
igor@49 748 if sys.argv[1] == 'start-all':
igor@0 749 start_all()
igor@49 750 elif sys.argv[1] == 'start-domains':
igor@49 751 start_domains()
igor@49 752 elif sys.argv[1] == 'start-bridges':
igor@49 753 start_bridges()
igor@49 754 elif sys.argv[1] == 'stop-all':
igor@0 755 stop_all()
igor@49 756 elif sys.argv[1] == 'stop-domains':
igor@49 757 stop_domains()
igor@49 758 elif sys.argv[1] == 'stop-bridges':
igor@49 759 stop_bridges()
igor@49 760 elif sys.argv[1] == 'restart-all':
igor@49 761 restart_all()
igor@0 762 elif sys.argv[1] == 'screen':
igor@50 763 screen()
igor@0 764 elif sys.argv[1] == 'graph':
igor@0 765 graph()
igor@0 766 elif sys.argv[1] == 'shell':
igor@0 767 shell()
igor@60 768 elif sys.argv[1] == 'info':
igor@60 769 info()
igor@49 770 else:
igor@49 771 show_usage()
igor@49 772 sys.exit(1)
igor@49 773 elif len(sys.argv) == 3:
igor@49 774 if sys.argv[1] == 'start':
igor@49 775 start_domain(sys.argv[2])
igor@49 776 elif sys.argv[1] == 'stop':
igor@49 777 stop_domain(sys.argv[2])
igor@49 778 elif sys.argv[1] == 'restart':
igor@49 779 stop_domain(sys.argv[2])
igor@49 780 start_domain(sys.argv[2])
igor@49 781 else:
igor@49 782 show_usage()
igor@49 783 sys.exit(1)
igor@0 784 else:
igor@0 785 show_usage()
igor@0 786 sys.exit(1)
igor@0 787
igor@0 788 sys.exit(0)
igor@0 789
igor@0 790