9 from optparse import OptionParser
11 from sfa.client.sfi import Sfi
12 from sfa.util.sfalogging import sfa_logger,sfa_logger_goes_to_console
13 import sfa.util.xmlrpcprotocol as xmlrpcprotocol
15 m_url_with_proto=re.compile("\w+://(?P<hostname>[\w\-\.]+):(?P<port>[0-9]+).*")
16 m_url_without_proto=re.compile("(?P<hostname>[\w\-\.]+):(?P<port>[0-9]+).*")
17 def url_to_hostname_port (url):
18 match=m_url_with_proto.match(url)
20 return (match.group('hostname'),match.group('port'))
21 match=m_url_without_proto.match(url)
23 return (match.group('hostname'),match.group('port'))
24 return ('undefined','???')
29 def __init__ (self,url):
31 (self.hostname,self.port)=url_to_hostname_port(url)
32 self.ip=socket.gethostbyname(self.hostname)
37 self.hostname="undefined"
43 return "http://%s:%s/"%(self.hostname,self.port)
46 return "%s:%s"%(self.ip,self.port)
48 # connect to server and trigger GetVersion
49 def get_version(self):
52 # dummy to meet Sfi's expectations for its 'options' field
55 options=DummyOptions()
60 key_file = client.get_key_file()
61 cert_file = client.get_cert_file(key_file)
62 url="http://%s:%s/"%(self.hostname,self.port)
63 sfa_logger().info('issuing get version at %s'%url)
64 server=xmlrpcprotocol.get_server(url, key_file, cert_file, options)
65 self._version=server.GetVersion()
66 sfa_logger().info("get_version at %s returned %r"%(url,self._version))
68 sfa_logger().info("get_version at %s failed"%(url))
73 abbrevs = {"registry": "REG", "slicemgr":"SM", "aggregate":"AM"}
74 shapes = {"registry": "diamond", "slicemgr":"ellipse", "aggregate":"box", 'default':'plaintext'}
77 version=self.get_version()
78 if 'hrn' not in version: return self.url()
80 interface=version['interface']
81 abbrev=Interface.abbrevs.get(interface,"XXX")
82 result="%s %s"%(hrn,abbrev)
83 if 'code_tag' in version: result += " %s"%version['code_tag']
87 default=Interface.shapes['default']
89 version=self.get_version()
90 return Interface.shapes.get(version['interface'],default)
96 # provide the entry points (a list of interfaces)
100 def graph (self,entry_points):
101 graph=pygraphviz.AGraph(directed=True)
102 self.scan(entry_points,graph)
105 # scan from the given interfaces as entry points
106 def scan(self,interfaces,graph):
107 if not isinstance(interfaces,list):
108 interfaces=[interfaces]
110 # remember node to interface mapping
112 # add entry points right away using the interface uid's as a key
115 sfa_logger().info("adding initial node %s"%i.uid())
116 graph.add_node(i.uid())
117 node2interface[graph.get_node(i.uid())]=i
119 # keep on looping until we reach a fixed point
120 # don't worry about abels and shapes that will get fixed later on
122 for interface in to_scan:
123 # performing xmlrpc call
124 version=interface.get_version()
126 # proceed with neighbours
127 for (next_name,next_url) in version['peers'].items():
128 next_interface=Interface(next_url)
129 # locate or create node in graph
131 # if found, we're good with this one
132 next_node=graph.get_node(next_interface.uid())
134 # otherwise, let's move on with it
135 graph.add_node(next_interface.uid())
136 next_node=graph.get_node(next_interface.uid())
137 node2interface[next_node]=next_interface
138 to_scan.append(next_interface)
139 graph.add_edge(interface.uid(),next_interface.uid())
140 scanned.append(interface)
141 to_scan.remove(interface)
142 # we've scanned the whole graph, let's get the labels and shapes right
143 for node in graph.nodes():
144 interface=node2interface.get(node,None)
146 node.attr['label']=interface.get_name()
147 node.attr['shape']=interface.get_shape()
149 sfa_logger().info("MISSED interface with node %s"%node)
152 default_entry_points=["http://www.planet-lab.eu:12345/"]
153 default_outfiles=['sfa.png']
156 sfa_logger_goes_to_console()
157 parser=OptionParser()
158 parser.add_option("-e","--entry",action='append',dest='entry_points',default=[],
159 help="Specify entry points - defaults are %r"%default_entry_points)
160 parser.add_option("-o","--output",action='append',dest='outfiles',default=[],
161 help="Output filenames - defaults are %r"%default_outfiles)
162 (options,args)=parser.parse_args()
166 if not options.entry_points:
167 options.entry_points=default_entry_points
168 if not options.outfiles:
169 options.outfiles=default_outfiles
171 entries = [ Interface(entry) for entry in options.entry_points ]
172 g=scanner.graph(entries)
173 sfa_logger().info("creating layout")
175 for outfile in options.outfiles:
176 sfa_logger().info("drawing in %s"%outfile)
178 sfa_logger().info("done")
180 if __name__ == '__main__':