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