tracer.cxx File Reference

#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <iostream>
#include <sys/types.h>
#include <sys/syscall.h>
#include <sys/ptrace.h>
#include <sys/wait.h>
#include <unistd.h>
#include <errno.h>
#include <sys/user.h>
#include <assert.h>
#include <sys/select.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <pthread.h>
#include "tls.hh"
#include "memoryprotection/filter.h"
#include "memoryprotection/UDSignals.h"
#include "memoryprotection/MemorySegment.h"

Go to the source code of this file.

Macros
#define	MAXTHREADS   128

Functions
void	FetchMap (pid_t pid)
char *	getLibName (long int address)
void	processParameters (int, char *argv[])
void	test (pid_t pid, int wait_val)
int	SingleSteps (pid_t pid)
int	traceloop (pid_t pid)
void *	tracer_thread (void *pid_ptr)
void	startThreadTracers (pid_t pid)
void	waitForAllTracers (pid_t pid)
void	openFilter ()
void	closeFilter ()
int	main (int argc, char *argv[])

Variables
int	fildes [2]
int	status
char	cmd [1024]
int	protect = 0
pid_t	filter
int	numberOfThreads
pid_t	threads [MAXTHREADS]
pthread_t	tids [MAXTHREADS]
G4ThreadLocal struct user_regs_struct	regs
G4ThreadLocal struct user_regs_struct	regssinglestep
G4ThreadLocal int	detectionFlag = 0
pthread_mutex_t	segFaultHandlingLock = PTHREAD_MUTEX_INITIALIZER

Macro Definition Documentation

#define MAXTHREADS   128

Definition at line 28 of file tracer.cxx.

Function Documentation

void closeFilter

(

)

Definition at line 478 of file tracer.cxx.

References fildes, and filter.

Referenced by main().

{
  printf("tracer stopping...\n");
  unsigned long exitCode = 0;

  printf("tracer: notifying filter to exit\n");
  write(fildes[1], &exitCode, sizeof(exitCode));

  waitpid(filter, NULL, 0);
  printf("tracer: filter stopped...%d\n", filter);
  close(fildes[1]);
}

void FetchMap

(

pid_t

pid

)

Definition at line 70 of file tracer.cxx.

References test::fname.

Referenced by startThreadTracers().

{
  char fname[128], line[4096];
  sprintf(fname, "/proc/%d/maps", pid);
  FILE *fp = fopen(fname, "r");

  char *n1, *n2, *p;

  char lastlibname[1024];
    
  while (fgets(line, sizeof(line), fp) != NULL) {
    p = NULL;
    p = strstr(line, "/");
    
    if (p != NULL)
    {
      strcpy(lastlibname, p);
    }

    p = strstr(line, "[");

    if (p != NULL)
    {
      strcpy(lastlibname, p);
    }

    strcpy(libname[numOfSegments], lastlibname);

    n1 = strtok(line, "- ");
    n2 = strtok(NULL, "- ");
    memorySegments[numOfSegments].start = (void*)strtoul(n1, NULL, 16);
    memorySegments[numOfSegments].end = (void*)strtoul(n2, NULL, 16);

    numOfSegments++;
  }
  fclose(fp);

  int i;
  for (i = 0 ; i < numOfSegments ; i++)
  {
    std::cout << "Maps: " << memorySegments[i].start << ", "
              << memorySegments[i].end << ", " << libname[i] << std::endl;
  }
}

char* getLibName

(

long int

address

)

Definition at line 115 of file tracer.cxx.

Referenced by traceloop().

{
  int i;
  for (i = 0 ; i < numOfSegments ; i++)
  {
    if ((void*)address >= memorySegments[i].start && (void*)address <= memorySegments[i].end)
      return libname[i];
  }
  return NULL;
}

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 491 of file tracer.cxx.

References closeFilter(), openFilter(), processParameters(), startThreadTracers(), threads, traceloop(), and waitForAllTracers().

{
  isMain = 1;
  processParameters(argc, argv);
  openFilter();
  startThreadTracers(threads[0]);
  traceloop(threads[0]);
  waitForAllTracers(threads[0]);

  closeFilter();
  return 0;
}

void openFilter

(

)

Definition at line 436 of file tracer.cxx.

References cmd, fildes, filter, generategdbscripts(), insertAddress(), outputAddresses(), and status.

Referenced by main().

{
  status = pipe(fildes);

  filter = fork();

  if (filter == 0) //child process
  {
    close(fildes[1]);
    unsigned long code;
    fd_set rfds;
    FD_ZERO(&rfds);
    FD_SET(fildes[0], &rfds);

    while (1)
    {
      int len;
      select(fildes[0] + 1, &rfds, NULL, NULL, NULL);
      if ((len = read(fildes[0], &code, sizeof(code))) < 0) break;
      else
      {
        if (len > 0)
        {
          if (code > 0) insertAddress(code);
      else
            break;
    }
      }
      len = 0;
    }
    close(fildes[0]);
    outputAddresses(cmd);
    generategdbscripts();
    printf("tracer: filter stopping...\n");
    exit(0);
  }
  else
  {
    close(fildes[0]);
  }
}

void processParameters	(	int	,
		char *	argv[]
	)

Definition at line 126 of file tracer.cxx.

References cmd, numberOfThreads, and threads.

Referenced by main().

{
  strcpy(cmd, argv[1]);
  printf("tracer: traced process executable file: %s:: %s %s\n", cmd, argv[1], argv[2]);

  char tmpnum[1024];
  char *current;
  current = argv[2];
  int i = 0;
  while (*current != ' ')
  {
    tmpnum[i] = *current;
    current++; i++;
  }

  tmpnum[i] = 0;
  current++;
  numberOfThreads = atoi(tmpnum);
  printf("tracer number of threads: %d.\n", numberOfThreads);

  int j;
  for (j = 0 ; j < numberOfThreads; j++)
  {
    i = 0;
    while (*current != ' ')
    {
      tmpnum[i] = *current;
      current++; i++;
    }
    tmpnum[i] = 0;
    current++;
    threads[j] = atoi(tmpnum);
    printf("tracer threads[%d] = %d\n", j, threads[j]);
  }
}

int SingleSteps

(

pid_t

pid

)

Definition at line 193 of file tracer.cxx.

References assert, fildes, regssinglestep, and test().

Referenced by traceloop().

{
  int wait_val;           /*  traced's return value        */

#if defined(__linux__) 
  if (ptrace(PT_GETREGS, pid, 0, &regssinglestep) != 0)
  {
    printf("tracer for %d ptrace getregs", pid);
    exit(-1);
  }
#endif

#ifdef __x86_64__ 
  write(fildes[1], &(regssinglestep.rip), sizeof(regssinglestep.rip));
#else
  write(fildes[1], &(regssinglestep.eip), sizeof(regssinglestep.eip));
#endif

  if (ptrace(PT_STEP, pid, 0, 0) != 0)
  {
    printf("tracer for %d ptrace singlestep", pid);
    exit(-1);
  }
  assert(pid == my_wait(pid, &wait_val));
  if (wait_val == 1407) return 0;
  if (WSTOPSIG(wait_val) == SIGTERM)
  {
    printf("tracer for %d terminated signal\n", pid);
    exit(-1);
  }
  else if (wait_val == 0)
  {
    printf("tracer for %d finished signal\n", pid);
    exit(-1);
  }
  else
  {
    test(pid, wait_val);
    exit(-1);
  }
  return 0;
}

void startThreadTracers

(

pid_t

pid

)

Definition at line 414 of file tracer.cxx.

References assert, FetchMap(), numberOfThreads, threads, tids, and tracer_thread().

Referenced by main().

{
  FetchMap(pid);
  int j;
  for (j = 1 ; j < numberOfThreads; j++)
  {
    printf("tracer for %d start threads[%d] = %d\n", pid, j, threads[j]);
    assert(0 == pthread_create(&tids[j], NULL, tracer_thread, &threads[j]));
  }
}

void test	(	pid_t	pid,
		int	wait_val
	)

Definition at line 162 of file tracer.cxx.

Referenced by SingleSteps(), and traceloop().

{
  printf("tracer for %d WIFEXITED: %d\n", pid, WIFEXITED(wait_val));
  printf("tracer for %d WEXITSTATUS: %d\n", pid, WEXITSTATUS(wait_val));
  printf("tracer for %d WIFSIGNALED: %d\n", pid, WIFSIGNALED(wait_val));
  printf("tracer for %d WTERMSIG: %d\n", pid, WTERMSIG(wait_val));
  printf("tracer for %d WIFSTOPPED: %d\n", pid, WIFSTOPPED(wait_val));
  printf("tracer for %d WSTOPSIG: %d\n", pid, WSTOPSIG(wait_val));
  printf("tracer for %d WIFCONTINUED: %d\n", pid, WIFCONTINUED(wait_val));
}

int traceloop

(

pid_t

pid

)

Definition at line 236 of file tracer.cxx.

References assert, cmd, detectionFlag, fildes, getLibName(), protect, regssinglestep, segFaultHandlingLock, SingleSteps(), and test().

Referenced by main(), and tracer_thread().

{
  int wait_val;

  if (ptrace(PT_ATTACH, pid, 0, 0) != 0)
  {
    printf("tracer for %d ptrace attach", pid);
    exit(-1);
  }
  assert(pid == my_wait(pid, &wait_val));

  printf("tracer for %d after attach\n", pid);

  if (isMain)
  {
    if (ptrace(PT_CONTINUE, pid, 0, SIGUSR2) != 0)
    {
      printf("tracer for %d ptrace continue with SIGUSR2", pid);
      exit(-1);
    }

    assert(pid == my_wait(pid, &wait_val));
    assert(SIGUSR2 == WSTOPSIG(wait_val));

    protect = 1;
    printf("tracer for %d after memory protection\n", pid);

    if (ptrace(PT_CONTINUE, pid, 0, 0) != 0)
    {
      printf("tracer for %d ptrace continue without SIGUSR2", pid);
      exit(-1);
    }

  }
  else
  {
    printf("tracer for %d Waiting for the memory is protected\n", pid);
    while (!protect) usleep(10);

    if (ptrace(PT_CONTINUE, pid, 0, 0) != 0)
    {
      printf("tracer for %d ptrace continue without SIGUSR2", pid);
      exit(-1);
    }

    printf("tracer for %d after continue\n", pid);

  }

  int tmp = 0;

  do
  {
    // printf("tracer for %d tracerloop\n", pid);
    // pid_t frompid = my_wait(pid, &wait_val);

    if (WSTOPSIG(wait_val) == SIGSEGV)
    {
      pthread_mutex_lock(&segFaultHandlingLock);

#if defined(__linux__) 
      if (ptrace(PT_GETREGS, pid, 0, &regssinglestep) != 0)
      {
        printf("tracer for %d ptrace getregs", pid);
      }
#endif

      if (detectionFlag && !isMain)
      {
#ifdef __x86_64__ 
        write(fildes[1], &(regssinglestep.rip), sizeof(regssinglestep.rip));
        printf("tracer for %d Address: %lx in %s", pid, regssinglestep.rip, getLibName(regssinglestep.rip));
#else
        write(fildes[1], &(regssinglestep.eip), sizeof(regssinglestep.eip));
        printf("tracer for %d Address: %lx in %s", pid, regssinglestep.eip, getLibName(regssinglestep.eip));
#endif
        char convertcmd[1024];
        memset(convertcmd, 0, 1024);
#ifdef __x86_64__ 
        sprintf(convertcmd, "addr2line -e %s 0x%lx", cmd, regssinglestep.rip);  
#else
        sprintf(convertcmd, "addr2line -e %s 0x%lx", cmd, regssinglestep.eip);  
#endif
        system(convertcmd);
      }

      if (ptrace(PT_CONTINUE, pid, 0, SIGSEGV) != 0)
      {
        printf("tracer for %d ptrace continue with SIGSEGV", pid);
    tmp = -6;
      }

      assert(pid == my_wait(pid, &wait_val));
      assert(SIGUSR2 == WSTOPSIG(wait_val));


      if (SIGUSR2 != WSTOPSIG(wait_val))
      {
        test(pid, wait_val);
        tmp = -3; 
      }
      else
      {
        SingleSteps(pid);
      //      counter++;
      //      printf("Counter: %d\n", counter);

        if (ptrace(PT_CONTINUE, pid, 0, SIGUSR1) != 0)
        {
          printf("tracer for %d ptrace continue with SIGUSR1\n", pid);
          tmp = -5;
        }
      }

      pthread_mutex_unlock(&segFaultHandlingLock);

    }
    else
    {
      if ( wait_val == 0 )
      {
    //        test(pid, wait_val);
        tmp = -1;
      }
      else if (WSTOPSIG(wait_val) == SIGUSR1)
      {
    //        test(pid, wait_val);
        detectionFlag = 0;
        if (ptrace(PT_DETACH, pid, 0, 0) != 0)
        {
          printf("tracer for %d ptrace detach\n", pid);
          tmp = -6;
        }
      }
      else if (WSTOPSIG(wait_val) == SIGUSR2)
      {
        detectionFlag = 1;
        if (ptrace(PT_CONTINUE, pid, 0, 0) != 0)
        {
          printf("tracer for %d ptrace continue without SIGSEGV\n", pid);
          tmp = -4;
        }
      }
      else
      {
        if (ptrace(PT_CONTINUE, pid, 0, 0) != 0)
        {
          printf("tracer for %d ptrace continue without SIGSEGV\n", pid);
          tmp = -4;
        }
      }
      //      break;
    }


  } while (tmp == 0);

  printf("tracer for %d Exit code: %d\n", pid, tmp);

  if (tmp != -6)
  {
    if (ptrace(PT_DETACH, pid, 0, 0) != 0)
      printf("tracer for %d ptrace detach", pid);
  }

  return 0;
}

void* tracer_thread

(

void *

pid_ptr

)

Definition at line 404 of file tracer.cxx.

References traceloop().

Referenced by startThreadTracers().

{
  isMain = 0;
  pid_t pid;
  pid = * ((pid_t *) pid_ptr);

  traceloop(pid);
  return 0;
}

void waitForAllTracers

(

pid_t

pid

)

Definition at line 425 of file tracer.cxx.

References assert, numberOfThreads, threads, and tids.

Referenced by main().

{
  int j;
  for (j = 1 ; j < numberOfThreads; j++)
  {
    printf("tracer for %d join threads[%d] = %d\n", pid, j, threads[j]);
    assert(0 == pthread_join(tids[j], NULL));
    printf("tracer for %d after join threads[%d] = %d\n", pid, j, threads[j]);
  }
}

Variable Documentation

char cmd[1024]

Definition at line 25 of file tracer.cxx.

Referenced by G4FileUtilities::CopyFile(), G4UIcommandTree::CreateHTML(), G4GenericMessenger::DeclareMethod(), G4GenericMessenger::DeclareMethodWithUnit(), G4GenericMessenger::DeclareProperty(), G4GenericMessenger::DeclarePropertyWithUnit(), G4FileUtilities::DeleteFile(), CexmcRunManager::DoEventLoop(), G4VMPIsession::ExecCommand(), G4ModelCmdAddInterval< M >::G4ModelCmdAddInterval(), G4ModelCmdAddIntervalContext< M >::G4ModelCmdAddIntervalContext(), G4ModelCmdAddValue< M >::G4ModelCmdAddValue(), G4ModelCmdAddValueContext< M >::G4ModelCmdAddValueContext(), G4ModelCmdSetAuxPtsFillStyle< M >::G4ModelCmdSetAuxPtsFillStyle(), G4ModelCmdSetAuxPtsSizeType< M >::G4ModelCmdSetAuxPtsSizeType(), G4ModelCmdSetAuxPtsType< M >::G4ModelCmdSetAuxPtsType(), G4ModelCmdSetStepPtsFillStyle< M >::G4ModelCmdSetStepPtsFillStyle(), G4ModelCmdSetStepPtsSizeType< M >::G4ModelCmdSetStepPtsSizeType(), G4ModelCmdSetStepPtsType< M >::G4ModelCmdSetStepPtsType(), G4ModelCmdSetTimeSliceInterval< M >::G4ModelCmdSetTimeSliceInterval(), G4PersistencyCenterMessenger::G4PersistencyCenterMessenger(), openFilter(), processParameters(), CexmcRunManager::SetupConstructionHook(), and traceloop().

G4ThreadLocal int detectionFlag = 0

Definition at line 64 of file tracer.cxx.

Referenced by traceloop().

int fildes[2]

Definition at line 23 of file tracer.cxx.

Referenced by closeFilter(), openFilter(), SingleSteps(), and traceloop().

pid_t filter

Definition at line 30 of file tracer.cxx.

Referenced by G4AttributeFilterT< T >::Clear(), closeFilter(), G4AttributeFilterT< T >::Evaluate(), G4ScoreQuantityMessenger::FParticleWithEnergyCommand(), G4AttFilterUtils::GetNewFilter(), openFilter(), G4AttributeFilterT< T >::Print(), G4PropagatorInField::SetTrajectoryFilter(), and G4AttributeFilterT< T >::~G4AttributeFilterT().

int numberOfThreads

Definition at line 32 of file tracer.cxx.

Referenced by processParameters(), startThreadTracers(), and waitForAllTracers().

int protect = 0

Definition at line 26 of file tracer.cxx.

Referenced by traceloop().

G4ThreadLocal struct user_regs_struct regs

Definition at line 62 of file tracer.cxx.

Referenced by ExTGRCRegionCutsMgr::AddRegionCuts(), and ExTGRCRegionCutsMgr::FindRegionData().

G4ThreadLocal struct user_regs_struct regssinglestep

Definition at line 63 of file tracer.cxx.

Referenced by SingleSteps(), and traceloop().

pthread_mutex_t segFaultHandlingLock = PTHREAD_MUTEX_INITIALIZER

Definition at line 66 of file tracer.cxx.

Referenced by traceloop().

int status

Definition at line 24 of file tracer.cxx.

Referenced by CCalStackingAction::ClassifyNewTrack(), deflateEnd(), dminit(), G4FRofstream::DoesFileExist(), dpmjet(), filenu(), G4GIDI::freeTarget(), g4dpmjet_open_fort6(), g4dpmjet_open_nuclear_bin(), RE01Trajectory::GetAttDefs(), G4Pythia6Decayer::ImportDecayProducts(), openFilter(), phocin(), phoerr(), phorep(), G4HadronElasticProcess::PostStepDoIt(), prblm2(), readffky(), smr_setMessageError(), smr_setMessageInfo(), smr_vsetMessageError(), smr_vsetMessageInfo(), tpia_angular_SampleMu(), tpia_angularEnergy_getFromElement(), tpia_frame_setFromString(), tpia_Legendre_getFromElement(), tpia_map_findAllOfTarget(), tpia_misc_setMessageError_Element(), tpia_product_getDecayChannelFromElement(), and tpia_target_read().

pid_t threads[MAXTHREADS]

Definition at line 33 of file tracer.cxx.

Referenced by main(), processParameters(), startThreadTracers(), and waitForAllTracers().

pthread_t tids[MAXTHREADS]

Definition at line 34 of file tracer.cxx.

Referenced by startThreadTracers(), and waitForAllTracers().