auto-creamapi-2/auto-creamapi/Services/CreamDllService.cs

115 lines
4.1 KiB
C#
Raw Normal View History

2020-12-19 19:14:40 -05:00
using System;
using System.Collections.Generic;
using System.IO;
using System.Security.Cryptography;
using System.Threading.Tasks;
using auto_creamapi.Models;
2020-12-19 19:14:40 -05:00
using auto_creamapi.Utils;
namespace auto_creamapi.Services
2020-12-19 19:14:40 -05:00
{
public interface ICreamDllService
2020-12-19 19:14:40 -05:00
{
public string TargetPath { get; set; }
public Task Initialize();
public void Save();
public void CheckIfDllExistsAtTarget();
public bool CreamApiApplied();
}
2020-12-19 19:14:40 -05:00
public class CreamDllService : ICreamDllService
{
2020-12-19 19:14:40 -05:00
private const string X86Arch = "x86";
private const string X64Arch = "x64";
private static readonly string HashPath = Path.Combine(Directory.GetCurrentDirectory(), "cream_api.md5");
2020-12-19 19:14:40 -05:00
private readonly Dictionary<string, CreamDll> _creamDlls = new Dictionary<string, CreamDll>();
2020-12-19 19:14:40 -05:00
private bool _x64Exists;
private bool _x86Exists;
public string TargetPath { get; set; }
public async Task Initialize()
{
MyLogger.Log.Debug("CreamDllService: Initialize begin");
2020-12-19 19:14:40 -05:00
_creamDlls.Add(X86Arch, new CreamDll("steam_api.dll", "steam_api_o.dll"));
_creamDlls.Add(X64Arch, new CreamDll("steam_api64.dll", "steam_api64_o.dll"));
if (!File.Exists(HashPath))
{
MyLogger.Log.Information("Writing md5sum file...");
await File.WriteAllLinesAsync(HashPath,
new[]
{
$"{_creamDlls[X86Arch].Hash} {_creamDlls[X86Arch].Filename}",
$"{_creamDlls[X64Arch].Hash} {_creamDlls[X64Arch].Filename}"
});
2020-12-19 19:14:40 -05:00
}
MyLogger.Log.Debug("CreamDllService: Initialize end");
2020-12-19 19:14:40 -05:00
}
public void Save()
{
if (_x86Exists) CopyDll(X86Arch);
if (_x64Exists) CopyDll(X64Arch);
}
public void CheckIfDllExistsAtTarget()
{
var x86file = Path.Combine(TargetPath, "steam_api.dll");
var x64file = Path.Combine(TargetPath, "steam_api64.dll");
_x86Exists = File.Exists(x86file);
_x64Exists = File.Exists(x64file);
if (_x86Exists) MyLogger.Log.Information($"x86 SteamAPI DLL found: {x86file}");
if (_x64Exists) MyLogger.Log.Information($"x64 SteamAPI DLL found: {x64file}");
}
public bool CreamApiApplied()
{
var a = CreamApiApplied("x86");
var b = CreamApiApplied("x64");
return a | b;
}
2020-12-19 19:14:40 -05:00
private void CopyDll(string arch)
{
var sourceSteamApiDll = _creamDlls[arch].Filename;
var targetSteamApiDll = Path.Combine(TargetPath, _creamDlls[arch].Filename);
var targetSteamApiOrigDll = Path.Combine(TargetPath, _creamDlls[arch].OrigFilename);
var targetSteamApiDllBackup = Path.Combine(TargetPath, $"{_creamDlls[arch].Filename}.backup");
2020-12-23 06:47:55 -05:00
MyLogger.Log.Information($"Setting up CreamAPI DLL @ {TargetPath} (arch :{arch})");
2020-12-19 19:14:40 -05:00
// Create backup of steam_api.dll
File.Copy(targetSteamApiDll, targetSteamApiDllBackup, true);
// Check if steam_api_o.dll already exists
// If missing rename original file
if (!File.Exists(targetSteamApiOrigDll))
File.Move(targetSteamApiDll, targetSteamApiOrigDll, true);
// Copy creamapi dll
File.Copy(sourceSteamApiDll, targetSteamApiDll, true);
}
public bool CreamApiApplied(string arch)
{
var a = File.Exists(Path.Combine(TargetPath, _creamDlls[arch].OrigFilename));
var b = GetHash(Path.Combine(TargetPath, _creamDlls[arch].Filename)).Equals(_creamDlls[arch].Hash);
2020-12-19 19:14:40 -05:00
return a & b;
}
private string GetHash(string filename)
{
if (File.Exists(filename))
{
using var md5 = MD5.Create();
using var stream = File.OpenRead(filename);
return BitConverter
.ToString(md5.ComputeHash(stream))
.Replace("-", string.Empty);
}
return "";
2020-12-19 19:14:40 -05:00
}
}
}