IngestProfiles.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2011-2017 Basis Technology Corp.
 * Contact: 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.ingest;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.openide.util.Exceptions;
import org.sleuthkit.autopsy.coreutils.ModuleSettings;
import org.sleuthkit.autopsy.coreutils.PlatformUtil;

public final class IngestProfiles {

    private static final String PROFILE_FOLDER = "IngestProfiles";
    private static final String PROFILE_NAME_KEY = "Profile_Name";
    private static final String PROFILE_DESC_KEY = "Profile_Description";
    private static final String PROFILE_FILTER_KEY = "Profile_Filter";
    private static final String PROFILE_FILE_EXT = ".properties";

    public synchronized static List<IngestProfile> getIngestProfiles() {
        File dir = Paths.get(PlatformUtil.getUserConfigDirectory(), PROFILE_FOLDER).toFile();
        File[] directoryListing = dir.listFiles();
        List<IngestProfile> profileList = new ArrayList<>();
        if (directoryListing != null) {
            for (File child : directoryListing) {
                String name = FilenameUtils.removeExtension(child.getName()); 
                String context = PROFILE_FOLDER + File.separator + name;
                String desc = ModuleSettings.getConfigSetting(context, PROFILE_DESC_KEY);
                String fileIngestFilter = ModuleSettings.getConfigSetting(context, PROFILE_FILTER_KEY);
                profileList.add(new IngestProfile(name, desc, fileIngestFilter));
            }
        }
        return profileList;
    }

    synchronized static void setProfiles(List<IngestProfile> profiles) {
        for (IngestProfile profile : profiles) {
            IngestProfile.saveProfile(profile);
        }
    }

    public static final class IngestProfile {

        private final String name;
        private final String description;
        private final String fileIngestFilter;

        IngestProfile(String name, String desc, String selectedFilter) {
            this.name = name;
            this.description = desc;
            this.fileIngestFilter = selectedFilter;
        }

        @Override
        public String toString() {
            return getName();
        }

        String getName() {
            return name;
        }

        public String getDescription() {
            return description;
        }

        public String getFileIngestFilter() {
            return fileIngestFilter;
        }

        synchronized static void deleteProfile(IngestProfile selectedProfile) {
            try {
                Files.deleteIfExists(Paths.get(PlatformUtil.getUserConfigDirectory(), PROFILE_FOLDER, selectedProfile.getName() + PROFILE_FILE_EXT));
                Files.deleteIfExists(Paths.get(PlatformUtil.getUserConfigDirectory(), selectedProfile.getName() + PROFILE_FILE_EXT));
                FileUtils.deleteDirectory(IngestJobSettings.getSavedModuleSettingsFolder(selectedProfile.getName() + File.separator).toFile());
            } catch (IOException ex) {
                Exceptions.printStackTrace(ex);
            }
        }

        synchronized static void renameProfile(String oldName, String newName) {
            if (!oldName.equals(newName)) { //if renameProfile was called with the new name being the same as the old name, it is complete already
                File oldFile = Paths.get(PlatformUtil.getUserConfigDirectory(), PROFILE_FOLDER, oldName + PROFILE_FILE_EXT).toFile();
                File newFile = Paths.get(PlatformUtil.getUserConfigDirectory(), PROFILE_FOLDER, newName + PROFILE_FILE_EXT).toFile();
                oldFile.renameTo(newFile);
                oldFile = Paths.get(PlatformUtil.getUserConfigDirectory(), oldName + PROFILE_FILE_EXT).toFile();
                newFile = Paths.get(PlatformUtil.getUserConfigDirectory(), newName + PROFILE_FILE_EXT).toFile();
                oldFile.renameTo(newFile);
                oldFile = IngestJobSettings.getSavedModuleSettingsFolder(oldName + File.separator).toFile();
                newFile = IngestJobSettings.getSavedModuleSettingsFolder(newName + File.separator).toFile();
                oldFile.renameTo(newFile);
            }
        }

        synchronized static void saveProfile(IngestProfile profile) {
            String context = PROFILE_FOLDER + File.separator + profile.getName();
            ModuleSettings.setConfigSetting(context, PROFILE_NAME_KEY, profile.getName());
            ModuleSettings.setConfigSetting(context, PROFILE_DESC_KEY, profile.getDescription());
            ModuleSettings.setConfigSetting(context, PROFILE_FILTER_KEY, profile.getFileIngestFilter());
        }
    }
}