CentralRepoPostgresSettingsUtil.java

Go to the documentation of this file.

/*
 *
 * Autopsy Forensic Browser
 *
 * Copyright 2012-2020 Basis Technology Corp.
 *
 * Copyright 2012 42six Solutions.
 * Contact: aebadirad <at> 42six <dot> com
 * Project Contact/Architect: carrier <at> sleuthkit <dot> org
 *
 * 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.
 */
package org.sleuthkit.autopsy.centralrepository.datamodel;

import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;
import org.sleuthkit.autopsy.centralrepository.CentralRepoSettings;
import org.sleuthkit.autopsy.core.UserPreferences;
import org.sleuthkit.autopsy.core.UserPreferencesException;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.autopsy.coreutils.ModuleSettings;
import org.sleuthkit.autopsy.coreutils.TextConverter;
import org.sleuthkit.autopsy.coreutils.TextConverterException;
import org.sleuthkit.datamodel.CaseDbConnectionInfo;

public class CentralRepoPostgresSettingsUtil {
    private final static Logger LOGGER = Logger.getLogger(CentralRepoPostgresSettingsUtil.class.getName());

    private static final String PASSWORD_KEY = "db.postgresql.password";
    private static final String BULK_THRESHOLD_KEY = "db.postgresql.bulkThreshold";
    private static final String PORT_KEY = "db.postgresql.port";
    private static final String USER_KEY = "db.postgresql.user";
    private static final String DBNAME_KEY = "db.postgresql.dbName";
    private static final String HOST_KEY = "db.postgresql.host";

    private static final String MODULE_KEY = CentralRepoSettings.getInstance().getModuleSettingsKey();

    private static CentralRepoPostgresSettingsUtil instance = null;
    
    public static synchronized CentralRepoPostgresSettingsUtil getInstance() {
        if (instance == null)
            instance = new CentralRepoPostgresSettingsUtil();
        
        return instance;
    }
    
    private CentralRepoPostgresSettingsUtil() {}
    
    private void setValOrLog(ValueSetter setter, String value) {
        // ignore null values as they indicate a setting that is not set yet
        if (value == null || value.isEmpty())
            return;

        try {
            setter.set(value);
        }
        catch (CentralRepoException | NumberFormatException e) {
            LOGGER.log(Level.WARNING, "There was an error in converting central repo postgres settings", e);
        }
    }
    
    private interface ValueSetter {
        void set(String value) throws CentralRepoException, NumberFormatException;
    }
    
    public PostgresConnectionSettings loadMultiUserSettings() {
        PostgresConnectionSettings settings = new PostgresConnectionSettings();
        
        CaseDbConnectionInfo muConn;
        try {
            muConn = UserPreferences.getDatabaseConnectionInfo();
        } catch (UserPreferencesException ex) {
            LOGGER.log(Level.SEVERE, "Failed to import settings from multi-user settings.", ex);
            return settings;
        }
        
        setValOrLog((v) -> settings.setHost(v), muConn.getHost());
        setValOrLog((v) -> settings.setUserName(v), muConn.getUserName());
        setValOrLog((v) -> settings.setPassword(v), muConn.getPassword());
        
        setValOrLog((v) -> settings.setPort(Integer.parseInt(v)), muConn.getPort());
                
        return settings;
    }
    
    
    public PostgresConnectionSettings loadCustomSettings() {
        PostgresConnectionSettings settings = new PostgresConnectionSettings();
        Map<String, String> keyVals = ModuleSettings.getConfigSettings(MODULE_KEY);
        
        
        setValOrLog((v) -> settings.setHost(v), keyVals.get(HOST_KEY));
        setValOrLog((v) -> settings.setDbName(v), keyVals.get(DBNAME_KEY));
        setValOrLog((v) -> settings.setUserName(v), keyVals.get(USER_KEY));
        
        setValOrLog((v) -> settings.setPort(Integer.parseInt(v)), keyVals.get(PORT_KEY));
        setValOrLog((v) -> settings.setBulkThreshold(Integer.parseInt(v)), keyVals.get((BULK_THRESHOLD_KEY)));
        
        String passwordHex = keyVals.get(PASSWORD_KEY);
        if (passwordHex != null) {
            String password;
            try {
                password = TextConverter.convertHexTextToText(passwordHex);
            } catch (TextConverterException ex) {
                LOGGER.log(Level.WARNING, "Failed to convert password from hex text to text.", ex);
                password = null;
            }

            final String finalPassword = password;
            setValOrLog((v) -> settings.setPassword(v), finalPassword);
        }
        
        return settings;
    }

    public void saveCustomSettings(PostgresConnectionSettings settings) {
        Map<String, String> map = new HashMap<String, String>();
        map.put(HOST_KEY, settings.getHost());
        map.put(PORT_KEY, Integer.toString(settings.getPort()));
        map.put(DBNAME_KEY, settings.getDbName());
        map.put(BULK_THRESHOLD_KEY, Integer.toString(settings.getBulkThreshold()));
        map.put(USER_KEY, settings.getUserName());
        try {
            map.put(PASSWORD_KEY, TextConverter.convertTextToHexText(settings.getPassword())); // NON-NLS
        } catch (TextConverterException ex) {
            LOGGER.log(Level.SEVERE, "Failed to convert password from text to hex text.", ex);
        }
        
        ModuleSettings.setConfigSettings(MODULE_KEY, map);
    }

    public boolean areCustomSettingsChanged(PostgresConnectionSettings settings) {
        PostgresConnectionSettings saved = loadCustomSettings();
        return saved.equals(settings);
    }
}