//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//

//----------------------------------------------------------
// nearDiffer.h - differ that handles code that is very similar
//----------------------------------------------------------
#ifndef _nearDiffer
#define _nearDiffer

#include "methodcontext.h"
#include "compileresult.h"

class NearDiffer
{
public:

    NearDiffer(const char *targetArch, bool useCorDisTools)
        : TargetArchitecture(targetArch)
        , UseCoreDisTools(useCorDisTools)
#ifdef USE_COREDISTOOLS
        , corAsmDiff(nullptr)
#endif // USE_COREDISTOOLS
    {
    }

    ~NearDiffer();

    void InitAsmDiff();

    bool compare(MethodContext *mc, CompileResult *cr1,CompileResult *cr2);

    const char* TargetArchitecture;
    const bool UseCoreDisTools;

private:

    void DumpCodeBlock(unsigned char *block, ULONG blocksize, void *originalAddr);

    bool compareCodeSection(
        MethodContext *mc,
        CompileResult *cr1,
        CompileResult *cr2,
        unsigned char *block1,
        ULONG blocksize1,
        unsigned char *datablock1,
        ULONG datablockSize1,
        void *originalBlock1,
        void *originalDataBlock1,
        void *otherCodeBlock1,
        ULONG otherCodeBlockSize1,
        unsigned char *block2,
        ULONG blocksize2,
        unsigned char *datablock2,
        ULONG datablockSize2,
        void *originalBlock2,
        void *originalDataBlock2,
        void *otherCodeBlock2,
        ULONG otherCodeBlockSize2);

    bool compareReadOnlyDataBlock(MethodContext *mc, CompileResult *cr1, CompileResult *cr2,
        unsigned char *block1, ULONG blocksize1, void *originalDataBlock1,
        unsigned char *block2, ULONG blocksize2, void *originalDataBlock2);
    bool compareEHInfo(MethodContext *mc, CompileResult *cr1, CompileResult *cr2);
    bool compareGCInfo(MethodContext *mc, CompileResult *cr1, CompileResult *cr2);
    bool compareVars(MethodContext *mc, CompileResult *cr1, CompileResult *cr2);
    bool compareBoundaries(MethodContext *mc, CompileResult *cr1, CompileResult *cr2);

    static bool compareOffsets(const void *payload,
                               size_t blockOffset,
                               size_t instrLen,
                               uint64_t offset1,
                               uint64_t offset2);

#ifdef USE_COREDISTOOLS
    CorAsmDiff *corAsmDiff;
#endif // USE_COREDISTOOLS

#ifdef USE_MSVCDIS
    DIS* GetMsVcDis();
#endif // USE_MSVCDIS

};

#endif // _nearDiffer