add basic auth httpserver

Author: tongyy

.startHttpBasicAuthServer.sh.swp

@@ -0,0 +1,65 @@
+import BaseHTTPServer
+from SimpleHTTPServer import SimpleHTTPRequestHandler
+import sys
+import base64
+
+key = ""
+
+class AuthHandler(SimpleHTTPRequestHandler):
+    def do_HEAD(self):
+        self.send_response(200)
+        self.send_header('Content-type', 'text/html')
+        self.end_headers()
+
+    def do_AUTHHEAD(self):
+        self.send_response(401)
+        self.send_header('WWW-Authenticate', 'Basic realm=\"Test\"')
+        self.send_header('Content-type', 'text/html')
+        self.end_headers()
+
+    def do_GET(self):
+        print(self.headers)
+        global key
+        ''' Present frontpage with user authentication. '''
+        if self.headers.getheader('Authorization') == None:
+            self.do_AUTHHEAD()
+            self.wfile.write('no auth header received')
+            pass
+        elif self.headers.getheader('Authorization') == 'Basic '+key:
+            SimpleHTTPRequestHandler.do_GET(self)
+            pass
+        else:
+            self.do_AUTHHEAD()
+            self.wfile.write(self.headers.getheader('Authorization'))
+            self.wfile.write('not authenticated')
+            pass
+
+    def do_POST(self):
+        body = self.rfile.read(int(self.headers['Content-Length']))
+        print(body)
+        print(self.headers)
+        global key
+        ''' Present frontpage with user authentication. '''
+        if self.headers.getheader('Authorization') == None:
+            self.do_AUTHHEAD()
+            self.wfile.write('no auth header received')
+            pass
+        elif self.headers.getheader('Authorization') == 'Basic '+key:
+            SimpleHTTPRequestHandler.do_GET(self)
+            pass
+        else:
+            self.do_AUTHHEAD()
+            self.wfile.write(self.headers.getheader('Authorization'))
+            self.wfile.write('not authenticated')
+            pass
+def test(HandlerClass = AuthHandler,
+         ServerClass = BaseHTTPServer.HTTPServer):
+    BaseHTTPServer.test(HandlerClass, ServerClass)
+
+
+if __name__ == '__main__':
+    if len(sys.argv)<3:
+        print "usage HTTPBasicAuthServer.py [port] [username:password]"
+        sys.exit()
+    key = base64.b64encode(sys.argv[2])
+    test()

HTTPBasicAuth_Server.py

@@ -1,5 +1,4 @@
 import BaseHTTPServer,SimpleHTTPServer,SocketServer
-import logging
 import cgi
 import sys
 import socket
@@ -8,22 +7,22 @@
 PORT = int(sys.argv[1])
 
 def staticHttpServer(req):
-	logging.warning(req.headers)
+	print(req.headers)
 	SimpleHTTPServer.SimpleHTTPRequestHandler.do_GET(req)
 
 class ServerHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):
 
 	def do_GET(self):
-		"do something for GET"
 		staticHttpServer(self)
 	def do_POST(self):
-		"do something for POST"
+                body = self.rfile.read(int(self.headers['Content-Length']))
+                print(body)
 		staticHttpServer(self)
 
 Handler = ServerHandler
 
 httpd = BaseHTTPServer.HTTPServer(("", PORT),Handler)
 httpd.socket = ssl.wrap_socket (httpd.socket, certfile='../cert.pem', server_side=True)
-ip = socket.gethostbyname(socket.gethostname())
+ip = "127.0.0.1" 
 print "Serving at: https://%(interface)s:%(port)s" % dict(interface=ip or "localhost", port=PORT)
 httpd.serve_forever()

HTTPS_Server.py

@@ -1,28 +1,27 @@
 import SimpleHTTPServer
 import SocketServer
-import logging
 import cgi
 import sys
 import socket
 
 PORT = int(sys.argv[1])
 
 def staticHttpServer(req):
-	logging.warning(req.headers)
+        print(req.headers)
 	SimpleHTTPServer.SimpleHTTPRequestHandler.do_GET(req)
 
 class ServerHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):
 
 	def do_GET(self):
-		"do something for GET"
 		staticHttpServer(self)
 	def do_POST(self):
-		"do something for POST"
+                body = self.rfile.read(int(self.headers['Content-Length']))
+                print(body)
 		staticHttpServer(self)
 
 Handler = ServerHandler
 
 httpd = SocketServer.TCPServer(("", PORT), Handler)
-ip = socket.gethostbyname(socket.gethostname())
+ip = "127.0.0.1"
 print "Serving at: http://%(interface)s:%(port)s" % dict(interface=ip or "localhost", port=PORT)
-httpd.serve_forever()
+httpd.serve_forever()

HTTP_Server.py

@@ -3,11 +3,6 @@ Easy to use and make fake response over http/https
 
 Just four steps as follows
 
-* Install dependent module
-
-```shell
-npm install
-```
 
 * Run build.sh script (This step is for https protocal)
 
@@ -20,7 +15,7 @@ sh buils.sh
 sh startHttpServer.sh
 ```
 
-* :+1: Open your browser to test (https://localhost:8001 or http://localhost:8002)
+* :+1: Open your browser to test (https://localhost:8001/res.json , http://localhost:8002/res.json and http://localhost:8003/res.json with Basic Authentication(default tony/1234)
 
 
 

README.md

@@ -0,0 +1,2 @@
+pushd ./resource
+python ../HTTPBasicAuth_Server.py 8003 tony:1234 

startHttpBasicAuthServer.sh

@@ -1,2 +1,2 @@
 pushd ./resource
-nohup python ../HTTP_Server.py 8002 &> ../nohup.out &
+python ../HTTP_Server.py 8002 

startHttpServer.sh

@@ -1,2 +1,2 @@
 pushd ./resource
-nohup python ../HTTPS_Server.py 8001 &> ../nohup.out &
+python ../HTTPS_Server.py 8001 

startHttpsServer.sh

@@ -0,0 +1 @@
+kill $(ps aux | grep '[H]TTPBasicAuth_Server' | awk '{print $2}')