Trace record

#654
2016-05-25 11:29:13 +00:00 · 2016-05-25 11:29:13 +00:00 · 09270e501a
parent e6502e236e
commit 09270e501a
1 changed files with 89 additions and 0 deletions
--- a/src/dbg/TraceRecord.h
+++ b/src/dbg/TraceRecord.h
@ -0,0 +1,89 @@
+#ifndef TRACERECORD_H
+#define TRACERECORD_H
+#include "_global.h"
+#include "_dbgfunctions.h"
+
+class TraceRecordManager
+{
+public:
+    enum TraceRecordByteType{
+        InstructionBody = 0,
+        InstructionHeading = 1,
+        InstructionTailing = 2,
+        InstructionOverlapped = 3, // The byte was executed with differing instruction base addresses
+        DataByte,  // This and the following is not implemented yet.
+        DataWord,
+        DataDWord,
+        DataQWord,
+        DataFloat,
+        DataDouble,
+        DataLongDouble,
+        DataXMM,
+        DataYMM,
+        DataMMX,
+        DataMixed, //the byte is accessed in multiple ways
+        InstructionDataMixed //the byte is both executed and written
+    };
+
+    /***************************************************************
+     * Trace record data layout
+     * TraceRecordNonoe: disable trace record
+     * TraceRecordBitExec: single-bit, executed.
+     * TraceRecordByteWithExecTypeAndCounter: 8-bit, YYXXXXXX YY:=TraceRecordByteType_2bit, XXXXXX:=Hit count(6bit)
+     * TraceRecordWordWithExecTypeAndCounter: 16-bit, YYXXXXXX XXXXXXXX YY:=TraceRecordByteType_2bit, XX:=Hit count(14bit)
+     * Other: reserved for future expanding
+     **************************************************************/
+    enum TraceRecordType{
+        TraceRecordNone,
+        TraceRecordBitExec,
+        TraceRecordByteWithExecTypeAndCounter,
+        TraceRecordWordWithExecTypeAndCounter
+    };
+
+    TraceRecordManager();
+    ~TraceRecordManager();
+    void clear();
+
+    bool setTraceRecordType(duint pageAddress, TraceRecordType type);
+    TraceRecordType getTraceRecordType(duint pageAddress);
+
+    void TraceExecute(duint address, duint size);
+    //void TraceAccess(duint address, unsigned char size, TraceRecordByteType accessType);
+
+    unsigned int getHitCount(duint address);
+    TraceRecordByteType getByteType(duint address);
+    void increaseInstructionCounter();
+
+    void saveToDb(JSON root);
+    void loadFromDb(JSON root);
+private:
+    enum TraceRecordByteType_2bit{
+        _InstructionBody = 0,
+        _InstructionHeading = 1,
+        _InstructionTailing = 2,
+        _InstructionOverlapped = 3
+    };
+
+    struct TraceRecordPage{
+        void* rawPtr;
+        duint rva;
+        TraceRecordType dataType;
+        unsigned int moduleIndex;
+    };
+
+    //Key := page base, value := trace record raw data
+    std::map<duint, TraceRecordPage> TraceRecord;
+    std::vector<std::string> ModuleNames;
+    unsigned int getModuleIndex(std::string moduleName);
+    unsigned long long instructionCounter;
+};
+
+extern TraceRecordManager TraceRecord;
+void _dbg_dbgtraceexecute(duint CIP);
+//exported to bridge
+unsigned int _dbg_dbggetTraceRecordHitCount(duint address);
+TRACERECORDBYTETYPE _dbg_dbggetTraceRecordByteType(duint address);
+bool _dbg_dbgsetTraceRecordType(duint pageAddress, TRACERECORDTYPE type);
+TRACERECORDTYPE _dbg_dbggetTraceRecordType(duint pageAddress);
+
+#endif // TRACERECORD_H