【m】框架大更新

Assets/03.YooAsset/Samples~/Test Sample/Runtime/CryptoSample/TestFileStreamEncryption.cs

@@ -0,0 +1,159 @@
+using System;
+using System.IO;
+using System.Text;
+using System.Collections;
+using UnityEngine;
+using YooAsset;
+
+/// <summary>
+/// 资源文件解密流
+/// </summary>
+public class BundleStream : FileStream
+{
+    public const byte KEY = 64;
+
+    public BundleStream(string path, FileMode mode, FileAccess access, FileShare share) : base(path, mode, access, share)
+    {
+    }
+    public BundleStream(string path, FileMode mode) : base(path, mode)
+    {
+    }
+
+    public override int Read(byte[] array, int offset, int count)
+    {
+        var index = base.Read(array, offset, count);
+        for (int i = 0; i < array.Length; i++)
+        {
+            array[i] ^= KEY;
+        }
+        return index;
+    }
+}
+
+/// <summary>
+/// 文件流加密方式
+/// </summary>
+public class TestFileStreamEncryption : IEncryptionServices
+{
+    public EncryptResult Encrypt(EncryptFileInfo fileInfo)
+    {
+        // 说明：对TestRes3资源目录进行加密
+        if (fileInfo.BundleName.Contains("_testres3_"))
+        {
+            var fileData = File.ReadAllBytes(fileInfo.FileLoadPath);
+            for (int i = 0; i < fileData.Length; i++)
+            {
+                fileData[i] ^= BundleStream.KEY;
+            }
+
+            EncryptResult result = new EncryptResult();
+            result.Encrypted = true;
+            result.EncryptedData = fileData;
+            return result;
+        }
+        else
+        {
+            EncryptResult result = new EncryptResult();
+            result.Encrypted = false;
+            return result;
+        }
+    }
+}
+
+/// <summary>
+/// 资源文件流解密类
+/// </summary>
+public class TestFileStreamDecryption : IDecryptionServices
+{
+    /// <summary>
+    /// 同步方式获取解密的资源包对象
+    /// </summary>
+    DecryptResult IDecryptionServices.LoadAssetBundle(DecryptFileInfo fileInfo)
+    {
+        BundleStream bundleStream = new BundleStream(fileInfo.FileLoadPath, FileMode.Open, FileAccess.Read, FileShare.Read);
+        DecryptResult decryptResult = new DecryptResult();
+        decryptResult.ManagedStream = bundleStream;
+        decryptResult.Result = AssetBundle.LoadFromStream(bundleStream, fileInfo.FileLoadCRC, GetManagedReadBufferSize());
+        return decryptResult;
+    }
+
+    /// <summary>
+    /// 异步方式获取解密的资源包对象
+    /// </summary>
+    DecryptResult IDecryptionServices.LoadAssetBundleAsync(DecryptFileInfo fileInfo)
+    {
+        BundleStream bundleStream = new BundleStream(fileInfo.FileLoadPath, FileMode.Open, FileAccess.Read, FileShare.Read);
+        DecryptResult decryptResult = new DecryptResult();
+        decryptResult.ManagedStream = bundleStream;
+        decryptResult.CreateRequest = AssetBundle.LoadFromStreamAsync(bundleStream, fileInfo.FileLoadCRC, GetManagedReadBufferSize());
+        return decryptResult;
+    }
+
+    /// <summary>
+    /// 后备方式获取解密的资源包
+    /// 注意：当正常解密方法失败后，会触发后备加载！
+    /// 说明：建议通过LoadFromMemory()方法加载资源包作为保底机制。
+    /// </summary>
+    DecryptResult IDecryptionServices.LoadAssetBundleFallback(DecryptFileInfo fileInfo)
+    {
+        byte[] fileData = File.ReadAllBytes(fileInfo.FileLoadPath);
+        var assetBundle = AssetBundle.LoadFromMemory(fileData);
+        DecryptResult decryptResult = new DecryptResult();
+        decryptResult.Result = assetBundle;
+        return decryptResult;
+    }
+
+    /// <summary>
+    /// 获取解密的字节数据
+    /// </summary>
+    byte[] IDecryptionServices.ReadFileData(DecryptFileInfo fileInfo)
+    {
+        throw new System.NotImplementedException();
+    }
+
+    /// <summary>
+    /// 获取解密的文本数据
+    /// </summary>
+    string IDecryptionServices.ReadFileText(DecryptFileInfo fileInfo)
+    {
+        throw new System.NotImplementedException();
+    }
+
+    private static uint GetManagedReadBufferSize()
+    {
+        return 1024;
+    }
+}
+
+/// <summary>
+/// WebGL平台解密类
+/// 注意：WebGL平台支持内存解密
+/// </summary>
+public class TestWebFileMemoryDecryption : IWebDecryptionServices
+{
+    public WebDecryptResult LoadAssetBundle(WebDecryptFileInfo fileInfo)
+    {
+        /*
+        byte[] copyData = new byte[fileInfo.FileData.Length];
+        Buffer.BlockCopy(fileInfo.FileData, 0, copyData, 0, fileInfo.FileData.Length);
+
+        for (int i = 0; i < copyData.Length; i++)
+        {
+            copyData[i] ^= BundleStream.KEY;
+        }
+
+        WebDecryptResult decryptResult = new WebDecryptResult();
+        decryptResult.Result = AssetBundle.LoadFromMemory(copyData);
+        return decryptResult;
+        */
+
+        for (int i = 0; i < fileInfo.FileData.Length; i++)
+        {
+            fileInfo.FileData[i] ^= BundleStream.KEY;
+        }
+
+        WebDecryptResult decryptResult = new WebDecryptResult();
+        decryptResult.Result = AssetBundle.LoadFromMemory(fileInfo.FileData);
+        return decryptResult;
+    }
+}