Антивирус

#include <windows.h>
#include <iostream>
#include <string>
#include <map>
 
#define BUFSIZE 1024
#define MD5LEN  16
 
int main()
{
    setlocale(LC_ALL, "Russian");
 
    WIN32_FIND_DATA winFileData;
    DWORD dwStatus = 0;
    DWORD cbRead = 0;
    DWORD cbHash = 0;
    HCRYPTPROV hProv = 0;
    HCRYPTHASH hHash = 0;
    HANDLE dFile = NULL;
    HANDLE hFile = NULL;
    BOOL bResult = FALSE;
    BYTE rgbFile[BUFSIZE];
    BYTE rgbHash[MD5LEN];
    CHAR rgbDigits[] = "0123456789abcdef";
    CHAR md5str[MD5LEN * 2 + 1] = "\0";
    INT allfiles = 0, copies = 0;
    WCHAR szPath[MAX_PATH], exp[10];
    WCHAR szPathM[MAX_PATH];
    WCHAR szPathTo[MAX_PATH] = L"Копии";
    int i = 0;
 
    std::multimap <std::string, std::wstring> md5filename;
 
    if (!GetCurrentDirectoryW(sizeof(szPath), szPath))
    {
        std::cout << L"Невозможно получить текущую папку" << std::endl;
        system("pause");
        return 0;
    }
    std::wcout << L"Текущая папка : " << szPath << L'\\' << std::endl << std::endl << L"Введите расширение файлов : ";
 
    lstrcat(szPath, L"\\*.");
 
    std::wcin >> exp;
 
    lstrcat(szPath, exp);
 
    std::wcout << std::endl;
 
    if ((dFile = FindFirstFile(szPath, &winFileData)) == INVALID_HANDLE_VALUE)
    {
        std::cout << L"Нет файлов в папке" << std::endl << std::endl;
        system("pause");
        return 0;
    }
    while (!CreateDirectory(szPathTo, NULL)) wsprintf(szPathTo, L"Копии (%i)", ++i); // Создать папку для копий
 
 
    do // Цикл по файлам
    {
        if (winFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) continue; // Если это не файл - тогда дальше
 
        hFile = CreateFile(winFileData.cFileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
 
        if (hFile == INVALID_HANDLE_VALUE) // Попытка открыть файл для чтения 
        {
            std::wcout << L"Невозможно открыть файл " << winFileData.cFileName << std::endl;
            continue;
        }
        if (!CryptAcquireContext(&hProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
        {
            dwStatus = GetLastError();
            std::wcout << std::endl << L"Ошибка, не удалось получить контекст: " << dwStatus << std::endl << std::endl;
            CloseHandle(hFile);
            continue;
        }
        if (!CryptCreateHash(hProv, CALG_MD5, 0, 0, &hHash))
        {
            dwStatus = GetLastError();
            std::wcout << std::endl << L"Ошибка, не удалось получить контекст: " << dwStatus << std::endl << std::endl;
            CloseHandle(hFile);
            CryptReleaseContext(hProv, 0);
            continue;
        }
        while (bResult = ReadFile(hFile, rgbFile, BUFSIZE, &cbRead, NULL))
        {
            if (cbRead == 0) break;
 
            if (!CryptHashData(hHash, rgbFile, cbRead, 0))
            {
                dwStatus = GetLastError();
                std::wcout << std::endl << L"Ошибка, не удалось получить хэш: " << dwStatus << std::endl << std::endl;
                CryptReleaseContext(hProv, 0);
                CryptDestroyHash(hHash);
                CloseHandle(hFile);
                continue;
            }
        }
        if (!bResult)
        {
            dwStatus = GetLastError();
            std::wcout << std::endl << L"Ошибка, не удалось прочитать файл: " << dwStatus << std::endl << std::endl;
            CryptReleaseContext(hProv, 0);
            CryptDestroyHash(hHash);
            CloseHandle(hFile);
            continue;
        }
        cbHash = MD5LEN;
 
        if (CryptGetHashParam(hHash, HP_HASHVAL, rgbHash, &cbHash, 0))
        {
 
            for (int j = 0, i = 0; i < cbHash; i++)
            {
                md5str[j++] = rgbDigits[rgbHash[i] >> 4];
                md5str[j++] = rgbDigits[rgbHash[i] & 0xf];
            }
            md5filename.insert(std::pair<std::string, std::wstring>((std::string)md5str, (std::wstring)winFileData.cFileName));
            allfiles++;
        }
        else
        {
            dwStatus = GetLastError();
            std::wcout << std::endl << L"Ошибка, не удалось получить параметры : " << dwStatus << std::endl << std::endl;
        }
        CryptDestroyHash(hHash);
        CryptReleaseContext(hProv, 0);
        CloseHandle(hFile);
    } while (FindNextFile(dFile, &winFileData));
 
    std::wcout << std::endl << std::endl << "-----------------------------------------------------" << std::endl;
 
    std::string previos, first; // Для хранения предыдущего и текущего хэша
    std::wstring pname, fname;  // Для хранения предыдущего и текущего имёни файла
 
    for (auto it = md5filename.begin(); it != md5filename.end(); it++) // Вывод одинаковых файлов
    {
        first = it->first;          // Запись текущего хэша
 
        if (first == previos)       // Если текущий хэш равен предыдущему
        {
            fname = it->second;     // Получить имя файла (текущее)
            std::wcout << fname << std::endl;   // Вывести имя файла
 
            wsprintfW(szPathM, L"%s\\%s", szPathTo, fname.c_str()); // Форматирование нового пути файла
            MoveFileEx(fname.c_str(), szPathM, MOVEFILE_REPLACE_EXISTING);  // Переместить копию в папку копий
 
            copies++;               // Посчитать копию
        }
        else                        // Если хэш не равен предыдущему
        {
            previos = first;        // Записать новый хеш
            pname = it->second;     // Получить имя файла (предыдущее)
 
            std::cout << std::endl << first;
            std::wcout << L" : Копий - " << md5filename.count(first) << std::endl; // Вывести хеш и число копий
            std::wcout << pname << std::endl; // Вывести имя файла
        }
    }
    std::wcout << std::endl << std::endl << L"Всего найдено файлов - " << allfiles << L" Найдено копий - " << copies << std::endl << std::endl;
 
    if (!copies) RemoveDirectory(szPathTo);
    system("pause");
    return 0;
}