+init

7 files changed, 439 insertions, 0 deletions

diff --git a/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/AmongUsModifier.cs b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/AmongUsModifier.cs
new file mode 100644
index 0000000..95f5524
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/AmongUsModifier.cs
@@ -0,0 +1,247 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Net;
+using System.Net.Sockets;
+using System.Runtime.InteropServices;
+using System.Threading.Tasks;
+using Gameloop.Vdf;
+using Gameloop.Vdf.Linq;
+using Impostor.Patcher.Shared.Events;
+using Impostor.Patcher.Shared.Innersloth;
+using ErrorEventArgs = Impostor.Patcher.Shared.Events.ErrorEventArgs;
+
+namespace Impostor.Patcher.Shared
+{
+    public class AmongUsModifier
+    {
+        private const uint AppId = 945360;
+        public const string DefaultRegionName = "Impostor";
+        public const ushort DefaultPort = 22023;
+
+        private readonly string _amongUsDir;
+        private readonly string _regionFile;
+
+        public string RegionName { get; set; } = DefaultRegionName;
+
+        public AmongUsModifier()
+        {
+            var appData = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "..", "LocalLow");
+
+            if (!Directory.Exists(appData))
+            {
+                appData = FindProtonAppData();
+            }
+
+            if (appData == null)
+                return;
+
+            _amongUsDir = Path.Combine(appData, "Innersloth", "Among Us");
+            _regionFile = Path.Combine(_amongUsDir, "regionInfo.dat");
+        }
+
+        private string FindProtonAppData()
+        {
+            string steamApps;
+
+            if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
+            {
+                steamApps = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".steam", "steam", "steamapps");
+            }
+            else if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
+            {
+                steamApps = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), "Library", "Application Support", "Steam", "steamapps");
+            }
+            else
+            {
+                return null;
+            }
+
+            if (!Directory.Exists(steamApps))
+                return null;
+
+            var libraries = new List<string>
+            {
+                steamApps
+            };
+
+            var vdf = Path.Combine(steamApps, "libraryfolders.vdf");
+            if (File.Exists(vdf))
+            {
+                var libraryFolders = VdfConvert.Deserialize(File.ReadAllText(vdf));
+
+                foreach (var libraryFolder in libraryFolders.Value.Children<VProperty>())
+                {
+                    if (!int.TryParse(libraryFolder.Key, out _))
+                        continue;
+
+                    libraries.Add(Path.Combine(libraryFolder.Value.Value<string>(), "steamapps"));
+                }
+            }
+
+            foreach (var library in libraries)
+            {
+                var path = Path.Combine(library, "compatdata", AppId.ToString(), "pfx", "drive_c", "users", "steamuser", "AppData", "LocalLow");
+                if (Directory.Exists(path))
+                {
+                    return path;
+                }
+            }
+
+            return null;
+        }
+
+        public async Task<bool> SaveIpAsync(string input)
+        {
+            // Filter out whitespace.
+            input = input.Trim();
+
+            // Split port from ip.
+            // Only IPv4 is supported so just do it simple.
+            var ip = string.Empty;
+            var port = DefaultPort;
+
+            var parts = input.Split(':');
+            if (parts.Length >= 1)
+            {
+                ip = parts[0];
+            }
+
+            if (parts.Length >= 2)
+            {
+                ushort.TryParse(parts[1], out port);
+            }
+
+            // Check if a valid IP address was entered.
+            if (!IPAddress.TryParse(ip, out var ipAddress))
+            {
+                // Attempt to resolve DNS.
+                try
+                {
+                    var hostAddresses = await Dns.GetHostAddressesAsync(ip);
+                    if (hostAddresses.Length == 0)
+                    {
+                        OnError("Invalid IP Address entered");
+                        return false;
+                    }
+
+                    // Use first IPv4 result.
+                    ipAddress = hostAddresses.First(x => x.AddressFamily == AddressFamily.InterNetwork);
+                }
+                catch (SocketException)
+                {
+                    OnError("Failed to resolve hostname.");
+                    return false;
+                }
+            }
+
+            // Only IPv4.
+            if (ipAddress.AddressFamily == AddressFamily.InterNetworkV6)
+            {
+                OnError("Invalid IP Address entered, only IPv4 is allowed.");
+                return false;
+            }
+
+            return WriteIp(ipAddress, port);
+        }
+
+        /// <summary>
+        ///     Writes an IP Address to the Among Us region file.
+        /// </summary>
+        /// <param name="ipAddress">The IPv4 address to write.</param>
+        /// <param name="port"></param>
+        private bool WriteIp(IPAddress ipAddress, ushort port)
+        {
+            if (ipAddress == null ||
+                ipAddress.AddressFamily != AddressFamily.InterNetwork)
+            {
+                throw new ArgumentException(nameof(ipAddress));
+            }
+
+            if (!Directory.Exists(_amongUsDir))
+            {
+                OnError("Among Us directory was not found, is it installed? Try running it once.");
+                return false;
+            }
+
+            using (var file = File.Open(_regionFile, FileMode.Create, FileAccess.Write))
+            using (var writer = new BinaryWriter(file))
+            {
+                var ip = ipAddress.ToString();
+                var region = new RegionInfo(RegionName, ip, new[]
+                {
+                    new ServerInfo($"{RegionName}-Master-1", ip, port)
+                });
+
+                region.Serialize(writer);
+
+                OnSaved(ip, port);
+                return true;
+            }
+        }
+
+        /// <summary>
+        ///     Loads the existing region info from the Among Us.
+        /// </summary>
+        public bool TryLoadRegionInfo(out RegionInfo regionInfo)
+        {
+            regionInfo = null;
+
+            if (!File.Exists(_regionFile))
+            {
+                return false;
+            }
+
+            using (var file = File.Open(_regionFile, FileMode.Open, FileAccess.Read))
+            using (var reader = new BinaryReader(file))
+            {
+                try
+                {
+                    regionInfo = RegionInfo.Deserialize(reader);
+                    return true;
+                }
+                catch (Exception exception)
+                {
+                    OnError("Couldn't parse region info\n" + exception);
+                    return false;
+                }
+            }
+        }
+
+        /// <summary>
+        ///     Loads the existing IP Address from the Among Us region file
+        ///     if it was set by Impostor before.
+        /// </summary>
+        public bool TryLoadIp(out string ipAddress)
+        {
+            ipAddress = null;
+
+            if (!TryLoadRegionInfo(out var regionInfo))
+            {
+                return false;
+            }
+
+            if ((regionInfo.Name == RegionName || regionInfo.Name == DefaultRegionName) && regionInfo.Servers.Count >= 1)
+            {
+                ipAddress = regionInfo.Servers.ElementAt(0).Ip;
+                return true;
+            }
+
+            return false;
+        }
+
+        private void OnError(string message)
+        {
+            Error?.Invoke(this, new ErrorEventArgs(message));
+        }
+
+        private void OnSaved(string ipAddress, ushort port)
+        {
+            Saved?.Invoke(this, new SavedEventArgs(ipAddress, port));
+        }
+
+        public event EventHandler<ErrorEventArgs> Error;
+        public event EventHandler<SavedEventArgs> Saved;
+    }
+}
\ No newline at end of file
diff --git a/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Configuration.cs b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Configuration.cs
new file mode 100644
index 0000000..b256156
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Configuration.cs
@@ -0,0 +1,65 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+
+namespace Impostor.Patcher.Shared
+{
+    public class Configuration
+    {
+        private const string FileRecentIps = @"recent_ips.txt";
+        private const int MaxRecentIps = 5;
+
+        private readonly string _baseDir;
+        private readonly string _recentIpsPath;
+        private readonly List<string> _recentIps;
+
+        public Configuration()
+        {
+            var appData = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData));
+
+            _baseDir = Path.Combine(appData, "Impostor");
+            _recentIpsPath = Path.Combine(_baseDir, FileRecentIps);
+            _recentIps = new List<string>();
+        }
+
+        public IReadOnlyList<string> RecentIps => _recentIps;
+
+        public void Load()
+        {
+            if (File.Exists(_recentIpsPath))
+            {
+                _recentIps.AddRange(File.ReadAllLines(_recentIpsPath));
+            }
+        }
+
+        public void Save()
+        {
+            Directory.CreateDirectory(_baseDir);
+
+            if (!Directory.Exists(_baseDir))
+            {
+                return;
+            }
+
+            if (_recentIps.Count > 0)
+            {
+                File.WriteAllLines(_recentIpsPath, _recentIps);
+            }
+        }
+
+        public void AddIp(string ip)
+        {
+            if (_recentIps.Contains(ip))
+            {
+                _recentIps.Remove(ip);
+            }
+
+            _recentIps.Insert(0, ip);
+
+            if (_recentIps.Count > MaxRecentIps)
+            {
+                _recentIps.RemoveAt(MaxRecentIps);
+            }
+        }
+    }
+}
\ No newline at end of file
diff --git a/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Events/ErrorEventArgs.cs b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Events/ErrorEventArgs.cs
new file mode 100644
index 0000000..7211d5d
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Events/ErrorEventArgs.cs
@@ -0,0 +1,14 @@
+using System;
+
+namespace Impostor.Patcher.Shared.Events
+{
+    public class ErrorEventArgs : EventArgs
+    {
+        public ErrorEventArgs(string message)
+        {
+            Message = message;
+        }
+
+        public string Message { get; }
+    }
+}
\ No newline at end of file
diff --git a/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Events/SavedEventArgs.cs b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Events/SavedEventArgs.cs
new file mode 100644
index 0000000..c91d071
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Events/SavedEventArgs.cs
@@ -0,0 +1,16 @@
+using System;
+
+namespace Impostor.Patcher.Shared.Events
+{
+    public class SavedEventArgs : EventArgs
+    {
+        public SavedEventArgs(string ipAddress, ushort port)
+        {
+            IpAddress = ipAddress;
+            Port = port;
+        }
+
+        public string IpAddress { get; }
+        public ushort Port { get; }
+    }
+}
\ No newline at end of file
diff --git a/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Impostor.Patcher.Shared.csproj b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Impostor.Patcher.Shared.csproj
new file mode 100644
index 0000000..e480870
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Impostor.Patcher.Shared.csproj
@@ -0,0 +1,12 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <TargetFrameworks>netstandard2.0;netstandard2.1</TargetFrameworks>
+        <Version>1.0.0</Version>
+    </PropertyGroup>
+
+    <ItemGroup>
+        <PackageReference Include="Gameloop.Vdf" Version="0.6.1" />
+    </ItemGroup>
+    
+</Project>
diff --git a/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Innersloth/RegionInfo.cs b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Innersloth/RegionInfo.cs
new file mode 100644
index 0000000..01b74d1
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Innersloth/RegionInfo.cs
@@ -0,0 +1,48 @@
+using System.Collections.Generic;
+using System.IO;
+
+namespace Impostor.Patcher.Shared.Innersloth
+{
+    public class RegionInfo
+    {
+        public RegionInfo(string name, string ping, IReadOnlyList<ServerInfo> servers)
+        {
+            Name = name;
+            Ping = ping;
+            Servers = servers;
+        }
+
+        public string Name { get; }
+        public string Ping { get; }
+        public IReadOnlyList<ServerInfo> Servers { get; }
+
+        public void Serialize(BinaryWriter writer)
+        {
+            writer.Write(0);
+            writer.Write(Name);
+            writer.Write(Ping);
+            writer.Write(Servers.Count);
+
+            foreach (var server in Servers)
+            {
+                server.Serialize(writer);
+            }
+        }
+
+        public static RegionInfo Deserialize(BinaryReader reader)
+        {
+            var unknown = reader.ReadInt32();
+            var name = reader.ReadString();
+            var ping = reader.ReadString();
+            var servers = new List<ServerInfo>();
+            var serverCount = reader.ReadInt32();
+
+            for (var i = 0; i < serverCount; i++)
+            {
+                servers.Add(ServerInfo.Deserialize(reader));
+            }
+
+            return new RegionInfo(name, ping, servers);
+        }
+    }
+}
\ No newline at end of file
diff --git a/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Innersloth/ServerInfo.cs b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Innersloth/ServerInfo.cs
new file mode 100644
index 0000000..7203c84
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Patcher/Impostor.Patcher.Shared/Innersloth/ServerInfo.cs
@@ -0,0 +1,37 @@
+using System.IO;
+using System.Net;
+
+namespace Impostor.Patcher.Shared.Innersloth
+{
+    public class ServerInfo
+    {
+        public string Name { get; }
+        public string Ip { get; }
+        public ushort Port { get; }
+
+        public ServerInfo(string name, string ip, ushort port)
+        {
+            Name = name;
+            Ip = ip;
+            Port = port;
+        }
+        
+        public void Serialize(BinaryWriter writer)
+        {
+            writer.Write(Name);
+            writer.Write(IPAddress.Parse(Ip).GetAddressBytes());
+            writer.Write(Port);
+            writer.Write(0);
+        }
+
+        public static ServerInfo Deserialize(BinaryReader reader)
+        {
+            var name = reader.ReadString();
+            var ip = new IPAddress(reader.ReadBytes(4)).ToString();
+            var port = reader.ReadUInt16();
+            var unknown = reader.ReadInt32();
+            
+            return new ServerInfo(name, ip, port);
+        }
+    }
+}
\ No newline at end of file