通讯编程

开发平台：
Visual C++

dosdbellasim.c：源码内容
							//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_filenamen",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 %fn",&diameter,&bn_bw,&bn_delay);
  fscanf(filer,"%d %lf %lfn",&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 %lfn",&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,"%lfn",&finish_time);
  printf("Topology Generated Using:n Diameter=%d BottleNeck: BW=%.3fMbps Delay=%.3fmsn 
Background Traffic: TCP Nodes=%d BW=%.3lfMbps Starttime=%.1lfsn 
Dos Traffic       : Attacker= %d BW=%.3lfMbps Starttime=%.1lfsn 
PacketSize=%dB  Master = %d  Finishtime = %.1lfsn", 
	 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 datann");
}
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 %dn",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 %dn", i, c+1,c+2);
		c=c+2;
	}
}
void TCL_Write_DDOS_Agent(FILE* filew){
	int i;
	double ddosbw;	
	fprintf(filew,"# Ddos stuffn");
	
	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 %dn",topo.TCPNodes.NoNodes+1,
		((int)(ddosbw*125)));
}
void TCL_Write_Final(FILE *filew){
  fprintf(filew,"nn");
  fprintf(filew,"set outfile [open qtrace.tr w]n$ns trace-queue $BotLeft $BotRight $outfilennn");
  fprintf(filew,"$ns at %lf "finish"n",finish_time);
  fprintf(filew, "$ns runn");
}

						