ProtectedMemory.cc File Reference

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

Go to the source code of this file.

Macros
#define	MAXTHREADS   128

Functions
void *	AllocateInProtectedHeap (size_t size)
void	PhaseChange ()
void	BuildProtectedMemory (size_t totalHeap)
void	addThread ()
void	setuphandlers ()
void	starttracer ()
void	finishtracer ()
void	StartDetection ()
void	FinishDetection ()

Variables
int	numberOfThreads = 0
pid_t	threads [MAXTHREADS]
pid_t	tracerpid

Macro Definition Documentation

#define MAXTHREADS   128

Definition at line 456 of file ProtectedMemory.cc.

Referenced by addThread().

Function Documentation

void addThread

(

)

Definition at line 462 of file ProtectedMemory.cc.

References assert, and MAXTHREADS.

{
  numberOfThreads++;
  assert(numberOfThreads <= MAXTHREADS);
  threads[numberOfThreads-1] = gettid();
}

void* AllocateInProtectedHeap

(

size_t

size

)

Definition at line 36 of file ProtectedMemory.cc.

References malloc().

{
  if (startPosition != 0)
  {
    currentPosition = nextPosition;
    nextPosition = &nextPosition[size];
    if (nextPosition >= endPosition)
    {
      nextPosition = currentPosition;
      return malloc(size);
    }
    return currentPosition;
  }
  else
  {
    return malloc(size);
  }
}

void BuildProtectedMemory

(

size_t

totalHeap

)

Definition at line 183 of file ProtectedMemory.cc.

Referenced by starttracer().

{
  pagesize = getpagesize();

  pid_t myselfpid = getpid();
  char outputfilename[1024];  
  outputfilename[0] = 0;
  sprintf(outputfilename, "./memoryhotpot.out.%d.%d", phase, myselfpid);
  outputFile = open(outputfilename, O_CREAT|O_RDWR, S_IRWXU);

  show_stackframe();

  #ifdef PROTECTEDMEMORYDEBUG
  printf("Page size: %d\n", pagesize);
  #endif

  long tmp;
  for (logpagesize = -1, tmp = pagesize; tmp > 0; logpagesize++)
    tmp = tmp >> 1;

  #ifdef PROTECTEDMEMORYDEBUG
  printf("log(page size): %d\n", logpagesize);
  #endif

  if (totalHeap > 0)
  {
    AllocateProtectedHeap(totalHeap, pagesize);

    memorySegments[numOfSegments].start = getPageNumber(nextPosition);
    memorySegments[numOfSegments].end = getPageNumber(endPosition); 
    numOfSegments++;
  }

  //force to load backtrace library
  //  show_stackframe();

  //open output file
  //  assert(-1 != (stdOutFile = open("./memoryhotspot.out", O_CREAT, S_IRWXU)));

  char buf1[1024];
  char buf2[51200];
  char *tmp2 = buf2;
  static char filename2[1024];
  void *start;
  void *end;
  int writable;

  if (-1 == readFile((char *) "/proc/self/cmdline", buf1, 1024)) {
    perror("readFile: /proc/self/cmdline");
    abort();
  }
  memset(application, 0, 1024);
  strcpy(application, buf1);

  //  cout << "command line: " << application << endl;

  if (-1 == readFile((char *) "/proc/self/maps", buf2, 51200)) {
    perror("readFile: /proc/self/maps");
    abort();
  }
  static char lastfilename[1024];
  do {
    tmp2 = readMapsLine(tmp2, &start, &end, &writable, filename2);
    if (strlen(filename2) > 0)
    {
      lastfilename[0] = 0;
      strcpy(lastfilename, filename2); 
    }
    else
    {
      strcpy(filename2, lastfilename); 
    }

    //    if (writable == 1 && (long int) start < 0x700000000000 && 
    //        strncmp(filename2, "/lib64/", 7) != 0 && strncmp(filename2, "/opt/", 5) != 0 &&
    //        ((long int) nextPosition < (long int) start || (long int) nextPosition > (long int ) end) && 
    //        ((long int) endPosition < (long int) start || (long int) endPosition > (long int ) end))
    if (writable ==  1)
    {
      //ignore 
      if (strstr(filename2, "ld") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "vdso") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "stack") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "libc") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "libProtectedMemory.so") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "libParRunManagerMP.so") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "libmymalloc.so") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "libtpmallocstub.so") != 0)
      {
        std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                  << std::endl;
      }
      else if (strstr(filename2, "heap") != 0)
      {
        static int Heapcount = 0;
        //Only one is the data segment follows with [heap] [thread private heap, empty, thread local storage]*          
        Heapcount++;
        if (Heapcount == 1)
        {
          std::cout << "Protected: " << numOfSegments << ", " << start << ", " << end << ", "
                                     << writable << ", " << filename2
                    << std::endl;
          memorySegments[numOfSegments].start = start;
          memorySegments[numOfSegments].end = end;
          numOfSegments++;
        }
        else
        {
          std::cout << "Not protected: " << start << ", " << end << ", "
                                       << writable << ", " << filename2
                    << std::endl;
        }
      }
      else if (strstr(filename2, "ParA01") != 0)
      {
        static int Parmaincount = 0;
    //Only one is the data segment follows with [heap] [thread private heap, empty, thread local storage]*
        Parmaincount++;
        printf("Determine: %d\n", Parmaincount);
    //        if (Parmaincount == 1 || Parmaincount == 2)
    { 
          std::cout << "Protected: " << numOfSegments << ", " << start << ", " << end << ", "
                                     << writable << ", " << filename2
                    << std::endl;
          memorySegments[numOfSegments].start = start;
          memorySegments[numOfSegments].end = end;
          numOfSegments++;
    }
      }
      else if (strstr(filename2, "geant4") != 0)
      {
        std::cout << "Protected: " << numOfSegments << ", " << start << ", " << end << ", "
                                   << writable << ", " << filename2
                  << std::endl;
        memorySegments[numOfSegments].start = start;
        memorySegments[numOfSegments].end = end;
        numOfSegments++;
      }

    }
    else
    {
      std::cout << "Not protected: " << start << ", " << end << ", "
                                     << writable << ", " << filename2
                << std::endl;
    }

  } while ((*tmp2) != '\0' && numOfSegments < 4096);
}

void FinishDetection

(

)

Definition at line 525 of file ProtectedMemory.cc.

References assert.

{
  //move to main thread
  //  UnSetUpAllProtect();

  assert(0 == raise(SIGUSR1));
  //For each thread, can not do this
}

void finishtracer

(

)

Definition at line 505 of file ProtectedMemory.cc.

References assert.

{
  UnSetUpAllProtect();
  close(outputFile);
  assert(0 == raise(SIGUSR1));
  printf("Before wait for child process: %d\n", tracerpid);
  waitpid(tracerpid, NULL, 0);
  printf("After wait for child process: %d\n", tracerpid);
}

void PhaseChange

(

)

Definition at line 104 of file ProtectedMemory.cc.

{
  int tmpoutputFile;
  tmpoutputFile = outputFile;

  phase++;
  pid_t myselfpid = getpid();
  char outputfilename[1024];  
  outputfilename[0] = 0;
  sprintf(outputfilename, "./memoryhotpot.out.%d.%d", phase, myselfpid);

  outputFile = open(outputfilename, O_CREAT|O_RDWR, S_IRWXU);
  close(tmpoutputFile);
}

void setuphandlers

(

)

Definition at line 570 of file ProtectedMemory.cc.

Referenced by starttracer().

{
  setuphandler0();
  setuphandler1();
  setuphandler2();
}

void StartDetection

(

)

Definition at line 516 of file ProtectedMemory.cc.

References assert.

{
  printf("waiting for the tracer\n");
  while (isCalled == 0) usleep(100);
  printf("the tracer ready\n");
  assert(0 == raise(SIGUSR2));
}

void starttracer

(

)

Definition at line 474 of file ProtectedMemory.cc.

References BuildProtectedMemory(), numberOfThreads, and setuphandlers().

{
  BuildProtectedMemory(0);
  setuphandlers();

  if ((tracerpid = fork()) == 0) //child process
  {
    char processid[10240];
    char tmptid[1024];
    processid[0] = 0;
    tmptid[0] = 0;
    sprintf(tmptid, "%d", numberOfThreads);
    strcat(processid, tmptid);
    strcat(processid, " ");
    int i;
    for (i = 0; i < numberOfThreads; i++)
    {
      tmptid[0] = 0;
      sprintf(tmptid, "%d", threads[i]);
      strcat(processid, tmptid);
      strcat(processid, " ");
    }
    execl("./memoryprotection/tracer", "./memoryprotection/tracer", application, processid, (char *)0);
    exit(0);
  }

}

Variable Documentation

int numberOfThreads = 0

Definition at line 457 of file ProtectedMemory.cc.

Referenced by starttracer().

pid_t threads[MAXTHREADS]

Definition at line 458 of file ProtectedMemory.cc.

pid_t tracerpid

Definition at line 459 of file ProtectedMemory.cc.