owlps/owlps-positioning/src/outputtcpsocketevaal.cc

#include "outputtcpsocketevaal.hh"
#include "request.hh"
#include "resultlist.hh"
#include "area.hh"
#include "stock.hh"
#include "posexcept.hh"

#include <iostream>
#include <sstream>
#include <cstdio> // For perror()
#include <boost/lexical_cast.hpp>

using namespace std ;



/* *** Constructors *** */


OutputTCPSocketEvAAL::OutputTCPSocketEvAAL(
  const string &_remote_host,
  const uint_fast16_t _remote_port):
  remote_host(_remote_host), remote_port(_remote_port)
{
  if (! init_socket())
    throw error_opening_output_file("TCP socket") ;
}


OutputTCPSocketEvAAL::~OutputTCPSocketEvAAL()
{
  close_socket() ;
}



/* *** Operations *** */


/**
 * @return \em true if the socket were successfully opened.
 * @return \em false in case of error.
 */
bool OutputTCPSocketEvAAL::init_socket()
{
  sockfd = socket(AF_INET, SOCK_STREAM, 0) ;
  if (sockfd < 0)
    {
      perror("TCP socket creation failed") ;
      return false ;
    }

  struct sockaddr_in server_description ;
  memset(&server_description, 0, sizeof(server_description)) ;
  server_description.sin_family = AF_INET ;
  // Server IP address:
  server_description.sin_addr.s_addr = inet_addr(remote_host.c_str()) ;
  // Listening port on the server:
  server_description.sin_port = htons(remote_port) ;

  int ret = connect(sockfd, (struct sockaddr *)&server_description,
                    sizeof(server_description)) ;
  if (ret < 0)
    {
      perror("Cannot bind the TCP socket") ;
      return false ;
    }

  return true ;
}


/**
 * Normally, the socket is closed automatically by the destructor. Use
 * this if you want to close the socket prematurely.
 * #sockfd is set to -1, even in case of error.
 * @return \em true if the socket were successfully closed or were not
 * opened.
 * @return \em false in case of error.
 */
bool OutputTCPSocketEvAAL::close_socket()
{
  if (sockfd >= 0)
    {
      if (close(sockfd))
        {
          perror("Cannot close TCP socket") ;
          return false ;
        }

      sockfd = -1 ;
    }

  return true ;
}


/**
 * @param results Must contain only one element, since the EvAAL format
 * accepts only one algorithm result.
 */
void OutputTCPSocketEvAAL::write(const ResultList &results)
{
  assert(results.get_results().size() == 1) ;
  write(results.get_results()[0]) ;
}


void OutputTCPSocketEvAAL::write(const Result &result)
{
  const Point3D &position = result.get_position() ;
  Timestamp request_time = result.get_request()->get_time_sent() ;
  int area_of_interest = area_of_interest_number(position) ;

  ostringstream str ;
  str
      << "OwlPS "
      << position.get_x() << ' '
      << position.get_y() << ' '
      << static_cast<uint64_t>(request_time) << ' '
      << area_of_interest << '\n' ;

  if (send_data(str.str()))
    acknowledge() ;
}


int OutputTCPSocketEvAAL::
area_of_interest_number(const Point3D &position) const
{
  const Area *area = Stock::in_which_area_is(position) ;
  if (area == NULL)
    return 0 ;

  int aoi_number = 0 ;
  try
    {
      aoi_number = boost::lexical_cast<int>(area->get_name()) ;
    }
  catch (boost::bad_lexical_cast &e)
    {
      cerr
          << "Cannot convert the area of interest's name (« "
          << area->get_name() << " ») into an integer value: "
          << e.what() << '\n' ;
    }

  return aoi_number ;
}


/**
 * Sends the text buffer 'data'.
 */
bool OutputTCPSocketEvAAL::send_data(const string &data) const
{
  ssize_t nsent = send(sockfd, data.c_str(), data.size(), 0) ;
  if (nsent != static_cast<ssize_t>(data.size()))
    {
      perror("Error sending result data") ;
      return false ;
    }
  return true ;
}


bool OutputTCPSocketEvAAL::acknowledge() const
{
  char ack_buf[10] ;
  ssize_t nrecv = recv(sockfd, ack_buf, 10, 0) ;
  if (nrecv < 0)
    perror("Error receiving acknowledgement") ;

  ack_buf[9] = '\0' ;
  string ack(ack_buf) ;
  if (ack == "Published")
    return true ;

  if (ack == "NotParsed")
    {
      cerr << "TCP server error: string not parsed.\n" ;
      return false ;
    }

  cerr
      << "TCP server error: unknown acknoledgment string « "
      << ack << " ».\n" ;
  return false ;
}