dirforce.cpp

#include <iostream>
#include <string>
#include <cstring>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <vector>
#include <thread>
#include <cstdlib>
#include <unistd.h>
#include <mutex>
#include <fstream>
#include <openssl/bio.h>
#include <openssl/ssl.h>
#include <openssl/err.h>
#include <iomanip>
#include <sstream>

using namespace std;
string version = "v0.3";

int wordlistcount;
ifstream wordlist;
int sslOption; //check if ssl is enabled
string host;
string address;
mutex wordlist_lock;
mutex progresslock;
int progressCounter = 0;
string percent;
string http_method;
int requestOption;
string readrequest;
string suffixe;
string preffixe;
string port;
string wordlistfile;
int threadCounter;
bool filterOption = false;
string filterStr;
bool liteOutput = false;
bool robotxtOption = true;
vector<string> extension;
string custom_header_str;

#include "progress.cpp"
#include "lineCounter.cpp"
#include "header_extractor.cpp"
#include "request.cpp"
#include "arguments.cpp"
#include "robotxt.cpp"

int main(int argc, char* argv[]) {
    arguments(argc, argv);
    wordlist.open(wordlistfile);
    wordlistcount = lineCounter(wordlistfile);
    if (!liteOutput) {
        cout << "      ██ ██          ████                            " << version
             << endl << "     ░██░░          ░██░                                 "
             << endl << "     ░██ ██ ██████ ██████  ██████  ██████  █████   █████ "
             << endl << "  ██████░██░░██░░█░░░██░  ██░░░░██░░██░░█ ██░░░██ ██░░░██"
             << endl << " ██░░░██░██ ░██ ░   ░██  ░██   ░██ ░██ ░ ░██  ░░ ░███████"
             << endl << "░██  ░██░██ ░██     ░██  ░██   ░██ ░██   ░██   ██░██░░░░ "
             << endl << "░░██████░██░███     ░██  ░░██████ ░███   ░░█████ ░░██████"
             << endl << " ░░░░░░ ░░ ░░░      ░░    ░░░░░░  ░░░     ░░░░░   ░░░░░░ \n";
    
        cout << " Created by Guendouz Aimed \n youtube: www.youtube.com/@code0109 \n instagram: www.instagram.com/cod_e010\n";
        cout << "=========================================================\n\n"
             << "url : " << host << address << endl 
             << "wordlist : " << wordlistfile << " line : " << wordlistcount * extension.size() << endl 
             << "http method : " << http_method << "\n"
             << "thread : " << threadCounter << "\n"
             << "extension : ";
        for (auto& e : extension) {
            cout << ", " << e;
        }
        cout << "\n\n=========================================================\n\n";
    }

    vector<thread> threads;
    cout << "\033[1;37m[-] brute-forcing directory: \n\033[0;37m";
    // Resolve the host name to an IP address
    struct addrinfo hints = {0};
    hints.ai_family = AF_INET;
    hints.ai_socktype = SOCK_STREAM;
    struct addrinfo* result;
    if (sslOption == 1) port = "443"; else port = "80";
    int rc = getaddrinfo(host.c_str(), port.c_str(), &hints, &result);
    if (rc != 0) {
        std::cerr << "Error resolving hostname " << gai_strerror(rc) << "   " << port << host << std::endl;
        exit(1);
        return -1;
    }
    stringstream output;
    output << left << setw(8) << setfill(' ') << "status" << setw(custom_header_str.size() + 2) << setfill(' ') <<  custom_header_str << setw(15) << setfill(' ') << "path" << setw(15) << setfill(' ') << "redirect\n";
    cout << output.str() << endl;
    for (int i = 0; i < threadCounter; i++) {
        threads.emplace_back([&] {request(result->ai_addr, result->ai_addrlen);});
    }

    for(auto& t : threads) {
        t.join();
    }

    if (robotxtOption)
        robotxt(host, sslOption);

    printf("\n\n\t\033[1;31mscaning completed");
    return 0;
}