-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathUtils.cpp
More file actions
162 lines (145 loc) · 3.82 KB
/
Copy pathUtils.cpp
File metadata and controls
162 lines (145 loc) · 3.82 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
#include "Utils.h"
using namespace std;
/* ===================================================================== */
// Utilities
/* ===================================================================== */
// Shamelessly taken and slightly modified from debugtrace example. Totally UNTESTED Btw.
VOID ShowN(THREADID threadid, const char * disasm, ADDRINT eip, UINT32 size, VOID *ea )
{
UINT8 b[512];
UINT8* x;
if (size > 512)
{
x = new UINT8[size];
}
else
{
x = b;
}
PIN_SafeCopy( x, static_cast<UINT8*>(ea), size );
fprintf(g_outfile, "WR [tid:%d] eip:0x%lx %s ; [ea:0x%lx] = ", threadid, eip, disasm, ea);
for ( UINT32 i = 0; i < size; i++ )
{
fprintf(g_outfile, "%02x", static_cast<UINT32>(x[size-i-1]));
if ( ( ( reinterpret_cast<ADDRINT>(ea)+size-i-1 )&0x3 ) ==0 && i<size-1 )
{
fprintf(g_outfile, "_");
}
}
if (size > 512)
{
delete [] x;
}
fprintf(g_outfile, " [size:%d]\r\n", size);
}
void UpdateBaseAddress( ADDRINT base, rapidjson::Value *val, const std::string msg, const std::string value_type)
{
for (rapidjson::Value::ValueIterator itr = val->Begin(); itr != val->End(); ++itr)
{
if ( itr->IsUint() )
{
itr->SetUint( base + itr->GetUint() );
fprintf(g_outfile, "%s %s @ 0x%lx\r\n", msg.c_str(), value_type.c_str(), itr->GetUint());
}
else if (itr->IsUint64() )
{
itr->SetUint64( base + itr->GetUint64() );
fprintf(g_outfile, "%s 64 %s @ 0x%lx\r\n", msg.c_str(), value_type.c_str(), itr->GetUint64());
}
}
}
void UpdateDataAdd( ADDRINT base )
{
UpdateBaseAddress( base, g_vars.data_add, "Adding", "data" );
}
void UpdateCodeAdd( ADDRINT base )
{
UpdateBaseAddress( base, g_vars.code_add, "Adding", "code" );
}
void UpdateIgnoredCode( ADDRINT base )
{
UpdateBaseAddress( base, g_vars.code_ignore, "Ignoring", "code" );
}
void UpdateIgnoredData( ADDRINT base )
{
UpdateBaseAddress( base, g_vars.data_ignore, "Ignoring", "data" );
}
bool compareFiles(const std::string imageName, const std::string targetImage)
{
if ( imageName.compare(targetImage) == 0 )
{
return true;
}
string fileOnly = imageName.substr(imageName.find_last_of("/\\")+1);
if ( targetImage.compare(fileOnly) == 0 )
{
return true;
}
return false;
}
const char *
dumpInstruction(INS ins)
{
ADDRINT address = INS_Address(ins);
std::stringstream ss;
// Generate instruction byte encoding
for (size_t i=0;i<INS_Size(ins);i++)
{
ss << setfill('0') << setw(2) << hex << (((unsigned int) *(unsigned char*)(address + i)) & 0xFF) << " ";
}
for (size_t i=INS_Size(ins);i<8;i++)
{
ss << " ";
}
// Generate diassembled string
ss << INS_Disassemble(ins);
return _strdup(ss.str().c_str());
}
std::list<REG> *
listRegisters(INS ins)
{
std::list<REG> *registers = new std::list<REG>;
for (UINT32 i = 0; i < INS_OperandCount(ins); i++) {
if (INS_OperandIsReg(ins, i)) {
REG reg = INS_OperandReg(ins, i);
registers->push_back(reg);
}
}
return registers;
}
int validateVars()
{
fprintf(g_outfile, "Validating config...\r\n");
if ( g_vars.code_start >= g_vars.code_end )
{
cerr << "ERROR: code start is >= code end.\r\n" << endl;
return -1;
}
if ( g_vars.data_start >= g_vars.data_end )
{
cerr << "ERROR: data start " << g_vars.data_start << " is >= data end " << g_vars.data_end << ".\r\n" << endl;
return -1;
}
return 0;
}
/*!
* Print out help message.
*/
INT32 Usage(const std::string err)
{
cerr << err << endl;
cerr << KNOB_BASE::StringKnobSummary() << endl;
return -1;
}
/*!
* Print out analysis results.
* This function is called when the application exits.
* @param[in] code exit code of the application
* @param[in] v value specified by the tool in the
* PIN_AddFiniFunction function call
*/
VOID Fini(INT32 code, VOID *v)
{
fflush( g_outfile );
fclose( g_outfile );
}