Skip to content

Instantly share code, notes, and snippets.

@fallive

fallive/SimpleAuthServer.py

Forked from fxsjy/SimpleAuthServer.py
Created November 29, 2016 23:31
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save fallive/79ffce6596d3c5e695e531ee317d1831 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save fallive/79ffce6596d3c5e695e531ee317d1831 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @fxsjy fxsjy created this gist Apr 26, 2013.
    49 changes: 49 additions & 0 deletions SimpleAuthServer.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,49 @@
    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()