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

Removed prematuer merge of some new tags api files

Browse Source
This commit is contained in:
Richard Cordovano 2013-10-22 17:21:44 -04:00
parent 572923a733
commit 8b487ca56d
4 changed files with 151 additions and 444 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Core/src/org/sleuthkit/autopsy/casemodule/services/Services.java
View File

@ -41,26 +41,18 @@ public class Services implements Closeable {
// services
private FileManager fileManager;
private TagsManager tagsManager;
public Services(SleuthkitCase tskCase) {
this.tskCase = tskCase;
//create and initialize FileManager as early as possibly in the new/opened Case
fileManager = new FileManager(tskCase);
services.add(fileManager);
tagsManager = new TagsManager(tskCase);
services.add(tagsManager);
}
public FileManager getFileManager() {
return fileManager;
}
public TagsManager getTagsManager() {
return tagsManager;
}
@Override
public void close() throws IOException {
// close all services

247
Core/src/org/sleuthkit/autopsy/casemodule/services/TagsManager.java
View File

@ -1,247 +0,0 @@
/*
* Autopsy Forensic Browser
*
* Copyright 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.casemodule.services;
import java.io.Closeable;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.sleuthkit.autopsy.coreutils.ModuleSettings;
import org.sleuthkit.datamodel.BlackboardArtifact;
import org.sleuthkit.datamodel.BlackboardArtifactTag;
import org.sleuthkit.datamodel.Content;
import org.sleuthkit.datamodel.ContentTag;
import org.sleuthkit.datamodel.SleuthkitCase;
import org.sleuthkit.datamodel.TagType;
import org.sleuthkit.datamodel.TskCoreException;
/**
* A singleton instance of this class functions as an Autopsy service that
* manages the creation, updating, and deletion of tags applied to Content and
* BlackboardArtifacts objects by users.
*/
public class TagsManager implements Closeable {
private static final String TAGS_SETTINGS_FILE_NAME = "tags";
private static final String TAG_TYPES_SETTING_KEY = "tagTypes";
private final SleuthkitCase tskCase;
private final HashMap<String, TagType> tagTypes = new HashMap<>();
TagsManager(SleuthkitCase tskCase) {
this.tskCase = tskCase;
loadTagTypesFromTagSettings();
}
private void loadTagTypesFromTagSettings() {
// Get any tag types already added to the current case.
try {
List<TagType> currentTagTypes = tskCase.getTagTypes();
for (TagType tagType : currentTagTypes) {
tagTypes.put(tagType.getDisplayName(), tagType);
}
}
catch (TskCoreException ex) {
Logger.getLogger(TagsManager.class.getName()).log(Level.SEVERE, "Failed to get tag types from the current case", ex);
}
// Read the saved tag types, if any, from the tags settings file and
// add them to the current case if they haven't already been added, e.g,
// when the case was last opened.
String setting = ModuleSettings.getConfigSetting(TAGS_SETTINGS_FILE_NAME, TAG_TYPES_SETTING_KEY);
if (null != setting && !setting.isEmpty()) {
// Read the tag types setting and break in into tag type tuples.
List<String> tagTypeTuples = Arrays.asList(setting.split(";"));
// Parse each tuple and add the tag types to the current case, one
// at a time to gracefully discard any duplicates or corrupt tuples.
for (String tagTypeTuple : tagTypeTuples) {
String[] tagTypeAttributes = tagTypeTuple.split(",");
if (!tagTypes.containsKey(tagTypeAttributes[0])) {
TagType tagType = new TagType(tagTypeAttributes[0], tagTypeAttributes[1], TagType.HTML_COLOR.getColorByName(tagTypeAttributes[2]));
try {
tskCase.addTagType(tagType);
tagTypes.put(tagType.getDisplayName(),tagType);
}
catch(TskCoreException ex) {
Logger.getLogger(TagsManager.class.getName()).log(Level.WARNING, "Failed to add saved " + tagType.getDisplayName() + " tag type to the current case", ex);
}
}
}
saveTagTypesToTagsSettings();
}
}
private void saveTagTypesToTagsSettings() {
if (!tagTypes.isEmpty()) {
StringBuilder setting = new StringBuilder();
for (TagType tagType : tagTypes.values()) {
if (setting.length() != 0) {
setting.append(";");
}
setting.append(tagType.getDisplayName()).append(",");
setting.append(tagType.getDescription()).append(",");
setting.append(tagType.getColor().name());
}
ModuleSettings.setConfigSetting(TAGS_SETTINGS_FILE_NAME, TAG_TYPES_SETTING_KEY, setting.toString());
}
}
/**
* Gets a list of all tag types currently available for tagging content or
* blackboard artifacts.
* @return A list, possibly empty, of TagType data transfer objects (DTOs).
* @throws TskCoreException
*/
public List<TagType> getTagTypes() throws TskCoreException {
return tskCase.getTagTypes();
}
/**
* Adds a new tag type to the current case and to the tags settings file.
* @param displayName The display name for the new tag type.
* @return A TagType object representing the new type on success, null on failure.
* @throws TskCoreException
*/
public TagType addTagType(String displayName) throws TagTypeAlreadyExistsException, TskCoreException {
return addTagType(displayName, "", TagType.HTML_COLOR.NONE);
}
/**
* Adds a new tag type to the current case and to the tags settings file.
* @param displayName The display name for the new tag type.
* @param description The description for the new tag type.
* @return A TagType object representing the new type on success, null on failure.
* @throws TskCoreException
*/
public TagType addTagType(String displayName, String description) throws TagTypeAlreadyExistsException, TskCoreException {
return addTagType(displayName, description, TagType.HTML_COLOR.NONE);
}
/**
* Adds a new tag type to the current case and to the tags settings file.
* @param displayName The display name for the new tag type.
* @param description The description for the new tag type.
* @param color The HTML color to associate with the new tag type.
* @return A TagType object representing the new type.
* @throws TskCoreException
*/
public synchronized TagType addTagType(String displayName, String description, TagType.HTML_COLOR color) throws TagTypeAlreadyExistsException, TskCoreException {
if (tagTypes.containsKey(displayName)) {
throw new TagTypeAlreadyExistsException();
}
TagType newTagType = new TagType(displayName, description, color);
tskCase.addTagType(newTagType);
tagTypes.put(newTagType.getDisplayName(), newTagType);
saveTagTypesToTagsSettings();
return newTagType;
}
public class TagTypeAlreadyExistsException extends Exception {
}
/**
* Tags a Content object.
* @param content The Content to tag.
* @param tagType The type of tag to add.
* @throws TskCoreException
*/
public void addContentTag(Content content, TagType tagType) throws TskCoreException {
addContentTag(content, tagType, "", 0, content.getSize() - 1);
}
/**
* Tags a Content object.
* @param content The Content to tag.
* @param tagType The type of tag to add.
* @param comment A comment to store with the tag.
* @throws TskCoreException
*/
public void addContentTag(Content content, TagType tagType, String comment) throws TskCoreException {
addContentTag(content, tagType, comment, 0, content.getSize() - 1);
}
/**
* Tags a Content object or a portion of a content object.
* @param content The Content to tag.
* @param tagType The type of tag to add.
* @param comment A comment to store with the tag.
* @param beginByteOffset Designates the beginning of a tagged extent.
* @param endByteOffset Designates the end of a tagged extent.
* @throws TskCoreException
*/
public void addContentTag(Content content, TagType tagType, String comment, long beginByteOffset, long endByteOffset) throws IllegalArgumentException, TskCoreException {
if (beginByteOffset < 0) {
throw new IllegalArgumentException("Content extent incorrect: beginByteOffset < 0");
}
if (endByteOffset <= beginByteOffset) {
throw new IllegalArgumentException("Content extent incorrect: endByteOffset <= beginByteOffset");
}
if (endByteOffset > content.getSize() - 1) {
throw new IllegalArgumentException("Content extent incorrect: endByteOffset exceeds content size");
}
tskCase.addContentTag(new ContentTag(content, tagType, comment, beginByteOffset, endByteOffset));
}
/**
* Deletes a content tag.
* @param tag The tag to delete.
* @throws TskCoreException
*/
public void deleteContentTag(ContentTag tag) throws TskCoreException {
tskCase.deleteContentTag(tag);
}
/**
* Tags a BlackboardArtifact object.
* @param artifact The BlackboardArtifact to tag.
* @param tagType The type of tag to add.
* @throws TskCoreException
*/
public void addBlackboardArtifactTag(BlackboardArtifact artifact, TagType tagType) throws TskCoreException {
addBlackboardArtifactTag(artifact, tagType, "");
}
/**
* Tags a BlackboardArtifact object.
* @param artifact The BlackboardArtifact to tag.
* @param tagType The type of tag to add.
* @param comment A comment to store with the tag.
* @throws TskCoreException
*/
public void addBlackboardArtifactTag(BlackboardArtifact artifact, TagType tagType, String comment) throws TskCoreException {
tskCase.addBlackboardArtifactTag(new BlackboardArtifactTag(artifact, tagType, comment));
}
void deleteBlackboardArtifactTag(BlackboardArtifactTag tag) throws TskCoreException {
tskCase.deleteBlackboardArtifactTag(tag);
}
@Override
public void close() throws IOException {
saveTagTypesToTagsSettings();
}
}

19
Core/src/org/sleuthkit/autopsy/directorytree/TagAbstractFileAction.java
View File

@ -27,11 +27,9 @@ import javax.swing.JOptionPane;
import org.openide.util.Utilities;
import org.openide.util.actions.Presenter;
import org.sleuthkit.autopsy.casemodule.Case;
import org.sleuthkit.autopsy.casemodule.services.TagsManager;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.autopsy.datamodel.Tags;
import org.sleuthkit.datamodel.AbstractFile;
import org.sleuthkit.datamodel.TagType;
import org.sleuthkit.datamodel.TskCoreException;
public class TagAbstractFileAction extends AbstractAction implements Presenter.Popup {
@ -68,26 +66,9 @@ public class TagAbstractFileAction extends AbstractAction implements Presenter.P
@Override
protected void applyTag(String tagDisplayName, String comment) {
try {
TagsManager tagsManager = Case.getCurrentCase().getServices().getTagsManager();
TagType tagType = tagsManager.addTagType(tagDisplayName);
Collection<? extends AbstractFile> selectedFiles = Utilities.actionsGlobalContext().lookupAll(AbstractFile.class);
for (AbstractFile file : selectedFiles) {
Tags.createTag(file, tagDisplayName, comment);
// try {
// tagsManager.addContentTag(file, tagType);
// }
// catch (TskCoreException ex) {
// Logger.getLogger(TagAbstractFileMenu.class.getName()).log(Level.SEVERE, "Error tagging content", ex);
// }
}
}
catch (TagsManager.TagTypeAlreadyExistsException ex) {
JOptionPane.showMessageDialog(null, "A " + tagDisplayName + " tag type has already been defined.", "Duplicate Tag Type", JOptionPane.ERROR_MESSAGE);
}
catch (TskCoreException ex) {
Logger.getLogger(TagAbstractFileMenu.class.getName()).log(Level.SEVERE, "Error adding " + tagDisplayName + " tag type", ex);
}
}
}

19
Core/src/org/sleuthkit/autopsy/directorytree/TagBlackboardArtifactAction.java
View File

@ -27,11 +27,9 @@ import javax.swing.JOptionPane;
import org.openide.util.Utilities;
import org.openide.util.actions.Presenter;
import org.sleuthkit.autopsy.casemodule.Case;
import org.sleuthkit.autopsy.casemodule.services.TagsManager;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.autopsy.datamodel.Tags;
import org.sleuthkit.datamodel.BlackboardArtifact;
import org.sleuthkit.datamodel.TagType;
import org.sleuthkit.datamodel.TskCoreException;
public class TagBlackboardArtifactAction extends AbstractAction implements Presenter.Popup {
@ -69,26 +67,9 @@ public class TagBlackboardArtifactAction extends AbstractAction implements Prese
@Override
protected void applyTag(String tagDisplayName, String comment) {
try {
TagsManager tagsManager = Case.getCurrentCase().getServices().getTagsManager();
TagType tagType = tagsManager.addTagType(tagDisplayName);
Collection<? extends BlackboardArtifact> selectedArtifacts = Utilities.actionsGlobalContext().lookupAll(BlackboardArtifact.class);
for (BlackboardArtifact artifact : selectedArtifacts) {
Tags.createTag(artifact, tagDisplayName, comment);
try {
tagsManager.addBlackboardArtifactTag(artifact, tagType);
}
catch (TskCoreException ex) {
Logger.getLogger(TagBlackboardArtifactMenu.class.getName()).log(Level.SEVERE, "Error tagging result", ex);
}
}
}
catch (TagsManager.TagTypeAlreadyExistsException ex) {
JOptionPane.showMessageDialog(null, "A " + tagDisplayName + " tag type has already been defined.", "Duplicate Tag Type", JOptionPane.ERROR_MESSAGE);
}
catch (TskCoreException ex) {
Logger.getLogger(TagBlackboardArtifactMenu.class.getName()).log(Level.SEVERE, "Error adding " + tagDisplayName + " tag type", ex);
}
}
}