dcce4223f6
git-svn-id: svn://localhost/Users/andi/Downloads/code/trunk@28 eddbe33b-e435-4246-ac25-f5eb65f9a13c
107 lines
4.6 KiB
C#
107 lines
4.6 KiB
C#
/* This file is part of CustomizeMii
|
|
* Copyright (C) 2009 WiiCrazy / I.R.on
|
|
*
|
|
* CustomizeMii is free software: you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as published
|
|
* by the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* CustomizeMii is distributed in the hope that it will be
|
|
* useful, but WITHOUT ANY WARRANTY; without even the implied warranty
|
|
* of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
using System;
|
|
using System.IO;
|
|
using System.IO.Compression;
|
|
using System.Security.Cryptography;
|
|
|
|
namespace CustomizeMiiInstaller
|
|
{
|
|
public class InstallerHelper
|
|
{
|
|
public static MemoryStream CreateInstaller(string wadFile, byte iosToUse)
|
|
{
|
|
const int injectionPosition = 0x5A74C;
|
|
const int maxAllowedSizeForWads = 4 * 1024 * 1024 - 32; //(Max 4MB-32bytes )
|
|
|
|
//0. Read length of the wad to ensure it has an allowed size
|
|
byte[] wadFileBytes = File.ReadAllBytes(wadFile);
|
|
uint wadLength = (uint)wadFileBytes.Length;
|
|
|
|
if (wadLength > maxAllowedSizeForWads)
|
|
{
|
|
throw new ArgumentException(String.Format("The file {0} is sized above the max allowed limit of {1} for network installation.", wadFile, maxAllowedSizeForWads));
|
|
}
|
|
|
|
//1. Open the stub installer from resources
|
|
MemoryStream compressedStubInstallerStream = LoadCompressedStubInstaller("CustomizeMiiInstaller.dol.z");
|
|
compressedStubInstallerStream.Seek(0, SeekOrigin.Begin);
|
|
|
|
//2. Decompress compressed installer
|
|
MemoryStream uncompressedStubInstallerStream = new MemoryStream();
|
|
|
|
using (GZipStream gzipStream = new GZipStream(compressedStubInstallerStream, CompressionMode.Decompress))
|
|
{
|
|
byte[] decompressedBuff = new byte[1024];
|
|
while (true)
|
|
{
|
|
int length = gzipStream.Read(decompressedBuff, 0, 1024);
|
|
|
|
if (length == 0) break;
|
|
|
|
uncompressedStubInstallerStream.Write(decompressedBuff, 0, length);
|
|
}
|
|
|
|
}
|
|
|
|
//3. Take SHA of the wad and store it in the stub installer along with the size of the wad
|
|
|
|
byte[] shaHash;
|
|
using (SHA1 shaGen = SHA1.Create())
|
|
{
|
|
shaHash = shaGen.ComputeHash(wadFileBytes);
|
|
}
|
|
|
|
//4. Inject the data into the installer
|
|
|
|
//Write out the wad size
|
|
uncompressedStubInstallerStream.Seek(injectionPosition, SeekOrigin.Begin);
|
|
uncompressedStubInstallerStream.WriteByte((byte)((wadLength >> 24) & 0xff));
|
|
uncompressedStubInstallerStream.WriteByte((byte)((wadLength >> 16) & 0xff));
|
|
uncompressedStubInstallerStream.WriteByte((byte)((wadLength >> 8) & 0xff));
|
|
uncompressedStubInstallerStream.WriteByte((byte)(wadLength & 0xff));
|
|
|
|
//Write out the SHA1 value (Against corruption of the file on the network, this value will be checked by the installer)
|
|
uncompressedStubInstallerStream.Write(shaHash, 0, 20);
|
|
|
|
//Write out the ios to be used...
|
|
uncompressedStubInstallerStream.WriteByte(iosToUse);
|
|
|
|
//pad it with three zeroes (to align it into 32-bit)
|
|
uncompressedStubInstallerStream.WriteByte(0); uncompressedStubInstallerStream.WriteByte(0); uncompressedStubInstallerStream.WriteByte(0);
|
|
|
|
|
|
//Write out to be installed wad file's contents...
|
|
uncompressedStubInstallerStream.Write(wadFileBytes, 0, (int)wadLength);
|
|
|
|
return uncompressedStubInstallerStream;
|
|
}
|
|
|
|
private static MemoryStream LoadCompressedStubInstaller(string installerResourceName)
|
|
{
|
|
using (BinaryReader resLoader = new BinaryReader(System.Reflection.Assembly.GetExecutingAssembly().GetManifestResourceStream("CustomizeMiiInstaller.Resources." + installerResourceName)))
|
|
{
|
|
MemoryStream ms = new MemoryStream();
|
|
byte[] temp = resLoader.ReadBytes((int)resLoader.BaseStream.Length);
|
|
ms.Write(temp, 0, temp.Length);
|
|
return ms;
|
|
}
|
|
}
|
|
}
|
|
}
|