93 lines
2.7 KiB
C#
93 lines
2.7 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.IO;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
|
|
namespace YooAsset.Editor
|
|
{
|
|
[TaskAttribute("资源包加密")]
|
|
public class TaskEncryption : IBuildTask
|
|
{
|
|
public class EncryptionContext : IContextObject
|
|
{
|
|
public List<string> EncryptList;
|
|
|
|
/// <summary>
|
|
/// 检测是否为加密文件
|
|
/// </summary>
|
|
public bool IsEncryptFile(string bundleName)
|
|
{
|
|
return EncryptList.Contains(bundleName);
|
|
}
|
|
}
|
|
|
|
void IBuildTask.Run(BuildContext context)
|
|
{
|
|
var buildParameters = context.GetContextObject<BuildParametersContext>();
|
|
var buildMapContext = context.GetContextObject<BuildMapContext>();
|
|
|
|
var buildMode = buildParameters.Parameters.BuildMode;
|
|
if (buildMode == EBuildMode.ForceRebuild || buildMode == EBuildMode.IncrementalBuild)
|
|
{
|
|
EncryptionContext encryptionContext = new EncryptionContext();
|
|
encryptionContext.EncryptList = EncryptFiles(buildParameters, buildMapContext);
|
|
context.SetContextObject(encryptionContext);
|
|
}
|
|
else
|
|
{
|
|
EncryptionContext encryptionContext = new EncryptionContext();
|
|
encryptionContext.EncryptList = new List<string>();
|
|
context.SetContextObject(encryptionContext);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 加密文件
|
|
/// </summary>
|
|
private List<string> EncryptFiles(BuildParametersContext buildParameters, BuildMapContext buildMapContext)
|
|
{
|
|
var encryptionServices = buildParameters.Parameters.EncryptionServices;
|
|
|
|
// 加密资源列表
|
|
List<string> encryptList = new List<string>();
|
|
|
|
// 如果没有设置加密类
|
|
if (encryptionServices == null)
|
|
return encryptList;
|
|
|
|
int progressValue = 0;
|
|
foreach (var bundleInfo in buildMapContext.BundleInfos)
|
|
{
|
|
if (encryptionServices.Check(bundleInfo.BundleName))
|
|
{
|
|
if (bundleInfo.IsRawFile)
|
|
{
|
|
UnityEngine.Debug.LogWarning($"Encryption not support raw file : {bundleInfo.BundleName}");
|
|
continue;
|
|
}
|
|
|
|
encryptList.Add(bundleInfo.BundleName);
|
|
|
|
// 注意：通过判断文件合法性，规避重复加密一个文件
|
|
string filePath = $"{buildParameters.PipelineOutputDirectory}/{bundleInfo.BundleName}";
|
|
byte[] fileData = File.ReadAllBytes(filePath);
|
|
if (EditorTools.CheckBundleFileValid(fileData))
|
|
{
|
|
byte[] bytes = encryptionServices.Encrypt(fileData);
|
|
File.WriteAllBytes(filePath, bytes);
|
|
BuildRunner.Log($"文件加密完成：{filePath}");
|
|
}
|
|
}
|
|
|
|
// 进度条
|
|
EditorTools.DisplayProgressBar("加密资源包", ++progressValue, buildMapContext.BundleInfos.Count);
|
|
}
|
|
EditorTools.ClearProgressBar();
|
|
|
|
if(encryptList.Count == 0)
|
|
UnityEngine.Debug.LogWarning($"没有发现需要加密的文件！");
|
|
return encryptList;
|
|
}
|
|
}
|
|
} |