main.cpp File Reference

#include <sys/wait.h>
#include <getopt.h>
#include <iostream>
#include <execinfo.h>
#include "RioShell.h"
#include "RioError.h"

Go to the source code of this file.

Functions
void	Usage (char *progname)
	Prints the usage.
void	handle_SIGINT (int)
	Handles Signal Int.
void	show_stackframe ()
	Handles the SEGFAULT.
void	segfaultHandler (int sig)
int	main (int argc, char *argv[])
Variables
static RioShell *	rioshell

---

Function Documentation

void handle_SIGINT

(

int

)

Handles Signal Int.

Definition at line 345 of file clients/riosh/src/main.cpp.

{
    rioshell->handle_SIGINT( );
}

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 106 of file clients/riosh/src/main.cpp.

{
    char *UserName    = NULL;
    char *MachineName = NULL;
    char *Password    = NULL;
    bool commandMode  = false;

    #ifdef USE_QT
    bool textMode     = false;
    #endif

    char *cmdLine1    = NULL;
    bool stdOut       = false;
    bool pipe_mode    = false; //if in pipe mode, riosh must abort when any
                               //command fail, returning error
    char optch;
    int ind = -1;
    int status;

    #ifdef USE_QT
    static char opts_short[] = "c:hp:os:tu:v";
    #else
    static char opts_short[] = "c:hp:os:u:v";
    #endif

    static struct option opts_int[] =
    {
       {"command"        ,1,0,'c'},
       {"help"           ,0,0,'h'},
       {"clean-output"   ,0,0,'o'},
       {"server"         ,1,0,'s'},

       #ifdef USE_QT
       {"text-mode"      ,0,0,'t'},
       #endif

       {"user"           ,1,0,'u'},
       {"password"       ,1,0,'p'},
       {"version"        ,0,0,'v'},
       {0,0,0,0}
    };

    signal( SIGSEGV, segfaultHandler );

    while( ( optch = getopt_long( argc, argv, opts_short, opts_int, &ind ) ) != -1 )
    {
        switch( optch )
        {
            case 'o':
                stdOut = true;
                break;

            case 's':
                if( MachineName == NULL )
                    MachineName = strdup( optarg );
                break;

            #ifdef USE_QT
            case 't':
                textMode = true;
                break;
            #endif

            case 'u':
                if( UserName == NULL )
                    UserName = strdup( optarg );
                break;

            case 'c':
                commandMode = true;
                cmdLine1 = strdup( optarg );
                break;

            case 'p':
                Password = strdup( optarg );
                if( strlen( Password ) > MAXPASSWORDSIZE )
                {
                    cout << argv[ 0 ] << ": password too long!" << endl;
                    exit( -1 );
                }
                break;

            case 'v':
                cout << "riosh version: " << VERSION << endl;
                exit( 0 );

            case 'h':
                Usage( argv[ 0 ] );
                return( 0 );

            default:
                Usage( argv[ 0 ] );
                return( -1 );
        }
    }

    // pipe mode identified by a '-' as last argument
    // it makes getopt_long abort as if there were no more arguments, so we
    // must check it manually after the getopt loop is done
    if( strcmp( argv[argc - 1], "-" ) == 0 )
    {
        pipe_mode = true;
    } 
    
    if( !pipe_mode && ( argc - optind > 0 ) ) 
    {
        int i, length;
        commandMode = true;
        length      = 0;

        for( i = optind; i < argc; i++ )
            length = strlen( argv[ i ] );

        cmdLine1 = (char *)malloc( ( length + ( argc - optind ) + 2 )
                                   * sizeof( char ) );
        for( i = optind; i < argc; i++ )
        {
            strcpy( cmdLine1 + strlen( cmdLine1 ), argv[ i ] );
            strcat( cmdLine1, " " );
        }

        cmdLine1[ strlen( cmdLine1 ) - 1 ] = 0;
    }
    if( pipe_mode && ( argc - optind > 1 ) )
    {
        int i, length;
        commandMode = true;
        length      = 0;

        for( i = optind; i < argc - 1; i++ )
            length = strlen( argv[ i ] );

        cmdLine1 = (char *)malloc( ( length + ( argc - optind ) + 1 )
                                   * sizeof( char ) );
        for( i = optind; i < argc - 1; i++ )
        {
            strcpy( cmdLine1 + strlen( cmdLine1 ), argv[ i ] );
            strcat( cmdLine1, " " );
        }

        cmdLine1[ strlen( cmdLine1 ) - 1 ] = 0;
    }

    #ifdef USE_QT
    // Options -o and -c and pipe mode must be used with -t
    if( ( stdOut || commandMode || pipe_mode || ( Password != NULL ) ) &&
          ! textMode )
        Usage( argv[ 0 ] );

    // Graphic Interface Implementation
    if( !textMode )
    {
        QApplication app( argc, argv );

        // Handles Child's Termination
        signal( SIGCHLD, handle_SIGCHLD );
        rioqt = new RioQt();
        app.setMainWidget( rioqt );

        return app.exec();
    }
    else
    {
        // Remove Password from command line
        for( int i = 1; i < argc; i++ )
        {
            if( strcmp( argv[i], "--password" ) == 0 ||
                strcmp( argv[i], "-p" ) == 0 )
            {
                int len = strlen( argv[i + 1] );
                for( int j = 0; j < len; j++ )
                    argv[i + 1][j] = '*';
            }
        }

        rioshell = new RioShell( MachineName, UserName, Password );
        if( rioshell->connect() )
        {
            signal( SIGINT, handle_SIGINT );
            status = rioshell->exec( commandMode, stdOut, cmdLine1, pipe_mode );
            delete rioshell;
            return status;
        }
        else
        {
            return RIOSHELL_ERROR;
        }
    }
    #else
    // Remove Password from command line
    for( int i = 1; i < argc; i++ )
    {
        if( strcmp( argv[i], "--password" ) == 0 ||
            strcmp( argv[i], "-p" ) == 0 )
        {
            int len = strlen( argv[i + 1] );
            for( int j = 0; j < len; j++ )
                argv[i + 1][j] = '*';
        }
    }

    rioshell = new RioShell( MachineName, UserName, Password );
    if( rioshell->connect() )
    {
        signal( SIGINT, handle_SIGINT );
        status = rioshell->exec( commandMode, stdOut, cmdLine1, pipe_mode );
        delete rioshell;
        return status;
    }
    else
    {
        return RIOSHELL_ERROR;
    }
    #endif

    return 0;
}

void segfaultHandler

(

int

sig

)

Definition at line 98 of file clients/riosh/src/main.cpp.

{
    RioErr << "[Riosh-main] SEGMENTATION FAULT!" << endl;
    show_stackframe();

    _exit( 1 );
}

void show_stackframe

(

)

Handles the SEGFAULT.

Definition at line 84 of file clients/riosh/src/main.cpp.

{
    void *trace[16];
    char **messages = (char **)NULL;
    int i, trace_size = 0;

    trace_size = backtrace( trace, 16 );
    messages = backtrace_symbols( trace, trace_size );
    RioErr << "[bt] Execution path:" << endl;
    for( i = 0; i < trace_size; ++i )
        RioErr << "[bt(" << i << "/" << trace_size << ")] " << messages[i] 
               << endl;
}

void Usage

(

char *

progname

)

Prints the usage.

Definition at line 53 of file clients/riosh/src/main.cpp.

{
    #ifdef USE_QT
    cout << "\nUsage: " << progname
         << " -s <server_name> -u <user_name> [-p <password>] [-t | -t -o] -c \"command [-]\"\n";
    #else     
    cout << "\nUsage: " << progname
         << " -s <server_name> -u <user_name> [-p <password>] [-o] -c \"command [-]\"\n";
    #endif     

    cout << "Options:" << endl;
    cout << "-s|--server <server_name>: Log in to server 'server_name'." << endl;
    cout << "-u|--user <user_name>:     Log in using the user name 'user_name'." << endl;

    #ifdef USE_QT
    cout << "-t|--text-mode:            Log in using the user name user_name." << endl;
    cout << "-o|--clean-output:         Just output the commands results. Only valid with -t" << endl;
    cout << "-p|--password:             User Password. Only valid with -t" << endl;
    cout << "-c|--command \"command\":    Execute command 'command' on the server and exit. Only valid with -t" << endl;
    cout << "-:                         Pass commands to riosh through a pipe. Only valid with -t" << endl;
    #else
    cout << "-o|--clean-output:         Just output the commands results." << endl;
    cout << "-p|--password:             User Password." << endl;
    cout << "-c|--command \"command\":    Execute command 'command' on the server and exit." << endl;
    cout << "-:                         Pass commands to riosh through a pipe." << endl;
    #endif
}

---

Variable Documentation

RioShell* rioshell [static]

Definition at line 39 of file clients/riosh/src/main.cpp.