load_elf.h

Go to the documentation of this file.

 
#ifndef SIMPLEKERNEL_LOAD_ELF_H
#define SIMPLEKERNEL_LOAD_ELF_H
 
#include <array>
#include <elf.h>
#include <span>
#include <utility>
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include "efi.h"
#include "efilib.h"
 
#ifdef __cplusplus
}
#endif
 
class Graphics {
public:
  Graphics();
 
  ~Graphics() = default;
 
  Graphics(const Graphics &) = delete;
  Graphics(Graphics &&) = delete;
  auto operator=(const Graphics &) -> Graphics & = delete;
  auto operator=(Graphics &&) -> Graphics & = delete;
 
  void set_mode(
      EFI_GRAPHICS_PIXEL_FORMAT _format = PixelBlueGreenRedReserved8BitPerColor,
      uint32_t _width = DEFAULT_WIDTH, uint32_t _height = DEFAULT_HEIGHT) const;
 
  void print_info() const;
 
private:
  static constexpr const uint32_t DEFAULT_WIDTH = 1920;
  static constexpr const uint32_t DEFAULT_HEIGHT = 1080;
  EFI_GRAPHICS_OUTPUT_PROTOCOL *gop = nullptr;
};
 
class Memory {
public:
  Memory();
 
  ~Memory() = default;
 
  Memory(const Memory &) = delete;
  Memory(Memory &&) = delete;
  auto operator=(const Memory &) -> Memory & = delete;
  auto operator=(Memory &&) -> Memory & = delete;
 
  void print_info();
 
private:
  uint64_t desc_count = 0;
  EFI_MEMORY_DESCRIPTOR *memory_map = nullptr;
  uint64_t map_key = 0;
  uint64_t desc_size = 0;
  uint32_t desc_version = 0;
 
  void flush_desc();
};
 
class Elf {
public:
  explicit Elf(wchar_t *_kernel_image_filename);
 
  ~Elf();
 
  Elf(const Elf &) = delete;
  Elf(Elf &&) = delete;
  auto operator=(const Elf &) -> Elf & = delete;
  auto operator=(Elf &&) -> Elf & = delete;
 
  [[nodiscard]] auto load_kernel_image() const -> uint64_t;
 
  auto load() const -> uintptr_t;
 
private:
  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *file_system_protocol = nullptr;
  EFI_FILE *root_file_system = nullptr;
  EFI_FILE *elf = nullptr;
  size_t elf_file_size = 0;
  void *elf_file_buffer = nullptr;
  std::span<uint8_t> file = {};
  Elf64_Ehdr ehdr = {};
  std::span<Elf64_Phdr> phdr = {};
  std::span<Elf64_Shdr> shdr = {};
  static constexpr const size_t SECTION_BUF_SIZE = 1024;
  std::array<uint8_t, SECTION_BUF_SIZE> shstrtab_buf = {};
 
  [[nodiscard]] auto get_file_size() const -> size_t;
 
  [[nodiscard]] auto check_elf_identity() const -> bool;
 
  void get_ehdr();
 
  void print_ehdr() const;
 
  void get_phdr();
 
  void print_phdr() const;
 
  void get_shdr();
 
  void print_shdr() const;
 
  void load_sections(const Elf64_Phdr &_phdr) const;
 
  void load_program_sections() const;
};
 
#endif /* SIMPLEKERNEL_LOAD_ELF_H */