108 lines
4.2 KiB
C#
108 lines
4.2 KiB
C#
using System.IO;
|
|
using System.Windows;
|
|
using JetBrains.Annotations;
|
|
using Livia.Models;
|
|
using Livia.Models.Data;
|
|
using Livia.Properties;
|
|
using Livia.Utility.DependencyInjection;
|
|
using Livia.ViewModels;
|
|
using Livia.Views;
|
|
using Livia.Views.Utility;
|
|
using MdpC2.ViewModels;
|
|
using MdpC2.Views.Controls;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace MdpC2;
|
|
|
|
/// <summary>
|
|
/// Interaction logic for App.xaml
|
|
/// </summary>
|
|
public partial class App
|
|
{
|
|
|
|
private readonly ILogger _logger;
|
|
private readonly IServiceProvider _serviceProvider;
|
|
private readonly IWarningSystem _warningSystem;
|
|
private readonly ISerialKeyManager _serialKeyManager;
|
|
private readonly IProcessManager _processManager;
|
|
[UsedImplicitly] private Mutex? _mutex;
|
|
|
|
public App()
|
|
{
|
|
ServiceConfigurations.AppName = "cereflow";
|
|
ServiceProviderFactory.Init(sc =>
|
|
{
|
|
sc.AddSingleton<ILiviaControlViewModel, MdpC2ControlViewModel>();
|
|
sc.AddSingleton<IServerHandler, MdpCServerHandler>();
|
|
});
|
|
|
|
InitializeComponent();
|
|
_serviceProvider = ServiceProviderFactory.ServiceProvider;
|
|
_logger = ActivatorUtilities.GetServiceOrCreateInstance<ILogger<App>>(ServiceProviderFactory.ServiceProvider);
|
|
_warningSystem = ActivatorUtilities.GetServiceOrCreateInstance<IWarningSystem>(ServiceProviderFactory.ServiceProvider);
|
|
_serialKeyManager = ActivatorUtilities.GetServiceOrCreateInstance<ISerialKeyManager>(ServiceProviderFactory.ServiceProvider);
|
|
_processManager = ActivatorUtilities.GetServiceOrCreateInstance<IProcessManager>(ServiceProviderFactory.ServiceProvider);
|
|
_logger.LogInformation("App Starting");
|
|
SettingsLogger.Init();
|
|
|
|
AppHelper helper = ActivatorUtilities.GetServiceOrCreateInstance<AppHelper>(ServiceProviderFactory.ServiceProvider);
|
|
|
|
Dispatcher.UnhandledException += helper.OnDispatcherUnhandledException;
|
|
helper.UpdateSettings();
|
|
helper.LoadStrings();
|
|
}
|
|
|
|
private void OnStartup(object sender, StartupEventArgs eventArgs)
|
|
{
|
|
_mutex = new Mutex(true, "Livia", out bool isNewInstance);
|
|
if (!isNewInstance)
|
|
{
|
|
_warningSystem.ShowDialog(WarningWindowKind.Warning, true, "InstanceIsRunningError");
|
|
Current.Shutdown();
|
|
}
|
|
|
|
//set to registration version
|
|
SettingsWindowViewModel.LanguageSelectionEnabled = true;
|
|
Settings.Default.ShowUpdateButton = false;
|
|
MainWindowViewModel.ShowCheckReportsButton = false;
|
|
MainWindowViewModel.ShowRemainingQuota = false;
|
|
Settings.Default.ShowArchiveDownloadButton = false;
|
|
Settings.Default.ShowSortedDownloadButton = false;
|
|
LoginControlViewModel.ShowMessage2 = true;
|
|
Settings.Default.ShowSelectReportModuleDialog = false;
|
|
Settings.Default.ServerAddress = "127.0.0.1";
|
|
Settings.Default.ServerPort = 8080;
|
|
|
|
bool success = _processManager.Start([
|
|
new ServerFileLocation("astroke_back-sqlite.exe", Path.Join(Directory.GetCurrentDirectory(), "astroke_cmd", "astroke_back-sqlite")),
|
|
new ServerFileLocation("data-storage-sqlite.exe", Path.Join(Directory.GetCurrentDirectory(), "astroke_cmd", "data-storage-sqlite"))
|
|
]);
|
|
if (!success)
|
|
{
|
|
_warningSystem.ShowDialog(WarningWindowKind.Error, true, "SystemCrashing");
|
|
_logger.LogError("Cannot start process");
|
|
Current.Shutdown();
|
|
return;
|
|
}
|
|
|
|
MainWindow mainWindow = ActivatorUtilities.CreateInstance<MainWindow>(_serviceProvider);
|
|
mainWindow.AddMainControl(new MdpC2Control(ActivatorUtilities.GetServiceOrCreateInstance<ILiviaControlViewModel>(ServiceProviderFactory.ServiceProvider)));
|
|
MainWindow = mainWindow;
|
|
|
|
if (!_serialKeyManager.Validate())
|
|
{
|
|
ValidateSerialKeyDialog dialog = new();
|
|
bool result = dialog.ShowDialog() ?? false;
|
|
if (!result)
|
|
{
|
|
_logger.LogWarning("ValidateSerialKeyDialog closed without success, shutting down");
|
|
Current.Shutdown();
|
|
}
|
|
}
|
|
|
|
mainWindow.Show();
|
|
|
|
_logger.LogInformation("App Started");
|
|
}
|
|
} |