sfa/client/sfadump.py

   1 #! /usr/bin/env python
   2 from __future__ import with_statement
   3
   4 import sys
   5 import os, os.path
   6 import tempfile
   7 import xmlrpclib
   8 from types import StringTypes, ListType
   9 from optparse import OptionParser
  10
  11 from sfa.trust.certificate import Certificate
  12 from sfa.trust.credential import Credential
  13 from sfa.trust.gid import GID
  14 from sfa.util.record import SfaRecord
  15 from sfa.util.rspec import RSpec
  16 from sfa.util.sfalogging import logger
  17
  18 def determine_sfa_filekind(fn):
  19
  20     if fn.endswith('.gid'): return 'gid'
  21     elif fn.endswith('.cert'): return 'certificate'
  22     elif fn.endswith('cred'): return 'credential'
  23
  24     try:
  25         cred=Credential(filename=fn)
  26         return 'credential'
  27     except: pass
  28
  29     try:
  30         gid=GID(filename=fn)
  31         if gid.uuid: return 'gid'
  32     except: pass
  33
  34     try:
  35         cert = Certificate(filename = fn)
  36         return 'certificate'
  37     except: pass
  38
  39     # to be completed
  40 #    if "gidCaller" in dict:
  41 #        return "credential"
  42 #
  43 #    if "uuid" in dict:
  44 #        return "gid"
  45
  46     return "unknown"
  47
  48 def save_gid(gid):
  49    hrn = gid.get_hrn()
  50    lastpart = hrn.split(".")[-1]
  51    filename = lastpart + ".gid"
  52
  53    if os.path.exists(filename):
  54        print filename, ": already exists... skipping"
  55        return
  56
  57    print filename, ": extracting gid of", hrn
  58
  59    gid.save_to_file(filename, save_parents = True)
  60
  61 def extract_gids(cred, extract_parents):
  62    gidCaller = cred.get_gid_caller()
  63    if gidCaller:
  64        save_gid(gidCaller)
  65
  66    gidObject = cred.get_gid_object()
  67    if gidObject and ((gidCaller == None) or (gidCaller.get_hrn() != gidObject.get_hrn())):
  68        save_gid(gidObject)
  69
  70    # no such method Credential.get_parent
  71 #   if extract_parents:
  72 #       parent = cred.get_parent()
  73 #       if parent:
  74 #           extract_gids(parent, extract_parents)
  75
  76 def handle_input (filename, options):
  77     kind = determine_sfa_filekind(filename)
  78     handle_input_kind (filename,options,kind)
  79
  80 def handle_input_kind (filename, options, kind):
  81
  82
  83 # dump methods current do 'print' so let's go this road for now
  84     if kind=="certificate":
  85         cert=Certificate (filename=filename)
  86         print '--------------------',filename,'IS A',kind
  87         cert.dump(show_extensions=options.show_extensions)
  88     elif kind=="credential":
  89         cred = Credential(filename = filename)
  90         print '--------------------',filename,'IS A',kind
  91         cred.dump(dump_parents = options.dump_parents)
  92         if options.extract_gids:
  93             print '--------------------',filename,'embedded GIDS'
  94             extract_gids(cred, extract_parents = options.dump_parents)
  95     elif kind=="gid":
  96         gid = GID(filename = filename)
  97         print '--------------------',filename,'IS A',kind
  98         gid.dump(dump_parents = options.dump_parents)
  99     else:
 100         print "%s: unknown filekind '%s'"% (filename,kind)
 101
 102 def main():
 103     usage = """%prog file1 [ .. filen]
 104 display info on input files"""
 105     parser = OptionParser(usage=usage)
 106
 107     parser.add_option("-g", "--extract-gids", action="store_true", dest="extract_gids", default=False, help="Extract GIDs from credentials")
 108     parser.add_option("-p", "--dump-parents", action="store_true", dest="dump_parents", default=False, help="Show parents")
 109     parser.add_option("-e", "--extensions", action="store_true", dest="show_extensions", default="False", help="Show certificate extensions")
 110     parser.add_option("-v", "--verbose", action='count', dest='verbose', default=0)
 111     (options, args) = parser.parse_args()
 112
 113     logger.setLevelFromOptVerbose(options.verbose)
 114     if len(args) <= 0:
 115         parser.print_help()
 116         sys.exit(1)
 117     for f in args:
 118         handle_input(f,options)
 119
 120 if __name__=="__main__":
 121    main()