ExplorerNodeActionVisitor.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2011-2014 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.directorytree;

import org.openide.util.NbBundle;
import org.sleuthkit.autopsy.actions.AddContentTagAction;
import java.awt.Toolkit;
import java.awt.Dimension;
import java.awt.Font;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.logging.Level;
import javax.swing.AbstractAction;
import javax.swing.Action;
import javax.swing.JDialog;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JTable;
import javax.swing.table.DefaultTableModel;
import org.sleuthkit.autopsy.coreutils.ContextMenuExtensionPoint;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.datamodel.Content;
import org.sleuthkit.datamodel.ContentVisitor;
import org.sleuthkit.datamodel.DerivedFile;
import org.sleuthkit.datamodel.Directory;
import org.sleuthkit.datamodel.FileSystem;
import org.sleuthkit.datamodel.Image;
import org.sleuthkit.datamodel.LocalFile;
import org.sleuthkit.datamodel.VirtualDirectory;
import org.sleuthkit.datamodel.Volume;

public class ExplorerNodeActionVisitor extends ContentVisitor.Default<List<? extends Action>> {

    private static ExplorerNodeActionVisitor instance = new ExplorerNodeActionVisitor();

    public static List<Action> getActions(Content c) {
        List<Action> actions = new ArrayList<>();

        actions.addAll(c.accept(instance));
        //TODO: fix this
        /*
         * while (c.isOnto()) { try { List<? extends Content> children =
         * c.getChildren(); if (!children.isEmpty()) { c =
         * c.getChildren().get(0); } else { return actions; } } catch
         * (TskException ex) {
         * Log.get(ExplorerNodeActionVisitor.class).log(Level.WARNING, "Error
         * getting show detail actions.", ex); return actions; }
         * actions.addAll(c.accept(instance));
         }
         */
        return actions;
    }

    ExplorerNodeActionVisitor() {
    }

    @Override
    public List<? extends Action> visit(final Image img) {
        List<Action> lst = new ArrayList<>();
        lst.add(new ImageDetails(
                NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.action.imgDetails.title"), img));
        //TODO lst.add(new ExtractAction("Extract Image", img));
        lst.add(new ExtractUnallocAction(
                NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.action.extUnallocToSingleFiles"), img));
        return lst;
    }

    @Override
    public List<? extends Action> visit(final FileSystem fs) {
        return Collections.singletonList(new FileSystemDetails(
                NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.action.fileSystemDetails.title"), fs));
    }

    @Override
    public List<? extends Action> visit(final Volume vol) {
        List<AbstractAction> lst = new ArrayList<>();
        lst.add(new VolumeDetails(
                NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.action.volumeDetails.title"), vol));
        lst.add(new ExtractUnallocAction(
                NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.action.extUnallocToSingleFile"), vol));
        return lst;
    }

    @Override
    public List<? extends Action> visit(final Directory d) {
        List<Action> actions = new ArrayList<>();
        actions.add(AddContentTagAction.getInstance());
        actions.addAll(ContextMenuExtensionPoint.getActions());
        return actions;
    }

    @Override
    public List<? extends Action> visit(final VirtualDirectory d) {
        List<Action> actions = new ArrayList<>();
        actions.add(ExtractAction.getInstance());
        actions.addAll(ContextMenuExtensionPoint.getActions());
        return actions;
    }

    @Override
    public List<? extends Action> visit(final DerivedFile d) {
        List<Action> actions = new ArrayList<>();
        actions.add(ExtractAction.getInstance());
        actions.add(AddContentTagAction.getInstance());
        actions.addAll(ContextMenuExtensionPoint.getActions());
        return actions;
    }

    @Override
    public List<? extends Action> visit(final LocalFile d) {
        List<Action> actions = new ArrayList<>();
        actions.add(ExtractAction.getInstance());
        actions.add(AddContentTagAction.getInstance());
        actions.addAll(ContextMenuExtensionPoint.getActions());
        return actions;
    }

    @Override
    public List<? extends Action> visit(final org.sleuthkit.datamodel.File d) {
        List<Action> actions = new ArrayList<>();
        actions.add(ExtractAction.getInstance());
        actions.add(AddContentTagAction.getInstance());
        actions.addAll(ContextMenuExtensionPoint.getActions());
        return actions;
    }

    @Override
    protected List<? extends Action> defaultVisit(Content di) {
        return Collections.<Action>emptyList();
    }

    //Below here are classes regarding node-specific actions
    private class VolumeDetails extends AbstractAction {

        private final String title;
        private final Volume vol;

        VolumeDetails(String title, Volume vol) {
            super(title);
            this.title = title;
            this.vol = vol;
        }

        @Override
        public void actionPerformed(ActionEvent e) {
            final JFrame frame = new JFrame(title);
            final JDialog popUpWindow = new JDialog(frame, title, true); // to make the popUp Window to be modal

            Dimension screenDimension = Toolkit.getDefaultToolkit().getScreenSize();

            // set the popUp window / JFrame
            popUpWindow.setSize(800, 400);

            int w = popUpWindow.getSize().width;
            int h = popUpWindow.getSize().height;

            // set the location of the popUp Window on the center of the screen
            popUpWindow.setLocation((screenDimension.width - w) / 2, (screenDimension.height - h) / 2);

            VolumeDetailsPanel volumeDetailPanel = new VolumeDetailsPanel();
            Boolean counter = false;

            volumeDetailPanel.setVolumeIDValue(Long.toString(vol.getAddr()));
            volumeDetailPanel.setStartValue(Long.toString(vol.getStart()));
            volumeDetailPanel.setLengthValue(Long.toString(vol.getLength()));
            volumeDetailPanel.setDescValue(vol.getDescription());
            volumeDetailPanel.setFlagsValue(vol.getFlagsAsString());
            counter = true;

            if (counter) {
                // add the volume detail panel to the popUp window
                popUpWindow.add(volumeDetailPanel);
            } else {
                // error handler if no volume matches
                JLabel error = new JLabel(
                        NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.volDetail.noVolMatchErr"));
                error.setFont(error.getFont().deriveFont(Font.BOLD, 24));
                popUpWindow.add(error);
            }

            // add the command to close the window to the button on the Volume Detail Panel
            volumeDetailPanel.setOKButtonActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent e) {
                    popUpWindow.dispose();
                }
            });
            popUpWindow.pack();
            popUpWindow.setResizable(false);
            popUpWindow.setVisible(true);

        }
    }

    private class ImageDetails extends AbstractAction {

        final String title;
        final Image img;

        ImageDetails(String title, Image img) {
            super(title);
            this.title = title;
            this.img = img;
        }

        @Override
        public void actionPerformed(ActionEvent e) {
            final JFrame frame = new JFrame(title);
            final JDialog popUpWindow = new JDialog(frame, title, true); // to make the popUp Window to be modal
            // if we select the Image Details menu

            Dimension screenDimension = Toolkit.getDefaultToolkit().getScreenSize();

            // set the popUp window / JFrame
            popUpWindow.setSize(750, 400);

            int w = popUpWindow.getSize().width;
            int h = popUpWindow.getSize().height;

            // set the location of the popUp Window on the center of the screen
            popUpWindow.setLocation((screenDimension.width - w) / 2, (screenDimension.height - h) / 2);

            ImageDetailsPanel imgDetailPanel = new ImageDetailsPanel();
            Boolean counter = false;

            imgDetailPanel.setImgNameValue(img.getName());
            imgDetailPanel.setImgTypeValue(img.getType().getName());
            imgDetailPanel.setImgSectorSizeValue(Long.toString(img.getSsize()));
            imgDetailPanel.setImgTotalSizeValue(Long.toString(img.getSize()));
            String hash = img.getMd5();
            // don't show the hash if there isn't one
            imgDetailPanel.setVisibleHashInfo(hash != null);
            imgDetailPanel.setImgHashValue(hash);

            counter = true;

            if (counter) {
                // add the volume detail panel to the popUp window
                popUpWindow.add(imgDetailPanel);
            } else {
                // error handler if no volume matches
                JLabel error = new JLabel(
                        NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.imgDetail.noVolMatchesErr"));
                error.setFont(error.getFont().deriveFont(Font.BOLD, 24));
                popUpWindow.add(error);
            }

            // add the command to close the window to the button on the Volume Detail Panel
            imgDetailPanel.setOKButtonActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent e) {
                    popUpWindow.dispose();
                }
            });

            popUpWindow.pack();
            popUpWindow.setResizable(false);
            popUpWindow.setVisible(true);
        }
    }

    private class FileSystemDetails extends AbstractAction {

        private final FileSystem fs;
        private final String title;

        FileSystemDetails(String title, FileSystem fs) {
            super(title);
            this.title = title;
            this.fs = fs;
        }

        @Override
        public void actionPerformed(ActionEvent e) {
            Dimension screenDimension = Toolkit.getDefaultToolkit().getScreenSize();

            final JFrame frame = new JFrame(title);
            final JDialog popUpWindow = new JDialog(frame, title, true); // to make the popUp Window to be modal

            // set the popUp window / JFrame
            popUpWindow.setSize(1000, 500);

            int w = popUpWindow.getSize().width;
            int h = popUpWindow.getSize().height;

            // set the location of the popUp Window on the center of the screen
            popUpWindow.setLocation((screenDimension.width - w) / 2, (screenDimension.height - h) / 2);

            String[] columnNames = new String[]{
                "fs_id", //NON-NLS
                "img_offset", //NON-NLS
                "par_id", //NON-NLS
                "fs_type", //NON-NLS
                "block_size", //NON-NLS
                "block_count", //NON-NLS
                "root_inum", //NON-NLS
                "first_inum", //NON-NLS
                "last_inum" //NON-NLS
            };

            Object[][] rowValues = new Object[1][9];

            Content parent = null;
            try {
                parent = fs.getParent();
            } catch (Exception ex) {
                throw new RuntimeException(
                        NbBundle.getMessage(this.getClass(), "ExplorerNodeActionVisitor.exception.probGetParent.text",
                                FileSystem.class.getName(), fs), ex);
            }
            long id = -1;
            if (parent != null) {
                id = parent.getId();
            }

            Arrays.fill(rowValues, 0, 1, new Object[]{
                fs.getId(),
                fs.getImageOffset(),
                id,
                fs.getFsType(),
                fs.getBlock_size(),
                fs.getBlock_count(),
                fs.getRoot_inum(),
                fs.getFirst_inum(),
                fs.getLastInum()
            });

            JTable table = new JTable(new DefaultTableModel(rowValues, columnNames));

            FileSystemDetailsPanel fsdPanel = new FileSystemDetailsPanel();

            // add the command to close the window to the button on the Volume Detail Panel
            fsdPanel.setOKButtonActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent e) {
                    popUpWindow.dispose();
                }
            });

            try {
                fsdPanel.setFileSystemTypeValue(table.getValueAt(0, 3).toString());
                fsdPanel.setImageOffsetValue(table.getValueAt(0, 1).toString());
                fsdPanel.setVolumeIDValue(table.getValueAt(0, 2).toString());  //TODO: fix this to parent id, not vol id
                fsdPanel.setBlockSizeValue(table.getValueAt(0, 4).toString());
                fsdPanel.setBlockCountValue(table.getValueAt(0, 5).toString());
                fsdPanel.setRootInumValue(table.getValueAt(0, 6).toString());
                fsdPanel.setFirstInumValue(table.getValueAt(0, 7).toString());
                fsdPanel.setLastInumValue(table.getValueAt(0, 8).toString());

                popUpWindow.add(fsdPanel);
            } catch (Exception ex) {
                Logger.getLogger(ExplorerNodeActionVisitor.class.getName()).log(Level.WARNING, "Error setting up File System Details panel.", ex); //NON-NLS
            }

            popUpWindow.pack();
            popUpWindow.setResizable(false);
            popUpWindow.setVisible(true);

        }
    }
}