src/io/RtServerSocket.cpp

/*=========================================================================
 *  RtServerSocket.cpp defines a class that implements server io operations on a
 *  socket
 *
 *  Copyright 2007-2013, the MURFI dev team.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/

#include"RtServerSocket.h"

// constructor with port and host
RtServerSocket::RtServerSocket(unsigned short portNum) {
  messageTerminationChar = '\n';
  address.set(portNum);
}

// destructor
RtServerSocket::~RtServerSocket() {
  if (acceptor.isOpen) {
    close();
  }
}

// open and start accepting
bool RtServerSocket::open(RtConfig &config) {
  if (acceptor.open(address, 1) == -1) {
    cerr << "couldn't open acceptor to listen on " << address.get_host_name()
         << ":" << address.get_port_number() << endl;
    acceptor.isOpen = false;
    return false;
  }
  cerr << "opened acceptor to listen on " << address.get_host_name()
       << ":" << address.get_port_number() << endl;

  acceptor.isOpen = true;
  return true;
}

// close and clean up
bool RtServerSocket::close() {
  acceptor.close();
  return true;
}

// receive a message
// in   string received
// out  string response
string RtServerSocket::receiveMessage(string &message,
                                      ACE_SOCK_Stream &stream) {
  cout << "in wrong version" << endl;
  return message;
}

// send a message to a client
// in
//  message to send
//  stream to send on
// out
//  success or failure
bool RtServerSocket::sendMessage(const string &message, ACE_SOCK_Stream &stream) {
  unsigned int sent = stream.send_n(message.c_str(), message.length());
  if (sent < message.length()) {
    cerr << "incomplete send" << endl;
    return false;
  }

  stream.send_n(&messageTerminationChar, 1);

  cout << "sent " << message << endl;

  return true;
}

// listen for connections
int RtServerSocket::svc() {
  // continuously try to accept connections
  while (acceptor.isOpen && acceptor.accept(stream) != -1) {
    stringstream message;

    // get characters until null character
    char last = 'a';
    while (last != messageTerminationChar) {
      if (0 == stream.recv_n(&last, 1)) {
        cout << "received empty message" << endl;
        break;
      }

      message << last;
    }

    // receive the message, store the response
    string received = message.str();
    string response = receiveMessage(received, stream);

    // send the response
    if(!response.empty()) {
      sendMessage(response, stream);
    }

    // close the stream (scanner connects anew for each image)
    stream.close();
  }
  acceptor.isOpen = false;

  return 0;
}