WPF 使用 SHGetFileInfo 造成內存泄漏問題

使用WPF自己做一個文件管理。其中，需要回去文件圖標進行顯示，使用win32接口進行，代碼如下

/// <summary>
		/// 獲取文件圖標需要的結構體，作爲出參，不需要初始化
		/// </summary>
		[StructLayout(LayoutKind.Sequential)]
		public struct SHFILEINFO
		{
			public IntPtr hIcon;                                        //文件的圖標句柄
			public int iIcon;                                              //文件圖標的系統索引號
			public uint dwAttributes;                                //文件的屬性值

			[MarshalAs(UnmanagedType.ByValTStr, SizeConst = 260)]
			public string szDisplayName;                           //文件的顯示名

			[MarshalAs(UnmanagedType.ByValTStr, SizeConst = 80)]
			public string szTypeName;                               //文件的類型名
		}

		/// <summary>
		/// 使用win32程序，查看文件信息，主要是獲取圖標，包括文件圖標，文件夾圖標，驅動器圖標
		/// </summary>
		/// <param name="strFilePath">文件路徑</param>
		/// <param name="dwFileAttributes">文件屬性，一般區分文件和文件夾</param>
		/// <param name="lpFileInfo">出參，保存圖標等信息的結構體</param>
		/// <param name="cbFileInfoSize">結構體大小</param>
		/// <param name="uFlags">核心變量，通過不同的標誌獲取不同的信息</param>
		/// <returns></returns>
		[DllImport("shell32.dll", SetLastError = true)]
		public static extern int SHGetFileInfo(string strFilePath, uint dwFileAttributes, ref SHFILEINFO lpFileInfo, uint cbFileInfoSize, uint uFlags);

		[DllImport("user32.dll", SetLastError = true)]
		public static extern bool DestroyIcon(IntPtr hIcon);

		//獲取圖標
		private const uint SHGFI_ICON = 0x100;

		//大圖標 32 x 32
		private const uint SHGFI_LARGEICON = 0x0;

		//小圖標 16 x 16
		private const uint SHGFI_SMALLICON = 0x1;

		//使用use passed dwFileAttribute
		private const uint SHGFI_USEFILEATTRIBUTES = 0x10;

		private const uint FILE_ATTRIBUTE_NORMAL = 0x80;
		private const uint FILE_ATTRIBUTE_DIRECTORY = 0x10;
		private const uint SHGFI_DISPLAYNAME = 0x200;
		private const uint SHGFI_SYSICONINDEX = 0x400;

		/// <summary>
		/// 自定義函數，獲取文件的圖標，可以指定大小圖標，或者文件夾圖標
		/// </summary>
		/// <param name="strFilePath">文件名</param>
		/// <param name="bSmallOrLarge">true 小圖標  false 大圖標</param>
		/// <param name="bDirectory">true  文件夾  false 文件</param>
		/// <returns></returns>
		public static ImageSource GetIcon(string strFilePath, bool bSmallOrLarge, bool bDirectory)
		{
			uint uFlag = SHGFI_ICON | SHGFI_USEFILEATTRIBUTES | SHGFI_DISPLAYNAME;
			if (bSmallOrLarge)
				uFlag |= SHGFI_SMALLICON;

			uint uAttribute = FILE_ATTRIBUTE_NORMAL;
			if (bDirectory)
				uAttribute |= FILE_ATTRIBUTE_DIRECTORY;

			SHFILEINFO fileInfo = new SHFILEINFO();

			if (0 != SHGetFileInfo(strFilePath, uAttribute, ref fileInfo, (uint)Marshal.SizeOf(typeof(SHFILEINFO)), uFlag))
			{
				if(fileInfo.hIcon != IntPtr.Zero)
				{
					BitmapSource bmpSource = System.Windows.Interop.Imaging.CreateBitmapSourceFromHIcon(fileInfo.hIcon, Int32Rect.Empty, BitmapSizeOptions.FromEmptyOptions());
					return bmpSource;
				}
			}

			return null;
		}

然後運行界面，進行基本的文件顯示和操作。

 

過一段時間之後，界面卡死了，查看任務管理器，發現 GDI 佔用高達9999

 

網上搜了一下，原來需要對 SHGetFileInfo 獲取到的圖標進行釋放。否則，會一直存在內存中，系統不會自己進行回收。

 

修改代碼如下：

/// <summary>
		/// 自定義函數，獲取文件的圖標，可以指定大小圖標，或者文件夾圖標
		/// </summary>
		/// <param name="strFilePath">文件名</param>
		/// <param name="bSmallOrLarge">true 小圖標  false 大圖標</param>
		/// <param name="bDirectory">true  文件夾  false 文件</param>
		/// <returns></returns>
		public static ImageSource GetIcon(string strFilePath, bool bSmallOrLarge, bool bDirectory)
		{
			uint uFlag = SHGFI_ICON | SHGFI_USEFILEATTRIBUTES | SHGFI_DISPLAYNAME;
			if (bSmallOrLarge)
				uFlag |= SHGFI_SMALLICON;

			uint uAttribute = FILE_ATTRIBUTE_NORMAL;
			if (bDirectory)
				uAttribute |= FILE_ATTRIBUTE_DIRECTORY;

			SHFILEINFO fileInfo = new SHFILEINFO();

			if (0 != SHGetFileInfo(strFilePath, uAttribute, ref fileInfo, (uint)Marshal.SizeOf(typeof(SHFILEINFO)), uFlag))
			{
				if(fileInfo.hIcon != IntPtr.Zero)
				{
					BitmapSource bmpSource = System.Windows.Interop.Imaging.CreateBitmapSourceFromHIcon(fileInfo.hIcon, Int32Rect.Empty, BitmapSizeOptions.FromEmptyOptions());
					DestroyIcon(fileInfo.hIcon);
					return bmpSource;
				}
			}

			return null;
		}

注意，增加了這樣一條：

DestroyIcon(fileInfo.hIcon);

 

然後，GDI保持在正常的水平，不會繼續增長了。

 

點擊這裏可以下載文件管理代碼