override.hh

Go to the documentation of this file.

/* ###
 * IP: GHIDRA
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef __OVERRIDE__
#define __OVERRIDE__

#include "database.hh"

class FuncCallSpecs;        // Forward declaration

class Override {
public:
  enum {
    NONE = 0,           
    BRANCH = 1,         
    CALL = 2,           
    CALL_RETURN = 3,        
    RETURN = 4          
  };
private:
  map<Address,Address> forcegoto;   
  vector<int4> deadcodedelay;       
  map<Address,Address> indirectover;    
  map<Address,FuncProto *> protoover;   
  vector<Address> multistagejump;   
  map<Address,uint4> flowoverride;  
  void clear(void);         
  static string generateDeadcodeDelayMessage(int4 index,Architecture *glb);
public:
  ~Override(void) { clear(); }      
  void insertForceGoto(const Address &targetpc,const Address &destpc);
  void insertDeadcodeDelay(AddrSpace *spc,int4 delay);
  bool hasDeadcodeDelay(AddrSpace *spc) const;
  void insertIndirectOverride(const Address &callpoint,const Address &directcall);
  void insertProtoOverride(const Address &callpoint,FuncProto *p);
  void insertMultistageJump(const Address &addr);
  void insertFlowOverride(const Address &addr,uint4 type);

  void applyPrototype(Funcdata &data,FuncCallSpecs &fspecs) const;
  void applyIndirect(Funcdata &data,FuncCallSpecs &fspecs) const;
  bool queryMultistageJumptable(const Address &addr) const;
  void applyDeadCodeDelay(Funcdata &data) const;
  void applyForceGoto(Funcdata &data) const;
  bool hasFlowOverride(void) const { return (!flowoverride.empty()); }  
  uint4 getFlowOverride(const Address &addr) const;
  void printRaw(ostream &s,Architecture *glb) const;
  void generateOverrideMessages(vector<string> &messagelist,Architecture *glb) const;
  void saveXml(ostream &s,Architecture *glb) const;
  void restoreXml(const Element *el,Architecture *glb);
  static string typeToString(uint4 tp);         
  static uint4 stringToType(const string &nm);      
};

#endif