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.
154 lines
3.6 KiB
154 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.
|
|
*/
|
|
|
|
/*! @file
|
|
*
|
|
* A test for attach to single threaded application.
|
|
* The application launches Pin and waits until Pin attaches
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <sched.h>
|
|
#include <signal.h>
|
|
#include <stdlib.h>
|
|
#include <string>
|
|
#include <list>
|
|
#include <assert.h>
|
|
using std::list;
|
|
using std::string;
|
|
|
|
|
|
typedef bool (*FUNPTR)();
|
|
|
|
/* Pin doesn't kill the process if failed to attach, exit on SIGALRM */
|
|
void ExitOnAlarm(int sig)
|
|
{
|
|
fprintf(stderr, "Pin is not attached, exit on SIGALRM\n");
|
|
exit(0);
|
|
}
|
|
|
|
extern "C" bool PinAttached()
|
|
{
|
|
return false;
|
|
}
|
|
|
|
/*
|
|
* Expected command line: <this exe> [-th_num NUM] -pin $PIN -pinarg <pin args > -t tool <tool args>
|
|
*/
|
|
|
|
void ParseCommandLine(int argc, char *argv[], list < string>* pinArgs)//, list < string>* pinBinaryArgs)
|
|
{
|
|
string pinBinary;
|
|
string beforePinArgs;
|
|
for (int i=1; i<argc; i++)
|
|
{
|
|
string arg = string(argv[i]);
|
|
if (arg == "-pin")
|
|
{
|
|
pinBinary = argv[++i];
|
|
if ((i + 1) < argc)
|
|
{
|
|
arg = string(argv[i + 1]);
|
|
if ("-pinarg" != arg)
|
|
{
|
|
beforePinArgs = argv[++i];
|
|
}
|
|
}
|
|
}
|
|
else if (arg == "-pinarg")
|
|
{
|
|
//Add -pinarg to point start of pin args.
|
|
pinArgs->push_back(arg);
|
|
for (int parg = ++i; parg < argc; parg++)
|
|
{
|
|
pinArgs->push_back(string(argv[parg]));
|
|
++i;
|
|
}
|
|
}
|
|
}
|
|
if (!beforePinArgs.empty())
|
|
{
|
|
pinArgs->push_front(beforePinArgs);
|
|
}
|
|
assert(!pinBinary.empty());
|
|
pinArgs->push_front(pinBinary);
|
|
}
|
|
|
|
void StartPin(list <string>* pinArgs)
|
|
{
|
|
pid_t appPid = getpid();
|
|
pid_t child = fork();
|
|
if (child > 0)
|
|
{
|
|
return;
|
|
}
|
|
else if (0 == child)
|
|
{
|
|
// start Pin from child
|
|
char **inArgv = new char*[pinArgs->size()+10];
|
|
|
|
// Pin binary in the first
|
|
list <string>::iterator pinArgIt = pinArgs->begin();
|
|
string pinBinary = *pinArgIt;
|
|
pinArgIt++;
|
|
|
|
// build pin arguments:
|
|
// pin -pid appPid [pinarg]
|
|
unsigned int idx = 0;
|
|
inArgv[idx++] = (char *)pinBinary.c_str();
|
|
if ("-pinarg" != *pinArgIt)
|
|
{
|
|
inArgv[idx++] = (char *)(pinArgIt++)->c_str();
|
|
}
|
|
pinArgIt++;
|
|
inArgv[idx++] = (char*)"-pid";
|
|
inArgv[idx] = (char *)malloc(10);
|
|
sprintf(inArgv[idx++], "%d", appPid);
|
|
|
|
for (; pinArgIt != pinArgs->end(); pinArgIt++)
|
|
{
|
|
inArgv[idx++]= (char *)pinArgIt->c_str();
|
|
}
|
|
inArgv[idx] = 0;
|
|
|
|
execvp(inArgv[0], inArgv);
|
|
fprintf(stderr, "ERROR: execv %s failed\n", inArgv[0]);
|
|
exit(1);
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr, "ERROR: fork failed\n");
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char * argv[])
|
|
{
|
|
list <string> pinArgs;
|
|
volatile FUNPTR pinAttached = PinAttached;
|
|
|
|
ParseCommandLine(argc, argv, &pinArgs);
|
|
StartPin(&pinArgs);
|
|
|
|
/* Exit in 20 sec */
|
|
signal(SIGALRM, ExitOnAlarm);
|
|
alarm(20);
|
|
|
|
while (!pinAttached())
|
|
{
|
|
sched_yield();
|
|
sleep(2);
|
|
}
|
|
|
|
return 0;
|
|
}
|