just an identation error correction
[nodemanager.git] / bootauth.py
1 #!/usr/bin/python
2 #
3 # Obtaining a node session key. Usually, the Boot
4 # Manager obtains it, then writes it to /etc/planetlab/session.
5 #
6 # Mark Huang <mlhuang@cs.princeton.edu>
7 # Copyright (C) 2006 The Trustees of Princeton University
8 #
9
10 import os, sys
11 import getopt
12
13 from config import Config
14 from plcapi import PLCAPI
15
16 def main():
17     # Defaults
18     config = None
19     node_id = None
20     key = None
21
22     # Help
23     def usage():
24         print "Usage: %s [OPTION]..." % sys.argv[0]
25         print "Options:"
26         print "     -f, --config=FILE       PLC configuration file (default: /etc/planetlab/plc_config)"
27         print "     -n, --node-id=FILE      Node ID (or file)"
28         print "     -k, --key=FILE          Node key (or file)"
29         print "     --help                  This message"
30         sys.exit(1)
31
32     # Get options
33     try:
34         (opts, argv) = getopt.getopt(sys.argv[1:], "f:n:k:h",
35                                      ["config=", "cfg=", "file=",
36                                       "node=", "nodeid=", "node-id", "node_id",
37                                       "key=",
38                                       "help"])
39     except getopt.GetoptError, err:
40         print "Error: " + err.msg
41         usage()
42
43     for (opt, optval) in opts:
44         if opt == "-f" or opt == "--config" or opt == "--cfg" or opt == "--file":
45             config = Config(optval)
46         elif opt == "-n" or opt == "--node" or opt == "--nodeid" or opt == "--node-id" or opt == "--node_id":
47             if os.path.exists(optval):
48                 node_id = file(optval).read().strip()
49             else:
50                 node_id = int(optval)
51         elif opt == "-k" or opt == "--key":
52             if os.path.exists(optval):
53                 key = file(optval).read().strip()
54             else:
55                 key = optval
56         else:
57             usage()
58
59     if config is None:
60         config = Config()
61
62     if node_id is None or \
63        key is None:
64         usage()
65
66     # Authenticate as the Boot Manager would and get a session key
67     plc = PLCAPI(config.plc_api_uri, config.cacert, (node_id, key))
68     session = plc.BootGetNodeDetails()['session']
69
70     plc = PLCAPI(config.plc_api_uri, config.cacert, session)
71     assert session == plc.GetSession()
72
73     print session
74
75 if __name__ == '__main__':
76     main()