com.yammer.dropwizard.authenticator.AutoclosingLdapContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dropwizard-auth-ldap Show documentation

Dropwizard Authentication Module for LDAP using JNDI

There is a newer version: 1.0.5

package com.yammer.dropwizard.authenticator;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.naming.NamingException;
import javax.naming.ldap.InitialLdapContext;
import javax.naming.ldap.StartTlsRequest;
import javax.naming.ldap.StartTlsResponse;
import java.io.IOException;
import java.util.Hashtable;

public class AutoclosingLdapContext extends InitialLdapContext implements AutoCloseable {
    private static final Logger LOGGER = LoggerFactory.getLogger(AutoclosingLdapContext.class);
    private StartTlsResponse tls = null;

    protected AutoclosingLdapContext() throws IOException, NamingException {
        this(new Hashtable<>(), TlsOption.NONE);
    }

    public AutoclosingLdapContext(Hashtable environment, TlsOption negotiateTls) throws IOException, NamingException {
        super(environment, null);
        switch (negotiateTls) {
            case ATTEMPT:
                try {
                    tls = (StartTlsResponse) this.extendedOperation(new StartTlsRequest());
                    tls.negotiate();
                } catch (Exception err) {
                    LOGGER.info("Could not negotiate TLS", err);
                }
                break;
            case STRICT:
                tls = (StartTlsResponse) this.extendedOperation(new StartTlsRequest());
                tls.negotiate();
                break;
            case NONE:
                break;
        }
    }

    @Override
    public void close() throws NamingException {
        if (tls != null) {
            try {
                tls.close();
            } catch (Exception err) {
                LOGGER.error("Could not close TLS", err);
            }
        }
        super.close();
    }
}