You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
3.6 KiB
125 lines
3.6 KiB
/*
|
|
* Copyright 2002-2019 Intel Corporation.
|
|
*
|
|
* This software is provided to you as Sample Source Code as defined in the accompanying
|
|
* End User License Agreement for the Intel(R) Software Development Products ("Agreement")
|
|
* section 1.L.
|
|
*
|
|
* This software and the related documents are provided as is, with no express or implied
|
|
* warranties, other than those that are expressly stated in the License.
|
|
*/
|
|
|
|
#if !defined(_ALIGNCHK_H_)
|
|
#define _ALIGNCHK_H_
|
|
#include <iostream>
|
|
#include <iomanip>
|
|
|
|
|
|
#ifndef ALIGNMENT_CHECK_KNOB_DEFAULT
|
|
# define ALIGNMENT_CHECK_KNOB_DEFAULT "1"
|
|
#endif
|
|
|
|
namespace INSTLIB
|
|
{
|
|
using namespace std;
|
|
|
|
class ALIGN_CHECK
|
|
{
|
|
public:
|
|
ALIGN_CHECK():
|
|
enableKnob(KNOB_MODE_WRITEONCE, "pintool", "alignchk",
|
|
ALIGNMENT_CHECK_KNOB_DEFAULT,
|
|
"Check alignment of data for movdqa instructions.")
|
|
{
|
|
}
|
|
|
|
bool IsActive()
|
|
{
|
|
return enableKnob;
|
|
}
|
|
|
|
INT32 Activate()
|
|
{
|
|
if (enableKnob == 0)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
// Register Instruction to be called to instrument instructions
|
|
TRACE_AddInstrumentFunction(InstrumentTrace, this);
|
|
|
|
return 1;
|
|
}
|
|
private:
|
|
|
|
KNOB<BOOL> enableKnob;
|
|
|
|
// Pin calls this function every time a new trace is encountered
|
|
// Goal: Check alignment of MOVDQA instructions
|
|
// NOTE: We are using TRACE instrumentation because it has higher
|
|
// precedence than INS instrumentation.
|
|
|
|
static VOID
|
|
InstrumentTrace(TRACE trace, VOID *v)
|
|
{
|
|
ALIGN_CHECK* xthis = reinterpret_cast<ALIGN_CHECK*>(v);
|
|
if (xthis->enableKnob)
|
|
{
|
|
for (BBL bbl = TRACE_BblHead(trace); BBL_Valid(bbl); bbl = BBL_Next(bbl))
|
|
{
|
|
for (INS ins = BBL_InsHead(bbl); INS_Valid(ins); ins = INS_Next(ins))
|
|
{
|
|
xed_iclass_enum_t iclass = static_cast<xed_iclass_enum_t>(INS_Opcode(ins));
|
|
if (iclass == XED_ICLASS_MOVDQA)
|
|
{
|
|
if (INS_IsMemoryWrite(ins))
|
|
{
|
|
INS_InsertCall(ins, IPOINT_BEFORE, (AFUNPTR)check_alignment,
|
|
IARG_MEMORYWRITE_EA,
|
|
IARG_INST_PTR,
|
|
IARG_UINT32, 0,
|
|
IARG_END);
|
|
}
|
|
else if (INS_IsMemoryRead(ins))
|
|
{
|
|
INS_InsertCall(ins, IPOINT_BEFORE, (AFUNPTR)check_alignment,
|
|
IARG_MEMORYREAD_EA,
|
|
IARG_INST_PTR,
|
|
IARG_UINT32, 1,
|
|
IARG_END);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
static VOID
|
|
check_alignment(ADDRINT memaddr, ADDRINT pc, UINT32 read)
|
|
{
|
|
// movdqa's must be 16b aligned. So the low 4 bits of the address
|
|
// MUST be zero.
|
|
// cerr << "Checking " << hex << memaddr << " @ " << pc << dec << endl;
|
|
if ((memaddr & 0xF) != 0)
|
|
{
|
|
// misaligned
|
|
cerr << "Misaligned MOVDQA at instruction address: "
|
|
<< hex
|
|
<< pc
|
|
<< " "
|
|
<< (read ? "read" : "write")
|
|
<< " data address: "
|
|
<< memaddr
|
|
<< dec
|
|
<< endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
};
|
|
|
|
} // namespace INSTLIB
|
|
#endif
|