Subversion Repositories shark

/*
 * Project: S.Ha.R.K.
 *
 * Coordinators:
 *   Giorgio Buttazzo    <giorgio@sssup.it>
 *   Paolo Gai           <pj@gandalf.sssup.it>
 *
 * Authors     :
 *   Giacomo Guidi       <giacomo@gandalf.sssup.it>
 *
 * ReTiS Lab (Scuola Superiore S.Anna - Pisa - Italy)
 *
 * http://www.sssup.it
 * http://retis.sssup.it
 * http://shark.sssup.it
 */

#ifndef __TRACER_H__
#define __TRACER_H__

#ifdef __OLD_TRACER__

#include <ll/sys/types.h>
#include "types.h"
#include "trace.h"
#include "FTrace.h"

#define TRACER_LOGEVENT oldtrace

extern int (*trc_register_eventclass)
     (int class,int num,int(*func)(trc_event_t *, int event, void *ptr));

extern void (*trc_logevent)(int event, void *ptr);

extern int (*trc_resume)(void);
extern int (*trc_suspend)(void);

#define NOT_DEFINED 0xFF

extern __inline__ void oldtrace(BYTE type, BYTE flag, DWORD par1, DWORD par2)
{

  static int oldtype;
  static int oldpar;
  extern BYTE conv[256]; // Conversion table: ctable.c
  BYTE ttype = 0;
 
  ttype = type & 0x0F;
  type = type >> 4;
  type |= (ttype << 4);

  oldtype = conv[type];

  if (oldtype != NOT_DEFINED) trc_logevent(oldtype,&oldpar);

}


#else
#ifdef __NEW_TRACER__

#include <ll/sys/types.h>
#include <ll/i386/hw-instr.h>
#include "FTrace.h"

#define TRACER_NO_OUTPUT  0
#define TRACER_UDP_OUTPUT 1

#define TRACER_LOGEVENT fast_logevent

int tracer_initialize(int MemorySize);
void tracer_enable();
void tracer_disable();
void tracer_print_statistics();

//UDP

int init_udp_tracer(char *local_ip, char *target_ip);

#define TRACER_SERIALIZE

/* Save bytes => 1(type) + 4(tsc_high) + 4(tsc_low) + 1(size)
                if (flag & 1) + 4(par1)
                if (flag & 3) + 4(par2)
 * Return 0 -> Success
 * Return 1 -> Inactive
 * Return 2 -> Error
 */
extern __inline__ int fast_logevent(BYTE type, BYTE flag, DWORD par1, DWORD par2)
{

  extern void *StartTracerBuffer;
  extern void *EndTracerBuffer;
  extern void *LastBeforeEndTracerBuffer;
  extern void *CurrentTracerBuffer;
  extern void *FirstTracerBuffer;

  extern int TracerActive;
  extern unsigned long long TracerEventsRecorded;
  extern unsigned int TracerEventsSaved;

  SYS_FLAGS f;

  DWORD tsc_low, tsc_high;
  BYTE size = 10;

  f = ll_fsave();

  if (!TracerActive) {
    ll_frestore(f);
    return 1;
  }

  #ifdef TRACER_SERIALIZE
    __asm__("xorl %%eax,%%eax\n\t"
            "cpuid\n\t"
            "rdtsc\n\t"
            :"=a" (tsc_low), "=d" (tsc_high)
            :
            :"ebx","ecx");
  #else
    __asm__("rdtsc\n\t"
            :"=a" (tsc_low), "=d" (tsc_high)
            ::);
  #endif

  if (flag & 1) size += 4;
  if (flag & 3) size += 4;

  // Adjust FirstTracerBuffer
  if (FirstTracerBuffer >= CurrentTracerBuffer && TracerEventsSaved != 0) {
    while ((CurrentTracerBuffer + size - 1) >= FirstTracerBuffer &&
            FirstTracerBuffer != StartTracerBuffer) {
      FirstTracerBuffer += *(BYTE *)(FirstTracerBuffer + 9);
      if (FirstTracerBuffer >= LastBeforeEndTracerBuffer) FirstTracerBuffer = StartTracerBuffer;
    }
  } else {
    TracerEventsSaved++;
  }

  // Check if we overcome EndTracerBuffer
  if ((CurrentTracerBuffer + size - 1) > EndTracerBuffer) {
    //Cyclical Buffer implementation
    LastBeforeEndTracerBuffer = CurrentTracerBuffer;
    CurrentTracerBuffer = StartTracerBuffer;
    //Set the First Event
    while ((CurrentTracerBuffer + size - 1) >= FirstTracerBuffer) {
      FirstTracerBuffer += *(BYTE *)(FirstTracerBuffer + 9);
      if (FirstTracerBuffer >= LastBeforeEndTracerBuffer) FirstTracerBuffer = StartTracerBuffer;
    }
  }

  // Add the new event
  *(BYTE *)CurrentTracerBuffer = type;
  CurrentTracerBuffer++;
  *(DWORD *)(CurrentTracerBuffer) = tsc_high;
  CurrentTracerBuffer+=4;
  *(DWORD *)(CurrentTracerBuffer) = tsc_low;
  CurrentTracerBuffer+=4;
  *(BYTE *)(CurrentTracerBuffer) = size;
  CurrentTracerBuffer++;
  if (flag & 1) {
    *(DWORD *)(CurrentTracerBuffer) = par1;
    CurrentTracerBuffer+=4;
  }
  if (flag & 3) {
    *(DWORD *)(CurrentTracerBuffer) = par2;
    CurrentTracerBuffer+=4;
  }
 
  if (CurrentTracerBuffer > EndTracerBuffer) {
    ll_frestore(f);
    return 2;
  }

  TracerEventsRecorded++;

  ll_frestore(f);
  return 0;

}

#else

#define TRACER_LOGEVENT(a,b,c,d)

#endif

#endif

#endif