/* -*- Mode:C++; c-basic-offset:8; tab-width:8; indent-tabs-mode:t -*- */
/*
* Copyright (c) Xerox Corporation 1997. All rights reserved.
*
* 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.,
* 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* Linking this file statically or dynamically with other modules is making
* a combined work based on this file. Thus, the terms and conditions of
* the GNU General Public License cover the whole combination.
*
* In addition, as a special exception, the copyright holders of this file
* give you permission to combine this file 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
* file 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 file 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.
*
* $Header: /nfs/jade/vint/CVSROOT/ns-2/adc/sa.cc,v 1.16 2005/08/26 05:05:28 tomh Exp $
*/
//packets after it succeeds in a3-way handshake from the receiver
// should be connected with Agent/SignalAck class
#include "udp.h"
#include "sa.h"
#include "ip.h"
#include "random.h"
#define SAMPLERATE 8000
SA_Agent::SA_Agent() : Agent(PT_UDP), trafgen_(0), rtd_(0), callback_(0),
sa_timer_(this), nextPkttime_(-1), running_(0), seqno_(-1)
{
bind_bw("rate_",&rate_);
bind("bucket_",&bucket_);
bind("packetSize_", &size_);
}
SA_Agent::~SA_Agent()
{
if (callback_)
delete[] callback_;
}
static class SA_AgentClass : public TclClass {
public:
SA_AgentClass() : TclClass("Agent/SA") {}
TclObject* create(int, const char*const*) {
return (new SA_Agent());
}
} class_signalsource_agent;
int SA_Agent::command(int argc, const char*const* argv)
{
Tcl& tcl = Tcl::instance();
if (argc==3) {
if (strcmp(argv[1], "target") == 0) {
target_ = (NsObject*)TclObject::lookup(argv[2]);
if (target_ == 0) {
tcl.resultf("no such object %s", argv[2]);
return (TCL_ERROR);
}
ctrl_target_=target_;
return (TCL_OK);
}
else if (strcmp(argv[1],"ctrl-target")== 0) {
ctrl_target_=(NsObject*)TclObject::lookup(argv[2]);
if (ctrl_target_ == 0) {
tcl.resultf("no such object %s", argv[2]);
return (TCL_ERROR);
}
return (TCL_OK);
}
if (strcmp(argv[1], "stoponidle") == 0) {
stoponidle(argv[2]);
return(TCL_OK);
}
if (strcmp(argv[1], "attach-traffic") == 0) {
trafgen_ =(TrafficGenerator*)TclObject::lookup(argv[2]);
if (trafgen_ == 0) {
tcl.resultf("no such node %s", argv[2]);
return(TCL_ERROR);
}
return(TCL_OK);
}
}
if (argc == 2) {
if (strcmp(argv[1], "start") == 0) {
start();
return(TCL_OK);
} else if (strcmp(argv[1], "stop") == 0) {
stop();
return(TCL_OK);
}
}
return (Agent::command(argc,argv));
}
void SA_Agent::start()
{
//send the request packet
if (trafgen_) {
trafgen_->init();
//running_=1;
sendreq();
}
}
void SA_Agent::stop()
{
sendteardown();
if (running_ != 0) {
sa_timer_.cancel();
running_ =0;
}
}
void SA_Agent::sendreq()
{
Packet *p = allocpkt();
hdr_cmn* ch= hdr_cmn::access(p);
ch->ptype()=PT_REQUEST;
ch->size()=20;
//also put in the r,b parameters for the flow in the packet
hdr_resv* rv=hdr_resv::access(p);
rv->decision() =1;
rv->rate()=rate_;
rv->bucket()=bucket_;
ctrl_target_->recv(p);
}
void SA_Agent::sendteardown()
{
Packet *p = allocpkt();
hdr_cmn* ch= hdr_cmn::access(p);
ch->ptype()=PT_TEARDOWN;
ch->size()=20;
//also put in the r,b parameters for the flow in the packet
hdr_resv* rv=hdr_resv::access(p);
rv->decision() =1;
rv->rate()=rate_;
rv->bucket()=bucket_;
ctrl_target_->recv(p);
}
void SA_Agent::recv(Packet *p, Handler *)
{
hdr_cmn *ch= hdr_cmn::access(p);
hdr_resv *rv=hdr_resv::access(p);
hdr_ip * iph = hdr_ip::access(p);
if ( ch->ptype() == PT_ACCEPT || ch->ptype() == PT_REJECT ) {
ch->ptype() = PT_CONFIRM;
// turn the packet around by swapping src and dst
// (address and port)
ns_addr_t tmp;
tmp = iph->src();
iph->src() = iph->dst();
iph->dst() = tmp;
ctrl_target_->recv(p);
}
// put an additional check here to see if admission was granted
if (rv->decision()) {
//printf("Flow %d accepted @ %f\n",iph->flowid(),Scheduler::instance().clock());
fflush(stdout);
double t = trafgen_->next_interval(size_);
running_=1;
sa_timer_.resched(t);
}
else {
//printf("Flow %d rejected @ %f\n",iph->flowid(),Scheduler::instance().clock());
fflush(stdout);
//Currently the flow is stopped if rejected
running_=0;
}
//make an upcall to sched a stoptime for this flow from now
Tcl::instance().evalf("%s sched-stop %d",name(),rv->decision());
}
void SA_Agent::stoponidle(const char *s)
{
callback_ = new char[strlen(s)+1];
strcpy(callback_, s);
if (trafgen_->on()) {
// Tcl::instance().evalf("puts \"%s waiting for burst at %f\"", name(), Scheduler::instance().clock());
rtd_ = 1;
}
else {
stop();
Tcl::instance().evalf("%s %s", name(), callback_);
}
}
void SA_Timer::expire(Event* /*e*/) {
a_->timeout(0);
}
void SA_Agent::timeout(int)
{
if (running_) {
/* send a packet */
sendpkt();
/* figure out when to send the next one */
nextPkttime_ = trafgen_->next_interval(size_);
/* schedule it */
sa_timer_.resched(nextPkttime_);
/* hack: if we are waiting for a current burst to end
* before stopping . . .
*/
if (rtd_) {
if (trafgen_->on() == 0) {
stop();
//Tcl::instance().evalf("puts \"%s burst over at %f\"",
// name(), Scheduler::instance().clock());
Tcl::instance().evalf("%s sched-stop %d", name(), 0);
}
}
}
}
void SA_Agent::sendpkt()
{
Packet* p = allocpkt();
hdr_rtp* rh = hdr_rtp::access(p);
rh->seqno() = ++seqno_;
rh->flags()=0;
double local_time=Scheduler::instance().clock();
/*put in "rtp timestamps" and begining of talkspurt labels */
hdr_cmn* ch = hdr_cmn::access(p);
ch->timestamp()=(u_int32_t)(SAMPLERATE*local_time);
ch->size()=size_;
if ((nextPkttime_ != trafgen_->interval()) || (nextPkttime_ == -1))
rh->flags() |= RTP_M;
target_->recv(p);
}
void SA_Agent::sendmsg(int nbytes, const char* /*flags*/)
{
Packet *p;
int n;
assert (size_ > 0);
n = nbytes / size_;
if (nbytes == -1) {
start();
return;
}
while (n-- > 0) {
p = allocpkt();
hdr_rtp* rh = hdr_rtp::access(p);
rh->seqno() = seqno_;
target_->recv(p);
}
n = nbytes % size_;
if (n > 0) {
p = allocpkt();
hdr_rtp* rh = hdr_rtp::access(p);
rh->seqno() = seqno_;
target_->recv(p);
}
idle();
}
syntax highlighted by Code2HTML, v. 0.9.1