/// <summary>
/// Extracts the texts and other data inside the codec.dat file from MGS.
/// </summary>
/// <param name="Data">The Stream with the data to be extracted</param>
/// <param name="OutputFolder">The output folder where the extracted data will be placed</param>
public static void Extract(Stream Data, string OutputFolder, MGSGame Game)
{
if (Game == MGSGame.MGS3)
{
MGSText.Initialize();
}
BinaryReader Reader = new BinaryReader(Data);
List <uint> AddressTable = new List <uint>();
uint Delimiter = Reader.ReadUInt32();
Data.Seek(0, SeekOrigin.Begin);
while (Data.Position < Data.Length)
{
uint Position = (uint)Data.Position;
uint Value = Reader.ReadUInt32();
if (Value == Delimiter)
{
AddressTable.Add(Position);
}
}
AddressTable.Add((uint)Data.Length);
for (int Index = 0; Index < AddressTable.Count - 1; Index++)
{
string OutputPath = Path.Combine(OutputFolder, string.Format("Text_{0:D5}", Index));
if (!Directory.Exists(OutputPath))
{
Directory.CreateDirectory(OutputPath);
}
uint HeaderLength = 0;
Data.Seek(AddressTable[Index], SeekOrigin.Begin);
while (Reader.ReadInt32() != -1)
{
HeaderLength += 4;
}
uint BaseOffset = (uint)Data.Position;
uint ScriptOffset = Reader.ReadUInt32() + BaseOffset;
uint DialogsTableOffset = Reader.ReadUInt32() + BaseOffset;
uint DialogsTextOffset = Reader.ReadUInt32() + BaseOffset;
uint ScriptHeaderOffset = Reader.ReadUInt32() + BaseOffset;
uint Key = Reader.ReadUInt32();
//Read header (with script pointers)
Data.Seek(AddressTable[Index], SeekOrigin.Begin);
byte[] Header = new byte[HeaderLength];
Reader.Read(Header, 0, Header.Length);
MGSCrypto Crypto = new MGSCrypto(Key);
StringBuilder Text = new StringBuilder();
//Read texts
if (DialogsTableOffset != DialogsTextOffset)
{
int TextIndex = 0;
bool HasText = true;
while (HasText)
{
Data.Seek(DialogsTableOffset + TextIndex++ *4, SeekOrigin.Begin);
uint Offset = (Reader.ReadUInt32() & 0x7fffffff) + DialogsTextOffset;
uint NextOffset = (Reader.ReadUInt32() & 0x7fffffff) + DialogsTextOffset;
if (Data.Position > DialogsTextOffset)
{
NextOffset = ScriptHeaderOffset;
HasText = false;
}
Data.Seek(Offset, SeekOrigin.Begin);
byte[] TextBuffer = new byte[NextOffset - Offset];
Reader.Read(TextBuffer, 0, TextBuffer.Length);
TextBuffer = UnpadText(Crypto.ProcessBuffer(TextBuffer));
Text.Append(MGSText.Buffer2Text(TextBuffer, Game));
Text.Append(TextSeparator);
}
}
if (Game == MGSGame.MGS3)
{
//This is probably not a header, but I don't know what this data is, so...
string ScriptHeaderFile = Path.Combine(OutputPath, "ScriptHeader.bin");
Data.Seek(ScriptHeaderOffset, SeekOrigin.Begin);
uint ScriptHeaderLength = ScriptOffset - ScriptHeaderOffset;
byte[] ScriptHeader = new byte[ScriptHeaderLength];
Reader.Read(ScriptHeader, 0, ScriptHeader.Length);
File.WriteAllBytes(ScriptHeaderFile, ScriptHeader);
}
//Read script
Data.Seek(ScriptOffset, SeekOrigin.Begin);
uint ScriptLength = AddressTable[Index + 1] - ScriptOffset;
byte[] Script = new byte[ScriptLength];
Reader.Read(Script, 0, Script.Length);
string HeaderFile = Path.Combine(OutputPath, "Header.bin");
string TextFile = Path.Combine(OutputPath, "Text.txt");
string ScriptFile = Path.Combine(OutputPath, "Script.bin");
File.WriteAllBytes(HeaderFile, Header);
File.WriteAllText(TextFile, Text.ToString());
File.WriteAllBytes(ScriptFile, Script);
if (OnStatusChanged != null)
{
float Percentage = (float)(Index + 1) / (AddressTable.Count - 1);
OnStatusChanged(Percentage);
}
}
}
/// <summary>
/// Creates a codec.dat file from a folder with the extracted data.
/// </summary>
/// <param name="Data">The output stream where the newly created data will be placed</param>
/// <param name="InputFolder">The input folder with the extracted data</param>
public static void Create(Stream Data, string InputFolder, MGSGame Game)
{
if (Game == MGSGame.MGS3)
{
MGSText.Initialize();
}
BinaryWriter Writer = new BinaryWriter(Data);
string[] Folders = Directory.GetDirectories(InputFolder);
int CurrentFolder = 0;
foreach (string Folder in Folders)
{
string HeaderFile = Path.Combine(Folder, "Header.bin");
string TextFile = Path.Combine(Folder, "Text.txt");
string ScriptFile = Path.Combine(Folder, "Script.bin");
if (File.Exists(HeaderFile) && File.Exists(TextFile) && File.Exists(ScriptFile))
{
const uint Key = 0x4805d94d;
MGSCrypto Crypto = new MGSCrypto(Key);
Writer.Write(File.ReadAllBytes(HeaderFile));
Writer.Write(-1);
byte[] ScriptHeader = null;
uint ScriptHeaderLength = 0;
if (Game == MGSGame.MGS3)
{
string ScriptHeaderFile = Path.Combine(Folder, "ScriptHeader.bin");
if (File.Exists(ScriptHeaderFile))
{
ScriptHeader = File.ReadAllBytes(ScriptHeaderFile);
ScriptHeaderLength = (uint)ScriptHeader.Length;
}
}
using (MemoryStream TextBlock = new MemoryStream())
{
BinaryWriter TextWriter = new BinaryWriter(TextBlock);
string Text = File.ReadAllText(TextFile);
string[] Texts = Text.Split(new string[] { TextSeparator }, StringSplitOptions.RemoveEmptyEntries);
int BaseTextOffset = 0x14 + Texts.Length * 4;
int TextOffset = 0;
TextBlock.Seek(BaseTextOffset, SeekOrigin.Begin);
for (int Index = 0; Index < Texts.Length; Index++)
{
TextBlock.Seek(0x14 + Index * 4, SeekOrigin.Begin);
TextWriter.Write((uint)(TextOffset | 0x80000000));
TextBlock.Seek(BaseTextOffset + TextOffset, SeekOrigin.Begin);
byte[] Buffer = MGSText.Text2Buffer(Texts[Index], Game);
TextWriter.Write(Crypto.ProcessBuffer(Buffer));
TextOffset += Buffer.Length;
}
int PadIndex = 0;
while ((TextBlock.Position & 3) != 0)
{
byte Encrypted = Crypto.ProcessByte((byte)Padding[PadIndex++]);
TextBlock.WriteByte(Encrypted);
}
uint ScriptHeaderOffset = (uint)TextBlock.Position;
uint ScriptOffset = ScriptHeaderOffset + ScriptHeaderLength;
TextBlock.Seek(0, SeekOrigin.Begin);
TextWriter.Write(ScriptOffset);
TextWriter.Write(0x14);
TextWriter.Write(BaseTextOffset);
TextWriter.Write(ScriptHeaderOffset);
TextWriter.Write(Key);
Writer.Write(TextBlock.ToArray());
}
if (ScriptHeaderLength > 0)
{
Writer.Write(ScriptHeader);
}
Writer.Write(File.ReadAllBytes(ScriptFile));
while ((Data.Position & 0xf) != 0)
{
Data.WriteByte(0);
}
}
if (OnStatusChanged != null)
{
float Percentage = (float)++CurrentFolder / Folders.Length;
OnStatusChanged(Percentage);
}
}
}
