vpClient.cpp

/****************************************************************************
 *
 * This file is part of the ViSP software.
 * Copyright (C) 2005 - 2017 by Inria. All rights reserved.
 *
 * This software is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * ("GPL") version 2 as published by the Free Software Foundation.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ViSP with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * See http://visp.inria.fr for more information.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at visp@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Description:
 * TCP Client
 *
 * Authors:
 * Aurelien Yol
 *
 *****************************************************************************/

#include <visp3/core/vpClient.h>

vpClient::vpClient() :  vpNetwork(), numberOfAttempts(0)
{
}

vpClient::~vpClient()
{
  stop();
}

bool vpClient::connectToHostname(const std::string &hostname, const unsigned int &port_serv)
{
  // get server host information from hostname
  struct hostent *server = gethostbyname( hostname.c_str() );
  
  if ( server == NULL )
  {
    std::string noSuchHostMessage( "ERROR, " );
    noSuchHostMessage.append( hostname );
    noSuchHostMessage.append( ": no such host\n" );
    vpERROR_TRACE( noSuchHostMessage.c_str(),
                             "vpClient::connectToHostname(const std::string &hostname, const int &port_serv)" );
    return false;
  }

  vpNetwork::vpReceptor serv;
  
  serv.socketFileDescriptorReceptor = socket( AF_INET, SOCK_STREAM, 0 );

#if !defined(_WIN32) && (defined(__unix__) || defined(__unix) || (defined(__APPLE__) && defined(__MACH__))) // UNIX
  if ( serv.socketFileDescriptorReceptor < 0){
#else
  if ( serv.socketFileDescriptorReceptor == INVALID_SOCKET){
#endif 
    vpERROR_TRACE( "ERROR opening socket",
                 "vpClient::connectToHostname()" );
    return false;
  }
  
  memset((char *) &serv.receptorAddress, '\0', sizeof(serv.receptorAddress));
  serv.receptorAddress.sin_family = AF_INET;
  memmove( (char *) &serv.receptorAddress.sin_addr.s_addr, (char *) server->h_addr, (unsigned)server->h_length );
  serv.receptorAddress.sin_port = htons( (unsigned short)port_serv );
  serv.receptorIP = inet_ntoa(*(in_addr *)server->h_addr);

  return connectServer(serv);
}

bool vpClient::connectToIP(const std::string &ip, const unsigned int &port_serv)
{
  vpNetwork::vpReceptor serv;
  
  serv.socketFileDescriptorReceptor = socket( AF_INET, SOCK_STREAM, 0 );

#if !defined(_WIN32) && (defined(__unix__) || defined(__unix) || (defined(__APPLE__) && defined(__MACH__))) // UNIX
  if ( serv.socketFileDescriptorReceptor < 0){
#else
  if ( serv.socketFileDescriptorReceptor == INVALID_SOCKET){
#endif
    vpERROR_TRACE( "ERROR opening socket",
                 "vpClient::connectToIP()" );
    return false;
  }
  
  memset((char *) &serv.receptorAddress, '\0', sizeof(serv.receptorAddress));
  serv.receptorAddress.sin_family = AF_INET;
  serv.receptorAddress.sin_addr.s_addr = inet_addr(ip.c_str());
  serv.receptorAddress.sin_port = htons( (unsigned short)port_serv );
  
  return connectServer(serv);
}

void vpClient::deconnect(const unsigned int &index)
{   
  if(index < receptor_list.size())
  {
#if !defined(_WIN32) && (defined(__unix__) || defined(__unix) || (defined(__APPLE__) && defined(__MACH__))) // UNIX
    shutdown( receptor_list[index].socketFileDescriptorReceptor, SHUT_RDWR );
#else // _WIN32
    shutdown( receptor_list[index].socketFileDescriptorReceptor, SD_BOTH );
#endif
    receptor_list.erase(receptor_list.begin()+(int)index);
  }  
}

void vpClient::stop()
{
  for(unsigned int i = 0 ; i < receptor_list.size() ; i++){
#if !defined(_WIN32) && (defined(__unix__) || defined(__unix) || (defined(__APPLE__) && defined(__MACH__))) // UNIX
    shutdown( receptor_list[i].socketFileDescriptorReceptor, SHUT_RDWR );
#else // _WIN32
    shutdown( receptor_list[i].socketFileDescriptorReceptor, SD_BOTH );
#endif
    receptor_list.erase(receptor_list.begin()+(int)i);
    i--;
  }
}

void vpClient::print()
{
  vpNetwork::print("Server");
}

//Private function
bool vpClient::connectServer(vpNetwork::vpReceptor &serv)
{
  serv.receptorAddressSize = sizeof( serv.receptorAddress );
  
  numberOfAttempts = 15;
  unsigned int ind = 1;
  int connectionResult=-1;
  
  while(ind <= numberOfAttempts){
    std::cout << "Attempt number " << ind << "..." << std::endl;
    
    connectionResult = connect( serv.socketFileDescriptorReceptor,
                                    (sockaddr*) &serv.receptorAddress,
                                    serv.receptorAddressSize );
    if(connectionResult >= 0)
      break;
    
    ind++;
    vpTime::wait(1000);
  }
  
  if( connectionResult< 0 )
  {
    vpERROR_TRACE( "ERROR connecting, the server may not be waiting for connection at this port.",
                 "vpClient::connectServer()");
    
    return false;
  }
  
  receptor_list.push_back(serv);

#ifdef SO_NOSIGPIPE
  // Mac OS X does not have the MSG_NOSIGNAL flag. It does have this
  // connections based version, however.
  if (serv.socketFileDescriptorReceptor > 0) {
    int set_option = 1;
    if (0 == setsockopt(serv.socketFileDescriptorReceptor, SOL_SOCKET, SO_NOSIGPIPE, &set_option, sizeof(set_option))) {
    } else {
      std::cout << "Failed to set socket signal option" << std::endl;
    }
  }
#endif // SO_NOSIGPIPE

  std::cout << "Connected!" << std::endl;
  return true;
}