dosdbellasim.c

跑leach需要的

//This file outputs a ns file for a dumbell network to be used for
//DDOS experiments.
/*the input file is a script file which is read by the program to get
the parameters of the network.

A typical file example
d (diameter of the network) H (hurtz parameter)  
n1 (no of incoming TCP nodes) rate (rate) 
n2 (no. of DDOS attackers in the subtree) master (0/1) rate (rate)
*/

#include<stdio.h>
#include<stdlib.h>
#include "dosdbell.h"


void set_Topology(FILE *filer);
void TCL_Write_Initial(FILE *);
void TCL_Write_Nodes(FILE *);
void TCL_Write_Connections(FILE*);
void TCL_Write_Agents(FILE *);
void TCL_Write_DDOS_Agent(FILE*);
void TCL_Write_Final(FILE*);

struct Topology topo;
int diameter;
double TCP_start_time;
double DDOS_start_time;
double finish_time;
double bw;
int pack_size;
float bn_delay;
float bn_bw ; 

main(int argc, char **argv){
  FILE *filer;
 
  if(argc !=3){
    printf("USAGE: %s parameter_filename tcloutput_filename\n",argv[0]);
    exit(0);
  }
  else{
    filer = fopen(argv[1],"r");
    set_Topology(filer);
    fclose(filer);
    filer = fopen(argv[2],"w");
    TCL_Write_Initial(filer);
    TCL_Write_Nodes(filer);
    TCL_Write_Connections(filer);
    TCL_Write_Agents(filer);
    TCL_Write_DDOS_Agent(filer);
    /* TCL_Write_Final(filer);*/
    fclose(filer);
  }  
}


  
void set_Topology(FILE *filer){
  double perT, perFTP, perCBR, perV;
  int i;
  double H;
  double rnd;
  int attackNo,master;
  double DDOSRate;
  double hop;  

  fscanf(filer,"%d %f %f\n",&diameter,&bn_bw,&bn_delay);
  fscanf(filer,"%d %lf %lf\n",&topo.TCPNodes.NoNodes,&bw,&TCP_start_time);  
  topo.TCPNodes.TCPNode = (struct TCP_Node_Info *)malloc(topo.TCPNodes.NoNodes*sizeof(struct TCP_Node_Info));

  for(i=0;i<topo.TCPNodes.NoNodes;i++){
    
    /* Based on the topology, get the number of hops from the source to destination.
       Since a dumbbell has a minimum of three hops, the number of hops returned 
       should be greater than 3 
    */ 

    do{
      topo.TCPNodes.TCPNode[i].delayFrom = (double) Inet_default_no_hops(diameter);
    }while(topo.TCPNodes.TCPNode[i].delayFrom<=3);

    /* Calculate the delay in for both the links */
    topo.TCPNodes.TCPNode[i].delayTo = ceil(drand48()*(topo.TCPNodes.TCPNode[i].delayFrom-2));
    topo.TCPNodes.TCPNode[i].delayFrom -= topo.TCPNodes.TCPNode[i].delayTo;
    
    /* Maximum delay between the start of all sources */
    rnd = ((double)(drand48()*TCP_start_time));
    topo.TCPNodes.TCPNode[i].startTime = rnd;
  }

  fscanf(filer,"%d %lf %d %d %lf\n",&topo.CBRTraffic.NoAttackers,&topo.CBRTraffic.DDOSRate,&pack_size,&master,&DDOS_start_time); 
  CBRTrafficInet(diameter,topo.CBRTraffic.NoAttackers, master, topo.CBRTraffic.DDOSRate, &topo.CBRTraffic.DDOSTraffic);
  if(master==0)
    topo.CBRTraffic.master = NoMaster;
  else
    topo.CBRTraffic.master = Master;
  topo.CBRTraffic.MaxTime = topo.CBRTraffic.DDOSTraffic.Events[topo.CBRTraffic.DDOSTraffic.NoEvents-1].delay;
  fscanf(filer,"%lf\n",&finish_time);
  printf("Topology Generated Using:\n Diameter=%d BottleNeck: BW=%.3fMbps Delay=%.3fms\n \
Background Traffic: TCP Nodes=%d BW=%.3lfMbps Starttime=%.1lfs\n \
Dos Traffic       : Attacker= %d BW=%.3lfMbps Starttime=%.1lfs\n \
PacketSize=%dB  Master = %d  Finishtime = %.1lfs\n", \
	 diameter,bn_bw,bn_delay,topo.TCPNodes.NoNodes,bw,TCP_start_time, \
	 topo.CBRTraffic.NoAttackers, \
         topo.CBRTraffic.DDOSRate, DDOS_start_time,pack_size,master,finish_time);
}

void TCL_Write_Initial(FILE *filew){
  fprintf(filew, "# Dumping asim data\n\n");
}


void TCL_Write_Nodes(FILE *filew){
	
}

void TCL_Write_Connections(FILE *filew){
  int i;
  int bw_;
  double del_;
  int c;

  bw_ = ((int)bn_bw * 125);
  del_ = bn_delay*1.0/1000;
  fprintf(filew,"m %d\n",topo.TCPNodes.NoNodes*2+3);
  fprintf(filew,"link 1  %.3lf %d %d 50 \n",del_,bw_,bw_);

  c=3;
  for(i=0;i<topo.TCPNodes.NoNodes;i++){
	  bw_ = ((int)(bw*125));
	  del_= topo.TCPNodes.TCPNode[i].delayTo*bn_delay/1000;
	  fprintf(filew,"link %d  %.3lf %d %d 50 \n", c+1, del_,bw_,bw_);
	  fprintf(filew,"link %d  %.3lf %d %d 50 \n", c+2, del_,bw_,bw_);
	  c=c+2;
  }
}


void TCL_Write_Agents(FILE *filew){
	int i;
	double time;
	
	int c = 3;
	fprintf(filew,"n %d \n",topo.TCPNodes.NoNodes+1);
	for(i=1; i<=topo.TCPNodes.NoNodes; i++){
		fprintf(filew,"route %d 3 %d 1 %d\n", i, c+1,c+2);
		c=c+2;
	}
}



void TCL_Write_DDOS_Agent(FILE* filew){

	int i;
	double ddosbw;	
	fprintf(filew,"# Ddos stuff\n");
	
	ddosbw =  topo.CBRTraffic.DDOSTraffic.Events[0].NetCBR;
	for(i=1;i<topo.CBRTraffic.DDOSTraffic.NoEvents;i++){
		ddosbw+=topo.CBRTraffic.DDOSTraffic.Events[i].NetCBR;
	}
	fprintf(filew,"route %d 3 2 1 3 cbr %d\n",topo.TCPNodes.NoNodes+1,
		((int)(ddosbw*125)));

}


void TCL_Write_Final(FILE *filew){
  fprintf(filew,"\n\n");
  fprintf(filew,"set outfile [open qtrace.tr w]\n$ns trace-queue $BotLeft $BotRight $outfile\n\n\n");
  fprintf(filew,"$ns at %lf \"finish\"\n",finish_time);
  fprintf(filew, "$ns run\n");
}