xentaur: aaf034af3a35 xentaur.py

xentaur

view xentaur.py @ 66:aaf034af3a35

Merge of Xgurulla into Xentaur code. Not completed yet!!!

Now Xentaur can work with Amazon EC2,
but only with. Local domains management
is switched off temporarily.

author	Igor Chubin <igor@chub.in>
date	Sat Jan 09 20:20:08 2010 +0200 (2010-01-09)
parents	cf823d27b029
children	6c145935ece5

line source

1 #!/usr/bin/python

2 # vim: set fileencoding=utf-8 :

4 import sys,os,time

5 from IPython.Shell import IPShellEmbed

7 path_xentaur="%s/hg/xentaur" % os.environ['HOME']

8 path_shapes=path_xentaur+'/shapes'

9 path_scripts=path_xentaur+'/files'

11 network='net1'

13 path_network=os.environ['HOME']+"/.xentaur/"+network

14 if not os.path.exists(path_network):

15 os.makedirs(path_network)

16 screenrc=path_network+"/.screenrc_xentaur"

18 sys.path.append('/etc/xen')

19 sys.path.append(path_xentaur)

20 sys.path.append(path_network)

21 sys.path.append('.')

23 node_object={}

24 link_object={}

25 bridge_object={}

26 ec2_node={}

28 domain='dyn1'

29 from xendomain import *

31 bridges_turned_down=[]

33 def run(program, *args):

34 pid = os.fork()

35 if not pid:

36 os.execvp(program, (program,) + args)

37 return os.wait()[0]

39 def run_command(line):

40 #cmds=line.split()

41 #run(cmds[0],*cmds[1:])

42 run("/bin/sh", "-c", line)

44 def run_command_return_stdout(line):

45 p = os.popen(line)

46 output = p.read()

47 p.close()

48 return output

50 ################################################################################

51 #Xentaur command-line commands

53 ## Start

55 def start_bridges():

56 unbound_bridges=set(bridges)-set(real_bridges)

57 script=""

58 script="\n".join(map(lambda x: "sudo brctl show | awk '{print $1}' | grep -qx "+x+" || sudo brctl addbr "+x, unbound_bridges))

59 script+="\n"+"\n".join(map(lambda x: "sudo brctl stp "+x+" off", unbound_bridges))

60 script+="\n"+"\n".join(map(lambda x: "sudo ip link set "+x+" up", unbound_bridges))

62 print """#!/bin/sh

63 # create unbound bridges

64 %s

65 """ % (script)

67 def start_domain(domain):

68 print "sudo xm create "+path_xentaur+"/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"

70 def start_domains(doms=domains):

71 for domain in doms:

72 if not domain in real_nodes:

73 start_domain(domain)

75 def start_all():

76 graph()

77 screen()

78 html()

79 start_bridges()

80 start_domains()

82 def ec2_assign_nodes_to_instances():

83 domain_number=0

84 for dom in domains:

85 node_name=network+"-node%s"%(domain_number/2)

86 ec2_node[dom]=node_name

87 domain_number+=1

89 def make_start_emulators():

90 s=""

91 #s+="cat screenrc_template > screenrc_$NETWORK\n"

92 #s+="echo \"screen -t console 0 sh -c 'cd ~/xentaur; ./xentaur.py shell'\" >> screenrc_$NETWORK\n"

93 ios="/mnt/ios/C7200-AD.BIN"

94 ec2_assign_nodes_to_instances()

95 for dom in domains:

96 node_name=ec2_node[dom]

97 i=0

98 line='dynamips '+ios

99 line2=''

100 for iface in vbridges_table[dom]:

101 bridge=vbridges_table[dom][i]

102 if i>0:

103 line += " -p %s:PA-FE-TX "%i

104 line += " -s %s:0:tap:%s_%s "% (i,dom,iface)

105 line2 += "sleep 2; brctl addif %s %s; "%(bridge, dom+"_"+iface)

106 line2 +='ifconfig %s up; ifconfig %s promisc ;' %(bridge, bridge)

107 line2 += "ifconfig %s_%s up; "%(dom, iface)

108 i+=1

109 s += "ssh %s \"mkdir -p /mnt/%s; cd /mnt/%s; screen -S %s- -d -m %s\"\n"%(node_name,dom,dom,dom,line)

110 s += "ssh %s \"%s\"\n" % (node_name, line2)

111 #s += "echo \"screen %s ssh -t %s 'screen -r %s-'\" >> screenrc_%s\n" % (domain_number, node_name, dom, network)

112 #s += "echo \"screen -t %s %s connect-ec2-session INSTANCE EC2_NODE %s %s %s\" >> screenrc_$NETWORK\n" % (dom, domain_number+1, node_name, network, dom)

113 return s

114

115 def make_start_instances_config(net=network):

116 f = open(path_network+"/start-instances-"+network, "w");

117 f.write("INSTANCES_NUMBER=%s\n"%((len(domains)+1)/2))

118 f.write("start_emulators()\n{\n%s\n}\n"%make_start_emulators())

119 f.close()

120

121 def start_network(net=network):

122 make_start_instances_config()

123 #run_command("cd %s; env NETWORK=%s sh %s instances start" % (path_network, network, path_scrips+"/ec2-instances"))

124