From 56891e8a2f5df747951cbd746d377493aa2b9d52 Mon Sep 17 00:00:00 2001
From: Anairkoen Schno <aaron.cirr.com@gmail.com>
Date: Sat, 17 Aug 2019 17:17:05 -0500
Subject: [PATCH] Added basic anti-piracy checks

---
 IPA.Injector/AntiPiracy.cs       | 28 ++++++++++++++++++++++++++++
 IPA.Injector/GameVersionEarly.cs |  2 +-
 IPA.Injector/IPA.Injector.csproj | 11 ++++++-----
 IPA.Injector/Injector.cs         |  7 +++++++
 4 files changed, 42 insertions(+), 6 deletions(-)
 create mode 100644 IPA.Injector/AntiPiracy.cs

diff --git a/IPA.Injector/AntiPiracy.cs b/IPA.Injector/AntiPiracy.cs
new file mode 100644
index 00000000..dfb94024
--- /dev/null
+++ b/IPA.Injector/AntiPiracy.cs
@@ -0,0 +1,28 @@
+using System;
+using System.IO;
+using IPA.Utilities;
+#if NET3
+using Net3_Proxy;
+using Path = Net3_Proxy.Path;
+using File = Net3_Proxy.File;
+using Directory = Net3_Proxy.Directory;
+#endif
+
+namespace IPA.Injector
+{
+    internal class AntiPiracy
+    {
+        public static bool IsInvalid(string path)
+        {
+            var dataPlugins = Path.Combine(GameVersionEarly.ResolveDataPath(path), "Plugins");
+
+            return 
+                File.Exists(Path.Combine(path, "IGG-GAMES.COM.url")) ||
+                File.Exists(Path.Combine(path, "SmartSteamEmu.ini")) ||
+                File.Exists(Path.Combine(path, "GAMESTORRENT.CO.url")) ||
+                File.Exists(Path.Combine(dataPlugins, "BSteam crack.dll")) ||
+                File.Exists(Path.Combine(dataPlugins, "HUHUVR_steam_api64.dll")) ||
+                Directory.GetFiles(BeatSaber.PluginsPath, "*.ini", SearchOption.TopDirectoryOnly).Length > 0;
+        }
+    }
+}
diff --git a/IPA.Injector/GameVersionEarly.cs b/IPA.Injector/GameVersionEarly.cs
index ece652f7..78de36c9 100644
--- a/IPA.Injector/GameVersionEarly.cs
+++ b/IPA.Injector/GameVersionEarly.cs
@@ -14,7 +14,7 @@ namespace IPA.Injector
 {
     internal static class GameVersionEarly
     {
-        private static string ResolveDataPath(string installDir) => 
+        internal static string ResolveDataPath(string installDir) => 
             Path.Combine(Directory.EnumerateDirectories(installDir, "*_Data").First(), "globalgamemanagers");
 
         internal static string GetGameVersion()
diff --git a/IPA.Injector/IPA.Injector.csproj b/IPA.Injector/IPA.Injector.csproj
index eeda835b..bd4a6719 100644
--- a/IPA.Injector/IPA.Injector.csproj
+++ b/IPA.Injector/IPA.Injector.csproj
@@ -13,10 +13,10 @@
     <Deterministic>true</Deterministic>
     <PathMap>$(SolutionDir)=C:\</PathMap>
     <DebugType>portable</DebugType>
-    
     <ShowTargetFrameworks>false</ShowTargetFrameworks>
     <ShowTargetFrameworks Condition=" '$(BuildingInsideVisualStudio)' != 'true' Or '$(CI)' == 'True' ">true</ShowTargetFrameworks>
     <TargetFrameworks Condition=" '$(ShowTargetFrameworks)' == 'true' ">net461;net35</TargetFrameworks>
+    <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
   </PropertyGroup>
   <PropertyGroup Condition=" '$(Configuration)' == 'Debug' ">
     <DebugSymbols>true</DebugSymbols>
@@ -60,6 +60,7 @@
     </Reference>
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="AntiPiracy.cs" />
     <Compile Include="Backups\BackupManager.cs" />
     <Compile Include="Backups\BackupUnit.cs" />
     <Compile Include="Bootstrapper.cs" />
@@ -75,12 +76,12 @@
       <Project>{5ad344f0-01a0-4ca8-92e5-9d095737744d}</Project>
       <Name>IPA.Loader</Name>
     </ProjectReference>
-    <ProjectReference Include="..\Net3-Proxy\Net3-Proxy.csproj" Condition=" '$(Platform)' == 'Net3' " >
+    <ProjectReference Include="..\Net3-Proxy\Net3-Proxy.csproj" Condition=" '$(Platform)' == 'Net3' ">
       <Project>{642F52DA-90F9-40E3-8784-6964F36752FB}</Project>
       <Name>Net3-Proxy</Name>
     </ProjectReference>
   </ItemGroup>
-  <ItemGroup Condition=" '$(Platform)' == 'Net4' " >
+  <ItemGroup Condition=" '$(Platform)' == 'Net4' ">
     <Content Include="..\Libs\I18N.Net4.dll">
       <Link>Libraries\Mono\I18N.dll</Link>
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
@@ -89,7 +90,7 @@
       <Link>Libraries\Mono\I18N.West.dll</Link>
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </Content>
-    <Content Include="..\Libs\Microsoft.CSharp.dll" >
+    <Content Include="..\Libs\Microsoft.CSharp.dll">
       <Link>Libraries\Mono\Microsoft.CSharp.dll</Link>
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </Content>
@@ -98,7 +99,7 @@
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </Content>
   </ItemGroup>
-  <ItemGroup Condition=" '$(Platform)' == 'Net3' " >
+  <ItemGroup Condition=" '$(Platform)' == 'Net3' ">
     <Content Include="..\Libs\I18N.Net3.dll">
       <Link>Libraries\Mono\I18N.dll</Link>
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
diff --git a/IPA.Injector/Injector.cs b/IPA.Injector/Injector.cs
index f8f05623..106d2418 100644
--- a/IPA.Injector/Injector.cs
+++ b/IPA.Injector/Injector.cs
@@ -76,6 +76,13 @@ namespace IPA.Injector
                 SelfConfig.ReadCommandLine(Environment.GetCommandLineArgs());
                 DisabledConfig.Load();
 
+                if (AntiPiracy.IsInvalid(Environment.CurrentDirectory))
+                {
+                    loader.Error("Invalid installation; please buy the game to run BSIPA.");
+
+                    return;
+                }
+
                 loader.Debug("Prepping bootstrapper");
                 
                 // updates backup