COFF/Parser.hpp

Go to the documentation of this file.

/* Copyright 2017 - 2026 R. Thomas
 * Copyright 2017 - 2026 Quarkslab
 *
 * 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 LIEF_COFF_PARSER_H
#define LIEF_COFF_PARSER_H
#include "LIEF/BinaryStream/BinaryStream.hpp"
#include "LIEF/compiler_attributes.hpp"
#include "LIEF/errors.hpp"
#include "LIEF/path.hpp"
#include "LIEF/visibility.h"
 
#include <string_view>
#include <map>
 
#include "LIEF/COFF/Header.hpp"
#include "LIEF/COFF/ParserConfig.hpp"
 
 
namespace LIEF::COFF {
class Binary;
class Section;
class String;
class Symbol;
 
class Parser {
  public:
  static constexpr size_t MAX_NB_SECTIONS = 1 << 20;

  static LIEF_API std::unique_ptr<Binary>
      parse(std::unique_ptr<BinaryStream> stream,
            const ParserConfig& config = ParserConfig::default_conf());

  static LIEF_API std::unique_ptr<Binary>
      parse(std::string_view file,
            const ParserConfig& config = ParserConfig::default_conf());

  template<class PathT, enable_if_path_t<PathT> = 0>
  static std::unique_ptr<Binary>
      parse(const PathT& file,
            const ParserConfig& config = ParserConfig::default_conf()) {
    return parse(file.string(), config);
  }

  struct SymSec {
    size_t sec_idx = 0;
    Symbol* symbol = nullptr;
 
    friend bool operator<(SymSec lhs, SymSec rhs) {
      return lhs.sec_idx < rhs.sec_idx;
    }
 
    friend bool operator==(SymSec lhs, SymSec rhs) {
      return lhs.sec_idx == rhs.sec_idx && lhs.symbol == rhs.symbol;
    }
 
    friend bool operator!=(SymSec lhs, SymSec rhs) {
      return !(lhs == rhs);
    }
  };

  using SymSecMap = std::vector<SymSec>;

  LIEF_LOCAL void memoize(String str);

  LIEF_LOCAL String* find_coff_string(uint32_t offset) const;
 
  ~Parser();
 
  private:
  Parser(std::unique_ptr<BinaryStream> stream, const ParserConfig& config,
         Header::KIND kind);
 
  ok_error_t process();
  ok_error_t parse_header();
  ok_error_t parse_optional_header();
  ok_error_t parse_sections();
  ok_error_t parse_relocations(Section& section);
  ok_error_t parse_symbols();
  ok_error_t parse_string_table();
 
  std::unique_ptr<BinaryStream> stream_;
  std::unique_ptr<Binary> bin_;
  Header::KIND kind_ = Header::KIND::UNKNOWN;
 
  std::map<uint32_t, size_t> memoize_coff_str_;
  std::map<size_t, Symbol*> symbol_idx_;
  SymSecMap symsec_;
 
  LIEF_MAYBE_UNUSED ParserConfig config_;
};
}
 
#endif