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 sfa_logger, sfa_logger_goes_to_console
  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     sfa_logger_goes_to_console()
 104     usage = """%prog file1 [ .. filen]
 105 display info on input files"""
 106     parser = OptionParser(usage=usage)
 107
 108     parser.add_option("-g", "--extract-gids", action="store_true", dest="extract_gids", default=False, help="Extract GIDs from credentials")
 109     parser.add_option("-p", "--dump-parents", action="store_true", dest="dump_parents", default=False, help="Show parents")
 110     parser.add_option("-e", "--extensions", action="store_true", dest="show_extensions", default="False", help="Show certificate extensions")
 111     parser.add_option("-v", "--verbose", action='count', dest='verbose', default=0)
 112     (options, args) = parser.parse_args()
 113
 114     sfa_logger().setLevelFromOptVerbose(options.verbose)
 115     if len(args) <= 0:
 116         parser.print_help()
 117         sys.exit(1)
 118     for f in args:
 119         handle_input(f,options)
 120
 121 if __name__=="__main__":
 122    main()