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

Finished reports code

Browse Source
This commit is contained in:
Oliver Spohngellert 2016-02-10 11:18:15 -05:00
parent f7b2dc6cfc
commit b75a5c4751
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
Core/src/org/sleuthkit/autopsy/report/ReportGenerator.java
View File

@ -533,16 +533,19 @@ class ReportGenerator {
} }
// Get the column headers appropriate for the artifact type. // Get the column headers appropriate for the artifact type.
List<Cell> columnHeaders = getArtifactTableColumnHeaders(type.getTypeID(), attrTypeSet); List<Cell> columnHeaders = getArtifactTableColumnHeaders(type.getTypeID(), attrTypeSet);
for (ArtifactData d : artifactList) { if(columnHeaders.isEmpty()) {
d.setColumnHeaders(columnHeaders); continue;
}
for (ArtifactData artData : artifactList) {
artData.setColumnHeaders(columnHeaders);
} }
// The most efficient way to sort all the Artifacts is to add them to a List, and then // The most efficient way to sort all the Artifacts is to add them to a List, and then
// sort that List based off a Comparator. Adding to a TreeMap/Set/List sorts the list // sort that List based off a Comparator. Adding to a TreeMap/Set/List sorts the list
// each time an element is added, which adds unnecessary overhead if we only need it sorted once. // each time an element is added, which adds unnecessary overhead if we only need it sorted once.
Collections.sort(artifactList); Collections.sort(artifactList);
List<String> columnHeaderNames = new ArrayList<>(); List<String> columnHeaderNames = new ArrayList<>();
for (Cell c : columnHeaders) { for (Cell currCell : columnHeaders) {
columnHeaderNames.add(c.getColumnHeader()); columnHeaderNames.add(currCell.getColumnHeader());
} }
for (TableReportModule module : tableModules) { for (TableReportModule module : tableModules) {
@ -1042,8 +1045,8 @@ class ReportGenerator {
module.addSetElement(currentKeyword); module.addSetElement(currentKeyword);
List<Cell> columnHeaders = getArtifactTableColumnHeaders(ARTIFACT_TYPE.TSK_KEYWORD_HIT.getTypeID(), new HashSet<BlackboardAttribute.Type>()); List<Cell> columnHeaders = getArtifactTableColumnHeaders(ARTIFACT_TYPE.TSK_KEYWORD_HIT.getTypeID(), new HashSet<BlackboardAttribute.Type>());
List<String> columnHeaderNames = new ArrayList<>(); List<String> columnHeaderNames = new ArrayList<>();
for (Cell c : columnHeaders) { for (Cell currCell : columnHeaders) {
columnHeaderNames.add(c.getColumnHeader()); columnHeaderNames.add(currCell.getColumnHeader());
} }
module.startTable(columnHeaderNames); module.startTable(columnHeaderNames);
} }
@ -1179,8 +1182,8 @@ class ReportGenerator {
module.startSet(currentSet); module.startSet(currentSet);
List<Cell> columnHeaders = getArtifactTableColumnHeaders(ARTIFACT_TYPE.TSK_HASHSET_HIT.getTypeID(), new HashSet<BlackboardAttribute.Type>()); List<Cell> columnHeaders = getArtifactTableColumnHeaders(ARTIFACT_TYPE.TSK_HASHSET_HIT.getTypeID(), new HashSet<BlackboardAttribute.Type>());
List<String> columnHeaderNames = new ArrayList<>(); List<String> columnHeaderNames = new ArrayList<>();
for (Cell c : columnHeaders) { for (Cell currCell : columnHeaders) {
columnHeaderNames.add(c.getColumnHeader()); columnHeaderNames.add(currCell.getColumnHeader());
} }
module.startTable(columnHeaderNames); module.startTable(columnHeaderNames);
tableProgress.get(module).updateStatusLabel( tableProgress.get(module).updateStatusLabel(
@ -2348,7 +2351,7 @@ class ReportGenerator {
public List<String> getRow() { public List<String> getRow() {
if (rowData == null) { if (rowData == null) {
try { try {
rowData = getOrderedRowDataAsStrings(this.columnHeaders); rowData = getOrderedRowDataAsStrings();
// replace null values if attribute was not defined // replace null values if attribute was not defined
for (int i = 0; i < rowData.size(); i++) { for (int i = 0; i < rowData.size(); i++) {
if (rowData.get(i) == null) { if (rowData.get(i) == null) {
@ -2377,7 +2380,7 @@ class ReportGenerator {
* *
* @throws TskCoreException * @throws TskCoreException
*/ */
private List<String> getOrderedRowDataAsStrings(List<Cell> columnHeaders) throws TskCoreException { private List<String> getOrderedRowDataAsStrings() throws TskCoreException {
Map<Integer, String> mappedAttributes = getMappedAttributes(); Map<Integer, String> mappedAttributes = getMappedAttributes();
List<String> orderedRowData = new ArrayList<>(); List<String> orderedRowData = new ArrayList<>();
if (ARTIFACT_TYPE.TSK_WEB_BOOKMARK.getTypeID() == getArtifact().getArtifactTypeID()) { if (ARTIFACT_TYPE.TSK_WEB_BOOKMARK.getTypeID() == getArtifact().getArtifactTypeID()) {
@ -2595,8 +2598,8 @@ class ReportGenerator {
orderedRowData.add(mappedAttributes.get(ATTRIBUTE_TYPE.TSK_LOCAL_PATH.getTypeID())); orderedRowData.add(mappedAttributes.get(ATTRIBUTE_TYPE.TSK_LOCAL_PATH.getTypeID()));
orderedRowData.add(mappedAttributes.get(ATTRIBUTE_TYPE.TSK_REMOTE_PATH.getTypeID())); orderedRowData.add(mappedAttributes.get(ATTRIBUTE_TYPE.TSK_REMOTE_PATH.getTypeID()));
} else { } else {
for (Cell c : columnHeaders) { for (Cell currCell : this.columnHeaders) {
String rowData = c.getRowData(this); String rowData = currCell.getRowData(this);
orderedRowData.add(rowData); orderedRowData.add(rowData);
} }
return orderedRowData; return orderedRowData;