cmdline/sfi.py

   1 #! /usr/bin/env python
   2 from __future__ import with_statement
   3
   4 # sfi -- slice-based facility interface
   5
   6 import sys
   7 import os, os.path
   8 import tempfile
   9 from optparse import OptionParser
  10 from geni.util.cert import Keypair, Certificate
  11 from geni.util.credential import Credential
  12 from geni.util.geniclient import GeniClient, ServerException
  13 from geni.util.gid import create_uuid
  14 from geni.util.record import GeniRecord
  15
  16 sfi_dir = os.path.expanduser("~/.sfi/")
  17 slicemgr = None
  18 registry = None
  19 user = None
  20 authority = None
  21 verbose = False
  22
  23 #
  24 # Establish Connection to SliceMgr and Registry Servers
  25 #
  26 def set_servers(options):
  27    global slicemgr
  28    global registry
  29    global user
  30    global authority
  31
  32    # Set SliceMgr URL
  33    if (options.sm is not None):
  34       sm_url = options.sm
  35    elif ("SFI_SM" in os.environ):
  36       sm_url = os.environ["SFI_SM"]
  37    else:
  38       print "No Known Slice Manager"
  39       sys.exit(1)
  40
  41    # Set Registry URL
  42    if (options.registry is not None):
  43       reg_url = options.registry
  44    elif ("SFI_REGISTRY" in os.environ):
  45       reg_url = os.environ["SFI_REGISTRY"]
  46    else:
  47       print "No Known Registry Server"
  48       sys.exit(1)
  49
  50    if options.verbose :
  51       print "Contacting Slice Manager at:", sm_url
  52       print "Contacting Registry at:", reg_url
  53
  54    # Set user HRN
  55    if (options.user is not None):
  56       user = options.user
  57    elif ("SFI_USER" in os.environ):
  58       user = os.environ["SFI_USER"]
  59    else:
  60       print "No Known User Name"
  61       sys.exit(1)
  62
  63    # Set authority HRN
  64    if (options.auth is not None):
  65       authority = options.auth
  66    elif ("SFI_AUTH" in os.environ):
  67       authority = os.environ["SFI_AUTH"]
  68    else:
  69       authority = None
  70
  71    # Get key and certificate
  72    key_file = get_key_file()
  73    cert_file = get_cert_file(key_file)
  74
  75    # Establish connection to server(s)
  76    slicemgr = GeniClient(sm_url, key_file, cert_file)
  77    registry = GeniClient(reg_url, key_file, cert_file)
  78    return
  79
  80 #
  81 # Get various credential and spec files
  82 #
  83 # Establishes limiting conventions
  84 #   - conflates MAs and SAs
  85 #   - assumes last token in slice name is unique
  86 #
  87 # Bootstraps credentials
  88 #   - bootstrap user credential from self-signed certificate
  89 #   - bootstrap authority credential from user credential
  90 #   - bootstrap slice credential from user credential
  91 #
  92
  93 def get_leaf(name):
  94    parts = name.split(".")
  95    return parts[-1]
  96
  97 def get_key_file():
  98    file = os.path.join(sfi_dir, get_leaf(user) + ".pkey")
  99    if (os.path.isfile(file)):
 100       return file
 101    else:
 102       print "Key file", file, "does not exist"
 103       sys.exit(-1)
 104    return
 105
 106 def get_cert_file(key_file):
 107    global verbose
 108
 109    file = os.path.join(sfi_dir, get_leaf(user) + ".cert")
 110    if (os.path.isfile(file)):
 111       return file
 112    else:
 113       k = Keypair(filename = key_file)
 114       cert = Certificate(subject=user)
 115       cert.set_pubkey(k)
 116       cert.set_issuer(k, user)
 117       cert.sign()
 118       if verbose :
 119          print "Writing self-signed certificate to", file
 120       cert.save_to_file(file)
 121       return file
 122
 123 def get_user_cred():
 124    global user
 125
 126    file = os.path.join(sfi_dir, get_leaf(user) + ".cred")
 127    if (os.path.isfile(file)):
 128       user_cred = Credential(filename=file)
 129       return user_cred
 130    else:
 131       # bootstrap user credential
 132       user_cred = registry.get_credential(None, "user", user)
 133       if user_cred:
 134          user_cred.save_to_file(file, save_parents=True)
 135          if verbose:
 136             print "Writing user credential to", file
 137          return user_cred
 138       else:
 139          print "Failed to get user credential"
 140          sys.exit(-1)
 141
 142 def get_auth_cred():
 143    global authority
 144
 145    if not authority:
 146       print "no authority specified. Use -a or set SF_AUTH"
 147       sys.exit(-1)
 148
 149    file = os.path.join(sfi_dir, get_leaf("authority") +".cred")
 150    if (os.path.isfile(file)):
 151       auth_cred = Credential(filename=file)
 152       return auth_cred
 153    else:
 154       # bootstrap authority credential from user credential
 155       user_cred = get_user_cred()
 156       auth_cred = registry.get_credential(user_cred, "sa", authority)
 157       if auth_cred:
 158          auth_cred.save_to_file(file, save_parents=True)
 159          if verbose:
 160             print "Writing authority credential to", file
 161          return auth_cred
 162       else:
 163          print "Failed to get authority credential"
 164          sys.exit(-1)
 165
 166 def get_slice_cred(name):
 167    file = os.path.join(sfi_dir, "slice_" + get_leaf(name) + ".cred")
 168    if (os.path.isfile(file)):
 169       slice_cred = Credential(filename=file)
 170       return slice_cred
 171    else:
 172       # bootstrap slice credential from user credential
 173       user_cred = get_user_cred()
 174       slice_cred = registry.get_credential(user_cred, "slice", name)
 175       if slice_cred:
 176          slice_cred.save_to_file(file, save_parents=True)
 177          if verbose:
 178             print "Writing slice credential to", file
 179          return slice_cred
 180       else:
 181          print "Failed to get slice credential"
 182          sys.exit(-1)
 183
 184 def get_rspec_file(rspec):
 185    if (os.path.isabs(rspec)):
 186       file = rspec
 187    else:
 188       file = os.path.join(sfi_dir, rspec)
 189    if (os.path.isfile(file)):
 190       return file
 191    else:
 192       print "No such rspec file", rspec
 193       sys.exit(1)
 194
 195 def get_record_file(record):
 196    if (os.path.isabs(record)):
 197       file = record
 198    else:
 199       file = os.path.join(sfi_dir, record)
 200    if (os.path.isfile(file)):
 201       return file
 202    else:
 203       print "No such registry record file", record
 204       sys.exit(1)
 205
 206 def load_publickey_string(fn):
 207    f = file(fn,"r")
 208    key_string = f.read()
 209
 210    # if the filename is a private key file, then extract the public key
 211    if "PRIVATE KEY" in key_string:
 212        outfn = tempfile.mktemp()
 213        cmd = "openssl rsa -in " + fn + " -pubout -outform PEM -out " + outfn
 214        os.system(cmd)
 215        f = file(outfn, "r")
 216        key_string = f.read()
 217        os.remove(outfn)
 218
 219    return key_string
 220
 221 #
 222 # Generate sub-command parser
 223 #
 224 def create_cmd_parser(command, additional_cmdargs = None):
 225    cmdargs = {"list": "name",
 226               "show": "name",
 227               "remove": "name",
 228               "add": "record",
 229               "update": "record",
 230               "slices": "",
 231               "resources": "[name]",
 232               "create": "name rspec",
 233               "delete": "name",
 234               "reset": "name",
 235               "start": "name",
 236               "stop": "name"
 237              }
 238
 239    if additional_cmdargs:
 240       cmdargs.update(additional_cmdargs)
 241
 242    if command not in cmdargs:
 243       print "Invalid command\n"
 244       print "Commands: ",
 245       for key in cmdargs.keys():
 246           print key+",",
 247       print ""
 248       sys.exit(2)
 249
 250    parser = OptionParser(usage="sfi [sfi_options] %s [options] %s" \
 251       % (command, cmdargs[command]))
 252    if command in ("resources"):
 253       parser.add_option("-f", "--format", dest="format",type="choice",
 254            help="display format (dns|ip|rspec)",default="rspec",
 255            choices=("dns","ip","rspec"))
 256    if command in ("list", "show", "remove"):
 257       parser.add_option("-t", "--type", dest="type",type="choice",
 258            help="type filter (user|slice|sa|ma|node|aggregate)",
 259            choices=("user","slice","sa","ma","node","aggregate", "all"),
 260            default="all")
 261    if command in ("show", "list", "resources"):
 262       parser.add_option("-o", "--output", dest="file",
 263            help="output XML to file", metavar="FILE", default=None)
 264    return parser
 265
 266 def create_parser():
 267    # Generate command line parser
 268    parser = OptionParser(usage="sfi [options] command [command_options] [command_args]",
 269         description="Commands: list,show,remove,add,update,nodes,slices,resources,create,delete,start,stop,reset")
 270    parser.add_option("-r", "--registry", dest="registry",
 271         help="root registry", metavar="URL", default=None)
 272    parser.add_option("-s", "--slicemgr", dest="sm",
 273         help="slice manager", metavar="URL", default=None)
 274    parser.add_option("-d", "--dir", dest="dir",
 275         help="working directory", metavar="PATH", default = sfi_dir)
 276    parser.add_option("-u", "--user", dest="user",
 277         help="user name", metavar="HRN", default=None)
 278    parser.add_option("-a", "--auth", dest="auth",
 279         help="authority name", metavar="HRN", default=None)
 280    parser.add_option("-v", "--verbose",
 281         action="store_true", dest="verbose", default=False,
 282         help="verbose mode")
 283    parser.disable_interspersed_args()
 284
 285    return parser
 286
 287 def dispatch(command, cmd_opts, cmd_args):
 288    globals()[command](cmd_opts, cmd_args)
 289
 290 #
 291 # Main: parse arguments and dispatch to command
 292 #
 293 def main():
 294    global verbose
 295
 296    parser = create_parser()
 297    (options, args) = parser.parse_args()
 298
 299    if len(args) <= 0:
 300         print "No command given. Use -h for help."
 301         return -1
 302
 303    command = args[0]
 304    (cmd_opts, cmd_args) = create_cmd_parser(command).parse_args(args[1:])
 305    verbose = options.verbose
 306    if verbose :
 307       print "Resgistry %s, sm %s, dir %s, user %s, auth %s" % (options.registry,
 308                                             options.sm,
 309                                             options.dir,
 310                                             options.user,
 311                                             options.auth)
 312       print "Command %s" %command
 313       if command in ("resources"):
 314          print "resources cmd_opts %s" %cmd_opts.format
 315       elif command in ("list","show","remove"):
 316          print "cmd_opts.type %s" %cmd_opts.type
 317       print "cmd_args %s" %cmd_args
 318
 319    set_servers(options)
 320
 321    try:
 322       dispatch(command, cmd_opts, cmd_args)
 323    except KeyError:
 324       print "Command not found:", command
 325       sys.exit(1)
 326
 327    return
 328
 329 #
 330 # Following functions implement the commands
 331 #
 332 # Registry-related commands
 333 #
 334
 335 # list entires in named authority registry
 336 def list(opts, args):
 337    global registry
 338    user_cred = get_user_cred()
 339    list = registry.list(user_cred, args[0])
 340    list = filter_records(opts.type, list)
 341    display_records(list)
 342    if opts.file:
 343        save_records_to_file(opts.file, list)
 344    return
 345
 346 # show named registry record
 347 def show(opts, args):
 348    global registry
 349    user_cred = get_user_cred()
 350    records = registry.resolve(user_cred, args[0])
 351    records = filter_records(opts.type, records)
 352    if not records:
 353       print "No record of type", opts.type
 354    display_records(records, True)
 355    if opts.file:
 356        save_records_to_file(opts.file, records)
 357    return
 358
 359 # removed named registry record
 360 #   - have to first retrieve the record to be removed
 361 def remove(opts, args):
 362    global registry
 363    auth_cred = get_auth_cred()
 364    return registry.remove(auth_cred, opts.type, args[0])
 365
 366 # add named registry record
 367 def add(opts, args):
 368    global registry
 369    auth_cred = get_auth_cred()
 370    rec_file = get_record_file(args[0])
 371    record = load_record_from_file(rec_file)
 372
 373    # check and see if we need to create a gid for this record. The creator
 374    # of the record signals this by filling in the create_gid, create_gid_hrn,
 375    # and create_gid_key members.
 376    # (note: we'd use an unsigned GID in the record instead, but pyOpenSSL is
 377    #   broken and has no way for us to get the key back out of the gid)
 378    geni_info = record.get_geni_info()
 379    if "create_gid" in geni_info:
 380        gid = registry.create_gid(auth_cred, geni_info["create_gid_hrn"], create_uuid(), geni_info["create_gid_key"])
 381        record.set_gid(gid)
 382
 383        del geni_info["create_gid"]
 384        del geni_info["create_gid_hrn"]
 385        del geni_info["create_gid_key"]
 386
 387    return registry.register(auth_cred, record)
 388
 389 # update named registry entry
 390 def update(opts, args):
 391    global registry
 392    user_cred = get_user_cred()
 393    rec_file = get_record_file(args[0])
 394    record = load_record_from_file(rec_file)
 395    if record.get_type() == "user":
 396        if record.get_name() == user_cred.get_gid_object().get_hrn():
 397           cred = user_cred
 398        else:
 399           create = get_auth_cred()
 400    elif record.get_type() in ["slice"]:
 401        try:
 402            cred = get_slice_cred(record.get_name())
 403        except ServerException, e:
 404            if "PermissionError" in e.args[0]:
 405                cred = get_auth_cred()
 406            else:
 407                raise
 408    elif record.get_type() in ["sa", "ma", "node"]:
 409        cred = get_auth_cred()
 410    else:
 411        raise "unknown record type" + record.get_type()
 412    return registry.update(cred, record)
 413
 414 #
 415 # Slice-related commands
 416 #
 417
 418 # list available nodes -- now use 'resources' w/ no argument instead
 419 #def nodes(opts, args):
 420 #   global slicemgr
 421 #   user_cred = get_user_cred()
 422 #   if not opts.format:
 423 #      context = None
 424 #   else:
 425 #      context = opts.format
 426 #   results = slicemgr.list_nodes(user_cred)
 427 #   if opts.format in ['rspec']:
 428 #      display_rspec(results)
 429 #   else:
 430 #      display_list(results)
 431 #   if (opts.file is not None):
 432 #      rspec = slicemgr.list_nodes(user_cred)
 433 #      save_rspec_to_file(rspec, opts.file)
 434 #   return
 435
 436 # list instantiated slices
 437 def slices(opts, args):
 438    global slicemgr
 439    user_cred = get_user_cred()
 440    results = slicemgr.list_slices(user_cred)
 441    display_list(results)
 442    return
 443
 444 # show rspec for named slice
 445 def resources(opts, args):
 446    global slicemgr
 447    if args:
 448        slice_cred = get_slice_cred(args[0])
 449        result = slicemgr.get_resources(slice_cred, args[0])
 450    else:
 451        user_cred = get_user_cred()
 452        result = slicemgr.get_resources(user_cred)
 453    display_rspec(result)
 454    if (opts.file is not None):
 455       save_rspec_to_file(opts.file, result)
 456    return
 457
 458 # created named slice with given rspec
 459 def create(opts, args):
 460    global slicemgr
 461    slice_hrn = args[0]
 462    slice_cred = get_slice_cred(slice_hrn)
 463    rspec_file = get_rspec_file(args[1])
 464    with open(rspec_file) as f:
 465       rspec = f.read()
 466    return slicemgr.create_slice(slice_cred, slice_hrn, rspec)
 467
 468 # delete named slice
 469 def delete(opts, args):
 470    global slicemgr
 471    slice_hrn = args[0]
 472    slice_cred = get_slice_cred(slice_hrn)
 473
 474    return slicemgr.delete_slice(slice_cred, slice_hrn)
 475
 476 # start named slice
 477 def start(opts, args):
 478    global slicemgr
 479    slice_cred = get_slice_cred(args[0])
 480    return slicemgr.start_slice(slice_cred)
 481
 482 # stop named slice
 483 def stop(opts, args):
 484    global slicemgr
 485    slice_cred = get_slice_cred(args[0])
 486    return slicemgr.stop_slice(slice_cred)
 487
 488 # reset named slice
 489 def reset(opts, args):
 490    global slicemgr
 491    slice_cred = get_slice_cred(args[0])
 492    return slicemgr.reset_slice(slice_cred)
 493
 494 #
 495 #
 496 # Display, Save, and Filter RSpecs and Records
 497 #   - to be replace by EMF-generated routines
 498 #
 499 #
 500
 501 def display_rspec(rspec):
 502    print rspec
 503    return
 504
 505 def display_list(results):
 506     for result in results:
 507         print result
 508
 509 def save_rspec_to_file(rspec, filename):
 510    if not filename.startswith(os.sep):
 511        filename = sfi_dir + filename
 512    if not filename.endswith(".rspec"):
 513        filename = filename + ".rspec"
 514
 515    f = open(filename, 'w')
 516    f.write(rspec)
 517    f.close()
 518    return
 519
 520 def display_records(recordList, dump = False):
 521    for record in recordList:
 522       display_record(record, dump)
 523
 524 def display_record(record, dump = False):
 525    if dump:
 526        record.dump(False)
 527    else:
 528        info = record.getdict()
 529        print "%s (%s)" % (info['hrn'], info['type'])
 530    return
 531
 532 def filter_records(type, records):
 533    filtered_records = []
 534    for record in records:
 535        if (record.get_type() == type) or (type == "all"):
 536            filtered_records.append(record)
 537    return filtered_records
 538
 539 def save_records_to_file(filename, recordList):
 540    index = 0
 541    for record in recordList:
 542        if index>0:
 543            save_record_to_file(filename + "." + str(index), record)
 544        else:
 545            save_record_to_file(filename, record)
 546        index = index + 1
 547
 548 def save_record_to_file(filename, record):
 549    if not filename.startswith(os.sep):
 550        filename = sfi_dir + filename
 551    str = record.save_to_string()
 552    file(filename, "w").write(str)
 553    return
 554
 555 def load_record_from_file(filename):
 556    str = file(filename, "r").read()
 557    record = GeniRecord(string=str)
 558    return record
 559
 560 if __name__=="__main__":
 561    main()