| #ifndef _PRAGMA_COPYRIGHT_ |
| #define _PRAGMA_COPYRIGHT_ |
| #pragma comment(copyright, "%Z% %I% %W% %D% %T%\0") |
| #endif /* _PRAGMA_COPYRIGHT_ */ |
| /**************************************************************************** |
| |
| * Copyright (c) 2008, 2010 IBM Corporation. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v1.0s |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v10.html |
| |
| Classes: Listener |
| |
| Description: Listener Thread. |
| |
| Author: Tu HongJ, Liu Wei, Nicole Nie |
| |
| History: |
| Date Who ID Description |
| -------- --- --- ----------- |
| 10/06/08 tuhongj Initial code (D153875) |
| |
| ****************************************************************************/ |
| |
| #include "listener.hpp" |
| #include <assert.h> |
| #include <stdlib.h> |
| #include <sys/socket.h> |
| #include <sys/wait.h> |
| |
| #include "log.hpp" |
| #include "stream.hpp" |
| #include "exception.hpp" |
| |
| #include "ctrlblock.hpp" |
| #include "statemachine.hpp" |
| #include "socket.hpp" |
| #include "readerproc.hpp" |
| #include "writerproc.hpp" |
| #include "queue.hpp" |
| |
| Listener:: Listener(int hndl) |
| : Thread(hndl), bindPort(-1) |
| { |
| socket = new Socket(); |
| } |
| |
| Listener::~Listener() |
| { |
| delete socket; |
| } |
| |
| int Listener::init() |
| { |
| bindPort = 0; |
| char *envp = ::getenv("SCI_DEVICE_NAME"); |
| if (envp) { |
| string ifname = envp; |
| sockfd = socket->listen(bindPort, ifname); |
| } else { |
| sockfd = socket->listen(bindPort); |
| } |
| |
| log_debug("listener binded to port %d", bindPort); |
| |
| return bindPort; |
| } |
| |
| int Listener::stop() |
| { |
| setState(false); |
| ::shutdown(sockfd, SHUT_RDWR); |
| ::close(sockfd); |
| |
| return 0; |
| } |
| |
| void Listener::run() |
| { |
| int child = -1; |
| int hndl = -1; |
| int key; |
| |
| while (getState()) { |
| try { |
| child = socket->accept(sockfd); |
| } catch (SocketException &e) { |
| log_warn("Listener: socket broken: %s", e.getErrMsg().c_str()); |
| break; |
| } catch (...) { |
| log_warn("Listener: unknown exception: %s"); |
| break; |
| } |
| if (child < 0) { |
| // invalid connection |
| continue; |
| } |
| if (!gCtrlBlock->isEnabled()) { |
| log_debug("Listener: uninitialized, rejected this connection"); |
| break; |
| } |
| |
| log_debug("Listener: accepted a child agent sockfd %d", child); |
| |
| try { |
| Stream *stream = new Stream(); |
| stream->init(child); |
| *stream >> key; |
| if (key != gCtrlBlock->getJobKey()) { |
| log_warn("Listener: client with invalid credential is trying to connect."); |
| stream->stop(); |
| delete stream; |
| continue; |
| } |
| |
| *stream >> hndl >> endl; |
| if (hndl >= 0) { |
| log_debug("Listener: back end %d is connected", hndl); |
| } else { |
| log_debug("Listener: agent %d is connected", hndl); |
| } |
| |
| gStateMachine->parse(StateMachine::CLIENT_CONNECTED); |
| |
| char name[32]; |
| MessageQueue *inQ = gCtrlBlock->queryQueue(hndl); |
| if (NULL == inQ) { |
| inQ = new MessageQueue(); |
| if (hndl >= 0) { |
| ::sprintf(name, "BE%d_inQ", hndl); |
| } else { |
| ::sprintf(name, "Agent%d_inQ", hndl); |
| } |
| inQ->setName(string(name)); |
| gCtrlBlock->registerQueue(inQ); |
| gCtrlBlock->mapQueue(hndl, inQ); |
| gCtrlBlock->genSelfInfo(inQ, false); |
| } |
| |
| ReaderProcessor *reader = new ReaderProcessor(hndl); |
| reader->setInStream(stream); |
| reader->setOutQueue(gCtrlBlock->getFilterInQueue()); |
| ::sprintf(name, "Reader%d", hndl); |
| reader->setName(string(name)); |
| reader->setOutErrorQueue(gCtrlBlock->getErrorQueue()); |
| |
| WriterProcessor *writer = new WriterProcessor(hndl); |
| writer->setInQueue(inQ); |
| writer->setOutStream(stream); |
| ::sprintf(name, "Writer%d", hndl); |
| writer->setName(string(name)); |
| |
| // writer is a peer processor of reader |
| reader->setPeerProcessor(writer); |
| |
| reader->start(); |
| writer->start(); |
| |
| gCtrlBlock->registerProcessor(reader); |
| gCtrlBlock->registerProcessor(writer); |
| |
| gCtrlBlock->registerStream(stream); |
| } catch (Exception &e) { |
| log_error("Listener: exception %s", e.getErrMsg()); |
| gStateMachine->parse(StateMachine::FATAL_EXCEPTION); |
| break; |
| } catch (ThreadException &e) { |
| log_error("Listener: thread exception %d", e.getErrCode()); |
| gStateMachine->parse(StateMachine::FATAL_EXCEPTION); |
| break; |
| } catch (SocketException &e) { |
| log_error("Listener: socket exception: %s", e.getErrMsg().c_str()); |
| gStateMachine->parse(StateMachine::FATAL_EXCEPTION); |
| break; |
| } catch (std::bad_alloc) { |
| log_error("Listener: out of memory"); |
| gStateMachine->parse(StateMachine::FATAL_EXCEPTION); |
| break; |
| } catch (...) { |
| log_error("Listener: unknown exception"); |
| gStateMachine->parse(StateMachine::FATAL_EXCEPTION); |
| break; |
| } |
| } |
| |
| ::close(sockfd); |
| setState(false); |
| } |
| |