Autopsy  4.0
Graphical digital forensics platform for The Sleuth Kit and other tools.
HashDbCreateDatabaseDialog.java
Go to the documentation of this file.
1 /*
2  * Autopsy Forensic Browser
3  *
4  * Copyright 2013 Basis Technology Corp.
5  * Contact: carrier <at> sleuthkit <dot> org
6  *
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  * http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  */
19 package org.sleuthkit.autopsy.modules.hashdatabase;
20 
21 import java.awt.Dimension;
22 import java.awt.Toolkit;
23 import java.io.File;
24 import java.io.IOException;
25 import java.util.logging.Level;
26 
27 import org.openide.util.NbBundle;
28 import org.sleuthkit.autopsy.coreutils.Logger;
29 import javax.swing.JFileChooser;
30 import javax.swing.JOptionPane;
31 import javax.swing.JFrame;
32 import org.apache.commons.io.FilenameUtils;
33 import org.sleuthkit.autopsy.coreutils.ModuleSettings;
34 import org.sleuthkit.datamodel.TskCoreException;
35 import org.sleuthkit.autopsy.modules.hashdatabase.HashDbManager.HashDb;
36 import org.sleuthkit.autopsy.modules.hashdatabase.HashDbManager.HashDb.KnownFilesType;
37 import org.sleuthkit.autopsy.modules.hashdatabase.HashDbManager.HashDbManagerException;
38 
44 final class HashDbCreateDatabaseDialog extends javax.swing.JDialog {
45 
46  private static final String DEFAULT_FILE_NAME = NbBundle
47  .getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.defaultFileName");
48  private JFileChooser fileChooser = null;
49  private HashDb newHashDb = null;
50  private final static String LAST_FILE_PATH_KEY = "HashDbCreate_Path";
51 
57  HashDbCreateDatabaseDialog() {
58  super(new JFrame(), NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.createHashDbMsg"), true);
59  initFileChooser();
60  initComponents();
61  display();
62  }
63 
69  HashDb getHashDatabase() {
70  return newHashDb;
71  }
72 
73  private void initFileChooser() {
74  fileChooser = new JFileChooser() {
75  @Override
76  public void approveSelection() {
77  File selectedFile = getSelectedFile();
78  if (!FilenameUtils.getExtension(selectedFile.getName()).equalsIgnoreCase(HashDbManager.getHashDatabaseFileExtension())) {
79  if (JOptionPane.showConfirmDialog(this,
80  NbBundle.getMessage(this.getClass(),
81  "HashDbCreateDatabaseDialog.hashDbMustHaveFileExtensionMsg",
82  HashDbManager.getHashDatabaseFileExtension()),
83  NbBundle.getMessage(this.getClass(),
84  "HashDbCreateDatabaseDialog.fileNameErr"),
85  JOptionPane.OK_CANCEL_OPTION) == JOptionPane.CANCEL_OPTION) {
86  cancelSelection();
87  }
88  return;
89  }
90  if (selectedFile.exists()) {
91  if (JOptionPane.showConfirmDialog(this,
92  NbBundle.getMessage(this.getClass(),
93  "HashDbCreateDatabaseDialog.fileNameAlreadyExistsMsg"),
94  NbBundle.getMessage(this.getClass(),
95  "HashDbCreateDatabaseDialog.fileExistsErr"),
96  JOptionPane.OK_CANCEL_OPTION) == JOptionPane.CANCEL_OPTION) {
97  cancelSelection();
98  }
99  return;
100  }
101  super.approveSelection();
102  }
103  };
104  fileChooser.setFileSelectionMode(JFileChooser.FILES_ONLY);
105  fileChooser.setDragEnabled(false);
106  fileChooser.setMultiSelectionEnabled(false);
107  }
108 
109  private void display() {
110  Dimension screenDimension = Toolkit.getDefaultToolkit().getScreenSize();
111  setLocation((screenDimension.width - getSize().width) / 2, (screenDimension.height - getSize().height) / 2);
112  setVisible(true);
113  }
114 
120  @SuppressWarnings("unchecked")
121  // <editor-fold defaultstate="collapsed" desc="Generated Code">//GEN-BEGIN:initComponents
122  private void initComponents() {
123 
124  buttonGroup1 = new javax.swing.ButtonGroup();
125  saveAsButton = new javax.swing.JButton();
126  cancelButton = new javax.swing.JButton();
127  knownRadioButton = new javax.swing.JRadioButton();
128  knownBadRadioButton = new javax.swing.JRadioButton();
129  jLabel1 = new javax.swing.JLabel();
130  hashSetNameTextField = new javax.swing.JTextField();
131  jLabel2 = new javax.swing.JLabel();
132  sendIngestMessagesCheckbox = new javax.swing.JCheckBox();
133  jLabel3 = new javax.swing.JLabel();
134  databasePathTextField = new javax.swing.JTextField();
135  okButton = new javax.swing.JButton();
136 
137  setDefaultCloseOperation(javax.swing.WindowConstants.DISPOSE_ON_CLOSE);
138 
139  org.openide.awt.Mnemonics.setLocalizedText(saveAsButton, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.saveAsButton.text")); // NOI18N
140  saveAsButton.addActionListener(new java.awt.event.ActionListener() {
141  public void actionPerformed(java.awt.event.ActionEvent evt) {
142  saveAsButtonActionPerformed(evt);
143  }
144  });
145 
146  org.openide.awt.Mnemonics.setLocalizedText(cancelButton, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.cancelButton.text")); // NOI18N
147  cancelButton.addActionListener(new java.awt.event.ActionListener() {
148  public void actionPerformed(java.awt.event.ActionEvent evt) {
149  cancelButtonActionPerformed(evt);
150  }
151  });
152 
153  buttonGroup1.add(knownRadioButton);
154  org.openide.awt.Mnemonics.setLocalizedText(knownRadioButton, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.knownRadioButton.text")); // NOI18N
155  knownRadioButton.addActionListener(new java.awt.event.ActionListener() {
156  public void actionPerformed(java.awt.event.ActionEvent evt) {
157  knownRadioButtonActionPerformed(evt);
158  }
159  });
160 
161  buttonGroup1.add(knownBadRadioButton);
162  knownBadRadioButton.setSelected(true);
163  org.openide.awt.Mnemonics.setLocalizedText(knownBadRadioButton, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.knownBadRadioButton.text")); // NOI18N
164  knownBadRadioButton.addActionListener(new java.awt.event.ActionListener() {
165  public void actionPerformed(java.awt.event.ActionEvent evt) {
166  knownBadRadioButtonActionPerformed(evt);
167  }
168  });
169 
170  org.openide.awt.Mnemonics.setLocalizedText(jLabel1, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.jLabel1.text")); // NOI18N
171 
172  hashSetNameTextField.setText(org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.hashSetNameTextField.text")); // NOI18N
173 
174  org.openide.awt.Mnemonics.setLocalizedText(jLabel2, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.jLabel2.text")); // NOI18N
175 
176  sendIngestMessagesCheckbox.setSelected(true);
177  org.openide.awt.Mnemonics.setLocalizedText(sendIngestMessagesCheckbox, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.sendIngestMessagesCheckbox.text")); // NOI18N
178  sendIngestMessagesCheckbox.addActionListener(new java.awt.event.ActionListener() {
179  public void actionPerformed(java.awt.event.ActionEvent evt) {
180  sendIngestMessagesCheckboxActionPerformed(evt);
181  }
182  });
183 
184  org.openide.awt.Mnemonics.setLocalizedText(jLabel3, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.jLabel3.text")); // NOI18N
185 
186  databasePathTextField.setEditable(false);
187  databasePathTextField.setText(org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.databasePathTextField.text")); // NOI18N
188 
189  org.openide.awt.Mnemonics.setLocalizedText(okButton, org.openide.util.NbBundle.getMessage(HashDbCreateDatabaseDialog.class, "HashDbCreateDatabaseDialog.okButton.text")); // NOI18N
190  okButton.addActionListener(new java.awt.event.ActionListener() {
191  public void actionPerformed(java.awt.event.ActionEvent evt) {
192  okButtonActionPerformed(evt);
193  }
194  });
195 
196  javax.swing.GroupLayout layout = new javax.swing.GroupLayout(getContentPane());
197  getContentPane().setLayout(layout);
198  layout.setHorizontalGroup(
199  layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
200  .addGroup(layout.createSequentialGroup()
201  .addContainerGap()
202  .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
203  .addComponent(sendIngestMessagesCheckbox)
204  .addGroup(javax.swing.GroupLayout.Alignment.TRAILING, layout.createSequentialGroup()
205  .addGap(0, 0, Short.MAX_VALUE)
206  .addComponent(okButton)
207  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
208  .addComponent(cancelButton))
209  .addComponent(jLabel2)
210  .addGroup(layout.createSequentialGroup()
211  .addGap(20, 20, 20)
212  .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
213  .addComponent(knownRadioButton)
214  .addComponent(knownBadRadioButton)))
215  .addGroup(layout.createSequentialGroup()
216  .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING, false)
217  .addGroup(layout.createSequentialGroup()
218  .addComponent(jLabel1)
219  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
220  .addComponent(databasePathTextField))
221  .addGroup(javax.swing.GroupLayout.Alignment.LEADING, layout.createSequentialGroup()
222  .addComponent(jLabel3)
223  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
224  .addComponent(hashSetNameTextField, javax.swing.GroupLayout.PREFERRED_SIZE, 272, javax.swing.GroupLayout.PREFERRED_SIZE)))
225  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
226  .addComponent(saveAsButton)))
227  .addContainerGap())
228  );
229 
230  layout.linkSize(javax.swing.SwingConstants.HORIZONTAL, new java.awt.Component[] {cancelButton, okButton});
231 
232  layout.setVerticalGroup(
233  layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
234  .addGroup(javax.swing.GroupLayout.Alignment.TRAILING, layout.createSequentialGroup()
235  .addGap(2, 2, 2)
236  .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
237  .addComponent(jLabel3)
238  .addComponent(hashSetNameTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
239  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
240  .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
241  .addComponent(databasePathTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
242  .addComponent(saveAsButton)
243  .addComponent(jLabel1))
244  .addGap(7, 7, 7)
245  .addComponent(jLabel2)
246  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
247  .addComponent(knownRadioButton)
248  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
249  .addComponent(knownBadRadioButton)
250  .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
251  .addComponent(sendIngestMessagesCheckbox)
252  .addGap(26, 26, 26)
253  .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
254  .addComponent(cancelButton)
255  .addComponent(okButton))
256  .addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
257  );
258 
259  pack();
260  }// </editor-fold>//GEN-END:initComponents
261 
262  private void knownRadioButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_knownRadioButtonActionPerformed
263  sendIngestMessagesCheckbox.setSelected(false);
264  sendIngestMessagesCheckbox.setEnabled(false);
265  }//GEN-LAST:event_knownRadioButtonActionPerformed
266 
267  private void knownBadRadioButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_knownBadRadioButtonActionPerformed
268  sendIngestMessagesCheckbox.setSelected(true);
269  sendIngestMessagesCheckbox.setEnabled(true);
270  }//GEN-LAST:event_knownBadRadioButtonActionPerformed
271 
272  private void cancelButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_cancelButtonActionPerformed
273  dispose();
274  }//GEN-LAST:event_cancelButtonActionPerformed
275 
276  private void saveAsButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_saveAsButtonActionPerformed
277  try {
278  String lastBaseDirectory = "";
279  if (ModuleSettings.settingExists(ModuleSettings.MAIN_SETTINGS, LAST_FILE_PATH_KEY)) {
280  lastBaseDirectory = ModuleSettings.getConfigSetting(ModuleSettings.MAIN_SETTINGS, LAST_FILE_PATH_KEY);
281  }
282  StringBuilder path = new StringBuilder();
283  path.append(lastBaseDirectory);
284  if (!hashSetNameTextField.getText().isEmpty()) {
285  path.append(File.separator).append(hashSetNameTextField.getText());
286  } else {
287  path.append(File.separator).append(DEFAULT_FILE_NAME);
288  }
289  path.append(".").append(HashDbManager.getHashDatabaseFileExtension());
290  fileChooser.setSelectedFile(new File(path.toString()));
291  if (fileChooser.showSaveDialog(this) == JFileChooser.APPROVE_OPTION) {
292  File databaseFile = fileChooser.getSelectedFile();
293  databasePathTextField.setText(databaseFile.getCanonicalPath());
294  ModuleSettings.setConfigSetting(ModuleSettings.MAIN_SETTINGS, LAST_FILE_PATH_KEY, databaseFile.getParent());
295  }
296  } catch (IOException ex) {
297  Logger.getLogger(HashDbCreateDatabaseDialog.class.getName()).log(Level.WARNING, "Couldn't get selected file path.", ex); //NON-NLS
298  }
299  }//GEN-LAST:event_saveAsButtonActionPerformed
300 
301  private void okButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_okButtonActionPerformed
302  // Note that the error handlers in this method call return without disposing of the
303  // dialog to allow the user to try again, if desired.
304 
305  if (hashSetNameTextField.getText().isEmpty()) {
306  JOptionPane.showMessageDialog(this,
307  NbBundle.getMessage(this.getClass(),
308  "HashDbCreateDatabaseDialog.mustEnterHashSetNameMsg"),
309  NbBundle.getMessage(this.getClass(),
310  "HashDbCreateDatabaseDialog.createHashDbErr"),
311  JOptionPane.ERROR_MESSAGE);
312  return;
313  }
314 
315  if (databasePathTextField.getText().isEmpty()) {
316  JOptionPane.showMessageDialog(this,
317  NbBundle.getMessage(this.getClass(),
318  "HashDbCreateDatabaseDialog.mustEnterHashDbPathMsg"),
319  NbBundle.getMessage(this.getClass(),
320  "HashDbCreateDatabaseDialog.createHashDbErr"),
321  JOptionPane.ERROR_MESSAGE);
322  return;
323  }
324 
325  KnownFilesType type;
326  if (knownRadioButton.isSelected()) {
327  type = KnownFilesType.KNOWN;
328  } else {
329  type = KnownFilesType.KNOWN_BAD;
330  }
331 
332  String errorMessage = NbBundle
333  .getMessage(this.getClass(), "HashDbCreateDatabaseDialog.errMsg.hashDbCreationErr");
334  try {
335  newHashDb = HashDbManager.getInstance().addNewHashDatabaseInternal(hashSetNameTextField.getText(), fileChooser.getSelectedFile().getCanonicalPath(), true, sendIngestMessagesCheckbox.isSelected(), type);
336  } catch (IOException ex) {
337  Logger.getLogger(HashDbCreateDatabaseDialog.class.getName()).log(Level.WARNING, errorMessage, ex);
338  JOptionPane.showMessageDialog(this,
339  NbBundle.getMessage(this.getClass(),
340  "HashDbCreateDatabaseDialog.cannotCreateFileAtLocMsg"),
341  NbBundle.getMessage(this.getClass(),
342  "HashDbCreateDatabaseDialog.createHashDbErr"),
343  JOptionPane.ERROR_MESSAGE);
344  return;
345  } catch (HashDbManagerException ex) {
346  Logger.getLogger(HashDbCreateDatabaseDialog.class.getName()).log(Level.WARNING, errorMessage, ex);
347  JOptionPane.showMessageDialog(this,
348  ex.getMessage(),
349  NbBundle.getMessage(this.getClass(),
350  "HashDbCreateDatabaseDialog.createHashDbErr"),
351  JOptionPane.ERROR_MESSAGE);
352  return;
353  } catch (TskCoreException ex) {
354  Logger.getLogger(HashDbCreateDatabaseDialog.class.getName()).log(Level.SEVERE, errorMessage, ex);
355  JOptionPane.showMessageDialog(this,
356  NbBundle.getMessage(this.getClass(),
357  "HashDbCreateDatabaseDialog.failedToCreateHashDbMsg"),
358  NbBundle.getMessage(this.getClass(),
359  "HashDbCreateDatabaseDialog.createHashDbErr"),
360  JOptionPane.ERROR_MESSAGE);
361  return;
362  }
363 
364  dispose();
365  }//GEN-LAST:event_okButtonActionPerformed
366 
367  private void sendIngestMessagesCheckboxActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_sendIngestMessagesCheckboxActionPerformed
368  // TODO add your handling code here:
369  }//GEN-LAST:event_sendIngestMessagesCheckboxActionPerformed
370 
371  // Variables declaration - do not modify//GEN-BEGIN:variables
372  private javax.swing.ButtonGroup buttonGroup1;
373  private javax.swing.JButton cancelButton;
374  private javax.swing.JTextField databasePathTextField;
375  private javax.swing.JTextField hashSetNameTextField;
376  private javax.swing.JLabel jLabel1;
377  private javax.swing.JLabel jLabel2;
378  private javax.swing.JLabel jLabel3;
379  private javax.swing.JRadioButton knownBadRadioButton;
380  private javax.swing.JRadioButton knownRadioButton;
381  private javax.swing.JButton okButton;
382  private javax.swing.JButton saveAsButton;
383  private javax.swing.JCheckBox sendIngestMessagesCheckbox;
384  // End of variables declaration//GEN-END:variables
385 }

Copyright © 2012-2015 Basis Technology. Generated on: Wed Apr 6 2016
This work is licensed under a Creative Commons Attribution-Share Alike 3.0 United States License.