1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198
|
/*
** The Sleuth Kit
**
** Brian Carrier [carrier <at> sleuthkit [dot] org]
** Copyright (c) 2010-2013 Brian Carrier. All Rights reserved
**
** This software is distributed under the Common Public License 1.0
**
*/
/**
* \file case_db.cpp
* Contains class definition for TskCaseDb class to handle creating/opening a case
* database and adding images to it.
*/
#include "tsk_case_db.h"
#include "tsk_auto_i.h"
TskCaseDb::TskCaseDb(TskDb * a_db)
{
m_tag = TSK_CASE_DB_TAG;
m_db = a_db;
m_NSRLDb = NULL;
m_knownBadDb = NULL;
}
TskCaseDb::~TskCaseDb()
{
if (m_db != NULL) {
delete m_db;
m_db = NULL;
}
if (m_NSRLDb != NULL) {
tsk_hdb_close(m_NSRLDb);
m_NSRLDb = NULL;
}
if (m_knownBadDb != NULL) {
tsk_hdb_close(m_knownBadDb);
m_knownBadDb = NULL;
}
m_tag = 0;
}
/**
* Creates a new single-user case with a new database and initializes its tables.
* Fails if there's already a file at the given path.
*
* @param path Full path to create new database at.
* @returns Pointer to a new TskCaseDb object, NULL on error
*/
TskCaseDb *
TskCaseDb::newDb(const TSK_TCHAR * const path)
{
TskDb *db = new TskDbSqlite(path, true);
// Check if the database already exsists
if (db->dbExists()) {
tsk_error_reset();
tsk_error_set_errno(TSK_ERR_AUTO_DB);
tsk_error_set_errstr("Database %" PRIttocTSK
" already exists. Must be deleted first.", path);
delete(db);
return NULL;
}
// Open the database.
if (db->open(true)) {
delete(db);
return NULL;
}
return new TskCaseDb(db);
}
/**
* Opens a single-user case from an existing database.
*
* @param path Full path to open database from.
* @returns Pointer to a new TskCaseDb object, NULL on error
*/
TskCaseDb *
TskCaseDb::openDb(const TSK_TCHAR * path)
{
TskDb *db = new TskDbSqlite(path, true);
// Confirm that database already exsists
if (!db->dbExists()) {
tsk_error_reset();
tsk_error_set_errno(TSK_ERR_AUTO_DB);
tsk_error_set_errstr("Database %" PRIttocTSK
" does not exist. Must be created first.", path);
delete(db);
return NULL;
}
// Open the database.
if (db->open(false)) {
delete(db);
return NULL;
}
return new TskCaseDb(db);
}
/**
* Prepares the process to add an image to the database. This method
* allows the caller to specify options to be used during the ingest.
* @returns TskAutDb object that can be used to add the image.
*/
TskAutoDb *
TskCaseDb::initAddImage()
{
return new TskAutoDb(m_db, m_NSRLDb, m_knownBadDb);
}
/**
* Add an image to the database. This method does not allow you
* to customize any of the settings for ingest (such as hash calculation,
* and block map population). Use TskCaseDb::initAddImage() to set
* these values.
*
* @param numImg Number of images to add
* @param imagePaths Paths to the image splits to open.
* @param imgType TYpe of image format
* @param sSize Sector size of image
* @returns 1 on error and 0 on success
*/
uint8_t
TskCaseDb::addImage(int numImg, const TSK_TCHAR * const imagePaths[],
TSK_IMG_TYPE_ENUM imgType, unsigned int sSize)
{
TskAutoDb autoDb(m_db, m_NSRLDb, m_knownBadDb);
if (autoDb.startAddImage(numImg, imagePaths, imgType, sSize)) {
autoDb.revertAddImage();
return 1;
}
if (autoDb.commitAddImage()) {
return 1;
}
return 0;
}
/*
* Specify the NSRL index used for determining "known" files.
* @param images Path to index.
* @returns 1 on error and 0 on success
*/
uint8_t
TskCaseDb::setNSRLHashDb(TSK_TCHAR * const indexFile ) {
if (m_NSRLDb != NULL) {
tsk_hdb_close(m_NSRLDb);
m_NSRLDb = NULL;
}
TSK_HDB_OPEN_ENUM flags = TSK_HDB_OPEN_IDXONLY;
m_NSRLDb = tsk_hdb_open(indexFile, flags);
return m_NSRLDb != NULL;
}
/*
* Specify an index for determining "known bad" files.
* @param images Path to index.
* @returns 1 on error and 0 on success
*/
uint8_t
TskCaseDb::setKnownBadHashDb(TSK_TCHAR * const indexFile) {
if (m_knownBadDb != NULL) {
tsk_hdb_close(m_knownBadDb);
m_knownBadDb = NULL;
}
TSK_HDB_OPEN_ENUM flags = TSK_HDB_OPEN_IDXONLY;
m_knownBadDb = tsk_hdb_open(indexFile, flags);
return m_knownBadDb != NULL;
}
/*
* Clear set lookup databases.
* @param images Path to index.
*/
void
TskCaseDb::clearLookupDatabases() {
if (m_NSRLDb != NULL) {
tsk_hdb_close(m_NSRLDb);
m_NSRLDb = NULL;
}
if (m_knownBadDb != NULL) {
tsk_hdb_close(m_knownBadDb);
m_knownBadDb = NULL;
}
}
|