sfa/server/sfa-clean-peer-records.py

   1 #!/usr/bin/python
   2
   3 import sys
   4 import os
   5 import traceback
   6 from sfa.util.namespace import *
   7 from sfa.util.table import SfaTable
   8 from sfa.util.prefixTree import prefixTree
   9 from sfa.plc.api import SfaAPI
  10 from sfa.util.config import Config
  11 from sfa.trust.certificate import Keypair
  12 from sfa.trust.hierarchy import Hierarchy
  13 from sfa.util.report import trace, error
  14 from sfa.server.registry import Registries
  15 import sfa.util.xmlrpcprotocol as xmlrpcprotocol
  16 import socket
  17
  18 def main():
  19     config = Config()
  20     if not config.SFA_REGISTRY_ENABLED:
  21         sys.exit(0)
  22
  23     # Get the path to the sfa server key/cert files from
  24     # the sfa hierarchy object
  25     sfa_hierarchy = Hierarchy()
  26     sfa_key_path = sfa_hierarchy.basedir
  27     key_file = os.path.join(sfa_key_path, "server.key")
  28     cert_file = os.path.join(sfa_key_path, "server.cert")
  29     key = Keypair(filename=key_file)
  30
  31     # get a connection to our local sfa registry
  32     # and a valid credential
  33     authority = config.SFA_INTERFACE_HRN
  34     url = 'http://%s:%s/' %(config.SFA_REGISTRY_HOST, config.SFA_REGISTRY_PORT)
  35     registry = xmlrpcprotocol.get_server(url, key_file, cert_file)
  36     sfa_api = SfaAPI(key_file = key_file, cert_file = cert_file, interface='registry')
  37     credential = sfa_api.getCredential()
  38
  39     # get peer registries
  40     registries = Registries(sfa_api)
  41     tree = prefixTree()
  42     tree.load(registries.keys())
  43
  44     # get local peer records
  45     table = SfaTable()
  46     peer_records = table.find({'~peer_authority': None})
  47     found_records = []
  48     hrn_dict = {}
  49     for record in peer_records:
  50         registry_hrn = tree.best_match(record['hrn'])
  51         if registry_hrn not in hrn_dict:
  52             hrn_dict[registry_hrn] = []
  53         hrn_dict[registry_hrn].append(record['hrn'])
  54
  55     # attempt to resolve the record at the authoritative interface
  56     for registry_hrn in hrn_dict:
  57         if registry_hrn in registries:
  58             records = []
  59             target_hrns = hrn_dict[registry_hrn]
  60             try:
  61                 records = registries[registry_hrn].Resolve(target_hrns, credential)
  62                 found_records.extend([record['hrn'] for record in records])
  63             except ServerException:
  64                 # an exception will be thrown if the record doenst exist
  65                 # if so remove the record from the local registry
  66                 continue
  67             except:
  68                 # this deosnt necessarily mean the records dont exist
  69                 # lets give them the benefit of the doubt here (for now)
  70                 found_records.extend(target_hrns)
  71                 traceback.print_exc()
  72
  73     # remove what wasnt found
  74     for peer_record in peer_records:
  75         if peer_record['hrn'] not in found_records:
  76             registries[sfa_api.hrn].Remove(peer_record['hrn'], credential, peer_record['type'])
  77
  78 if __name__ == '__main__':
  79     main()