577 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			577 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| # vim: tabstop=4 shiftwidth=4 softtabstop=4
 | |
| 
 | |
| # Copyright 2010 United States Government as represented by the
 | |
| # Administrator of the National Aeronautics and Space Administration.
 | |
| # All Rights Reserved.
 | |
| #
 | |
| #    Licensed under the Apache License, Version 2.0 (the "License"); you may
 | |
| #    not use this file except in compliance with the License. You may obtain
 | |
| #    a copy of the License at
 | |
| #
 | |
| #         http://www.apache.org/licenses/LICENSE-2.0
 | |
| #
 | |
| #    Unless required by applicable law or agreed to in writing, software
 | |
| #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 | |
| #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 | |
| #    License for the specific language governing permissions and limitations
 | |
| #    under the License.
 | |
| 
 | |
| """
 | |
| Auth driver for ldap.  Includes FakeLdapDriver.
 | |
| 
 | |
| It should be easy to create a replacement for this driver supporting
 | |
| other backends by creating another class that exposes the same
 | |
| public methods.
 | |
| """
 | |
| 
 | |
| import logging
 | |
| import sys
 | |
| 
 | |
| from nova import exception
 | |
| from nova import flags
 | |
| 
 | |
| 
 | |
| FLAGS = flags.FLAGS
 | |
| flags.DEFINE_integer('ldap_schema_version', 2,
 | |
|                      'Current version of the LDAP schema')
 | |
| flags.DEFINE_string('ldap_url', 'ldap://localhost',
 | |
|                     'Point this at your ldap server')
 | |
| flags.DEFINE_string('ldap_password', 'changeme', 'LDAP password')
 | |
| flags.DEFINE_string('ldap_user_dn', 'cn=Manager,dc=example,dc=com',
 | |
|                     'DN of admin user')
 | |
| flags.DEFINE_string('ldap_user_id_attribute', 'uid', 'Attribute to use as id')
 | |
| flags.DEFINE_string('ldap_user_name_attribute', 'cn',
 | |
|                     'Attribute to use as name')
 | |
| flags.DEFINE_string('ldap_user_unit', 'Users', 'OID for Users')
 | |
| flags.DEFINE_string('ldap_user_subtree', 'ou=Users,dc=example,dc=com',
 | |
|                     'OU for Users')
 | |
| flags.DEFINE_boolean('ldap_user_modify_only', False,
 | |
|                     'Modify attributes for users instead of creating/deleting')
 | |
| flags.DEFINE_string('ldap_project_subtree', 'ou=Groups,dc=example,dc=com',
 | |
|                     'OU for Projects')
 | |
| flags.DEFINE_string('role_project_subtree', 'ou=Groups,dc=example,dc=com',
 | |
|                     'OU for Roles')
 | |
| 
 | |
| # NOTE(vish): mapping with these flags is necessary because we're going
 | |
| #             to tie in to an existing ldap schema
 | |
| flags.DEFINE_string('ldap_cloudadmin',
 | |
|     'cn=cloudadmins,ou=Groups,dc=example,dc=com', 'cn for Cloud Admins')
 | |
| flags.DEFINE_string('ldap_itsec',
 | |
|     'cn=itsec,ou=Groups,dc=example,dc=com', 'cn for ItSec')
 | |
| flags.DEFINE_string('ldap_sysadmin',
 | |
|     'cn=sysadmins,ou=Groups,dc=example,dc=com', 'cn for Sysadmins')
 | |
| flags.DEFINE_string('ldap_netadmin',
 | |
|     'cn=netadmins,ou=Groups,dc=example,dc=com', 'cn for NetAdmins')
 | |
| flags.DEFINE_string('ldap_developer',
 | |
|     'cn=developers,ou=Groups,dc=example,dc=com', 'cn for Developers')
 | |
| 
 | |
| 
 | |
| # TODO(vish): make an abstract base class with the same public methods
 | |
| #             to define a set interface for AuthDrivers. I'm delaying
 | |
| #             creating this now because I'm expecting an auth refactor
 | |
| #             in which we may want to change the interface a bit more.
 | |
| 
 | |
| 
 | |
| class LdapDriver(object):
 | |
|     """Ldap Auth driver
 | |
| 
 | |
|     Defines enter and exit and therefore supports the with/as syntax.
 | |
|     """
 | |
| 
 | |
|     project_pattern = '(owner=*)'
 | |
|     isadmin_attribute = 'isNovaAdmin'
 | |
|     project_attribute = 'owner'
 | |
|     project_objectclass = 'groupOfNames'
 | |
| 
 | |
|     def __init__(self):
 | |
|         """Imports the LDAP module"""
 | |
|         self.ldap = __import__('ldap')
 | |
|         self.conn = None
 | |
|         if FLAGS.ldap_schema_version == 1:
 | |
|             LdapDriver.project_pattern = '(objectclass=novaProject)'
 | |
|             LdapDriver.isadmin_attribute = 'isAdmin'
 | |
|             LdapDriver.project_attribute = 'projectManager'
 | |
|             LdapDriver.project_objectclass = 'novaProject'
 | |
| 
 | |
|     def __enter__(self):
 | |
|         """Creates the connection to LDAP"""
 | |
|         self.conn = self.ldap.initialize(FLAGS.ldap_url)
 | |
|         self.conn.simple_bind_s(FLAGS.ldap_user_dn, FLAGS.ldap_password)
 | |
|         return self
 | |
| 
 | |
|     def __exit__(self, exc_type, exc_value, traceback):
 | |
|         """Destroys the connection to LDAP"""
 | |
|         self.conn.unbind_s()
 | |
|         return False
 | |
| 
 | |
|     def get_user(self, uid):
 | |
|         """Retrieve user by id"""
 | |
|         attr = self.__get_ldap_user(uid)
 | |
|         return self.__to_user(attr)
 | |
| 
 | |
|     def get_user_from_access_key(self, access):
 | |
|         """Retrieve user by access key"""
 | |
|         query = '(accessKey=%s)' % access
 | |
|         dn = FLAGS.ldap_user_subtree
 | |
|         return self.__to_user(self.__find_object(dn, query))
 | |
| 
 | |
|     def get_project(self, pid):
 | |
|         """Retrieve project by id"""
 | |
|         dn = 'cn=%s,%s' % (pid,
 | |
|                            FLAGS.ldap_project_subtree)
 | |
|         attr = self.__find_object(dn, LdapDriver.project_pattern)
 | |
|         return self.__to_project(attr)
 | |
| 
 | |
|     def get_users(self):
 | |
|         """Retrieve list of users"""
 | |
|         attrs = self.__find_objects(FLAGS.ldap_user_subtree,
 | |
|                                   '(objectclass=novaUser)')
 | |
|         users = []
 | |
|         for attr in attrs:
 | |
|             user = self.__to_user(attr)
 | |
|             if user is not None:
 | |
|                 users.append(user)
 | |
|         return users
 | |
| 
 | |
|     def get_projects(self, uid=None):
 | |
|         """Retrieve list of projects"""
 | |
|         pattern = LdapDriver.project_pattern
 | |
|         if uid:
 | |
|             pattern = "(&%s(member=%s))" % (pattern, self.__uid_to_dn(uid))
 | |
|         attrs = self.__find_objects(FLAGS.ldap_project_subtree,
 | |
|                                     pattern)
 | |
|         return [self.__to_project(attr) for attr in attrs]
 | |
| 
 | |
|     def create_user(self, name, access_key, secret_key, is_admin):
 | |
|         """Create a user"""
 | |
|         if self.__user_exists(name):
 | |
|             raise exception.Duplicate("LDAP user %s already exists" % name)
 | |
|         if FLAGS.ldap_user_modify_only:
 | |
|             if self.__ldap_user_exists(name):
 | |
|                 # Retrieve user by name
 | |
|                 user = self.__get_ldap_user(name)
 | |
|                 # Entry could be malformed, test for missing attrs.
 | |
|                 # Malformed entries are useless, replace attributes found.
 | |
|                 attr = []
 | |
|                 if 'secretKey' in user.keys():
 | |
|                     attr.append((self.ldap.MOD_REPLACE, 'secretKey',
 | |
|                     [secret_key]))
 | |
|                 else:
 | |
|                     attr.append((self.ldap.MOD_ADD, 'secretKey',
 | |
|                     [secret_key]))
 | |
|                 if 'accessKey' in user.keys():
 | |
|                     attr.append((self.ldap.MOD_REPLACE, 'accessKey',
 | |
|                     [access_key]))
 | |
|                 else:
 | |
|                     attr.append((self.ldap.MOD_ADD, 'accessKey',
 | |
|                     [access_key]))
 | |
|                 if LdapDriver.isadmin_attribute in user.keys():
 | |
|                     attr.append((self.ldap.MOD_REPLACE,
 | |
|                     LdapDriver.isadmin_attribute, [str(is_admin).upper()]))
 | |
|                 else:
 | |
|                     attr.append((self.ldap.MOD_ADD,
 | |
|                     LdapDriver.isadmin_attribute, [str(is_admin).upper()]))
 | |
|                 self.conn.modify_s(self.__uid_to_dn(name), attr)
 | |
|                 return self.get_user(name)
 | |
|             else:
 | |
|                 raise exception.NotFound("LDAP object for %s doesn't exist"
 | |
|                                          % name)
 | |
|         else:
 | |
|             attr = [
 | |
|                 ('objectclass', ['person',
 | |
|                                  'organizationalPerson',
 | |
|                                  'inetOrgPerson',
 | |
|                                  'novaUser']),
 | |
|                 ('ou', [FLAGS.ldap_user_unit]),
 | |
|                 (FLAGS.ldap_user_id_attribute, [name]),
 | |
|                 ('sn', [name]),
 | |
|                 (FLAGS.ldap_user_name_attribute, [name]),
 | |
|                 ('secretKey', [secret_key]),
 | |
|                 ('accessKey', [access_key]),
 | |
|                 (LdapDriver.isadmin_attribute, [str(is_admin).upper()]),
 | |
|             ]
 | |
|             self.conn.add_s(self.__uid_to_dn(name), attr)
 | |
|             return self.__to_user(dict(attr))
 | |
| 
 | |
|     def create_project(self, name, manager_uid,
 | |
|                        description=None, member_uids=None):
 | |
|         """Create a project"""
 | |
|         if self.__project_exists(name):
 | |
|             raise exception.Duplicate("Project can't be created because "
 | |
|                                       "project %s already exists" % name)
 | |
|         if not self.__user_exists(manager_uid):
 | |
|             raise exception.NotFound("Project can't be created because "
 | |
|                                      "manager %s doesn't exist" % manager_uid)
 | |
|         manager_dn = self.__uid_to_dn(manager_uid)
 | |
|         # description is a required attribute
 | |
|         if description is None:
 | |
|             description = name
 | |
|         members = []
 | |
|         if member_uids is not None:
 | |
|             for member_uid in member_uids:
 | |
|                 if not self.__user_exists(member_uid):
 | |
|                     raise exception.NotFound("Project can't be created "
 | |
|                                              "because user %s doesn't exist"
 | |
|                                              % member_uid)
 | |
|                 members.append(self.__uid_to_dn(member_uid))
 | |
|         # always add the manager as a member because members is required
 | |
|         if not manager_dn in members:
 | |
|             members.append(manager_dn)
 | |
|         attr = [
 | |
|             ('objectclass', [LdapDriver.project_objectclass]),
 | |
|             ('cn', [name]),
 | |
|             ('description', [description]),
 | |
|             (LdapDriver.project_attribute, [manager_dn]),
 | |
|             ('member', members)]
 | |
|         self.conn.add_s('cn=%s,%s' % (name, FLAGS.ldap_project_subtree), attr)
 | |
|         return self.__to_project(dict(attr))
 | |
| 
 | |
|     def modify_project(self, project_id, manager_uid=None, description=None):
 | |
|         """Modify an existing project"""
 | |
|         if not manager_uid and not description:
 | |
|             return
 | |
|         attr = []
 | |
|         if manager_uid:
 | |
|             if not self.__user_exists(manager_uid):
 | |
|                 raise exception.NotFound("Project can't be modified because "
 | |
|                                          "manager %s doesn't exist" %
 | |
|                                          manager_uid)
 | |
|             manager_dn = self.__uid_to_dn(manager_uid)
 | |
|             attr.append((self.ldap.MOD_REPLACE, LdapDriver.project_attribute,
 | |
|                          manager_dn))
 | |
|         if description:
 | |
|             attr.append((self.ldap.MOD_REPLACE, 'description', description))
 | |
|         self.conn.modify_s('cn=%s,%s' % (project_id,
 | |
|                                          FLAGS.ldap_project_subtree),
 | |
|                            attr)
 | |
| 
 | |
|     def add_to_project(self, uid, project_id):
 | |
|         """Add user to project"""
 | |
|         dn = 'cn=%s,%s' % (project_id, FLAGS.ldap_project_subtree)
 | |
|         return self.__add_to_group(uid, dn)
 | |
| 
 | |
|     def remove_from_project(self, uid, project_id):
 | |
|         """Remove user from project"""
 | |
|         dn = 'cn=%s,%s' % (project_id, FLAGS.ldap_project_subtree)
 | |
|         return self.__remove_from_group(uid, dn)
 | |
| 
 | |
|     def is_in_project(self, uid, project_id):
 | |
|         """Check if user is in project"""
 | |
|         dn = 'cn=%s,%s' % (project_id, FLAGS.ldap_project_subtree)
 | |
|         return self.__is_in_group(uid, dn)
 | |
| 
 | |
|     def has_role(self, uid, role, project_id=None):
 | |
|         """Check if user has role
 | |
| 
 | |
|         If project is specified, it checks for local role, otherwise it
 | |
|         checks for global role
 | |
|         """
 | |
|         role_dn = self.__role_to_dn(role, project_id)
 | |
|         return self.__is_in_group(uid, role_dn)
 | |
| 
 | |
|     def add_role(self, uid, role, project_id=None):
 | |
|         """Add role for user (or user and project)"""
 | |
|         role_dn = self.__role_to_dn(role, project_id)
 | |
|         if not self.__group_exists(role_dn):
 | |
|             # create the role if it doesn't exist
 | |
|             description = '%s role for %s' % (role, project_id)
 | |
|             self.__create_group(role_dn, role, uid, description)
 | |
|         else:
 | |
|             return self.__add_to_group(uid, role_dn)
 | |
| 
 | |
|     def remove_role(self, uid, role, project_id=None):
 | |
|         """Remove role for user (or user and project)"""
 | |
|         role_dn = self.__role_to_dn(role, project_id)
 | |
|         return self.__remove_from_group(uid, role_dn)
 | |
| 
 | |
|     def get_user_roles(self, uid, project_id=None):
 | |
|         """Retrieve list of roles for user (or user and project)"""
 | |
|         if project_id is None:
 | |
|             # NOTE(vish): This is unneccesarily slow, but since we can't
 | |
|             #             guarantee that the global roles are located
 | |
|             #             together in the ldap tree, we're doing this version.
 | |
|             roles = []
 | |
|             for role in FLAGS.allowed_roles:
 | |
|                 role_dn = self.__role_to_dn(role)
 | |
|                 if self.__is_in_group(uid, role_dn):
 | |
|                     roles.append(role)
 | |
|             return roles
 | |
|         else:
 | |
|             project_dn = 'cn=%s,%s' % (project_id, FLAGS.ldap_project_subtree)
 | |
|             query = ('(&(&(objectclass=groupOfNames)(!%s))(member=%s))' %
 | |
|                       (LdapDriver.project_pattern, self.__uid_to_dn(uid)))
 | |
|             roles = self.__find_objects(project_dn, query)
 | |
|             return [role['cn'][0] for role in roles]
 | |
| 
 | |
|     def delete_user(self, uid):
 | |
|         """Delete a user"""
 | |
|         if not self.__user_exists(uid):
 | |
|             raise exception.NotFound("User %s doesn't exist" % uid)
 | |
|         self.__remove_from_all(uid)
 | |
|         if FLAGS.ldap_user_modify_only:
 | |
|             # Delete attributes
 | |
|             attr = []
 | |
|             # Retrieve user by name
 | |
|             user = self.__get_ldap_user(uid)
 | |
|             if 'secretKey' in user.keys():
 | |
|                 attr.append((self.ldap.MOD_DELETE, 'secretKey',
 | |
|                              user['secretKey']))
 | |
|             if 'accessKey' in user.keys():
 | |
|                 attr.append((self.ldap.MOD_DELETE, 'accessKey',
 | |
|                              user['accessKey']))
 | |
|             if LdapDriver.isadmin_attribute in user.keys():
 | |
|                 attr.append((self.ldap.MOD_DELETE,
 | |
|                              LdapDriver.isadmin_attribute,
 | |
|                              user[LdapDriver.isadmin_attribute]))
 | |
|             self.conn.modify_s(self.__uid_to_dn(uid), attr)
 | |
|         else:
 | |
|             # Delete entry
 | |
|             self.conn.delete_s(self.__uid_to_dn(uid))
 | |
| 
 | |
|     def delete_project(self, project_id):
 | |
|         """Delete a project"""
 | |
|         project_dn = 'cn=%s,%s' % (project_id, FLAGS.ldap_project_subtree)
 | |
|         self.__delete_roles(project_dn)
 | |
|         self.__delete_group(project_dn)
 | |
| 
 | |
|     def modify_user(self, uid, access_key=None, secret_key=None, admin=None):
 | |
|         """Modify an existing user"""
 | |
|         if not access_key and not secret_key and admin is None:
 | |
|             return
 | |
|         attr = []
 | |
|         if access_key:
 | |
|             attr.append((self.ldap.MOD_REPLACE, 'accessKey', access_key))
 | |
|         if secret_key:
 | |
|             attr.append((self.ldap.MOD_REPLACE, 'secretKey', secret_key))
 | |
|         if admin is not None:
 | |
|             attr.append((self.ldap.MOD_REPLACE, LdapDriver.isadmin_attribute,
 | |
|                          str(admin).upper()))
 | |
|         self.conn.modify_s(self.__uid_to_dn(uid), attr)
 | |
| 
 | |
|     def __user_exists(self, uid):
 | |
|         """Check if user exists"""
 | |
|         return self.get_user(uid) is not None
 | |
| 
 | |
|     def __ldap_user_exists(self, uid):
 | |
|         """Check if the user exists in ldap"""
 | |
|         return self.__get_ldap_user(uid) is not None
 | |
| 
 | |
|     def __project_exists(self, project_id):
 | |
|         """Check if project exists"""
 | |
|         return self.get_project(project_id) is not None
 | |
| 
 | |
|     def __get_ldap_user(self, uid):
 | |
|         """Retrieve LDAP user entry by id"""
 | |
|         attr = self.__find_object(self.__uid_to_dn(uid),
 | |
|                                 '(objectclass=novaUser)')
 | |
|         return attr
 | |
| 
 | |
|     def __find_object(self, dn, query=None, scope=None):
 | |
|         """Find an object by dn and query"""
 | |
|         objects = self.__find_objects(dn, query, scope)
 | |
|         if len(objects) == 0:
 | |
|             return None
 | |
|         return objects[0]
 | |
| 
 | |
|     def __find_dns(self, dn, query=None, scope=None):
 | |
|         """Find dns by query"""
 | |
|         if scope is None:
 | |
|             # One of the flags is 0!
 | |
|             scope = self.ldap.SCOPE_SUBTREE
 | |
|         try:
 | |
|             res = self.conn.search_s(dn, scope, query)
 | |
|         except self.ldap.NO_SUCH_OBJECT:
 | |
|             return []
 | |
|         # Just return the DNs
 | |
|         return [dn for dn, _attributes in res]
 | |
| 
 | |
|     def __find_objects(self, dn, query=None, scope=None):
 | |
|         """Find objects by query"""
 | |
|         if scope is None:
 | |
|             # One of the flags is 0!
 | |
|             scope = self.ldap.SCOPE_SUBTREE
 | |
|         try:
 | |
|             res = self.conn.search_s(dn, scope, query)
 | |
|         except self.ldap.NO_SUCH_OBJECT:
 | |
|             return []
 | |
|         # Just return the attributes
 | |
|         return [attributes for dn, attributes in res]
 | |
| 
 | |
|     def __find_role_dns(self, tree):
 | |
|         """Find dns of role objects in given tree"""
 | |
|         query = ('(&(objectclass=groupOfNames)(!%s))' %
 | |
|                  LdapDriver.project_pattern)
 | |
|         return self.__find_dns(tree, query)
 | |
| 
 | |
|     def __find_group_dns_with_member(self, tree, uid):
 | |
|         """Find dns of group objects in a given tree that contain member"""
 | |
|         query = ('(&(objectclass=groupOfNames)(member=%s))' %
 | |
|                   self.__uid_to_dn(uid))
 | |
|         dns = self.__find_dns(tree, query)
 | |
|         return dns
 | |
| 
 | |
|     def __group_exists(self, dn):
 | |
|         """Check if group exists"""
 | |
|         query = '(objectclass=groupOfNames)'
 | |
|         return self.__find_object(dn, query) is not None
 | |
| 
 | |
|     @staticmethod
 | |
|     def __role_to_dn(role, project_id=None):
 | |
|         """Convert role to corresponding dn"""
 | |
|         if project_id is None:
 | |
|             return FLAGS.__getitem__("ldap_%s" % role).value
 | |
|         else:
 | |
|             return 'cn=%s,cn=%s,%s' % (role,
 | |
|                                        project_id,
 | |
|                                        FLAGS.ldap_project_subtree)
 | |
| 
 | |
|     def __create_group(self, group_dn, name, uid,
 | |
|                        description, member_uids=None):
 | |
|         """Create a group"""
 | |
|         if self.__group_exists(group_dn):
 | |
|             raise exception.Duplicate("Group can't be created because "
 | |
|                                       "group %s already exists" % name)
 | |
|         members = []
 | |
|         if member_uids is not None:
 | |
|             for member_uid in member_uids:
 | |
|                 if not self.__user_exists(member_uid):
 | |
|                     raise exception.NotFound("Group can't be created "
 | |
|                             "because user %s doesn't exist" % member_uid)
 | |
|                 members.append(self.__uid_to_dn(member_uid))
 | |
|         dn = self.__uid_to_dn(uid)
 | |
|         if not dn in members:
 | |
|             members.append(dn)
 | |
|         attr = [
 | |
|             ('objectclass', ['groupOfNames']),
 | |
|             ('cn', [name]),
 | |
|             ('description', [description]),
 | |
|             ('member', members)]
 | |
|         self.conn.add_s(group_dn, attr)
 | |
| 
 | |
|     def __is_in_group(self, uid, group_dn):
 | |
|         """Check if user is in group"""
 | |
|         if not self.__user_exists(uid):
 | |
|             raise exception.NotFound("User %s can't be searched in group "
 | |
|                     "because the user doesn't exist" % uid)
 | |
|         if not self.__group_exists(group_dn):
 | |
|             return False
 | |
|         res = self.__find_object(group_dn,
 | |
|                                  '(member=%s)' % self.__uid_to_dn(uid),
 | |
|                                  self.ldap.SCOPE_BASE)
 | |
|         return res is not None
 | |
| 
 | |
|     def __add_to_group(self, uid, group_dn):
 | |
|         """Add user to group"""
 | |
|         if not self.__user_exists(uid):
 | |
|             raise exception.NotFound("User %s can't be added to the group "
 | |
|                     "because the user doesn't exist" % uid)
 | |
|         if not self.__group_exists(group_dn):
 | |
|             raise exception.NotFound("The group at dn %s doesn't exist" %
 | |
|                                      group_dn)
 | |
|         if self.__is_in_group(uid, group_dn):
 | |
|             raise exception.Duplicate("User %s is already a member of "
 | |
|                                       "the group %s" % (uid, group_dn))
 | |
|         attr = [(self.ldap.MOD_ADD, 'member', self.__uid_to_dn(uid))]
 | |
|         self.conn.modify_s(group_dn, attr)
 | |
| 
 | |
|     def __remove_from_group(self, uid, group_dn):
 | |
|         """Remove user from group"""
 | |
|         if not self.__group_exists(group_dn):
 | |
|             raise exception.NotFound("The group at dn %s doesn't exist" %
 | |
|                                      group_dn)
 | |
|         if not self.__user_exists(uid):
 | |
|             raise exception.NotFound("User %s can't be removed from the "
 | |
|                     "group because the user doesn't exist" % uid)
 | |
|         if not self.__is_in_group(uid, group_dn):
 | |
|             raise exception.NotFound("User %s is not a member of the group" %
 | |
|                                      uid)
 | |
|         # NOTE(vish): remove user from group and any sub_groups
 | |
|         sub_dns = self.__find_group_dns_with_member(
 | |
|                 group_dn, uid)
 | |
|         for sub_dn in sub_dns:
 | |
|             self.__safe_remove_from_group(uid, sub_dn)
 | |
| 
 | |
|     def __safe_remove_from_group(self, uid, group_dn):
 | |
|         """Remove user from group, deleting group if user is last member"""
 | |
|         # FIXME(vish): what if deleted user is a project manager?
 | |
|         attr = [(self.ldap.MOD_DELETE, 'member', self.__uid_to_dn(uid))]
 | |
|         try:
 | |
|             self.conn.modify_s(group_dn, attr)
 | |
|         except self.ldap.OBJECT_CLASS_VIOLATION:
 | |
|             logging.debug("Attempted to remove the last member of a group. "
 | |
|                           "Deleting the group at %s instead.", group_dn)
 | |
|             self.__delete_group(group_dn)
 | |
| 
 | |
|     def __remove_from_all(self, uid):
 | |
|         """Remove user from all roles and projects"""
 | |
|         if not self.__user_exists(uid):
 | |
|             raise exception.NotFound("User %s can't be removed from all "
 | |
|                     "because the user doesn't exist" % uid)
 | |
|         role_dns = self.__find_group_dns_with_member(
 | |
|                 FLAGS.role_project_subtree, uid)
 | |
|         for role_dn in role_dns:
 | |
|             self.__safe_remove_from_group(uid, role_dn)
 | |
|         project_dns = self.__find_group_dns_with_member(
 | |
|                 FLAGS.ldap_project_subtree, uid)
 | |
|         for project_dn in project_dns:
 | |
|             self.__safe_remove_from_group(uid, project_dn)
 | |
| 
 | |
|     def __delete_group(self, group_dn):
 | |
|         """Delete Group"""
 | |
|         if not self.__group_exists(group_dn):
 | |
|             raise exception.NotFound("Group at dn %s doesn't exist" % group_dn)
 | |
|         self.conn.delete_s(group_dn)
 | |
| 
 | |
|     def __delete_roles(self, project_dn):
 | |
|         """Delete all roles for project"""
 | |
|         for role_dn in self.__find_role_dns(project_dn):
 | |
|             self.__delete_group(role_dn)
 | |
| 
 | |
|     @staticmethod
 | |
|     def __to_user(attr):
 | |
|         """Convert ldap attributes to User object"""
 | |
|         if attr is None:
 | |
|             return None
 | |
|         if ('accessKey' in attr.keys() and 'secretKey' in attr.keys() \
 | |
|             and LdapDriver.isadmin_attribute in attr.keys()):
 | |
|             return {
 | |
|                 'id': attr[FLAGS.ldap_user_id_attribute][0],
 | |
|                 'name': attr[FLAGS.ldap_user_name_attribute][0],
 | |
|                 'access': attr['accessKey'][0],
 | |
|                 'secret': attr['secretKey'][0],
 | |
|                 'admin': (attr[LdapDriver.isadmin_attribute][0] == 'TRUE')}
 | |
|         else:
 | |
|             return None
 | |
| 
 | |
|     def __to_project(self, attr):
 | |
|         """Convert ldap attributes to Project object"""
 | |
|         if attr is None:
 | |
|             return None
 | |
|         member_dns = attr.get('member', [])
 | |
|         return {
 | |
|             'id': attr['cn'][0],
 | |
|             'name': attr['cn'][0],
 | |
|             'project_manager_id':
 | |
|                 self.__dn_to_uid(attr[LdapDriver.project_attribute][0]),
 | |
|             'description': attr.get('description', [None])[0],
 | |
|             'member_ids': [self.__dn_to_uid(x) for x in member_dns]}
 | |
| 
 | |
|     @staticmethod
 | |
|     def __dn_to_uid(dn):
 | |
|         """Convert user dn to uid"""
 | |
|         return dn.split(',')[0].split('=')[1]
 | |
| 
 | |
|     @staticmethod
 | |
|     def __uid_to_dn(uid):
 | |
|         """Convert uid to dn"""
 | |
|         return FLAGS.ldap_user_id_attribute + '=%s,%s' \
 | |
|             % (uid, FLAGS.ldap_user_subtree)
 | |
| 
 | |
| 
 | |
| class FakeLdapDriver(LdapDriver):
 | |
|     """Fake Ldap Auth driver"""
 | |
| 
 | |
|     def __init__(self):  # pylint: disable-msg=W0231
 | |
|         __import__('nova.auth.fakeldap')
 | |
|         self.ldap = sys.modules['nova.auth.fakeldap']
 | 
