/*
|
* Copyright (c) 2015-2019 Beebyte Limited. All rights reserved.
|
*/
|
#if !BEEBYTE_OBFUSCATOR_DISABLE
|
|
using System;
|
using System.IO;
|
using System.Net;
|
using UnityEditor;
|
using UnityEngine;
|
using UnityEditor.Callbacks;
|
#if UNITY_5_6_OR_NEWER
|
using UnityEditor.Build;
|
#endif
|
#if UNITY_2018_1_OR_NEWER
|
using UnityEditor.Build.Reporting;
|
#endif
|
|
namespace Beebyte.Obfuscator
|
{
|
/**
|
* Hooks into the Unity build pipeline and delegates to perform obfuscation / restoration if requested.
|
*/
|
#if UNITY_2018_2_OR_NEWER
|
public class PipelineHook : IPreprocessBuildWithReport, IPostBuildPlayerScriptDLLs, IFilterBuildAssemblies
|
#elif UNITY_2018_1_OR_NEWER
|
public class PipelineHook : IPreprocessBuildWithReport, IPostprocessBuildWithReport
|
#elif UNITY_5_6_OR_NEWER
|
public class PipelineHook : IPreprocessBuild, IPostprocessBuild
|
#else
|
public class PipelineHook
|
#endif
|
{
|
private static Project _project = new Project(false);
|
|
#if UNITY_5_6_OR_NEWER
|
public int callbackOrder
|
{
|
get { return 1; }
|
}
|
#endif
|
|
[PostProcessBuild(2)]
|
public static void PostProcessBuild(BuildTarget target, string pathToBuildProject)
|
{
|
_project.ObfuscateAssets(target, pathToBuildProject);
|
}
|
|
#if UNITY_2018_2_OR_NEWER
|
public void OnPostBuildPlayerScriptDLLs(BuildReport report)
|
#elif UNITY_2018_1_OR_NEWER
|
public void OnPostprocessBuild(BuildReport report)
|
#elif UNITY_5_6_OR_NEWER
|
public void OnPostprocessBuild(BuildTarget target, string path)
|
#else
|
[PostProcessBuild(1)]
|
private static void PostBuildHook(BuildTarget buildTarget, string pathToBuildProject)
|
#endif
|
{
|
if (_project.HasMonoBehaviourAssetsThatNeedReverting()) RestoreUtils.RestoreMonobehaviourSourceFiles();
|
RestoreUtils.RestoreOriginalDlls();
|
|
try
|
{
|
#if UNITY_2018_1_OR_NEWER
|
if ((report.summary.options & BuildOptions.BuildScriptsOnly) != 0 &&
|
(report.summary.options & BuildOptions.Development) != 0 && // BuildScriptsOnly is only applied in development builds
|
_project.ShouldObfuscate())
|
{
|
Debug.LogError("Obfuscation is not supported for the \"Scripts Only Build\" option");
|
return;
|
}
|
#endif
|
if (_project.IsSuccess()) return;
|
|
if (_project.HasCSharpScripts())
|
{
|
Debug.LogError("Failed to obfuscate");
|
#if UNITY_5_6_OR_NEWER
|
throw new BuildFailedException("Failed to obfuscate");
|
#else
|
throw new OperationCanceledException("Failed to obfuscate");
|
#endif
|
}
|
|
Debug.LogWarning("No obfuscation required because no C# scripts were found");
|
}
|
finally
|
{
|
Clear();
|
}
|
}
|
|
#if UNITY_2018_1_OR_NEWER
|
public void OnPreprocessBuild(BuildReport report)
|
{
|
_project = new Project((report.summary.options & BuildOptions.Development) != 0);
|
}
|
#elif UNITY_5_6_OR_NEWER
|
public void OnPreprocessBuild(BuildTarget target, string path)
|
{
|
_project = new Project(Debug.isDebugBuild);
|
}
|
#endif
|
|
#if UNITY_2018_2_OR_NEWER
|
public string[] OnFilterAssemblies(BuildOptions buildOptions, string[] assemblies)
|
{
|
_project.ObfuscateIfNeeded();
|
return assemblies;
|
}
|
#else
|
[PostProcessScene(1)]
|
public static void Obfuscate()
|
{
|
_project.ObfuscateIfNeeded();
|
}
|
#endif
|
|
public static void ClearProjectViaUpdate()
|
{
|
Clear();
|
EditorApplication.update -= ClearProjectViaUpdate;
|
}
|
|
private static void Clear()
|
{
|
_project = new Project(false);
|
}
|
}
|
|
[InitializeOnLoad]
|
public static class RestorationStatic
|
{
|
/*
|
* Often Unity's EditorApplication.update delegate is reset. Because it's so important to restore
|
* renamed MonoBehaviour assets we assign here where it will be called after scripts are compiled.
|
*/
|
static RestorationStatic()
|
{
|
#if UNITY_2018_2_OR_NEWER
|
#else
|
EditorApplication.update += RestoreUtils.RestoreMonobehaviourSourceFiles;
|
#endif
|
EditorApplication.update += RestoreUtils.RestoreOriginalDlls;
|
}
|
}
|
}
|
#endif
|
