File: dosdbell.c

package info (click to toggle)
ns2 2.35%2Bdfsg-2.1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 78,780 kB
  • ctags: 27,490
  • sloc: cpp: 172,923; tcl: 107,130; perl: 6,391; sh: 6,143; ansic: 5,846; makefile: 816; awk: 525; csh: 355
file content (218 lines) | stat: -rw-r--r-- 7,617 bytes parent folder | download | duplicates (8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
//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, "set ns [new Simulator]\n\n\n");
  /* Incase we need nam support */
  //fprintf(filew, "set nf [open out.nam w]\n");
  //fprintf(filew, "$ns namtrace-all $nf\n\n\n");
  //fprintf(filew, "proc finish {} { \n  global ns nf \n  $ns flush-trace \n  close $nf \n  exec nam out.nam & \n  exit 0 \n  }\n\n\n");
  /***till here***/
  fprintf(filew, "proc finish {} {\n  exit 0 \n  }\n\n\n");
}


void TCL_Write_Nodes(FILE *filew){
  fprintf(filew, "for {set i 0} {$i < %d} {incr i} { \n  set TCPSend($i) [$ns node] \n  }\n\n",topo.TCPNodes.NoNodes);
  fprintf(filew, "for {set i 0} {$i < %d} {incr i} { \n  set TCPRecv($i) [$ns node] \n  }\n\n",topo.TCPNodes.NoNodes);

  
   fprintf(filew,"set BotLeft [$ns node] \n");
   fprintf(filew,"set BotRight [$ns node] \n");
   fprintf(filew,"set DDOSNode [$ns node] \n");
   fprintf(filew,"set Victim [$ns node] \n");

   fprintf(filew,"\n\n");
}


void TCL_Write_Connections(FILE *filew){
  int i;
  for(i=0;i<topo.TCPNodes.NoNodes;i++){
    fprintf(filew,"$ns duplex-link $TCPSend(%d) $BotLeft %.3lfMb %.3lfms DropTail\n",i,bw,topo.TCPNodes.TCPNode[i].delayTo*bn_delay); 
    fprintf(filew,"$ns duplex-link $TCPRecv(%d) $BotRight %.3lfMb %.3lfms DropTail\n",i,bw,topo.TCPNodes.TCPNode[i].delayFrom*bn_delay); 
  }

   fprintf(filew,"\n\n");
   fprintf(filew,"$ns duplex-link $BotLeft $BotRight %.3lfMb %.3lfms DropTail\n",bn_bw,bn_delay); 
   fprintf(filew,"$ns duplex-link $BotLeft $DDOSNode %.3lfMb %.3lfms DropTail\n",10*bn_bw,bn_delay);
   fprintf(filew,"$ns duplex-link $BotRight $Victim %.3lfMb %.3lfms DropTail\n",10*bn_bw,bn_delay);
   fprintf(filew,"\n\n");
}


void TCL_Write_Agents(FILE *filew){
  int i;
  double time;

  fprintf(filew, "for {set i 0} {$i < %d} {incr i} { \n    \
set TCPSendAg($i) [new Agent/TCP] \n  \
$ns attach-agent $TCPSend($i) $TCPSendAg($i) \n     \
set TCPSendTraffic($i) [new Application/FTP] \n     \
$TCPSendTraffic($i) attach-agent $TCPSendAg($i) \n   \
set TCPRecvAg($i) [new Agent/TCPSink] \n    \
$ns attach-agent $TCPRecv($i) $TCPRecvAg($i) \n   \
$ns connect $TCPSendAg($i) $TCPRecvAg($i) \n  \
$TCPSendAg($i) set class_ 1 \n}\n\n",topo.TCPNodes.NoNodes);

  fprintf(filew,"\n\n\n");

  for(i=0;i<topo.TCPNodes.NoNodes;i++){
    fprintf(filew,"$ns at %lf \"$TCPSendTraffic(%d) start\"\n",topo.TCPNodes.TCPNode[i].startTime,i);
  }
  fprintf(filew,"\n\n");
}



void TCL_Write_DDOS_Agent(FILE* filew){
  int i;

  fprintf(filew,"\n\n");
  fprintf(filew,"set DDOSUDP [new Agent/UDP]\n");
  fprintf(filew,"$ns attach-agent $DDOSNode $DDOSUDP\n");
  fprintf(filew,"set DDOSTraffic [new Application/Traffic/CBR]\n");
  fprintf(filew,"$DDOSTraffic attach-agent $DDOSUDP\n");
  fprintf(filew,"set VictimAgent [new Agent/Null]\n");
  fprintf(filew,"$ns attach-agent $Victim $VictimAgent\n");
  fprintf(filew,"$ns connect $DDOSUDP $VictimAgent\n");
  fprintf(filew,"$DDOSUDP set class_ 2\n");


  fprintf(filew,"\n\n\n");

  DDOS_start_time*=1000;
  fprintf(filew,"$ns at %lf \"$DDOSTraffic set rate_ %lfMb \"\n",ceil(DDOS_start_time+topo.CBRTraffic.DDOSTraffic.Events[0].delay*bn_delay)/1000, \
	                                   topo.CBRTraffic.DDOSTraffic.Events[0].NetCBR);
  fprintf(filew,"$ns at %lf \"$DDOSTraffic set packet_size_ %d \"\n",\
	                                   ceil(DDOS_start_time+topo.CBRTraffic.DDOSTraffic.Events[0].delay*bn_delay)/1000,pack_size);
  fprintf(filew,"$ns at %lf \"$DDOSTraffic start\"\n",ceil(DDOS_start_time+topo.CBRTraffic.DDOSTraffic.Events[0].delay*bn_delay)/1000);

  for(i=1;i<topo.CBRTraffic.DDOSTraffic.NoEvents;i++){
    fprintf(filew,"$ns at %lf \"$DDOSTraffic set rate_ %lfMb\"\n",\
	                                   ceil(DDOS_start_time+topo.CBRTraffic.DDOSTraffic.Events[i].delay*bn_delay)/1000, 
	                                   topo.CBRTraffic.DDOSTraffic.Events[i].NetCBR);
  }

  fprintf(filew,"\n\n");

}


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");
}