/*
|
* CDDL HEADER START
|
*
|
* The contents of this file are subject to the terms of the
|
* Common Development and Distribution License, Version 1.0 only
|
* (the "License"). You may not use this file except in compliance
|
* with the License.
|
*
|
* You can obtain a copy of the license at
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE
|
* or https://OpenDS.dev.java.net/OpenDS.LICENSE.
|
* See the License for the specific language governing permissions
|
* and limitations under the License.
|
*
|
* When distributing Covered Code, include this CDDL HEADER in each
|
* file and include the License file at
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE. If applicable,
|
* add the following below this CDDL HEADER, with the fields enclosed
|
* by brackets "[]" replaced with your own identifying information:
|
* Portions Copyright [yyyy] [name of copyright owner]
|
*
|
* CDDL HEADER END
|
*
|
*
|
* Copyright 2009 Sun Microsystems, Inc.
|
*/
|
|
package com.sun.opends.sdk.ldap;
|
|
|
|
import java.io.IOException;
|
import java.net.SocketAddress;
|
import java.util.concurrent.ExecutionException;
|
|
import javax.net.ssl.SSLEngine;
|
|
import org.opends.sdk.*;
|
import org.opends.sdk.requests.Requests;
|
import org.opends.sdk.requests.StartTLSExtendedRequest;
|
import org.opends.sdk.responses.ExtendedResult;
|
import org.opends.sdk.responses.Responses;
|
import org.opends.sdk.responses.Result;
|
|
import com.sun.grizzly.CompletionHandler;
|
import com.sun.grizzly.Connection;
|
import com.sun.grizzly.EmptyCompletionHandler;
|
import com.sun.grizzly.filterchain.DefaultFilterChain;
|
import com.sun.grizzly.filterchain.FilterChain;
|
import com.sun.grizzly.filterchain.TransportFilter;
|
import com.sun.grizzly.nio.transport.TCPNIOTransport;
|
import com.sun.opends.sdk.util.CompletedFutureResult;
|
import com.sun.opends.sdk.util.FutureResultTransformer;
|
import com.sun.opends.sdk.util.RecursiveFutureResult;
|
|
|
|
/**
|
* LDAP connection factory implementation.
|
*/
|
public final class LDAPConnectionFactoryImpl extends AbstractConnectionFactory
|
implements ConnectionFactory
|
{
|
|
@SuppressWarnings("unchecked")
|
private final class FutureResultImpl implements CompletionHandler<Connection>
|
{
|
private final FutureResultTransformer<Result, AsynchronousConnection> futureStartTLSResult;
|
|
private final RecursiveFutureResult<LDAPConnection, ExtendedResult> futureConnectionResult;
|
|
private LDAPConnection connection;
|
|
|
|
private FutureResultImpl(
|
final ResultHandler<? super AsynchronousConnection> handler)
|
{
|
this.futureStartTLSResult = new FutureResultTransformer<Result, AsynchronousConnection>(
|
handler)
|
{
|
|
@Override
|
protected ErrorResultException transformErrorResult(
|
final ErrorResultException errorResult)
|
{
|
// Ensure that the connection is closed.
|
try
|
{
|
connection.close();
|
}
|
catch (final Exception e)
|
{
|
// Ignore.
|
}
|
return errorResult;
|
}
|
|
|
|
@Override
|
protected LDAPConnection transformResult(final Result result)
|
throws ErrorResultException
|
{
|
return connection;
|
}
|
|
};
|
|
this.futureConnectionResult = new RecursiveFutureResult<LDAPConnection, ExtendedResult>(
|
futureStartTLSResult)
|
{
|
|
@Override
|
protected FutureResult<? extends ExtendedResult> chainResult(
|
final LDAPConnection innerResult,
|
final ResultHandler<? super ExtendedResult> handler)
|
throws ErrorResultException
|
{
|
connection = innerResult;
|
|
if (options.getSSLContext() != null && options.useStartTLS())
|
{
|
final StartTLSExtendedRequest startTLS = Requests
|
.newStartTLSExtendedRequest(options.getSSLContext());
|
return connection.extendedRequest(startTLS, handler);
|
}
|
|
if (options.getSSLContext() != null)
|
{
|
try
|
{
|
connection.startTLS(options.getSSLContext(),
|
new EmptyCompletionHandler<SSLEngine>()
|
{
|
@Override
|
public void completed(final SSLEngine result)
|
{
|
handler.handleResult(null);
|
}
|
|
|
|
@Override
|
public void failed(final Throwable throwable)
|
{
|
final Result errorResult = Responses.newResult(
|
ResultCode.CLIENT_SIDE_CONNECT_ERROR).setCause(
|
throwable).setDiagnosticMessage(
|
throwable.getMessage());
|
handler.handleErrorResult(ErrorResultException
|
.wrap(errorResult));
|
}
|
});
|
return null;
|
}
|
catch (final IOException ioe)
|
{
|
final Result errorResult = Responses.newResult(
|
ResultCode.CLIENT_SIDE_CONNECT_ERROR).setCause(ioe)
|
.setDiagnosticMessage(ioe.getMessage());
|
throw ErrorResultException.wrap(errorResult);
|
}
|
}
|
handler.handleResult(null);
|
return new CompletedFutureResult<ExtendedResult>(
|
(ExtendedResult) null);
|
}
|
|
};
|
|
futureStartTLSResult.setFutureResult(futureConnectionResult);
|
}
|
|
|
|
/**
|
* {@inheritDoc}
|
*/
|
public void cancelled()
|
{
|
// Ignore this.
|
}
|
|
|
|
/**
|
* {@inheritDoc}
|
*/
|
public void completed(final Connection connection)
|
{
|
futureConnectionResult.handleResult(adaptConnection(connection));
|
}
|
|
|
|
/**
|
* {@inheritDoc}
|
*/
|
public void failed(final Throwable throwable)
|
{
|
futureConnectionResult
|
.handleErrorResult(adaptConnectionException(throwable));
|
}
|
|
|
|
/**
|
* {@inheritDoc}
|
*/
|
public void updated(final Connection connection)
|
{
|
// Ignore this.
|
}
|
|
}
|
|
|
|
private final SocketAddress socketAddress;
|
|
private final TCPNIOTransport transport;
|
|
private final FilterChain defaultFilterChain;
|
|
private final LDAPClientFilter clientFilter;
|
|
private final LDAPOptions options;
|
|
|
|
/**
|
* Creates a new LDAP connection factory implementation which can be used to
|
* create connections to the Directory Server at the provided host and port
|
* address using provided connection options.
|
*
|
* @param address
|
* The address of the Directory Server to connect to.
|
* @param options
|
* The LDAP connection options to use when creating connections.
|
*/
|
public LDAPConnectionFactoryImpl(final SocketAddress address,
|
final LDAPOptions options)
|
{
|
TCPNIOTransport tmpTransport = null;
|
if (options instanceof GrizzlyLDAPOptions)
|
{
|
tmpTransport = ((GrizzlyLDAPOptions) options).getTCPNIOTransport();
|
}
|
if (tmpTransport == null)
|
{
|
tmpTransport = GlobalTransportFactory.getInstance().createTCPTransport();
|
}
|
this.transport = tmpTransport;
|
|
this.socketAddress = address;
|
this.options = new LDAPOptions(options);
|
this.clientFilter = new LDAPClientFilter(new LDAPReader(this.options
|
.getDecodeOptions()), 0);
|
this.defaultFilterChain = new DefaultFilterChain();
|
this.defaultFilterChain.add(new TransportFilter());
|
this.defaultFilterChain.add(clientFilter);
|
}
|
|
|
|
/**
|
* {@inheritDoc}
|
*/
|
@Override
|
public FutureResult<AsynchronousConnection> getAsynchronousConnection(
|
final ResultHandler<AsynchronousConnection> handler)
|
{
|
final FutureResultImpl future = new FutureResultImpl(handler);
|
|
try
|
{
|
future.futureConnectionResult.setFutureResult(transport.connect(
|
socketAddress, future));
|
return future.futureStartTLSResult;
|
}
|
catch (final IOException e)
|
{
|
final ErrorResultException result = adaptConnectionException(e);
|
return new CompletedFutureResult<AsynchronousConnection>(result);
|
}
|
}
|
|
|
|
private LDAPConnection adaptConnection(final Connection<?> connection)
|
{
|
// Test shows that its much faster with non block writes but risk
|
// running out of memory if the server is slow.
|
connection.configureBlocking(true);
|
connection.setProcessor(defaultFilterChain);
|
|
final LDAPConnection ldapConnection = new LDAPConnection(connection,
|
options);
|
clientFilter.registerConnection(connection, ldapConnection);
|
return ldapConnection;
|
}
|
|
|
|
private ErrorResultException adaptConnectionException(Throwable t)
|
{
|
if (t instanceof ExecutionException)
|
{
|
t = t.getCause();
|
}
|
|
final Result result = Responses.newResult(
|
ResultCode.CLIENT_SIDE_CONNECT_ERROR).setCause(t).setDiagnosticMessage(
|
t.getMessage());
|
return ErrorResultException.wrap(result);
|
}
|
}
|
