load_method_by_regex.cpp

Go to the documentation of this file.

 
/*******************************************************************\
 
Module: Java Bytecode
 
Author: Diffblue Ltd.
 
\*******************************************************************/
 
#include "load_method_by_regex.h"
 
#include <util/prefix.h>
#include <util/symbol_table_base.h>
 
#include <regex>

static std::regex build_regex_from_pattern(const std::string &pattern)
{
  std::string modified_pattern = pattern;
  if(does_pattern_miss_descriptor(pattern))
    modified_pattern += R"(:\‍(.*\).*)";
 
  if(!has_prefix(pattern, "java::"))
    modified_pattern = "java::" + modified_pattern;
 
  return std::regex{modified_pattern};
}

bool does_pattern_miss_descriptor(const std::string &pattern)
{
  const size_t descriptor_index = pattern.rfind(':');
  if(descriptor_index == std::string::npos)
    return true;
 
  const std::string java_prefix = "java::";
  return descriptor_index == java_prefix.length() - 1 &&
         has_prefix(pattern, java_prefix);
}

std::function<std::vector<irep_idt>(const symbol_table_baset &symbol_table)>
build_load_method_by_regex(const std::string &pattern)
{
  std::regex regex = build_regex_from_pattern(pattern);
 
  return [=](const symbol_table_baset &symbol_table) {
    std::vector<irep_idt> matched_methods;
    for(const auto &symbol : symbol_table.symbols)
    {
      if(
        symbol.second.is_function() &&
        std::regex_match(id2string(symbol.first), regex))
      {
        matched_methods.push_back(symbol.first);
      }
    }
    return matched_methods;
  };
}