поиск файлов по маске в каталоге c++ - C++ - Ответ 5424306

#include "stdafx.h"
#include "Windows.h" 
#include <iostream>
#include <string> 
#include <fstream>
 
#define MAXTHREAD 10
#define LogName "log.txt"
 
HANDLE hSemaphore; 
 
//чтение
void readFile()
{
    FILE *file;
    char arr[20];
    file = fopen(LogName, "r");
    while (fgets (arr, 20, file) != NULL)
    printf("%s", arr);
    printf("\n");
    fclose(file);
}
 
//запись
void appendLog(const char* filename)
{
     FILE* log = fopen(LogName,"a"); 
     fputs(filename,log);
     fputs("\r\n",log);
     fclose(log);
}
 
DWORD WINAPI FolderSearch( LPVOID param)
{
    WIN32_FIND_DATAA fd;
    //ищем первый файл
    HANDLE hf = FindFirstFileA(
                        (LPCSTR) param, // имя файла
                        &fd             // буфер данных
                              );
    if (hf != INVALID_HANDLE_VALUE)
    {         
            //захватываем и/или ожидаем семафор
            WaitForSingleObject(hSemaphore, INFINITE);  
            //записываем в лог
            appendLog(fd.cFileName);
            //отпускаем семафор +1      
            ReleaseSemaphore(hSemaphore,1,NULL);
        
            while ( FindNextFileA(hf, &fd) ) 
            { 
                //захватываем и/или ожидаем семафор
                WaitForSingleObject(hSemaphore, INFINITE);  
                //записываем в лог                
                appendLog(fd.cFileName);
                //отпускаем семафор +1
                ReleaseSemaphore(hSemaphore,1,NULL);
            } 
            //освобождаем ресурсы поиска файлов
            FindClose(hf); 
    }
    return 0;
}
 
int main(int argc, _TCHAR* argv[])
{
    
    printf("Input file mask for search:  ");
    std::string mask;
    //ввод маски
    std::cin >> mask;
    printf("Input folders count: ");
    int foldersCount;
    //ввод количества папок
    std::cin >> foldersCount;
    std::string folders[MAXTHREAD];
    //запрос папок
    for (int i=0; i<foldersCount; i++){
        printf("Input folder %d with \\:",i+1);
        //ввод папки
        std::cin >> folders[i];
    }
    //массив хэндлов потоков
    HANDLE aThread[MAXTHREAD];
    
    //содаём семафор с счётчиком 1, максимальное кол-во обращений - 1
    hSemaphore = CreateSemaphore(NULL,1,1,NULL);  
 
    //делаем пустой файл
     FILE* log = fopen(LogName,"w"); 
     fclose(log);
 
    //запускаем потоки с поиском
    DWORD ThreadID;
    for (int i=0; i<foldersCount; i++){
        folders[i].append(mask);
        aThread[i] = CreateThread( 
                        NULL,                                   // дескриптор защиты
                        0,                                      // начальный размер стека
                        (LPTHREAD_START_ROUTINE) FolderSearch,  // функция потока 
                        (LPVOID)folders[i].c_str(),             // параметр потока
                        0,                                      // опции создания
                        &ThreadID                               // идентификатор потокка
                                 );
    }
 
    //ждём пока все потоки завершатся
    WaitForMultipleObjects(foldersCount, aThread, TRUE, INFINITE);
 
    //освобождаем ресурсы потоков
    for(int i=0; i < foldersCount; i++ ) CloseHandle(aThread[i]);
 
    //освобождаем ресурсы семафора
    CloseHandle(hSemaphore);
    printf("Found files:\n");
    readFile();
    system("Pause");
    return 0;
}