c7919d5ac825f2bffae5a65fb32a6ae6994b82a7
[nepi.git] / src / nepi / core / execute.py
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3
4 from nepi.core.attributes import Attribute, AttributesMap
5 from nepi.core.connector import ConnectorTypeBase
6 from nepi.util import validation
7 from nepi.util.constants import STATUS_FINISHED, TIME_NOW
8 from nepi.util.parser._xml import XmlExperimentParser
9 import sys
10 import re
11 import threading
12 import ConfigParser
13 import os
14 import collections
15 import functools
16
17 ATTRIBUTE_PATTERN_BASE = re.compile(r"\{#\[(?P<label>[-a-zA-Z0-9._]*)\](?P<expr>(?P<component>\.addr\[[0-9]+\]|\.route\[[0-9]+\]|\.trace\[[0-9]+\])?.\[(?P<attribute>[-a-zA-Z0-9._]*)\])#}")
18 ATTRIBUTE_PATTERN_GUID_SUB = r"{#[%(guid)s]%(expr)s#}"
19 COMPONENT_PATTERN = re.compile(r"(?P<kind>[a-z]*)\[(?P<index>.*)\]")
20
21 class ConnectorType(ConnectorTypeBase):
22     def __init__(self, testbed_id, factory_id, name, max = -1, min = 0):
23         super(ConnectorType, self).__init__(testbed_id, factory_id, name, max, min)
24         # from_connections -- connections where the other connector is the "From"
25         # to_connections -- connections where the other connector is the "To"
26         # keys in the dictionary correspond to the 
27         # connector_type_id for possible connections. The value is a tuple:
28         # (can_cross, connect)
29         # can_cross: indicates if the connection is allowed accros different
30         #    testbed instances
31         # code: is the connection function to be invoked when the elements
32         #    are connected
33         self._from_connections = dict()
34         self._to_connections = dict()
35
36     def add_from_connection(self, testbed_id, factory_id, name, can_cross, 
37             init_code, compl_code):
38         type_id = self.make_connector_type_id(testbed_id, factory_id, name)
39         self._from_connections[type_id] = (can_cross, init_code, compl_code)
40
41     def add_to_connection(self, testbed_id, factory_id, name, can_cross, 
42             init_code, compl_code):
43         type_id = self.make_connector_type_id(testbed_id, factory_id, name)
44         self._to_connections[type_id] = (can_cross, init_code, compl_code)
45
46     def can_connect(self, testbed_id, factory_id, name, count, 
47             must_cross):
48         connector_type_id = self.make_connector_type_id(testbed_id, factory_id, name)
49         for lookup_type_id in self._type_resolution_order(connector_type_id):
50             if lookup_type_id in self._from_connections:
51                 (can_cross, init_code, compl_code) = self._from_connections[lookup_type_id]
52             elif lookup_type_id in self._to_connections:
53                 (can_cross, init_code, compl_code) = self._to_connections[lookup_type_id]
54             else:
55                 # keep trying
56                 continue
57             return not must_cross or can_cross
58         else:
59             return False
60
61     def _connect_to_code(self, testbed_id, factory_id, name,
62             must_cross):
63         connector_type_id = self.make_connector_type_id(testbed_id, factory_id, name)
64         for lookup_type_id in self._type_resolution_order(connector_type_id):
65             if lookup_type_id in self._to_connections:
66                 (can_cross, init_code, compl_code) = self._to_connections[lookup_type_id]
67                 if not must_cross or can_cross:
68                     return (init_code, compl_code)
69         else:
70             return (False, False)
71     
72     def connect_to_init_code(self, testbed_id, factory_id, name, must_cross):
73         return self._connect_to_code(testbed_id, factory_id, name, must_cross)[0]
74
75     def connect_to_compl_code(self, testbed_id, factory_id, name, must_cross):
76         return self._connect_to_code(testbed_id, factory_id, name, must_cross)[1]
77
78 class Factory(AttributesMap):
79     def __init__(self, factory_id, create_function, start_function, 
80             stop_function, status_function, 
81             configure_function, preconfigure_function,
82             prestart_function,
83             allow_addresses = False, has_addresses = False,
84             allow_routes = False, has_routes = False):
85         super(Factory, self).__init__()
86         self._factory_id = factory_id
87         self._allow_addresses = bool(allow_addresses)
88         self._allow_routes = bool(allow_routes)
89         self._has_addresses = bool(has_addresses) or self._allow_addresses
90         self._has_routes = bool(has_routes) or self._allow_routes
91         self._create_function = create_function
92         self._start_function = start_function
93         self._stop_function = stop_function
94         self._status_function = status_function
95         self._configure_function = configure_function
96         self._preconfigure_function = preconfigure_function
97         self._prestart_function = prestart_function
98         self._connector_types = dict()
99         self._traces = list()
100         self._tags = list()
101         self._box_attributes = AttributesMap()
102
103     @property
104     def factory_id(self):
105         return self._factory_id
106
107     @property
108     def allow_addresses(self):
109         return self._allow_addresses
110
111     @property
112     def allow_routes(self):
113         return self._allow_routes
114
115     @property
116     def has_addresses(self):
117         return self._has_addresses
118
119     @property
120     def has_routes(self):
121         return self._has_routes
122
123     @property
124     def box_attributes(self):
125         return self._box_attributes
126
127     @property
128     def create_function(self):
129         return self._create_function
130
131     @property
132     def prestart_function(self):
133         return self._prestart_function
134
135     @property
136     def start_function(self):
137         return self._start_function
138
139     @property
140     def stop_function(self):
141         return self._stop_function
142
143     @property
144     def status_function(self):
145         return self._status_function
146
147     @property
148     def configure_function(self):
149         return self._configure_function
150
151     @property
152     def preconfigure_function(self):
153         return self._preconfigure_function
154
155     @property
156     def traces(self):
157         return self._traces
158
159     @property
160     def tags(self):
161         return self._tags
162
163     def connector_type(self, name):
164         return self._connector_types[name]
165
166     def add_connector_type(self, connector_type):
167         self._connector_types[connector_type.name] = connector_type
168
169     def add_trace(self, trace_id):
170         self._traces.append(trace_id)
171
172     def add_tag(self, tag_id):
173         self._tags.append(tag_id)
174
175     def add_box_attribute(self, name, help, type, value = None, range = None,
176         allowed = None, flags = Attribute.NoFlags, validation_function = None,
177         category = None):
178         self._box_attributes.add_attribute(name, help, type, value, range, 
179                 allowed, flags, validation_function, category)
180
181 class TestbedController(object):
182     def __init__(self, testbed_id, testbed_version):
183         self._testbed_id = testbed_id
184         self._testbed_version = testbed_version
185
186     @property
187     def testbed_id(self):
188         return self._testbed_id
189
190     @property
191     def testbed_version(self):
192         return self._testbed_version
193
194     @property
195     def guids(self):
196         raise NotImplementedError
197
198     def defer_configure(self, name, value):
199         """Instructs setting a configuartion attribute for the testbed instance"""
200         raise NotImplementedError
201
202     def defer_create(self, guid, factory_id):
203         """Instructs creation of element """
204         raise NotImplementedError
205
206     def defer_create_set(self, guid, name, value):
207         """Instructs setting an initial attribute on an element"""
208         raise NotImplementedError
209
210     def defer_factory_set(self, guid, name, value):
211         """Instructs setting an attribute on a factory"""
212         raise NotImplementedError
213
214     def defer_connect(self, guid1, connector_type_name1, guid2, 
215             connector_type_name2): 
216         """Instructs creation of a connection between the given connectors"""
217         raise NotImplementedError
218
219     def defer_cross_connect(self, 
220             guid, connector_type_name,
221             cross_guid, cross_testbed_guid,
222             cross_testbed_id, cross_factory_id,
223             cross_connector_type_name):
224         """
225         Instructs creation of a connection between the given connectors 
226         of different testbed instances
227         """
228         raise NotImplementedError
229
230     def defer_add_trace(self, guid, trace_id):
231         """Instructs the addition of a trace"""
232         raise NotImplementedError
233
234     def defer_add_address(self, guid, address, netprefix, broadcast): 
235         """Instructs the addition of an address"""
236         raise NotImplementedError
237
238     def defer_add_route(self, guid, destination, netprefix, nexthop):
239         """Instructs the addition of a route"""
240         raise NotImplementedError
241
242     def do_setup(self):
243         """After do_setup the testbed initial configuration is done"""
244         raise NotImplementedError
245
246     def do_create(self):
247         """
248         After do_create all instructed elements are created and 
249         attributes setted
250         """
251         raise NotImplementedError
252
253     def do_connect_init(self):
254         """
255         After do_connect_init all internal connections between testbed elements
256         are initiated
257         """
258         raise NotImplementedError
259
260     def do_connect_compl(self):
261         """
262         After do_connect all internal connections between testbed elements
263         are completed
264         """
265         raise NotImplementedError
266
267     def do_preconfigure(self):
268         """
269         Done just before resolving netrefs, after connection, before cross connections,
270         useful for early stages of configuration, for setting up stuff that might be
271         required for netref resolution.
272         """
273         raise NotImplementedError
274
275     def do_configure(self):
276         """After do_configure elements are configured"""
277         raise NotImplementedError
278
279     def do_prestart(self):
280         """Before do_start elements are prestart-configured"""
281         raise NotImplementedError
282
283     def do_cross_connect_init(self, cross_data):
284         """
285         After do_cross_connect_init initiation of all external connections 
286         between different testbed elements is performed
287         """
288         raise NotImplementedError
289
290     def do_cross_connect_compl(self, cross_data):
291         """
292         After do_cross_connect_compl completion of all external connections 
293         between different testbed elements is performed
294         """
295         raise NotImplementedError
296
297     def start(self):
298         raise NotImplementedError
299
300     def stop(self):
301         raise NotImplementedError
302
303     def set(self, guid, name, value, time = TIME_NOW):
304         raise NotImplementedError
305
306     def get(self, guid, name, time = TIME_NOW):
307         raise NotImplementedError
308     
309     def get_route(self, guid, index, attribute):
310         """
311         Params:
312             
313             guid: guid of box to query
314             index: number of routing entry to fetch
315             attribute: one of Destination, NextHop, NetPrefix
316         """
317         raise NotImplementedError
318
319     def get_address(self, guid, index, attribute='Address'):
320         """
321         Params:
322             
323             guid: guid of box to query
324             index: number of inteface to select
325             attribute: one of Address, NetPrefix, Broadcast
326         """
327         raise NotImplementedError
328
329     def get_attribute_list(self, guid):
330         raise NotImplementedError
331
332     def get_tags(self, guid):
333         raise NotImplementedError
334
335     def action(self, time, guid, action):
336         raise NotImplementedError
337
338     def status(self, guid):
339         raise NotImplementedError
340
341     def trace(self, guid, trace_id, attribute='value'):
342         raise NotImplementedError
343
344     def shutdown(self):
345         raise NotImplementedError
346
347 class ExperimentController(object):
348     def __init__(self, experiment_xml, root_dir):
349         self._experiment_xml = experiment_xml
350         self._testbeds = dict()
351         self._deployment_config = dict()
352         self._netrefs = collections.defaultdict(set)
353         self._testbed_netrefs = collections.defaultdict(set)
354         self._cross_data = dict()
355         self._root_dir = root_dir
356         self._netreffed_testbeds = set()
357         self._guids_in_testbed_cache = dict()
358
359         self.persist_experiment_xml()
360
361     @property
362     def experiment_xml(self):
363         return self._experiment_xml
364
365     @property
366     def guids(self):
367         guids = list()
368         for testbed_guid in self._testbeds.keys():
369             _guids = self._guids_in_testbed(testbed_guid)
370             if _guids:
371                 guids.extend(_guids)
372         return guids
373
374     def persist_experiment_xml(self):
375         xml_path = os.path.join(self._root_dir, "experiment.xml")
376         f = open(xml_path, "w")
377         f.write(self._experiment_xml)
378         f.close()
379
380     def trace(self, guid, trace_id, attribute='value'):
381         testbed = self._testbed_for_guid(guid)
382         if testbed != None:
383             return testbed.trace(guid, trace_id, attribute)
384         raise RuntimeError("No element exists with guid %d" % guid)    
385
386     @staticmethod
387     def _parallel(callables):
388         excs = []
389         def wrap(callable):
390             @functools.wraps(callable)
391             def wrapped(*p, **kw):
392                 try:
393                     callable(*p, **kw)
394                 except:
395                     import traceback
396                     traceback.print_exc(file=sys.stderr)
397                     excs.append(sys.exc_info())
398             return wrapped
399         threads = [ threading.Thread(target=wrap(callable)) for callable in callables ]
400         for thread in threads:
401             thread.start()
402         for thread in threads:
403             thread.join()
404         for exc in excs:
405             eTyp, eVal, eLoc = exc
406             raise eTyp, eVal, eLoc
407
408     def start(self):
409         parser = XmlExperimentParser()
410         data = parser.from_xml_to_data(self._experiment_xml)
411         
412         self._init_testbed_controllers(data)
413         
414         # persist testbed connection data, for potential recovery
415         self._persist_testbed_proxies()
416         
417         def steps_to_configure(self, allowed_guids):
418             # perform setup in parallel for all test beds,
419             # wait for all threads to finish
420             self._parallel([testbed.do_setup 
421                             for guid,testbed in self._testbeds.iteritems()
422                             if guid in allowed_guids])
423             
424             # perform create-connect in parallel, wait
425             # (internal connections only)
426             self._parallel([testbed.do_create
427                             for guid,testbed in self._testbeds.iteritems()
428                             if guid in allowed_guids])
429
430             self._parallel([testbed.do_connect_init
431                             for guid,testbed in self._testbeds.iteritems()
432                             if guid in allowed_guids])
433
434             self._parallel([testbed.do_connect_compl
435                             for guid,testbed in self._testbeds.iteritems()
436                             if guid in allowed_guids])
437
438             self._parallel([testbed.do_preconfigure
439                             for guid,testbed in self._testbeds.iteritems()
440                             if guid in allowed_guids])
441             
442         steps_to_configure(self, self._testbeds)
443         
444         if self._netreffed_testbeds:
445             # initally resolve netrefs
446             self.do_netrefs(data, fail_if_undefined=False)
447             
448             # rinse and repeat, for netreffed testbeds
449             netreffed_testbeds = set(self._netreffed_testbeds)
450
451             self._init_testbed_controllers(data)
452             
453             # persist testbed connection data, for potential recovery
454             self._persist_testbed_proxies()
455
456             # configure dependant testbeds
457             steps_to_configure(self, netreffed_testbeds)
458             
459         # final netref step, fail if anything's left unresolved
460         self.do_netrefs(data, fail_if_undefined=True)
461         
462         self._program_testbed_cross_connections(data)
463         
464         # perform do_configure in parallel for al testbeds
465         # (it's internal configuration for each)
466         self._parallel([testbed.do_configure
467                         for testbed in self._testbeds.itervalues()])
468
469         
470         #print >>sys.stderr, "DO IT"
471         #import time
472         #time.sleep(60)
473         
474         # cross-connect (cannot be done in parallel)
475         for guid, testbed in self._testbeds.iteritems():
476             cross_data = self._get_cross_data(guid)
477             testbed.do_cross_connect_init(cross_data)
478         for guid, testbed in self._testbeds.iteritems():
479             cross_data = self._get_cross_data(guid)
480             testbed.do_cross_connect_compl(cross_data)
481        
482         # Last chance to configure (parallel on all testbeds)
483         self._parallel([testbed.do_prestart
484                         for testbed in self._testbeds.itervalues()])
485
486         # After this point no new elements will be craeted. Cleaning cache for safety.
487         self._guids_in_testbed_cache = dict()
488
489         # start experiment (parallel start on all testbeds)
490         self._parallel([testbed.start
491                         for testbed in self._testbeds.itervalues()])
492
493     def _persist_testbed_proxies(self):
494         TRANSIENT = ('Recover',)
495         
496         # persist access configuration for all testbeds, so that
497         # recovery mode can reconnect to them if it becomes necessary
498         conf = ConfigParser.RawConfigParser()
499         for testbed_guid, testbed_config in self._deployment_config.iteritems():
500             testbed_guid = str(testbed_guid)
501             conf.add_section(testbed_guid)
502             for attr in testbed_config.attributes_list:
503                 if attr not in TRANSIENT:
504                     conf.set(testbed_guid, attr, 
505                         testbed_config.get_attribute_value(attr))
506         
507         f = open(os.path.join(self._root_dir, 'deployment_config.ini'), 'w')
508         conf.write(f)
509         f.close()
510     
511     def _load_testbed_proxies(self):
512         TYPEMAP = {
513             STRING : 'get',
514             INTEGER : 'getint',
515             FLOAT : 'getfloat',
516             BOOLEAN : 'getboolean',
517         }
518         
519         # deferred import because proxy needs
520         # our class definitions to define proxies
521         import nepi.util.proxy as proxy
522         
523         conf = ConfigParser.RawConfigParser()
524         conf.read(os.path.join(self._root_dir, 'deployment_config.ini'))
525         for testbed_guid in conf.sections():
526             testbed_config = proxy.AccessConfiguration()
527             for attr in conf.options(testbed_guid):
528                 testbed_config.set_attribute_value(attr, 
529                     conf.get(testbed_guid, attr) )
530                 
531             testbed_guid = str(testbed_guid)
532             conf.add_section(testbed_guid)
533             for attr in testbed_config.attributes_list:
534                 if attr not in TRANSIENT:
535                     getter = getattr(conf, TYPEMAP.get(
536                         testbed_config.get_attribute_type(attr),
537                         'get') )
538                     testbed_config.set_attribute_value(
539                         testbed_guid, attr, getter(attr))
540     
541     def _unpersist_testbed_proxies(self):
542         try:
543             os.remove(os.path.join(self._root_dir, 'deployment_config.ini'))
544         except:
545             # Just print exceptions, this is just cleanup
546             import traceback
547             traceback.print_exc(file=sys.stderr)
548
549     def stop(self):
550        for testbed in self._testbeds.values():
551            testbed.stop()
552        self._unpersist_testbed_proxies()
553    
554     def recover(self):
555         # reload perviously persisted testbed access configurations
556         self._load_testbed_proxies()
557         
558         # recreate testbed proxies by reconnecting only
559         self._init_testbed_controllers(recover = True)
560         
561         # another time, for netrefs
562         self._init_testbed_controllers(recover = True)
563
564     def is_finished(self, guid):
565         testbed = self._testbed_for_guid(guid)
566         if testbed != None:
567             return testbed.status(guid) == STATUS_FINISHED
568         raise RuntimeError("No element exists with guid %d" % guid)    
569
570     def set(self, guid, name, value, time = TIME_NOW):
571         testbed = self._testbed_for_guid(guid)
572         if testbed != None:
573             testbed.set(guid, name, value, time)
574         raise RuntimeError("No element exists with guid %d" % guid)    
575
576     def get(self, guid, name, time = TIME_NOW):
577         testbed = self._testbed_for_guid(guid)
578         if testbed != None:
579             return testbed.get(guid, name, time)
580         raise RuntimeError("No element exists with guid %d" % guid)    
581
582     def get_tags(self, guid):
583         testbed = self._testbed_for_guid(guid)
584         if testbed != None:
585             return testbed.get_tags(guid)
586         raise RuntimeError("No element exists with guid %d" % guid)    
587
588     def shutdown(self):
589         for testbed in self._testbeds.values():
590             testbed.shutdown()
591
592     def _testbed_for_guid(self, guid):
593         for testbed_guid in self._testbeds.keys():
594             if guid in self._guids_in_testbed(testbed_guid):
595                 return self._testbeds[testbed_guid]
596         return None
597
598     def _guids_in_testbed(self, testbed_guid):
599         if testbed_guid not in self._testbeds:
600             return set()
601         if testbed_guid not in self._guids_in_testbed_cache:
602             self._guids_in_testbed_cache[testbed_guid] = \
603                 set(self._testbeds[testbed_guid].guids)
604         return self._guids_in_testbed_cache[testbed_guid]
605
606     @staticmethod
607     def _netref_component_split(component):
608         match = COMPONENT_PATTERN.match(component)
609         if match:
610             return match.group("kind"), match.group("index")
611         else:
612             return component, None
613
614     _NETREF_COMPONENT_GETTERS = {
615         'addr':
616             lambda testbed, guid, index, name: 
617                 testbed.get_address(guid, int(index), name),
618         'route' :
619             lambda testbed, guid, index, name: 
620                 testbed.get_route(guid, int(index), name),
621         'trace' :
622             lambda testbed, guid, index, name: 
623                 testbed.trace(guid, index, name),
624         '' : 
625             lambda testbed, guid, index, name: 
626                 testbed.get(guid, name),
627     }
628     
629     def resolve_netref_value(self, value, failval = None):
630         match = ATTRIBUTE_PATTERN_BASE.search(value)
631         if match:
632             label = match.group("label")
633             if label.startswith('GUID-'):
634                 ref_guid = int(label[5:])
635                 if ref_guid:
636                     expr = match.group("expr")
637                     component = (match.group("component") or "")[1:] # skip the dot
638                     attribute = match.group("attribute")
639                     
640                     # split compound components into component kind and index
641                     # eg: 'addr[0]' -> ('addr', '0')
642                     component, component_index = self._netref_component_split(component)
643
644                     # find object and resolve expression
645                     for ref_testbed_guid, ref_testbed in self._testbeds.iteritems():
646                         if component not in self._NETREF_COMPONENT_GETTERS:
647                             raise ValueError, "Malformed netref: %r - unknown component" % (expr,)
648                         elif ref_guid not in self._guids_in_testbed(ref_testbed_guid):
649                             pass
650                         else:
651                             ref_value = self._NETREF_COMPONENT_GETTERS[component](
652                                 ref_testbed, ref_guid, component_index, attribute)
653                             if ref_value:
654                                 return value.replace(match.group(), ref_value)
655         # couldn't find value
656         return failval
657     
658     def do_netrefs(self, data, fail_if_undefined = False):
659         # element netrefs
660         for (testbed_guid, guid), attrs in self._netrefs.items():
661             testbed = self._testbeds.get(testbed_guid)
662             if testbed is not None:
663                 for name in set(attrs):
664                     value = testbed.get(guid, name)
665                     if isinstance(value, basestring):
666                         ref_value = self.resolve_netref_value(value)
667                         if ref_value is not None:
668                             testbed.set(guid, name, ref_value)
669                             attrs.remove(name)
670                         elif fail_if_undefined:
671                             raise ValueError, "Unresolvable netref in: %r=%r" % (name,value,)
672                 if not attrs:
673                     del self._netrefs[(testbed_guid, guid)]
674         
675         # testbed netrefs
676         for testbed_guid, attrs in self._testbed_netrefs.items():
677             tb_data = dict(data.get_attribute_data(testbed_guid))
678             if data:
679                 for name in set(attrs):
680                     value = tb_data.get(name)
681                     if isinstance(value, basestring):
682                         ref_value = self.resolve_netref_value(value)
683                         if ref_value is not None:
684                             data.set_attribute_data(testbed_guid, name, ref_value)
685                             attrs.remove(name)
686                         elif fail_if_undefined:
687                             raise ValueError, "Unresolvable netref in: %r" % (value,)
688                 if not attrs:
689                     del self._testbed_netrefs[testbed_guid]
690         
691
692     def _init_testbed_controllers(self, data, recover = False):
693         blacklist_testbeds = set(self._testbeds)
694         element_guids = list()
695         label_guids = dict()
696         data_guids = data.guids
697
698         # create testbed controllers
699         for guid in data_guids:
700             if data.is_testbed_data(guid):
701                 if guid not in self._testbeds:
702                     self._create_testbed_controller(guid, data, element_guids,
703                             recover)
704             else:
705                 (testbed_guid, factory_id) = data.get_box_data(guid)
706                 if testbed_guid not in blacklist_testbeds:
707                     element_guids.append(guid)
708                     label = data.get_attribute_data(guid, "label")
709                     if label is not None:
710                         if label in label_guids:
711                             raise RuntimeError, "Label %r is not unique" % (label,)
712                         label_guids[label] = guid
713
714         # replace references to elements labels for its guid
715         self._resolve_labels(data, data_guids, label_guids)
716     
717         # program testbed controllers
718         if not recover:
719             self._program_testbed_controllers(element_guids, data)
720
721     def _resolve_labels(self, data, data_guids, label_guids):
722         netrefs = self._netrefs
723         testbed_netrefs = self._testbed_netrefs
724         for guid in data_guids:
725             for name, value in data.get_attribute_data(guid):
726                 if isinstance(value, basestring):
727                     match = ATTRIBUTE_PATTERN_BASE.search(value)
728                     if match:
729                         label = match.group("label")
730                         if not label.startswith('GUID-'):
731                             ref_guid = label_guids.get(label)
732                             if ref_guid is not None:
733                                 value = ATTRIBUTE_PATTERN_BASE.sub(
734                                     ATTRIBUTE_PATTERN_GUID_SUB % dict(
735                                         guid = 'GUID-%d' % (ref_guid,),
736                                         expr = match.group("expr"),
737                                         label = label), 
738                                     value)
739                                 data.set_attribute_data(guid, name, value)
740                                 
741                                 # memorize which guid-attribute pairs require
742                                 # postprocessing, to avoid excessive controller-testbed
743                                 # communication at configuration time
744                                 # (which could require high-latency network I/O)
745                                 if not data.is_testbed_data(guid):
746                                     (testbed_guid, factory_id) = data.get_box_data(guid)
747                                     netrefs[(testbed_guid, guid)].add(name)
748                                 else:
749                                     testbed_netrefs[guid].add(name)
750
751     def _create_testbed_controller(self, guid, data, element_guids, recover):
752         (testbed_id, testbed_version) = data.get_testbed_data(guid)
753         deployment_config = self._deployment_config.get(guid)
754         
755         # deferred import because proxy needs
756         # our class definitions to define proxies
757         import nepi.util.proxy as proxy
758         
759         if deployment_config is None:
760             # need to create one
761             deployment_config = proxy.AccessConfiguration()
762             
763             for (name, value) in data.get_attribute_data(guid):
764                 if value is not None and deployment_config.has_attribute(name):
765                     # if any deployment config attribute has a netref, we can't
766                     # create this controller yet
767                     if isinstance(value, basestring) and ATTRIBUTE_PATTERN_BASE.search(value):
768                         # remember to re-issue this one
769                         self._netreffed_testbeds.add(guid)
770                         return
771                     
772                     # copy deployment config attribute
773                     deployment_config.set_attribute_value(name, value)
774             
775             # commit config
776             self._deployment_config[guid] = deployment_config
777         
778         if deployment_config is not None:
779             # force recovery mode 
780             deployment_config.set_attribute_value("recover",recover)
781         
782         testbed = proxy.create_testbed_controller(testbed_id, 
783                 testbed_version, deployment_config)
784         for (name, value) in data.get_attribute_data(guid):
785             testbed.defer_configure(name, value)
786         self._testbeds[guid] = testbed
787         if guid in self._netreffed_testbeds:
788             self._netreffed_testbeds.remove(guid)
789
790     def _program_testbed_controllers(self, element_guids, data):
791         for guid in element_guids:
792             (testbed_guid, factory_id) = data.get_box_data(guid)
793             testbed = self._testbeds.get(testbed_guid)
794             if testbed:
795                 testbed.defer_create(guid, factory_id)
796                 for (name, value) in data.get_attribute_data(guid):
797                     # Try to resolve create-time netrefs, if possible
798                     if isinstance(value, basestring) and ATTRIBUTE_PATTERN_BASE.search(value):
799                         try:
800                             nuvalue = self.resolve_netref_value(value)
801                         except:
802                             # Any trouble means we're not in shape to resolve the netref yet
803                             nuvalue = None
804                         if nuvalue is not None:
805                             # Only if we succeed we remove the netref deferral entry
806                             value = nuvalue
807                             data.set_attribute_data(guid, name, value)
808                             if (testbed_guid, guid) in self._netrefs:
809                                 self._netrefs[(testbed_guid, guid)].discard(name)
810                     testbed.defer_create_set(guid, name, value)
811
812         for guid in element_guids: 
813             (testbed_guid, factory_id) = data.get_box_data(guid)
814             testbed = self._testbeds.get(testbed_guid)
815             if testbed:
816                 for (connector_type_name, cross_guid, cross_connector_type_name) \
817                         in data.get_connection_data(guid):
818                     (testbed_guid, factory_id) = data.get_box_data(guid)
819                     (cross_testbed_guid, cross_factory_id) = data.get_box_data(
820                             cross_guid)
821                     if testbed_guid == cross_testbed_guid:
822                         testbed.defer_connect(guid, connector_type_name, 
823                                 cross_guid, cross_connector_type_name)
824                 for trace_id in data.get_trace_data(guid):
825                     testbed.defer_add_trace(guid, trace_id)
826                 for (autoconf, address, netprefix, broadcast) in \
827                         data.get_address_data(guid):
828                     if address != None:
829                         testbed.defer_add_address(guid, address, netprefix, 
830                                 broadcast)
831                 for (destination, netprefix, nexthop) in data.get_route_data(guid):
832                     testbed.defer_add_route(guid, destination, netprefix, nexthop)
833     
834     def _program_testbed_cross_connections(self, data):
835         data_guids = data.guids
836
837         for guid in data_guids: 
838             if not data.is_testbed_data(guid):
839                 (testbed_guid, factory_id) = data.get_box_data(guid)
840                 testbed = self._testbeds.get(testbed_guid)
841                 if testbed:
842                     for (connector_type_name, cross_guid, cross_connector_type_name) \
843                             in data.get_connection_data(guid):
844                         (testbed_guid, factory_id) = data.get_box_data(guid)
845                         (cross_testbed_guid, cross_factory_id) = data.get_box_data(
846                                 cross_guid)
847                         if testbed_guid != cross_testbed_guid:
848                             cross_testbed = self._testbeds[cross_testbed_guid]
849                             cross_testbed_id = cross_testbed.testbed_id
850                             testbed.defer_cross_connect(guid, connector_type_name, cross_guid, 
851                                     cross_testbed_guid, cross_testbed_id, cross_factory_id, 
852                                     cross_connector_type_name)
853                             # save cross data for later
854                             self._add_crossdata(testbed_guid, guid, cross_testbed_guid,
855                                     cross_guid)
856                 
857     def _add_crossdata(self, testbed_guid, guid, cross_testbed_guid, cross_guid):
858         if testbed_guid not in self._cross_data:
859             self._cross_data[testbed_guid] = dict()
860         if cross_testbed_guid not in self._cross_data[testbed_guid]:
861             self._cross_data[testbed_guid][cross_testbed_guid] = set()
862         self._cross_data[testbed_guid][cross_testbed_guid].add(cross_guid)
863
864     def _get_cross_data(self, testbed_guid):
865         cross_data = dict()
866         if not testbed_guid in self._cross_data:
867             return cross_data
868         for cross_testbed_guid, guid_list in \
869                 self._cross_data[testbed_guid].iteritems():
870             cross_data[cross_testbed_guid] = dict()
871             cross_testbed = self._testbeds[cross_testbed_guid]
872             for cross_guid in guid_list:
873                 elem_cross_data = dict(
874                     _guid = cross_guid,
875                     _testbed_guid = cross_testbed_guid,
876                     _testbed_id = cross_testbed.testbed_id,
877                     _testbed_version = cross_testbed.testbed_version)
878                 cross_data[cross_testbed_guid][cross_guid] = elem_cross_data
879                 attributes_list = cross_testbed.get_attribute_list(cross_guid)
880                 for attr_name in attributes_list:
881                     attr_value = cross_testbed.get(cross_guid, attr_name)
882                     elem_cross_data[attr_name] = attr_value
883         return cross_data
884