IngestTaskPipeline.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2021 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 static java.lang.Thread.sleep;
import java.time.DayOfWeek;
import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import org.openide.util.NbBundle;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.autopsy.coreutils.MessageNotifyUtil;

abstract class IngestTaskPipeline<T extends IngestTask> {

    private static final Logger logger = Logger.getLogger(IngestTaskPipeline.class.getName());
    private final IngestJobPipeline ingestJobPipeline;
    private final List<IngestModuleTemplate> moduleTemplates;
    private final List<PipelineModule<T>> modules;
    private volatile Date startTime;
    private volatile boolean running;
    private volatile PipelineModule<T> currentModule;

    IngestTaskPipeline(IngestJobPipeline ingestPipeline, List<IngestModuleTemplate> moduleTemplates) {
        this.ingestJobPipeline = ingestPipeline;
        /*
         * The creation of ingest modules from the ingest module templates has
         * been deliberately deferred to the startUp() method so that any and
         * all errors in module construction or start up can be reported to the
         * client code.
         */
        this.moduleTemplates = moduleTemplates;
        modules = new ArrayList<>();
    }

    boolean isEmpty() {
        return modules.isEmpty();
    }

    boolean isRunning() {
        return running;
    }

    List<IngestModuleError> startUp() {
        List<IngestModuleError> errors = new ArrayList<>();
        if (!running) {
            /*
             * The creation of ingest modules from the ingest module templates
             * has been deliberately deferred to the startUp() method so that
             * any and all errors in module construction or start up can be
             * reported to the client code.
             */
            createIngestModules(moduleTemplates);
            errors.addAll(startUpIngestModules());
        } else {
            errors.add(new IngestModuleError("Ingest Task Pipeline", new IngestTaskPipelineException("Pipeline already started"))); //NON-NLS                        
        }
        return errors;
    }

    private void createIngestModules(List<IngestModuleTemplate> moduleTemplates) {
        if (modules.isEmpty()) {
            for (IngestModuleTemplate template : moduleTemplates) {
                Optional<PipelineModule<T>> module = acceptModuleTemplate(template);
                if (module.isPresent()) {
                    modules.add(module.get());
                }
            }
        }
    }

    abstract Optional<PipelineModule<T>> acceptModuleTemplate(IngestModuleTemplate template);

    private List<IngestModuleError> startUpIngestModules() {
        List<IngestModuleError> errors = new ArrayList<>();
        startTime = new Date();
        running = true;
        for (PipelineModule<T> module : modules) {
            try {
                module.startUp(new IngestJobContext(ingestJobPipeline));
            } catch (Throwable ex) {
                /*
                 * A catch-all exception firewall. Start up errors for all of
                 * the ingest modules, whether checked exceptions or runtime
                 * exceptions, are reported to allow correction of all of the
                 * error conditions in one go.
                 */
                errors.add(new IngestModuleError(module.getDisplayName(), ex));
            }
        }
        return errors;
    }

    Date getStartTime() {
        Date reportedStartTime = null;
        if (startTime != null) {
            reportedStartTime = new Date(startTime.getTime());
        }
        return reportedStartTime;
    }

    List<IngestModuleError> executeTask(T task) {
        List<IngestModuleError> errors = new ArrayList<>();
        if (running) {
            if (!ingestJobPipeline.isCancelled()) {
                pauseIfScheduled();
                if (ingestJobPipeline.isCancelled()) {
                    return errors;
                }
                try {
                    prepareForTask(task);
                } catch (IngestTaskPipelineException ex) {
                    errors.add(new IngestModuleError("Ingest Task Pipeline", ex)); //NON-NLS
                    return errors;
                }
                for (PipelineModule<T> module : modules) {
                    pauseIfScheduled();
                    if (ingestJobPipeline.isCancelled()) {
                        break;
                    }
                    try {
                        currentModule = module;
                        currentModule.setProcessingStartTime();
                        module.executeTask(ingestJobPipeline, task);
                    } catch (Throwable ex) {
                        /*
                         * A catch-all exception firewall. Note that a runtime
                         * exception from a single module does not stop
                         * processing of the task by the other modules in the
                         * pipeline.
                         */
                        errors.add(new IngestModuleError(module.getDisplayName(), ex));
                    }
                    if (ingestJobPipeline.isCancelled()) {
                        break;
                    }
                }
            }
            try {
                cleanUpAfterTask(task);
            } catch (IngestTaskPipelineException ex) {
                errors.add(new IngestModuleError("Ingest Task Pipeline", ex)); //NON-NLS
            }
        } else {
            errors.add(new IngestModuleError("Ingest Task Pipeline", new IngestTaskPipelineException("Pipeline not started or shut down"))); //NON-NLS                        
        }
        currentModule = null;
        return errors;
    }

    private void pauseIfScheduled() {
        if (ScheduledIngestPauseSettings.getPauseEnabled() == true) {
            /*
             * Calculate the date/time for the scheduled pause start by
             * "normalizing" the day of week to the current week and then
             * adjusting the hour and minute to match the scheduled hour and
             * minute.
             */
            LocalDateTime pauseStart = LocalDateTime.now();
            DayOfWeek pauseDayOfWeek = ScheduledIngestPauseSettings.getPauseDayOfWeek();
            while (pauseStart.getDayOfWeek() != pauseDayOfWeek) {
                pauseStart = pauseStart.minusDays(1);
            }
            pauseStart = pauseStart.withHour(ScheduledIngestPauseSettings.getPauseStartTimeHour());
            pauseStart = pauseStart.withMinute(ScheduledIngestPauseSettings.getPauseStartTimeMinute());
            pauseStart = pauseStart.withSecond(0);

            /*
             * Calculate the pause end date/time.
             */
            LocalDateTime pauseEnd = pauseStart.plusMinutes(ScheduledIngestPauseSettings.getPauseDurationMinutes());

            /*
             * Check whether the current date/time is in the pause interval. If
             * it is, register the ingest thread this code is running in so it
             * can be interrupted if the job is canceled, and sleep until
             * whatever time remains in the pause interval has expired.
             */
            LocalDateTime timeNow = LocalDateTime.now();
            if ((timeNow.equals(pauseStart) || timeNow.isAfter(pauseStart)) && timeNow.isBefore(pauseEnd)) {
                ingestJobPipeline.registerPausedIngestThread(Thread.currentThread());
                try {
                    long timeRemainingMillis = ChronoUnit.MILLIS.between(timeNow, pauseEnd);
                    logger.log(Level.INFO, String.format("%s pausing at %s for ~%d minutes", Thread.currentThread().getName(), LocalDateTime.now(), TimeUnit.MILLISECONDS.toMinutes(timeRemainingMillis)));
                    sleep(timeRemainingMillis);
                    logger.log(Level.INFO, String.format("%s resuming at %s", Thread.currentThread().getName(), LocalDateTime.now()));
                } catch (InterruptedException notLogged) {
                    logger.log(Level.INFO, String.format("%s resuming at %s due to sleep interrupt (ingest job canceled)", Thread.currentThread().getName(), LocalDateTime.now()));
                } finally {
                    ingestJobPipeline.unregisterPausedIngestThread(Thread.currentThread());
                }
            }
        }
    }

    abstract void prepareForTask(T task) throws IngestTaskPipelineException;

    PipelineModule<T> getCurrentlyRunningModule() {
        return currentModule;
    }

    List<IngestModuleError> shutDown() {
        List<IngestModuleError> errors = new ArrayList<>();
        if (running == true) {
            for (PipelineModule<T> module : modules) {
                try {
                    module.shutDown();
                } catch (Throwable ex) { // Catch-all exception firewall
                    errors.add(new IngestModuleError(module.getDisplayName(), ex));
                    String msg = ex.getMessage();
                    if (msg == null) {
                        /*
                         * Jython run-time errors don't seem to have a message,
                         * but have details in the string returned by
                         * toString().
                         */
                        msg = ex.toString();
                    }
                    MessageNotifyUtil.Notify.error(NbBundle.getMessage(this.getClass(), "FileIngestPipeline.moduleError.title.text", module.getDisplayName()), msg);
                }
            }
        }
        running = false;
        return errors;

    }

    abstract void cleanUpAfterTask(T task) throws IngestTaskPipelineException;

    static abstract class PipelineModule<T extends IngestTask> implements IngestModule {

        private final IngestModule module;
        private final String displayName;
        private volatile Date processingStartTime;

        PipelineModule(IngestModule module, String displayName) {
            this.module = module;
            this.displayName = displayName;
            this.processingStartTime = new Date();
        }

        String getClassName() {
            return module.getClass().getCanonicalName();
        }

        String getDisplayName() {
            return displayName;
        }

        void setProcessingStartTime() {
            processingStartTime = new Date();
        }

        Date getProcessingStartTime() {
            return new Date(processingStartTime.getTime());
        }

        @Override
        public void startUp(IngestJobContext context) throws IngestModuleException {
            module.startUp(context);
        }

        abstract void executeTask(IngestJobPipeline ingestJobPipeline, T task) throws IngestModuleException;

        @Override
        public void shutDown() {
            module.shutDown();
        }

    }

    public static class IngestTaskPipelineException extends Exception {

        private static final long serialVersionUID = 1L;

        public IngestTaskPipelineException(String message) {
            super(message);
        }

        public IngestTaskPipelineException(String message, Throwable cause) {
            super(message, cause);
        }

    }

}