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 update_binding_xml(sgt0, prefix0, sgt1, prefix1, ip, domain_name):
 612     """Generate xml for Update Binding request
 613
 614     :param sgt0: Original Source Group Tag
 615     :type sgt0: str
 616     :param prefix0: Original Ipv4/6 prefix
 617     :type prefix0: str
 618     :param sgt1: New Source Group Tag
 619     :type sgt1: str
 620     :param prefix1: New Ipv4/6 prefix
 621     :type prefix1: str
 622     :param ip: Ipv4 address of node
 623     :type ip: str
 624     :param domain_name: Name of Domain
 625     :type domain_name: str
 626     :returns: String containing xml data for request
 627
 628     """
 629     templ = Template('''<input>
 630   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 631   $domain
 632   <original-binding xmlns="urn:opendaylight:sxp:controller">
 633     <sgt>$sgt0</sgt>
 634     <ip-prefix>$prefix0</ip-prefix>
 635   </original-binding>
 636   <new-binding xmlns="urn:opendaylight:sxp:controller">
 637     <sgt>$sgt1</sgt>
 638     <ip-prefix>$prefix1</ip-prefix>
 639   </new-binding>
 640 </input>''')
 641     data = templ.substitute(
 642         {'sgt0': sgt0, 'sgt1': sgt1, 'prefix0': prefix0, 'prefix1': prefix1, 'ip': ip,
 643          'domain': get_domain_name(domain_name)})
 644     return data
 645
 646
 647 def delete_binding_xml(sgt, prefix, ip, domain_name):
 648     """Generate xml for Delete Binding request
 649
 650     :param sgt: Source Group Tag
 651     :type sgt: str
 652     :param prefix: Ipv4/6 prefix
 653     :type prefix: str
 654     :param ip: Ipv4 address of node
 655     :type ip: str
 656     :param domain_name: Name of Domain
 657     :type domain_name: str
 658     :returns: String containing xml data for request
 659
 660     """
 661     templ = Template('''<input>
 662   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 663   <sgt xmlns="urn:opendaylight:sxp:controller">$sgt</sgt>
 664   <ip-prefix xmlns="urn:opendaylight:sxp:controller">$prefix</ip-prefix>
 665   $domain
 666 </input>''')
 667     data = templ.substitute({'sgt': sgt, 'prefix': prefix, 'ip': ip, 'domain': get_domain_name(domain_name)})
 668     return data
 669
 670
 671 def add_peer_group_xml(name, peers, ip):
 672     """Generate xml for Add PeerGroups request
 673
 674     :param name: Name of PeerGroup
 675     :type name: str
 676     :param peers: XML formatted peers that will be added to group
 677     :type peers: str
 678     :param ip: Ipv4 address of node
 679     :type ip: str
 680     :returns: String containing xml data for request
 681
 682     """
 683     templ = Template('''<input>
 684   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 685   <sxp-peer-group xmlns="urn:opendaylight:sxp:controller">
 686     <name xmlns="urn:opendaylight:sxp:controller">$name</name>
 687     <sxp-peers xmlns="urn:opendaylight:sxp:controller">$peers</sxp-peers>
 688     </sxp-peer-group>
 689 </input>''')
 690     data = templ.substitute({'name': name, 'peers': peers, 'ip': ip})
 691     return data
 692
 693
 694 def delete_peer_group_xml(name, ip):
 695     """Generate xml for Delete PeerGroup request
 696
 697     :param name: Name of PeerGroup
 698     :type name: str
 699     :param ip: Ipv4 address of node
 700     :type ip: str
 701     :returns: String containing xml data for request
 702
 703     """
 704     templ = Template('''<input>
 705   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 706   <peer-group-name xmlns="urn:opendaylight:sxp:controller">$name</peer-group-name>
 707 </input>''')
 708     data = templ.substitute({'name': name, 'ip': ip})
 709     return data
 710
 711
 712 def get_peer_groups_from_node_xml(ip):
 713     """Generate xml for Get PeerGroups request
 714
 715     :param ip: Ipv4 address of node
 716     :type ip: str
 717     :returns: String containing xml data for request
 718
 719     """
 720     templ = Template('''<input>
 721    <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 722 </input>''')
 723     data = templ.substitute({'ip': ip})
 724     return data
 725
 726
 727 def add_filter_xml(group, filter_type, entries, ip, policy=None):
 728     """Generate xml for Add Filter request
 729
 730     :param group: Name of group containing filter
 731     :type group: str
 732     :param filter_type: Type of filter
 733     :type filter_type: str
 734     :param entries: XML formatted entries that will be added in filter
 735     :type entries: str
 736     :param ip: Ipv4 address of node
 737     :type ip: str
 738     :param policy: Policy of filter update mechanism
 739     :type policy: str
 740     :returns: String containing xml data for request
 741
 742     """
 743     if policy:
 744         policy = "<filter-policy>" + policy + "</filter-policy>"
 745     else:
 746         policy = ""
 747     templ = Template('''<input>
 748   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 749   <peer-group-name xmlns="urn:opendaylight:sxp:controller">$group</peer-group-name>
 750   <sxp-filter xmlns="urn:opendaylight:sxp:controller">
 751     $filter_policy
 752     <filter-type>$filter_type</filter-type>$entries
 753   </sxp-filter>
 754 </input>''')
 755     data = templ.substitute(
 756         {'group': group, 'filter_type': filter_type, 'ip': ip, 'entries': entries, 'filter_policy': policy})
 757     return data
 758
 759
 760 def add_domain_filter_xml(domain, domains, entries, ip, filter_name=None):
 761     """Generate xml for Add Domain Filter request
 762
 763     :param domain: Name of Domain containing filter
 764     :type domain: str
 765     :param domains: Domains on which filter will be applied
 766     :type domains: str
 767     :param entries: XML formatted entries that will be added in filter
 768     :type entries: str
 769     :param ip: Ipv4 address of node
 770     :type ip: str
 771     :param filter_name: Name of filter
 772     :type filter_name: str
 773     :returns: String containing xml data for request
 774
 775     """
 776     if filter_name:
 777         filter_name = "<filter-name>" + filter_name + "</filter-name>"
 778     templ = Template('''<input>
 779   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 780   <domain-name xmlns="urn:opendaylight:sxp:controller">$domain</domain-name>
 781   <sxp-domain-filter xmlns="urn:opendaylight:sxp:controller">
 782     $filter_name
 783     <domains>$domains</domains>
 784     $entries
 785   </sxp-domain-filter>
 786 </input>''')
 787     data = templ.substitute(
 788         {'domain': domain, 'domains': domains, 'ip': ip, 'entries': entries, 'filter_name': filter_name})
 789     return data
 790
 791
 792 def delete_filter_xml(group, filter_type, ip):
 793     """Generate xml for Delete Filter request
 794
 795     :param group: Name of group containing filter
 796     :type group: str
 797     :param filter_type: Type of filter
 798     :type filter_type: str
 799     :param ip: Ipv4 address of node
 800     :type ip: str
 801     :returns: String containing xml data for request
 802
 803     """
 804     templ = Template('''<input>
 805   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 806   <peer-group-name xmlns="urn:opendaylight:sxp:controller">$group</peer-group-name>
 807   <filter-type xmlns="urn:opendaylight:sxp:controller">$filter_type</filter-type>
 808 </input>''')
 809     data = templ.substitute(
 810         {'group': group, 'filter_type': filter_type, 'ip': ip})
 811     return data
 812
 813
 814 def delete_domain_filter_xml(domain, ip, filter_name=None):
 815     """Generate xml for Delete Filter request
 816
 817     :param domain: Name of Domain containing filter
 818     :type domain: str
 819     :param ip: Ipv4 address of node
 820     :type ip: str
 821     :param filter_name: Name of filter
 822     :type filter_name: str
 823     :returns: String containing xml data for request
 824
 825     """
 826     if filter_name:
 827         filter_name = '<filter-name xmlns="urn:opendaylight:sxp:controller">' + filter_name + "</filter-name>"
 828     templ = Template('''<input>
 829   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 830   <domain-name xmlns="urn:opendaylight:sxp:controller">$domain</domain-name>
 831   $filter_name
 832 </input>''')
 833     data = templ.substitute(
 834         {'domain': domain, 'ip': ip, 'filter_name': filter_name})
 835     return data
 836
 837
 838 def get_connections_from_node_xml(ip, domain_name):
 839     """Generate xml for Get Connections request
 840
 841     :param ip: Ipv4 address of node
 842     :type ip: str
 843     :param domain_name: Name of Domain
 844     :type domain_name: str
 845     :returns: String containing xml data for request
 846
 847     """
 848     templ = Template('''<input>
 849    <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 850    $domain
 851 </input>''')
 852     data = templ.substitute({'ip': ip, 'domain': get_domain_name(domain_name)})
 853     return data
 854
 855
 856 def get_bindings_from_node_xml(ip, binding_range, domain_name):
 857     """Generate xml for Get Bindings request
 858
 859     :param binding_range: All or only Local bindings
 860     :type binding_range: str
 861     :param ip: Ipv4 address of node
 862     :type ip: str
 863     :param domain_name: Name of Domain
 864     :type domain_name: str
 865     :returns: String containing xml data for request
 866
 867     """
 868     templ = Template('''<input>
 869   <requested-node xmlns="urn:opendaylight:sxp:controller">$ip</requested-node>
 870   <bindings-range xmlns="urn:opendaylight:sxp:controller">$range</bindings-range>
 871   $domain
 872 </input>''')
 873     data = templ.substitute({'ip': ip, 'range': binding_range, 'domain': get_domain_name(domain_name)})
 874     return data
 875
 876
 877 def add_node_xml(node_id, port, password, version, node_ip=None, expansion=0, bindings_timeout=0, keystores=None,
 878                  retry_open_timer=1):
 879     """Generate xml for Add Node request
 880
 881     :param node_id: Ipv4 address formatted node id
 882     :type node_id: str
 883     :param node_ip: Ipv4 address of node
 884     :type node_ip: strl
 885     :param port: Node port number
 886     :type port: int
 887     :param password: TCP-MD5 password
 888     :type password: str
 889     :param version: Sxp device version
 890     :type version: str
 891     :param expansion: Bindings expansion
 892     :type expansion: int
 893     :param bindings_timeout: Specifies DHD and Reconciliation timers
 894     :type bindings_timeout: int
 895     :param keystores: SSL keystore and truststore specification
 896     :type keystores: dict
 897     :returns: String containing xml data for request
 898
 899     """
 900     tls = ''
 901     if keystores:
 902         tls = Template('''
 903         <tls>
 904             <keystore>
 905               <location>$keystore</location>
 906               <type>JKS</type>
 907               <path-type>PATH</path-type>
 908               <password>$passwd</password>
 909             </keystore>
 910             <truststore>
 911               <location>$truststore</location>
 912               <type>JKS</type>
 913               <path-type>PATH</path-type>
 914               <password>$passwd</password>
 915             </truststore>
 916             <certificate-password>$passwd</certificate-password>
 917         </tls>
 918     ''').substitute(
 919             {'keystore': keystores['keystore'], 'truststore': keystores['truststore'], 'passwd': keystores['password']})
 920
 921     templ = Template('''<input xmlns="urn:opendaylight:sxp:controller">
 922     <node-id>$id</node-id>
 923     <timers>
 924         <retry-open-time>$retry_open_timer</retry-open-time>
 925         <hold-time-min-acceptable>120</hold-time-min-acceptable>
 926         <delete-hold-down-time>$timeout</delete-hold-down-time>
 927         <hold-time-min>90</hold-time-min>
 928         <reconciliation-time>$timeout</reconciliation-time>
 929         <hold-time>90</hold-time>
 930         <hold-time-max>180</hold-time-max>
 931         <keep-alive-time>30</keep-alive-time>
 932     </timers>
 933     <mapping-expanded>$expansion</mapping-expanded>
 934     <security>
 935         $tls
 936         <password>$password</password>
 937     </security>
 938     <tcp-port>$port</tcp-port>
 939     <version>$version</version>
 940     <description>ODL SXP Controller</description>
 941     <source-ip>$ip</source-ip>
 942 </input>''')
 943     data = templ.substitute(
 944         {'ip': node_ip or node_id, 'id': node_id, 'port': port, 'password': password,
 945          'version': version, 'expansion': expansion, 'timeout': bindings_timeout, 'tls': tls,
 946          'retry_open_timer': retry_open_timer})
 947     return data
 948
 949
 950 def delete_node_xml(node_id):
 951     """Generate xml for Delete node request
 952
 953     :param node_id: Ipv4 address formatted node id
 954     :type node_id: str
 955     :returns: String containing xml data for request
 956
 957     """
 958     templ = Template('''<input xmlns="urn:opendaylight:sxp:controller">
 959   <node-id>$id</node-id>
 960 </input>''')
 961     data = templ.substitute({'id': node_id})
 962     return data
 963
 964
 965 def add_domain_xml(node_id, name):
 966     """Generate xml for Add Domain request
 967
 968     :param node_id: Id of node
 969     :type node_id: str
 970     :param name: Name of Domain
 971     :type name: str
 972     :returns: String containing xml data for request
 973
 974     """
 975     templ = Template('''<input>
 976   <node-id xmlns="urn:opendaylight:sxp:controller">$id</node-id>
 977   <domain-name xmlns="urn:opendaylight:sxp:controller">$name</domain-name>
 978 </input>''')
 979     data = templ.substitute({'name': name, 'id': node_id})
 980     return data
 981
 982
 983 def delete_domain_xml(node_id, name):
 984     """Generate xml for Remove Domain request
 985
 986     :param node_id: Id of node
 987     :type node_id: str
 988     :param name: Name of Domain
 989     :type name: str
 990     :returns: String containing xml data for request
 991
 992     """
 993     return add_domain_xml(node_id, name)
 994
 995
 996 def get_domain_name(domain_name):
 997     """Generate xml for Get Bindings request
 998
 999     :param domain_name: Name of Domain
1000     :type domain_name: str
1001     :returns: String containing xml data for request
1002
1003     """
1004     if domain_name == 'global':
1005         return ''
1006     else:
1007         return '<domain-name xmlns="urn:opendaylight:sxp:controller">' + domain_name + '</domain-name>'
1008
1009
1010 def add_bindings_xml(node_id, domain, sgt, prefixes):
1011     """Generate xml for Add Bindings request
1012
1013     :param node_id: Id of node
1014     :type node_id: str
1015     :param domain: Name of Domain
1016     :type domain: str
1017     :param sgt: Security group
1018     :type sgt: int
1019     :param prefixes: List of ip-prefixes
1020     :type prefixes: str
1021     :returns: String containing xml data for request
1022
1023     """
1024     bindings = ''
1025     for prefix in prefixes.split(','):
1026         bindings += '\n' + '<ip-prefix>' + prefix + '</ip-prefix>'
1027     templ = Template('''<input>
1028   <node-id xmlns="urn:opendaylight:sxp:controller">$id</node-id>
1029   <domain-name xmlns="urn:opendaylight:sxp:controller">$name</domain-name>
1030   <binding xmlns="urn:opendaylight:sxp:controller">
1031       <sgt>$sgt</sgt>
1032       $bindings
1033   </binding>
1034 </input>''')
1035     data = templ.substitute({'name': domain, 'id': node_id, 'sgt': sgt, 'bindings': bindings})
1036     return data
1037
1038
1039 def delete_bindings_xml(node_id, domain, sgt, prefixes):
1040     """Generate xml for Remove Bindings request
1041
1042     :param node_id: Id of node
1043     :type node_id: str
1044     :param domain: Name of Domain
1045     :type domain: str
1046     :param sgt: Security group
1047     :type sgt: int
1048     :param prefixes: List of ip-prefixes
1049     :type prefixes: str
1050     :returns: String containing xml data for request
1051
1052     """
1053     return add_bindings_xml(node_id, domain, sgt, prefixes)
1054
1055
1056 def prefix_range(start, end):
1057     """Generate and concatenate ip-prefixes
1058
1059     :param start: Start index
1060     :type start: str
1061     :param end: End index
1062     :type end: str
1063     :returns: String containing concatenated ip-prefixes
1064
1065     """
1066     start = int(start)
1067     end = int(end)
1068     index = 0
1069     prefixes = ''
1070     while index < end:
1071         prefixes += get_ip_from_number(index + start) + '/32'
1072         index += 1
1073         if index < end:
1074             prefixes += ','
1075     return prefixes
1076
1077
1078 def route_definition_xml(virtual_ip, net_mask, interface):
1079     """Generate xml for Add Bindings request
1080
1081     :param interface: Network interface name
1082     :type interface: str
1083     :param net_mask: NetMask of virtual ip
1084     :type net_mask: str
1085     :param virtual_ip: Virtual ip
1086     :type virtual_ip: str
1087     :returns: String containing xml data for request
1088
1089     """
1090     templ = Template('''
1091     <routing-definition>
1092         <ip-address>$vip</ip-address>
1093         <interface>$interface</interface>
1094         <netmask>$mask</netmask>
1095     </routing-definition>
1096     ''')
1097     data = templ.substitute({'mask': net_mask, 'vip': virtual_ip, 'interface': interface})
1098     return data
1099
1100
1101 def route_definitions_xml(routes, old_routes=None):
1102     """Generate xml for Add Bindings request
1103
1104     :param routes: XML formatted data containing RouteDefinitions
1105     :type routes: str
1106     :param old_routes: Routes add to request that needs to persist
1107     :type old_routes: str
1108     :returns: String containing xml data for request
1109
1110     """
1111     if old_routes and "</sxp-cluster-route>" in old_routes:
1112         templ = Template(old_routes.replace("</sxp-cluster-route>", "$routes</sxp-cluster-route>"))
1113     else:
1114         templ = Template('''<sxp-cluster-route xmlns="urn:opendaylight:sxp:cluster:route">
1115     $routes
1116 </sxp-cluster-route>
1117     ''')
1118     data = templ.substitute({'routes': routes})
1119     return data