Beesnest/Main.cpp

Go to the documentation of this file.

/*
Beesnest - Presonal Application Server (beesnest-server.org)
Copyright 2007 Erez Bibi (erezbibi@users.sourceforge.net)
This file is part of Beesnest.

Beesnest is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

Beesnest is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Beesnest; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/


/* Main.cpp: Main file for the Beesnest Server. */

#include "DcGlobals.h"
#include "DcParameters.h"
#include "DcConnectionThread.h"
#include "DcLoadController.h"
#include "DcEnginesBank.h"
#include "DcLogger.h"
#include "Interfaces/DcAdapter.h"
#include _GET_INCLUDE_PATH (DcSocket.h)
#include _GET_INCLUDE_PATH (DcSocketServer.h)
#include _GET_INCLUDE_PATH (DcDemon.h)


/* Global Variables. */
/** True when the program runs as a regular application. */
bool gbRunAsApp;
/** The Logger object (just one for the entire application). */
CdcLogger oLogger;

/* Just make the logger available to all. */
void PrintString (CdcString str, int level)
{ oLogger.PrintString (str, level, gbRunAsApp); }

int ProgramMain ()
{
    if (gbRunAsApp == false)
        if (CdcDemon::StartPending () == false)
        {
            CdcDemon::Stop (-1);
            return -1;
        }

    CdcSocketServer* pserver = NULL;
    CdcSocket* psock = NULL;
    int ret_val = -1;
    try
    {   /* Initialize. */
        InitProgram();
        int port = CdcParameters::ToInt (
                CdcParameters::GetParam ("Server->Port"));
        pserver = CdcSocketServer::GetSocketServer (port);
        PrintString ("Server started", PS_GEN);
        if (gbRunAsApp == false)
            CdcDemon::MarkRun ();
        /* Listen and create thread to each request. */
        while (CdcDemon::IsRunning () || gbRunAsApp)
        {
            psock = pserver->Listen ();
            if (CdcLoadController::AddThread (psock) == true)
            {
                CdcConnectionThread* pconnection = new
                    CdcConnectionThread (psock);
            }
        }
    }
    catch (CdcException& e)
    {
        PrintString (e.ToStr (), PS_GEN);
        ret_val = -1;
        if (gbRunAsApp == false)
            CdcDemon::Stop (-1);
    }
    PrintString ("Server closed", PS_GEN);
    if(pserver != NULL)
        delete pserver;
    CdcSocketServer::Cleanup ();
    CdcCrypto::Destroy ();
    CdcEnginesBank::Term ();
    return ret_val;
}


void InitProgram()
{
    CdcHttpTime::Init ();
    CdcCrypto::Init ();
    bool is_ok = CdcParameters::ReadParams (PARM_FILE_EXT);
    int reqlen = CdcParameters::ToInt (CdcParameters::GetParam
        ("Server->Max-Req-Leng"));
    reqlen = reqlen < 1024 ? 1024 : reqlen;
    int reqtime = CdcParameters::ToInt (CdcParameters::GetParam
        ("Server->Rec-Timeout"));
    CdcSocketServer::Startup (reqlen, reqtime);
    CdcString file = CdcFileUtil::GetWorkPath (
        CdcParameters::GetParam ("Login-Manager->Users-File"));
    CdcLoginManager::Load (file.GetBuffer ());
    CdcAdaptersBank::Init ();
    CdcLoadController::Init ();
    CdcRetrieveFile::Init ();
    CdcEnginesBank::Init ();
    oLogger.Init ();
    if (is_ok == false)
        PrintString ("Warning - XML file is not Well-Formed!", PS_GEN);
    /* Set high priority for the listening (main) thread. */
    SetThreadPriority (GetCurrentThread (), THREAD_PRIORITY_HIGHEST);
}


void ControlUsers (int argc, char* argv[])
{
    int is_ok = false;
    if (argc < 5)
    {
        cout << "usage: beesnest logins (add <username> <password> | " << endl;
        cout << "\t\t\tchange <username> <oldpassword> <newpassword> | " << endl;
        cout << "\t\t\tremove <username> <password>)" << endl;
        return;
    }

    try
    {   /* Some code from InitProgram is copied here. */
        bool is_ok = CdcParameters::ReadParams (PARM_FILE_EXT);
        CdcString file = CdcFileUtil::GetWorkPath (
            CdcParameters::GetParam ("Login-Manager->Users-File"));
        CdcLoginManager::Load (file.GetBuffer ());
        if (is_ok == false)
            PrintString ("Warning - XML file is not Well-Formed!", PS_GEN);
    }
    catch (CdcException& e)
    { PrintString (e.GetExpMessage (), PS_GEN); return; }

    if (strcmp (argv[2], "add") == 0)
        is_ok = CdcLoginManager::AddUser (argv[3], argv[4]);
    else if (strcmp (argv[2], "change") == 0 && argc > 5)
        is_ok = CdcLoginManager::ChangePassword (argv[3], argv[4], argv[5]);
    else if (strcmp (argv[2], "remove") == 0)
        is_ok = CdcLoginManager::RemoveUser (argv[3], argv[4]);

    if (is_ok == false)
        PrintString ("Cannot execute the request!", PS_GEN);
    else
        PrintString ("Request executed - OK!", PS_GEN);

    CdcString file = CdcFileUtil::GetWorkPath (
        CdcParameters::GetParam ("Login-Manager->Users-File"));
    CdcLoginManager::Save (file.GetBuffer ());
}

void main (int argc, char* argv[])
{
    gbRunAsApp = true;
    if (argc > 1)
    {
        if (strcmp (argv[1], "install") == 0)
            CdcDemon::Install ();
        else if (strcmp (argv[1], "uninstall") == 0)
            CdcDemon::Uninstall ();
        else if (strcmp (argv[1], "run") == 0)
            ProgramMain ();
        else if (strcmp (argv[1], "logins") == 0)
            ControlUsers (argc, argv);
        else
            cout << "usage: beesnest [run | install | uninstall | logins...]" << endl;
    }
    else
    {
        gbRunAsApp = false;
        CdcDemon::RunMain ();
    }
}