tcpface.cpp

来自「一个语言识别引擎」· C++ 代码 · 共 104 行

// -*- mode:C++; tab-width:4; c-basic-offset:4; indent-tabs-mode:nil -*-

/*
 * Copyright (C) 2006 Paul Fitzpatrick
 * CopyPolicy: Released under the terms of the GNU GPL v2.0.
 *
 */


#include <yarp/TcpFace.h>
#include <yarp/Logger.h>
#include <yarp/SocketTwoWayStream.h>
#include <yarp/Protocol.h>

#include <ace/OS_NS_stdio.h>

using namespace yarp;

static Logger tcpFaceLog("TcpFace", Logger::get());


TcpFace::~TcpFace() {
    closeFace();
}


void TcpFace::open(const Address& address) {
    YARP_DEBUG(tcpFaceLog,String("TcpFace opening for address ") + address.toString());

    this->address = address;
    ACE_INET_Addr	serverAddr(address.getPort());
    int result = peerAcceptor.open(serverAddr,1);
    if (result==-1) {
        throw IOException("cannot listen on specified tcp address");
    }
}

void TcpFace::close() {
    closeFace();
}

void TcpFace::closeFace() {
    peerAcceptor.close();

    /*
      if (!closed) {
      closed = true;
      OutputProtocol *op = NULL;
      try {
      op = write(address);
      //ACE_OS::printf("write done, gave %ld\n", (long int)op);
      if (op!=NULL) {
      op->close();
      }
      } catch (IOException e) {
      // no problem
      ACE_OS::printf("exception during write\n");
      }
      if (op!=NULL) {
      delete op;
      op = NULL;
      }
      peerAcceptor.close();
      }
    */
}


/**
 * return NULL on failure.  No exceptions thrown.
 */
InputProtocol *TcpFace::read() {

    SocketTwoWayStream *stream  = new SocketTwoWayStream();
    YARP_ASSERT(stream!=NULL);

    try {
        stream->open(peerAcceptor);
    } catch (IOException e) {
        ACE_OS::printf("exception on tcp stream read: %s\n", e.toString().c_str());
        stream->close();
        delete stream;
        stream = NULL;
    }

    if (stream!=NULL) {
        return new Protocol(stream);
    }
    return NULL;
}


OutputProtocol *TcpFace::write(const Address& address) {
    SocketTwoWayStream *stream  = new SocketTwoWayStream();
    int result = stream->open(address);
    if (result==-1) {
        stream->close();
        delete stream;
        return NULL;
    }
    return new Protocol(stream);
}