专栏首页流川疯编写程序的艺术windows下,c /c++实现磁盘扫描,结合配置文件,读取特定后缀文件目录代码

windows下,c /c++实现磁盘扫描,结合配置文件,读取特定后缀文件目录代码

#include <windows.h>
#include <iostream>
#include <fstream> 
#include <locale.h>
#include "stdio.h"
#include "OperatingIni.h"

using namespace std;

class ScanDisk //磁盘搜索类
{
public:

	ScanDisk(TCHAR *Expansion,TCHAR *FileName);//构造函数
	~ScanDisk();

   	TCHAR DriveString[MAX_PATH];// 驱动器列表
	TCHAR Driver[MAX_PATH];//驱动器名
	TCHAR Expansion[MAX_PATH];//后缀名
	TCHAR FileName[MAX_PATH];//构造函数使用生成的文件名
	TCHAR Name[MAX_PATH];//还未传送的文件路径
	TCHAR ConfigName[MAX_PATH];//要使用的配置文件名
	DWORD count;//文件个数
	DWORD Transform_count;//已传送文件个数

	CIniReader *Reader;
	CIniWriter *Writer;
	FILE *fp;//文件指针,创建路径文件
public:

TCHAR * GetFirstFile();//得到第一个文件路径
//void ModifyPath(TCHAR *path);//修改路径字符串
void SearchforAllDriver(); //搜索所有驱动器
void GetDriverList();//得到驱动器列表
bool Search(TCHAR *Path,TCHAR *File);//递归搜索
bool TcharMarch(TCHAR *fileName,TCHAR *Extension);//文件后缀名匹配
void SetExpansion(TCHAR *Expansion);//设置新的文件后缀
void SetConfigName(TCHAR *ConfigName);//设置需要操作的配置文件名
void InitOperateIni(TCHAR *ConfigName);//初始化配置信息类
void GetAllExpansion();//得到所有后缀名并且检索目录写入文件
};

ScanDisk::ScanDisk(TCHAR *Expansion,TCHAR *FileName)//初始化工作
{
	memset(this->DriveString,0,sizeof(this->DriveString));
	memset(this->Driver,0,sizeof(this->Driver));
	memset(this->Expansion,0,sizeof(this->Expansion));
	memset(this->FileName,0,sizeof(this->FileName));
	memset(this->Name,0,sizeof(this->Name));
	memset(this->ConfigName,0,sizeof(this->ConfigName));
	this->count=0;//文件个数
	this->Transform_count=0;//已传送文件个数为0

	memcpy(this->Expansion,Expansion,wcslen(Expansion)*2);
	memcpy(this->FileName,FileName,wcslen(FileName)*2);
	//MessageBox(NULL,this->FileName,NULL,MB_OK);
//	MessageBox(NULL,this->Expansion,NULL,MB_OK);

}

ScanDisk::~ScanDisk()
{
	fclose(this->fp);

}

void ScanDisk::SetExpansion(TCHAR *Expansion)
{
	memset(this->Expansion,0,sizeof(this->Expansion));
	memcpy(this->Expansion,Expansion,wcslen(Expansion)*2);
}

void ScanDisk::SetConfigName(TCHAR *ConfigName)
{
	memset(this->ConfigName,0,sizeof(this->ConfigName));
	memcpy(this->ConfigName,ConfigName,wcslen(ConfigName)*2);
}

void ScanDisk::InitOperateIni(TCHAR *ConfigName)
{
	memset(this->ConfigName,0,sizeof(this->ConfigName));
	memcpy(this->ConfigName,ConfigName,wcslen(ConfigName)*2);
	this->Writer=new CIniWriter(this->ConfigName);
	this->Reader=new CIniReader(this->ConfigName);
	(this->Writer)->WriteInteger(L"Setting",L"count",this->count);
	(this->Writer)->WriteInteger(L"Setting",L"Transform_count",this->Transform_count);
}

void ScanDisk::GetAllExpansion()//读取配置文件中的每一个后缀名,遍历磁盘写入文件
{
	TCHAR *expansion=(this->Reader)->ReadString(L"Setting", L"extension", L"");//此处设计不是很好
	int length=lstrlen(expansion)+1;//没有斜杠零
	int i=0;
	TCHAR temp[MAX_PATH]={0};
	for (int j=0;j<length;j++)
	{

		if (((*expansion)!=L',')&&((*expansion)!=L'\0'))
		{
			memcpy(&temp[i],expansion,sizeof(TCHAR));
			temp[i++];
			expansion++;
		}
		
		if (((*expansion)==L',')||((*expansion)==L'\0'))
		{ 

			temp[i]=L'\0';
			this->SetExpansion(temp);
			this->SearchforAllDriver();
			if ((*expansion)==L'\0')
			{
				break;
			}
			expansion++;
			i=0;
			memset(temp,0,sizeof(temp));
				
		}

	 }
	 

 
 
}


TCHAR * ScanDisk::GetFirstFile()
{
	DWORD number=(this->Reader)->ReadInteger(L"Setting",L"Transform_count",this->Transform_count);//看看读到第几个文件了
	this->fp=_wfopen(this->FileName,L"r"); //读的方式打开
	if(!this->fp) 
	{ 
		cout<<L"Can not open the .txt file"<<endl; 
	}
	else
	{
		cout<<"the file is opened !"<<endl; 
	}
	//TCHAR path[MAX_PATH]={0};
	for (int i=0;i<=number;i++)
	{
			fgetws(this->Name,MAX_PATH,this->fp);//
	}
	//fgetws(this->Name,MAX_PATH,this->fp);//
	this->Name[lstrlen(this->Name)-1]=0;//去掉文件最后的0A
	wprintf(this->Name);
	//MessageBox(NULL,this->Name,NULL,MB_OK);
	this->Transform_count++;
	(this->Writer)->WriteInteger(L"Setting",L"Transform_count",this->Transform_count);
	fclose(this->fp);
	
	return this->Name;

}

void ScanDisk::SearchforAllDriver()
{

	memset(this->Driver,0,sizeof(this->Driver));
	
	this->GetDriverList();
	int driverCount=0;
	TCHAR * pDrive= this->DriveString;
	while( *pDrive )
	{
		pDrive += wcslen( pDrive ) + 1;
		driverCount++;
		
	} 
//	printf("%d\n",driverCount);//总共几个驱动器
	pDrive= this->DriveString;
	
	this->fp=_wfopen(this->FileName,L"a+"); //追加的方式打开
	if(!this->fp) 
	{ 
		cout<<L"Can not open the .txt file"<<endl; 
	}
	else
	{
		cout<<"the file is opened !"<<endl; 
	}
	//for (int i=0;i<driverCount;i++)
	//{
	while( * pDrive )
	{
		memcpy(this->Driver,pDrive,wcslen(this->DriveString)+1);//控制字符长度,和缓冲区
		//MessageBox(NULL,this->Driver,NULL,MB_OK);
		this->Search(this->Driver,this->Expansion);
		fflush(this->fp);
		pDrive=pDrive+wcslen(pDrive)+1;
	}
	//}
	(this->Writer)->WriteInteger(L"Setting",L"count",this->count);

}

void ScanDisk::GetDriverList()
{
	TCHAR	DriveString[MAX_PATH];
	// 前一个字节为令牌,后面的52字节为驱动器跟相关属性
	GetLogicalDriveStrings(sizeof(DriveString), DriveString);
	memcpy(this->DriveString,DriveString,sizeof(this->DriveString));
}


bool ScanDisk::TcharMarch(TCHAR *fileName,TCHAR *Extension)//文件后缀名匹配
{
	int length_of_ext=wcslen(Extension);
	int length_of_name=wcslen(fileName);
	int i=0;
	while(i<length_of_ext)
	{
		if (fileName[i+(length_of_name-length_of_ext)]!=Extension[i])
		{
			return false;
		}
		else
			i++;
	}
	return true;

}

bool ScanDisk::Search(TCHAR *Path,TCHAR *File)
{
	HANDLE hFind;
	WIN32_FIND_DATA wfd;
	
	ZeroMemory(&wfd,sizeof(WIN32_FIND_DATA));
	TCHAR PathTemp[MAX_PATH];
	memset(PathTemp,0,sizeof(PathTemp));

	swprintf(PathTemp,L"%s\\*.*",Path);
	hFind=FindFirstFile(PathTemp,&wfd);

	if(INVALID_HANDLE_VALUE==hFind)
	{
		//MessageBox(NULL,L"INVALID_HANDLE_VALUE",L"FindFirstFile",MB_OK);
		return false;
	}
	
	do
	{
		if('.'==wfd.cFileName[0])
		{
			continue;
		}
		
		if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
		{
			swprintf(PathTemp,L"%s\\%s",Path,wfd.cFileName);
			//MessageBox(NULL,PathTemp,"Directory",MB_OK);
			//wprintf(PathTemp);
			//printf("\n");
			Search(PathTemp,File);
			fflush(this->fp);
		}
		else
		{
			
			if (TcharMarch(wfd.cFileName,File))
			{
			
				swprintf(PathTemp,L"%s\\%s",Path,wfd.cFileName);
				//MessageBox(NULL,L"Found",PathTemp,MB_OK);
				//printf(PathTemp);
				//wprintf(PathTemp);
				//printf("\n");
///////////////////////////////////////////////////////////////////////////////////
//				TCHAR temp[MAX_PATH];
//				memcpy(temp," ",sizeof(temp));
//				temp[MAX_PATH-2]=L'\0';
//				memcpy(temp,PathTemp,lstrlen(PathTemp)*2);
//////////////////////////////////////////////////////////////////////////////////
				fwprintf(this->fp,L"%s",PathTemp);
				
				fwprintf(this->fp,L"\n");
				this->count++;//文件个数加1
				
			}
		}
		
	}while(FindNextFile(hFind,&wfd));
	
	FindClose(hFind);
	
	return true;
	
}

这个代码必须在unicode工程下使用,支持中文路径,

"OperatingIni.h"//这个头文件,如下所示
#include <tchar.h>
#include <iostream>
#include <Windows.h>

class CIniReader
{
public:
	CIniReader(TCHAR * szFileName); 
	int ReadInteger(TCHAR* szSection, TCHAR* szKey, int iDefaultValue);
	//float ReadFloat(TCHAR* szSection, TCHAR* szKey, float fltDefaultValue);
	bool ReadBoolean(TCHAR* szSection, TCHAR* szKey, bool bolDefaultValue);
	TCHAR* ReadString(TCHAR* szSection, TCHAR* szKey, const TCHAR* szDefaultValue);
private:
	TCHAR m_szFileName[MAX_PATH];
};

class CIniWriter
{
public:
	CIniWriter(TCHAR* szFileName); 
	void WriteInteger(TCHAR* szSection, TCHAR* szKey, int iValue);
	//void WriteFloat(TCHAR* szSection, TCHAR* szKey, float fltValue);
	void WriteBoolean(TCHAR* szSection, TCHAR* szKey, bool bolValue);
	void WriteString(TCHAR* szSection, TCHAR* szKey, TCHAR* szValue);
	void DeleteString(TCHAR* szSection, TCHAR* szKey);
private:
	TCHAR m_szFileName[MAX_PATH];
};


CIniReader::CIniReader(TCHAR* szFileName)
{
	memset(m_szFileName, 0x00, MAX_PATH);
	memcpy(m_szFileName, szFileName, wcslen(szFileName)*2);//注意此处
}

int CIniReader::ReadInteger(TCHAR* szSection, TCHAR* szKey, int iDefaultValue)
{
	int iResult = GetPrivateProfileInt(szSection,  szKey, iDefaultValue, m_szFileName); 
	return iResult;
}

//float CIniReader::ReadFloat(TCHAR* szSection, TCHAR* szKey, float fltDefaultValue)
//{
//TCHAR szResult[255];
//TCHAR szDefault[255];
//float fltResult;
//	swprintf(szDefault, L"%f",fltDefaultValue);
//	GetPrivateProfileString(szSection,  szKey, szDefault, szResult, 255, m_szFileName); 
//	fltResult =  atof(szResult);
//	return fltResult;
//}

bool CIniReader::ReadBoolean(TCHAR* szSection, TCHAR* szKey, bool bolDefaultValue)
{
	TCHAR szResult[MAX_PATH];
	TCHAR szDefault[MAX_PATH];
	bool bolResult;
	swprintf(szDefault, L"%s", bolDefaultValue? L"True" : L"False");
	GetPrivateProfileString(szSection, szKey, szDefault, szResult, 255, m_szFileName); 
	bolResult =  (wcscmp(szResult, L"True") == 0 || 
		wcscmp(szResult, L"true") == 0) ? true : false;
	return bolResult;
}

TCHAR* CIniReader::ReadString(TCHAR* szSection, TCHAR* szKey, const TCHAR* szDefaultValue)
{
	TCHAR* szResult = new TCHAR[MAX_PATH];
	memset(szResult, 0x00, MAX_PATH);
	GetPrivateProfileString(szSection,  szKey, 
		szDefaultValue, szResult, MAX_PATH, m_szFileName); 
	return szResult;
}

CIniWriter::CIniWriter(TCHAR* szFileName)
{
	memset(m_szFileName, 0x00, MAX_PATH);
	memcpy(m_szFileName, szFileName, wcslen(szFileName)*2);
}

void CIniWriter::WriteInteger(TCHAR* szSection, TCHAR* szKey, int iValue)
{
	TCHAR szValue[MAX_PATH];
	swprintf(szValue, L"%d", iValue);
	WritePrivateProfileString(szSection,  szKey, szValue, m_szFileName); 
}

//void CIniWriter::WriteFloat(TCHAR* szSection, TCHAR* szKey, float fltValue)
//{
//TCHAR szValue[255];
//swprintf(szValue,L"%f", fltValue);
//WritePrivateProfileString(szSection,  szKey, szValue, m_szFileName); 
//}

void CIniWriter::WriteBoolean(TCHAR* szSection, TCHAR* szKey, bool bolValue)
{
	TCHAR szValue[MAX_PATH];
	swprintf(szValue, L"%s", bolValue ? L"True" : L"False");
	WritePrivateProfileString(szSection,  szKey, szValue, m_szFileName); 
}

void CIniWriter::WriteString(TCHAR* szSection, TCHAR* szKey, TCHAR* szValue)
{
	WritePrivateProfileString(szSection,  szKey, szValue, m_szFileName);
}

void CIniWriter::DeleteString(TCHAR* szSection, TCHAR* szKey)
{
	WritePrivateProfileString(szSection,szKey,NULL,m_szFileName);
}

主函数可以这么写:

#include "ScanDisk.h"
//#include "OperatingIni.h"

int main()
{
	//_wsetlocale设置中文语言环境
	_wsetlocale(LC_ALL,L"chs");
	
	
	

  //*/
	//_wsetlocale设置中文语言环境
//	_wsetlocale(LC_ALL,L"chs");
	
	CIniWriter iniWriter(L".//Log.ini");
	iniWriter.WriteString(L"Setting",L"extension", L".txt");   
	//iniWriter.WriteInteger(L"Setting", L"count", 2); 
	////iniWriter.WriteFloat(L"Setting", L"Height", 1.82f); 
	//iniWriter.WriteBoolean(L"Setting", L"Marriage", false);  
	//iniWriter.WriteBoolean(L"Setting", L"Marriage", NULL); 
	//iniWriter.DeleteString(L"Setting",L"Marriage");
	//iniWriter.WriteString("ff", "Name", "jia");   
	//	iniWriter.WriteInteger("ff", "Age", 8); 
	//iniWriter.WriteFloat("ff", "Height", 1.82f); 
	//iniWriter.WriteBoolean("ff", "Marriage", false); 
	
	CIniReader iniReader(L".//Log.ini");
	
	TCHAR *szName = iniReader.ReadString(L"Setting", L"extension", L"");   
//	int iAge = iniReader.ReadInteger(L"Setting", L"Age", 0); 
	//float fltHieght = iniReader.ReadFloat(L"Setting", L"Height", 1.80f); 
//	bool bMarriage = iniReader.ReadBoolean(L"Setting", L"Marriage", true); 
	
	//std::cout<<L"Name:"<<szName<<std::endl
	//	<<L"Age:"<<iAge<<std::endl 
	//<<L"Height:"<<fltHieght<<std::endl 
	//<<L"Marriage:"<<bMarriage<<std::endl; 
	//wprintf(szName);
//	wprintf(L"\n");
	//wprintf(L"%d\n",iAge);
    _wremove(L"Logger.log");//删除ini文件
	TCHAR *file= L"Logger.log";
	TCHAR *ext=szName;
	ScanDisk sd(ext,file);
	sd.InitOperateIni(L".//Log.ini");
	sd.GetAllExpansion();
	sd.SearchforAllDriver();
	sd.GetFirstFile();
	sd.GetFirstFile();
	//sd.SetExpansion(L".dsp");
	//sd.SearchforAllDriver();
	//sd.GetFirstFile();
//	MessageBox(NULL,sd.GetFirstFile(),NULL,MB_OK);
	


 
	//fclose(fp);
	//fclose(sd.fp);
	system("pause");
	return 0;
}

配置文件:

部分搜索结果:

资源下载:

http://download.csdn.net/detail/wangyaninglm/8301303

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 大数据ETL实践探索(4)---- 搜索神器Elastic search

    本系列文章主要针对ETL大数据处理这一典型场景,基于python语言使用Oracle、aws、Elastic search 、Spark 相关组件进行一些基本的...

    流川疯
  • 数据可视化的开源方案: Superset vs Redash vs Metabase (一)

    人是视觉动物,要用数据把一个故事讲活,图表是必不可少的。如果你经常看到做数据分析同事,在SQL客户端里执行完查询,把结果复制/粘贴到 Excel 里再做成图表,...

    流川疯
  • 《一本书读懂24种互联网思维》 摘抄

    虽然书中很多内容,在IT 老司机来看都是老生常谈,但对于我们每个有互联网思维的小伙伴来说还是常读常新的。

    流川疯
  • ES6——Class基础语法

      而在ES6中,引入了Class这个概念来作为对象的模板。而class只是一个语法糖写法,还是基于ES5封装而来的。上面的例子用class来改写如下:

    流眸
  • Python 切片(Slice)

    在实际开发中,经常遇到下面的需求:在线性表(数组)中提取若干个元素的操作,提取规则有很多,比如说提取前5个、提取后5个、提取奇数/偶数位元素等等。 ...

    Steve Wang
  • 绿色教学,让幼教专用NComputing云终端解决方案来守护孩子的成长

    对幼儿教育来说,电脑是必需品,云课堂的活跃气氛是不可少的。而作为人们日常工作生活所必不可少的工具,电脑就有着一些让人诟病的问题。比如那硕大的机箱无时无刻...

    nc云终端yun0101.com
  • django表单提交

    HTML表单是网站交互性的经典方式。 本章将介绍如何用Django对用户提交的表单数据进行处理。

    py3study
  • SCI写作专题第三期:Materials and Methods

    Scientific writing的主要任务是将研究者的信息清晰、简明地传达给科学界,一是为了科学论文的发表,二是推动了科学和循证决策的进展,在过去的几十年里...

    百味科研芝士
  • 美国国防部正在试验软件定义网络 进展十分有限

    软件定义网络(software-defined networking)有很大的希望提高设置和控制网络的效率,将引发信息技术的这一关键方面发生根本性的改变。虽然国...

    静一
  • 前端-手摸手,带你用合理的姿势使用webpack4(下)

    webpack 4 最大的改动就是废除了 CommonsChunkPlugin 引入了 optimization.splitChunks。

    grain先森

扫码关注云+社区

领取腾讯云代金券