C/C++ 實現動態資源文件釋放

當我們開發Windows應用程序時,通常會涉及到使用資源(Resource)的情況。資源可以包括圖標、位圖、字符串等,它們以二進制形式嵌入到可執行文件中。在某些情況下,我們可能需要從可執行文件中提取自定義資源並保存爲獨立的文件。在這篇博客文章中,我們將討論如何使用C++和WinAPI實現這個目標。

簡介

首先,讓我們考慮一個場景:我們有一個 Windows 應用程序,其中包含了一個自定義的二進制資源比如默認的配置文件,我們希望將這個資源提取出來並保存爲一個獨立的文件以用於初始化程序配置項。爲了實現這個目標,我們可以使用Windows API提供的相關函數,來完成對資源的釋放工作。

關鍵函數概述

GetModuleHandle

該函數用於獲取指定模塊的句柄。模塊可以是一個可執行文件(例如 .exe 文件)或一個動態鏈接庫(例如 .dll 文件)。該函數返回指定模塊的實例句柄,以便在後續的操作中使用。

以下是 GetModuleHandle 函數的一般形式:

HMODULE GetModuleHandle(
  LPCTSTR lpModuleName
);

參數說明:

  • lpModuleName:指定要獲取句柄的模塊的名稱。如果爲 NULL,則返回調用線程的可執行模塊句柄。

在許多情況下,GetModuleHandle 主要用於獲取當前進程的模塊句柄,以便在後續的操作中使用該句柄。模塊句柄通常用於在進程中查找資源、定位函數地址等目的。

FindResource

該函數用於定位並返回指定模塊(通常是 .exe 或 .dll 文件)中的資源。資源可以是諸如位圖、圖標、對話框模板、字符串等等的數據。

以下是 FindResource 函數的一般形式:

HRSRC FindResource(
  HMODULE hModule,
  LPCTSTR lpName,
  LPCTSTR lpType
);

參數說明:

  • hModule:指定包含資源的模塊的句柄。如果爲 NULL,則表示使用當前可執行模塊的句柄。
  • lpName:指定資源的名稱或標識符。可以是字符串或整數標識符。
  • lpType:指定資源的類型。通常是一個字符串,如 "RT_BITMAP" 表示位圖資源。

如果找到,則返回指向資源的句柄(HRSRC)。這個句柄可以用於後續的資源加載和操作,函數的第二個參數經常配合MAKEINTRESOURCE一起使用,MAKEINTRESOURCE 是一個宏(macro),用於將整數標識符(ID)轉換爲字符串指針。在 Windows 編程中,通常用於標識資源的 ID。

#define MAKEINTRESOURCE(i) ((LPCTSTR)((DWORD)((WORD)(i))))

這個宏接受一個整數參數 i,然後將其轉換爲字符串指針。在資源標識符上下文中,通常將整數標識符轉換爲字符串是爲了在使用相關資源函數時傳遞正確的參數。

舉個例子,如果有一個字符串資源的標識符是 IDR_MYSTRING,則可以使用 MAKEINTRESOURCE 將其轉換爲字符串:

LPCTSTR pszResourceName = MAKEINTRESOURCE(IDR_MYSTRING);

在這裏,pszResourceName 將指向字符串 "IDR_MYSTRING"。

在前面提到的 FindResource 中,通常將 MAKEINTRESOURCE(IDR_MYSTRING) 作爲 lpName 參數傳遞給 FindResource。這是因爲 FindResource 函數期望資源名稱是字符串類型,而 IDR_MYSTRING 可能是一個整數標識符。通過使用 MAKEINTRESOURCE,則可以將整數標識符轉換爲字符串,以便正確地在資源中查找。

SizeofResource

該函數用於獲取指定資源的大小。它返回資源的字節數,可以用於確定加載資源所需的內存大小。

以下是 SizeofResource 函數的一般形式:

DWORD SizeofResource(
  HMODULE hModule,
  HRSRC   hResInfo
);

參數說明:

  • hModule:指定包含資源的模塊的句柄。如果爲 NULL,則表示使用當前可執行模塊的句柄。
  • hResInfo:指定資源的句柄,通常由 FindResource 返回。

SizeofResource 返回資源的大小,以字節爲單位。這個函數在加載資源之前可以用來分配足夠的內存空間。

LoadResource

該函數用於加載指定資源的數據。該函數返回一個全局內存塊的句柄,該內存塊包含了資源的實際數據,你可以通過 LockResource 函數獲取該內存塊的指針來訪問資源數據。

以下是 LoadResource 函數的一般形式:

HGLOBAL LoadResource(
  HMODULE hModule,
  HRSRC   hResInfo
);

參數說明:

  • hModule:指定包含資源的模塊的句柄。如果爲 NULL,則表示使用當前可執行模塊的句柄。
  • hResInfo:指定資源的句柄,通常由 FindResource 返回。

LoadResource 用於將資源數據加載到全局內存塊中,並返回該內存塊的句柄。在加載資源後,可以使用 LockResource 函數獲取指向資源數據的指針。

LockResource

用於獲取指定資源的數據指針。它接受一個全局內存塊的句柄,該內存塊通常由 LoadResource 函數返回,然後返回一個指向資源數據的指針。

以下是 LockResource 函數的一般形式:

LPVOID LockResource(
  HGLOBAL hResData
);

參數說明:

  • hResData:指定資源數據的全局內存塊句柄,通常由 LoadResource 函數返回。

LockResource 用於鎖定指定資源的全局內存塊,並返回指向資源數據的指針。請注意,這個函數實際上並不執行拷貝,而是返回指向內存塊的指針,因此對返回指針的任何修改都會直接影響到內存塊本身。

FreeResource

用於釋放由 LoadResource 函數加載的資源。這個函數通常用於釋放不再需要的資源,以防止資源泄漏。

以下是 FreeResource 函數的一般形式:

BOOL FreeResource(
  HGLOBAL hResData
);

參數說明:

  • hResData:指定要釋放的全局內存塊句柄,通常由 LoadResource 函數返回。

FreeResource 用於釋放之前由 LoadResource 加載的資源。請注意,這個函數通常在資源的生命週期結束時調用,以確保釋放資源佔用的內存。但在實際應用中,現代 Windows 應用通常不需要顯式調用 FreeResource,因爲 Windows 會在程序退出時自動釋放資源。

在實際的應用程序中,FindResource 可以與 LoadResourceLockResource 等函數一起使用,用於加載和操作資源數據。當數據資源被加載到內存之後則可以直接通過fwrite函數將其直接寫出到磁盤中,以此來實現釋放資源的目的。

代碼功能實現

首先新建一個控制檯程序以作爲本次的測試環境,接着準備好我們需要寫出的數據,這裏就準備一個lyshark.ini配置文件,在項目中右鍵選擇添加並添加資源,此時會彈出如下圖所示的提示信息;

此時會彈出添加資源菜單,通過點擊導入按鈕並輸入資源類型爲LYSHARK點擊確定保存這個更改,如下圖所示;

此時我們在主程序中引入#include "resource.h"包含資源頭文件,並修改FindResource中的特定位置使其指向我們導入的配置文件,在釋放時同樣需要保持fopen("map\\lyshark.ini", "wb+")配置文件的格式。

這段資源釋放的完整代碼如下所示;

#define _CRT_SECURE_NO_WARNINGS
#include <Windows.h>
#include <iostream>
#include <WinUser.h>
#include "resource.h"

BOOL UseCustomResource()
{
	// 定位我們的自定義資源
	HMODULE hModule = GetModuleHandle(NULL);
	if (hModule == NULL)
	{
		std::cerr << "錯誤:獲取模塊句柄失敗。" << std::endl;
		return FALSE;
	}

	HRSRC hRsrc = FindResource(hModule, MAKEINTRESOURCE(IDR_LYSHARK1), TEXT("LYSHARK"));
	if (hRsrc == NULL)
	{
		std::cerr << "錯誤:無法找到資源。" << std::endl;
		return FALSE;
	}

	// 獲取資源大小
	DWORD dwSize = SizeofResource(hModule, hRsrc);
	if (dwSize == 0)
	{
		std::cerr << "錯誤:無效的資源大小。" << std::endl;
		return FALSE;
	}

	// 加載資源
	HGLOBAL hGlobal = LoadResource(hModule, hRsrc);
	if (hGlobal == NULL)
	{
		std::cerr << "錯誤:無法加載資源。" << std::endl;
		return FALSE;
	}

	// 鎖定資源
	LPVOID lpVoid = LockResource(hGlobal);
	if (lpVoid == NULL)
	{
		std::cerr << "錯誤:無法鎖定資源。" << std::endl;
		FreeResource(hGlobal);  // 在返回前釋放資源
		return FALSE;
	}

	// 如果不存在,創建一個“map”目錄
	if (!CreateDirectory("map", NULL) && GetLastError() != ERROR_ALREADY_EXISTS)
	{
		std::cerr << "錯誤:無法創建目錄。" << std::endl;
		FreeResource(hGlobal);
		return FALSE;
	}

	// 將資源寫入文件
	FILE* fp = fopen("map\\lyshark.ini", "wb+");
	if (fp == NULL)
	{
		std::cerr << "錯誤:無法創建或打開文件。" << std::endl;
		FreeResource(hGlobal);
		return FALSE;
	}

	fwrite(lpVoid, sizeof(char), dwSize, fp);
	fclose(fp);

	// 釋放資源
	FreeResource(hGlobal);

	return TRUE;
}

int main(int argc, char* argv[])
{
	BOOL ref = UseCustomResource();
	std::cout << "釋放狀態: " << ref << std::endl;

	system("pause");
	return 0;
}

以管理員模式運行上述程序,並等待,此時會釋放一個目錄幷包含一個配置文件,如下圖所示的輸出結果;

結語

通過以上的代碼實現,我們成功地將自定義資源提取並保存爲一個獨立的文件。這種技術在一些特殊情況下可能會很有用,例如需要動態加載或替換資源的情況。希望這篇博客對你理解如何使用 C++ 和 Windows API 進行資源操作有所幫助。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章