parser.cpp

#include <iostream>
#include <fstream>
#include <string>
#include <cstring>
#include <vector>
#include <map>
#include <iomanip>
#include <sstream>

using namespace std;
//using namespace ofstream;

//конвертор hex байта string в unsigned char...
unsigned char hstob(char data)
{
    if(data>=48 && data<=57)
        return (unsigned char)(data-48);
    if(data>=65 && data<=70)
        return (unsigned char)(data-55);
    if(data>=97 && data<=102)
        return (unsigned char)(data-87);

    return 0;
}

//распечатка сортировки...
void print_sort(map<unsigned int, vector<vector<unsigned char>>> & data)
{
    for(pair plspair : data)
    {
        cout << "size " << plspair.first << " | contains objects " << plspair.second.size() << endl;
    }
}

//печать заголовка, нумерация байт...
void print_head(unsigned int key)
{
    for(unsigned int i=0;i<=key;i++)
        cout << setw(2) << std::setfill('0') << i << " ";
    cout << endl;

    for(unsigned int i=0;i<=((key+1)*2)+key;i++)
        cout << "-";
    cout << endl;
}

//печать отсортированной группы по ключу...
void print_group(map<unsigned int, vector<vector<unsigned char>>> & data, unsigned int key)
{
    print_head(key);

    vector<vector<unsigned char>> & outer_vec = data[key];

    // Вывод элементов векторов  
    for (const auto & inner_vec : outer_vec) 
    {  
        for (unsigned char value : inner_vec) {  
            cout << hex << setw(2) <<static_cast<int>(value) << " " << dec;  
        }  
        cout << endl;  
    }
}

void analitic(map<unsigned int, vector<vector<unsigned char>>> & data, unsigned char key)
{
    cout << ">>>> analitic..." << endl;

    vector<vector<unsigned char>> & outer_vec = data[key];

    //vector<unsigned int> differences;
    map<unsigned int,unsigned int> differences;

    //ищем несовпадения...
    for(unsigned int i=0;i<key;i++)
    {
        unsigned char temp = data[key][0][i];

        for(vector<unsigned char> & inner_vec : outer_vec) 
        {  
            if(temp!=inner_vec[i])
            {
                if(!differences.count(temp))
                {
                    differences.insert(make_pair(i,i));
                }
            }
        }
    }

        if(differences.size())
        {
            cout << "found " << differences.size() << " differences" << endl;
            cout << "are different ";

            for(pair tmp : differences)
            {
                cout << dec << tmp.first << " ";
            }

            cout << "bytes" << endl;
        }
        else
            {
                cout << ">>>> payload equial!" << endl;
            }
}

void print_help()
{
    cout << endl << "<<< JSON parser + sort + find differences v0.1.1>>>" << endl << endl;
    cout << "USING: parser <filename>" << endl << endl;
    cout << "COMMANDs:..." << endl;
    cout << "  prints | ps       - print result sort" << endl;
    cout << "  printg | pg <grp> - print group payload data lenght" << endl;
    cout << "analitic | al <grp> - analysis differences in group" << endl;
    cout << "   clear | cls      - clear console" << endl;
    cout << "    quit | q        - exit programm" << endl << endl;
}

int main(int argc, char* argv[])
{
    /*
    *   обработка аргументов
    */
    if(argc<2)
    {
        print_help();
        exit(0);
    }
    
    if(!memcmp(argv[1],"--help",6))
    {
        print_help();
        exit(0);
    }


    string path = argv[1];

    /*
    *   открываем файл
    */
    ifstream fin;
    fin.open(path);
    
    if(fin.is_open())
    {
        cout << "file " << path << " is open" << endl;
    }
    else
        {
            cout << "file " << path << " not exist" << endl;
            exit(-1);
        }
    
    vector<string> json_str;
    vector<string> payload_str;
    vector<vector<unsigned char>> payload_hex;

    /*
    *   вычитываем строки из файла в вектор
    */ 
    while(!fin.eof())
    {
        string strtmp = "";
        getline(fin,strtmp);
        json_str.push_back(strtmp);
    }

    cout << "lines readed " << json_str.size() << "..."<< endl;

    int index = 0;

    /*
    *   чистим строки
    */
    for(string strtmp : json_str)
    {
        if(strtmp.find("udp.payload")!=string::npos)
        {
            index++;
            strtmp.erase(0,26);

            for(auto i=strtmp.begin();i!=strtmp.end();i++)
                if(*i==':') strtmp.erase(i);

            payload_str.push_back(strtmp);
            //cout << strtmp << endl << endl;
        }
    }

    cout << "find " << index << " string..." << endl;

    /*
    *   конвертирование hex строки в число...
    */
    index = 0;
    for(string strtmp : payload_str)
    {
        index++;
        vector<unsigned char> temp;

        for(size_t i=0;i<strtmp.size();i+=2)
        {
            unsigned char _hex = (unsigned char)(hstob(strtmp[i])*16 + hstob(strtmp[i+1]));
            temp.push_back(_hex);
        }
        
        payload_hex.push_back(temp);
    }

    cout << "convert " << index << " string..." << endl;

    /*
    *   сортировка...
    */
    map<unsigned int, vector<vector<unsigned char>>> payload_sort;

    for(vector<unsigned char> uchtmp : payload_hex)
    {
        unsigned int size = (unsigned int)uchtmp.size()-1;

        if(payload_sort.count(size))
        {
            payload_sort[size].push_back(uchtmp);
        }
        else
            {
                // Используем emplace для создания ключа и первоначального значения
                //без make_pair!
                payload_sort.emplace(size, vector<vector<unsigned char>>());  
    
                // Добавляем вектор в значение по ключу 1  
                payload_sort[size].push_back(uchtmp);  
            }
    }

    cout << "sorted " << payload_sort.size() << " uniq elements..." << endl;

    /*
    *   командный блок...
    */
    string cmd;
    string arg1(10,'\0');
    string arg2(10,'\0');
    
    while(1)
    {
        cmd.clear();
        getline(cin, cmd);

        sscanf(cmd.c_str(),"%s %s",arg1.data(),arg2.data());
        unsigned int key = (unsigned int)atoi(arg2.c_str());

        //печать группы пакетов...    
        if(!memcmp(arg1.c_str(),"printg",6)||!memcmp(arg1.c_str(),"pg",2))
        {
            if(key>0)
            {
                if(payload_sort.count(key))
                {
                    print_group(payload_sort,key);
                    continue;
                }
                else
                    {
                        cout << ">>>> illegal key!" << endl;
                        continue;
                    }
            }
        }

        //анализ группы пакетов...    
        if(!memcmp(arg1.c_str(),"analitic",8)||!memcmp(arg1.c_str(),"al",2))
        {
            analitic(payload_sort,(unsigned char)key);
            continue;
        }

        //печать списка сортировки...
        if(cmd=="prints"||cmd=="ps")
        {
            print_sort(payload_sort);
            continue;
        }

        //очистка консоли...
        if(cmd=="clear"||cmd=="cls")
        {
            system("cls");
            continue;
        }


        //пустая комманда...
        if(cmd=="")
        {
            continue;
        }

        //выход...
        if(cmd=="quit"||cmd=="q")
        {
            break;
        }

        cout << ">>>> illegal cmd!" << endl;

    }

    /*
    *   закрываем файл...
    */
    fin.close();
}

// pair<const unsigned int, vector<unsigned char>> //для пары из map

/* for(unsigned char c : temp)
{
    printf("%2x ",(unsigned char)c);
}
cout << endl; */