windows外设盘符管理模块

最新推荐文章于 2024-09-06 11:00:51 发布

飞刀探花郎

最新推荐文章于 2024-09-06 11:00:51 发布

阅读量196

点赞数 1

分类专栏： Windows_C++ 文章标签： U盘进出判断光盘判断盘符 MFC 外设消息 windows

本文链接：https://blog.csdn.net/l1040515635/article/details/103908340

版权

Windows_C++ 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

/*	本文件对磁盘进行管理，windows下一般情况下最多磁盘数a-z
	对这些磁盘进行可用性和磁盘类型备份，就可以判断出磁盘的变化情况。
	本文件主要是为了管理外设，目前支持单个单种磁盘的变化，不尽完善。
	如果所指定的磁盘类型在机器上已存在，可能会出现bug。


	基本思路：如何判断一个磁盘是新插入的磁盘？
	解答：如果一个磁盘的可使用状态改变了，那就说明它有可能是新插入的磁盘，当然，再加上磁盘类型确认，基本上就可以确认这个磁盘是新进入的磁盘。
	并且，外设消息来的时候，进行状态检测，可保证插入的盘符的正确性。
	当然，反之亦然。

注：m_strDiskRootPath是对上次调用检测插入磁盘的磁盘盘符缓存，方便在磁盘弹出的时候做判断。
m_DiskType 是你需要检测的盘符类型，U盘对应DRIVE_REMOVABLE 光盘对应DRIVE_CDROM
*/
#pragma once
#include <Windows.h>
#include <string>
#include <vector>
#include <iostream>
using namespace std;



class DiskManager
{
public:
	std::string m_strDiskRootPath;
	int m_DiskType;
private:
	std::vector<std::string> vStrOldDiskList;
	std::vector<bool> vBoolUsable;
	std::vector<int> viDiskType;
	static const int iDisk = 26;
public:
	DiskManager(int DiskType = DRIVE_CDROM)
	{
		for (size_t i = 0; i < iDisk; i++)
		{
			char buff[100] = { 0 };
			sprintf_s(buff, 100, "%c:\\\\", 65 + i);
			vStrOldDiskList.push_back(buff);
		}
		
		for (size_t i = 0; i < iDisk; i++)
		{
			bool usable = false;
			int DiskType = -1;

			unsigned _int64 i64FreeBytesToCaller = 0;
			unsigned _int64 i64TotalBytes = 0;
			unsigned _int64 i64FreeBytes = 0;
			DiskType = GetDriveTypeA(vStrOldDiskList[i].c_str());
			usable = GetDiskFreeSpaceExA(
				vStrOldDiskList[i].c_str(),
				(PULARGE_INTEGER)&i64FreeBytesToCaller,
				(PULARGE_INTEGER)&i64TotalBytes,
				(PULARGE_INTEGER)&i64FreeBytes);
			vBoolUsable.push_back(usable);
			viDiskType.push_back(DiskType);
		}
		m_DiskType = DiskType ;

	}
	~DiskManager() {};

	void UpdateList() 
	{
		vBoolUsable.clear();
		viDiskType.clear();
		for (size_t i = 0; i < iDisk; i++)
		{
			bool usable = false;
			int DiskType = -1;

			unsigned _int64 i64FreeBytesToCaller = 0;
			unsigned _int64 i64TotalBytes = 0;
			unsigned _int64 i64FreeBytes = 0;
			DiskType = GetDriveTypeA(vStrOldDiskList[i].c_str());
			usable = GetDiskFreeSpaceExA(
				vStrOldDiskList[i].c_str(),
				(PULARGE_INTEGER)&i64FreeBytesToCaller,
				(PULARGE_INTEGER)&i64TotalBytes,
				(PULARGE_INTEGER)&i64FreeBytes);
			vBoolUsable.push_back(usable);
			viDiskType.push_back(DiskType);
		}
	}
	std::string GetNewDiskIn(int& outDiskType ) 
	{
		m_strDiskRootPath.clear();
		outDiskType = -1;
		for (size_t i = 0; i < iDisk; i++)
		{
			bool usable = false;
			int DiskType = -1;

			unsigned _int64 i64FreeBytesToCaller = 0;
			unsigned _int64 i64TotalBytes = 0;
			unsigned _int64 i64FreeBytes = 0;
			DiskType = GetDriveTypeA(vStrOldDiskList[i].c_str());
			usable = GetDiskFreeSpaceExA(
				vStrOldDiskList[i].c_str(),
				(PULARGE_INTEGER)&i64FreeBytesToCaller,
				(PULARGE_INTEGER)&i64TotalBytes,
				(PULARGE_INTEGER)&i64FreeBytes);
			if (m_DiskType == DiskType &&vBoolUsable.at(i) == false && usable /*&&i64FreeBytes>0&&i64TotalBytes>0*/)//一个磁盘从不可用变成可用，类型为DVD，即可确定此磁盘是新进的DVD
			{
				outDiskType = DiskType;
				UpdateList();
				m_strDiskRootPath = vStrOldDiskList[i];
				return vStrOldDiskList[i];
			}
		}

		return "";
	};

	std::string GetNewDiskOut(int& outDiskType ) 
	{
		m_strDiskRootPath.clear();
		outDiskType = -1;
		for (size_t i = 0; i < iDisk; i++)
		{
			bool usable = false;
			int DiskType = -1;

			unsigned _int64 i64FreeBytesToCaller = 0;
			unsigned _int64 i64TotalBytes = 0;
			unsigned _int64 i64FreeBytes = 0;
			DiskType = GetDriveTypeA(vStrOldDiskList[i].c_str());
			usable = GetDiskFreeSpaceExA(
				vStrOldDiskList[i].c_str(),
				(PULARGE_INTEGER)&i64FreeBytesToCaller,
				(PULARGE_INTEGER)&i64TotalBytes,
				(PULARGE_INTEGER)&i64FreeBytes);
			if (m_DiskType == DiskType &&vBoolUsable.at(i) == true && !usable /*&&i64FreeBytes>0&&i64TotalBytes>0*/)//一个磁盘从可用变成不可用，类型为DVD，即可确定此磁盘是退出的DVD
			{
				outDiskType = DiskType;
				UpdateList();
				//m_strDiskRootPath = vStrOldDiskList[i];
				return vStrOldDiskList[i];
			}
		}

		return "";
	};
};

1 使用方式，mfc增加消息

WM_DEVICECHANGE

#include <dbt.h>

BOOL CUsbDeviceFerryClientDlg::OnDeviceChange(UINT nEventType, DWORD dwData)
{
    if (DBT_DEVICEARRIVAL== nEventType)
    {
       int diskType = -1;
		std::string diskInfo = m_diskManager.GetNewDiskIn(diskType);

		if (diskType == DRIVE_CDROM)
            {
                //do InSomething
            }
    }

    if (DBT_DEVICEREMOVECOMPLETE == nEventType)
    {
       std::string StrDvd = m_diskManager.m_strDiskRootPath;
		int diskType = -1;
		std::string diskInfo = m_diskManager.GetNewDiskOut(diskType);
		LOG("光盘已弹出,已经关闭拷贝线程"+diskInfo);
		if (!StrDvd.empty() && 0 == strcmp(StrDvd.c_str(),diskInfo.c_str()))
		{
            //do OutSomething
        }
    }
    return TRUE;
}