Import.hpp

/* Copyright 2017 - 2024 R. Thomas
 * Copyright 2017 - 2024 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 "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<ImportEntry>;
  using it_entries       = ref_iterator<entries_t&>;
  using it_const_entries = const_ref_iterator<const entries_t&>;
 
  Import(const details::pe_import& import);
  Import(std::string name) :
    name_(std::move(name))
  {}
  Import() = default;
  ~Import() override = default;
 
  Import(const Import& other) = default;
  Import(Import&& other) noexcept = default;
  Import& operator=(Import&& other) noexcept = default;
  Import& operator=(const Import& other) = 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 import_address_table_RVA_;
  }
 
  uint32_t import_lookup_table_rva() const {
    return import_lookup_table_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);
  }
 
  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);
 
  ImportEntry& add_entry(const std::string& name);
 
  void import_lookup_table_rva(uint32_t rva);
  void import_address_table_rva(uint32_t rva);
 
  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         import_lookup_table_RVA_ = 0;
  uint32_t         timedatestamp_ = 0;
  uint32_t         forwarder_chain_ = 0;
  uint32_t         name_RVA_ = 0;
  uint32_t         import_address_table_RVA_ = 0;
  std::string      name_;
  PE_TYPE          type_ = PE_TYPE::PE32;
};
 
}
}
 
#endif