#include "stackinfo.h"
#include "debugger.h"
#include "memory.h"
#include "disasm_helper.h"
#include "BeaEngine\BeaEngine.h"
#include "addrinfo.h"
#include "_exports.h"

bool stackcommentget(uint addr, STACK_COMMENT* comment)
{
    uint data=0;
    memset(comment, 0, sizeof(STACK_COMMENT));
    memread(fdProcessInfo->hProcess, (const void*)addr, &data, sizeof(uint), 0);
    if(memisvalidreadptr(fdProcessInfo->hProcess, data)) //valid memory
    {
        uint size=0;
        uint base=memfindbaseaddr(fdProcessInfo->hProcess, data, &size);
        uint readStart=data-16*4;
        if(readStart<base)
            readStart=base;
        unsigned char disasmData[256];
        memread(fdProcessInfo->hProcess, (const void*)readStart, disasmData, sizeof(disasmData), 0);
        unsigned int prev=disasmback(disasmData, 0, sizeof(disasmData), data-readStart, 1);
        uint previousInstr=readStart+prev;
        DISASM disasm;
        disasm.Options=NoformatNumeral;
#ifdef _WIN64
        disasm.Archi=64;
#endif // _WIN64
        disasm.VirtualAddr=previousInstr;
        disasm.EIP=(UIntPtr)(disasmData+prev);
        int len=Disasm(&disasm);
        static char instruction[INSTRUCT_LENGTH]="";
        if(len!=UNKNOWN_OPCODE && disasm.Instruction.BranchType==CallType) //call
        {
            DISASM_INSTR instr;
            disasmget((unsigned char*)disasm.EIP, previousInstr, &instr);
            
            char label[MAX_LABEL_SIZE]="";
            ADDRINFO addrinfo;
            addrinfo.flags=flaglabel;
            if(_dbg_addrinfoget(data, SEG_DEFAULT, &addrinfo))
                strcpy(label, addrinfo.label);
            char module[MAX_MODULE_SIZE]="";
            modnamefromaddr(data, module, false);
            char returnToAddr[MAX_COMMENT_SIZE]="";
            if(*module)
                sprintf(returnToAddr, "%s.", module);
            if(!*label)
                sprintf(label, fhex, data);
            strcat(returnToAddr, label);

            data=instr.arg[0].value;
            *label=0;
            addrinfo.flags=flaglabel;
            if(_dbg_addrinfoget(data, SEG_DEFAULT, &addrinfo))
                strcpy(label, addrinfo.label);
            *module=0;
            modnamefromaddr(data, module, false);
            char returnFromAddr[MAX_COMMENT_SIZE]="";
            if(*module)
                sprintf(returnFromAddr, "%s.", module);
            if(!*label)
                sprintf(label, fhex, data);
            strcat(returnFromAddr, label);

            sprintf(comment->comment, "return to %s from %s", returnToAddr, returnFromAddr);
            strcpy(comment->color, "#ff0000");
            return true;
        }
        else
        {
            char label[MAX_LABEL_SIZE]="";
            ADDRINFO addrinfo;
            addrinfo.flags=flaglabel;
            if(_dbg_addrinfoget(data, SEG_DEFAULT, &addrinfo))
                strcpy(label, addrinfo.label);
            char module[MAX_MODULE_SIZE]="";
            modnamefromaddr(data, module, false);
            char addrInfo[MAX_COMMENT_SIZE]="";
            if(*module) //module
            {
                if(*label) //+label
                    sprintf(comment->comment, "%s.%s", module, label);
                else //module only
                    sprintf(comment->comment, "%s."fhex, module, data);
                return true;
            }
            else if(*label) //label only
            {
                sprintf(comment->comment, "<%s>", label);
                return true;
            }
        }
    }
    return false;
}