Remove usage of removed RPC update-entry
[integration/test.git] / csit / libraries / Sxp.py
1 import json
2 from netaddr import IPAddress
3 from string import Template
4
5
6 def get_active_controller_from_json(resp, service):
7     """Gets index of active controller running specified service
8
9     :param resp: JSON formatted response from EOS
10     :type resp: str
11     :param service: EOS Service to look for
12     :type service: str
13     :return: Index of controller
14     """
15     entities = json.loads(resp)['entity-owners']['entity-type']
16     for entity in entities:
17         if entity['type'] == "org.opendaylight.mdsal.ServiceEntityType":
18             for instance in entity['entity']:
19                 if service in instance['id']:
20                     return int(instance['owner'][-1:])
21     return 0
22
23
24 def mod(num, base):
25     """Gets modulo of number
26
27     :param num: Number to be used
28     :type num: str
29     :param base: Base used
30     :type base: str
31     :returns: Int representing modulo of specified numbers.
32
33     """
34     return int(num) % int(base)
35
36
37 def get_average_of_items(items):
38     """Gets average of items in provided list
39
40     :param items: To be proceed
41     :return: Average value
42
43     """
44     return sum(items) / len(items)
45
46
47 def get_opposing_mode(mode):
48     """Generate string representing opposing SXP peer mode
49
50         :param mode: SXP peer mode
51         :type mode: str
52         :returns: String with opposing SXP peer mode.
53
54         """
55     if 'speaker' == mode:
56         return 'listener'
57     elif 'listener' == mode:
58         return 'speaker'
59     return 'both'
60
61
62 def get_ip_from_number(n, base=2130706432):
63     """Generate string representing Ipv4 from specified number plus base value
64
65     :param n: Number to be converted
66     :type n: int
67     :param base: Starting index
68     :type base: int
69     :returns: String containing Ipv4.
70
71     """
72     ip = IPAddress(int(base) + n)
73     return str(ip)
74
75
76 def get_ip_from_number_and_ip(n, ip_address):
77     """Generate string representing Ipv4 from specified number and IPAddress
78
79     :param n: Number to be converted
80     :type n: int
81     :param ip_address: Base address
82     :type ip_address: str
83     :returns: String containing Ipv4.
84
85     """
86     ip = IPAddress(int(IPAddress(ip_address)) + n)
87     return str(ip)
88
89
90 def lower_version(ver1, ver2):
91     """Generate xml containing SGT mach data
92
93     :param ver1: Version of SXP protocol for compare
94     :type ver1: str
95     :param ver2: Version of SXP protocol for compare
96     :type ver2: str
97     :returns: String containing lower from those two specified versions.
98
99     """
100     v1 = int(ver1[-1:])
101     v2 = int(ver2[-1:])
102     if v1 <= v2:
103         return ver1
104     else:
105         return ver2
106
107
108 def get_filter_entry(seq, entry_type, sgt="", esgt="", acl="", eacl="", pl="", epl="", ps=""):
109     """Generate xml containing FilterEntry data
110
111     :param seq: Sequence of entry
112     :type seq: str
113     :param entry_type: Type of entry (permit/deny)
114     :type entry_type: str
115     :param sgt: SGT matches to be added to entry
116     :type sgt: str
117     :param esgt: SGT ranges match to be added to entry
118     :type esgt: str
119     :param acl: ACL matches to be added to entry
120     :type acl: str
121     :param eacl: EACL matches to be added to entry
122     :type eacl: str
123     :param pl: PrefixList matches to be added to entry
124     :type pl: str
125     :param epl: ExtendedPrefixList matches to be added to entry
126     :type epl: str
127     :param ps: PeerSequence matches to be added to entry
128     :type ps: str
129     :returns: String containing xml data for request
130
131     """
132     entries = ""
133     # Generate XML request containing combination of Matches of different types
134     if sgt:
135         args = sgt.split(',')
136         entries += add_sgt_matches_xml(args)
137     elif esgt:
138         args = esgt.split(',')
139         entries += add_sgt_range_xml(args[0], args[1])
140     if pl:
141         entries += add_pl_entry_xml(pl)
142     elif epl:
143         args = epl.split(',')
144         entries += add_epl_entry_xml(args[0], args[1], args[2])
145     if acl:
146         args = acl.split(',')
147         entries += add_acl_entry_xml(args[0], args[1])
148     elif eacl:
149         args = eacl.split(',')
150         entries += add_eacl_entry_xml(args[0], args[1], args[2], args[3])
151     if ps:
152         args = ps.split(',')
153         entries += add_ps_entry_xml(args[0], args[1])
154     # Wrap entries in ACL/PrefixList according to specified values
155     if pl or epl:
156         return add_pl_entry_default_xml(seq, entry_type, entries)
157     elif ps:
158         return add_ps_entry_default_xml(seq, entry_type, entries)
159     return add_acl_entry_default_xml(seq, entry_type, entries)
160
161
162 def add_peers(*args):
163     """Generate xml containing Peer mach data
164
165     :param args: Peers data
166     :type args: dict
167     :returns: String containing xml data for request
168
169     """
170     templ = Template('''
171         <sxp-peer>
172             <peer-address>$ip</peer-address>
173         </sxp-peer>''')
174     peers = ""
175     for count, value in enumerate(args):
176         peers += templ.substitute({'ip': value})
177     return peers
178
179
180 def add_domains(*args):
181     """Generate xml containing Domain mach data
182
183     :param args: Domain data
184     :type args: dict
185     :returns: String containing xml data for request
186
187     """
188     templ = Template('''
189         <domain>
190             <name>$name</name>
191         </domain>''')
192     peers = ""
193     for count, value in enumerate(args):
194         peers += templ.substitute({'name': value})
195     return peers
196
197
198 def add_sgt_matches_xml(sgt_entries):
199     """Generate xml containing SGT mach data
200
201     :param sgt_entries: SGT matches
202     :type sgt_entries: str
203     :returns: String containing xml data for request
204
205     """
206     templ = Template('''
207         <matches>$sgt</matches>''')
208     matches = ""
209     for sgt in sgt_entries:
210         matches += templ.substitute({'sgt': sgt})
211     return matches
212
213
214 def add_sgt_range_xml(start, end):
215     """Generate xml containing SGT RangeMach data
216
217     :param start: Start range of SGT
218     :type start: str
219     :param end: End range of SGT
220     :type end: str
221     :returns: String containing xml data for request
222
223     """
224     templ = Template('''
225         <sgt-start>$start</sgt-start>
226         <sgt-end>$end</sgt-end>''')
227     match = templ.substitute({'start': start, 'end': end})
228     return match
229
230
231 def add_acl_entry_default_xml(seq, entry_type, acl_entries):
232     """Generate xml containing AccessList data
233
234     :param seq: Sequence of PrefixList entry
235     :type seq: str
236     :param entry_type: Entry type (permit/deny)
237     :type entry_type: str
238     :param acl_entries: XML data containing AccessList entries
239     :type acl_entries: str
240     :returns: String containing xml data for request
241
242     """
243     templ = Template('''
244         <acl-entry>
245             <entry-type>$entry_type</entry-type>
246             <entry-seq>$seq</entry-seq>$acl_entries
247         </acl-entry>''')
248     matches = templ.substitute(
249         {'seq': seq, 'entry_type': entry_type, 'acl_entries': acl_entries})
250     return matches
251
252
253 def add_acl_entry_xml(ip, mask):
254     """Generate xml containing AccessList data
255
256     :param ip: Ipv4/6 address
257     :type ip: str
258     :param mask: Ipv4/6 wildcard mask
259     :type mask: str
260     :returns: String containing xml data for request
261
262     """
263     templ = Template('''
264         <acl-match>
265             <ip-address>$ip</ip-address>
266             <wildcard-mask>$mask</wildcard-mask>
267         </acl-match>''')
268     return templ.substitute({'ip': ip, 'mask': mask})
269
270
271 def add_eacl_entry_xml(ip, mask, amask, wmask):
272     """Generate xml containing ExtendedAccessList data
273
274     :param ip: Ipv4/6 address
275     :type ip: str
276     :param mask: Ipv4/6 wildcard mask
277     :type mask: str
278     :param amask: Ipv4/6 address mask
279     :type amask: str
280     :param wmask: Ipv4/6 address wildcard mask
281     :type wmask: str
282     :returns: String containing xml data for request
283
284     """
285     templ = Template('''
286         <acl-match>
287             <ip-address>$ip</ip-address>
288             <wildcard-mask>$mask</wildcard-mask>
289             <mask>
290               <address-mask>$amask</address-mask>
291               <wildcard-mask>$wmask</wildcard-mask>
292             </mask>
293         </acl-match>''')
294     return templ.substitute({'ip': ip, 'mask': mask, 'amask': amask, 'wmask': wmask})
295
296
297 def add_ps_entry_default_xml(seq, entry_type, ps_entries):
298     """Generate xml containing PeerSequence data
299
300     :param seq: Sequence of PrefixList entry
301     :type seq: str
302     :param entry_type: Entry type (permit/deny)
303     :type entry_type: str
304     :param ps_entries: XML data containing PeerSequence entries
305     :type ps_entries: str
306     :returns: String containing xml data for request
307
308     """
309     templ = Template('''
310     <peer-sequence-entry xmlns="urn:opendaylight:sxp:controller">
311           <entry-type>$entry_type</entry-type>
312           <entry-seq>$seq</entry-seq>$ps_entries
313     </peer-sequence-entry>''')
314     return templ.substitute({'seq': seq, 'entry_type': entry_type, 'ps_entries': ps_entries})
315
316
317 def add_pl_entry_default_xml(seq, entry_type, pl_entries):
318     """Generate xml containing PrefixList data
319
320     :param seq: Sequence of PrefixList entry
321     :type seq: str
322     :param entry_type: Entry type (permit/deny)
323     :type entry_type: str
324     :param pl_entries: XML data containing PrefixList entries
325     :type pl_entries: str
326     :returns: String containing xml data for request
327
328     """
329     templ = Template('''
330     <prefix-list-entry xmlns="urn:opendaylight:sxp:controller">
331           <entry-type>$entry_type</entry-type>
332           <entry-seq>$seq</entry-seq>$pl_entries
333     </prefix-list-entry>''')
334     return templ.substitute({'seq': seq, 'entry_type': entry_type, 'pl_entries': pl_entries})
335
336
337 def add_pl_entry_xml(prefix):
338     """Generate xml containing PrefixList data
339
340     :param prefix: Ipv4/6 prefix
341     :type prefix: str
342     :returns: String containing xml data for request
343
344     """
345     templ = Template('''
346         <prefix-list-match>
347             <ip-prefix>$prefix</ip-prefix>
348         </prefix-list-match>''')
349     return templ.substitute({'prefix': prefix})
350
351
352 def add_epl_entry_xml(prefix, op, mask):
353     """Generate xml containing Extended PrefixList data
354
355     :param prefix: Ipv4/6 prefix
356     :type prefix: str
357     :param op: PrefixList option (ge/le/eq)
358     :type op: str
359     :param mask: Ipv4/6 Mask
360     :type mask: str
361     :returns: String containing xml data for request
362
363     """
364     templ = Template('''
365         <prefix-list-match>
366             <ip-prefix>$prefix</ip-prefix>
367             <mask>
368                 <mask-range>$op</mask-range>
369                 <mask-value>$mask</mask-value>
370             </mask>
371         </prefix-list-match>''')
372     return templ.substitute({'prefix': prefix, 'mask': mask, 'op': op})
373
374
375 def add_ps_entry_xml(op, length):
376     """Generate xml containing Extended PrefixList data
377
378     :param op: PrefixList option (ge/le/eq)
379     :type op: str
380     :param length: PeerSequence length
381     :type length: str
382     :returns: String containing xml data for request
383
384     """
385     templ = Template('''
386         <peer-sequence-length>$length</peer-sequence-length>
387         <peer-sequence-range>$op</peer-sequence-range>
388         ''')
389     return templ.substitute({'length': length, 'op': op})
390
391
392 def parse_peer_groups(groups_json):
393     """Parse JSON string into Array of PeerGroups
394
395     :param groups_json: JSON containing PeerGroups
396     :type groups_json: str
397     :returns: Array containing PeerGroups.
398
399     """
400     data = json.loads(groups_json)
401     groups = data['output']
402     output = []
403     for group in groups.values():
404         output += group
405     return output
406
407
408 def parse_connections(connections_json):
409     """Parse JSON string into Array of Connections
410
411     :param connections_json: JSON containing Connections
412     :type connections_json: str
413     :returns: Array containing Connections.
414
415     """
416     data = json.loads(connections_json)
417     connections = data['output']['connections']
418     output = []
419     for connection in connections.values():
420         output += connection
421     return output
422
423
424 def find_connection(connections_json, version, mode, ip, port, state):
425     """Test if Connection with specified values is contained in JSON
426
427     :param connections_json: JSON containing Connections
428     :type connections_json: str
429     :param version: Version of SXP protocol (version1/2/3/4)
430     :type version: str
431     :param mode: Mode of SXP peer (speaker/listener/both)
432     :type mode: str
433     :param ip: Ipv4/6 address of remote peer
434     :type ip: str
435     :param port: Port on with remote peer listens
436     :type port: str
437     :param state: State of connection (on/off/pendingOn/deleteHoldDown)
438     :type state: str
439     :returns: True if Connection with specified params was found, otherwise False.
440
441     """
442     for connection in parse_connections(connections_json):
443         if (connection['peer-address'] == ip and connection['tcp-port'] == int(port) and (
444                 mode.strip() == 'any' or connection['mode'] == mode) and connection['version'] == version):
445             if state == 'none':
446                 return True
447             elif connection['state'] == state:
448                 return True
449     return False
450
451
452 def parse_bindings(bindings_json):
453     """Parse JSON string into Array of Bindings
454
455     :param bindings_json: JSON containing Bindings
456     :type bindings_json: str
457     :returns: Array containing Bindings.
458
459     """
460     data = json.loads(bindings_json)
461     output = []
462     for bindings_json in data['output'].values():
463         for binding in bindings_json:
464             output.append(binding)
465     return output
466
467
468 def find_binding(bindings, sgt, prefix):
469     """Test if Binding with specified values is contained in JSON
470
471     :param bindings: JSON containing Bindings
472     :type bindings: str
473     :param sgt: Source Group Tag
474     :type sgt: str
475     :param prefix: Ipv4/6 prefix
476     :type prefix: str
477     :returns: True if Binding with specified params was found, otherwise False.
478
479     """
480     for binding in parse_bindings(bindings):
481         if binding['sgt'] == int(sgt):
482             for ip_prefix in binding['ip-prefix']:
483                 if ip_prefix == prefix:
484                     return True
485     return False
486
487
488 def parse_prefix_groups(prefix_groups_json, source_):
489     """Parse JSON string into Array of PrefixGroups
490
491     :param prefix_groups_json: JSON containing PrefixGroups
492     :type prefix_groups_json: str
493     :param source_: Source of PrefixGroups (sxp/local)
494     :type source_: str
495     :returns: Array containing PrefixGroups.
496
497     """
498     data = json.loads(prefix_groups_json)
499     bindings = data['sxp-node:master-database']
500     output = []
501     for binding in bindings.values():
502         for binding_source in binding:
503             if source_ == "any" or binding_source['binding-source'] == source_:
504                 for prefix_group in binding_source['prefix-group']:
505                     output.append(prefix_group)
506     return output
507
508
509 def find_binding_legacy(prefix_groups_json, sgt, prefix, source_, action):
510     """Test if Binding with specified values is contained in JSON
511
512     :param prefix_groups_json: JSON containing Bindings and PrefixGroups
513     :type prefix_groups_json: str
514     :param sgt: Source Group Tag
515     :type sgt: str
516     :param prefix: Ipv4/6 prefix
517     :type prefix: str
518     :param source_: Source of binding (local/sxp)
519     :type source_: str
520     :param action: Action for binding (add/delete)
521     :type action: str
522     :returns: True if Binding with specified params was found, otherwise False.
523
524     """
525     found = False
526     for prefixgroup in parse_prefix_groups(prefix_groups_json, source_):
527         if prefixgroup['sgt'] == int(sgt):
528             for binding in prefixgroup['binding']:
529                 if binding['ip-prefix'] == prefix and binding['action'] == action:
530                     found = True
531     return found
532
533
534 def add_connection_xml(version, mode, ip, port, node, password_, domain_name, bindings_timeout=0, security_mode=''):
535     """Generate xml for Add Connection request
536
537     :param version: Version of SXP protocol (version1/2/3/4)
538     :type version: str
539     :param mode: Mode of SXP peer (speaker/listener/both)
540     :type mode: str
541     :param ip: Ipv4/6 address of remote peer
542     :type ip: str
543     :param port: Port on with remote peer listens
544     :type port: str
545     :param node: Ipv4 address of node
546     :type node: str
547     :param password_: Password type (none/default)
548     :type password_: str
549     :param domain_name: Name of Domain
550     :type domain_name: str
551     :param security_mode: Default/TSL security
552     :type security_mode: str
553     :param bindings_timeout: Specifies DHD and Reconciliation timers
554     :type bindings_timeout: int
555     :returns: String containing xml data for request
556
557     """
558     templ = Template('''<input>
559    <requested-node xmlns="urn:opendaylight:sxp:controller">$node</requested-node>
560    $domain
561    <connections xmlns="urn:opendaylight:sxp:controller">
562       <connection>
563          <peer-address>$ip</peer-address>
564          <tcp-port>$port</tcp-port>
565          <password>$password_</password>
566          <mode>$mode</mode>
567          <version>$version</version>
568          <description>Connection to ISR-G2</description>
569          $security_type
570          <connection-timers>
571             <hold-time-min-acceptable>45</hold-time-min-acceptable>
572             <keep-alive-time>30</keep-alive-time>
573             <reconciliation-time>$timeout</reconciliation-time>
574             <delete-hold-down-time>$timeout</delete-hold-down-time>
575          </connection-timers>
576       </connection>
577    </connections>
578 </input>
579 ''')
580     data = templ.substitute(
581         {'ip': ip, 'port': port, 'mode': mode, 'version': version, 'node': node,
582          'password_': password_, 'domain': get_domain_name(domain_name), 'timeout': bindings_timeout,
583          'security_type': '<security-type>' + security_mode + '</security-type>' if security_mode else ''})
584     return data
585
586
587 def delete_connections_xml(address, port, node, domain_name):
588     """Generate xml for Delete Connection request
589
590     :param address: Ipv4/6 address of remote peer
591     :type address: str
592     :param port: Port on with remote peer listens
593     :type port: str
594     :param node: Ipv4 address of node
595     :type node: str
596     :param domain_name: Name of Domain
597     :type domain_name: str
598     :returns: String containing xml data for request
599
600     """
601     templ = Template('''<input>
602    <requested-node xmlns="urn:opendaylight:sxp:controller">$node</requested-node>
603    $domain
604    <peer-address xmlns="urn:opendaylight:sxp:controller">$address</peer-address>
605    <tcp-port xmlns="urn:opendaylight:sxp:controller">$port</tcp-port>
606 </input>''')
607     data = templ.substitute({'address': address, 'port': port, 'node': node, 'domain': get_domain_name(domain_name)})
608     return data
609
610
611 def delete_binding_xml(sgt, prefix, ip, domain_name):
612     """Generate xml for Delete Binding request
613
614     :param sgt: Source Group Tag
615     :type sgt: str
616     :param prefix: Ipv4/6 prefix
617     :type prefix: str
618     :param ip: Ipv4 address of node
619     :type ip: str
620     :param domain_name: Name of Domain
621     :type domain_name: str
622     :returns: String containing xml data for request
623
624     """
625     templ = Template('''<input>
626   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
627   <sgt xmlns="urn:opendaylight:sxp:controller">$sgt</sgt>
628   <ip-prefix xmlns="urn:opendaylight:sxp:controller">$prefix</ip-prefix>
629   $domain
630 </input>''')
631     data = templ.substitute({'sgt': sgt, 'prefix': prefix, 'ip': ip, 'domain': get_domain_name(domain_name)})
632     return data
633
634
635 def add_peer_group_xml(name, peers, ip):
636     """Generate xml for Add PeerGroups request
637
638     :param name: Name of PeerGroup
639     :type name: str
640     :param peers: XML formatted peers that will be added to group
641     :type peers: str
642     :param ip: Ipv4 address of node
643     :type ip: str
644     :returns: String containing xml data for request
645
646     """
647     templ = Template('''<input>
648   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
649   <sxp-peer-group xmlns="urn:opendaylight:sxp:controller">
650     <name xmlns="urn:opendaylight:sxp:controller">$name</name>
651     <sxp-peers xmlns="urn:opendaylight:sxp:controller">$peers</sxp-peers>
652     </sxp-peer-group>
653 </input>''')
654     data = templ.substitute({'name': name, 'peers': peers, 'ip': ip})
655     return data
656
657
658 def delete_peer_group_xml(name, ip):
659     """Generate xml for Delete PeerGroup request
660
661     :param name: Name of PeerGroup
662     :type name: str
663     :param ip: Ipv4 address of node
664     :type ip: str
665     :returns: String containing xml data for request
666
667     """
668     templ = Template('''<input>
669   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
670   <peer-group-name xmlns="urn:opendaylight:sxp:controller">$name</peer-group-name>
671 </input>''')
672     data = templ.substitute({'name': name, 'ip': ip})
673     return data
674
675
676 def get_peer_groups_from_node_xml(ip):
677     """Generate xml for Get PeerGroups request
678
679     :param ip: Ipv4 address of node
680     :type ip: str
681     :returns: String containing xml data for request
682
683     """
684     templ = Template('''<input>
685    <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
686 </input>''')
687     data = templ.substitute({'ip': ip})
688     return data
689
690
691 def add_filter_xml(group, filter_type, entries, ip, policy=None):
692     """Generate xml for Add Filter request
693
694     :param group: Name of group containing filter
695     :type group: str
696     :param filter_type: Type of filter
697     :type filter_type: str
698     :param entries: XML formatted entries that will be added in filter
699     :type entries: str
700     :param ip: Ipv4 address of node
701     :type ip: str
702     :param policy: Policy of filter update mechanism
703     :type policy: str
704     :returns: String containing xml data for request
705
706     """
707     if policy:
708         policy = "<filter-policy>" + policy + "</filter-policy>"
709     else:
710         policy = ""
711     templ = Template('''<input>
712   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
713   <peer-group-name xmlns="urn:opendaylight:sxp:controller">$group</peer-group-name>
714   <sxp-filter xmlns="urn:opendaylight:sxp:controller">
715     $filter_policy
716     <filter-type>$filter_type</filter-type>$entries
717   </sxp-filter>
718 </input>''')
719     data = templ.substitute(
720         {'group': group, 'filter_type': filter_type, 'ip': ip, 'entries': entries, 'filter_policy': policy})
721     return data
722
723
724 def add_domain_filter_xml(domain, domains, entries, ip, filter_name=None):
725     """Generate xml for Add Domain Filter request
726
727     :param domain: Name of Domain containing filter
728     :type domain: str
729     :param domains: Domains on which filter will be applied
730     :type domains: str
731     :param entries: XML formatted entries that will be added in filter
732     :type entries: str
733     :param ip: Ipv4 address of node
734     :type ip: str
735     :param filter_name: Name of filter
736     :type filter_name: str
737     :returns: String containing xml data for request
738
739     """
740     if filter_name:
741         filter_name = "<filter-name>" + filter_name + "</filter-name>"
742     templ = Template('''<input>
743   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
744   <domain-name xmlns="urn:opendaylight:sxp:controller">$domain</domain-name>
745   <sxp-domain-filter xmlns="urn:opendaylight:sxp:controller">
746     $filter_name
747     <domains>$domains</domains>
748     $entries
749   </sxp-domain-filter>
750 </input>''')
751     data = templ.substitute(
752         {'domain': domain, 'domains': domains, 'ip': ip, 'entries': entries, 'filter_name': filter_name})
753     return data
754
755
756 def delete_filter_xml(group, filter_type, ip):
757     """Generate xml for Delete Filter request
758
759     :param group: Name of group containing filter
760     :type group: str
761     :param filter_type: Type of filter
762     :type filter_type: str
763     :param ip: Ipv4 address of node
764     :type ip: str
765     :returns: String containing xml data for request
766
767     """
768     templ = Template('''<input>
769   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
770   <peer-group-name xmlns="urn:opendaylight:sxp:controller">$group</peer-group-name>
771   <filter-type xmlns="urn:opendaylight:sxp:controller">$filter_type</filter-type>
772 </input>''')
773     data = templ.substitute(
774         {'group': group, 'filter_type': filter_type, 'ip': ip})
775     return data
776
777
778 def delete_domain_filter_xml(domain, ip, filter_name=None):
779     """Generate xml for Delete Filter request
780
781     :param domain: Name of Domain containing filter
782     :type domain: str
783     :param ip: Ipv4 address of node
784     :type ip: str
785     :param filter_name: Name of filter
786     :type filter_name: str
787     :returns: String containing xml data for request
788
789     """
790     if filter_name:
791         filter_name = '<filter-name xmlns="urn:opendaylight:sxp:controller">' + filter_name + "</filter-name>"
792     templ = Template('''<input>
793   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
794   <domain-name xmlns="urn:opendaylight:sxp:controller">$domain</domain-name>
795   $filter_name
796 </input>''')
797     data = templ.substitute(
798         {'domain': domain, 'ip': ip, 'filter_name': filter_name})
799     return data
800
801
802 def get_connections_from_node_xml(ip, domain_name):
803     """Generate xml for Get Connections request
804
805     :param ip: Ipv4 address of node
806     :type ip: str
807     :param domain_name: Name of Domain
808     :type domain_name: str
809     :returns: String containing xml data for request
810
811     """
812     templ = Template('''<input>
813    <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
814    $domain
815 </input>''')
816     data = templ.substitute({'ip': ip, 'domain': get_domain_name(domain_name)})
817     return data
818
819
820 def get_bindings_from_node_xml(ip, binding_range, domain_name):
821     """Generate xml for Get Bindings request
822
823     :param binding_range: All or only Local bindings
824     :type binding_range: str
825     :param ip: Ipv4 address of node
826     :type ip: str
827     :param domain_name: Name of Domain
828     :type domain_name: str
829     :returns: String containing xml data for request
830
831     """
832     templ = Template('''<input>
833   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
834   <bindings-range xmlns="urn:opendaylight:sxp:controller">$range</bindings-range>
835   $domain
836 </input>''')
837     data = templ.substitute({'ip': ip, 'range': binding_range, 'domain': get_domain_name(domain_name)})
838     return data
839
840
841 def add_node_xml(node_id, port, password, version, node_ip=None, expansion=0, bindings_timeout=0, keystores=None,
842                  retry_open_timer=1):
843     """Generate xml for Add Node request
844
845     :param node_id: Ipv4 address formatted node id
846     :type node_id: str
847     :param node_ip: Ipv4 address of node
848     :type node_ip: strl
849     :param port: Node port number
850     :type port: int
851     :param password: TCP-MD5 password
852     :type password: str
853     :param version: Sxp device version
854     :type version: str
855     :param expansion: Bindings expansion
856     :type expansion: int
857     :param bindings_timeout: Specifies DHD and Reconciliation timers
858     :type bindings_timeout: int
859     :param keystores: SSL keystore and truststore specification
860     :type keystores: dict
861     :returns: String containing xml data for request
862
863     """
864     tls = ''
865     if keystores:
866         tls = Template('''
867         <tls>
868             <keystore>
869               <location>$keystore</location>
870               <type>JKS</type>
871               <path-type>PATH</path-type>
872               <password>$passwd</password>
873             </keystore>
874             <truststore>
875               <location>$truststore</location>
876               <type>JKS</type>
877               <path-type>PATH</path-type>
878               <password>$passwd</password>
879             </truststore>
880             <certificate-password>$passwd</certificate-password>
881         </tls>
882     ''').substitute(
883             {'keystore': keystores['keystore'], 'truststore': keystores['truststore'], 'passwd': keystores['password']})
884
885     templ = Template('''<input xmlns="urn:opendaylight:sxp:controller">
886     <node-id>$id</node-id>
887     <timers>
888         <retry-open-time>$retry_open_timer</retry-open-time>
889         <hold-time-min-acceptable>120</hold-time-min-acceptable>
890         <delete-hold-down-time>$timeout</delete-hold-down-time>
891         <hold-time-min>90</hold-time-min>
892         <reconciliation-time>$timeout</reconciliation-time>
893         <hold-time>90</hold-time>
894         <hold-time-max>180</hold-time-max>
895         <keep-alive-time>30</keep-alive-time>
896     </timers>
897     <mapping-expanded>$expansion</mapping-expanded>
898     <security>
899         $tls
900         <password>$password</password>
901     </security>
902     <tcp-port>$port</tcp-port>
903     <version>$version</version>
904     <description>ODL SXP Controller</description>
905     <source-ip>$ip</source-ip>
906 </input>''')
907     data = templ.substitute(
908         {'ip': node_ip or node_id, 'id': node_id, 'port': port, 'password': password,
909          'version': version, 'expansion': expansion, 'timeout': bindings_timeout, 'tls': tls,
910          'retry_open_timer': retry_open_timer})
911     return data
912
913
914 def delete_node_xml(node_id):
915     """Generate xml for Delete node request
916
917     :param node_id: Ipv4 address formatted node id
918     :type node_id: str
919     :returns: String containing xml data for request
920
921     """
922     templ = Template('''<input xmlns="urn:opendaylight:sxp:controller">
923   <node-id>$id</node-id>
924 </input>''')
925     data = templ.substitute({'id': node_id})
926     return data
927
928
929 def add_domain_xml(node_id, name):
930     """Generate xml for Add Domain request
931
932     :param node_id: Id of node
933     :type node_id: str
934     :param name: Name of Domain
935     :type name: str
936     :returns: String containing xml data for request
937
938     """
939     templ = Template('''<input>
940   <node-id xmlns="urn:opendaylight:sxp:controller">$id</node-id>
941   <domain-name xmlns="urn:opendaylight:sxp:controller">$name</domain-name>
942 </input>''')
943     data = templ.substitute({'name': name, 'id': node_id})
944     return data
945
946
947 def delete_domain_xml(node_id, name):
948     """Generate xml for Remove Domain request
949
950     :param node_id: Id of node
951     :type node_id: str
952     :param name: Name of Domain
953     :type name: str
954     :returns: String containing xml data for request
955
956     """
957     return add_domain_xml(node_id, name)
958
959
960 def get_domain_name(domain_name):
961     """Generate xml for Get Bindings request
962
963     :param domain_name: Name of Domain
964     :type domain_name: str
965     :returns: String containing xml data for request
966
967     """
968     if domain_name == 'global':
969         return ''
970     else:
971         return '<domain-name xmlns="urn:opendaylight:sxp:controller">' + domain_name + '</domain-name>'
972
973
974 def add_bindings_xml(node_id, domain, sgt, prefixes):
975     """Generate xml for Add Bindings request
976
977     :param node_id: Id of node
978     :type node_id: str
979     :param domain: Name of Domain
980     :type domain: str
981     :param sgt: Security group
982     :type sgt: int
983     :param prefixes: List of ip-prefixes
984     :type prefixes: str
985     :returns: String containing xml data for request
986
987     """
988     bindings = ''
989     for prefix in prefixes.split(','):
990         bindings += '\n' + '<ip-prefix>' + prefix + '</ip-prefix>'
991     templ = Template('''<input>
992   <node-id xmlns="urn:opendaylight:sxp:controller">$id</node-id>
993   <domain-name xmlns="urn:opendaylight:sxp:controller">$name</domain-name>
994   <binding xmlns="urn:opendaylight:sxp:controller">
995       <sgt>$sgt</sgt>
996       $bindings
997   </binding>
998 </input>''')
999     data = templ.substitute({'name': domain, 'id': node_id, 'sgt': sgt, 'bindings': bindings})
1000     return data
1001
1002
1003 def delete_bindings_xml(node_id, domain, sgt, prefixes):
1004     """Generate xml for Remove Bindings request
1005
1006     :param node_id: Id of node
1007     :type node_id: str
1008     :param domain: Name of Domain
1009     :type domain: str
1010     :param sgt: Security group
1011     :type sgt: int
1012     :param prefixes: List of ip-prefixes
1013     :type prefixes: str
1014     :returns: String containing xml data for request
1015
1016     """
1017     return add_bindings_xml(node_id, domain, sgt, prefixes)
1018
1019
1020 def prefix_range(start, end):
1021     """Generate and concatenate ip-prefixes
1022
1023     :param start: Start index
1024     :type start: str
1025     :param end: End index
1026     :type end: str
1027     :returns: String containing concatenated ip-prefixes
1028
1029     """
1030     start = int(start)
1031     end = int(end)
1032     index = 0
1033     prefixes = ''
1034     while index < end:
1035         prefixes += get_ip_from_number(index + start) + '/32'
1036         index += 1
1037         if index < end:
1038             prefixes += ','
1039     return prefixes
1040
1041
1042 def route_definition_xml(virtual_ip, net_mask, interface):
1043     """Generate xml for Add Bindings request
1044
1045     :param interface: Network interface name
1046     :type interface: str
1047     :param net_mask: NetMask of virtual ip
1048     :type net_mask: str
1049     :param virtual_ip: Virtual ip
1050     :type virtual_ip: str
1051     :returns: String containing xml data for request
1052
1053     """
1054     templ = Template('''
1055     <routing-definition>
1056         <ip-address>$vip</ip-address>
1057         <interface>$interface</interface>
1058         <netmask>$mask</netmask>
1059     </routing-definition>
1060     ''')
1061     data = templ.substitute({'mask': net_mask, 'vip': virtual_ip, 'interface': interface})
1062     return data
1063
1064
1065 def route_definitions_xml(routes, old_routes=None):
1066     """Generate xml for Add Bindings request
1067
1068     :param routes: XML formatted data containing RouteDefinitions
1069     :type routes: str
1070     :param old_routes: Routes add to request that needs to persist
1071     :type old_routes: str
1072     :returns: String containing xml data for request
1073
1074     """
1075     if old_routes and "</sxp-cluster-route>" in old_routes:
1076         templ = Template(old_routes.replace("</sxp-cluster-route>", "$routes</sxp-cluster-route>"))
1077     else:
1078         templ = Template('''<sxp-cluster-route xmlns="urn:opendaylight:sxp:cluster:route">
1079     $routes
1080 </sxp-cluster-route>
1081     ''')
1082     data = templ.substitute({'routes': routes})
1083     return data