Exemplo n.º 1
0
static BOOL FindNextChildDir (HANDLE hFindFile,
   WIN32_FIND_DATA *lpFindData) {

   BOOL fFound = FALSE;

   do {
      fFound = FindNextFile(hFindFile, lpFindData);
   } while (fFound && !IsChildDir(lpFindData));

   return(fFound);
}
Exemplo n.º 2
0
bool getDirs(string pathString, DirsFiles& vec){
	
	WIN32_FIND_DATA FindFileData;
    HANDLE hFind;
	wstring wStr_tmp;
	string str_tmp;
	wstring wstr(pathString.length(),L'');
	copy(pathString.begin(),pathString.end(),wstr.begin());
	const wchar_t* path = wstr.c_str();
	if(!SetCurrentDirectory(path)){
        printf("%s Not a directory.\n",path);
        return false;
    }
    hFind = FindFirstFile(L"*", &FindFileData);
    if (hFind == INVALID_HANDLE_VALUE) {
        cerr << "Invalid file handle. Error code: " 
              << GetLastError() << endl;
        return false;
    } 
    else {
        //wcout << L"File found: " << FindFileData.cFileName << endl;
        while(FindNextFile(hFind, &FindFileData)){
            //wcout << L"File found: " << FindFileData.cFileName << endl;
			if(IsChildDir(&FindFileData)){
				wStr_tmp = FindFileData.cFileName;
				str_tmp = string(wStr_tmp.begin(), wStr_tmp.end());
				str_tmp.assign(wStr_tmp.begin(), wStr_tmp.end());
				vec.push_back(str_tmp);
				//vec.push_back(FindFileData.cFileName); 
			}
        }
        FindClose(hFind);
 
        if (GetLastError() == ERROR_NO_MORE_FILES){
            //cout << "\nSearch finished." << endl;
        }
        else{
            cout << "\nUnknown error." << endl;
            return false;
        }
    }
    return true;
}
Exemplo n.º 3
0
static HANDLE FindFirstChildDir (LPTSTR szPath,
   WIN32_FIND_DATA *lpFindData) {

   BOOL fFound;
   HANDLE hFindFile = FindFirstFile(szPath, lpFindData);

   if (hFindFile != INVALID_HANDLE_VALUE) {
      fFound = IsChildDir(lpFindData);

      if (!fFound)
         fFound = FindNextChildDir(hFindFile, lpFindData);

      if (!fFound) {
         FindClose(hFindFile);
         hFindFile = INVALID_HANDLE_VALUE;
      }
   }
   return(hFindFile);
}
Exemplo n.º 4
0
// Walk the directory structure and fill a list box with
// filenames. If pDW->fRecurse is set, list any child
// directories by recursively calling DirWalkRecurse.
static void DirWalkRecurse (LPDIRWALKDATA pDW) {
   HANDLE hFind;

   pDW->nDepth++;

   pDW->nIndent = 3 * pDW->nDepth;
   _stprintf(pDW->szBuf, __TEXT("%*s"), pDW->nIndent,
      __TEXT(""));

   GetCurrentDirectory(chDIMOF(pDW->szBuf) - pDW->nIndent,
      &pDW->szBuf[pDW->nIndent]);
   //store data in filenamelist
   /*
   _FileNameData fnd;
   memcpy(fnd.buf, pDW->szBuf, BUF_SIZE);
   fileNameList.push_back(fnd);
   */
   //end
   hFind = FindFirstFile(__TEXT("*.*"), &pDW->FindData);
   pDW->fOk = (hFind != INVALID_HANDLE_VALUE);

   while (pDW->fOk) {
      pDW->fIsDir = pDW->FindData.dwFileAttributes &
         FILE_ATTRIBUTE_DIRECTORY;
      if (!pDW->fIsDir ||
         (!pDW->fRecurse && IsChildDir(&pDW->FindData))) {

         _stprintf(pDW->szBuf,
            pDW->fIsDir ? __TEXT("%*s[%s]") : __TEXT("%*s%s"),
            pDW->nIndent, __TEXT(""),
            pDW->FindData.cFileName);

         //ListBox_AddString(pDW->hwndTreeLB, pDW->szBuf);
           _FileNameData fnd;
          memcpy(fnd.buf, pDW->szBuf, BUF_SIZE);
          fileNameList.push_back(fnd);
      }
      pDW->fOk = FindNextFile(hFind, &pDW->FindData);
   }
   if (hFind != INVALID_HANDLE_VALUE)
      FindClose(hFind);

   if (pDW->fRecurse) {
      // Get the first child directory.
      hFind = FindFirstChildDir(
         __TEXT("*.*"), &pDW->FindData);
      pDW->fOk = (hFind != INVALID_HANDLE_VALUE);
      while (pDW->fOk) {
         // Change into the child directory.
         if (SetCurrentDirectory(pDW->FindData.cFileName)) {

            // Perform the recursive walk into the child
            // directory.  Remember that some members of pDW 
            // will be overwritten by this call.
            DirWalkRecurse(pDW);

            // Change back to the child's parent directory.
            SetCurrentDirectory(__TEXT(".."));
         }

         pDW->fOk = FindNextChildDir(hFind, &pDW->FindData);
      }

      if (hFind != INVALID_HANDLE_VALUE)
         FindClose(hFind);
   }
   pDW->nDepth--;
}
Exemplo n.º 5
0
static void DirWalkRecurse(LPDIRWALKDATA pDW, LPCTSTR pszRootPath, LPCTSTR pszDesPath)	//pointer to DIRWALKDATA as parameter, so the param in the struct can be constantly updated such as nDepth
{
	HANDLE hFind;						//FindFirstFile()返回的句柄  作为FindNextFile()的参数,多次查找的关联

	LPTSTR src = (LPTSTR)malloc(100 * sizeof(LPTSTR));
	LPTSTR des = (LPTSTR)malloc(100 * sizeof(LPTSTR));

	pDW->nDepth++;						//recursive depth + 1
	pDW->nIndent = 3 * pDW->nDepth;		//控制缓冲区写入位置
	_stprintf_s(pDW->szBuf, _TEXT("%*s"), pDW->nIndent, _TEXT(""));

	GetCurrentDirectory(chDIMOF(pDW->szBuf) - pDW->nIndent, &pDW->szBuf[pDW->nIndent]);		//将当前目录存入缓冲区

	//printf("%s\n", pDW->szBuf);

	hFind = FindFirstFile(_TEXT("*.*"), &pDW->FindData);						//system API
	pDW->fOk = (hFind != INVALID_HANDLE_VALUE);

	while (pDW->fOk)		//当前目录搜索完毕则跳出
	{
		pDW->fIsDir = pDW->FindData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;	//是否是目录
		if (!pDW->fIsDir || (!pDW->fRecurse && IsChildDir(&pDW->FindData)))			//为文件非目录 || (非递归 && 为一个目录文件)
		{
			_stprintf_s(pDW->szBuf,
				pDW->fIsDir ? _TEXT("%*s[%s]") : _TEXT("%*s%s"),
				pDW->nIndent,
				_TEXT(""),
				pDW->FindData.cFileName);

			//封装路径
			lstrcpy(src, pszRootPath);
			lstrcat(src, "\\");
			lstrcat(src, pDW->FindData.cFileName);

			lstrcpy(des, pszDesPath);
			lstrcat(des, "\\");
			lstrcat(des, pDW->FindData.cFileName);

			//创建/打开文件
			HANDLE desFile = CreateFile(
				des,
				GENERIC_READ | GENERIC_WRITE,
				0,
				NULL,
				OPEN_ALWAYS,
				pDW->FindData.dwFileAttributes,
				NULL);
			HANDLE srcFile = CreateFile(
				src,
				GENERIC_READ | GENERIC_WRITE,
				0,
				NULL,
				OPEN_EXISTING,
				pDW->FindData.dwFileAttributes,
				NULL);

			DWORD filesize = GetFileSize(srcFile, NULL);
			char* buffer = new char[filesize + 1];
			DWORD readsize;

			ReadFile(srcFile, buffer, filesize, &readsize, NULL);
			WriteFile(desFile, buffer, filesize, &readsize, NULL);
			buffer[filesize] = 0;
			CloseHandle(srcFile);
			CloseHandle(desFile);

			//printf("%s\n", pDW->szBuf);
		}
		else {
			SECURITY_ATTRIBUTES attribute;
			attribute.nLength = sizeof(attribute);
			attribute.lpSecurityDescriptor = NULL;
			attribute.bInheritHandle = FALSE;
			SetCurrentDirectory(pszDesPath);
			CreateDirectory(pDW->FindData.cFileName, &attribute);
			SetCurrentDirectory(pszRootPath);
		}

		pDW->fOk = FindNextFile(hFind, &pDW->FindData);							//system API
	}

	if (hFind != INVALID_HANDLE_VALUE)
	{
		FindClose(hFind);	
	}

	if (pDW->fRecurse)
	{
		//Get the first child directory
		hFind = FindFirstChildDir(_TEXT("*.*"), &pDW->FindData);				//找到第一个子目录 FindFirstFile
		pDW->fOk = (hFind != INVALID_HANDLE_VALUE);
		
		while (pDW->fOk)
		{
			//change into the child directory
			if (SetCurrentDirectory(pDW->FindData.cFileName))
			{
				//封装路径
				lstrcpy(src, pszRootPath);
				lstrcat(src, "\\");
				lstrcat(src, pDW->FindData.cFileName);

				lstrcpy(des, pszDesPath);
				lstrcat(des, "\\");
				lstrcat(des, pDW->FindData.cFileName);
				DirWalkRecurse(pDW, src, des);

				//change back to the child's parent dir
				SetCurrentDirectory(_TEXT(".."));
			}

			pDW->fOk = FindNextChildDir(hFind, &pDW->FindData);					//将hFind句柄传入 FindNextFile
		}

		if (hFind != INVALID_HANDLE_VALUE)
		{
			FindClose(hFind);
		}
	}

	pDW->nDepth--;
}