public static bool Write(ImgHeader hdr, byte[] kernel, byte[] ramdisk, byte[] second, string OutFile, ref long sz)
{
if (hdr == null)
return false;
//Setting new sizes
hdr.kernel_size = (kernel != null) ? (uint)kernel.Length : 0;
hdr.ramdisk_size = (ramdisk != null) ? (uint)ramdisk.Length : 0;
hdr.second_size = (second != null) ? (uint)second.Length : 0;
//Generating data to calc SHA1
List<byte> data_to_hash = new List<byte>();
data_to_hash.AddRange(kernel);
data_to_hash.AddRange(BitConverter.GetBytes(hdr.kernel_size));
data_to_hash.AddRange(ramdisk);
data_to_hash.AddRange(BitConverter.GetBytes(hdr.ramdisk_size));
if (second != null)
data_to_hash.AddRange(second);
data_to_hash.AddRange(BitConverter.GetBytes(hdr.second_size));
SHA1 sha = new SHA1CryptoServiceProvider();
byte[] sha_bytes = sha.ComputeHash(data_to_hash.ToArray(), 0, data_to_hash.Count);
hdr.id = new byte[32];
Array.Copy(sha_bytes, 0, hdr.id, 0,
SHA_DIGEST_SIZE > hdr.id.Length ? hdr.id.Length : SHA_DIGEST_SIZE);
byte[] hdr_bytes = hdr.ToBytes();
try
{
FileStream fs = File.Open(OutFile, FileMode.Create, FileAccess.Write);
fs.Write(hdr_bytes, 0, hdr_bytes.Length);
write_padding(ref fs, hdr.page_size, (uint)hdr_bytes.Length);
fs.Write(kernel, 0, kernel.Length);
write_padding(ref fs, hdr.page_size, (uint)kernel.Length);
fs.Write(ramdisk, 0, ramdisk.Length);
write_padding(ref fs, hdr.page_size, (uint)ramdisk.Length);
if (second != null)
{
fs.Write(second, 0, second.Length);
write_padding(ref fs, hdr.page_size, (uint)second.Length);
}
sz = fs.Length;
fs.Dispose();
}
catch { return false; }
return true;
}
public static int Pack(WorkData Data)
{
string mKernel, mRamdisk, mSecondStage, mtkKernelHdr, mtkRamdiskHdr;
mKernel = mRamdisk = mSecondStage = mtkKernelHdr = mtkRamdiskHdr = string.Empty;
ImgHeader hdr;
//Для запаковки папки обязательно нужен исходный заголовок
if (Data.Mode == WorkMode.Pack)
{
string mHeader = Path.Combine(Data.pack_dir, "image.hdr");
mKernel = Path.Combine(Data.pack_dir, "zImage");
mRamdisk = Path.Combine(Data.pack_dir, "ram_disk.gz");
mSecondStage = Path.Combine(Data.pack_dir, "second.bin");
mtkKernelHdr = mKernel + ".mtkhdr";
mtkRamdiskHdr = mRamdisk + ".mtkhdr";
Console.WriteLine("Packing image...");
Console.WriteLine("Source folder: {0}", Data.pack_dir);
Console.WriteLine("Destination file: {0}", Data.output_path);
Console.Write("Looking for template header 'image.hdr'... ");
if (File.Exists(mHeader))
{
ConsoleEx.WriteLine(ConsoleColor.Green, "OK");
hdr = ImgHeader.FromBytes(File.ReadAllBytes(mHeader));
}
else
{
ConsoleEx.WriteLine(ConsoleColor.Red, "Error: file not found");
return 1;
}
}
else
hdr = new ImgHeader();
//Обновляем данные в заголовке согласно новым входным
UpdateHeader(ref hdr, Data);
Console.Write("Looking for source images... ");
//Пытаемся загрузить образы либо из файла пакета, либо из входного. Приоритет на входной.
byte[] bKernel = LoadImage(mKernel, Data.kernel_data);
byte[] bRamdisk = LoadImage(mRamdisk, Data.ramdisk_data);
byte[] bSecondStage = LoadImage(mSecondStage, Data.second_data);
//Если что-то не загрузилось ни из одного источника - возвращаеам ошибку
if (bKernel == null || bRamdisk == null)
{
ConsoleEx.WriteLine(ConsoleColor.Red, "FAILED");
return -2; // NO SOURCE IMAGES FOUND
}
ConsoleEx.WriteLine(ConsoleColor.Green, "OK.");
//Проверяем, есть ли хоть один источник заголовка МТК.
//Если хоть что-то есть, то считаем что наш образ полностью для МТК-устройства
//Также если установлен флаг mtk_force, то необходимо создавать заголовки вручную
if (File.Exists(mtkKernelHdr) || File.Exists(mtkRamdiskHdr)
|| Data.mtk_kernel_hdr_data != null || Data.mtk_ramdisk_hdr_data != null
|| Data.mtk_force)
{
ConsoleEx.WriteLine(ConsoleColor.White, "Image will be in MTK format!");
byte[] bMtkKernelHdr = LoadImage(mtkKernelHdr, Data.mtk_kernel_hdr_data);
byte[] bMtkRamdiskHdr = LoadImage(mtkRamdiskHdr, Data.mtk_ramdisk_hdr_data);
if (bMtkKernelHdr == null)
ConsoleEx.WriteLine(ConsoleColor.Yellow, "Template of MTK kernel header not found. Will be created a fresh one.");
if (bMtkRamdiskHdr == null)
ConsoleEx.WriteLine(ConsoleColor.Yellow, "Template of MTK ramdisk header not found. Will be created a fresh one.");
//если что-то не загрузили, то необходимо создать новый экземпляр
mtkSectionHeader smtkKernelHdr = (bMtkKernelHdr != null)
? mtkSectionHeader.FromBytes(bMtkKernelHdr)
: new mtkSectionHeader("KERNEL");
mtkSectionHeader smtkRamdiskHdr = (bMtkRamdiskHdr != null)
? mtkSectionHeader.FromBytes(bMtkRamdiskHdr)
: new mtkSectionHeader("ROOTFS");
//Если установлено имя секции рутфс, меняем
if (Data.mtk_ramdisk_name != null)
{
Console.WriteLine("Defining name of ramdisk as '{0}'", Data.mtk_ramdisk_name);
smtkRamdiskHdr.SetName(Data.mtk_ramdisk_name);
}
Console.WriteLine("Generating MTK sections...");
//создаем новые МТК-секции
bKernel = mtkSectionWriter.GetBytes(smtkKernelHdr, bKernel);
bRamdisk = mtkSectionWriter.GetBytes(smtkRamdiskHdr, bRamdisk);
}
//Показываем таблицу шаблона
Console.WriteLine("Creating image... ");
hdr.kernel_size = (uint)bKernel.Length;
hdr.ramdisk_size = (uint)bRamdisk.Length;
hdr.second_size = bSecondStage == null ? (uint)0 : (uint)bSecondStage.Length;
hdr.PrintInfo();
//Записываем итоговый
long sz = 0;
if (ImgWriter.Write(hdr, bKernel, bRamdisk, bSecondStage, Data.output_path, ref sz))
{
Console.WriteLine();
ConsoleEx.Write(ConsoleColor.Green, "Done!");
if (sz > 0x600000)
ConsoleEx.WriteLine(ConsoleColor.Yellow, " But be careful! Size of compiled image is more than 6MB!", sz);
return 0; //OK
}
ConsoleEx.WriteLine(ConsoleColor.Red, "Oops. Something is wrong.");
return -1; //WRITING FAILED
}
public static void UpdateHeader(ref ImgHeader hdr, WorkData Data)
{
uint OffsetBase = hdr.GetConjecturalBase();
if (Data.base_offset != null)
{
hdr.SetBase(OffsetBase, Data.base_offset.Value);
OffsetBase = Data.base_offset.Value;
}
if (Data.kernel_offset != null)
hdr.SetAddress(ref hdr.kernel_addr, OffsetBase, Data.kernel_offset.Value);
if (Data.ramdisk_offset != null)
hdr.SetAddress(ref hdr.ramdisk_addr, OffsetBase, Data.ramdisk_offset.Value);
if (Data.second_offset != null)
hdr.SetAddress(ref hdr.second_addr, OffsetBase, Data.second_offset.Value);
if (Data.tags_offset != null)
hdr.SetAddress(ref hdr.tags_addr, OffsetBase, Data.tags_offset.Value);
if (Data.pagesize != null)
hdr.page_size = Data.pagesize.Value;
if (Data.cmdline != null)
hdr.SetCmdLine(Data.cmdline);
if (Data.board != null)
hdr.SetName(Data.board);
}
