Ebr.cs

using System;
using System.Collections.Generic;
using MtkClient.Library.Utils;

namespace MtkClient.Library
{
    public class EbrPartition
    {
        public byte Status { get; set; }
        public byte Type { get; set; }
        public uint FirstLba { get; set; }
        public uint SectorCount { get; set; }
        public uint NextEbrLba { get; set; }

        public EbrPartition(byte[] data, int offset = 0)
        {
            if (data.Length < offset + 16) return;
            Status = data[offset];
            Type = data[offset + 4];
            FirstLba = BitConverter.ToUInt32(data, offset + 8);
            SectorCount = BitConverter.ToUInt32(data, offset + 12);
        }
    }

    public class Ebr : LogBase
    {
        public List<EbrPartition> Partitions { get; } = new List<EbrPartition>();
        public const ushort EBR_SIGNATURE = 0xAA55;

        public Ebr(byte[] data, uint startLba = 0, LogLevel logLevel = LogLevel.Info)
        {
            InitLogger(logLevel);
            Parse(data, startLba);
        }

        private void Parse(byte[] data, uint startLba)
        {
            if (data == null || data.Length < 512) return;

            uint currentLba = startLba;

            while (true)
            {
                int offset = (int)(currentLba * 512);
                if (offset + 512 > data.Length) break;

                var sig = BitConverter.ToUInt16(data, offset + 510);
                if (sig != EBR_SIGNATURE) break;

                var partition = new EbrPartition(data, offset + 446);
                if (partition.Type == 0) break;

                partition.FirstLba += currentLba;
                Partitions.Add(partition);

                var nextEntry = new EbrPartition(data, offset + 462);
                if (nextEntry.Type == 0 || nextEntry.FirstLba == 0) break;

                nextEntry.NextEbrLba = startLba + nextEntry.FirstLba;
                currentLba = nextEntry.NextEbrLba;
            }

            Info($"EBR parsed: {Partitions.Count} logical partitions");
        }
    }
}