|
|
|
@ -1,61 +1,56 @@
|
|
|
|
|
"""LDAP Authentication Provider Plugin"""
|
|
|
|
|
import io
|
|
|
|
|
import os
|
|
|
|
|
import ssl
|
|
|
|
|
from typing import Optional
|
|
|
|
|
from flask_ldapconn import LDAPConn
|
|
|
|
|
from flask import current_app as app
|
|
|
|
|
from ldap3.utils.hashed import hashed
|
|
|
|
|
from ldap3.core.exceptions import LDAPPasswordIsMandatoryError, LDAPBindError
|
|
|
|
|
from ldap3 import SUBTREE, MODIFY_REPLACE, MODIFY_ADD, MODIFY_DELETE
|
|
|
|
|
from ldap3 import SUBTREE, MODIFY_REPLACE, MODIFY_ADD, MODIFY_DELETE, HASHED_SALTED_MD5
|
|
|
|
|
from werkzeug.exceptions import BadRequest, InternalServerError, NotFound
|
|
|
|
|
|
|
|
|
|
from flaschengeist import logger
|
|
|
|
|
from flaschengeist.plugins import AuthPlugin, before_role_updated
|
|
|
|
|
from flaschengeist.plugins import AuthPlugin, after_role_updated
|
|
|
|
|
from flaschengeist.models.user import User, Role, _Avatar
|
|
|
|
|
import flaschengeist.controller.userController as userController
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AuthLDAP(AuthPlugin):
|
|
|
|
|
def __init__(self, config):
|
|
|
|
|
def __init__(self, cfg):
|
|
|
|
|
super().__init__()
|
|
|
|
|
config = {"port": 389, "use_ssl": False}
|
|
|
|
|
config.update(cfg)
|
|
|
|
|
app.config.update(
|
|
|
|
|
LDAP_SERVER=config.get("host", "localhost"),
|
|
|
|
|
LDAP_PORT=config.get("port", 389),
|
|
|
|
|
LDAP_BINDDN=config.get("bind_dn", None),
|
|
|
|
|
LDAP_SECRET=config.get("secret", None),
|
|
|
|
|
LDAP_USE_SSL=config.get("use_ssl", False),
|
|
|
|
|
# That's not TLS, its dirty StartTLS on unencrypted LDAP
|
|
|
|
|
LDAP_SERVER=config["host"],
|
|
|
|
|
LDAP_PORT=config["port"],
|
|
|
|
|
LDAP_BINDDN=config["bind_dn"],
|
|
|
|
|
LDAP_USE_TLS=False,
|
|
|
|
|
LDAP_TLS_VERSION=ssl.PROTOCOL_TLS,
|
|
|
|
|
LDAP_USE_SSL=config["use_ssl"],
|
|
|
|
|
LDAP_TLS_VERSION=ssl.PROTOCOL_TLSv1_2,
|
|
|
|
|
LDAP_REQUIRE_CERT=ssl.CERT_NONE,
|
|
|
|
|
FORCE_ATTRIBUTE_VALUE_AS_LIST=True,
|
|
|
|
|
)
|
|
|
|
|
logger.warning(app.config.get("LDAP_USE_SSL"))
|
|
|
|
|
if "ca_cert" in config:
|
|
|
|
|
app.config["LDAP_CA_CERTS_FILE"] = config["ca_cert"]
|
|
|
|
|
else:
|
|
|
|
|
# Default is CERT_REQUIRED
|
|
|
|
|
app.config["LDAP_REQUIRE_CERT"] = ssl.CERT_OPTIONAL
|
|
|
|
|
if "secret" in config:
|
|
|
|
|
app.config["LDAP_SECRET"] = config["secret"]
|
|
|
|
|
self.ldap = LDAPConn(app)
|
|
|
|
|
self.base_dn = config["base_dn"]
|
|
|
|
|
self.search_dn = config.get("search_dn", "ou=people,{base_dn}").format(base_dn=self.base_dn)
|
|
|
|
|
self.group_dn = config.get("group_dn", "ou=group,{base_dn}").format(base_dn=self.base_dn)
|
|
|
|
|
self.password_hash = config.get("password_hash", "SSHA").upper()
|
|
|
|
|
self.object_classes = config.get("object_classes", ["inetOrgPerson"])
|
|
|
|
|
self.user_attributes: dict = config.get("user_attributes", {})
|
|
|
|
|
self.dn = config["base_dn"]
|
|
|
|
|
self.default_gid = config["default_gid"]
|
|
|
|
|
|
|
|
|
|
# TODO: might not be set if modify is called
|
|
|
|
|
self.root_dn = config.get("root_dn", None)
|
|
|
|
|
self.root_secret = config.get("root_secret", None)
|
|
|
|
|
if "admin_dn" in config:
|
|
|
|
|
self.admin_dn = config["admin_dn"]
|
|
|
|
|
self.admin_secret = config["admin_secret"]
|
|
|
|
|
else:
|
|
|
|
|
self.admin_dn = None
|
|
|
|
|
|
|
|
|
|
@before_role_updated
|
|
|
|
|
@after_role_updated
|
|
|
|
|
def _role_updated(role, new_name):
|
|
|
|
|
logger.debug(f"LDAP: before_role_updated called with ({role}, {new_name})")
|
|
|
|
|
self.__modify_role(role, new_name)
|
|
|
|
|
|
|
|
|
|
def login(self, user, password):
|
|
|
|
|
if not user:
|
|
|
|
|
return False
|
|
|
|
|
return self.ldap.authenticate(user.userid, password, "uid", self.base_dn)
|
|
|
|
|
return self.ldap.authenticate(user.userid, password, "uid", self.dn)
|
|
|
|
|
|
|
|
|
|
def find_user(self, userid, mail=None):
|
|
|
|
|
attr = self.__find(userid, mail)
|
|
|
|
@ -69,41 +64,42 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
self.__update(user, attr)
|
|
|
|
|
|
|
|
|
|
def create_user(self, user, password):
|
|
|
|
|
if self.root_dn is None:
|
|
|
|
|
logger.error("root_dn missing in ldap config!")
|
|
|
|
|
if self.admin_dn is None:
|
|
|
|
|
logger.error("admin_dn missing in ldap config!")
|
|
|
|
|
raise InternalServerError
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
|
|
|
|
attributes = self.user_attributes.copy()
|
|
|
|
|
if "uidNumber" in attributes:
|
|
|
|
|
self.ldap.connection.search(
|
|
|
|
|
self.search_dn,
|
|
|
|
|
"(uidNumber=*)",
|
|
|
|
|
SUBTREE,
|
|
|
|
|
attributes=["uidNumber"],
|
|
|
|
|
)
|
|
|
|
|
resp = sorted(
|
|
|
|
|
self.ldap.response(),
|
|
|
|
|
key=lambda i: i["attributes"]["uidNumber"],
|
|
|
|
|
reverse=True,
|
|
|
|
|
)
|
|
|
|
|
attributes = resp[0]["attributes"]["uidNumber"] + 1 if resp else attributes["uidNumber"]
|
|
|
|
|
dn = self.dn_template.format(
|
|
|
|
|
firstname=user.firstname,
|
|
|
|
|
lastname=user.lastname,
|
|
|
|
|
userid=user.userid,
|
|
|
|
|
mail=user.mail,
|
|
|
|
|
display_name=user.display_name,
|
|
|
|
|
base_dn=self.base_dn,
|
|
|
|
|
ldap_conn = self.ldap.connect(self.admin_dn, self.admin_secret)
|
|
|
|
|
self.ldap.connection.search(
|
|
|
|
|
"ou=user,{}".format(self.dn),
|
|
|
|
|
"(uidNumber=*)",
|
|
|
|
|
SUBTREE,
|
|
|
|
|
attributes=["uidNumber"],
|
|
|
|
|
)
|
|
|
|
|
attributes.update({
|
|
|
|
|
"sn": user.lastname,
|
|
|
|
|
"givenName": user.firstname,
|
|
|
|
|
uid_number = (
|
|
|
|
|
sorted(self.ldap.response(), key=lambda i: i["attributes"]["uidNumber"], reverse=True,)[0][
|
|
|
|
|
"attributes"
|
|
|
|
|
]["uidNumber"]
|
|
|
|
|
+ 1
|
|
|
|
|
)
|
|
|
|
|
dn = f"cn={user.firstname} {user.lastname},ou=user,{self.dn}"
|
|
|
|
|
object_class = [
|
|
|
|
|
"inetOrgPerson",
|
|
|
|
|
"posixAccount",
|
|
|
|
|
"person",
|
|
|
|
|
"organizationalPerson",
|
|
|
|
|
]
|
|
|
|
|
attributes = {
|
|
|
|
|
"sn": user.firstname,
|
|
|
|
|
"givenName": user.lastname,
|
|
|
|
|
"gidNumber": self.default_gid,
|
|
|
|
|
"homeDirectory": f"/home/{user.userid}",
|
|
|
|
|
"loginShell": "/bin/bash",
|
|
|
|
|
"uid": user.userid,
|
|
|
|
|
"userPassword": self.__hash(password),
|
|
|
|
|
})
|
|
|
|
|
ldap_conn.add(dn, self.object_classes, attributes)
|
|
|
|
|
"userPassword": hashed(HASHED_SALTED_MD5, password),
|
|
|
|
|
"uidNumber": uid_number,
|
|
|
|
|
}
|
|
|
|
|
ldap_conn.add(dn, object_class, attributes)
|
|
|
|
|
self._set_roles(user)
|
|
|
|
|
except (LDAPPasswordIsMandatoryError, LDAPBindError):
|
|
|
|
|
raise BadRequest
|
|
|
|
@ -114,10 +110,10 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
if password:
|
|
|
|
|
ldap_conn = self.ldap.connect(dn, password)
|
|
|
|
|
else:
|
|
|
|
|
if self.root_dn is None:
|
|
|
|
|
logger.error("root_dn missing in ldap config!")
|
|
|
|
|
if self.admin_dn is None:
|
|
|
|
|
logger.error("admin_dn missing in ldap config!")
|
|
|
|
|
raise InternalServerError
|
|
|
|
|
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
|
|
|
|
ldap_conn = self.ldap.connect(self.admin_dn, self.admin_secret)
|
|
|
|
|
modifier = {}
|
|
|
|
|
for name, ldap_name in [
|
|
|
|
|
("firstname", "givenName"),
|
|
|
|
@ -128,7 +124,9 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
if hasattr(user, name):
|
|
|
|
|
modifier[ldap_name] = [(MODIFY_REPLACE, [getattr(user, name)])]
|
|
|
|
|
if new_password:
|
|
|
|
|
modifier["userPassword"] = [(MODIFY_REPLACE, [self.__hash(new_password)])]
|
|
|
|
|
# TODO: Use secure hash!
|
|
|
|
|
salted_password = hashed(HASHED_SALTED_MD5, new_password)
|
|
|
|
|
modifier["userPassword"] = [(MODIFY_REPLACE, [salted_password])]
|
|
|
|
|
ldap_conn.modify(dn, modifier)
|
|
|
|
|
self._set_roles(user)
|
|
|
|
|
except (LDAPPasswordIsMandatoryError, LDAPBindError):
|
|
|
|
@ -136,7 +134,7 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
|
|
|
|
|
def get_avatar(self, user):
|
|
|
|
|
self.ldap.connection.search(
|
|
|
|
|
self.search_dn,
|
|
|
|
|
"ou=user,{}".format(self.dn),
|
|
|
|
|
"(uid={})".format(user.userid),
|
|
|
|
|
SUBTREE,
|
|
|
|
|
attributes=["jpegPhoto"],
|
|
|
|
@ -152,8 +150,8 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
raise NotFound
|
|
|
|
|
|
|
|
|
|
def set_avatar(self, user, avatar: _Avatar):
|
|
|
|
|
if self.root_dn is None:
|
|
|
|
|
logger.error("root_dn missing in ldap config!")
|
|
|
|
|
if self.admin_dn is None:
|
|
|
|
|
logger.error("admin_dn missing in ldap config!")
|
|
|
|
|
raise InternalServerError
|
|
|
|
|
|
|
|
|
|
if avatar.mimetype != "image/jpeg":
|
|
|
|
@ -174,16 +172,16 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
raise BadRequest("Unsupported image format")
|
|
|
|
|
|
|
|
|
|
dn = user.get_attribute("DN")
|
|
|
|
|
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
|
|
|
|
ldap_conn = self.ldap.connect(self.admin_dn, self.admin_secret)
|
|
|
|
|
ldap_conn.modify(dn, {"jpegPhoto": [(MODIFY_REPLACE, [avatar.binary])]})
|
|
|
|
|
|
|
|
|
|
def __find(self, userid, mail=None):
|
|
|
|
|
"""Find attributes of an user by uid or mail in LDAP"""
|
|
|
|
|
con = self.ldap.connection
|
|
|
|
|
if not con:
|
|
|
|
|
con = self.ldap.connect(self.root_dn, self.root_secret)
|
|
|
|
|
con = self.ldap.connect(self.admin_dn, self.admin_secret)
|
|
|
|
|
con.search(
|
|
|
|
|
self.search_dn,
|
|
|
|
|
f"ou=user,{self.dn}",
|
|
|
|
|
f"(| (uid={userid})(mail={mail}))" if mail else f"(uid={userid})",
|
|
|
|
|
SUBTREE,
|
|
|
|
|
attributes=["uid", "givenName", "sn", "mail"],
|
|
|
|
@ -207,12 +205,12 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
role: Role,
|
|
|
|
|
new_name: Optional[str],
|
|
|
|
|
):
|
|
|
|
|
if self.root_dn is None:
|
|
|
|
|
logger.error("root_dn missing in ldap config!")
|
|
|
|
|
if self.admin_dn is None:
|
|
|
|
|
logger.error("admin_dn missing in ldap config!")
|
|
|
|
|
raise InternalServerError
|
|
|
|
|
try:
|
|
|
|
|
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
|
|
|
|
ldap_conn.search(self.group_dn, f"(cn={role.name})", SUBTREE, attributes=["cn"])
|
|
|
|
|
ldap_conn = self.ldap.connect(self.admin_dn, self.admin_secret)
|
|
|
|
|
ldap_conn.search(f"ou=group,{self.dn}", f"(cn={role.name})", SUBTREE, attributes=["cn"])
|
|
|
|
|
if len(ldap_conn.response) > 0:
|
|
|
|
|
dn = ldap_conn.response[0]["dn"]
|
|
|
|
|
if new_name:
|
|
|
|
@ -220,33 +218,13 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
else:
|
|
|
|
|
ldap_conn.delete(dn)
|
|
|
|
|
|
|
|
|
|
except LDAPPasswordIsMandatoryError:
|
|
|
|
|
except (LDAPPasswordIsMandatoryError, LDAPBindError):
|
|
|
|
|
raise BadRequest
|
|
|
|
|
except LDAPBindError:
|
|
|
|
|
logger.debug(f"Could not bind to LDAP server", exc_info=True)
|
|
|
|
|
raise InternalServerError
|
|
|
|
|
|
|
|
|
|
def __hash(self, password):
|
|
|
|
|
if self.password_hash == "ARGON2":
|
|
|
|
|
from argon2 import PasswordHasher
|
|
|
|
|
|
|
|
|
|
return f"{{ARGON2}}{PasswordHasher().hash(password)}"
|
|
|
|
|
else:
|
|
|
|
|
from hashlib import pbkdf2_hmac, sha1
|
|
|
|
|
import base64
|
|
|
|
|
|
|
|
|
|
salt = os.urandom(16)
|
|
|
|
|
if self.password_hash == "PBKDF2":
|
|
|
|
|
rounds = 200000
|
|
|
|
|
password_hash = base64.b64encode(pbkdf2_hmac("sha512", password.encode("utf-8"), salt, rounds)).decode()
|
|
|
|
|
return f"{{PBKDF2-SHA512}}{rounds}${base64.b64encode(salt).decode()}${password_hash}"
|
|
|
|
|
else:
|
|
|
|
|
return f"{{SSHA}}{base64.b64encode(sha1(password + salt) + salt)}"
|
|
|
|
|
|
|
|
|
|
def _get_groups(self, uid):
|
|
|
|
|
groups = []
|
|
|
|
|
self.ldap.connection.search(
|
|
|
|
|
self.group_dn,
|
|
|
|
|
"ou=group,{}".format(self.dn),
|
|
|
|
|
"(memberUID={})".format(uid),
|
|
|
|
|
SUBTREE,
|
|
|
|
|
attributes=["cn"],
|
|
|
|
@ -258,7 +236,7 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
|
|
|
|
|
def _get_all_roles(self):
|
|
|
|
|
self.ldap.connection.search(
|
|
|
|
|
self.group_dn,
|
|
|
|
|
f"ou=group,{self.dn}",
|
|
|
|
|
"(cn=*)",
|
|
|
|
|
SUBTREE,
|
|
|
|
|
attributes=["cn", "gidNumber", "memberUid"],
|
|
|
|
@ -267,7 +245,8 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
|
|
|
|
|
def _set_roles(self, user: User):
|
|
|
|
|
try:
|
|
|
|
|
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
|
|
|
|
ldap_conn = self.ldap.connect(self.admin_dn, self.admin_secret)
|
|
|
|
|
|
|
|
|
|
ldap_roles = self._get_all_roles()
|
|
|
|
|
|
|
|
|
|
gid_numbers = sorted(ldap_roles, key=lambda i: i["attributes"]["gidNumber"], reverse=True)
|
|
|
|
@ -276,7 +255,7 @@ class AuthLDAP(AuthPlugin):
|
|
|
|
|
for user_role in user.roles:
|
|
|
|
|
if user_role not in [role["attributes"]["cn"][0] for role in ldap_roles]:
|
|
|
|
|
ldap_conn.add(
|
|
|
|
|
f"cn={user_role},{self.group_dn}",
|
|
|
|
|
f"cn={user_role},ou=group,{self.dn}",
|
|
|
|
|
["posixGroup"],
|
|
|
|
|
attributes={"gidNumber": gid_number},
|
|
|
|
|
)
|
|
|
|
|