-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
c030528
commit 7ba20c5
Showing
15 changed files
with
568 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,287 @@ | ||
| using NyaFs; | ||
| using NyaFs.Processor; | ||
| using NyaFs.Processor.Scripting; | ||
| using NyaFsBmu.Loader; | ||
| using NyaFsBmu.Types; | ||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.IO; | ||
| using System.Text; | ||
| using static System.Net.Mime.MediaTypeNames; | ||
|
|
||
| namespace NyaFsBmu.Commands | ||
| { | ||
| class LoadBmu : ScriptStepGenerator | ||
| { | ||
| public LoadBmu() : base("loadbmu") | ||
| { | ||
| AddConfig(new ScriptArgsConfig(0, new ScriptArgsParam[] { | ||
| new NyaFs.Processor.Scripting.Params.FsPathScriptArgsParam() | ||
| })); | ||
| } | ||
|
|
||
| public override ScriptStep Get(ScriptArgs Args) | ||
| { | ||
| return new LoadBmuScriptStep(Args.RawArgs[0]); | ||
| } | ||
|
|
||
| public class LoadBmuScriptStep : ScriptStep | ||
| { | ||
| private string Path; | ||
|
|
||
| public LoadBmuScriptStep(string Path) : base("loadbmu") | ||
| { | ||
| this.Path = Path; | ||
| } | ||
|
|
||
| NyaFs.ImageFormat.Types.CPU FindExistArch(NyaFs.ImageFormat.BaseImageBlob Blob) | ||
| { | ||
| NyaFs.ImageFormat.Types.CPU arch = NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID; | ||
| if (Blob.IsProvidedKernel) | ||
| { | ||
| var K = Blob.GetKernel(); | ||
| if (K.Info.Architecture != NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| arch = K.Info.Architecture; | ||
| } | ||
| if (Blob.IsProvidedFs) | ||
| { | ||
| var Fs = Blob.GetFilesystem(); | ||
| if (Fs.Info.Architecture != NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| { | ||
| if (arch != NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| { | ||
| // Is detected os info differs to kernel | ||
| if (arch != Fs.Info.Architecture) | ||
| return NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID; | ||
| } | ||
| else | ||
| arch = Fs.Info.Architecture; | ||
| } | ||
| } | ||
| if (Blob.IsProvidedDTB) | ||
| { | ||
| var Dtb = Blob.GetDevTree(); | ||
| if (Dtb.Info.Architecture != NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| { | ||
| if (arch != NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| { | ||
| // Is detected os info differs to dtb | ||
| if (arch != Dtb.Info.Architecture) | ||
| return NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID; | ||
| } | ||
| else | ||
| arch = Dtb.Info.Architecture; | ||
| } | ||
| } | ||
|
|
||
| return arch; | ||
| } | ||
|
|
||
| NyaFs.ImageFormat.Types.OS FindExistOS(NyaFs.ImageFormat.BaseImageBlob Blob) | ||
| { | ||
| NyaFs.ImageFormat.Types.OS os = NyaFs.ImageFormat.Types.OS.IH_OS_INVALID; | ||
| if (Blob.IsProvidedKernel) | ||
| { | ||
| var K = Blob.GetKernel(); | ||
| if (K.Info.OperatingSystem != NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| os = K.Info.OperatingSystem; | ||
| } | ||
| if (Blob.IsProvidedFs) | ||
| { | ||
| var Fs = Blob.GetFilesystem(); | ||
| if (Fs.Info.OperatingSystem != NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| { | ||
| if (os != NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| { | ||
| // Is FS os info differs to kernel | ||
| if (os != Fs.Info.OperatingSystem) | ||
| return NyaFs.ImageFormat.Types.OS.IH_OS_INVALID; | ||
| } | ||
| else | ||
| os = Fs.Info.OperatingSystem; | ||
| } | ||
| } | ||
|
|
||
| return os; | ||
| } | ||
|
|
||
| private void AssumeAutoKernelParams(NyaFs.ImageFormat.Elements.Kernel.LinuxKernel Kernel) | ||
| { | ||
| /*if (Kernel.Info.Architecture == NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| { | ||
| var Type = NyaFs.ImageFormat.Helper.KernelHelper.DetectImageArch(Kernel.Image); | ||
| if (Type != NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| { | ||
| Log.Write(0, $"Assumed kernel architecture: {NyaFs.ImageFormat.Helper.FitHelper.GetCPUArchitecture(Type)}"); | ||
| Kernel.Info.Architecture = Type; | ||
| } | ||
| } | ||
| if (Kernel.Info.OperatingSystem == NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| { | ||
| var Os = Helper.KernelHelper.DetectImageOs(Kernel.Image); | ||
| if (Os != NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| { | ||
| Log.Write(0, $"Assumed kernel OS: {NyaFs.ImageFormat.Helper.FitHelper.GetOperatingSystem(Os)}"); | ||
| Kernel.Info.OperatingSystem = Os; | ||
| } | ||
| }*/ | ||
| } | ||
|
|
||
| private void AssumeAutoImageParams(ImageProcessor Processor) | ||
| { | ||
| var Blob = Processor.GetBlob(); | ||
| var Arch = FindExistArch(Blob); | ||
| if (Arch != NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| { | ||
| var K = Processor.GetKernel(); | ||
| if (K != null) | ||
| { | ||
| if (K.Info.Architecture == NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| K.Info.Architecture = Arch; | ||
| } | ||
| var Fs = Processor.GetFs(); | ||
| if (Fs != null) | ||
| { | ||
| if (Fs.Info.Architecture == NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| Fs.Info.Architecture = Arch; | ||
| } | ||
| var Dtb = Processor.GetDevTree(); | ||
| if (Dtb != null) | ||
| { | ||
| if (Dtb.Info.Architecture == NyaFs.ImageFormat.Types.CPU.IH_ARCH_INVALID) | ||
| Dtb.Info.Architecture = Arch; | ||
| } | ||
| } | ||
| var Os = FindExistOS(Blob); | ||
| if (Os != NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| { | ||
| var K = Processor.GetKernel(); | ||
| if (K != null) | ||
| { | ||
| if (K.Info.OperatingSystem == NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| K.Info.OperatingSystem = Os; | ||
| } | ||
| var Fs = Processor.GetFs(); | ||
| if (Fs != null) | ||
| { | ||
| if (Fs.Info.OperatingSystem == NyaFs.ImageFormat.Types.OS.IH_OS_INVALID) | ||
| Fs.Info.OperatingSystem = Os; | ||
| } | ||
| } | ||
| } | ||
|
|
||
| public override ScriptStepResult Exec(ImageProcessor Processor) | ||
| { | ||
| // Проверим наличие локального файла, содержимое которого надо загрузить в ФС | ||
| if (!System.IO.File.Exists(Path)) | ||
| return new ScriptStepResult(ScriptStepStatus.Error, $"{Path} not found!"); | ||
|
|
||
| var Data = File.ReadAllBytes(Path); | ||
| var Image = new BmuImage(Data); | ||
|
|
||
| if (Image.Correct) | ||
| { | ||
| var FsRes = ReadFs(Processor, Image); | ||
| if (FsRes.Status == ScriptStepStatus.Error) | ||
| return FsRes; | ||
|
|
||
| var KernelRes = ReadKernel(Processor, Image); | ||
| if (FsRes.Status == ScriptStepStatus.Error) | ||
| return KernelRes; | ||
|
|
||
| var DevtreeRes = ReadDtb(Processor, Image); | ||
| if (FsRes.Status == ScriptStepStatus.Error) | ||
| return DevtreeRes; | ||
|
|
||
| return new ScriptStepResult(ScriptStepStatus.Ok, $"BMU is loaded!"); | ||
| } | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Error, $"BMU is invalid image!"); | ||
| } | ||
|
|
||
| private ScriptStepResult ReadFs(ImageProcessor Processor, BmuImage Data) | ||
| { | ||
| var OldLoaded = Processor.GetFs()?.Loaded ?? false; | ||
| var Fs = new NyaFs.ImageFormat.Elements.Fs.LinuxFilesystem(); | ||
| var FsReader = new NyaFsBmu.Loader.BmuFsReader(Data); | ||
|
|
||
| FsReader.ReadToFs(Fs); | ||
| if (Fs.Loaded) | ||
| { | ||
| Processor.SetFs(Fs); | ||
| AssumeAutoImageParams(Processor); | ||
| //if (Fs.FilesystemType != ImageFormat.Types.FsType.Cpio) | ||
| //{ | ||
| // Fs.FilesystemType = ImageFormat.Types.FsType.Cpio; | ||
| // Log.Warning(0, "Filesystem is switched to cpio as only supported for packing."); | ||
| // if (Fs.Info.Compression == ImageFormat.Types.CompressionType.IH_COMP_NONE) | ||
| // { | ||
| // Fs.Info.Compression = ImageFormat.Types.CompressionType.IH_COMP_GZIP; | ||
| // Log.Warning(0, "Compression is switched to gzip."); | ||
| // } | ||
| //} | ||
| NyaFs.ImageFormat.Helper.LogHelper.RamfsInfo(Fs); | ||
|
|
||
| if (OldLoaded) | ||
| return new ScriptStepResult(ScriptStepStatus.Warning, $"BMU image is loaded as filesystem! Old filesystem is replaced by this."); | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Ok, $"BMU is loaded as filesystem!"); | ||
| } | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Error, $"Unknown fs format!"); | ||
| } | ||
|
|
||
| private ScriptStepResult ReadDtb(ImageProcessor Processor, BmuImage Data) | ||
| { | ||
| var OldLoaded = Processor.GetDevTree()?.Loaded ?? false; | ||
| var Dtb = new NyaFs.ImageFormat.Elements.Dtb.DeviceTree(); | ||
| var Reader = new NyaFsBmu.Loader.BmuDevtreeReader(Data); | ||
| if (Reader != null) | ||
| { | ||
| Reader.ReadToDevTree(Dtb); | ||
| if (Dtb.Loaded) | ||
| { | ||
| Processor.SetDeviceTree(Dtb); | ||
| AssumeAutoImageParams(Processor); | ||
| NyaFs.ImageFormat.Helper.LogHelper.DevtreeInfo(Dtb); | ||
| return new ScriptStepResult(ScriptStepStatus.Ok, $"Device tree is loaded from BMU file!"); | ||
| } | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Error, $"Device tree is not loaded!"); | ||
| } | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Error, $"Unknown devtree format!"); | ||
| } | ||
|
|
||
| private ScriptStepResult ReadKernel(ImageProcessor Processor, BmuImage Data) | ||
| { | ||
| var OldLoaded = Processor.GetKernel()?.Loaded ?? false; | ||
| var Kernel = new NyaFs.ImageFormat.Elements.Kernel.LinuxKernel(); | ||
| var Reader = new NyaFsBmu.Loader.BmuKernelReader(Data); | ||
| if (Reader != null) | ||
| { | ||
| Reader.ReadToKernel(Kernel); | ||
| if (Kernel.Loaded) | ||
| { | ||
| Processor.SetKernel(Kernel); | ||
| AssumeAutoKernelParams(Kernel); | ||
| AssumeAutoImageParams(Processor); | ||
| NyaFs.ImageFormat.Helper.LogHelper.KernelInfo(Kernel); | ||
|
|
||
| if (OldLoaded) | ||
| return new ScriptStepResult(ScriptStepStatus.Warning, $"BMU kernel image is loaded as kernel! Old kernel is replaced by this."); | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Ok, $"BMU kernel image is loaded as kernel!"); | ||
| } | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Error, $"Kernel image file is not loaded!"); | ||
| } | ||
| else | ||
| return new ScriptStepResult(ScriptStepStatus.Error, $"Unknown kernel format!"); | ||
| } | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,48 @@ | ||
| using NyaFs.Processor.Scripting; | ||
| using NyaFs.Processor; | ||
| using NyaFsBmu.Types; | ||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.IO; | ||
| using System.IO.Compression; | ||
| using System.Text; | ||
| using NyaFs.ImageFormat.Elements.Kernel; | ||
| using NyaFs; | ||
| using NyaFs.FlattenedDeviceTree; | ||
| using NyaFs.ImageFormat.Elements.Dtb; | ||
|
|
||
| namespace NyaFsBmu.Loader | ||
| { | ||
| class BmuDevtreeReader : NyaFs.ImageFormat.Elements.Dtb.Reader.Reader | ||
| { | ||
| BmuImage Image; | ||
|
|
||
| public BmuDevtreeReader(BmuImage Data) | ||
| { | ||
| Image = Data; | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Читаем в дерево устройств из внешнего источника | ||
| /// </summary> | ||
| /// <param name="Dst"></param> | ||
| public override void ReadToDevTree(DeviceTree Dst) | ||
| { | ||
| var Data = Image.GetImageByType(BmuImageType.DEVTREE); | ||
| if (Data != null) | ||
| { | ||
| var Dtb = new NyaFs.FlattenedDeviceTree.Reader.FDTReader(Data); | ||
| if (Dtb.Correct) | ||
| { | ||
| Dst.DevTree = Dtb.Read(); | ||
| Dst.Info.Type = NyaFs.ImageFormat.Types.ImageType.IH_TYPE_FLATDT; | ||
| Dst.Info.Compression = NyaFs.ImageFormat.Types.CompressionType.IH_COMP_NONE; | ||
| } | ||
| else | ||
| Log.Error(0, $"Invalid dtb header in BMU"); | ||
| } | ||
| else | ||
| Log.Error(0, $"No dtb in BMU"); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.