cachelistener.cxx

来自「vovida的软交换」· CXX 代码 · 共 193 行

/* ====================================================================
 * The Vovida Software License, Version 1.0 
 * 
 * Copyright (c) 2000 Vovida Networks, Inc.  All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 * 
 * 3. The names "VOCAL", "Vovida Open Communication Application Library",
 *    and "Vovida Open Communication Application Library (VOCAL)" must
 *    not be used to endorse or promote products derived from this
 *    software without prior written permission. For written
 *    permission, please contact [e-mail].
 *
 * 4. Products derived from this software may not be called "VOCAL", nor
 *    may "VOCAL" appear in their name, without prior written
 *    permission of Vovida Networks, Inc.
 * 
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE AND
 * NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL VOVIDA
 * NETWORKS, INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT DAMAGES
 * IN EXCESS OF $1,000, NOR FOR ANY INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
 * DAMAGE.
 * 
 * ====================================================================
 * 
 * This software consists of voluntary contributions made by Vovida
 * Networks, Inc. and many individuals on behalf of Vovida Networks,
 * Inc.  For more information on Vovida Networks, Inc., please see
 * <http://www.vovida.org/>.
 *
 */
 
#include "CacheListener.hxx"

#include <sys/time.h>
#include <sys/types.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "NetworkAddress.h"
#include "Sptr.hxx"
#include "Tcp_ServerSocket.hxx"
#include "Tcp_ClientSocket.hxx"
#include "Connection.hxx"
#include "VNetworkException.hxx"
#include "cpLog.h"

//const int CacheListener::FirstTryPort = 6006;

void
CacheListener::thread()
{        
  if (serverSocket != NULL) { 
    cpLog(LOG_DEBUG,"Starting to listen on port %d", _listenerPort);
    
    while (!(isShutdown())) { 
      if (((serverSocket->getServerConn().isReadReady()))) { 
	Connection conn;
	serverSocket->accept(conn);
	processRequest(conn);
	conn.close();
      }
    }
    cpLog(LOG_DEBUG,"Stopped listening on port %d", _listenerPort);
  } else { 
    cpLog(LOG_ERR, "No socket open at port %d", _listenerPort);
  }
}

CacheListener::~CacheListener()
{
  if (!(isShutdown())) { 
    shutdown();
    join();
  }
  if (serverSocket) delete serverSocket;
  serverSocket = NULL;
  
  if (updateRecord) delete updateRecord;
}

CacheListener::CacheListener(CacheStorage* newMyCache)
   : _listenerPort(0)
{
  myCache = newMyCache;
  updateRecord = NULL;
}

int CacheListener::openPort(int startPort)
{ 

  _listenerPort = startPort;
  bool socketSuccessful = false;

  while ( ! socketSuccessful ) {
    socketSuccessful = true;
    try {
      serverSocket = new TcpServerSocket(_listenerPort);
    }
    catch ( VNetworkException ) {
      // could not listen here, so retry
      sleep(1); 
      _listenerPort++;
      socketSuccessful = false;  // redo 
      
      cpLog(LOG_ALERT, "Trying port %d", _listenerPort);
    }
  }
  cpLog (LOG_DEBUG, "Opened %d for data", _listenerPort);
  return _listenerPort;
}

int
CacheListener::getListenPort()
{
  if (_listenerPort == 0) { 
    // No port open, open port
    openPort();
  }

  return _listenerPort;
}

int
CacheListener::setListenPort(int newPort)
{
  if (_listenerPort == 0) { 
    // No port open, open port
    openPort(newPort);
  }

  return _listenerPort;
}

void CacheListener::processRequest(Connection& conn)
{

  if (!isShutdown()) { // Don't process requests during shutdown 

    char buf[MAXLINE];
    string dRead;
    buf[0] = '\0';
    int nread = 0;

    while (conn.readLine(buf, sizeof(buf), nread) > 0)
    {
        buf[nread] = 0;
        dRead += buf;
    }

    cpLog(LOG_DEBUG, "Processing request: (%s)", dRead.c_str());

    // Handle request
    if (myCache) { 
      if (updateRecord) { 
	// The message will tell us the key name of the updated record, 
	// so ditch them from the cache 
	myCache->remove(updateRecord->getTriggerRecordType(), dRead);
	
	// Now call the call back function
	(updateRecord->getCallBackFunc())(updateRecord->getTriggerRecordType(), dRead);

      } else { 
	cpLog(LOG_ERR, "Listener recieved a signal (%s) but has no record of what to update", dRead.c_str());
      }
    } else { 
      cpLog(LOG_ERR, "Listener recieved a signal (%s) but has no access to the cache itself", dRead.c_str());
    }

    return;
  }

}