Cli.cc

Go to the documentation of this file.

#include <fstream>
#include <stdlib.h>
#include <boost/bind.hpp>
#include <boost/thread/thread.hpp>
#include <readline/readline.h>
#include <readline/history.h>
#include "include/Cli.h"
#include "include/generic.h"

using namespace std;
using namespace boost;

                
Cli::Cli(Configuration cfg, NetSniff *netsnf)
{
        // Parse configuration and init vars
        cout << endl << "VoIPHoney CLI interface ...." << endl << endl;
        prompt = "VoIPHoney@" + cfg.GetHostname() + ":CLI> ";
        hcmdpath = cfg.GetValue("helpcmdpath");
        // Get the sniffer control
        this->netsnf = netsnf;
}
bool Cli::SetDebugger(Debug *debugger)
{
        this->dbg = debugger;
        if (this->dbg != NULL)
                return true;
        else
                return false;
}

void Cli::MainLoop ()
{
        
        char *buf;
        string cmd;
        string validcommands = "debug test module help verbosity exit sniffer ";
        
        if (chdir (hcmdpath.c_str()) == -1)
        {
                cerr << "Error loading commands help on " << hcmdpath.c_str() << endl;
        }
        
        while((buf = readline(prompt.c_str()))!=NULL) 
        {
                cmd = buf;
                if (cmd.length() > 0)
                {
                
                        unsigned int len = cmd.length();
                        unsigned int last = cmd.find_last_of(" ");
                        
                        // Delete last char if it is a whitespace
                        if (last == len - 1)
                                cmd.erase(last);
                        
                        string keyword, args;
                        SplitStr(cmd, &keyword, &args);
                
                        if (validcommands.find(keyword + " ") != string::npos)
                                ProcessCommand(keyword, args);
                        else
                        {
                                // Switch CLI display colour to predefined settings 
                                cout << " Unknown command: " << FORE_RED << cmd.c_str() << ENDCOLOR << endl << endl;
                                cout << "Type help to display a command list" << endl;
                                cout << "Or type help [" << FORE_YELLOW << "command" << ENDCOLOR << "] for help on a specific command" << endl;
                                // Switch back display to normal settings
                                cout << endl;
                        }
                        
                        if (strcmp(buf,"exit")==0 || strcmp(buf,"exit ")==0)
                                break;
                        if (buf[0]!=0)
                                add_history(buf);
                
                }
                        
                free(buf);
        }

}

void Cli::ProcessCommand(string cmd, string args)
{
        if (cmd.compare("help") == 0)
                DisplayHelp(args);
        else if (cmd.compare("debug") == 0)
                DebugCmd(args);
        else if (cmd.compare("verbosity") == 0)
                VerbosityCmd(args);
        else if (cmd.compare("sniffer") == 0)
                SnifferCmd(args);               
}

void Cli::DebugCmd(string args)
{
        if (args.compare("show") == 0)
        {
                cout << "Debugging is set to: [";
                if (dbg->isDebugging())
                        cout << "yes";
                else
                        cout << "no";
                cout << "]" << endl;
        }
        else
        {
                string keyword, carg;
                SplitStr(args, &keyword, &carg);
                if (keyword.compare("set") == 0)
                {
                        if (carg.compare("yes") == 0)
                                dbg->StartDebug();
                        else if (carg.compare("no") == 0)
                                dbg->StopDebug();
                        else
                                cout << "Unknow value " << carg << " for setting debug" << endl;
                        
                }
                else
                        cout << "Sorry: debug " << args.c_str() << " not possible" << endl;
        }
}

void Cli::VerbosityCmd(string args)
{
        if (args.compare("show") == 0)
        {
                cout << "Verbosity is set to: [" << dbg->GetVerbosity().c_str();
                cout << "]" << endl;
        }
        else
        {
                string keyword, carg;
                SplitStr(args, &keyword, &carg);
                if (keyword.compare("set") == 0)
                {
                        int verb = -1;
                        istringstream buffer(carg);
                        buffer >> verb;
                        if (verb >=0 && verb <= 100)
                        {
                                dbg->SetVerbosity(carg);
                                cout << "Verbosity is now " << carg.c_str() << endl;
                        }
                        else
                                cout << "Verbosity value must be between 0 and 100" << endl;
                        
                }
                else
                        cout << "Sorry: verbosity " << args.c_str() << " not possible" << endl;
        }
}

void Cli::SnifferCmd(string args)
{
        if (args.compare("show") == 0)
        {
                if (netsnf->isReady())
                {
                        cout << "Sniffer " << FORE_GREEN << "READY " << ENDCOLOR;
                        if (netsnf->isSniffing())
                                cout << "AND " << FORE_GREEN << "SNIFFING" << ENDCOLOR;
                        else
                                cout << "AND " << FORE_RED "NOT SNIFFING" << ENDCOLOR;
                        cout << endl << netsnf->GetFullStatus();
                }
                else
                        cout << "Sniffer " << FORE_RED << "NOT READY" << ENDCOLOR;

                cout << endl;
        }
        else if (args.compare("start") == 0)
        {
                // Start sniffing in a new thread
                thread* sniffThread = new thread(bind(&network::NetSniff::startsniff,netsnf));
                if (sniffThread == NULL)
                        dbg->Prnterr(CODEAT, "Error creating thread");  
        }
        else if (args.compare("stop") == 0)
        {
                netsnf->stopsniff();    
        }
        else
                cout << "Sorry: sniffer " << args.c_str() << " not possible" << endl;
}

void Cli::SplitStr(string str, string *keyword, string *args)
{
        // Split command and args
        unsigned int first_white = str.find_first_of(" "); 
        *keyword = str.substr(0,first_white);
        if (first_white > str.length())
                *args = "";
        else
                *args = str.substr(first_white + 1, str.length()-1);
}

void Cli::DisplayHelp(string args)
{
        int content_len;
        char * content;
        cout << endl << "-----------------------------------------------------------------------" << endl;
        if (args.length() > 1)
                cout << "Help on topic " << args.c_str();
        else
        {
                cout << "Main Help ";
                args = "help";
        }
        cout << endl << "-----------------------------------------------------------------------" << endl;
        // Load file contents in the buffer

        ifstream is;

        is.open (args.c_str(), ios::binary );
        // Check if opening has failed
        if (is.fail())
        {
                cout << "Error loading help for " << args.c_str() << endl;
                return;
        }

        // get length of file
        is.seekg (0, ios::end);
        content_len = is.tellg();
        is.seekg (0, ios::beg);
        
        // allocate memory
        try {
                content = new char [content_len];
                // read data as a block
                is.read (content, content_len);
                is.close();
                
                cout.write(content,content_len);
                cout << endl << "-----------------------------------------------------------------------" << endl;
                
                // free memory
                delete[] content;
        }
        catch (std::bad_alloc&)
        {
                dbg->Prnterr(CODEAT, "Error allocating memory");
        }

}

Cli::~Cli()
{
        dbg->Prntdbg(CODEAT, 20, "Finish CLI ...");
}

---