blob: 52acae6a8c1401396bf5b7471775b3018aab37eb [file] [log] [blame]
#!/usr/bin/env python2.7
# -*- Mode: python; py-indent-offset: 4; indent-tabs-mode: nil; coding: utf-8; -*-
"""
Copyright (c) 2014 Regents of the University of California,
Arizona Board of Regents,
Colorado State University,
University Pierre & Marie Curie, Sorbonne University,
Washington University in St. Louis,
Beijing Institute of Technology
This file is part of NFD (Named Data Networking Forwarding Daemon).
See AUTHORS.md for complete list of NFD authors and contributors.
NFD is free software: you can redistribute it and/or modify it under the terms
of the GNU General Public License as published by the Free Software Foundation,
either version 3 of the License, or (at your option) any later version.
NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with
NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
"""
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
from SocketServer import ThreadingMixIn
import sys
import commands
import StringIO
import urlparse
import logging
import cgi
import argparse
import socket
class StatusHandler(BaseHTTPRequestHandler):
""" The handler class to handle requests."""
def do_GET(self):
# get the url info to decide how to respond
parsedPath = urlparse.urlparse(self.path)
resultMessage = ""
if parsedPath.path == "/robots.txt":
if self.server.robots == False:
# return User-agent: * Disallow: / to disallow robots
resultMessage = "User-agent: * \nDisallow: /\n"
self.send_response(200)
self.send_header("Content-type", "text/plain")
elif parsedPath.path == "/":
# get current nfd status, and use it as result message
resultMessage = self.getNfdStatus()
self.send_response(200)
self.send_header("Content-type", "text/html; charset=UTF-8")
else:
# non-existing content
resultMessage = '<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 '\
+ 'Transitional//EN" '\
+ '"http://www.w3.org/TR/html4/loose.dtd">\n'\
+ '<html><head><title>Object not '\
+ 'found!</title><meta http-equiv="Content-type" ' \
+ 'content="text/html;charset=UTF-8"></head>\n'\
+ '<body><h1>Object not found!</h1>'\
+ '<p>The requested URL was not found on this server.</p>'\
+ '<h2>Error 404</h2>'\
+ '</body></html>'
self.send_response(404)
self.send_header("Content-type", "text/html; charset=UTF-8")
self.end_headers()
self.wfile.write(resultMessage)
def log_message(self, format, *args):
if self.server.verbose:
logging.info("%s - %s\n" % (self.address_string(),
format % args))
def getNfdStatus(self):
"""
This function tries to call nfd-status command
to get nfd's current status, after convert it
to html format, return it to the caller
"""
(res, output) = commands.getstatusoutput('nfd-status')
htmlStr = '<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional'\
+ '//EN" "http://www.w3.org/TR/html4/loose.dtd">\n'\
+ '<html><head><title>NFD status</title>'\
+ '<meta http-equiv="Content-type" content="text/html;'\
+ 'charset=UTF-8"></head>\n<body>'
if res == 0:
# parse the output
buf = StringIO.StringIO(output)
firstLine = 0
for line in buf:
if line.endswith(":\n"):
if firstLine != 0:
htmlStr += "</ul>\n"
firstLine += 1
htmlStr += "<b>" + cgi.escape(line.strip()) + "</b><br>\n"\
+ "<ul>\n"
continue
line = line.strip()
htmlStr += "<li>" + cgi.escape(line) + "</li>\n"
buf.close()
htmlStr += "</ul>\n"
else:
# return connection error code
htmlStr = htmlStr + "<p>Cannot connect to NFD,"\
+ " Code = " + str(res) + "</p>\n"
htmlStr = htmlStr + "</body></html>"
return htmlStr
class ThreadHttpServer(ThreadingMixIn, HTTPServer):
""" Handle requests using threads """
def __init__(self, server, handler, verbose=False, robots=False):
serverAddr = server[0]
# socket.AF_UNSPEC is not supported, check whether it is v6 or v4
ipType = self.getIpType(serverAddr)
if ipType == socket.AF_INET6:
self.address_family = socket.AF_INET6
elif ipType == socket.AF_INET:
self.address_family == socket.AF_INET
else:
logging.error("The input IP address is neither IPv6 nor IPv4")
sys.exit(2)
try:
HTTPServer.__init__(self, server, handler)
except Exception as e:
logging.error(str(e))
sys.exit(2)
self.verbose = verbose
self.robots = robots
def getIpType(self, ipAddr):
""" Get ipAddr's address type """
# if ipAddr is an IPv6 addr, return AF_INET6
try:
socket.inet_pton(socket.AF_INET6, ipAddr)
return socket.AF_INET6
except socket.error:
pass
# if ipAddr is an IPv4 addr return AF_INET, if not, return None
try:
socket.inet_pton(socket.AF_INET, ipAddr)
return socket.AF_INET
except socket.error:
return None
# main function to start
def httpServer():
parser = argparse.ArgumentParser()
parser.add_argument("-p", type=int, metavar="port number",
help="Specify the HTTP server port number, default is 8080.",
dest="port", default=8080)
# if address is not specified, use 127.0.0.1
parser.add_argument("-a", default="127.0.0.1", metavar="IP address", dest="addr",
help="Specify the HTTP server IP address.")
parser.add_argument("-r", default=False, dest="robots", action="store_true",
help="Enable HTTP robots to crawl; disabled by default.")
parser.add_argument("-v", default=False, dest="verbose", action="store_true",
help="Verbose mode.")
args = vars(parser.parse_args())
localPort = args["port"]
localAddr = args["addr"]
verbose = args["verbose"]
robots = args["robots"]
# setting log message format
logging.basicConfig(format='%(asctime)s [%(levelname)s] %(message)s',
level=logging.INFO)
# if port is invalid, exit
if localPort <= 0 or localPort > 65535:
logging.error("Specified port number is invalid")
sys.exit(2)
httpd = ThreadHttpServer((localAddr, localPort), StatusHandler,
verbose, robots)
httpServerAddr = ""
if httpd.address_family == socket.AF_INET6:
httpServerAddr = "http://[%s]:%s" % (httpd.server_address[0],
httpd.server_address[1])
else:
httpServerAddr = "http://%s:%s" % (httpd.server_address[0],
httpd.server_address[1])
logging.info("Server started - at %s" % httpServerAddr)
try:
httpd.serve_forever()
except KeyboardInterrupt:
pass
httpd.server_close()
logging.info("Server stopped")
if __name__ == '__main__':
httpServer()