pivot.h

#ifndef __PIVOT_H
#define __PIVOT_H
/**
   @author Edward J. Schwartz
*/

#include "pin.H"
#include "pin_taint.h"
#include "pin_misc.h"
#include <set>
#include <stdint.h>

/** 
    Types of pivots.  

    Switchstack is ESP <- reg + c.
    Mempivot is ESP <- M[reg + c]. 
*/
typedef enum { UNKNOWN, SWITCHSTACK, MEMPIVOT } pivottype_t;

/** Pivot objects */
bool operator<(const struct pivot_s &a, const struct pivot_s &b);

typedef struct pivot_s {
  ADDRINT address; /** Gadget address. */
  pivottype_t t; /** Type of pivot. */
  REG base; /** Base register. */
  ADDRINT offset; /** Offset */
} pivot_t;

/** Sets of pivot objects */
typedef set<pivot_t> pivot_set;

/** Function headers */
pivot_set PIVOT_parseinput(istream &f);
void PIVOT_testpivot(pivot_set, CONTEXT *, pintrace::TaintTracker &);
ostream& operator<<(ostream &o, const pivot_s &);

#endif