stringmanage.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 __STRING_MANAGE__
#define __STRING_MANAGE__
 
#include "type.hh"
 
class Architecture;
 
class StringManager {
protected:
  class StringData {
  public:
    bool isTruncated;       
    vector<uint1> byteData; 
  };
  map<Address,StringData> stringMap;    
  int4 maximumChars;            
public:
  StringManager(int4 max);      
  virtual ~StringManager(void);     
 
  void clear(void) { stringMap.clear(); }           
 
  bool isString(const Address &addr,Datatype *charType);    // Determine if data at the given address is a string
 
  virtual const vector<uint1> &getStringData(const Address &addr,Datatype *charType,bool &isTrunc)=0;
 
  void saveXml(ostream &s) const;   
  void restoreXml(const Element *el,const AddrSpaceManager *m); 
 
  static bool hasCharTerminator(const uint1 *buffer,int4 size,int4 charsize);   
  static int4 readUtf16(const uint1 *buf,bool bigend);  
  static void writeUtf8(ostream &s,int4 codepoint); 
  static int4 getCodepoint(const uint1 *buf,int4 charsize,bool bigend,int4 &skip);  
};
 
class StringManagerUnicode : public StringManager {
  Architecture *glb;        
  uint1 *testBuffer;        
  int4 checkCharacters(const uint1 *buf,int4 size,int4 charsize) const; 
public:
  StringManagerUnicode(Architecture *g,int4 max);   
  virtual ~StringManagerUnicode(void);
 
  virtual const vector<uint1> &getStringData(const Address &addr,Datatype *charType,bool &isTrunc);
  bool writeUnicode(ostream &s,uint1 *buffer,int4 size,int4 charsize);  
};
 
#endif