robocop_funcs.cpp

#include <QFile>
#include <QDebug>
#include <QTextStream>
#include <QString>
#include <QProcess>
#include "player.h"
#include "team.h"
#include <QDirIterator>
#include "robocop_funcs.h"
#include <QRegExp>

enum STATION_PROPERTIES
{
    STATION_MAC = 0,
    FIRST_TIME_SEEN = 1,
    LAST_TIME_SEEN = 2,
    POWER = 3,
    PACKETS = 4,
    BYTES  = 5,
};

enum PLAYER_PROPERTIES{
    PLAYER_MAC = 0,
    PLAYER_NAME = 1,
};

#define MIN_PROPERTIES 6

vector<player *>   AllPlayers;
vector<team>     AllTeams;
static player* findStation(QString sta_mac);
static void checkTeams();





bool get_team_by_file(QString team_file){

    QFile team_descriptor(team_file);
    QTextStream tstream(&team_descriptor);
    team* team1;

    if(team_descriptor.open(QIODevice::ReadOnly | QIODevice::Text)!= true){
        qDebug() << "Error, Couldn't open file" << team_file << endl;
        return false;
    }

    while(!tstream.atEnd()){

        QString line = tstream.readLine();

        if(line==NULL)
        {
            qDebug() << "line is NULL so skip" << endl;
            continue;
        }

        if(line.at(0)=='#')
            continue;

        if(line.startsWith("Team Name:") == true){
            QStringList tname = line.split("Team Name:", QString::SkipEmptyParts);
            qDebug() << "Found Team Name: " <<  qPrintable((tname.at(0)).trimmed()) << endl;

            if(get_team_by_name(tname.at(0)) != NULL){
                qDebug("Team %s Already Exists in the system, will skip", qPrintable(tname.at(0).trimmed()));
                return false;
            }

            ROBOLOG_INFO("%s: Found TEAM: %s", TEAM_INFO, qPrintable(tname.at(0)));

            team1 = new team(tname.at(0));

            while(!tstream.atEnd()){

                QString line2 = tstream.readLine();

                if(line2==NULL || (line2.trimmed().isEmpty()==true))
                {
                    qDebug() << "line is NULL so skip" << endl;
                    continue;
                }
                if(line2.at(0)=='#')
                    continue;

                QStringList tplayer = line2.split(",", QString::SkipEmptyParts);
                if(tplayer.size() < 2){
                    qDebug() << "Invalid Player: " << qPrintable(line2) << endl;
                    continue;
                }

                qDebug() << "Found Candidate for Player: " <<  qPrintable((tplayer.at(1)).trimmed()) << " with MAC: "
                           <<  qPrintable((tplayer.at(0)).trimmed()) << endl;
                player *playerx;

                if(isValidMACaddr(tplayer.at(PLAYER_MAC).trimmed())==false){
                        qDebug("MAC %s is Invalid, skipping", qPrintable(tplayer.at(PLAYER_MAC).trimmed()));
                        continue;
                }



                if((playerx = findStation(tplayer.at(PLAYER_MAC).trimmed()))==NULL){
                    qDebug("STA_MAC %s not already in the player LIST will INSERT\n",qPrintable(tplayer.at(PLAYER_MAC).toUpper().trimmed()) );
                    player* player1 = new player(tplayer.at(PLAYER_NAME).trimmed(), tplayer.at(PLAYER_MAC).trimmed(), tname.at(0).trimmed());
                    AllPlayers.push_back(player1);
                    team1->insertPlayer(player1);
                    ROBOLOG_INFO("%s: Team %s PLAYER %s , %s ", TEAM_INFO, qPrintable(tname.at(0).trimmed()), qPrintable(tplayer.at(PLAYER_NAME).trimmed()), qPrintable(tplayer.at(PLAYER_MAC).trimmed()));

                }
                else{
                    qDebug("MAC %s already exists in data base in team %s, will skip",qPrintable(tplayer.at(PLAYER_MAC).trimmed()), qPrintable(playerx->team_name()));
                    continue;
                }

            }
        }
    }

    AllTeams.push_back(*team1);
    delete team1;
    team_descriptor.close();



    return true;
}

void get_all_teams(void)
{
    QDirIterator teamDir("./team", QDir::Files | QDir::NoDotAndDotDot | QDir::NoSymLinks, QDirIterator::NoIteratorFlags);
    while (teamDir.hasNext()) {
        get_team_by_file(teamDir.next());
    }
}

void printAllPlayers(void){

    qDebug() << "###############ALL PLAYERS LIST #################" << endl;

    for(unsigned int i = 0; i < AllPlayers.size(); i++){
        qDebug("Player %3d, Name %20s, MAC %20s, Team %10s, First Time Seen %s, Last Time Seen %s, Power %d dBm, Packets %d\n",
               i, qPrintable(AllPlayers[i]->name()), qPrintable(AllPlayers[i]->mac()),
               qPrintable(AllPlayers[i]->team_name()),
               qPrintable(AllPlayers[i]->firstTimeSeen().toString("dd-MM-yyyy hh:mm:ss")),
               qPrintable(AllPlayers[i]->lastTimeSeen().toString("dd-MM-yyyy hh:mm:ss")), AllPlayers[i]->power(), AllPlayers[i]->packets());
    }

}

void printAllTeams(void){

    qDebug() << "###############ALL TEAMS LIST #################" << endl;

    for(unsigned int i = 0; i<AllTeams.size(); i++){
       qDebug("**TEAM %s, size %d\n", qPrintable(AllTeams[i].name()), (int)AllTeams[i].get_team_size());
       for(int j = 0; j < AllTeams[i].get_team_size(); j++){
            player *playerx;
            AllTeams[i].get_player(j, &playerx);
            qDebug("Player %3d, Name %20s, MAC %20s, Team %10s\n", j, qPrintable(playerx->name()), qPrintable(playerx->mac()), qPrintable(playerx->team_name()));
       }
      }
}

void parseNetCapture(QString capture_file){

    QFile capture_descriptor(capture_file);
    QTextStream cap_stream(&capture_descriptor);

    if(capture_descriptor.open(QIODevice::ReadOnly | QIODevice::Text)!= true){
        qDebug() << "Error, Couldn't open file" << qPrintable(capture_file) << endl;
        return;
    }

    while(!cap_stream.atEnd()){
        QString line = cap_stream.readLine();
        //qDebug("%s\n",qPrintable(line));
       if(line.startsWith("Station MAC")==false)
           continue;
       while(!cap_stream.atEnd()){
           QString line2 = cap_stream.readLine();
           QStringList sta_props = line2.split(",", QString::SkipEmptyParts);
           if(sta_props.size()<MIN_PROPERTIES){
               qDebug("Invalid STATION %s\n",qPrintable(line2));
               continue;
           }
           QString sta_mac = sta_props.at(STATION_MAC).trimmed();


           QDateTime first_seen = QDateTime::fromString(sta_props.at(FIRST_TIME_SEEN).trimmed(),"yyyy-MM-dd hh:mm:ss");
           QDateTime last_seen = QDateTime::fromString(sta_props.at(LAST_TIME_SEEN).trimmed(),"yyyy-MM-dd hh:mm:ss");
           int power = sta_props.at(POWER).toInt();
           uint packets = sta_props.at(PACKETS).toUInt();
           uint bytes = sta_props.at(BYTES).toUInt();
           qDebug("##Candidate Station Found \n");
           qDebug("STA_MAC %s, First Time Seen %s, Last Time Seen %s, Power %d dBm, Packets %u, Bytes %u \n",
                  qPrintable(sta_mac), qPrintable(first_seen.toString("dd-MM-yyyy hh:mm:ss")), qPrintable(last_seen.toString("dd-MM-yyyy hh:mm:ss")), power, packets, bytes);

           if(isValidMACaddr(sta_mac)==false){
                   qDebug("MAC %s is Invalid, skipping",qPrintable(sta_mac));
                   continue;
           }
           player *player1;

           if((player1 = findStation(sta_mac) ) != NULL){
               qDebug("Station is already at database with %s, %s, %s", qPrintable((player1->mac())), qPrintable((player1->name())), qPrintable(player1->team_name()));
               player1->update(first_seen, last_seen, packets, power, bytes);
               qDebug("%s is transmitting %d pakets/s\n", qPrintable((player1->name())), (int)player1->pkts_second());
               qDebug("%s is transmitting %d bytes/sec \n", qPrintable((player1->name())), (int)player1->throughput());
               qDebug("Player %s - %s of team %s is %s", qPrintable(player1->mac()), qPrintable(player1->name()),
               qPrintable(player1->team_name()),(player1->isConnected()== true)?"connected" : "disconnected");
               team *p_team;

               if(player1->isConnected() ==  true){
                   if(((p_team = get_team_by_name(player1->team_name()))!=NULL) && (p_team->is_inGame() == true))
                       ROBOLOG_INFO("%s: Player %s, %s - %s is connected and transmitting %d kbits/s", STATION_INFO, qPrintable(player1->name()), qPrintable(player1->mac()), qPrintable(player1->team_name()),player1->throughput()*8/1000);
                   else
                       if(((p_team = get_team_by_name(player1->team_name()))!=NULL) && (p_team->is_inGame() == false))
                               ROBOLOG_ALERT("%s: OUT OF GAME Player/Station %s, %s - %s is connected and transmitting %d kbits/s", STATION_ALERT, qPrintable(player1->name()), qPrintable(player1->mac()), qPrintable(player1->team_name()),player1->throughput()*8/1000);
                   else
                           if(((p_team = get_team_by_name(player1->team_name()))==NULL))
                           ROBOLOG_ALERT("%s: OUT OF GAME Player/Station %s, %s - %s is connected and transmitting %d kbits/s", STATION_ALERT, qPrintable(player1->name()), qPrintable(player1->mac()), qPrintable(player1->team_name()), player1->throughput()*8/1000);
               }


           }else{
               qDebug("New Station Detected in Capture FILE %s, will insert in database and update stats \n", qPrintable(sta_mac));
               player *sta_new = new player(sta_mac);
               sta_new->update(first_seen, last_seen, packets, power, bytes);
               AllPlayers.push_back(sta_new);
               qDebug("%s is transmitting %d pakets/s\n", qPrintable((sta_new->name())), (int)sta_new->pkts_second());
               qDebug("%s is transmitting %d bytes/sec \n", qPrintable((sta_new->name())), (int)sta_new->throughput());
               qDebug("Player %s is %s", qPrintable(sta_new->mac()), (sta_new->isConnected()== true)?"connected" : "disconnected");

               ROBOLOG_ALERT("%s: New Station Detected %s, transmiting %d kbits/sec", STATION_ALERT, qPrintable(sta_new->mac()), sta_new->throughput()*8/1000);
           }

       }
    }
    capture_descriptor.close();
    checkTeams(); //update team info
}

static player* findStation(QString sta_mac){

    for(unsigned int i = 0; i < AllPlayers.size(); i++){
        if(QString::compare(sta_mac, AllPlayers[i]->mac(), Qt::CaseInsensitive) == 0)
            return (AllPlayers[i]);
    }

    return NULL;


}

team* get_team_by_name(QString tname){

    for(unsigned int i = 0; i < AllTeams.size(); i++){
        if(tname.trimmed() == AllTeams[i].name())
                return &AllTeams[i];
    }

    return NULL;
}

void print_team(team *a_team){
    if(a_team!=NULL){

    for(int i = 0; i<a_team->get_team_size(); i++){

        player *playerx;
        a_team->get_player(i, &playerx);
        qDebug("Player %3d, Name %20s, MAC %20s, Team %10s\n", i, qPrintable(playerx->name()), qPrintable(playerx->mac()), qPrintable(playerx->team_name()));

    }

   }

}

void clean_AllPlayers_stat(void){

    for(uint i = 0; i < AllPlayers.size(); i++){
        AllPlayers[i]->clean_stats();
    }
}

void clean_AllTeam_stat(void){
    for(uint i = 0; i < AllTeams.size(); i++){
        AllTeams[i].clean_stats();
    }
}


bool start_iw_mon(QString iw){
    QString command;
    command = QString::asprintf("sudo airmon-ng start %s", qPrintable(iw));

    return (QProcess::execute(command)==QProcess::NormalExit);

}

bool stop_iw_mon(QString iw){
    QString command;
    command = QString::asprintf("sudo airmon-ng stop %s", qPrintable(iw));
    return (QProcess::execute(command)==QProcess::NormalExit);
}

bool isValidMACaddr(QString new_mac){
    QRegExp macValidate("^([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})$", Qt::CaseInsensitive, QRegExp::RegExp);

    return macValidate.exactMatch(new_mac);
}

static void checkTeams(){

    for(uint i = 0; i<AllTeams.size(); i++){
        if(AllTeams[i].is_inGame() == true){
            AllTeams[i].updateTeam();
            if(AllTeams[i].bwAlarmed()){
                qDebug("ALARM: Team %s is over bw limit spending %d", qPrintable(AllTeams[i].name()),AllTeams[i].throughput()*8/1000);
                ROBOLOG_ALERT("%s Team %s is overspending bw with kbit/s %d",BW_ALERT,qPrintable(AllTeams[i].name()),AllTeams[i].throughput()*8/1000);

            }else{
                qDebug("INFO: Team %s is spending %d", qPrintable(AllTeams[i].name()),AllTeams[i].throughput()*8/1000);
                ROBOLOG_INFO("%s Team %s is spending bw with kbit/s %d",BW_INFO, qPrintable(AllTeams[i].name()), AllTeams[i].throughput()*8/1000);


            }

            }

    }





}