Feat: Add basic PE image methods

include/image.h

@@ -0,0 +1,152 @@
+/**
+ * @file image.h
+ * @brief basic PE image methods.
+ * 
+ * @author Chen Zhenshuo (chenzs108@outlook.com)
+ * @version 1.0
+ * @date 2020-01-07
+ * 
+ * @par GitHub
+ * https://github.com/czs108/
+ */
+
+#pragma once
+
+#include "extra_data.h"
+
+#include <windows.h>
+
+#include <stdbool.h>
+
+/**
+ * @brief Store PE image information, most methods are based on this structure.
+ * 
+ * @see https://docs.microsoft.com/en-us/windows/win32/debug/pe-format/
+ */
+typedef struct _PE_IMAGE_INFO
+{
+    //! The size of the PE image.
+    DWORD image_size;
+
+    //! The base address of the PE image when it is loaded in memory.
+    BYTE *image_base;
+
+    //! The base address of `IMAGE_NT_HEADERS` structure.
+    IMAGE_NT_HEADERS *nt_header;
+
+    //! The base address of the first `IMAGE_SECTION_HEADER` structure.
+    IMAGE_SECTION_HEADER *section_header;
+
+} PE_IMAGE_INFO;
+
+
+/**
+ * @brief Load the PE file to memory.
+ * 
+ * @public @memberof _PE_IMAGE_INFO
+ * 
+ * @param file_base         The base address of the file content.
+ * @param file_size         The size of the file.
+ * @param[out] image_info   The PE image related to the file.
+ * @param[out] extra_data   The extra data behind the PE image.
+ * @return `true` if the method succeeded, otherwise `false`.
+ * 
+ * @note The constructor of `_PE_IMAGE_INFO` structure.
+ * @note The constructor of `_EXTRA_DATA_VIEW` structure.
+ */
+bool LoadPeImage(
+    const BYTE *const file_base,
+    const DWORD file_size,
+    PE_IMAGE_INFO *const image_info,
+    EXTRA_DATA_VIEW *const extra_data);
+
+
+/**
+ * @brief Free the memory of the PE image.
+ * 
+ * @public @memberof _PE_IMAGE_INFO
+ * 
+ * @param[in, out] image_info   The PE image.
+ * 
+ * @note The destructor of `_PE_IMAGE_INFO` structure.
+ */
+void FreePeImage(
+    const PE_IMAGE_INFO *const image_info);
+
+
+/**
+ * @brief Write the PE image to a new file.
+ * 
+ * @public @memberof _PE_IMAGE_INFO
+ * 
+ * @param image_info    The PE image.
+ * @param file_name     The name of the new file.
+ * @return `true` if the method succeeded, otherwise `false`.
+ */
+bool WriteImageToFile(
+    const PE_IMAGE_INFO *const image_info,
+    const TCHAR *const file_name);
+
+
+/**
+ * @brief Convert the relative virtual address to a virtual address.
+ * 
+ * @public @memberof _PE_IMAGE_INFO
+ * 
+ * @param image_info    The PE image.
+ * @param rva           The relative virtual address.
+ * @return The virtual address.
+ */
+BYTE *RvaToVa(
+    const PE_IMAGE_INFO *const image_info,
+    const DWORD rva);
+
+
+/**
+ * @brief Check whether the PE image is x64 or x86.
+ * 
+ * @public @memberof _PE_IMAGE_INFO
+ * 
+ * @param image_info    The PE image.
+ * @return `true` if the image is a x64 file, otherwise `false`.
+ */
+bool IsPe64(
+    const PE_IMAGE_INFO *const image_info);
+
+
+/**
+ * @brief Align the value according to the alignment.
+ * 
+ * @param value     The value to be aligned.
+ * @param align     The value of alignment.
+ * @return DWORD    The value after alignment.
+ */
+DWORD Align(
+    const DWORD value,
+    const DWORD align);
+
+
+/**
+ * @brief Check if the file is a PE file.
+ * 
+ * @param file_base The base address of the file content.
+ * @return `true` if the file is a PE file, otherwise `false`.
+ */
+bool IsPeFile(
+    const BYTE *const file_base);
+
+
+/**
+ * @brief
+ * Calculate the size of PE headers,
+ * including `IMAGE_DOS_HEADER`, `IMAGE_NT_HEADERS` and all `IMAGE_SECTION_HEADER` structures.
+ * 
+ * @param file_base             The base address of the file content.
+ * @param[out] dos_header_size  The size of `IMAGE_DOS_HEADER` structure. It's optional.
+ * @param[out] nt_header_size   The size of `IMAGE_NT_HEADERS` structure. It's optional.
+ * @return The size of PE headers.
+ */
+DWORD CalcHeadersSize(
+    const BYTE *const file_base,
+    DWORD *const dos_header_size,
+    DWORD *const nt_header_size);

src/CMakeLists.txt

@@ -12,6 +12,9 @@ target_sources(PE-Packer
 
         ${CMAKE_CURRENT_LIST_DIR}/extra_data.c
         ${HEADER_PATH}/extra_data.h
+
+        ${CMAKE_CURRENT_LIST_DIR}/image.c
+        ${HEADER_PATH}/image.h
 )
 
 target_link_libraries(PE-Packer utility)