Import.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_PE_IMPORT_H
#define LIEF_PE_IMPORT_H
 
#include <string>
#include <ostream>
#include <memory>
 
#include "LIEF/errors.hpp"
#include "LIEF/Object.hpp"
#include "LIEF/visibility.h"
#include "LIEF/iterators.hpp"
#include "LIEF/PE/ImportEntry.hpp"
 
namespace LIEF {
namespace PE {
class Parser;
class Builder;
class DataDirectory;
 
namespace details {
struct pe_import;
}

class LIEF_API Import : public Object {
 
  friend class Parser;
  friend class Builder;
 
  public:
  using entries_t = std::vector<std::unique_ptr<ImportEntry>>;
  using it_entries = ref_iterator<entries_t&, ImportEntry*>;
  using it_const_entries =
      const_ref_iterator<const entries_t&, const ImportEntry*>;
 
  Import(const details::pe_import& import);
  Import(std::string name) :
    name_(std::move(name)) {}
  Import() = default;
  ~Import() override = default;
 
  Import(const Import& other);
  Import& operator=(const Import& other);
 
  Import(Import&& other) noexcept = default;
  Import& operator=(Import&& other) noexcept = default;

  uint32_t forwarder_chain() const {
    return forwarder_chain_;
  }

  uint32_t timedatestamp() const {
    return timedatestamp_;
  }

  it_const_entries entries() const {
    return entries_;
  }
 
  it_entries entries() {
    return entries_;
  }

  uint32_t import_address_table_rva() const {
    return iat_rva_;
  }

  uint32_t import_lookup_table_rva() const {
    return ilt_rva_;
  }

  result<uint32_t> get_function_rva_from_iat(const std::string& function) const;

  ImportEntry* get_entry(const std::string& name) {
    return const_cast<ImportEntry*>(
        static_cast<const Import*>(this)->get_entry(name)
    );
  }
  const ImportEntry* get_entry(const std::string& name) const;

  const std::string& name() const {
    return name_;
  }

  void name(std::string name) {
    name_ = std::move(name);
  }

  uint32_t name_rva() const {
    return name_rva_;
  }

  DataDirectory* directory() {
    return directory_;
  }
 
  const DataDirectory* directory() const {
    return directory_;
  }

  DataDirectory* iat_directory() {
    return iat_directory_;
  }
 
  const DataDirectory* iat_directory() const {
    return iat_directory_;
  }

  ImportEntry& add_entry(const ImportEntry& entry) {
    entries_.emplace_back(new ImportEntry(entry));
    return *entries_.back();
  }

  ImportEntry& add_entry(const std::string& name) {
    entries_.emplace_back(new ImportEntry(name));
    return *entries_.back();
  }

  bool remove_entry(const std::string& name);

  bool remove_entry(uint32_t ordinal);
 
  void import_lookup_table_rva(uint32_t rva) {
    ilt_rva_ = rva;
  }
 
  void import_address_table_rva(uint32_t rva) {
    iat_rva_ = rva;
  }

  LIEF_LOCAL size_t nb_original_func() const {
    return nb_original_func_;
  }
 
  void accept(Visitor& visitor) const override;
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const Import& entry);
 
  private:
  entries_t entries_;
  DataDirectory* directory_ = nullptr;
  DataDirectory* iat_directory_ = nullptr;
  uint32_t ilt_rva_ = 0;
  uint32_t timedatestamp_ = 0;
  uint32_t forwarder_chain_ = 0;
  uint32_t name_rva_ = 0;
  uint32_t iat_rva_ = 0;
  std::string name_;
  PE_TYPE type_ = PE_TYPE::PE32;
  size_t nb_original_func_ = 0;
};
 
}
}
 
#endif