Merge branch 'custom_artifact_addition' into custom_artifact_addition-1857

This commit is contained in:
Oliver Spohngellert 2016-01-14 12:24:42 -05:00
commit 03c53e62b6

View File

@ -11,6 +11,8 @@ import org.sleuthkit.autopsy.ingest.DataSourceIngestModule;
import org.sleuthkit.autopsy.ingest.DataSourceIngestModuleProgress; import org.sleuthkit.autopsy.ingest.DataSourceIngestModuleProgress;
import org.sleuthkit.autopsy.ingest.IngestJobContext; import org.sleuthkit.autopsy.ingest.IngestJobContext;
import org.sleuthkit.datamodel.BlackboardArtifact; import org.sleuthkit.datamodel.BlackboardArtifact;
import org.sleuthkit.datamodel.BlackboardAttribute;
import org.sleuthkit.datamodel.BlackboardAttribute.ATTRIBUTE_TYPE;
import org.sleuthkit.datamodel.Content; import org.sleuthkit.datamodel.Content;
import org.sleuthkit.datamodel.TskCoreException; import org.sleuthkit.datamodel.TskCoreException;
@ -28,9 +30,13 @@ public class UserArtifactIngestModule implements DataSourceIngestModule {
public ProcessResult process(Content dataSource, DataSourceIngestModuleProgress progressBar) { public ProcessResult process(Content dataSource, DataSourceIngestModuleProgress progressBar) {
progressBar.switchToDeterminate(2); progressBar.switchToDeterminate(2);
try { try {
dataSource.newArtifact(type1ID); BlackboardArtifact art1 = dataSource.newArtifact(type1ID);
art1.addAttribute(new BlackboardAttribute(ATTRIBUTE_TYPE.TSK_COUNT,
UserArtifactIngestModuleFactory.getModuleName(), 5));
progressBar.progress(1); progressBar.progress(1);
dataSource.newArtifact(type2ID); BlackboardArtifact art2 = dataSource.newArtifact(type2ID);
art2.addAttribute(new BlackboardAttribute(ATTRIBUTE_TYPE.TSK_MIN_COUNT,
UserArtifactIngestModuleFactory.getModuleName(), 4));
progressBar.progress(1); progressBar.progress(1);
return ProcessResult.OK; return ProcessResult.OK;
} }