P2pnsCache.h

Go to the documentation of this file.

//
// Copyright (C) 2007 Institut fuer Telematik, Universitaet Karlsruhe (TH)
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
//

#ifndef __P2PNSCACHE_H_
#define __P2PNSCACHE_H_

#include <set>
#include <vector>
#include <deque>
#include <sstream>

#include <omnetpp.h>

#include <NodeHandle.h>
#include <InitStages.h>
#include <BinaryValue.h>

struct P2pnsCacheEntry
{
    BinaryValue value;
    cMessage* ttlMessage;
    friend std::ostream& operator<<(std::ostream& Stream,
                                    const P2pnsCacheEntry entry);
};

enum P2pnsConnectionStates {
    CONNECTION_PENDING = 0,
    CONNECTION_ACTIVE = 1
};

class P2pnsIdCacheEntry
{
public:
    P2pnsIdCacheEntry(const OverlayKey& key) :
        key(key), state(CONNECTION_PENDING) {};

public:
    OverlayKey key;
    TransportAddress addr;
    P2pnsConnectionStates state;
    simtime_t lastUsage;
    std::deque<BinaryValue> payloadQueue;
};

typedef std::map<OverlayKey, P2pnsIdCacheEntry> P2pnsIdCache;

class P2pnsCache : public cSimpleModule
{
  public:
    virtual int numInitStages() const
    {
        return MAX_STAGE_APP;
    }

    virtual void initialize(int stage);
    virtual void handleMessage(cMessage* msg);

    virtual uint32_t getSize();

    virtual void clear();

    virtual bool isEmpty();


    virtual P2pnsIdCacheEntry* getIdCacheEntry(const OverlayKey& key);

    virtual P2pnsIdCacheEntry* addIdCacheEntry(const OverlayKey& key,
                                               const BinaryValue* payload = NULL);

    virtual void removeIdCacheEntry(const OverlayKey& key);

    virtual const BinaryValue& getData(const BinaryValue& name);

    virtual cMessage* getTtlMessage(const BinaryValue& name);

    virtual const BinaryValue& getDataAtPos(uint32_t pos = 0);

    virtual void addData(BinaryValue name, BinaryValue value, cMessage* ttlMessage = NULL);

    virtual void removeData(const BinaryValue& name);
    void display ();

  protected:
    std::map<BinaryValue, P2pnsCacheEntry> cache; 
    P2pnsIdCache idCache; 
    void updateDisplayString();

    void updateTooltip();
};

#endif