java_class_loader.h

Go to the documentation of this file.

/*******************************************************************\
 
Module:
 
Author: Daniel Kroening, kroening@kroening.com
 
\*******************************************************************/
 
 
#ifndef CPROVER_JAVA_BYTECODE_JAVA_CLASS_LOADER_H
#define CPROVER_JAVA_BYTECODE_JAVA_CLASS_LOADER_H
 
#include <map>
 
#include <util/fixed_keys_map_wrapper.h>
 
#include "java_bytecode_parse_tree.h"
#include "java_class_loader_base.h"
 
class java_class_loader_limitt;
class message_handlert;
 
class java_class_loadert : public java_class_loader_baset
{
public:
  typedef std::list<java_bytecode_parse_treet> parse_tree_with_overlayst;
  typedef std::map<irep_idt, parse_tree_with_overlayst>
    parse_tree_with_overridest_mapt;
 
  typedef std::function<std::vector<irep_idt>(const irep_idt &)>
    get_extra_class_refs_functiont;
 
  java_class_loadert()
  {
  }
 
  parse_tree_with_overlayst &
  operator()(const irep_idt &class_name, message_handlert &);
 
  bool can_load_class(const irep_idt &class_name, message_handlert &);
 
  parse_tree_with_overlayst &get_parse_tree(
    java_class_loader_limitt &class_loader_limit,
    const irep_idt &class_name,
    message_handlert &);
 
  void set_java_cp_include_files(const std::string &cp_include_files)
  {
    java_cp_include_files = cp_include_files;
  }
  void set_extra_class_refs_function(get_extra_class_refs_functiont func)
  {
    get_extra_class_refs = func;
  }
  void add_load_classes(const std::vector<irep_idt> &classes)
  {
    for(const auto &id : classes)
      java_load_classes.push_back(id);
  }
 
  std::vector<irep_idt>
  load_entire_jar(const std::string &jar_path, message_handlert &);
 
  fixed_keys_map_wrappert<parse_tree_with_overridest_mapt>
  get_class_with_overlays_map()
  {
    return fixed_keys_map_wrappert<parse_tree_with_overridest_mapt>(class_map);
  }
  const java_bytecode_parse_treet &get_original_class(
    const irep_idt &class_name)
  {
    return class_map.at(class_name).front();
  }
 
private:
  std::string java_cp_include_files;
 
  std::vector<irep_idt> java_load_classes;
  get_extra_class_refs_functiont get_extra_class_refs;
 
  parse_tree_with_overridest_mapt class_map;
 
  std::optional<std::vector<irep_idt>>
  read_jar_file(const std::string &jar_path, message_handlert &);
};
 
#endif // CPROVER_JAVA_BYTECODE_JAVA_CLASS_LOADER_H