saf7/main.cpp

/* 
 * File:   main.cpp
 * Author: SET - nmset@yandex.com
 * License : GPL v2
 * Copyright SET - © 2017
 *
 * Created on 6 mai 2017, 18:21
 */

#include "gcc_version.h"
#include <cstdlib>
#include <iostream>
#include "SAF.h"
#include "globals.h"
#include <string>
#include <fstream>
#include <signal.h>
#include <boost/program_options.hpp>
#include <gloox/client.h>
#include <gloox/disco.h>
#include "DBusSafWorkers.h"

using namespace std;
namespace po = boost::program_options;

string g_config = ""; // A configuration file.
string g_jid = "";
string g_password = "";
string g_resource = "SAF";
string g_server = "";
string g_tag = "default";
uint g_port = 5222;
uint g_retry = 0;
uint g_delay = 30;
uint g_wsping = 0;
bool g_verbose = false;

gloox::Client * g_client = NULL;
DBus::BusDispatcher g_dispatcher;
SAFConnListener * g_saf = NULL;
Connector * g_connector = NULL;
DBusSafAdaptor * g_adaptor_worker = NULL;

/**
 * TODO : Detect and react to vanishing network link.
 */

/*
 * http://www.radmangames.com/programming/how-to-use-boost-program_options
 * https://blog.knatten.org/2012/03/23/basic-gloox-tutorial/
 * https://camaya.net/api/gloox-1.0.20/
 */

/**
 * Cleaup and exit. Return code is 0 because we want to exit this way.
 * @param sn
 */
void signalhandler(int sn)
{
    delete g_connector;
    /* Cant delete adaptor_worker without a seg fault, this function is called from there.
     * Anyway, we are exiting.
     */
    //delete adaptor_worker;
    delete g_client;
    delete g_saf;
    if (g_verbose)
        cout << "Signaled to exit." << endl;
    exit(0);
}

bool GetAndValidateParams(int argc, char** argv)
{
    try
    {
        po::options_description desc(_ABOUT_);
        desc.add_options()
            ("help,h", "Print help messages")
            ("jid,j", po::value<string>(&g_jid), "Full JID without resource")
            ("password,w", po::value<string>(&g_password), "Password")
            ("resource,r", po::value<string>(&g_resource), "Resource (optional)")
            ("server,s", po::value<string>(&g_server), "Server (optional)")
            ("port,p", po::value<uint>(&g_port), "Port (optional), default = 5222")
            ("retry,y", po::value<uint>(&g_retry), "Number of times to try to connect to server (optional), default = 0 (for ever)")
            ("delay,d", po::value<uint>(&g_delay), "Number of seconds between each connection attempt (optional), default = 30")
            ("wsping,g", po::value<uint>(&g_wsping), "Number of seconds between each whitespace ping to the server (optional), default = 0 (never), at least 55")
            ("bustag,t", po::value<string>(&g_tag), "Bus name tag (optional), default=default")
            ("config,c", po::value<string>(&g_config), "Use specified configuration file (optional), else, pass all parameters on the command line")
            ("verbose,v", "Show messages (optional). Command line only.");

        po::variables_map vm;
        po::variables_map vmc;
        try
        {
            po::store(po::parse_command_line(argc, argv, desc), vm);
            if (vm.count("help"))
            {
                cout << desc << endl;
                return 0;
            }
            po::notify(vm);
            if (vm.count("verbose"))
            {
                g_verbose = true;
            }
            /*
             * If a config file is in the command line,
             * its content overrides all other parameters.
             */
            if (vm.count("config"))
            {
                ifstream f(g_config);
                if (f.fail())
                {
                    cerr << "Failure with file : " << g_config << endl;
                    return 0;
                }
                po::store(po::parse_config_file(f, desc), vmc);
                po::notify(vmc);
            }
        }
        catch (po::error& eb)
        {
            cerr << "Error : " << eb.what() << endl;
            return 0;
        }
    }
    catch (exception& e)
    {
        cerr << "Error : " << e.what() << endl;
        return 0;
    }

    if (g_jid.empty())
    {
        cout << "jid missing." << endl;
        return 0;
    }
    if (g_password.empty())
    {
        cout << "password missing." << endl;
        return 0;
    }
    if (g_wsping > 0 && g_wsping < 55 )
    {
        cout << "wsping must be at least 55 seconds." << endl;
        return 0;
    }
    return true;
}

DBusSafAdaptor* StartDBusAdaptor(Connector * connector, SAFConnListener * saf)
{
    DBus::_init_threading();
    DBus::default_dispatcher = &g_dispatcher;
#ifdef DEVTIME // Defined in Debug NetBeans configuration
    DBus::Connection bus = DBus::Connection::SessionBus();
#else
    DBus::Connection bus = DBus::Connection::SystemBus();
#endif
    const string busName = string("xmpp.saf-") + g_tag;
    bus.request_name(busName.data());
    DBusSafAdaptor * adaptor_worker = new DBusSafAdaptor(bus, connector, saf);

    return adaptor_worker;
}

int main(int argc, char** argv)
{

    if (!GetAndValidateParams(argc, argv))
        return 0;

    signal(SIGINT, signalhandler);
    signal(SIGTERM, signalhandler);

    JID ajid(g_jid + "/" + g_resource);
    g_client = new Client(ajid, g_password, g_port);
    if (!g_server.empty())
        g_client->setServer(g_server);
    g_client->setPort(g_port);
    g_client->disco()->setIdentity("client", "7", "saf");

    g_saf = new SAFConnListener(g_client);
    g_connector = new Connector(g_saf, g_retry, g_delay);
    // Listen on DBus
    g_adaptor_worker = StartDBusAdaptor(g_connector, g_saf);
    // Connect to server
    g_connector->DoConnect(); // Blocks

    //cout << g_client->getStatistics().encryption << endl;

    delete g_adaptor_worker;
    delete g_connector;
    delete g_client;
    delete g_saf;

    return 0;
}