mstoeck3/autopsy-flatpak
1
0
Fork 0
mirror of https://github.com/overcuriousity/autopsy-flatpak.git synced 2025-07-17 18:17:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

created central repo on first load

Browse Source
This commit is contained in:
U-BASIS\gregd 2020-02-21 16:08:26 -05:00
parent 851a3308ad
commit 1b18e39dcf
2 changed files with 85 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Core/src/org/sleuthkit/autopsy/centralrepository/datamodel/CentralRepoDbManager.java
View File

@ -255,6 +255,9 @@ public class CentralRepoDbManager {
break;
case SQLITE:
// save the new SQLite settings
logger.info(String.format("Attempting to set up sqlite database at path: %s with filename: %s",
dbSettingsSqlite.getDbDirectory(), dbSettingsSqlite.getDbName()));
dbSettingsSqlite.saveSettings();
// Load those newly saved settings into the sqlite db manager instance
// in case we are still using the same instance.

118
Core/src/org/sleuthkit/autopsy/centralrepository/eventlisteners/Installer.java
View File

@ -18,8 +18,12 @@
*/
package org.sleuthkit.autopsy.centralrepository.eventlisteners;
import java.lang.reflect.InvocationTargetException;
import java.util.logging.Level;
import javax.swing.JOptionPane;
import javax.swing.SwingUtilities;
import org.openide.modules.ModuleInstall;
import org.openide.util.Exceptions;
import org.openide.util.NbBundle;
import org.openide.windows.WindowManager;
import org.sleuthkit.autopsy.casemodule.Case;
@ -55,55 +59,95 @@ public class Installer extends ModuleInstall {
@NbBundle.Messages({
"Installer.centralRepoUpgradeFailed.title=Central repository disabled",
"Installer.initialCreateSqlite.title=Create Sqlite Central Repository?",
"Installer.initialCreateSqlite.message=The central repository allows a user to find matching artifacts both across cases " +
"and across data sources in the same case. Having data in the central repository is useful for file discovery. Would you " +
"like to create the default Central Repository now? If you choose not to at this time, this setting can be changed in the " +
"options panel."
"Installer.initialCreateSqlite.title=Enable Central Repository?",
"Installer.initialCreateSqlite.message=The Central Repository is not enabled. Would you like to enable it? " +
"It will store information about all hashes and identifiers that you process. You can use this to ignore previously " +
"seen files and make connections between cases."
})
@Override
public void restored() {
Case.addPropertyChangeListener(pcl);
ieListener.installListeners();
// Perform the database upgrade and inform the user if it fails
try {
String initialized = ModuleSettings.getConfigSetting("CentralRepository", "initialized");
if (!Boolean.parseBoolean(initialized)) {
String dialogText = "<html><body><p style='max-width: 400px;'>" +
NbBundle.getMessage(this.getClass(), "Installer.initialCreateSqlite.message") +
"</p></body></html>";
String initialized = ModuleSettings.getConfigSetting("CentralRepository", "initialized");
boolean setupSqlite = !RuntimeProperties.runningWithGUI() ||
JOptionPane.YES_OPTION == JOptionPane.showConfirmDialog(WindowManager.getDefault().getMainWindow(),
dialogText,
NbBundle.getMessage(this.getClass(), "Installer.initialCreateSqlite.title"),
JOptionPane.YES_NO_OPTION);
if (setupSqlite) {
CentralRepoDbManager manager = new CentralRepoDbManager();
manager.setupDefaultSqliteSettings();
manager.saveNewCentralRepo();
}
ModuleSettings.setConfigSetting("CentralRepository", "initialized", "true");
}
else {
CentralRepoDbManager.upgradeDatabase();
}
} catch (CentralRepoException ex) {
// if central repository hasn't been previously initialized, initialize it
if (!Boolean.parseBoolean(initialized)) {
// if running with a GUI, prompt the user
if (RuntimeProperties.runningWithGUI()) {
WindowManager.getDefault().invokeWhenUIReady(() -> {
JOptionPane.showMessageDialog(null,
ex.getUserMessage(),
NbBundle.getMessage(this.getClass(),
"Installer.centralRepoUpgradeFailed.title"),
JOptionPane.ERROR_MESSAGE);
});
try {
SwingUtilities.invokeAndWait(() -> {
try {
String dialogText = "<html><body><p style='width: 400px;'>"
+ NbBundle.getMessage(this.getClass(), "Installer.initialCreateSqlite.message")
+ "</p></body></html>";
if (JOptionPane.YES_OPTION == JOptionPane.showConfirmDialog(WindowManager.getDefault().getMainWindow(),
dialogText,
NbBundle.getMessage(this.getClass(), "Installer.initialCreateSqlite.title"),
JOptionPane.YES_NO_OPTION)) {
setupDefaultSqlite();
}
} catch (CentralRepoException ex) {
LOGGER.log(Level.SEVERE, "There was an error while initializing the central repository database", ex);
reportUpgradeError(ex);
}
});
} catch (Exception ex) {
LOGGER.log(Level.SEVERE, "There was an error while running the swing utility invoke later while creating the central repository database", ex);
}
} // if no GUI, just initialize
else {
try {
setupDefaultSqlite();
} catch (CentralRepoException ex) {
LOGGER.log(Level.SEVERE, "There was an error while initializing the central repository database", ex);
reportUpgradeError(ex);
}
}
ModuleSettings.setConfigSetting("CentralRepository", "initialized", "true");
}
// now run regular module startup code
try {
CentralRepoDbManager.upgradeDatabase();
} catch (CentralRepoException ex) {
LOGGER.log(Level.SEVERE, "There was an error while upgrading the central repository database", ex);
if (RuntimeProperties.runningWithGUI()) {
reportUpgradeError(ex);
}
}
}
private void setupDefaultSqlite() throws CentralRepoException {
CentralRepoDbUtil.setUseCentralRepo(true);
CentralRepoDbManager manager = new CentralRepoDbManager();
manager.setupDefaultSqliteSettings();
manager.createDb();
manager.saveNewCentralRepo();
}
private void reportUpgradeError(CentralRepoException ex) {
try {
SwingUtilities.invokeAndWait(() -> {
JOptionPane.showMessageDialog(null,
ex.getUserMessage(),
NbBundle.getMessage(this.getClass(),
"Installer.centralRepoUpgradeFailed.title"),
JOptionPane.ERROR_MESSAGE);
});
} catch (Exception e) {
LOGGER.warning("There was an error while running the swing utility invoke later while displaying an error " +
"for creating the central repository database: "
+ e.getMessage() + System.lineSeparator() + e.getStackTrace());
}
}
@Override
public boolean closing() {
//platform about to close