【框架-MFC】MFC 识别U盘，获取U盘的名称、序列号、文件系统、容量、已用空间、剩余空间

于 2016-04-05 15:53:51 发布
阅读量4.7k 收藏 5
点赞数 1
分类专栏： 语言-C++
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/chenlu5201314/article/details/51065307
版权 
	enum 
	{
		DiskSize_B = 0,
		DiskSize_K = 1,
		DiskSize_M = 2,
		DiskSize_G = 3,
		DiskSize_T = 4,
	};
	enum 
	{
		GDI_VOLUMENAME = 0,
		GDI_VOLUMESERIALNUMBER = 1,
		GDI_VOLUMEFILESYSTEM = 2,
		GDI_VOLUMETYPE = 3,
		GDI_VOLUMESIZE = 4,
		GDI_VOLUMEFREESIZE = 5,
	};
CString GetDiskSize(ULONGLONG nSize,int nType);
	CString GetDriverInfomation(BYTE bAlpha,int iGTI_TYPE);

CString GetDiskSize( ULONGLONG nSize,int nType )
	{
		CString tmp;
		if (nType == DiskSize_B)
		{
			ULONGLONG nG ,nM,nK,nB,nKB,nMKB,nT,nGMKB;
			nT =0;
			//nT = nSize/(ULONGLONG)(1024*1024*1024*1024);
			//nGMKB =  nSize%(ULONGLONG)(1024*1024*1024*1024);
			nG = nSize/(ULONGLONG)(1000*1000*1000);
			nMKB =  nSize%(ULONGLONG)(1000*1000*1000);
			nM = nMKB/(ULONGLONG)(1000*1000);
			nKB =nMKB%(ULONGLONG)(1000*1000);
			nK = nKB/(ULONGLONG)(1000);
			nB =nKB%(ULONGLONG)(1000);

			if ((int)nT != 0)
			{
				tmp.Format(L"%d,%03d,%03d,%03d,%03d B",(int)nT, (int)nG,(int)nM,(int)nK,(int)nB	);
			}
			else if ((int)nT == 0 &&(int)nG != 0 )
			{
				tmp.Format(L"%d,%03d,%03d,%03d B", (int)nG,(int)nM,(int)nK,(int)nB	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM != 0 )
			{
				tmp.Format(L"%d,%03d,%03d B",(int)nM,(int)nK,(int)nB	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM == 0 &&(int)nK != 0 )
			{
				tmp.Format(L"%d,%03d B",(int)nK,(int)nB	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM == 0 &&(int)nK == 0  &&(int)nB != 0 )
			{
				tmp.Format(L"%d B",(int)nB	);
			}
			else
			{
				tmp.Format(L"0 B");
			}
		}
		else if (nType == DiskSize_K)
		{
			ULONGLONG nG ,nM,nK,nB,nKB,nMKB,nT,nGMKB;
			nT =0;
			//nT = nSize/(ULONGLONG)(1024*1024*1024*1024);
			//nGMKB =  nSize%(ULONGLONG)(1024*1024*1024*1024);
			nG = nSize/(ULONGLONG)(1000*1000*1024);
			nMKB =  nSize%(ULONGLONG)(1000*1000*1024);
			nM = nMKB/(ULONGLONG)(1000*1024);
			nKB =nMKB%(ULONGLONG)(1000*1024);
			nK = nKB/(ULONGLONG)(1024);
			nB =nKB%(ULONGLONG)(1024);

			if ((int)nT != 0)
			{
				tmp.Format(L"%d,%03d,%03d,%03d.%02d K",(int)nT, (int)nG,(int)nM,(int)nK,(int)(((float)(int)nB)/10.24)	);
			}
			else if ((int)nT == 0 &&(int)nG != 0 )
			{
				tmp.Format(L"%d,%03d,%03d.%02d K", (int)nG,(int)nM,(int)nK,(int)(((float)(int)nB)/10.24)	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM != 0 )
			{
				tmp.Format(L"%d,%03d.%02d K",(int)nM,(int)nK,(int)(((float)(int)nB)/10.24)	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM == 0 &&(int)nK != 0 )
			{
				tmp.Format(L"%d.%02d K",(int)nK,(int)(((float)(int)nB)/10.24)	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM == 0 &&(int)nK == 0  &&(int)nB != 0 )
			{
				tmp.Format(L"0.%02d K",(int)(((float)(int)nB)/10.24)	);
			}
			else
			{
				tmp.Format(L"0.0 K");
			}
		}
		else if (nType == DiskSize_M)
		{
			ULONGLONG nG ,nM,nK,nB,nKB,nMKB,nT,nGMKB;
			nT =0;
			//nT = nSize/(ULONGLONG)(1024*1024*1024*1024);
			//nGMKB =  nSize%(ULONGLONG)(1024*1024*1024*1024);
			nG = nSize/(ULONGLONG)(1000*1024*1024);
			nMKB =  nSize%(ULONGLONG)(1000*1024*1024);
			nM = nMKB/(ULONGLONG)(1024*1024);
			nKB =nMKB%(ULONGLONG)(1024*1024);
			nK = nKB/(ULONGLONG)(1024);
			nB =nKB%(ULONGLONG)(1024);

			if ((int)nT != 0)
			{
				tmp.Format(L"%d,%03d,%03d.%02d M",(int)nT, (int)nG,(int)nM,(int)(((float)(int)nK)/10.24)	);
			}
			else if ((int)nT == 0 &&(int)nG != 0 )
			{
				tmp.Format(L"%d,%03d.%02d M", (int)nG,(int)nM,(int)(((float)(int)nK)/10.24)		);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM != 0 )
			{
				tmp.Format(L"%d.%02d M",(int)nM,(int)(((float)(int)nK)/10.24)	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM == 0 &&(int)nK != 0 )
			{
				tmp.Format(L"0.%02d M",(int)(((float)(int)nK)/10.24)	);
			}
			else
			{
				tmp.Format(L"0.0 K");
			}
		}
		else if (nType == DiskSize_G)
		{
			ULONGLONG nG ,nM,nK,nB,nKB,nMKB,nT,nGMKB;
			nT =0;
			//nT = nSize/(ULONGLONG)(1024*1024*1024*1024);
			//nGMKB =  nSize%(ULONGLONG)(1024*1024*1024*1024);
			nG = nSize/(ULONGLONG)(1024*1024*1024);
			nMKB =  nSize%(ULONGLONG)(1024*1024*1024);
			nM = nMKB/(ULONGLONG)(1024*1024);
			nKB =nMKB%(ULONGLONG)(1024*1024);
			nK = nKB/(ULONGLONG)(1024);
			nB =nKB%(ULONGLONG)(1024);

			if ((int)nT != 0)
			{
				tmp.Format(L"%d,%03d.%02d G",(int)nT, (int)nG,(int)(((float)(int)nM)/10.24));
			}
			else if ((int)nT == 0 &&(int)nG != 0 )
			{
				tmp.Format(L"%d.%02d G", (int)nG,(int)(((float)(int)nM)/10.24)	);
			}
			else if ((int)nT == 0 &&(int)nG == 0  &&(int)nM != 0 )
			{
				tmp.Format(L"0.%02d G", (int)(((float)(int)nM)/10.24)	);
			}
			else
			{
				tmp.Format(L"0.0 G");
			}
		}
		else if (nType == DiskSize_T)
		{
			ULONGLONG nG ,nM,nK,nB,nKB,nMKB,nT,nGMKB;
			nT =0;
			//nT = nSize/(ULONGLONG)(1024*1024*1024*1024);
			//nGMKB =  nSize%(ULONGLONG)(1024*1024*1024*1024);
			nG = nSize/(ULONGLONG)(1024*1024*1024);
			nMKB =  nSize%(ULONGLONG)(1024*1024*1024);
			nM = nMKB/(ULONGLONG)(1024*1024);
			nKB =nMKB%(ULONGLONG)(1024*1024);
			nK = nKB/(ULONGLONG)(1024);
			nB =nKB%(ULONGLONG)(1024);

			if ((int)nT != 0)
			{
				tmp.Format(L"%d.%03d T",(int)nT, (int)nG);
			}
			else if ((int)nT == 0 &&(int)nG != 0 )
			{
				tmp.Format(L"0.%03d G", (int)nG	);
			}
			else
			{
				tmp.Format(L"0.0 G");
			}
		}
		return tmp;
	}

	CString GetDriverInfomation( BYTE bAlpha,int iGTI_TYPE )
	{
		CString strResult = _T("");

		if(!::IsCharAlpha((TCHAR)bAlpha))
		{
			strResult = _T("驱动器参数无效!");
			return strResult;
		}
		else
		{
			/**********获取驱动器名字、序列号和文件系统部分**********/
			CString strRootPathName;
			strRootPathName.Format(_T("%c:\\"),bAlpha);
			LPCTSTR lpRootPathName = strRootPathName;
			LPTSTR lpVolumeNameBuffer = new wchar_t[_MAX_FNAME];
			DWORD nVolumeNameSize = _MAX_FNAME;
			DWORD nVolumeSerialNumber = 0;//便于驱动器无效时做判断
			DWORD nMaximumComponentLength;
			DWORD nFileSystemFlags;
			LPTSTR lpFileSystemNameBuffer = new wchar_t[20];//文件系统(NTFS,FAT)多大有定义好的宏吗
			DWORD nFileSystemNameSize = 20;
			GetVolumeInformation(
				lpRootPathName,
				lpVolumeNameBuffer,
				nVolumeNameSize,
				&nVolumeSerialNumber, 
				&nMaximumComponentLength,
				&nFileSystemFlags,
				lpFileSystemNameBuffer,
				nFileSystemNameSize);

			/**********获取驱动器类型部分**********/
			CString strDriveType;

			/**********获取驱动器总大小和剩余大小部分**********/
			LPCTSTR lpDirectoryName = new wchar_t[2];
			lpDirectoryName = (LPCTSTR)strRootPathName.Mid(0,2);
			_ULARGE_INTEGER FreeBytesAvailable,TotalNumberOfBytes,TotalNumberOfFreeBytes ;

			__int64 iVolumeSize = 0,iVolumeFreeSize = 0;

			GetDiskFreeSpaceEx(strRootPathName.Mid(0,2),&FreeBytesAvailable,&TotalNumberOfBytes,&TotalNumberOfFreeBytes );
			iVolumeSize = TotalNumberOfBytes.QuadPart / 1024 / 1024;
			iVolumeFreeSize = FreeBytesAvailable.QuadPart / 1024 / 1024;

			/**********根据参数得出对应的驱动器信息**********/
			switch(iGTI_TYPE)
			{
			case GDI_VOLUMENAME:
				if (lpVolumeNameBuffer != NULL)
					strResult.Format(_T("驱动器 %c 的名字为:%s."),bAlpha,lpVolumeNameBuffer);
				else
					strResult.Format(_T("驱动器 %c 的名字为:%s."),bAlpha,lpVolumeNameBuffer);
				//strResult.Format(_T("获取驱动器名字失败!"));
				break;
			case GDI_VOLUMESERIALNUMBER:
				if (nVolumeSerialNumber != 0)
					strResult.Format(_T("驱动器 %c 的序列号为:%X."),bAlpha,nVolumeSerialNumber);
				else
					strResult.Format(_T("获取驱动器序列号失败!"));
				break;
			case GDI_VOLUMEFILESYSTEM:
				if (lpFileSystemNameBuffer != NULL)
					strResult.Format(_T("驱动器 %c 的文件系统为:%s."),bAlpha,lpFileSystemNameBuffer);
				else
					strResult.Format(_T("获取驱动器文件系统失败!"));
				break;
			case GDI_VOLUMESIZE:
				if (iVolumeSize != 0)
					strResult.Format(_T("驱动器 %c 的总大小为:%.2fGB."),bAlpha,(float)iVolumeSize/1024);
				else
					strResult.Format(_T("获取驱动器总大小失败!"));
				break;
			case GDI_VOLUMEFREESIZE:
				if (iVolumeFreeSize != 0)
					strResult.Format(_T("驱动器 %c 的剩余大小为:%.2fGB."),bAlpha,(float)iVolumeFreeSize/1024);
				else
					strResult.Format(_T("获取驱动器剩余大小失败!"));
				break;
			case GDI_VOLUMETYPE:
				switch(GetDriveType(lpRootPathName))
				{
				case DRIVE_UNKNOWN:
					strDriveType = _T("Unknown Type!");
					break;
				case DRIVE_NO_ROOT_DIR:
					strResult = _T("获取驱动器类型时参数设置错误!");
					return strResult;
				case DRIVE_REMOVABLE:
					strDriveType = _T("Removable Disk");
					break;
				case DRIVE_FIXED:
					strDriveType = _T("Hard Disk");
					break;
				case DRIVE_REMOTE:
					strDriveType = _T("Remote Device");
					break;
				case DRIVE_CDROM:
					strDriveType = _T("CD");
					break;
				case DRIVE_RAMDISK:
					strDriveType = _T("RAM");
					break;
				default:
					strResult = _T("Unknown Mistake!");
					return strResult;
				}
				strResult.Format(_T("Driver %c Of Type:%s."),bAlpha,strDriveType);
				break;
			default:
				strResult = _T("获取驱动器信息时参数设置错误!");
				break;
			}

		}

		/**********返回所要求的驱动器的信息**********/
		return strResult;
	}

TCHAR FirstDriveFromMask(ULONG unitmask)  
{  
	char i;  
	for (i = 0; i < 26; ++i)  
	{  
		if (unitmask & 0x1)//看该驱动器的状态是否发生了变化  
			break;  
		unitmask = unitmask >> 1;  
	}  
	return (i + 'A');  
}

包含头文件：#include "dbt.h"

重写CDialogEx的DefWindowProc函数

LRESULT CxxxxSCDlg::DefWindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{
	// TODO: 在此添加专用代码和/或调用基类
	switch (message)
	{
		//WM_DEVICECHANGE，系统硬件改变发出的系统消息
	case WM_DEVICECHANGE:
		{
			PDEV_BROADCAST_HDR lpdb=(PDEV_BROADCAST_HDR)lParam;
			switch(wParam)
			{
			case WM_DEVICECHANGE:
				break;
			case DBT_DEVICEARRIVAL://DBT_DEVICEARRIVAL，设备检测结束，并且可以使用
				{
					if(lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)//逻辑卷
					{
						PDEV_BROADCAST_VOLUME lpdbv =  (PDEV_BROADCAST_VOLUME)lpdb;
						switch(lpdbv->dbcv_flags)
						{
						case 0:                //U盘
							{
								CString decDriver;
								decDriver = FirstDriveFromMask(lpdbv ->dbcv_unitmask);
								DWORD sizeStruct = lpdbv->dbcv_size;

								ULARGE_INTEGER FreeAv,TotalBytes,FreeBytes;
								if (GetDiskFreeSpaceEx(decDriver +L":\\", &FreeAv,&TotalBytes,&FreeBytes))
								{
									CString strTotalBytes = GetDiskSize(TotalBytes.QuadPart,DiskSize_B);
									CString strFreeBytes = GetDiskSize(FreeBytes.QuadPart,DiskSize_B);
									CString strFreeAV = GetDiskSize(TotalBytes.QuadPart - FreeAv.QuadPart,DiskSize_B);
									CString strTG = GetDiskSize(TotalBytes.QuadPart,DiskSize_G);
									CString strFG = GetDiskSize(FreeBytes.QuadPart,DiskSize_G);
									CString strYG = GetDiskSize(TotalBytes.QuadPart - FreeAv.QuadPart,DiskSize_G);
									CString strname = GetDriverInfomation(decDriver.GetAt(0),GDI_VOLUMENAME);
									CString strSeries = GetDriverInfomation(decDriver.GetAt(0),GDI_VOLUMESERIALNUMBER);
									CString strfilsystem = GetDriverInfomation(decDriver.GetAt(0),GDI_VOLUMEFILESYSTEM);
									CString strAllInfo;
									strAllInfo.Format(L"驱动器:[%s] \n容量：%s (%s)\n已用空间：%s (%s)\n可用空间: %s (%s)\n%s\n%s\n%s",
										decDriver +L":\\",  strTotalBytes ,strTG,strFreeAV,strYG,strFreeBytes,strFG,strname,strSeries,strfilsystem
										);/* 单位为G */
									AfxMessageBox(strAllInfo);
								}
								/*CString tmp;
								tmp.Format(L"检测到U盘:[%s]插入! %ld bytes\n",decDriver.GetBuffer(0),sizeStruct);
								AfxMessageBox(tmp);*/
							}
							break;
						case DBTF_MEDIA:    //光盘
							CString tmp;
							tmp.Format(L"检测到光盘:[%c]插入!\n",FirstDriveFromMask(lpdbv ->dbcv_unitmask));
							AfxMessageBox(tmp);
							break;
						}
					}
				}
				break;
			case DBT_DEVICEREMOVECOMPLETE://DBT_DEVICEREMOVECOMPLETE,设备卸载或者拔出
				{
					if(lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)//逻辑卷
					{
						PDEV_BROADCAST_VOLUME lpdbv =  (PDEV_BROADCAST_VOLUME)lpdb;
						switch(lpdbv->dbcv_flags)
						{
						case 0:                //U盘
							{
								CString decDriver;
								decDriver = FirstDriveFromMask(lpdbv ->dbcv_unitmask);
								CString tmp;
								tmp.Format(L"检测到U盘:[%s]拔出!\n",decDriver.GetBuffer(0));
								AfxMessageBox(tmp);
							}
							break;
						case DBTF_MEDIA:    //光盘

							break;
						}
					}
				}
				break;
			}
		}
		break;
	}
	return CDialogEx::DefWindowProc(message, wParam, lParam);
}


少莫千华
关注
专栏目录
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值