public_git
/
nginx-ldap-auth
mirror of https://github.com/nginxinc/nginx-ldap-auth.git
1
0
Fork 1
Code Issues Releases Wiki Activity

Set cookie on successful authentication

This commit is contained in:
Mattias Jiderhamn 2019-08-02 13:21:22 +02:00
parent 6fad4f3715
commit b654376e2a
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nginx-ldap-auth-daemon.py
View File

@ -50,7 +50,8 @@ class AuthHandler(BaseHTTPRequestHandler):
ctx['action'] = 'performing authorization' ctx['action'] = 'performing authorization'
auth_header = self.headers.get('Authorization') auth_header = self.headers.get('Authorization')
auth_cookie = self.get_cookie(ctx['cookiename']) cookiename = ctx['cookiename']
auth_cookie = self.get_cookie(cookiename)
if auth_cookie != None and auth_cookie != '': if auth_cookie != None and auth_cookie != '':
auth_header = "Basic " + auth_cookie auth_header = "Basic " + auth_cookie
@ -71,6 +72,9 @@ class AuthHandler(BaseHTTPRequestHandler):
ctx['action'] = 'decoding credentials' ctx['action'] = 'decoding credentials'
try: try:
if cookiename != None and cookiename != '':
ctx['cookievalue'] = auth_header[6:]
auth_decoded = base64.b64decode(auth_header[6:]) auth_decoded = base64.b64decode(auth_header[6:])
user, passwd = auth_decoded.split(':', 1) user, passwd = auth_decoded.split(':', 1)
@ -255,6 +259,10 @@ class LDAPAuthHandler(AuthHandler):
# Successfully authenticated user # Successfully authenticated user
self.send_response(200) self.send_response(200)
cookievalue = ctx['cookievalue']
if cookievalue != None:
# Cookie.SimpleCookie() causes issues with double quotes
self.send_header("Set-Cookie", "%s=%s" % (ctx['cookiename'], cookievalue))
self.end_headers() self.end_headers()
except: except: