mirror of
https://github.com/AdmiralCurtiss/WfcPatcher.git
synced 2026-04-25 15:35:25 -05:00
Basic overlay read/write code.
This commit is contained in:
Admiral H. Curtiss
parent
f623d12408
commit
c5e4b9d1e8
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -1,2 +1,3 @@
|
|||
bin
|
||||
obj
|
||||
*.user
|
||||
91
Program.cs
91
Program.cs
|
|
@ -6,6 +6,97 @@ using System.Text;
|
|||
namespace WfcPatcher {
|
||||
class Program {
|
||||
static void Main( string[] args ) {
|
||||
foreach ( string s in args ) {
|
||||
try {
|
||||
PatchFile( s );
|
||||
} catch ( Exception ex ) {
|
||||
Console.WriteLine( "Failed patching " + s );
|
||||
Console.WriteLine( ex.ToString() );
|
||||
Console.WriteLine();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void PatchFile( string filename ) {
|
||||
Console.WriteLine( "Reading & Copying ROM..." );
|
||||
var ndsSrc = new System.IO.FileStream( filename, System.IO.FileMode.Open );
|
||||
var nds = new System.IO.FileStream( filename + ".wfc.nds", System.IO.FileMode.Create );
|
||||
Util.CopyStream( ndsSrc, nds, (int)ndsSrc.Length );
|
||||
|
||||
// http://dsibrew.org/wiki/DSi_Cartridge_Header
|
||||
|
||||
// overlays
|
||||
Console.WriteLine( "Patching Overlays..." );
|
||||
nds.Position = 0x50;
|
||||
uint arm9overlayoff = nds.ReadUInt32();
|
||||
uint arm9overlaylen = nds.ReadUInt32();
|
||||
uint arm7overlayoff = nds.ReadUInt32();
|
||||
uint arm7overlaylen = nds.ReadUInt32();
|
||||
|
||||
PatchOverlay( nds, arm9overlayoff, arm9overlaylen );
|
||||
PatchOverlay( nds, arm7overlayoff, arm7overlaylen );
|
||||
}
|
||||
|
||||
static void PatchOverlay( System.IO.FileStream nds, uint pos, uint len ) {
|
||||
// http://sourceforge.net/p/devkitpro/ndstool/ci/master/tree/source/ndsextract.cpp
|
||||
// http://sourceforge.net/p/devkitpro/ndstool/ci/master/tree/source/overlay.h
|
||||
|
||||
nds.Position = 0x048;
|
||||
uint fatOffset = nds.ReadUInt32();
|
||||
|
||||
for ( uint i = 0; i < len; i += 0x20 ) {
|
||||
nds.Position = pos + i;
|
||||
uint id = nds.ReadUInt32();
|
||||
uint ramAddr = nds.ReadUInt32();
|
||||
uint ramSize = nds.ReadUInt32();
|
||||
uint bssSize = nds.ReadUInt32();
|
||||
uint sinitInit = nds.ReadUInt32();
|
||||
uint sinitInitEnd = nds.ReadUInt32();
|
||||
uint fileId = nds.ReadUInt32();
|
||||
uint reserved = nds.ReadUInt32();
|
||||
|
||||
nds.Position = fatOffset + 8 * id;
|
||||
uint overlayPositionStart = nds.ReadUInt32();
|
||||
uint overlayPositionEnd = nds.ReadUInt32();
|
||||
uint overlaySize = overlayPositionEnd - overlayPositionStart;
|
||||
|
||||
nds.Position = overlayPositionStart;
|
||||
byte[] data = new byte[overlaySize];
|
||||
nds.Read( data, 0, (int)overlaySize );
|
||||
|
||||
blz blz = new blz();
|
||||
byte[] decData = blz.BLZ_Decode( data );
|
||||
if ( ReplaceInData( decData ) ) {
|
||||
// if something was replaced, put it back into the ROM
|
||||
if ( blz.fileWasNotCompressed ) {
|
||||
data = decData; // which it should be anyway but yeah
|
||||
} else {
|
||||
data = blz.BLZ_Encode( decData, 0 );
|
||||
}
|
||||
nds.Position = overlayPositionStart;
|
||||
nds.Write( data, 0, data.Length );
|
||||
|
||||
// and write proper offsets
|
||||
overlayPositionEnd = (uint)nds.Position;
|
||||
byte[] newPosEndData = BitConverter.GetBytes( overlayPositionEnd );
|
||||
nds.Position = fatOffset + 8 * id + 4;
|
||||
nds.Write( newPosEndData, 0, 4 );
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
static bool ReplaceInData( byte[] data ) {
|
||||
string search = "https://";
|
||||
string replace = "http://";
|
||||
byte[] searchBytes = Encoding.ASCII.GetBytes( search );
|
||||
byte[] replaceBytes = Encoding.ASCII.GetBytes( replace );
|
||||
int requiredPadding = searchBytes.Length + replaceBytes.Length;
|
||||
|
||||
|
||||
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
399
Util.cs
Normal file
399
Util.cs
Normal file
|
|
@ -0,0 +1,399 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.IO;
|
||||
|
||||
namespace WfcPatcher {
|
||||
public static class Util {
|
||||
|
||||
#region SwapEndian
|
||||
public static Int16 SwapEndian( this Int16 x ) {
|
||||
return (Int16)SwapEndian( (UInt16)x );
|
||||
}
|
||||
public static UInt16 SwapEndian( this UInt16 x ) {
|
||||
return x = (UInt16)
|
||||
( ( x << 8 ) |
|
||||
( x >> 8 ) );
|
||||
}
|
||||
|
||||
public static Int32 SwapEndian( this Int32 x ) {
|
||||
return (Int32)SwapEndian( (UInt32)x );
|
||||
}
|
||||
public static UInt32 SwapEndian( this UInt32 x ) {
|
||||
return x = ( x << 24 ) |
|
||||
( ( x << 8 ) & 0x00FF0000 ) |
|
||||
( ( x >> 8 ) & 0x0000FF00 ) |
|
||||
( x >> 24 );
|
||||
}
|
||||
|
||||
public static Int64 SwapEndian( this Int64 x ) {
|
||||
return (Int64)SwapEndian( (UInt64)x );
|
||||
}
|
||||
public static UInt64 SwapEndian( this UInt64 x ) {
|
||||
return x = ( x << 56 ) |
|
||||
( ( x << 40 ) & 0x00FF000000000000 ) |
|
||||
( ( x << 24 ) & 0x0000FF0000000000 ) |
|
||||
( ( x << 8 ) & 0x000000FF00000000 ) |
|
||||
( ( x >> 8 ) & 0x00000000FF000000 ) |
|
||||
( ( x >> 24 ) & 0x0000000000FF0000 ) |
|
||||
( ( x >> 40 ) & 0x000000000000FF00 ) |
|
||||
( x >> 56 );
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region HexUtils
|
||||
public static byte ParseDecOrHexToByte( string s ) {
|
||||
s = s.Trim();
|
||||
|
||||
if ( s.StartsWith( "0x" ) ) {
|
||||
s = s.Substring( 2 );
|
||||
return Byte.Parse( s, System.Globalization.NumberStyles.HexNumber );
|
||||
} else {
|
||||
return Byte.Parse( s );
|
||||
}
|
||||
}
|
||||
|
||||
public static byte[] HexStringToByteArray( string hex ) {
|
||||
if ( hex.Length % 2 == 1 )
|
||||
throw new Exception( "The binary key cannot have an odd number of digits" );
|
||||
|
||||
byte[] arr = new byte[hex.Length >> 1];
|
||||
|
||||
for ( int i = 0; i < hex.Length >> 1; ++i ) {
|
||||
arr[i] = (byte)( ( GetHexVal( hex[i << 1] ) << 4 ) + ( GetHexVal( hex[( i << 1 ) + 1] ) ) );
|
||||
}
|
||||
|
||||
return arr;
|
||||
}
|
||||
|
||||
public static int GetHexVal( char hex ) {
|
||||
int val = (int)hex;
|
||||
//For uppercase A-F letters:
|
||||
//return val - (val < 58 ? 48 : 55);
|
||||
//For lowercase a-f letters:
|
||||
//return val - (val < 58 ? 48 : 87);
|
||||
//Or the two combined, but a bit slower:
|
||||
return val - ( val < 58 ? 48 : ( val < 97 ? 55 : 87 ) );
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region NumberUtils
|
||||
public static uint ToUInt24( byte[] File, int Pointer ) {
|
||||
byte b1 = File[Pointer];
|
||||
byte b2 = File[Pointer + 1];
|
||||
byte b3 = File[Pointer + 2];
|
||||
|
||||
return (uint)( b3 << 16 | b2 << 8 | b1 );
|
||||
}
|
||||
public static byte[] GetBytesForUInt24( uint Number ) {
|
||||
byte[] b = new byte[3];
|
||||
b[0] = (byte)( Number & 0xFF );
|
||||
b[1] = (byte)( ( Number >> 8 ) & 0xFF );
|
||||
b[2] = (byte)( ( Number >> 16 ) & 0xFF );
|
||||
return b;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// converts a 32-bit int that's actually a byte representation of a float
|
||||
/// to an actual float for use in calculations or whatever
|
||||
/// </summary>
|
||||
public static float UIntToFloat( uint integer ) {
|
||||
byte[] b = BitConverter.GetBytes( integer );
|
||||
float f = BitConverter.ToSingle( b, 0 );
|
||||
return f;
|
||||
}
|
||||
|
||||
public static int Align( this int Number, int Alignment ) {
|
||||
return (int)Align( (uint)Number, (uint)Alignment );
|
||||
}
|
||||
public static uint Align( this uint Number, uint Alignment ) {
|
||||
uint diff = Number % Alignment;
|
||||
if ( diff == 0 ) {
|
||||
return Number;
|
||||
} else {
|
||||
return ( Number + ( Alignment - diff ) );
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region TextUtils
|
||||
private static Encoding _ShiftJISEncoding = null;
|
||||
public static Encoding ShiftJISEncoding { get { if ( _ShiftJISEncoding == null ) { _ShiftJISEncoding = Encoding.GetEncoding( 932 ); } return _ShiftJISEncoding; } }
|
||||
public static String GetTextShiftJis( byte[] File, int Pointer ) {
|
||||
if ( Pointer == -1 ) return null;
|
||||
|
||||
try {
|
||||
int i = Pointer;
|
||||
while ( File[i] != 0x00 ) {
|
||||
i++;
|
||||
}
|
||||
String Text = ShiftJISEncoding.GetString( File, Pointer, i - Pointer );
|
||||
return Text;
|
||||
} catch ( Exception ) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
public static String GetTextAscii( byte[] File, int Pointer ) {
|
||||
if ( Pointer == -1 ) return null;
|
||||
|
||||
try {
|
||||
int i = Pointer;
|
||||
while ( File[i] != 0x00 ) {
|
||||
i++;
|
||||
}
|
||||
String Text = Encoding.ASCII.GetString( File, Pointer, i - Pointer );
|
||||
return Text;
|
||||
} catch ( Exception ) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
public static String GetTextUnicode( byte[] File, int Pointer, int MaxByteLength ) {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for ( int i = 0; i < MaxByteLength; i += 2 ) {
|
||||
ushort ch = BitConverter.ToUInt16( File, Pointer + i );
|
||||
if ( ch == 0 || ch == 0xFFFF ) { break; }
|
||||
sb.Append( (char)ch );
|
||||
}
|
||||
return sb.ToString();
|
||||
}
|
||||
public static String GetTextUTF8( byte[] File, int Pointer ) {
|
||||
int tmp;
|
||||
return GetTextUTF8( File, Pointer, out tmp );
|
||||
}
|
||||
public static String GetTextUTF8( byte[] File, int Pointer, out int NullLocation ) {
|
||||
if ( Pointer == -1 ) { NullLocation = -1; return null; }
|
||||
|
||||
try {
|
||||
int i = Pointer;
|
||||
while ( File[i] != 0x00 ) {
|
||||
i++;
|
||||
}
|
||||
String Text = Encoding.UTF8.GetString( File, Pointer, i - Pointer );
|
||||
NullLocation = i;
|
||||
return Text;
|
||||
} catch ( Exception ) {
|
||||
NullLocation = -1;
|
||||
return null;
|
||||
}
|
||||
}
|
||||
public static String TrimNull( this String s ) {
|
||||
int n = s.IndexOf( '\0', 0 );
|
||||
if ( n >= 0 ) {
|
||||
return s.Substring( 0, n );
|
||||
}
|
||||
return s;
|
||||
}
|
||||
public static byte[] StringToBytesShiftJis( String s ) {
|
||||
//byte[] bytes = ShiftJISEncoding.GetBytes(s);
|
||||
//return bytes.TakeWhile(subject => subject != 0x00).ToArray();
|
||||
return ShiftJISEncoding.GetBytes( s );
|
||||
}
|
||||
public static byte[] StringToBytesUTF16( String s ) {
|
||||
return Encoding.Unicode.GetBytes( s );
|
||||
}
|
||||
|
||||
public static string XmlEscape( string s ) {
|
||||
s = s.Replace( "&", "&" );
|
||||
s = s.Replace( "\"", """ );
|
||||
s = s.Replace( "'", "'" );
|
||||
s = s.Replace( "<", "<" );
|
||||
s = s.Replace( ">", ">" );
|
||||
return s;
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region TimeUtils
|
||||
public static DateTime UnixTimeToDateTime( ulong unixTime ) {
|
||||
return new DateTime( 1970, 1, 1, 0, 0, 0, 0, DateTimeKind.Utc ).AddSeconds( unixTime ).ToLocalTime();
|
||||
}
|
||||
public static ulong DateTimeToUnixTime( DateTime time ) {
|
||||
return (ulong)( time - new DateTime( 1970, 1, 1, 0, 0, 0, 0, DateTimeKind.Utc ).ToLocalTime() ).TotalSeconds;
|
||||
}
|
||||
public static DateTime PS3TimeToDateTime( ulong PS3Time ) {
|
||||
return new DateTime( 1, 1, 1, 0, 0, 0, 0, DateTimeKind.Utc ).AddMilliseconds( PS3Time / 1000 ).ToLocalTime();
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region ProgramUtils
|
||||
public static bool RunProgram( String prog, String args, bool displayCommandLine, bool displayOutput, bool useShell = false ) {
|
||||
if ( displayCommandLine ) {
|
||||
Console.Write( prog );
|
||||
Console.Write( " " );
|
||||
Console.WriteLine( args );
|
||||
}
|
||||
|
||||
// Use ProcessStartInfo class
|
||||
System.Diagnostics.ProcessStartInfo startInfo = new System.Diagnostics.ProcessStartInfo();
|
||||
startInfo.CreateNoWindow = false;
|
||||
startInfo.UseShellExecute = useShell;
|
||||
startInfo.FileName = prog;
|
||||
startInfo.WindowStyle = System.Diagnostics.ProcessWindowStyle.Hidden;
|
||||
startInfo.Arguments = args;
|
||||
startInfo.RedirectStandardOutput = !useShell;
|
||||
startInfo.RedirectStandardError = !useShell;
|
||||
//startInfo.RedirectStandardInput = !useShell;
|
||||
//startInfo.UserName = System.Security.Principal.WindowsIdentity.GetCurrent().Name;
|
||||
|
||||
using ( System.Diagnostics.Process exeProcess = System.Diagnostics.Process.Start( startInfo ) ) {
|
||||
exeProcess.WaitForExit();
|
||||
if ( useShell ) {
|
||||
return exeProcess.ExitCode == 0;
|
||||
}
|
||||
|
||||
string output;
|
||||
string err;
|
||||
output = exeProcess.StandardOutput.ReadToEnd();
|
||||
err = exeProcess.StandardError.ReadToEnd();
|
||||
int exitCode = exeProcess.ExitCode;
|
||||
|
||||
if ( exitCode != 0 ) {
|
||||
Console.WriteLine( prog + " returned nonzero:" );
|
||||
Console.WriteLine( output );
|
||||
throw new Exception( output );
|
||||
//return false;
|
||||
}
|
||||
|
||||
if ( displayOutput ) {
|
||||
Console.WriteLine( output );
|
||||
Console.WriteLine( err );
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region ArrayUtils
|
||||
public static void CopyByteArrayPart( IList<byte> from, int locationFrom, IList<byte> to, int locationTo, int count ) {
|
||||
for ( int i = 0; i < count; i++ ) {
|
||||
to[locationTo + i] = from[locationFrom + i];
|
||||
}
|
||||
}
|
||||
|
||||
public static void FillNull( IList<byte> Array, int Location, int Count ) {
|
||||
for ( int i = 0; i < Count; ++i ) {
|
||||
Array[Location + i] = 0x00;
|
||||
}
|
||||
}
|
||||
|
||||
public static bool IsByteArrayPartEqual( IList<byte> Array1, int Location1, IList<byte> Array2, int Location2, int count ) {
|
||||
for ( int i = 0; i < count; ++i ) {
|
||||
if ( Array1[i + Location1] != Array2[i + Location2] ) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region StreamUtils
|
||||
public static void CopyStream( System.IO.Stream input, System.IO.Stream output, int count ) {
|
||||
byte[] buffer = new byte[4096];
|
||||
int read;
|
||||
|
||||
int bytesLeft = count;
|
||||
while ( ( read = input.Read( buffer, 0, Math.Min( buffer.Length, bytesLeft ) ) ) > 0 ) {
|
||||
output.Write( buffer, 0, read );
|
||||
bytesLeft -= read;
|
||||
if ( bytesLeft <= 0 ) return;
|
||||
}
|
||||
}
|
||||
|
||||
public static uint ReadUInt32( this Stream s ) {
|
||||
int b1 = s.ReadByte();
|
||||
int b2 = s.ReadByte();
|
||||
int b3 = s.ReadByte();
|
||||
int b4 = s.ReadByte();
|
||||
|
||||
return (uint)( b4 << 24 | b3 << 16 | b2 << 8 | b1 );
|
||||
}
|
||||
public static uint PeekUInt32( this Stream s ) {
|
||||
long pos = s.Position;
|
||||
uint retval = s.ReadUInt32();
|
||||
s.Position = pos;
|
||||
return retval;
|
||||
}
|
||||
public static uint ReadUInt24( this Stream s ) {
|
||||
int b1 = s.ReadByte();
|
||||
int b2 = s.ReadByte();
|
||||
int b3 = s.ReadByte();
|
||||
|
||||
return (uint)( b3 << 16 | b2 << 8 | b1 );
|
||||
}
|
||||
public static uint PeekUInt24( this Stream s ) {
|
||||
long pos = s.Position;
|
||||
uint retval = s.ReadUInt24();
|
||||
s.Position = pos;
|
||||
return retval;
|
||||
}
|
||||
public static ushort ReadUInt16( this Stream s ) {
|
||||
int b1 = s.ReadByte();
|
||||
int b2 = s.ReadByte();
|
||||
|
||||
return (ushort)( b2 << 8 | b1 );
|
||||
}
|
||||
public static ushort PeekUInt16( this Stream s ) {
|
||||
long pos = s.Position;
|
||||
ushort retval = s.ReadUInt16();
|
||||
s.Position = pos;
|
||||
return retval;
|
||||
}
|
||||
|
||||
public static string ReadAsciiNullterm( this Stream s ) {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
int b = s.ReadByte();
|
||||
while ( b != 0 && b != -1 ) {
|
||||
sb.Append( (char)( b ) );
|
||||
b = s.ReadByte();
|
||||
}
|
||||
return sb.ToString();
|
||||
}
|
||||
public static string ReadAscii( this Stream s, int count ) {
|
||||
StringBuilder sb = new StringBuilder( count );
|
||||
int b;
|
||||
for ( int i = 0; i < count; ++i ) {
|
||||
b = s.ReadByte();
|
||||
sb.Append( (char)( b ) );
|
||||
}
|
||||
return sb.ToString();
|
||||
}
|
||||
public static string ReadUTF16Nullterm( this Stream s ) {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
byte[] b = new byte[2];
|
||||
int b0 = s.ReadByte();
|
||||
int b1 = s.ReadByte();
|
||||
while ( !( b0 == 0 && b1 == 0 ) && b1 != -1 ) {
|
||||
b[0] = (byte)b0; b[1] = (byte)b1;
|
||||
sb.Append( Encoding.Unicode.GetString( b, 0, 2 ) );
|
||||
b0 = s.ReadByte(); b1 = s.ReadByte();
|
||||
}
|
||||
return sb.ToString();
|
||||
}
|
||||
#endregion
|
||||
|
||||
public static string GuessFileExtension( Stream s ) {
|
||||
uint magic32 = s.PeekUInt32();
|
||||
uint magic24 = s.PeekUInt24();
|
||||
uint magic16 = s.PeekUInt16();
|
||||
|
||||
switch ( magic32 ) {
|
||||
case 0x46464952:
|
||||
return ".wav";
|
||||
case 0x474E5089:
|
||||
return ".png";
|
||||
case 0x5367674F:
|
||||
return ".ogg";
|
||||
}
|
||||
switch ( magic16 ) {
|
||||
case 0x4D42:
|
||||
return ".bmp";
|
||||
}
|
||||
|
||||
return "";
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
|
@ -46,6 +46,7 @@
|
|||
<Compile Include="blz.cs" />
|
||||
<Compile Include="Program.cs" />
|
||||
<Compile Include="Properties\AssemblyInfo.cs" />
|
||||
<Compile Include="Util.cs" />
|
||||
</ItemGroup>
|
||||
<Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
|
||||
<!-- To modify your build process, add your task inside one of the targets below and uncomment it.
|
||||
|
|
|
|||
7
blz.cs
7
blz.cs
|
|
@ -49,7 +49,8 @@ namespace WfcPatcher {
|
|||
// 0x00FFFFFF + 0x00200000 + 12 + padding
|
||||
|
||||
/*----------------------------------------------------------------------------*/
|
||||
public static int arm9;
|
||||
public int arm9;
|
||||
public bool fileWasNotCompressed = false;
|
||||
|
||||
/*----------------------------------------------------------------------------*/
|
||||
/*#define BREAK(text) { printf(text); return; }
|
||||
|
|
@ -154,6 +155,7 @@ namespace WfcPatcher {
|
|||
|
||||
/*----------------------------------------------------------------------------*/
|
||||
public byte[] BLZ_Decode( byte[] pak_buffer ) {
|
||||
fileWasNotCompressed = false;
|
||||
byte[] raw_buffer;
|
||||
uint pak, raw, pak_end, raw_end;
|
||||
uint pak_len, raw_len, len, pos, inc_len, hdr_len, enc_len, dec_len;
|
||||
|
|
@ -171,6 +173,9 @@ namespace WfcPatcher {
|
|||
dec_len = pak_len;
|
||||
pak_len = 0;
|
||||
raw_len = dec_len;
|
||||
|
||||
fileWasNotCompressed = true;
|
||||
return pak_buffer;
|
||||
} else {
|
||||
if ( pak_len < 8 ) throw new Exception( "File has a bad header" );
|
||||
hdr_len = pak_buffer[pak_len - 5];
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user