using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Windows;
|
|
using TrustedUninstaller.Shared;
|
|
|
|
namespace TrustedUninstaller.CLI
|
|
{
|
|
public class CLI
|
|
{
|
|
private static async System.Threading.Tasks.Task<int> Main(string[] args)
|
|
{
|
|
//Needed after defender removal's reboot, the "current directory" will be set to System32
|
|
//After the auto start up.
|
|
Directory.SetCurrentDirectory(AppDomain.CurrentDomain.BaseDirectory);
|
|
|
|
DualOut.Init();
|
|
|
|
if (!WinUtil.IsAdministrator())
|
|
{
|
|
System.Console.Error.WriteLine("This program must be launched as an Administrator!");
|
|
return -1;
|
|
}
|
|
#if !DEBUG
|
|
/*
|
|
if (!WinUtil.IsGenuineWindows())
|
|
{
|
|
System.Console.Error.WriteLine("This program only works on genuine Windows copies!");
|
|
return -1;
|
|
}
|
|
*/
|
|
#endif
|
|
|
|
if (args.Length < 1 || !Directory.Exists(args[0]))
|
|
{
|
|
Console.WriteLine("No Playbook selected: Use the GUI to select a playbook to run.");
|
|
return -1;
|
|
}
|
|
|
|
AmeliorationUtil.Playbook = await AmeliorationUtil.DeserializePlaybook(args[0]);
|
|
|
|
|
|
if (!Directory.Exists($"{AmeliorationUtil.Playbook.Path}\\Configuration"))
|
|
{
|
|
Console.WriteLine("Creating Configuration folder...");
|
|
Directory.CreateDirectory($"{AmeliorationUtil.Playbook.Path}\\Configuration");
|
|
}
|
|
|
|
if (Directory.GetFiles($"{AmeliorationUtil.Playbook.Path}\\Configuration").Length == 0)
|
|
{
|
|
Console.WriteLine("Configuration folder is empty, put YAML files in it and restart the application.");
|
|
Console.WriteLine($"Current directory: {Directory.GetCurrentDirectory()}");
|
|
return -1;
|
|
}
|
|
|
|
ExtractResourceFolder("resources", Directory.GetCurrentDirectory());
|
|
|
|
await AmeliorationUtil.StartAmelioration();
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
public static void ExtractResourceFolder(string resource, string dir, bool overwrite = false)
|
|
{
|
|
if (!Directory.Exists(dir)) Directory.CreateDirectory(dir);
|
|
|
|
Assembly assembly = Assembly.GetExecutingAssembly();
|
|
|
|
var resources = assembly.GetManifestResourceNames().Where(res => res.StartsWith($"TrustedUninstaller.CLI.Properties"));
|
|
|
|
foreach (var obj in resources)
|
|
{
|
|
using (UnmanagedMemoryStream stream = (UnmanagedMemoryStream)assembly.GetManifestResourceStream(obj))
|
|
{
|
|
int MB = 1024 * 1024;
|
|
int offset = -MB;
|
|
|
|
var file = dir + $"\\{obj.Substring($"TrustedUninstaller.CLI.Properties.{resource}.".Length).Replace("---", "\\")}";
|
|
if (file.EndsWith(".gitkeep")) continue;
|
|
|
|
var fileDir = Path.GetDirectoryName(file);
|
|
if (fileDir != null && !Directory.Exists(fileDir)) Directory.CreateDirectory(fileDir);
|
|
|
|
if (File.Exists(file) && !overwrite) continue;
|
|
if (File.Exists(file) && overwrite)
|
|
{
|
|
try
|
|
{
|
|
File.Delete(file);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
if (!Directory.Exists(Directory.GetCurrentDirectory() + "\\Logs"))
|
|
Directory.CreateDirectory(Directory.GetCurrentDirectory() + "\\Logs");
|
|
using (var writer = new StreamWriter(Path.Combine(Directory.GetCurrentDirectory(), "Logs\\ErrorLog.txt"), true))
|
|
{
|
|
writer.WriteLine($"Title: Could not delete existing resource file {file}.\r\nMessage: {e.Message}\r\n\r\nStackTrace: {e.StackTrace}");
|
|
writer.WriteLine("\r\nDate/Time: " + DateTime.Now);
|
|
writer.WriteLine("============================================");
|
|
}
|
|
continue;
|
|
}
|
|
}
|
|
|
|
using (FileStream fsDlst = new FileStream(file, FileMode.CreateNew, FileAccess.Write))
|
|
{
|
|
while (offset + MB < stream.Length)
|
|
{
|
|
var buffer = new byte[MB];
|
|
offset += MB;
|
|
|
|
if (offset + MB > stream.Length)
|
|
{
|
|
var bytesLeft = stream.Length - offset;
|
|
buffer = new byte[bytesLeft];
|
|
}
|
|
|
|
stream.Seek(offset, SeekOrigin.Begin);
|
|
stream.Read(buffer, 0, buffer.Length);
|
|
|
|
fsDlst.Seek(offset, SeekOrigin.Begin);
|
|
fsDlst.Write(buffer, 0, buffer.Length);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|