Meta.cs

using System;
using System.Text;
using MtkClient.Library.Config;
using MtkClient.Library.Connection;
using MtkClient.Library.State;
using MtkClient.Library.Utils;

namespace MtkClient.Library
{
    public class Meta : LogBase
    {
        private readonly MtkConfig _config;
        private readonly Port _port;
        private readonly MtkStateManager _stateManager;

        public const byte META_MODE = 0x01;
        public const byte FACTORY_MODE = 0x02;
        public const byte ATE_MODE = 0x05;

        // NVRAM commands
        public const byte CMD_NVRAM_READ = 0x30;
        public const byte CMD_NVRAM_WRITE = 0x31;
        public const byte CMD_NVRAM_RESET = 0x32;

        // IMEI commands
        public const byte CMD_READ_IMEI = 0x40;
        public const byte CMD_WRITE_IMEI = 0x41;

        public Meta(MtkConfig config, Port port, MtkStateManager stateManager = null,
            LogLevel logLevel = LogLevel.Info)
        {
            InitLogger(logLevel);
            _config = config;
            _port = port;
            _stateManager = stateManager;
        }

        public bool DetectMetaMode()
        {
            try
            {
                // Check if device is in META mode by probing
                _port.Cdc.Flush();
                var response = _port.UsbRead(4, 500);
                if (response != null && response.Length >= 4)
                {
                    string magic = Encoding.ASCII.GetString(response, 0, 4);
                    if (magic == "META" || magic == "FACT" || magic == "ATE\0")
                    {
                        Info($"META mode detected: {magic.TrimEnd('\0')}");
                        _stateManager?.TransitionTo(DeviceState.MetaMode, "META detected");
                        return true;
                    }
                }
            }
            catch { }
            return false;
        }

        public bool EnterMetaMode()
        {
            Info("Entering META mode...");
            try
            {
                _port.UsbWrite(new byte[] { Preloader.CMD_SET_META_BOOT, META_MODE });
                var ack = _port.RWord();
                if (ack == Preloader.ACK)
                {
                    Info("META mode entered successfully");
                    _stateManager?.TransitionTo(DeviceState.MetaMode, "META mode");
                    return true;
                }
                Error("Failed to enter META mode");
                return false;
            }
            catch (Exception ex)
            {
                Error($"META mode error: {ex.Message}");
                return false;
            }
        }

        public bool EnterFactoryMode()
        {
            Info("Entering Factory mode...");
            try
            {
                _port.UsbWrite(new byte[] { Preloader.CMD_SET_META_BOOT, FACTORY_MODE });
                var ack = _port.RWord();
                if (ack == Preloader.ACK)
                {
                    _stateManager?.TransitionTo(DeviceState.FactoryMode, "Factory mode");
                    return true;
                }
                return false;
            }
            catch (Exception ex)
            {
                Error($"Factory mode error: {ex.Message}");
                return false;
            }
        }

        public bool EnterAteMode()
        {
            Info("Entering ATE mode...");
            try
            {
                _port.UsbWrite(new byte[] { Preloader.CMD_SET_META_BOOT, ATE_MODE });
                var ack = _port.RWord();
                if (ack == Preloader.ACK)
                {
                    _stateManager?.TransitionTo(DeviceState.AteMode, "ATE mode");
                    return true;
                }
                return false;
            }
            catch (Exception ex)
            {
                Error($"ATE mode error: {ex.Message}");
                return false;
            }
        }

        public byte[] ReadNvram(int fileId, int offset = 0, int length = 0)
        {
            try
            {
                _port.UsbWrite(new byte[] { CMD_NVRAM_READ });
                _port.UsbWrite(MtkUtils.Pack32BE((uint)fileId));
                _port.UsbWrite(MtkUtils.Pack32BE((uint)offset));
                _port.UsbWrite(MtkUtils.Pack32BE((uint)length));

                var ack = _port.RWord();
                if (ack != Preloader.ACK) return null;

                var sizeData = _port.RdWord();
                if (sizeData == null) return null;

                var data = _port.UsbRead((int)sizeData.Value);
                Debug($"NVRAM read: fileId={fileId}, offset=0x{offset:X}, len={data?.Length ?? 0}");
                return data;
            }
            catch (Exception ex)
            {
                Error($"NVRAM read failed: {ex.Message}");
                return null;
            }
        }

        public bool WriteNvram(int fileId, byte[] data, int offset = 0)
        {
            try
            {
                _port.UsbWrite(new byte[] { CMD_NVRAM_WRITE });
                _port.UsbWrite(MtkUtils.Pack32BE((uint)fileId));
                _port.UsbWrite(MtkUtils.Pack32BE((uint)offset));
                _port.UsbWrite(MtkUtils.Pack32BE((uint)data.Length));
                _port.UsbWrite(data);

                var ack = _port.RWord();
                if (ack == Preloader.ACK)
                {
                    Debug($"NVRAM written: fileId={fileId}, offset=0x{offset:X}, len={data.Length}");
                    return true;
                }
                return false;
            }
            catch (Exception ex)
            {
                Error($"NVRAM write failed: {ex.Message}");
                return false;
            }
        }

        public string ReadImei(int slot = 0)
        {
            try
            {
                _port.UsbWrite(new byte[] { CMD_READ_IMEI });
                _port.UsbWrite(MtkUtils.Pack32BE((uint)slot));

                var ack = _port.RWord();
                if (ack != Preloader.ACK) return null;

                var lenData = _port.RdWord();
                if (lenData == null) return null;

                var imeiBytes = _port.UsbRead((int)lenData.Value);
                if (imeiBytes != null)
                {
                    string imei = Encoding.ASCII.GetString(imeiBytes).TrimEnd('\0');
                    Info($"IMEI slot {slot}: {imei}");
                    return imei;
                }
                return null;
            }
            catch (Exception ex)
            {
                Error($"Read IMEI failed: {ex.Message}");
                return null;
            }
        }

        public bool WriteImei(string imei, int slot = 0)
        {
            if (string.IsNullOrEmpty(imei) || imei.Length != 15)
            {
                Error("Invalid IMEI format (must be 15 digits)");
                return false;
            }

            try
            {
                var imeiBytes = Encoding.ASCII.GetBytes(imei);

                _port.UsbWrite(new byte[] { CMD_WRITE_IMEI });
                _port.UsbWrite(MtkUtils.Pack32BE((uint)slot));
                _port.UsbWrite(MtkUtils.Pack32BE((uint)imeiBytes.Length));
                _port.UsbWrite(imeiBytes);

                var ack = _port.RWord();
                if (ack == Preloader.ACK)
                {
                    Info($"IMEI slot {slot} written: {imei}");
                    return true;
                }
                return false;
            }
            catch (Exception ex)
            {
                Error($"Write IMEI failed: {ex.Message}");
                return false;
            }
        }

        public bool ResetNvram()
        {
            Info("Resetting NVRAM...");
            try
            {
                _port.UsbWrite(new byte[] { CMD_NVRAM_RESET });
                var ack = _port.RWord();
                if (ack == Preloader.ACK)
                {
                    Info("NVRAM reset successful");
                    return true;
                }
                Error("NVRAM reset failed");
                return false;
            }
            catch (Exception ex)
            {
                Error($"NVRAM reset failed: {ex.Message}");
                return false;
            }
        }
    }
}