Password8ParallelRandomPasswordNOWAIT.cpp

#include <atomic>
#include <chrono>
#include <crypt.h>
#include <iostream>
#include <string>
#include <cstring>
#include <iomanip>
#include <omp.h>
#include <random>

///////////////////////
/// MAIN FUNCTION ///
/////////////////////
int main(int argc, char* argv[]) {
    //// THREAD SETUP ////
    int num_threads;
    int max_threads = 999;
    if (argc > 1) {
        num_threads = std::atoi(argv[1]);

        if (num_threads <= 0 || num_threads > max_threads) {
            std::cerr << "ERRORE: Numero di thread non valido!\n";
            std::cerr << "Richiesto: " << num_threads << "\n";
            std::cerr << "Massimo disponibile sul sistema: " << max_threads << "\n";
            std::cerr << "Uso: " << argv[0] << " [num_threads]\n";
            return 1;
        }
    } else {
        std::cout << "=================================================\n";
        std::cout << "  Password Decryption - Brute Force Parallel\n";
        std::cout << "=================================================\n";
        std::cout << "Massimo numero di thread disponibili: " << max_threads << "\n";
        std::cout << "Inserisci il numero di thread da utilizzare (1-" << max_threads << "): ";
        std::cin >> num_threads;
        std::cout << "\n";
    }
    if (num_threads == 1) {
        std::cout << "\n⚠️  ATTENZIONE: Hai selezionato 1 thread!\n";
        std::cout << "Per prestazioni ottimali con esecuzione sequenziale,\n";
        std::cout << "usa il programma dedicato: ./Password8Sequenziale\n";
        std::cout << "(versione pura senza overhead OpenMP)\n\n";
    }

    //// SET NUMBER OF THREADS ////
    omp_set_num_threads(num_threads);

    //// HASH VARIABLES ////
    const std::string salt = "AB";
    const char* salt_cstr = salt.c_str();
    std::string found;
    std::atomic<bool> found_flag(false);
    bool correct = true;
    static std::random_device rd;
    static std::mt19937 gen(rd());
    //// TIMER START ////
    const auto start = std::chrono::high_resolution_clock::now();

    //// MAIN LOOP VARIABLES ////
    constexpr int NUM_ITER = 500;
    constexpr long long PASSWORDS_PER_ITER = 32LL * 13 * 2026;
    long long total_passwords_tested = 0;

    //// MAIN LOOP ////
    for (int i = 0; i < NUM_ITER; i++) {
        int numberIter = 0;
        found = "";
        found_flag.store(false, std::memory_order_release);

        /// GENERATE RANDOM PASSWORD TARGET ////
        std::uniform_int_distribution<int> giorno(1, 31);
        std::uniform_int_distribution<int> mese(1, 12);
        std::uniform_int_distribution<int> anno(0, 2025);
        const int g = giorno(gen);
        const int m = mese(gen);
        const int y = anno(gen);
        char target_password[9];
        target_password[0] = '0' + (g / 10);
        target_password[1] = '0' + (g % 10);
        target_password[2] = '0' + (m / 10);
        target_password[3] = '0' + (m % 10);
        target_password[4] = '0' + ((y / 1000) % 10);
        target_password[5] = '0' + ((y / 100) % 10);
        target_password[6] = '0' + ((y / 10) % 10);
        target_password[7] = '0' + (y % 10);
        target_password[8] = '\0';

        /// CRYPT TARGET PASSWORD ////
        const char* target = crypt(target_password, salt_cstr);

        /// BRUTE FORCE SEARCH WITH OPENMP NOWAIT ////
#pragma omp parallel default(none) shared(salt_cstr, found, target, found_flag) firstprivate(numberIter) reduction(+:total_passwords_tested) num_threads(25)
        {
            //// BUFFER FOR crypt_r ////
            struct crypt_data data{};
            data.initialized = 0;

            //// PARALLEL FOR WITH NOWAIT ////
#pragma omp for collapse(3) schedule(static) nowait
            for (int a = 0; a <= 31; ++a) {
                for (int b = 0; b <= 12; ++b) {
                    for (int c = 0; c <= 2025; ++c) {
                        if (found_flag.load(std::memory_order_relaxed)) {
                            continue; //// NO CANCELLATION POINT ////
                        }
                        numberIter++;
                        char date[9];
                        date[0] = '0' + (a / 10);
                        date[1] = '0' + (a % 10);
                        date[2] = '0' + (b / 10);
                        date[3] = '0' + (b % 10);
                        date[4] = '0' + ((c / 1000) % 10);
                        date[5] = '0' + ((c / 100) % 10);
                        date[6] = '0' + ((c / 10) % 10);
                        date[7] = '0' + (c % 10);
                        date[8] = '\0';

                        /// CRYPT GENERATED DATE ////
                        const char* h = crypt_r(date, salt_cstr, &data);

                        /// CHECK IF HASH MATCHES TARGET ////
                        if (h[3] == target[3] && h[4] == target[4]) {
                            if (strcmp(h, target) == 0) {
#pragma omp critical //// CRITICAL SECTION ////
                                {
                                    if (!found_flag.load(std::memory_order_relaxed)) {
                                        found = date;
                                        found_flag.store(true, std::memory_order_release);
                                    }
                                }
                            }
                        }
                    }
                }
            }
            total_passwords_tested += numberIter;
        }
    }

    //// TIMER END////
    auto end = std::chrono::high_resolution_clock::now();
    std::chrono::duration<double> elapsed = end - start;
    double passwords_per_second = total_passwords_tested / elapsed.count();
    //////// STATISTICS OUTPUT /////////
    std::cout << "========================================\n";
    std::cout << "Elaborazione completata!\n";
    std::cout << "========================================\n";
    std::cout << "Tempo totale impiegato: " << std::fixed << std::setprecision(2) << elapsed.count() << " secondi\n";
    std::cout << "Tempo medio per iterazione: " << std::fixed << std::setprecision(3) << (elapsed.count()/NUM_ITER) << " secondi\n";
    std::cout << "Iterazioni totali: " << NUM_ITER << "\n";
    std::cout << "Password testate totali: " << total_passwords_tested << "\n";
    std::cout << "Password testate/secondo: " << std::fixed << std::setprecision(0) << passwords_per_second << "\n";
    std::cout << "Thread utilizzati: " << num_threads << "\n";
    std::cout << "========================================\n";
    if (!found.empty()) {
        printf("✓ Password trovata: %s\n", found.c_str());
    } else {
        printf("✗ Password non trovata\n");
        correct=false;
    }
    //// VERIFY CORRECTNESS////
    if (correct) {
        std::cout << "✓ Tutte le password generate sono corrette\n";
    } else {
        std::cout << "✗ Alcune password generate non sono corrette\n";
    }

    return 0;
}