java_class_loader_base.cpp

Go to the documentation of this file.

/*******************************************************************\
 
Module:
 
Author: Daniel Kroening, kroening@kroening.com
 
\*******************************************************************/
 
#include "java_class_loader_base.h"
 
#include <util/message.h>
#include <util/prefix.h>
#include <util/suffix.h>
 
#include "jar_file.h"
#include "java_bytecode_parse_tree.h"
#include "java_bytecode_parser.h"
 
#include <filesystem>
#include <fstream>
 
void java_class_loader_baset::add_classpath_entry(
  const std::string &path,
  message_handlert &message_handler)
{
  messaget log(message_handler);
 
  if(has_suffix(path, ".jar"))
  {
    if(std::ifstream(path).good())
    {
      classpath_entries.push_back(
        classpath_entryt(classpath_entryt::JAR, path));
    }
    else
    {
      log.warning() << "Warning: failed to access JAR file "
                    << messaget::quote_begin << path << messaget::quote_end
                    << messaget::eom;
    }
  }
  else
  {
    if(std::filesystem::is_directory(path))
    {
      classpath_entries.push_back(
        classpath_entryt(classpath_entryt::DIRECTORY, path));
    }
    else
    {
      log.warning() << "Warning: failed to access directory "
                    << messaget::quote_begin << path << messaget::quote_end
                    << messaget::eom;
    }
  }
}
 
std::string java_class_loader_baset::file_to_class_name(const std::string &file)
{
  std::string result = file;
 
  // Strip .class. Note that the Java class loader would
  // not do that.
  if(has_suffix(result, ".class"))
    result.resize(result.size() - 6);
 
  // Strip a "./" prefix. Note that the Java class loader
  // would not do that.
#ifdef _WIN32
  while(has_prefix(result, ".\\"))
    result = std::string(result, 2, std::string::npos);
#else
  while(has_prefix(result, "./"))
    result = std::string(result, 2, std::string::npos);
#endif
 
  // slash to dot
  for(std::string::iterator it = result.begin(); it != result.end(); it++)
    if(*it == '/')
      *it = '.';
 
  return result;
}
 
std::string
java_class_loader_baset::class_name_to_jar_file(const irep_idt &class_name)
{
  std::string result = id2string(class_name);
 
  // dots (package name separators) to slash
  for(std::string::iterator it = result.begin(); it != result.end(); it++)
    if(*it == '.')
      *it = '/';
 
  // add .class suffix
  result += ".class";
 
  return result;
}
 
std::string
java_class_loader_baset::class_name_to_os_file(const irep_idt &class_name)
{
  std::string result = id2string(class_name);
 
  // dots (package name separators) to slash, depending on OS
  for(std::string::iterator it = result.begin(); it != result.end(); it++)
    if(*it == '.')
    {
#ifdef _WIN32
      *it = '\\';
#else
      *it = '/';
#endif
    }
 
  // add .class suffix
  result += ".class";
 
  return result;
}
 
std::optional<java_bytecode_parse_treet> java_class_loader_baset::load_class(
  const irep_idt &class_name,
  const classpath_entryt &cp_entry,
  message_handlert &message_handler)
{
  switch(cp_entry.kind)
  {
  case classpath_entryt::JAR:
    return get_class_from_jar(class_name, cp_entry.path, message_handler);
 
  case classpath_entryt::DIRECTORY:
    return get_class_from_directory(class_name, cp_entry.path, message_handler);
  }
 
  UNREACHABLE;
}
 
std::optional<java_bytecode_parse_treet>
java_class_loader_baset::get_class_from_jar(
  const irep_idt &class_name,
  const std::string &jar_file,
  message_handlert &message_handler)
{
  messaget log(message_handler);
 
  try
  {
    auto &jar = jar_pool(jar_file);
    auto data = jar.get_entry(class_name_to_jar_file(class_name));
 
    if(!data.has_value())
      return {};
 
    log.debug() << "Getting class '" << class_name << "' from JAR " << jar_file
                << messaget::eom;
 
    std::istringstream istream(*data);
    return java_bytecode_parse(istream, class_name, message_handler);
  }
  catch(const std::runtime_error &)
  {
    log.error() << "failed to open JAR file '" << jar_file << "'"
                << messaget::eom;
    return {};
  }
}
 
std::optional<java_bytecode_parse_treet>
java_class_loader_baset::get_class_from_directory(
  const irep_idt &class_name,
  const std::string &path,
  message_handlert &message_handler)
{
  // Look in the given directory
  const std::string class_file = class_name_to_os_file(class_name);
  const std::string full_path =
    std::filesystem::path(path).append(class_file).string();
 
  if(std::ifstream(full_path))
  {
    messaget log(message_handler);
    log.debug() << "Getting class '" << class_name << "' from file "
                << full_path << messaget::eom;
    return java_bytecode_parse(full_path, class_name, message_handler);
  }
  else
    return {};
}