| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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. |
| */ |
| package org.apache.knox.gateway.security.ldap; |
| |
| import org.apache.directory.api.ldap.model.entry.DefaultModification; |
| import org.apache.directory.api.ldap.model.entry.ModificationOperation; |
| import org.apache.directory.api.ldap.model.exception.LdapException; |
| import org.apache.directory.api.ldap.model.name.Dn; |
| import org.apache.directory.api.util.FileUtils; |
| import org.apache.directory.server.core.api.CoreSession; |
| import org.apache.directory.server.core.api.DirectoryService; |
| import org.apache.directory.server.core.api.partition.Partition; |
| import org.apache.directory.server.core.factory.DefaultDirectoryServiceFactory; |
| import org.apache.directory.server.core.factory.DirectoryServiceFactory; |
| import org.apache.directory.server.core.factory.JdbmPartitionFactory; |
| import org.apache.directory.server.core.factory.PartitionFactory; |
| import org.apache.directory.server.ldap.LdapServer; |
| import org.apache.directory.server.protocol.shared.store.LdifFileLoader; |
| import org.apache.directory.server.protocol.shared.transport.TcpTransport; |
| import org.apache.directory.server.protocol.shared.transport.Transport; |
| import org.apache.log4j.PropertyConfigurator; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import java.io.File; |
| import java.io.FileNotFoundException; |
| import java.net.ServerSocket; |
| import java.util.UUID; |
| |
| public class SimpleLdapDirectoryServer { |
| |
| private static final Logger LOG = LoggerFactory.getLogger(SimpleLdapDirectoryServer.class); |
| |
| private static final String DEFAULT_PORT = "33389"; |
| |
| private DirectoryService service; |
| |
| private LdapServer server; |
| |
| public SimpleLdapDirectoryServer( String rootDn, File usersLdif, Transport... transports ) throws Exception { |
| if( !usersLdif.exists() ) { |
| throw new FileNotFoundException( usersLdif.getAbsolutePath() ); |
| } |
| |
| DirectoryService directoryService; |
| try { |
| // creating the instance here so that |
| // we we can set some properties like accesscontrol, anon access |
| // before starting up the service |
| directoryService = new SimpleDirectoryService(); |
| |
| // no need to register a shutdown hook during tests because this |
| // starts a lot of threads and slows down test execution |
| directoryService.setShutdownHookEnabled( false ); |
| } catch ( Exception e ) { |
| throw new RuntimeException( e ); |
| } |
| |
| PartitionFactory partitionFactory; |
| try { |
| String typeName = System.getProperty( "apacheds.partition.factory" ); |
| |
| if ( typeName != null ) { |
| Class<? extends PartitionFactory> type = Class.forName( typeName ).asSubclass(PartitionFactory.class); |
| partitionFactory = type.getDeclaredConstructor().newInstance(); |
| } else { |
| partitionFactory = new JdbmPartitionFactory(); |
| } |
| } catch ( Exception e ) { |
| String msg = "Error instantiating custom partition factory"; |
| LOG.error(msg, e ); |
| throw new RuntimeException(msg, e); |
| } |
| |
| DirectoryServiceFactory factory = new DefaultDirectoryServiceFactory(directoryService, partitionFactory); |
| factory.init( UUID.randomUUID().toString() ); |
| service = factory.getDirectoryService(); |
| |
| enabledPosixSchema( service ); |
| |
| Partition partition = factory.getPartitionFactory().createPartition( |
| service.getSchemaManager(), service.getDnFactory(), "users", rootDn, 500, |
| service.getInstanceLayout().getInstanceDirectory() ); |
| service.addPartition( partition ); |
| |
| CoreSession session = service.getAdminSession(); |
| LdifFileLoader lfl = new LdifFileLoader( session, usersLdif, null ); |
| lfl.execute(); |
| |
| server = new LdapServer(); |
| server.setTransports( transports ); |
| server.setDirectoryService( service ); |
| } |
| |
| private static void enabledPosixSchema( DirectoryService service ) throws LdapException { |
| service.getSchemaManager().getLoadedSchema( "nis" ).enable(); |
| service.getAdminSession().modify( |
| new Dn( "cn=nis,ou=schema" ), |
| new DefaultModification( ModificationOperation.REPLACE_ATTRIBUTE, "m-disabled", "FALSE" ) ); |
| } |
| |
| public void start() throws Exception { |
| service.startup(); |
| server.start(); |
| } |
| |
| public void stop( boolean clean ) throws Exception { |
| server.stop(); |
| service.shutdown(); |
| if( clean ) { |
| FileUtils.deleteDirectory( service.getInstanceLayout().getInstanceDirectory() ); |
| } |
| } |
| |
| public static void main( String[] args ) throws Exception { |
| PropertyConfigurator.configure( System.getProperty( "log4j.configuration" ) ); |
| |
| SimpleLdapDirectoryServer ldap; |
| |
| File file; |
| if ( args.length < 1 ) { |
| file = new File( "conf/users.ldif" ); |
| } else { |
| File dir = new File( args[0] ); |
| if( !dir.exists() || !dir.isDirectory() ) { |
| throw new FileNotFoundException( dir.getAbsolutePath() ); |
| } |
| file = new File( dir, "users.ldif" ); |
| } |
| |
| if( !file.exists() || !file.canRead() ) { |
| throw new FileNotFoundException( file.getAbsolutePath() ); |
| } |
| |
| int port = Integer.parseInt(System.getProperty("ldap.port", DEFAULT_PORT)); |
| |
| // Make sure the port is free. |
| ServerSocket socket = new ServerSocket( port ); |
| socket.close(); |
| |
| TcpTransport transport = new TcpTransport( port ); |
| ldap = new SimpleLdapDirectoryServer( "dc=hadoop,dc=apache,dc=org", file, transport ); |
| ldap.start(); |
| } |
| |
| } |