Skip to content

Instantly share code, notes, and snippets.

@fxsjy
Created April 26, 2013 06:23
Show Gist options
  • Save fxsjy/5465353 to your computer and use it in GitHub Desktop.
Save fxsjy/5465353 to your computer and use it in GitHub Desktop.
SimpleAuthServer: A SimpleHTTPServer with authentication
import BaseHTTPServer
from SimpleHTTPServer import SimpleHTTPRequestHandler
import sys
import base64
key = ""
class AuthHandler(SimpleHTTPRequestHandler):
''' Main class to present webpages and authentication. '''
def do_HEAD(self):
print "send header"
self.send_response(200)
self.send_header('Content-type', 'text/html')
self.end_headers()
def do_AUTHHEAD(self):
print "send header"
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):
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 SimpleAuthServer.py [port] [username:password]"
sys.exit()
key = base64.b64encode(sys.argv[2])
test()
@Snape3058
Copy link

Snape3058 commented Jan 16, 2019

I updated to python 3 but it is not handled properly,
(Updated using 2to3.)

Traceback (most recent call last):
   File "SimpleAuthServer.py", line 41, in
     key = base64.b64encode (sys.argv [2])
   File "/opt/rh/rh-python36/root/usr/lib64/python3.6/base64.py", line 58, in b64encode
     encoded = binascii.b2a_base64 (s, newline = False)
TypeError: a bytes-like object is required, not 'str'

I did not try the 2to3 converted version, but by modifying the 95 line of SimpleHTTPAuthServer/main.py to

SimpleHTTPAuthHandler.KEY = base64.b64encode(args.key.encode("utf-8"))

can make the program start. View here for more info.
(Although some other errors happened.)

@mauler
Copy link

mauler commented Oct 31, 2019

Python3 Version, refactored to behave like python3 -m http.serve helper. Added --username and --password for basic HTTP Auth. (Authorization: Basic)

https://gist.github.com/mauler/593caee043f5fe4623732b4db5145a82

@ebrahimiali
Copy link

Is any chance to support post request as well, probably implement it by BaseHTTPRequestHandler?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment