static void EnumerateFolder(string strFldrPath, ref RemSigCountStats stats)
{
if (!strFldrPath.EndsWith(@"\"))
{
strFldrPath += @"\";
}
string[] strFiles = null;
try
{
strFiles = Directory.GetFiles(strFldrPath);
}
catch (UnauthorizedAccessException)
{
//Skip it
}
if (strFiles != null)
{
foreach (string strFile in strFiles)
{
Console.CursorTop = 0;
Console.WriteLine(strFile + "\t\t\t\t\t\t\t\t\t");
//Do the fuzzing of the file
FuzzFile(strFile, ref stats);
//Output stats
Console.CursorTop = 10;
Console.Write(
$"Success: {stats.nCount_Success}\n" +
$"Not PE: {stats.nCount_NotPE}\n" +
$"Failed Open: {stats.nCount_FailedToOpen}\n" +
$"No signature: {stats.nCount_NoSig}\n"
);
}
foreach (string strFldr in Directory.GetDirectories(strFldrPath))
{
EnumerateFolder(strFldr, ref stats);
}
}
}
static void FuzzFile(string strFilePath, ref RemSigCountStats stats)
{
try
{
Process proc = new Process();
proc.StartInfo.FileName = @"..\..\..\..\Debug\SigRemover.exe";
proc.StartInfo.Arguments = $"-i \"{ strFilePath }\"";
proc.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;
proc.Start();
proc.WaitForExit();
EXIT_CODES exitCode = (EXIT_CODES)proc.ExitCode;
//Check exit codes
if (exitCode == EXIT_CODES.XC_BinaryHasNoSignature)
{
stats.nCount_NoSig++;
}
else if (exitCode == EXIT_CODES.XC_FailedToOpen)
{
stats.nCount_FailedToOpen++;
}
else if (exitCode == EXIT_CODES.XC_Not_PE_File)
{
stats.nCount_NotPE++;
}
else if (exitCode == EXIT_CODES.XC_Success)
{
stats.nCount_Success++;
}
else
{
//This needs our attention!
Console.WriteLine($"Failed exit code: { exitCode } in file: { strFilePath}");
}
}
catch (Exception ex)
{
Console.WriteLine($"Exception: { ex.ToString() }");
}
}
static void Main(string[] args)
{
RemSigCountStats stats = new RemSigCountStats();
EnumerateFolder(@"C:\", ref stats);
}
