java_class_loader_limit.cpp

Go to the documentation of this file.

/*******************************************************************\
 
Module: limit class path loading
 
Author: Daniel Kroening, kroening@kroening.com
 
\*******************************************************************/
 
 
#include "java_class_loader_limit.h"
 
#include <json/json_parser.h>
 
#include <util/invariant.h>
 
void java_class_loader_limitt::setup_class_load_limit(
  const std::string &java_cp_include_files)
{
  PRECONDITION(!java_cp_include_files.empty());
 
  // '@' signals file reading with list of class files to load
  use_regex_match = java_cp_include_files[0] != '@';
  if(use_regex_match)
  {
    regex_matcher=std::regex(java_cp_include_files);
    log.debug() << "Limit loading to classes matching '"
                << java_cp_include_files << "'" << messaget::eom;
  }
  else
  {
    PRECONDITION(java_cp_include_files.length() > 1);
    jsont json_cp_config;
    if(parse_json(
         java_cp_include_files.substr(1),
         log.get_message_handler(),
         json_cp_config))
      throw "cannot read JSON input configuration for JAR loading";
    if(!json_cp_config.is_object())
      throw "the JSON file has a wrong format";
    jsont include_files=json_cp_config["classFiles"];
    if(!include_files.is_null() && !include_files.is_array())
      throw "the JSON file has a wrong format";
    for(const jsont &file_entry : to_json_array(include_files))
    {
      PRECONDITION(file_entry.is_string());
      set_matcher.insert(file_entry.value);
    }
  }
}
 
bool java_class_loader_limitt::load_class_file(const std::string &file_name)
{
  if(use_regex_match)
  {
    std::smatch string_matches;
    if(std::regex_match(file_name, string_matches, regex_matcher))
      return true;
    log.debug() << file_name + " discarded since not matching loader regexp"
                << messaget::eom;
    return false;
  }
  else
    // load .class file only if it is in the match set
    return set_matcher.find(file_name) != set_matcher.end();
}