DHTTestApp.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 __DHTTESTAPP_H_
#define __DHTTESTAPP_H_

#include <omnetpp.h>

#include <GlobalNodeList.h>
#include <GlobalStatistics.h>
#include <UnderlayConfigurator.h>
#include <TransportAddress.h>
#include <OverlayKey.h>
#include <InitStages.h>
#include <BinaryValue.h>
#include <BaseApp.h>
#include <set>
#include <sstream>

class GlobalDhtTestMap;

class DHTTestApp : public BaseApp
{
private:
    class DHTStatsContext : public cPolymorphic
    {
    public:
        bool measurementPhase;
        simtime_t requestTime;
        OverlayKey key;
        BinaryValue value;

        DHTStatsContext(bool measurementPhase,
                        simtime_t requestTime,
                        const OverlayKey& key,
                        const BinaryValue& value = BinaryValue::UNSPECIFIED_VALUE) :
            measurementPhase(measurementPhase), requestTime(requestTime),
            key(key), value(value) {};
    };

    void initializeApp(int stage);

    OverlayKey getRandomKey();

    BinaryValue generateRandomValue();

    void finishApp();

    virtual void handleGetResponse(DHTgetCAPIResponse* msg,
                                   DHTStatsContext* context);

    virtual void handlePutResponse(DHTputCAPIResponse* msg,
                                   DHTStatsContext* context);

    virtual void handleTimerEvent(cMessage* msg);

    virtual void handleTraceMessage(cMessage* msg);

    virtual void handleNodeLeaveNotification();

    // see RpcListener.h
    void handleRpcResponse(BaseResponseMessage* msg, const RpcState& state,
                           simtime_t rtt);

    UnderlayConfigurator* underlayConfigurator; 
    GlobalNodeList* globalNodeList; 
    GlobalStatistics* globalStatistics; 
    GlobalDhtTestMap* globalDhtTestMap; 
    // parameters
    bool debugOutput; 
    double mean; 
    double deviation; 
    int ttl; 
    bool p2pnsTraffic; 
    bool activeNetwInitPhase; 

    // statistics
    int numSent; 
    int numGetSent; 
    int numGetError; 
    int numGetSuccess; 
    int numPutSent; 
    int numPutError; 
    int numPutSuccess; 
    cMessage *dhttestput_timer, *dhttestget_timer, *dhttestmod_timer;
    bool nodeIsLeavingSoon; 

    static const int DHTTESTAPP_VALUE_LEN = 20;

public:
    DHTTestApp();

    virtual ~DHTTestApp();

};

#endif