You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

324 lines
11 KiB
C#

using System;
using System.Collections.Generic;
using System.IO;
using DotNetCTFDumper.MMFParser.EXE.Loaders;
using DotNetCTFDumper.MMFParser.EXE.Loaders.Banks;
using DotNetCTFDumper.Utils;
namespace DotNetCTFDumper.MMFParser.EXE
{
public class ChunkList
{
public List<Chunk> Chunks = new List<Chunk>();
public bool Verbose = false;
public List<Frame> Frames = new List<Frame>();
public void Read(ByteReader exeReader)
{
Chunks.Clear();
while (true)
{
Chunk chunk = new Chunk(Chunks.Count, this);
chunk.Verbose = Verbose;
chunk.Read(exeReader);
chunk.Loader = LoadChunk(chunk);
Chunks.Add(chunk);
if (chunk.Id == 8750)
{
chunk.BuildKey();
}
if (exeReader.Tell() >= exeReader.Size()) break;
if (chunk.Id == 32639) break; //LAST chunkID
}
}
public class Chunk
{
ChunkList _chunkList;
public string Name = "UNKNOWN";
int _uid;
public int Id = 0;
public ChunkLoader Loader;
public byte[] ChunkData;
public byte[] RawData;
public ChunkFlags Flag;
public int Size = 0;
public int DecompressedSize = -1;
public bool Verbose = false;
public Chunk(int actualuid, ChunkList actualChunkList)
{
_uid = actualuid;
_chunkList = actualChunkList;
}
public ByteReader get_reader()
{
return new ByteReader(ChunkData);
}
public void Read(ByteReader exeReader)
{
Id = exeReader.ReadInt16();
Name = this.ActualName();
Flag = (ChunkFlags) exeReader.ReadInt16();
Size = exeReader.ReadInt32();
if((Id!=26214&&Id!=26216)) //To prevent RAM from commiting suicide
{
RawData = exeReader.ReadBytes(Size);
exeReader.BaseStream.Position -= Size;
}
switch (Flag)
{
case ChunkFlags.Encrypted:
ChunkData = Decryption.DecodeChunk(exeReader.ReadBytes(Size),Size);
break;
case ChunkFlags.CompressedAndEncrypyed:
ChunkData = Decryption.DecodeMode3(exeReader.ReadBytes(Size), Size,Id,out DecompressedSize);
break;
case ChunkFlags.Compressed:
ChunkData = Decompressor.Decompress(exeReader,out DecompressedSize);
break;
case ChunkFlags.NotCompressed:
ChunkData = exeReader.ReadBytes(Size);
break;
}
int tempId=0;
int.TryParse(Name,out tempId);
if(tempId==Id)
{
//chunk_data.Log(true, "X2");
}
}
public void Save()
{
if (ChunkData != null)
{
string path = $"{Settings.ChunkPath}\\{Name}.chunk";
File.WriteAllBytes(path, ChunkData);
}
}
public void Print(bool extented)
{
if(extented)
{
Logger.Log($"Chunk: {Name} ({_uid})", true, ConsoleColor.DarkCyan);
Logger.Log($" ID: {Id} - 0x{Id.ToString("X")}", true, ConsoleColor.DarkCyan);
Logger.Log($" Flags: {Flag}", true, ConsoleColor.DarkCyan);
Logger.Log($" Loader: {(Loader != null ? Loader.GetType().Name : "Empty Loader")}", true,ConsoleColor.DarkCyan);
Logger.Log($" Size: {Size} B", true, ConsoleColor.DarkCyan);
Logger.Log($" Decompressed Size: {DecompressedSize} B", true, ConsoleColor.DarkCyan);
Logger.Log("---------------------------------------------", true, ConsoleColor.DarkCyan);
}
else
{
Logger.Log($"Chunk: {Name} ({_uid})", true, ConsoleColor.DarkCyan);
Logger.Log($" ID: {Id} - 0x{Id.ToString("X")}", true, ConsoleColor.DarkCyan);
Logger.Log($" Decompressed Size: {DecompressedSize} B", true, ConsoleColor.DarkCyan);
Logger.Log($" Flags: {Flag}", true, ConsoleColor.DarkCyan);
Logger.Log("---------------------------------------------", true, ConsoleColor.DarkCyan);
}
}
public void BuildKey()
{
string title = "";
string copyright = "";
string project = "";
var titleChunk = _chunkList.GetChunk<AppName>();
if (titleChunk != null) title = titleChunk.Value;
var copyrightChunk = _chunkList.GetChunk<Copyright>();
if (copyrightChunk != null) copyright = copyrightChunk.Value;
var projectChunk = _chunkList.GetChunk<EditorFilename>();
if (projectChunk != null) project = projectChunk.Value;
Settings.AppName=title;
Settings.Copyright = copyright;
Settings.ProjectPath = project;
if (Exe.Instance.GameData.ProductBuild > 284)
{
Decryption.MakeKey(title, copyright, project);
}
else
{
Decryption.MakeKey(project, title, copyright);
}
Logger.Log("New Key!");
}
}
public enum ChunkFlags
{
NotCompressed = 0,
Compressed = 1,
Encrypted = 2,
CompressedAndEncrypyed = 3
}
public ChunkLoader LoadChunk(Chunk chunk)
{
ChunkLoader loader = null;
switch (chunk.Id)
{
case 8739:
loader = new AppHeader(chunk);
break;
case 8740:
loader = new AppName(chunk);
break;
case 8741:
loader = new AppAuthor(chunk);
break;
case 8743:
loader = new ExtPath(chunk);
break;
case 8750:
loader = new EditorFilename(chunk);
break;
case 8751:
loader = new TargetFilename(chunk);
break;
case 8752:
loader = new AppDoc(chunk);
break;
case 8745:
loader = new FrameItems(chunk);
break;
case 8757:
loader = new AppIcon(chunk);
break;
case 8762:
loader = new AboutText(chunk);
break;
case 8763:
loader = new Copyright(chunk);
break;
case 13123:
loader = new DemoFilePath(chunk);
break;
case 13109:
loader = new FrameName(chunk);
break;
case 13107:
loader = new Frame(chunk);
Frames.Add((Frame)loader);
break;
case 13108:
loader = new FrameHeader(chunk);
break;
case 13111:
loader = new FramePalette(chunk);
break;
case 13112:
loader = new ObjectInstances(chunk);
break;
case 13115:
loader = new Transition(chunk);
break;
case 13116:
loader = new Transition(chunk);
break;
case 13121:
loader = new Layers(chunk);
break;
case 26214:
loader = new ImageBank(chunk);
break;
case 26216:
loader = new SoundBank(chunk);
break;
case 26217:
loader = new MusicBank(chunk);
break;
case 17477:
loader = new ObjectName(chunk);
break;
case 17476:
loader = new ObjectHeader(chunk);
break;
case 17478:
loader = new ObjectProperties(chunk);
return loader;
break;
case 8788:
//loader = new ObjectNames(chunk);
break;
case 8754:
loader = new GlobalValues(chunk);
break;
case 8755:
loader = new GlobalStrings(chunk);
break;
case 13117:
//loader = new Events(chunk);//NOT WORKING
break;
}
if (loader != null)
{
//Logger.Log($"Reading {loader.GetType().Name}...",true,ConsoleColor.Yellow);
loader.Read();
}
return loader;
}
public T GetChunk<T>() where T : ChunkLoader
{
foreach (Chunk chunk in Chunks)
{
if (chunk.Loader != null)
{
if (chunk.Loader.GetType().Name == typeof(T).Name)
{
return (T) chunk.Loader;
}
}
}
//Logger.Log($"ChunkLoader {typeof(T).Name} not found", true, ConsoleColor.Red);
return null;
}
public T PopChunk<T>() where T : ChunkLoader
{
for(int i=0;i<Chunks.Count;i++)
{
var chunk = Chunks[i];
if (chunk.Loader != null)
{
if (chunk.Loader.GetType().Name == typeof(T).Name)
{
return (T) chunk.Loader;
}
Chunks.Remove(chunk);
}
}
//Logger.Log($"ChunkLoader {typeof(T).Name} not found", true, ConsoleColor.Red);
return null;
}
}
}