mkdir var/log
[plcapi.git] / Server.py
1 #!/usr/bin/python
2 #
3 # Simple standalone HTTP server for testing PLCAPI
4 #
5 # Mark Huang <mlhuang@cs.princeton.edu>
6 # Copyright (C) 2006 The Trustees of Princeton University
7 #
8
9 import os
10 import sys
11 import getopt
12 import traceback
13 import BaseHTTPServer
14
15 # Append PLC to the system path
16 sys.path.append(os.path.dirname(os.path.realpath(sys.argv[0])))
17
18 from PLC.API import PLCAPI
19
20 class PLCAPIRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
21     """
22     Simple standalone HTTP request handler for testing PLCAPI.
23     """
24
25     def do_POST(self):
26         try:
27             # Read request
28             request = self.rfile.read(int(self.headers["Content-length"]))
29
30             # Handle request
31             response = self.server.api.handle(self.client_address, request)
32
33             # Write response
34             self.send_response(200)
35             self.send_header("Content-type", "text/xml")
36             self.send_header("Content-length", str(len(response)))
37             self.end_headers()
38             self.wfile.write(response)
39
40             self.wfile.flush()
41             self.connection.shutdown(1)
42
43         except Exception, e:
44             # Log error
45             sys.stderr.write(traceback.format_exc())
46             sys.stderr.flush()
47
48     def do_GET(self):
49         self.send_response(200)
50         self.send_header("Content-type", 'text/html')
51         self.end_headers()
52         self.wfile.write("""
53 <html><head>
54 <title>PLCAPI XML-RPC/SOAP Interface</title>
55 </head><body>
56 <h1>PLCAPI XML-RPC/SOAP Interface</h1>
57 <p>Please use XML-RPC or SOAP to access the PLCAPI.</p>
58 </body></html>
59 """)        
60         
61 class PLCAPIServer(BaseHTTPServer.HTTPServer):
62     """
63     Simple standalone HTTP server for testing PLCAPI.
64     """
65
66     def __init__(self, addr, config):
67         self.api = PLCAPI(config)
68         self.allow_reuse_address = 1
69         BaseHTTPServer.HTTPServer.__init__(self, addr, PLCAPIRequestHandler)
70
71 # Defaults
72 addr = "0.0.0.0"
73 port = 8000
74 config = "/etc/planetlab/plc_config"
75
76 def usage():
77     print "Usage: %s [OPTION]..." % sys.argv[0]
78     print "Options:"
79     print "     -p PORT, --port=PORT    TCP port number to listen on (default: %d)" % port
80     print "     -f FILE, --config=FILE  PLC configuration file (default: %s)" % config
81     print "     -h, --help              This message"
82     sys.exit(1)
83
84 # Get options
85 try:
86     (opts, argv) = getopt.getopt(sys.argv[1:], "p:f:h", ["port=", "config=", "help"])
87 except getopt.GetoptError, err:
88     print "Error: " + err.msg
89     usage()
90
91 for (opt, optval) in opts:
92     if opt == "-p" or opt == "--port":
93         try:
94             port = int(optval)
95         except ValueError:
96             usage()
97     elif opt == "-f" or opt == "--config":
98         config = optval
99     elif opt == "-h" or opt == "--help":
100         usage()
101
102 # Start server
103 PLCAPIServer((addr, port), config).serve_forever()