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.
126 lines
3.2 KiB
126 lines
3.2 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.
|
|
*/
|
|
|
|
/*
|
|
* This test verifies that the memalign() function returns correctly aligned pointers. It also
|
|
* verifies that the memory returned by memalign() can be freed by Pin's free() function.
|
|
*/
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
#include <stdlib.h>
|
|
#include "pin.H"
|
|
using std::dec;
|
|
using std::hex;
|
|
|
|
extern "C" void *memalign(size_t, size_t);
|
|
|
|
static BOOL TestAlign(size_t, size_t);
|
|
|
|
|
|
typedef std::vector<void *> POINTER_CONTAINER;
|
|
POINTER_CONTAINER Pointers;
|
|
|
|
|
|
int main(int argc, char * argv[])
|
|
{
|
|
PIN_Init(argc, argv);
|
|
|
|
// Test alignment of "small chunks"
|
|
for (int i = 0; i < 10; i++)
|
|
{
|
|
for (size_t align = 1; align <= 8192; align <<= 1)
|
|
{
|
|
if (!TestAlign(align, 4))
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
// Test alignment of "big chunks" that are larger than 1 page but smaller than 2 pages
|
|
for (int i = 0; i < 10; i++)
|
|
{
|
|
for (size_t align = 1; align <= 8192; align <<= 1)
|
|
{
|
|
if (!TestAlign(align, 1700))
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
// Pointers.clear();
|
|
|
|
// Test alignment of "big chunks" larger than one page in size and alignment larger than one page
|
|
for (int i = 0; i < 10; i++)
|
|
{
|
|
for (size_t align = 1024; align < 16*1024; align <<= 1)
|
|
{
|
|
if (!TestAlign(align, (i+1)*100000))
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
for (POINTER_CONTAINER::iterator it = Pointers.begin(); it != Pointers.end(); ++it)
|
|
free(*it);
|
|
|
|
// Test realloc to larger and smaller sizes
|
|
|
|
void * ptr = malloc(20);
|
|
if (ptr == NULL)
|
|
{
|
|
std::cerr << "Malloc failed\n";
|
|
exit(-1);
|
|
}
|
|
std::cerr << "Malloc : " << hex << ptr << dec << std::endl;
|
|
void * tmp = realloc(ptr, 21000);
|
|
if (tmp == NULL)
|
|
{
|
|
std::cerr << "Realloc to larger size failed\n";
|
|
exit(-1);
|
|
}
|
|
ptr = tmp;
|
|
std::cerr << "Realloc to larger size : " << hex << ptr << dec << std::endl;
|
|
tmp = realloc(ptr, 500);
|
|
if (tmp == NULL)
|
|
{
|
|
std::cerr << "Realloc to smaller size failed\n";
|
|
exit(-1);
|
|
}
|
|
ptr = tmp;
|
|
std::cerr << "Realloc to smaller size : " << hex << ptr << dec << std::endl;
|
|
free(ptr);
|
|
|
|
// Never returns
|
|
PIN_StartProgram();
|
|
return 0;
|
|
}
|
|
|
|
|
|
static BOOL TestAlign(size_t align, size_t size)
|
|
{
|
|
void *p = memalign(align, size);
|
|
if (!p)
|
|
{
|
|
std::cerr << "Returned NULL for alignment " << align << std::endl;
|
|
return FALSE;
|
|
}
|
|
|
|
Pointers.push_back(p);
|
|
|
|
ADDRINT addr = reinterpret_cast<ADDRINT>(p);
|
|
if (addr & (align-1))
|
|
{
|
|
std::cerr << "Incorrect alignment for " << align << " (p=" << p << ")" << std::endl;
|
|
return FALSE;
|
|
}
|
|
|
|
std::cout << "Alignment " << align << " : Size : " << size << " " << p << std::endl;
|
|
return TRUE;
|
|
}
|