mirror of
https://github.com/overcuriousity/autopsy-flatpak.git
synced 2025-07-12 16:06:15 +00:00
374 lines
20 KiB
Java
374 lines
20 KiB
Java
/*
|
|
* Autopsy Forensic Browser
|
|
*
|
|
* Copyright 2011 - 2013 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.hashdatabase;
|
|
|
|
import java.awt.Dimension;
|
|
import java.awt.Toolkit;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.logging.Level;
|
|
|
|
import org.openide.util.NbBundle;
|
|
import org.sleuthkit.autopsy.coreutils.Logger;
|
|
import javax.swing.JFileChooser;
|
|
import javax.swing.JOptionPane;
|
|
import javax.swing.filechooser.FileNameExtensionFilter;
|
|
import javax.swing.JFrame;
|
|
import org.sleuthkit.datamodel.TskCoreException;
|
|
import org.apache.commons.io.FilenameUtils;
|
|
import org.sleuthkit.autopsy.hashdatabase.HashDbManager.HashDb.KnownFilesType;
|
|
import org.sleuthkit.autopsy.hashdatabase.HashDbManager.HashDb;
|
|
import org.sleuthkit.autopsy.hashdatabase.HashDbManager.HashDbManagerException;
|
|
|
|
/**
|
|
* Instances of this class allow a user to select an existing hash database and
|
|
* add it to the set of hash databases used to classify files as unknown, known,
|
|
* or known bad.
|
|
*/
|
|
final class HashDbImportDatabaseDialog extends javax.swing.JDialog {
|
|
|
|
private JFileChooser fileChooser = new JFileChooser();
|
|
private String selectedFilePath = "";
|
|
private HashDb selectedHashDb = null;
|
|
|
|
/**
|
|
* Displays a dialog that allows a user to select an existing hash database
|
|
* and add it to the set of hash databases used to classify files as unknown,
|
|
* known, or known bad.
|
|
*/
|
|
HashDbImportDatabaseDialog() {
|
|
super(new JFrame(),
|
|
NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.importHashDbMsg"),
|
|
true);
|
|
initFileChooser();
|
|
initComponents();
|
|
display();
|
|
}
|
|
|
|
/**
|
|
* Get the hash database imported by the user, if any.
|
|
* @return A HashDb object or null.
|
|
*/
|
|
HashDb getHashDatabase() {
|
|
return selectedHashDb;
|
|
}
|
|
|
|
private void initFileChooser() {
|
|
fileChooser.setDragEnabled(false);
|
|
fileChooser.setFileSelectionMode(JFileChooser.FILES_ONLY);
|
|
String[] EXTENSION = new String[] { "txt", "kdb", "idx", "hash", "Hash", "hsh"}; //NON-NLS
|
|
FileNameExtensionFilter filter = new FileNameExtensionFilter(
|
|
NbBundle.getMessage(this.getClass(), "HashDbImportDatabaseDialog.fileNameExtFilter.text"), EXTENSION);
|
|
fileChooser.setFileFilter(filter);
|
|
fileChooser.setMultiSelectionEnabled(false);
|
|
}
|
|
|
|
private void display() {
|
|
Dimension screenDimension = Toolkit.getDefaultToolkit().getScreenSize();
|
|
setLocation((screenDimension.width - getSize().width) / 2, (screenDimension.height - getSize().height) / 2);
|
|
setVisible(true);
|
|
}
|
|
|
|
private static String shortenPath(String path) {
|
|
String shortenedPath = path;
|
|
if (shortenedPath.length() > 50){
|
|
shortenedPath = shortenedPath.substring(0, 10 + shortenedPath.substring(10).indexOf(File.separator) + 1) + "..." + shortenedPath.substring((shortenedPath.length() - 20) + shortenedPath.substring(shortenedPath.length() - 20).indexOf(File.separator));
|
|
}
|
|
return shortenedPath;
|
|
}
|
|
|
|
/**
|
|
* This method is called from within the constructor to initialize the form.
|
|
* WARNING: Do NOT modify this code. The content of this method is always
|
|
* regenerated by the Form Editor.
|
|
*/
|
|
@SuppressWarnings("unchecked")
|
|
// <editor-fold defaultstate="collapsed" desc="Generated Code">//GEN-BEGIN:initComponents
|
|
private void initComponents() {
|
|
|
|
buttonGroup1 = new javax.swing.ButtonGroup();
|
|
okButton = new javax.swing.JButton();
|
|
cancelButton = new javax.swing.JButton();
|
|
databasePathTextField = new javax.swing.JTextField();
|
|
openButton = new javax.swing.JButton();
|
|
knownRadioButton = new javax.swing.JRadioButton();
|
|
knownBadRadioButton = new javax.swing.JRadioButton();
|
|
jLabel1 = new javax.swing.JLabel();
|
|
hashSetNameTextField = new javax.swing.JTextField();
|
|
jLabel2 = new javax.swing.JLabel();
|
|
sendIngestMessagesCheckbox = new javax.swing.JCheckBox();
|
|
jLabel3 = new javax.swing.JLabel();
|
|
|
|
setDefaultCloseOperation(javax.swing.WindowConstants.DISPOSE_ON_CLOSE);
|
|
|
|
org.openide.awt.Mnemonics.setLocalizedText(okButton, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.okButton.text")); // NOI18N
|
|
okButton.addActionListener(new java.awt.event.ActionListener() {
|
|
public void actionPerformed(java.awt.event.ActionEvent evt) {
|
|
okButtonActionPerformed(evt);
|
|
}
|
|
});
|
|
|
|
org.openide.awt.Mnemonics.setLocalizedText(cancelButton, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.cancelButton.text")); // NOI18N
|
|
cancelButton.addActionListener(new java.awt.event.ActionListener() {
|
|
public void actionPerformed(java.awt.event.ActionEvent evt) {
|
|
cancelButtonActionPerformed(evt);
|
|
}
|
|
});
|
|
|
|
databasePathTextField.setEditable(false);
|
|
databasePathTextField.setText(org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.databasePathTextField.text")); // NOI18N
|
|
|
|
org.openide.awt.Mnemonics.setLocalizedText(openButton, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.openButton.text")); // NOI18N
|
|
openButton.addActionListener(new java.awt.event.ActionListener() {
|
|
public void actionPerformed(java.awt.event.ActionEvent evt) {
|
|
openButtonActionPerformed(evt);
|
|
}
|
|
});
|
|
|
|
buttonGroup1.add(knownRadioButton);
|
|
org.openide.awt.Mnemonics.setLocalizedText(knownRadioButton, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.knownRadioButton.text")); // NOI18N
|
|
knownRadioButton.addActionListener(new java.awt.event.ActionListener() {
|
|
public void actionPerformed(java.awt.event.ActionEvent evt) {
|
|
knownRadioButtonActionPerformed(evt);
|
|
}
|
|
});
|
|
|
|
buttonGroup1.add(knownBadRadioButton);
|
|
knownBadRadioButton.setSelected(true);
|
|
org.openide.awt.Mnemonics.setLocalizedText(knownBadRadioButton, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.knownBadRadioButton.text")); // NOI18N
|
|
knownBadRadioButton.addActionListener(new java.awt.event.ActionListener() {
|
|
public void actionPerformed(java.awt.event.ActionEvent evt) {
|
|
knownBadRadioButtonActionPerformed(evt);
|
|
}
|
|
});
|
|
|
|
org.openide.awt.Mnemonics.setLocalizedText(jLabel1, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.jLabel1.text")); // NOI18N
|
|
|
|
hashSetNameTextField.setText(org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.hashSetNameTextField.text")); // NOI18N
|
|
|
|
org.openide.awt.Mnemonics.setLocalizedText(jLabel2, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.jLabel2.text")); // NOI18N
|
|
|
|
sendIngestMessagesCheckbox.setSelected(true);
|
|
org.openide.awt.Mnemonics.setLocalizedText(sendIngestMessagesCheckbox, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.sendIngestMessagesCheckbox.text")); // NOI18N
|
|
sendIngestMessagesCheckbox.addActionListener(new java.awt.event.ActionListener() {
|
|
public void actionPerformed(java.awt.event.ActionEvent evt) {
|
|
sendIngestMessagesCheckboxActionPerformed(evt);
|
|
}
|
|
});
|
|
|
|
org.openide.awt.Mnemonics.setLocalizedText(jLabel3, org.openide.util.NbBundle.getMessage(HashDbImportDatabaseDialog.class, "HashDbImportDatabaseDialog.jLabel3.text")); // NOI18N
|
|
|
|
javax.swing.GroupLayout layout = new javax.swing.GroupLayout(getContentPane());
|
|
getContentPane().setLayout(layout);
|
|
layout.setHorizontalGroup(
|
|
layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
|
|
.addGroup(layout.createSequentialGroup()
|
|
.addContainerGap()
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
|
|
.addGroup(layout.createSequentialGroup()
|
|
.addGap(0, 0, Short.MAX_VALUE)
|
|
.addComponent(okButton)
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
|
|
.addComponent(cancelButton))
|
|
.addGroup(layout.createSequentialGroup()
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING)
|
|
.addGroup(javax.swing.GroupLayout.Alignment.LEADING, layout.createSequentialGroup()
|
|
.addComponent(jLabel1)
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
|
|
.addComponent(hashSetNameTextField, javax.swing.GroupLayout.DEFAULT_SIZE, 303, Short.MAX_VALUE))
|
|
.addGroup(javax.swing.GroupLayout.Alignment.LEADING, layout.createSequentialGroup()
|
|
.addComponent(jLabel3)
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
|
|
.addComponent(databasePathTextField)))
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
|
|
.addComponent(openButton))
|
|
.addGroup(layout.createSequentialGroup()
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
|
|
.addComponent(jLabel2)
|
|
.addGroup(layout.createSequentialGroup()
|
|
.addGap(19, 19, 19)
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
|
|
.addComponent(knownRadioButton)
|
|
.addComponent(knownBadRadioButton)))
|
|
.addComponent(sendIngestMessagesCheckbox))
|
|
.addGap(0, 0, Short.MAX_VALUE)))
|
|
.addContainerGap())
|
|
);
|
|
|
|
layout.linkSize(javax.swing.SwingConstants.HORIZONTAL, new java.awt.Component[] {cancelButton, okButton});
|
|
|
|
layout.setVerticalGroup(
|
|
layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
|
|
.addGroup(javax.swing.GroupLayout.Alignment.TRAILING, layout.createSequentialGroup()
|
|
.addContainerGap()
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
|
|
.addComponent(openButton)
|
|
.addComponent(databasePathTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
|
|
.addComponent(jLabel3))
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
|
|
.addComponent(jLabel1)
|
|
.addComponent(hashSetNameTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
|
|
.addGroup(layout.createSequentialGroup()
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED, 119, Short.MAX_VALUE)
|
|
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
|
|
.addComponent(okButton)
|
|
.addComponent(cancelButton))
|
|
.addContainerGap())
|
|
.addGroup(layout.createSequentialGroup()
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
|
|
.addComponent(jLabel2)
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
|
|
.addComponent(knownRadioButton)
|
|
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
|
|
.addComponent(knownBadRadioButton)
|
|
.addGap(18, 18, 18)
|
|
.addComponent(sendIngestMessagesCheckbox)
|
|
.addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))))
|
|
);
|
|
|
|
pack();
|
|
}// </editor-fold>//GEN-END:initComponents
|
|
|
|
private void openButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_openButtonActionPerformed
|
|
if (fileChooser.showOpenDialog(this) == JFileChooser.APPROVE_OPTION) {
|
|
File databaseFile = fileChooser.getSelectedFile();
|
|
try {
|
|
selectedFilePath = databaseFile.getCanonicalPath();
|
|
databasePathTextField.setText(shortenPath(selectedFilePath));
|
|
hashSetNameTextField.setText(FilenameUtils.removeExtension(databaseFile.getName()));
|
|
if (hashSetNameTextField.getText().toLowerCase().contains("nsrl")) { //NON-NLS
|
|
knownRadioButton.setSelected(true);
|
|
knownRadioButtonActionPerformed(null);
|
|
}
|
|
}
|
|
catch (IOException ex) {
|
|
Logger.getLogger(HashDbImportDatabaseDialog.class.getName()).log(Level.SEVERE, "Failed to get path of selected database", ex); //NON-NLS
|
|
JOptionPane.showMessageDialog(this,
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.failedToGetDbPathMsg"));
|
|
}
|
|
}
|
|
}//GEN-LAST:event_openButtonActionPerformed
|
|
|
|
private void knownRadioButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_knownRadioButtonActionPerformed
|
|
sendIngestMessagesCheckbox.setSelected(false);
|
|
sendIngestMessagesCheckbox.setEnabled(false);
|
|
}//GEN-LAST:event_knownRadioButtonActionPerformed
|
|
|
|
private void knownBadRadioButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_knownBadRadioButtonActionPerformed
|
|
sendIngestMessagesCheckbox.setSelected(true);
|
|
sendIngestMessagesCheckbox.setEnabled(true);
|
|
}//GEN-LAST:event_knownBadRadioButtonActionPerformed
|
|
|
|
private void cancelButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_cancelButtonActionPerformed
|
|
this.dispose();
|
|
}//GEN-LAST:event_cancelButtonActionPerformed
|
|
|
|
private void okButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_okButtonActionPerformed
|
|
// Note that the error handlers in this method call return without disposing of the
|
|
// dialog to allow the user to try again, if desired.
|
|
|
|
if(hashSetNameTextField.getText().isEmpty()) {
|
|
JOptionPane.showMessageDialog(this,
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbCreateDatabaseDialog.mustEnterHashSetNameMsg"),
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.importHashDbErr"),
|
|
JOptionPane.ERROR_MESSAGE);
|
|
return;
|
|
}
|
|
|
|
if(selectedFilePath.isEmpty()) {
|
|
JOptionPane.showMessageDialog(this,
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.mustSelectHashDbFilePathMsg"),
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.importHashDbErr"),
|
|
JOptionPane.ERROR_MESSAGE);
|
|
return;
|
|
}
|
|
File file = new File(selectedFilePath);
|
|
if (!file.exists()) {
|
|
JOptionPane.showMessageDialog(this,
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.hashDbDoesNotExistMsg"),
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.importHashDbErr"),
|
|
JOptionPane.ERROR_MESSAGE);
|
|
return;
|
|
}
|
|
|
|
KnownFilesType type;
|
|
if (knownRadioButton.isSelected()) {
|
|
type = KnownFilesType.KNOWN;
|
|
}
|
|
else {
|
|
type = KnownFilesType.KNOWN_BAD;
|
|
}
|
|
|
|
String errorMessage = NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.errorMessage.failedToOpenHashDbMsg",
|
|
selectedFilePath);
|
|
try {
|
|
selectedHashDb = HashDbManager.getInstance().addExistingHashDatabaseInternal(hashSetNameTextField.getText(), selectedFilePath, true, sendIngestMessagesCheckbox.isSelected(), type);
|
|
}
|
|
catch (HashDbManagerException ex) {
|
|
Logger.getLogger(HashDbImportDatabaseDialog.class.getName()).log(Level.WARNING, errorMessage, ex);
|
|
JOptionPane.showMessageDialog(this,
|
|
ex.getMessage(),
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.importHashDbErr"),
|
|
JOptionPane.ERROR_MESSAGE);
|
|
return;
|
|
}
|
|
catch (TskCoreException ex) {
|
|
Logger.getLogger(HashDbCreateDatabaseDialog.class.getName()).log(Level.SEVERE, errorMessage, ex);
|
|
JOptionPane.showMessageDialog(this,
|
|
errorMessage,
|
|
NbBundle.getMessage(this.getClass(),
|
|
"HashDbImportDatabaseDialog.importHashDbErr"),
|
|
JOptionPane.ERROR_MESSAGE);
|
|
return;
|
|
}
|
|
|
|
dispose();
|
|
}//GEN-LAST:event_okButtonActionPerformed
|
|
|
|
private void sendIngestMessagesCheckboxActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_sendIngestMessagesCheckboxActionPerformed
|
|
// TODO add your handling code here:
|
|
}//GEN-LAST:event_sendIngestMessagesCheckboxActionPerformed
|
|
|
|
// Variables declaration - do not modify//GEN-BEGIN:variables
|
|
private javax.swing.ButtonGroup buttonGroup1;
|
|
private javax.swing.JButton cancelButton;
|
|
private javax.swing.JTextField databasePathTextField;
|
|
private javax.swing.JTextField hashSetNameTextField;
|
|
private javax.swing.JLabel jLabel1;
|
|
private javax.swing.JLabel jLabel2;
|
|
private javax.swing.JLabel jLabel3;
|
|
private javax.swing.JRadioButton knownBadRadioButton;
|
|
private javax.swing.JRadioButton knownRadioButton;
|
|
private javax.swing.JButton okButton;
|
|
private javax.swing.JButton openButton;
|
|
private javax.swing.JCheckBox sendIngestMessagesCheckbox;
|
|
// End of variables declaration//GEN-END:variables
|
|
}
|