/**
 * Copyright (c) 2026 Alexey Gavrilov <alexey.gavrilov@mail.com>
 *
 * This file is part of ObjectiveOS.
 *
 * ObjectiveOS 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.
 *
 * ObjectiveOS 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
 * ObjectiveOS. If not, see <https://www.gnu.org/licenses/>. 
 */

// Tasks:
//         1) Implement msr write and save functions
//         2) Ia32ApicBase field

#include <MemoryOperations.h>

#ifndef MSR_H
#define MSR_H

// defines for Ia32ApicBase

#define BSP_FLAG 0x100
#define APIC_ENABLE_FLAG 0x800
#define BASE_FIELD_MASK_LOW 0xfffff000
#define UINT64_HIGH 0xffffffff00000000

//------------------------

class MsrCommand
{
    public:
        virtual uint32 getCommand() = 0;
        virtual void decode(uint64 rdmsrResult) = 0;
        virtual uint64 getMsr() = 0;
};

class Ia32ApicBase : public MsrCommand
{
    private:
        // Get through Cpuinfo instance
        uint8 maxPhyAddr;
        bool bspFlag, enableFlag;
        uint64 baseField;

    public:
        Ia32ApicBase();
        virtual uint32 getCommand() override final;
        virtual void decode(uint64 rdmrResult) override final;
        virtual uint64 getMsr() override final;
        void setBspFlag(bool flag);
        void setEnableFlag(bool flag);
        void setBaseField(uint64 fieldAddr);
};

class Msr
{
    private:
        bool msrSupport;
    public:
        Msr();
        // Return true if MSR not supported (CPUID.01H:EDX[5] = 0)
        bool rdmsr(MsrCommand *cmd);
        // Return value same as rdmsr method
        bool wrmsr(MsrCommand *cmd);
};

#endif