125 lines
3.9 KiB
C#
125 lines
3.9 KiB
C#
using System.Collections;
|
||
using System.Collections.Generic;
|
||
using System.IO;
|
||
using UnityEngine;
|
||
using System.Text;
|
||
using System.Reflection;
|
||
using System;
|
||
using System.Linq;
|
||
|
||
namespace UnityFS
|
||
{
|
||
/// <summary>
|
||
/// Unity 生成的二进制文件(本代码不支持5.x之前的版本)
|
||
/// </summary>
|
||
public unsafe class UnityBinFile
|
||
{
|
||
/*
|
||
* MonoManager: idx: 6;
|
||
* type: metaData.types[objects[6].typeID]
|
||
*/
|
||
public const int kMonoManagerIdx = 6;
|
||
|
||
public FileHeader header;
|
||
public MetaData metaData;
|
||
public ScriptsData scriptsData;
|
||
|
||
private Stream _originStream;
|
||
|
||
public void LoadFromStream(Stream source)
|
||
{
|
||
_originStream = source;
|
||
using (var br = new BinaryReader(source, Encoding.UTF8, true))
|
||
{
|
||
header.LoadFromStream(br);
|
||
// 按理说 metaData 应该新开一个buffer来避免加载时的对齐逻辑问题，但由于 sizeof(Header) = 20，已经对齐到4了，所以可以连续读
|
||
metaData.LoadFromStream(br, header.dataOffset);
|
||
scriptsData = metaData.GetScriptData(br);
|
||
}
|
||
}
|
||
|
||
public void Load(string path)
|
||
{
|
||
LoadFromStream(new MemoryStream(File.ReadAllBytes(path)));
|
||
}
|
||
|
||
public void AddScriptingAssemblies(List<string> assemblies)
|
||
{
|
||
foreach (string name in assemblies)
|
||
{
|
||
if (!scriptsData.dllNames.Contains(name))
|
||
{
|
||
scriptsData.dllNames.Add(name);
|
||
scriptsData.dllTypes.Add(16); // user dll type
|
||
Debug.Log($"[PatchScriptAssembliesJson] add dll:{name} to globalgamemanagers");
|
||
}
|
||
}
|
||
}
|
||
|
||
public byte[] CreatePatchedBytes()
|
||
{
|
||
var fsR = _originStream;
|
||
fsR.Position = 0;
|
||
var brR = new BinaryReader(fsR, Encoding.UTF8, true);
|
||
|
||
var ms = new MemoryStream((int)(header.fileSize * 1.5f));
|
||
var bw = new BinaryWriter(ms, Encoding.UTF8, true);
|
||
|
||
/*
|
||
* 开始写入data
|
||
* dll名称列表存储于 data 区段，修改其数据并不会影响 MetaData 大小，因此 dataOffset 不会改变
|
||
*/
|
||
ms.Position = header.dataOffset;
|
||
|
||
Dictionary<long, ObjectInfo> newObjInfos = new Dictionary<long, ObjectInfo>();
|
||
foreach (var kv in metaData.objects)
|
||
{
|
||
long objID = kv.Key;
|
||
ObjectInfo objInfo = kv.Value;
|
||
|
||
byte[] buff = new byte[objInfo.size];
|
||
fsR.Position = objInfo.realPos;
|
||
brR.Read(buff, 0, buff.Length);
|
||
|
||
|
||
{// unity 的数据偏移貌似会对齐到 8
|
||
int newPos = (((int)ms.Position + 7) >> 3) << 3;
|
||
int gapSize = newPos - (int)ms.Position;
|
||
|
||
for (int i = 0; i < gapSize; i++)
|
||
bw.Write((byte)0);
|
||
|
||
objInfo.dataPos = (uint)ms.Position - header.dataOffset; // 重定位数据偏移
|
||
}
|
||
|
||
if (objID != kMonoManagerIdx)
|
||
bw.Write(buff, 0, buff.Length);
|
||
else
|
||
objInfo.size = (uint)scriptsData.SaveToStream(bw);
|
||
|
||
newObjInfos.Add(objID, objInfo);
|
||
}
|
||
|
||
metaData.objects = newObjInfos;
|
||
header.fileSize = (uint)ms.Position;
|
||
|
||
ms.Position = 0;
|
||
header.SaveToStream(bw);
|
||
metaData.SaveToStream(bw);
|
||
|
||
brR.Close();
|
||
|
||
// 写入新文件
|
||
ms.Position = 0;
|
||
return ms.ToArray();
|
||
}
|
||
|
||
public void Save(string newPath)
|
||
{
|
||
byte[] patchedBytes = CreatePatchedBytes();
|
||
File.WriteAllBytes(newPath, patchedBytes);
|
||
}
|
||
}
|
||
|
||
}
|