TCPSrvHostApp.h

Go to the documentation of this file.

//
// Copyright 2004 Andras Varga
//
// This library is free software, you can redistribute it and/or modify
// it under  the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation;
// either version 2 of the License, or any later version.
// The library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//

#ifndef __INET_TCPSRVHOSTAPP_H
#define __INET_TCPSRVHOSTAPP_H

#include <omnetpp.h>
#include "TCPSocket.h"
#include "TCPSocketMap.h"



class TCPServerThreadBase;

class INET_API TCPSrvHostApp : public cSimpleModule
{
  protected:
    TCPSocket serverSocket;
    TCPSocketMap socketMap;

  protected:
    virtual void initialize();
    virtual void handleMessage(cMessage *msg);
    virtual void finish();

    virtual void updateDisplay();

  public:
    virtual void removeThread(TCPServerThreadBase *thread);
};

class INET_API TCPServerThreadBase : public cPolymorphic, public TCPSocket::CallbackInterface
{
  protected:
    TCPSrvHostApp *hostmod;
    TCPSocket *sock; // ptr into socketMap managed by TCPSrvHostApp

  protected:
    // internal: TCPSocket::CallbackInterface methods
    virtual void socketDataArrived(int, void *, cPacket *msg, bool urgent) {dataArrived(msg,urgent);}
    virtual void socketEstablished(int, void *)  {established();}
    virtual void socketPeerClosed(int, void *) {peerClosed();}
    virtual void socketClosed(int, void *) {closed();}
    virtual void socketFailure(int, void *, int code) {failure(code);}
    virtual void socketStatusArrived(int, void *, TCPStatusInfo *status) {statusArrived(status);}
  public:
    // internal: called by TCPSrvHostApp after creating this module
    virtual void init(TCPSrvHostApp *hostmodule, TCPSocket *socket) {hostmod=hostmodule; sock=socket;}

  public:
    TCPServerThreadBase()  {sock=NULL;}
    virtual ~TCPServerThreadBase() {}

    virtual TCPSocket *getSocket() {return sock;}

    virtual TCPSrvHostApp *getHostModule() {return hostmod;}

    virtual void scheduleAt(simtime_t t, cMessage *msg)  {msg->setContextPointer(this); hostmod->scheduleAt(t,msg);}

    virtual void cancelEvent(cMessage *msg)  {hostmod->cancelEvent(msg);}

    virtual void established() = 0;

    virtual void dataArrived(cMessage *msg, bool urgent) = 0;

    virtual void timerExpired(cMessage *timer) = 0;

    virtual void peerClosed() {getSocket()->close();}

    virtual void closed() {hostmod->removeThread(this);}

    virtual void failure(int code) {hostmod->removeThread(this);}

    virtual void statusArrived(TCPStatusInfo *status) {delete status;}
};

#endif