tpm.cc

Go to the documentation of this file.

/* -*-  Mode:C++; c-basic-offset:8; tab-width:8; indent-tabs-mode:t -*- */
//

/*
 * tpm.cc
 * Copyright (C) 2001 by the University of Southern California
 * $Id: tpm.cc,v 1.2 2005/08/25 18:58:02 johnh Exp $
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * 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.
 *
 *
 * The copyright of this module includes the following
 * linking-with-specific-other-licenses addition:
 *
 * In addition, as a special exception, the copyright holders of
 * this module give you permission to combine (via static or
 * dynamic linking) this module with free software programs or
 * libraries that are released under the GNU LGPL and with code
 * included in the standard release of ns-2 under the Apache 2.0
 * license or under otherwise-compatible licenses with advertising
 * requirements (or modified versions of such code, with unchanged
 * license).  You may copy and distribute such a system following the
 * terms of the GNU GPL for this module and the licenses of the
 * other code concerned, provided that you include the source code of
 * that other code when and as the GNU GPL requires distribution of
 * source code.
 *
 * Note that people who make modified versions of this module
 * are not obligated to grant this special exception for their
 * modified versions; it is their choice whether to do so.  The GNU
 * General Public License gives permission to release a modified
 * version without this exception; this exception also makes it
 * possible to release a modified version which carries forward this
 * exception.
 *
 */
//
// Other copyrights might apply to parts of this software and are so
// noted when applicable.
//
// Management function to playback TCPDump trace file (after post precssing)
// Xuan Chen (xuanc@isi.edu)
//
#include "tpm.h"

// Timer to send requests
TPMTimer::TPMTimer(TPM* t) {
    tpm = t;
};

void TPMTimer::expire(Event *e) {
    //if (e) 
    //  Packet::free((Packet *)e);

    tpm->run();
}


static class TPMClass : public TclClass {
public:
        TPMClass() : TclClass("TPM") {}
        TclObject* create(int, const char*const*) {
        return (new TPM());
    }
} class_tpm;

TPM::TPM() {
    tp_default_ = tp_num_ = 0;
    tp = NULL;
    
    // initialize next request
    next_p.time = 0;
    next_p.saddr = 0;
    next_p.sport = 0;
    next_p.daddr = 0;
    next_p.dport = 0;
    next_p.len = 0;

    // initialize request timer
    timer = new TPMTimer(this);
    start_t = 0;
}

TPM::~TPM() {
    if (fp)
        fclose(fp);
    if (timer)
        delete timer;
}

int TPM::loadLog(const char* filename) {
    fp = fopen(filename, "r");
    if (fp == 0)
        return(0);
    
    return(1);
}

int TPM::start() {
    start_t = Scheduler::instance().clock();
    processLog();
    return(1);
}

int TPM::processLog() {
    int time, sport, dport, len;
    unsigned int saddr, daddr;

    while (!feof(fp)) {
        fscanf(fp, "%d %u %d %u %d %d\n",
               &time, &saddr, &sport, &daddr, &dport, &len);

        if (time > 0) {
            // save information for next request
            next_p.time = time;
            next_p.saddr = saddr;
            next_p.sport = sport;
            next_p.daddr = daddr;
            next_p.dport = dport;
            next_p.len = len;   
            
            //double now = Scheduler::instance().clock();
            //double delay = time + start_t - now;
            double delay = time * 1.0 / 1000000.0;
            timer->resched(delay);
            
            return(1);
        } else {
            genPkt(saddr, sport, daddr, dport, len);
        }
    }
    return(1);
}

int TPM::run() {
    genPkt(next_p.saddr, next_p.sport, next_p.daddr, next_p.dport, next_p.len);
    processLog();
    return(1);
}

int TPM::genPkt(unsigned int saddr, int sport, unsigned int daddr, int dport, int len) {
    // we can send packets to anybody with twinks...
    if (saddr == daddr) {
           return(0);
    }

    // use the default TP agent unless specified.
    unsigned int tp_id = tp_default_;

    if (saddr < (unsigned int)tp_num_) {
        tp_id = saddr;
    }
    printf("tpid %d\n", tp_id);
    tp[tp_id]->sendto(len, saddr, sport, daddr, dport);
    
    return(1);
}

int TPM::command(int argc, const char*const* argv) {
    if (argc == 2) {
        if (strcmp(argv[1], "start") == 0) {
            if (start())
                return(TCL_OK);
            else
                return(TCL_ERROR);
            
        }
        
    } else if (argc == 3) {
        if (strcmp(argv[1], "loadLog") == 0) {
            if (loadLog(argv[2]))
                return(TCL_OK);
            else
                return(TCL_ERROR);

        } else if (strcmp(argv[1], "tp-num") == 0) {
            tp_num_ = atoi(argv[2]);
            tp_default_ = tp_num_ - 1;
            if (tp_default_ < 0)
              tp_default_ = 0;
            tp = new TPAgent*[tp_num_];
            return(TCL_OK);
        } else if (strcmp(argv[1], "tp-default") == 0) {
                tp_default_ = atoi(argv[2]);
            return(TCL_OK);
        }
    } else if (argc == 4) {
        if (strcmp(argv[1], "tp-reg") == 0) {
            int index = atoi(argv[2]);
            if (index >= tp_num_)
                return(TCL_ERROR);

            tp[index] = (TPAgent *) TclObject::lookup(argv[3]);
            return(TCL_OK);
        }
    }

    printf("Command specified does not exist\n");
    return(TCL_ERROR);
}

---