priq.cc

在网络仿真模拟工具下实现支持QoS的MAC层EDCF协议

#include <object.h>
#include <queue.h>
#include <packet.h>
#include <cmu-trace.h>
#include <iostream.h>
#include "priq.h"
#include "mac-802_11e.h"
#include <random.h>

typedef int (*PacketFilter)(Packet *, void *);

PriQ_List PriQ::prhead = { 0 };

static class PriQClass : public TclClass {
public:
  PriQClass() : TclClass("Queue/DTail/PriQ") {}
  TclObject* create(int, const char*const*) {
    return (new PriQ);
  }
} class_PriQ;


PriQ::PriQ() : DTail()
{
        bind("Prefer_Routing_Protocols", &Prefer_Routing_Protocols);
	bind("Max_Levels", &max_levels);
	bind("Levels", &levels);
	
	//added by smallko
	bind("choice", &choice);
	bind("threshold1", &threshold1);
        bind("threshold2", &threshold2);
        bind("prob0", &prob0);
        bind("prob1", &prob1);
        bind("prob2", &prob2);
        bind("qlen0", &qlen0);
        bind("qlen1", &qlen1);
        bind("qlen2", &qlen2);
        bind("qlen3", &qlen3); 
	
	LIST_INSERT_HEAD(&prhead, this, link);
	flag = 0;
	cntl = 0;
	level = 0;
}

int
PriQ::command(int argc, const char*const* argv)
{
    Scheduler &s = Scheduler::instance();
    if (argc == 5 )
    { 
	if(strcmp(argv[1], "Prio") == 0){
	    if(!(atoi(argv[2]) > levels)){
		level = atoi(argv[2]);
		if(strcmp(argv[3], "PF") == 0){
		    pri_[level].setPF(atoi(argv[4]));
		    return(TCL_OK);
		}
		if(strcmp(argv[3], "CW_MIN") == 0){
		    pri_[level].setCW_MIN(atoi(argv[4]));
		    return(TCL_OK);
		}
		if(strcmp(argv[3], "CW_MAX") == 0){
		    pri_[level].setCW_MAX(atoi(argv[4]));
		    return(TCL_OK);
		}
		if(strcmp(argv[3], "AIFS") == 0){
		    pri_[level].setAIFS(atoi(argv[4])); 
		    return(TCL_OK);	      
		}
		if(strcmp(argv[3], "TXOPLimit") == 0){
		    pri_[level].setTXOPLimit(atof(argv[4])); 
		    return(TCL_OK);	      
		}
	    }else return (TCL_ERROR);
	}
    }
    if (argc == 2 && strcasecmp(argv[1], "reset") == 0)
    {
	Terminate();
	//FALL-THROUGH to give parents a chance to reset
    }
    
    return DTail::command(argc, argv);
}

void
PriQ::recv(Packet *p, Handler *h)
{
    if(flag == 0) {
	((Mac802_11e*) target())->queue_ = this;
	flag = 1;
	for(int i = 0; i < MAX_PRI; i++) {
	    pri_[i].setdrop(drop_);    
	}
    }
    
    struct hdr_cmn *ch = HDR_CMN(p);
    if(Prefer_Routing_Protocols) {

                switch(ch->ptype()) {
                case PT_DSR:
                case PT_MESSAGE:
                case PT_TORA:
                case PT_AODV:
                        recvHighPriority(p, h);
                        break;

                default:
		    pri_recv(p, h);
                }
        }
        else {
	    pri_recv(p, h);
        }
}

void
PriQ::pri_recv(Packet *p, Handler *h)
{
/*
    printf("queue length of AC0:%f\n", pri_[0].getavgLen());
    printf("queue length of AC1:%f\n", pri_[1].getavgLen());
    printf("queue length of AC2:%f\n", pri_[2].getavgLen());
    printf("queue length of AC3:%f\n", pri_[3].getavgLen());
*/
    int done;
    double tmpx, prob;
    done=0;
    Scheduler &s = Scheduler::instance();
    level = PKT_LEVEL(p); 
    struct hdr_cmn *ch = HDR_CMN(p);
    hdr_ip* iph = hdr_ip::access(p);
    
    qlen0=pri_[0].getLen();
    qlen1=pri_[1].getLen();
    qlen2=pri_[2].getLen();
    qlen3=pri_[3].getLen();
    
    //printf("prio_:%d, frametype_:%d\n", iph->prio(), ch->frametype_);
      //printf("queue length of AC1:%f\n", pri_[1].getavgLen());
    //for AC1: video 1: I:AC1 P:AC2 B: AC3 static mapping
    if(choice==1 &&level==1 && ch->frametype_!=0){
    	//printf("choice=1, level=1, ch->frametype_!=0\n");
    	switch(ch->frametype_) {
  	case 1: 
  	  iph->prio()=1;
	  pri_[1].target_handle(target_);
    	  pri_[1].recv(p,h);
    	  done=1;
  	  break;
  	case 2: 
  	 iph->prio()=2;
	  pri_[2].target_handle(target_);
    	  pri_[2].recv(p,h);
    	  done=1;
  	  break;
  	case 3: 
  	  iph->prio()=3;
	  pri_[3].target_handle(target_);
    	  pri_[3].recv(p,h);
    	  done=1;
          break;
  	default:
  	  printf("Error in PriQ::pri_recv()\n");
          break;
  	}
    }
    
    //for AC1: video dynamic mapping
    if(choice==2 && level==1 && ch->frametype_!=0){
    	//printf("choice=2, level=1, ch->frametype_!=0\n");
    	switch(ch->frametype_) {
  	case 1: 
  	  if(int(pri_[1].getavgLen()< threshold1)){
  	  	iph->prio()=1;
	  	pri_[1].target_handle(target_);
    	   	pri_[1].recv(p,h);
    	   	done=1;
  	  } else if(int(pri_[1].getavgLen()< threshold2)) {	
  	  	prob= prob0*(pri_[1].getavgLen()-threshold1)/(threshold2-threshold1);
  	  	tmpx=Random::uniform(0.0,1.0);
  	  	if(tmpx<prob){
  	  		iph->prio()=2;
    	  		pri_[2].target_handle(target_);
    	  		pri_[2].recv(p,h);
    	  	}else {
    	  		iph->prio()=1;
	  		pri_[1].target_handle(target_);
    	  		pri_[1].recv(p,h);
    	  	}
    	  	done=1;
    	  } else {
    	  	tmpx=Random::uniform(0.0,1.0);
  	  	if(tmpx<prob0){
  	  		iph->prio()=3;
    	  		pri_[3].target_handle(target_);
    	  		pri_[3].recv(p,h);
    	  	}else {
    	  		iph->prio()=2;
	  		pri_[2].target_handle(target_);
    	  		pri_[2].recv(p,h);
    	  	}
    	  	done=1;
    	  	break;
    	  }
  	  break;
  	case 2: 
  	  if(int(pri_[1].getavgLen()< threshold1)){
  	  	iph->prio()=1;
	  	pri_[1].target_handle(target_);
    	   	pri_[1].recv(p,h);
    	   	done=1;
  	  } else if(int(pri_[1].getavgLen()< threshold2)) {	
  	  	prob= prob1*(pri_[1].getavgLen()-threshold1)/(threshold2-threshold1);
  	  	tmpx=Random::uniform(0.0,1.0);
  	  	if(tmpx<prob){
  	  		iph->prio()=2;
    	  		pri_[2].target_handle(target_);
    	  		pri_[2].recv(p,h);
    	  	}else {
    	  		iph->prio()=1;
	  		pri_[1].target_handle(target_);
    	  		pri_[1].recv(p,h);
    	  	}
    	  	done=1;
    	  } else {
    	  	tmpx=Random::uniform(0.0,1.0);
  	  	if(tmpx<prob1){
  	  		iph->prio()=3;
    	  		pri_[3].target_handle(target_);
    	  		pri_[3].recv(p,h);
    	  	}else {
    	  		iph->prio()=2;
	  		pri_[2].target_handle(target_);
    	  		pri_[2].recv(p,h);
    	  	}
    	  	done=1;
    	  	break;
    	  }
  	  break;
  	case 3: 
  	 if(int(pri_[1].getavgLen()< threshold1)){
  	  	iph->prio()=1;
	  	pri_[1].target_handle(target_);
    	   	pri_[1].recv(p,h);
    	   	done=1;
  	  } else if(int(pri_[1].getavgLen()< threshold2)) {	
  	  	prob= prob2*(pri_[1].getavgLen()-threshold1)/(threshold2-threshold1);
  	  	tmpx=Random::uniform(0.0,1.0);
  	  	if(tmpx<prob){
  	  		iph->prio()=2;
    	  		pri_[2].target_handle(target_);
    	  		pri_[2].recv(p,h);
    	  	}else {
    	  		iph->prio()=1;
	  		pri_[1].target_handle(target_);
    	  		pri_[1].recv(p,h);
    	  	}
    	  	done=1;
    	  } else {
    	  	tmpx=Random::uniform(0.0,1.0);
  	  	if(tmpx<prob2){
  	  		iph->prio()=3;
    	  		pri_[3].target_handle(target_);
    	  		pri_[3].recv(p,h);
    	  	}else {
    	  		iph->prio()=2;
	  		pri_[2].target_handle(target_);
    	  		pri_[2].recv(p,h);
    	  	}
    	  	done=1;
    	  	break;
    	  }
          break;
  	default:
  	  printf("Error in PriQ::pri_recv()\n");
          break;
  	}
    }
   
    if(done==1)
    	return;
    	
     /*  target_handle() is necessary to give the target to class Queue.  
     *  Otherwise the target is not known in class Queue 
     *  when dequeing packet :o(
     */
    pri_[level].target_handle(target_);
    pri_[level].recv(p,h);
}    


void 
PriQ::recvHighPriority(Packet *p, Handler *)
  // insert packet at front of queue
{
    pri_[cntl].q_->enqueHead(p);
    pri_[cntl].target_handle(target_);
    if (pri_[cntl].q_->length() >= qlim_)
    {
      Packet *to_drop = pri_[cntl].q_->lookup(pri_[cntl].q_->length()-1);
      pri_[cntl].q_->remove(to_drop);
      drop(to_drop);
    }
  if (!pri_[cntl].blocked()) {
    /*
     * We're not blocked.  Get a packet and send it on.
     * We perform an extra check because the queue
     * might drop the packet even if it was
     * previously empty!  (e.g., RED can do this.)
     */
  

p = pri_[cntl].deque();
    if (p != 0) {
      pri_[cntl].block();
      pri_[cntl].recvHighPri(p);
      //target_->recv(p, &qh_); <- done in d-tail.cc
    }
  } 
}
 
void 
PriQ::filter(PacketFilter filter, void * data)
  // apply filter to each packet in queue, 
  // - if filter returns 0 leave packet in queue
  // - if filter returns 1 remove packet from queue
{
    int i = 0;
    while (i < pri_[cntl].q_->length())
    {
      Packet *p = pri_[cntl].q_->lookup(i);
      if (filter(p,data))
	{
	  pri_[cntl].q_->remove(p); // decrements q len
	}
      else i++;
    }
}

Packet*
PriQ::filter(nsaddr_t id)
{
    Packet *p = 0;
    Packet *pp = 0;
    struct hdr_cmn *ch;
    
    for(p = pri_[cntl].q_->head(); p; p = p->next_) {
	ch = HDR_CMN(p);
	if(ch->next_hop() == id)
	    break;
	pp = p;
    }

	/*
	 * Deque Packet
	 */
     if(p) {
	if(pp == 0)
	    pri_[cntl].q_->remove(p);
	else
	    pri_[cntl].q_->remove(p, pp);
    }
    return p;
}



/*
 * Called at the end of the simulation to purge the IFQ.
 */
void
PriQ::Terminate()
{
    for(int i = 0; i< MAX_PRI; i++){
	Packet *p;
	while((p = pri_[i].deque())) {
	    drop(p, DROP_END_OF_SIMULATION);
	    //drop(p);
		
	}
    }
}

int PriQ::getLevels(){
    return levels;
}