G4FRClientServer.cc

Go to the documentation of this file.

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
//
// $Id: G4FRClientServer.cc 79795 2014-03-14 10:13:31Z gcosmo $
//
// Satoshi TANAKA, Wed Jul  3 14:14:29 JST 1996


//=================//
#ifdef G4VIS_BUILD_DAWN_DRIVER
//=================//

#include "G4VisManager.hh"
#include "G4FRClientServer.hh"

// #define DEBUG_CLIENT_SERVER

#include<sys/param.h>

    //----- const   
const  char DEFAULT_SUN_PATH[]      = "FR_TMP3"        ;
const  int  DEFAULT_PORT_NUMBER     = 40701            ;
//const  char   FR_ENV_SERVER_HOST_NAME[]   = "G4DAWN_HOST_NAME" ; // moved to .hh
const  int  MAX_CONNECT_TRIAL       = 10               ;
const  char     FR_DEFAULT_HOST_NAME[]          = "localhost"      ;      

    //----- G4FRClientServer::G4FRClientServer ()
G4FRClientServer::G4FRClientServer ( char terminator , char end_line ) :  
    TERMINATOR ( terminator ) , 
    END_OF_LINE( end_line   ) ,
    fSocketFd  ( -1 )           
{ 
    SetSunPath    ( DEFAULT_SUN_PATH ) ; // for Unix domain
    SetPortNumber ( DEFAULT_PORT_NUMBER ) ;
    ClearReceivedMessage () ;
}


    //----- G4FRClientServer::ConnectUnix()
int G4FRClientServer::ConnectUnix()
{
        //----- local
    int         flag_connected = 0 ; 
    struct sockaddr_un  server_address ;

        //----- make socket
    fSocketFd = socket( AF_UNIX, SOCK_STREAM, 0 );
    if( fSocketFd < 0 ) { Err("G4FRClientServer::ConnectUnix(),socket"); }

        //----- set server address
    memset( (char *)&server_address, '\0', sizeof(server_address)) ;
    server_address.sun_family = AF_UNIX ;
    strcpy( server_address.sun_path, SUN_PATH );

        //----- connection
    int connection_status = -1  ;
    int num_trial         = 0 ;
    while( connection_status < 0 && num_trial <= MAX_CONNECT_TRIAL ) {
        num_trial++ ; 
        connection_status = connect( fSocketFd, (struct sockaddr * )(&server_address), sizeof( server_address ) ) ;
        if( connection_status  <0 ) 
        {
#if defined DEBUG_CLIENT_SERVER
            Err("G4FRClientServer::ConnectUnix(),connect => RETRY");
#endif
            flag_connected = 0 ;
        } else {
            flag_connected = 1 ;
            break ;
        }

        sleep(1);
        
    } // while(connection_status...)

        //----- return status of connection
    return flag_connected ;

} // G4FRClientServer::ConnectUnix()


    //----- G4FRClientServer::Receive()
void    G4FRClientServer::Receive()
{
        //-----
    ClearReceivedMessage () ;
    if( recv( fSocketFd, fReceivedMessage, G4FRClientServer::RECV_BUFMAX , 0 ) < 0 ) 
    {
        Err("G4FRClientServer::Receive(), recv");
    }

#if defined DEBUG_CLIENT_SERVER
    if (G4VisManager::GetVerbosity() >= G4VisManager::errors)
      G4cout << ">>>>> receivedMessage = " << fReceivedMessage << G4endl;
#endif

}


    //----- G4FRClientServer::ReceiveLine()
void    G4FRClientServer::ReceiveLine()
{
        //----- local
    char    buf[1];
    int index = 0 ;

        //----- receive a line (until newline)
    memset(fReceivedMessage, '\0', RECV_BUFMAX) ;   
    while( read( fSocketFd, buf, 1 ) == 1 ) {
        fReceivedMessage[index++] = buf[0];
        if( IsEndOfLine(buf[0]) ) { break ;}
    }
} // G4FRClientServer::ReceiveLine()


    //----- G4FRClientServer::Send()
void    G4FRClientServer::Send()
{
    if( send( fSocketFd, fSendingMessage, strlen(fSendingMessage) , 0 ) < 0 ) 
    {
        Err("G4FRClientServer::Send(), send");
    }

#if defined DEBUG_CLIENT_SERVER
    if (G4VisManager::GetVerbosity() >= G4VisManager::errors)
      G4cout << "<<<<< SentMessage = " << fSendingMessage << G4endl;
#endif

} // G4FRClientServer::Send()


    //----- G4FRClientServer::Send( message )
void    G4FRClientServer::Send( const char* message ) 
{
    this->SetSendingMessage( message )      ;
    this->Send();

} // G4FRClientServer::Send( message )


    //----- G4FRClientServer::SendLine()
void    G4FRClientServer::SendLine( const char* message ) 
{
        //----- local
    int smsg_length ;

        //----- set message to sending buf
    this->SetSendingMessage( message )      ;
    smsg_length = GetSendingMessageLength() ;

        //----- add newline if necessary
    if( !IsEndOfLine( fSendingMessage[ (smsg_length - 1)] ) ) {
        fSendingMessage[ smsg_length ]      = GetEndOfLine() ;
        fSendingMessage[ (smsg_length +1) ] = '\0' ;
        smsg_length = GetSendingMessageLength();
    }

        //----- send
    this->Send();

}// G4FRClientServer::SendLine()


    //----- G4FRClientServer::DisConnect()
void G4FRClientServer::DisConnect()
{
        //----- close connection
    if( shutdown(fSocketFd,2) < 0 ) { 
        Err("G4FRClientServer::DisConnect,shutdown");
    }
    close( fSocketFd );

    this->Clear();
}



    //----- G4FRClientServer::Clear()
void G4FRClientServer::Clear()
{
    unlink(SUN_PATH) ;
    fSocketFd = -1   ;
}


    //----- G4FRClientServer::ConnectINET()
int G4FRClientServer::ConnectINET()
{
        //----- local
    int         flag_connected = 0 ;
    sockaddr_in     server_address ;
    char            server_hostname[32] ;
    hostent*        server_host_p ;

        //----- make socket
    fSocketFd = socket( AF_INET, SOCK_STREAM, 0 );
    if( fSocketFd < 0 ) { 
#if defined DEBUG_CLIENT_SERVER
      Err("G4FRClientServer::ConnectINET(),socket"); 
#endif
    }

        //----- get IP address of server from its name
    if( getenv( FR_ENV_SERVER_HOST_NAME ) != NULL ) 
    {
            //----- get server name
        strcpy( server_hostname, getenv( FR_ENV_SERVER_HOST_NAME ) );

            //----- get IP address of server from its name,
            //..... reading /etc/hosts
        server_host_p = gethostbyname( server_hostname ) ;
            
            //----- If the host specified by FR_ENV_SERVER_HOST_NAME
            //.....  is not written in /etc/hosts, 
            //...... server host is set to the same as the
            //...... client host
        if( !server_host_p ) {
#if defined DEBUG_CLIENT_SERVER
            Err("G4FRClientServer::ConnectINET(), gethostbyname");
#endif
            server_host_p = gethostbyname( FR_DEFAULT_HOST_NAME ) ;
        }

    } else {
            server_host_p = gethostbyname( FR_DEFAULT_HOST_NAME ) ;
    }



// #if defined DEBUG_CLIENT_SERVER
    if (G4VisManager::GetVerbosity() >= G4VisManager::errors)
      G4cout << "***** Trying connection to  " << server_hostname << G4endl;
// #endif 
    

        //----- connection and binding 
    memset( (char *)&server_address, '\0', sizeof(server_address)) ; 
                // clear server_address
    server_address.sin_family = AF_INET ;
    server_address.sin_port   = htons( PORT_NUMBER );
    memcpy( (char *)(&server_address.sin_addr ), 
        (char *)( server_host_p->h_addr   ), 
        server_host_p->h_length             ); 

    int connection_status = -1  ;
    int num_trial         = 0 ;
    while( connection_status < 0 && num_trial <= MAX_CONNECT_TRIAL ) {
        num_trial++ ; 
        connection_status = connect( fSocketFd, (struct sockaddr * )(&server_address), sizeof( server_address ) ) ;
        if( connection_status  <0 ) 
        {
#if defined DEBUG_CLIENT_SERVER
            Err("G4FRClientServer::ConnectINET(),connect => RETRY");
#endif 
            flag_connected  = 0 ;
        } else {
            flag_connected  = 1 ;
            break ;
        }

        sleep(1);

    } // while(connection_status...)

        //----- return status of connection
    return flag_connected ;

} // G4FRClientServer::ConnectINET()


    //----- G4FRClientServer::WaitSendBack()
void    G4FRClientServer::WaitSendBack( const char* command_string ) 
{
        //----- wait for sending back
    while(1) { 
        this->ReceiveLine();

        if( !strncmp(   this->GetReceivedMessage(), \
                command_string                , \
                (strlen(command_string))       )   )
        {
            break;
        } else {
            sleep(2);
        }

    } // while

        //----- clear buffer to receive message
    this->ClearReceivedMessage();   

} // G4FRClientServer::WaitSendBack()

#endif // G4VIS_BUILD_DAWN_DRIVER