inject_ghidra.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 __INJECT_GHIDRA__
#define __INJECT_GHIDRA__

#include "pcodeinject.hh"
#include "ghidra_arch.hh"

class InjectContextGhidra : public InjectContext {
public:
  virtual void saveXml(ostream &s) const;
};

class InjectPayloadGhidra : public InjectPayload {
  string source;        
public:
  InjectPayloadGhidra(const string &src,const string &nm,int4 tp) : InjectPayload(nm,tp) { source = src; }  
  virtual void inject(InjectContext &context,PcodeEmit &emit) const;
  virtual void restoreXml(const Element *el) {}
  virtual void printTemplate(ostream &s) const;
  virtual string getSource(void) const { return source; }
};

class InjectCallfixupGhidra : public InjectPayloadGhidra {
public:
  InjectCallfixupGhidra(const string &src,const string &nm);    
  virtual void restoreXml(const Element *el);
};

class InjectCallotherGhidra : public InjectPayloadGhidra {
public:
  InjectCallotherGhidra(const string &src,const string &nm);    
  virtual void restoreXml(const Element *el);
};

class ExecutablePcodeGhidra : public ExecutablePcode {
public:
  ExecutablePcodeGhidra(Architecture *g,const string &src,const string &nm);    
  virtual void inject(InjectContext &context,PcodeEmit &emit) const;
  virtual void restoreXml(const Element *el);
  virtual void printTemplate(ostream &s) const;
};

class PcodeInjectLibraryGhidra : public PcodeInjectLibrary {
  InjectContextGhidra contextCache;     
  vector<OpBehavior *> inst;            
  virtual int4 allocateInject(const string &sourceName,const string &name,int4 type);
  virtual void registerInject(int4 injectid);
public:
  PcodeInjectLibraryGhidra(ArchitectureGhidra *ghi);        
  virtual int4 manualCallFixup(const string &name,const string &snippet);
  virtual int4 manualCallOtherFixup(const string &name,const string &outname,const vector<string> &inname,
                    const string &snippet);
  virtual InjectContext &getCachedContext(void) { return contextCache; }
  virtual const vector<OpBehavior *> &getBehaviors(void);
};

#endif