|
|
|
@ -1,16 +1,36 @@
|
|
|
|
|
using System.Collections.Concurrent;
|
|
|
|
|
using System.Reflection;
|
|
|
|
|
using System.Security.Cryptography;
|
|
|
|
|
using Microsoft.Data.Sqlite;
|
|
|
|
|
using Serilog;
|
|
|
|
|
using Serilog.Events;
|
|
|
|
|
|
|
|
|
|
namespace Chksum.Utils;
|
|
|
|
|
public class ChksumUtils {
|
|
|
|
|
private ILogger logger = new LoggerConfiguration()
|
|
|
|
|
.MinimumLevel.Debug()
|
|
|
|
|
.WriteTo.Console(restrictedToMinimumLevel: LogEventLevel.Error)
|
|
|
|
|
.WriteTo.File("chksum.log")
|
|
|
|
|
.CreateLogger();
|
|
|
|
|
|
|
|
|
|
private int getFileCount() {
|
|
|
|
|
int fileCount = Directory.GetFiles(Directory.GetCurrentDirectory()).Length; // Get file count in current directory
|
|
|
|
|
return fileCount;
|
|
|
|
|
private int getTotalFileCount() {
|
|
|
|
|
int totalFileCount = Directory.GetFiles(Directory.GetCurrentDirectory(), "*", SearchOption.AllDirectories).Length;
|
|
|
|
|
logger.Debug("Total file count is {totalFileCount}", totalFileCount);
|
|
|
|
|
return totalFileCount - 3; // Remove the program, datbase and library from the totalFileCount
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private string[] indexFiles() {
|
|
|
|
|
string[] indexedFiles = Directory.GetFiles(Directory.GetCurrentDirectory(), "*", SearchOption.AllDirectories);
|
|
|
|
|
string[] filesToExclude = { "Chksum", "chksum.db", "libe_sqlite3.so" };
|
|
|
|
|
indexedFiles = indexedFiles.Where(file => !filesToExclude.Contains(Path.GetFileName(file))).ToArray();
|
|
|
|
|
logger.Information("All files were indexed");
|
|
|
|
|
return indexedFiles;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string DatabaseRoot { get; set; } = string.Empty;
|
|
|
|
|
public void getBaseDir() {
|
|
|
|
|
DatabaseRoot = AppDomain.CurrentDomain.BaseDirectory;
|
|
|
|
|
logger.Debug("DatabaseRoot is {DatabaseRoot}", DatabaseRoot);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string libraryPath { get; set; } = string.Empty;
|
|
|
|
@ -22,7 +42,9 @@ public class ChksumUtils {
|
|
|
|
|
byte[] buffer = new byte[resourceStream.Length];
|
|
|
|
|
resourceStream.Read(buffer, 0, buffer.Length);
|
|
|
|
|
File.WriteAllBytes(libraryPath, buffer);
|
|
|
|
|
logger.Debug("libe_sqlite3.so was successfully created");
|
|
|
|
|
} else {
|
|
|
|
|
logger.Error("libe_sqlite3.so could not be loaded");
|
|
|
|
|
throw new Exception(libraryPath + " could not be loaded");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
@ -30,6 +52,7 @@ public class ChksumUtils {
|
|
|
|
|
|
|
|
|
|
public void initializeDB() {
|
|
|
|
|
if (File.Exists("chksum.db")) {
|
|
|
|
|
logger.Information("A database already exits");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -48,6 +71,7 @@ public class ChksumUtils {
|
|
|
|
|
);
|
|
|
|
|
";
|
|
|
|
|
command.ExecuteNonQuery();
|
|
|
|
|
logger.Information("Database was successfully created");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -59,34 +83,46 @@ public class ChksumUtils {
|
|
|
|
|
vacuum;
|
|
|
|
|
";
|
|
|
|
|
command.ExecuteNonQuery();
|
|
|
|
|
logger.Debug("Database was successfully vacuumed");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private string CalculateMD5(string filename) {
|
|
|
|
|
using (var md5 = System.Security.Cryptography.MD5.Create()) {
|
|
|
|
|
private Dictionary<string, string> CalculateChecksums(string[] filenames) {
|
|
|
|
|
ConcurrentDictionary<string, string> checksums = new ConcurrentDictionary<string, string>();
|
|
|
|
|
|
|
|
|
|
Parallel.ForEach(filenames, (filename, state) => {
|
|
|
|
|
using (var md5 = MD5.Create()) {
|
|
|
|
|
using (var stream = File.OpenRead(filename)) {
|
|
|
|
|
var hash = md5.ComputeHash(stream);
|
|
|
|
|
return BitConverter.ToString(hash).Replace("-", "").ToLowerInvariant();
|
|
|
|
|
var checksum = BitConverter.ToString(hash).Replace("-", "").ToLowerInvariant();
|
|
|
|
|
|
|
|
|
|
lock (checksums) {
|
|
|
|
|
checksums.TryAdd(filename, checksum);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
logger.Debug("All files were checksummed");
|
|
|
|
|
return new Dictionary<string, string>(checksums);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void doTheThing() {
|
|
|
|
|
foreach (var directory in Directory.GetDirectories(Directory.GetCurrentDirectory()))
|
|
|
|
|
using (var connection = new SqliteConnection("Data Source=" + DatabaseRoot + "chksum.db;Mode=ReadWrite")) {
|
|
|
|
|
Directory.SetCurrentDirectory(directory); // Set new root
|
|
|
|
|
if (getFileCount() >= 1) {
|
|
|
|
|
DirectoryInfo dir = new DirectoryInfo(Directory.GetCurrentDirectory());
|
|
|
|
|
FileInfo[] files = dir.GetFiles();
|
|
|
|
|
foreach (FileInfo file in files) {
|
|
|
|
|
string fileName = file.Name;
|
|
|
|
|
string absolutePathToFile = Path.GetFullPath(fileName);
|
|
|
|
|
if (getTotalFileCount() < 1) {
|
|
|
|
|
logger.Information("There were no files to checksum");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
connection.Open();
|
|
|
|
|
Dictionary<string, string> fileHashes = CalculateChecksums(indexFiles());
|
|
|
|
|
|
|
|
|
|
foreach (var file in fileHashes) {
|
|
|
|
|
string absolutePathToFile = file.Key;
|
|
|
|
|
string fileName = Path.GetFileName(absolutePathToFile);
|
|
|
|
|
string pathToFile = Path.GetRelativePath(DatabaseRoot, absolutePathToFile);
|
|
|
|
|
string fileHash = CalculateMD5(fileName);
|
|
|
|
|
string fileHash = file.Value;
|
|
|
|
|
|
|
|
|
|
if (checkIfFileMovedAndUpdatePathToFile(fileHash, fileName, pathToFile) == false && checkIfFileAlreadyExistsInDatabase(fileHash, fileName) == false) {
|
|
|
|
|
connection.Open();
|
|
|
|
|
|
|
|
|
|
var command = connection.CreateCommand();
|
|
|
|
|
command.CommandText =
|
|
|
|
|
@"
|
|
|
|
@ -97,10 +133,10 @@ public class ChksumUtils {
|
|
|
|
|
command.Parameters.AddWithValue("$filename", fileName);
|
|
|
|
|
command.Parameters.AddWithValue("$pathtofile", pathToFile);
|
|
|
|
|
command.ExecuteNonQuery();
|
|
|
|
|
logger.Verbose("{fileName} which is located at {pathToFile} relative to the database with the hash {fileHash} was successfully inserted into the database", fileName, pathToFile, fileHash);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
doTheThing();
|
|
|
|
|
logger.Information("All files were successfully written to the database");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -125,9 +161,11 @@ public class ChksumUtils {
|
|
|
|
|
pathtofile = reader.GetString(1);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
logger.Verbose("{pathToFile} with the hash {fileHash} was successfully loaded", pathToFile, fileHash);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (fileHash == filehash) {
|
|
|
|
|
logger.Verbose("File with filehash {filehash} already exists in the database", filehash);
|
|
|
|
|
doesExist = true;
|
|
|
|
|
}
|
|
|
|
|
return doesExist;
|
|
|
|
@ -165,13 +203,14 @@ public class ChksumUtils {
|
|
|
|
|
command2.Parameters.AddWithValue("$filehash", fileHash);
|
|
|
|
|
command2.ExecuteNonQuery();
|
|
|
|
|
|
|
|
|
|
Console.WriteLine("File moved:");
|
|
|
|
|
Console.WriteLine("File moved or is a duplicate:");
|
|
|
|
|
Console.WriteLine($"\tfrom\t{pathToFile}");
|
|
|
|
|
Console.WriteLine($"\tto \t{pathtofile}\n");
|
|
|
|
|
wasMoved = true;
|
|
|
|
|
}
|
|
|
|
|
return wasMoved;
|
|
|
|
|
logger.Verbose("{fileName} which is located at {pathToFile} relative to the database with the hash {fileHash} was successfully checked", fileName, pathToFile, fileHash);
|
|
|
|
|
}
|
|
|
|
|
return wasMoved;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void checkIfFileWasDeleted() {
|
|
|
|
@ -190,7 +229,10 @@ public class ChksumUtils {
|
|
|
|
|
while (reader.Read()) {
|
|
|
|
|
pathToFile = reader.GetString(0);
|
|
|
|
|
|
|
|
|
|
if (!File.Exists(pathToFile)) {
|
|
|
|
|
if (File.Exists(pathToFile)) {
|
|
|
|
|
logger.Verbose("{pathToFile} exists", pathToFile);
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
var deleteCommand = connection.CreateCommand();
|
|
|
|
|
deleteCommand.CommandText =
|
|
|
|
|
@"
|
|
|
|
@ -202,9 +244,10 @@ public class ChksumUtils {
|
|
|
|
|
|
|
|
|
|
Console.WriteLine("File deleted:");
|
|
|
|
|
Console.WriteLine($"\t{pathToFile}\n");
|
|
|
|
|
logger.Verbose("File deleted: {pathToFile}", pathToFile);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
logger.Information("All deleted files were successfully removed from the database");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -230,12 +273,17 @@ public class ChksumUtils {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
logger.Debug("All filehashes were successfully retrived from the database");
|
|
|
|
|
return filehashesFromDatabase;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void compareDatabases(string filePathToOtherDatabase) {
|
|
|
|
|
if (!File.Exists(filePathToOtherDatabase)) {
|
|
|
|
|
logger.Error("No database could be found at {filePathToOtherDatabase}", filePathToOtherDatabase);
|
|
|
|
|
throw new Exception("No database could be found at " + filePathToOtherDatabase);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
List<string> filesThatDoNotExistsInTheRemote = getFilehashesFromDatabase("Data Source=" + DatabaseRoot + "chksum.db;Mode=ReadOnly").Except(getFilehashesFromDatabase("Data Source=" + filePathToOtherDatabase + ";Mode=ReadOnly")).ToList();
|
|
|
|
|
//List<string> filesThatDoNotExistsInTheOrigin = filehashesOfRemoteDatabase.Except(filehashesOfOriginDatabase).ToList();
|
|
|
|
|
|
|
|
|
|
foreach (string file in filesThatDoNotExistsInTheRemote) {
|
|
|
|
|
using (var connection = new SqliteConnection("Data Source=" + DatabaseRoot + "chksum.db;Mode=ReadOnly")) {
|
|
|
|
@ -256,13 +304,16 @@ public class ChksumUtils {
|
|
|
|
|
|
|
|
|
|
Console.WriteLine("File not found in remote:");
|
|
|
|
|
Console.WriteLine($"\t{filename}\n");
|
|
|
|
|
logger.Verbose("{filename} could not be found in the remote database", filename);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
logger.Information("Compared both databases successfully");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void cleanup() {
|
|
|
|
|
File.Delete(libraryPath);
|
|
|
|
|
logger.Debug("Successfully deleted libe_sqlite3.so");
|
|
|
|
|
}
|
|
|
|
|
}
|