usn.net.ssl.util.StarttlsHandlerPOSTGRES Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of install-cert Show documentation

A utility to add certs to the trust store from a connection attempt, originially written by Oracle

There is a newer version: 1.0.16

package usn.net.ssl.util;

import java.net.Socket;
import java.sql.Connection;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *
 * @author AO
 */
public class StarttlsHandlerPOSTGRES implements StarttlsHandler {

    private static final Logger LOG = LoggerFactory.getLogger(Starttls.class);

    @Override
    public boolean run(String host, int port, Socket tunnel) throws Exception {
        Object ds = null;
        try {
            ds = Class.forName("org.postgresql.ds.PGSimpleDataSource").newInstance();
        } catch (Throwable t) {
            if (LOG.isDebugEnabled()) {
                LOG.debug(t.getMessage(), t);
            }
            throw new Exception("Unable to classload postgres jdbc driver. Check to ensure it's on the classpath");
        }
        try {

            ds.getClass().getMethod("setServerName", String.class).invoke(ds, host);
            ds.getClass().getMethod("setPortNumber", int.class).invoke(ds, port);
            ds.getClass().getMethod("setSsl", boolean.class).invoke(ds, true);
            ds.getClass().getMethod("setSslfactory", String.class).invoke(ds, "usn.net.ssl.util.PostgresDumperFactory");
            Connection c = (Connection) ds.getClass().getMethod("getConnection").invoke(ds, (Object[]) null);

            c.close();
            return true;
        } catch (Exception ex) {
            LOG.warn(ex.getMessage(), ex);
            if (LOG.isDebugEnabled()) {
                LOG.debug(ex.getMessage(), ex);
            }
        }
        return false;
    }

}