projects/stargazer/plugins/capture/cap_debug/debug_cap.cpp

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18 Date: 18.09.2002
  19 */
  20
  21 /*
  22 * Author : Boris Mikhailenko <stg34@stg.dp.ua>
  23 */
  24
  25 /*
  26 $Revision: 1.21 $
  27 $Date: 2009/03/19 20:03:35 $
  28 $Author: faust $
  29 */
  30
  31 #include <stdio.h>
  32 #include <unistd.h>
  33 #include <signal.h>
  34 #include <sys/types.h>
  35 #include <unistd.h>
  36 #include <sys/socket.h>
  37 #include <netinet/in.h>
  38 #include <arpa/inet.h>
  39
  40 #include "libpal.h"
  41 #include "stg/plugin_creator.h"
  42 #include "stg/traffcounter.h"
  43 #include "debug_cap.h"
  44
  45 //-----------------------------------------------------------------------------
  46 void WriteStat(uint32_t u, uint32_t d)
  47 {
  48 FILE * f;
  49 f = fopen("/tmp/cap.stat", "at");
  50 fprintf(f, "up %5.2f Mbit, down %5.2f Mbit, sum %5.2f Mbit\n",
  51         u / (1000000*8.0),
  52         d / (1000000*8.0),
  53         (u + d) / (1000000*8.0));
  54 fclose(f);
  55 }
  56 //-----------------------------------------------------------------------------
  57 //-----------------------------------------------------------------------------
  58 //-----------------------------------------------------------------------------
  59 PLUGIN_CREATOR<DEBUG_CAP> cdc;
  60 //-----------------------------------------------------------------------------
  61 //-----------------------------------------------------------------------------
  62 //-----------------------------------------------------------------------------
  63 BASE_PLUGIN * GetPlugin()
  64 {
  65 return cdc.GetPlugin();
  66 }
  67 //-----------------------------------------------------------------------------
  68 //-----------------------------------------------------------------------------
  69 //-----------------------------------------------------------------------------
  70
  71 //-----------------------------------------------------------------------------
  72 RAW_PACKET MakeTCPPacket(const char * src,
  73                          const char * dst,
  74                          uint16_t sport,
  75                          uint16_t dport,
  76                          uint16_t len);
  77 const string DEBUG_CAP::GetVersion() const
  78 {
  79 return "Debug_cap v.0.01a";
  80 }
  81 //-----------------------------------------------------------------------------
  82 DEBUG_CAP::DEBUG_CAP()
  83 {
  84 isRunning = false;
  85 nonstop = false;
  86 }
  87 //-----------------------------------------------------------------------------
  88 void DEBUG_CAP::SetTraffcounter(TRAFFCOUNTER * tc)
  89 {
  90 traffCnt = tc;
  91 }
  92 //-----------------------------------------------------------------------------
  93 const string & DEBUG_CAP::GetStrError() const
  94 {
  95 return errorStr;
  96 }
  97 //-----------------------------------------------------------------------------
  98 int DEBUG_CAP::Start()
  99 {
 100 if (isRunning)
 101     return 0;
 102
 103 printfd(__FILE__, "DEBUG_CAP::Start()\n");
 104
 105 nonstop = true;
 106
 107 if (pthread_create(&thread, NULL, Run1, this) == 0)
 108     {
 109     return 0;
 110     }
 111
 112 errorStr = "Cannot create thread.";
 113 return -1;
 114 }
 115 //-----------------------------------------------------------------------------
 116 int DEBUG_CAP::Stop()
 117 {
 118 if (!isRunning)
 119     return 0;
 120
 121 nonstop = false;
 122
 123 //5 seconds to thread stops itself
 124 int i;
 125 for (i = 0; i < 25; i++)
 126     {
 127     if (!isRunning)
 128         break;
 129
 130     stgUsleep(200000);
 131     //printf(".");
 132     }
 133
 134 /*if (i)
 135     printf("\n");*/
 136
 137 //after 5 seconds waiting thread still running. now killing it
 138 if (isRunning)
 139     {
 140     //TODO pthread_cancel()
 141     if (pthread_kill(thread, SIGINT))
 142         {
 143         errorStr = "Cannot kill thread.";
 144         return -1;
 145         }
 146     }
 147
 148 return 0;
 149 }
 150 //-----------------------------------------------------------------------------
 151 bool DEBUG_CAP::IsRunning()
 152 {
 153 return nonstop;
 154 }
 155 //-----------------------------------------------------------------------------
 156 void * DEBUG_CAP::Run1(void * data)
 157 {
 158 printfd(__FILE__, "=====================| pid: %d |===================== \n", getpid());
 159
 160 DEBUG_CAP * dc = (DEBUG_CAP *)data;
 161 dc->isRunning = true;
 162
 163 RAW_PACKET rp;
 164 rp = MakeTCPPacket("192.168.1.1", "192.168.1.21", 255, 255, 200);
 165 int a = 0;
 166 sleep(3);
 167
 168 struct tm * tm;
 169 time_t t;
 170 uint32_t u = 0;
 171 uint32_t d = 0;
 172
 173 //2 upload : 3 download
 174
 175 int usize;
 176 int dsize;
 177
 178 t = stgTime;
 179 tm = localtime(&t);
 180 int min = tm->tm_min;
 181 int sec = tm->tm_sec;
 182
 183 char cliIP[20];
 184 char srvIP[20];
 185 char trashIP1[20];
 186 char trashIP2[20];
 187
 188 while (dc->nonstop)
 189     {
 190     for (int i = 8; i <= 252; i++)
 191         {
 192
 193         usize = random()%100 + 100;
 194                 dsize = random()%500 + 900;
 195
 196         for (int j = 2; j < 11; j++)
 197             {
 198                         sprintf(cliIP, "192.168.%d.%d", j, i);
 199             sprintf(srvIP, "10.1.%d.%d", random()%8, 1);
 200
 201             rp = MakeTCPPacket(srvIP, cliIP, 80, random()%2 + 2000, dsize);
 202             d += dsize;
 203             dc->traffCnt->Process(rp);
 204
 205             rp = MakeTCPPacket(cliIP, srvIP, random()%2 + 2000, 80, usize);
 206             u += usize;
 207             dc->traffCnt->Process(rp);
 208             }
 209         }
 210
 211     usleep(100000);
 212     t = stgTime;
 213
 214     if (min != localtime(&t)->tm_min)
 215         {
 216         min = localtime(&t)->tm_min;
 217         WriteStat(u, d);
 218         u = d = 0;
 219         }
 220
 221     a++;
 222
 223     }
 224
 225 dc->isRunning = false;
 226 return NULL;
 227 }
 228 //-----------------------------------------------------------------------------
 229 void * DEBUG_CAP::Run2(void * data)
 230 {
 231 printfd(__FILE__, "=====================| pid: %d |===================== \n", getpid());
 232
 233 DEBUG_CAP * dc = (DEBUG_CAP *)data;
 234 dc->isRunning = true;
 235
 236 RAW_PACKET rp;
 237 rp = MakeTCPPacket("192.168.1.1", "192.168.1.21", 255, 255, 200);
 238 int a = 0;
 239 sleep(3);
 240
 241 struct tm * tm;
 242 time_t t;
 243 uint32_t u = 0;
 244 uint32_t d = 0;
 245
 246 //2 upload : 3 download
 247
 248 int usize = 200;
 249 int dsize = 1500;
 250
 251 t = stgTime;
 252 tm = localtime(&t);
 253 int min = tm->tm_min;
 254
 255 char cliIP[20];
 256 char srvIP[20];
 257
 258 while (dc->nonstop)
 259     {
 260     for (int i = 101; i <= 150; i++)
 261         {
 262         sprintf(cliIP, "192.168.1.%d", i);
 263         for (int dp = 0; dp < 1; dp++)
 264             {
 265             //sprintf(srvIP, "10.1.%d.%d", i, 10 + dp);
 266             sprintf(srvIP, "10.1.%d.%d", i, 10 + dp);
 267
 268             rp = MakeTCPPacket(srvIP, cliIP, 80, 10000 + i + dp, dsize);
 269             d += dsize;
 270             dc->traffCnt->Process(rp);
 271
 272             rp = MakeTCPPacket(srvIP, cliIP, 80, 10000 + i + dp, dsize);
 273             d += dsize;
 274             dc->traffCnt->Process(rp);
 275
 276             rp = MakeTCPPacket(srvIP, cliIP, 80, 10000 + i + dp, dsize);
 277             dc->traffCnt->Process(rp);
 278             d += dsize;
 279
 280
 281             rp = MakeTCPPacket(cliIP, srvIP, 10000 + i + dp, 80, usize);
 282             u += usize;
 283             dc->traffCnt->Process(rp);
 284
 285             rp = MakeTCPPacket(cliIP, srvIP, 10000 + i + dp, 80, usize);
 286             u += usize;
 287             dc->traffCnt->Process(rp);
 288             }
 289         }
 290
 291     //usleep(20000);
 292     t = stgTime;
 293
 294     if (min != localtime(&t)->tm_min)
 295         {
 296         min = localtime(&t)->tm_min;
 297         WriteStat(u, d);
 298         u = d = 0;
 299         }
 300
 301     a++;
 302
 303     }
 304
 305 dc->isRunning = false;
 306 return NULL;
 307 }
 308 //-----------------------------------------------------------------------------
 309 void * DEBUG_CAP::Run3(void * data)
 310 {
 311 printfd(__FILE__, "=====================| pid: %d |===================== \n", getpid());
 312
 313 DEBUG_CAP * dc = (DEBUG_CAP *)data;
 314 dc->isRunning = true;
 315
 316 RAW_PACKET rp;
 317 rp = MakeTCPPacket("192.168.1.1", "192.168.1.21", 255, 255, 200);
 318 int a = 0;
 319 sleep(3);
 320
 321 struct tm * tm;
 322 time_t t;
 323 uint32_t u = 0;
 324 uint32_t d = 0;
 325
 326 //2 upload : 3 download
 327
 328 int usize = 200;
 329 int dsize = 1500;
 330
 331 t = stgTime;
 332 tm = localtime(&t);
 333
 334 char cliIP[20];
 335 char srvIP1[20];
 336 char srvIP2[20];
 337 char srvIP3[20];
 338
 339 int firstTime = true;
 340
 341 while (dc->nonstop)
 342     {
 343     if (firstTime)
 344         {
 345         sprintf(srvIP1, "10.1.%d.%d", random() % 14 + 153, random() % 11 + 35);
 346
 347         sprintf(srvIP2, "%d.%d.%d.%d",
 348                 random() % 20 + 81,
 349                 random() % 28 + 153,
 350                 random() % 28 + 37,
 351                 random() % 28 + 13);
 352
 353         sprintf(srvIP3, "%d.%d.%d.%d",
 354                 random() % 20 + 81,
 355                 random() % 28 + 153,
 356                 random() % 28 + 37,
 357                 random() % 28 + 13);
 358
 359         printfd(__FILE__, "firstTime=false\n");
 360         firstTime = false;
 361         }
 362
 363     int rnd = random() % 400;
 364     if (rnd < 5)
 365         {
 366         sprintf(srvIP1, "10.1.%d.%d", random() % 14 + 153, random() % 11 + 35);
 367         printfd(__FILE__, "srvIP1=%s\n", srvIP1);
 368         }
 369     if (rnd == 9)
 370         {
 371         sprintf(srvIP2, "%d.%d.%d.%d",
 372                 random() % 20 + 81,
 373                 random() % 28 + 153,
 374                 random() % 28 + 37,
 375                 random() % 28 + 13);
 376         printfd(__FILE__, "srvIP2=%s\n", srvIP2);
 377         }
 378     if (rnd == 5)
 379         {
 380         sprintf(srvIP2, "%d.%d.%d.%d",
 381                 random() % 20 + 81,
 382                 random() % 28 + 153,
 383                 random() % 28 + 37,
 384                 random() % 28 + 13);
 385         printfd(__FILE__, "srvIP3=%s\n", srvIP3);
 386         }
 387
 388     for (int i = 2; i < 52; i++)
 389         {
 390         sprintf(cliIP, "192.168.1.%d", i);
 391         for (int dp = 0; dp < 1; dp++)
 392             {
 393             usize = 50 + random() % 100;
 394             dsize = 1000 + random() % 400;
 395
 396             rp = MakeTCPPacket(srvIP1, cliIP, 80, 10000 + i + dp, dsize);
 397             dc->traffCnt->Process(rp);
 398
 399             rp = MakeTCPPacket(srvIP2, cliIP, 80, 10000 + i + dp, dsize);
 400             dc->traffCnt->Process(rp);
 401
 402             rp = MakeTCPPacket(srvIP3, cliIP, 80, 10000 + i + dp, dsize);
 403             dc->traffCnt->Process(rp);
 404
 405
 406             rp = MakeTCPPacket(cliIP, srvIP1, 10000 + i + dp, 80, usize);
 407             dc->traffCnt->Process(rp);
 408
 409             rp = MakeTCPPacket(cliIP, srvIP2, 10000 + i + dp, 80, usize);
 410             dc->traffCnt->Process(rp);
 411
 412             rp = MakeTCPPacket(cliIP, srvIP3, 10000 + i + dp, 80, usize);
 413             dc->traffCnt->Process(rp);
 414             }
 415         }
 416
 417     usleep(300000);
 418     /*t = stgTime;
 419
 420     if (min != localtime(&t)->tm_min)
 421         {
 422         min = localtime(&t)->tm_min;
 423         WriteStat(u, d);
 424         u = d = 0;
 425         }*/
 426
 427     a++;
 428
 429     }
 430
 431 dc->isRunning = false;
 432 return NULL;
 433 }
 434 //-----------------------------------------------------------------------------
 435 uint16_t DEBUG_CAP::GetStartPosition() const
 436 {
 437 return 0;
 438 }
 439 //-----------------------------------------------------------------------------
 440 uint16_t DEBUG_CAP::GetStopPosition() const
 441 {
 442 return 0;
 443 }
 444 //-----------------------------------------------------------------------------
 445 RAW_PACKET MakeTCPPacket(const char * src,
 446                          const char * dst,
 447                          uint16_t sport,
 448                          uint16_t dport,
 449                          uint16_t len)
 450 {
 451 struct packet pkt;
 452 if (pkt_init(&pkt, 0, 100))
 453     {
 454     printfd(__FILE__, "pkt_init error!\n");
 455     }
 456
 457 in_addr_t sip = inet_addr(src);
 458 in_addr_t dip = inet_addr(dst);
 459
 460 pkt_ip_header(&pkt,
 461               5,        //header len
 462               4,
 463               0,
 464               len,      //total len
 465               0,
 466               0,
 467               64,        //ttl
 468               6,         //TCP
 469               0,
 470               *(uint32_t*)&sip,
 471               *(uint32_t*)&dip);
 472
 473 pkt_move_actptr(&pkt, 20);
 474
 475 pkt_tcp_header(&pkt,
 476                sport,
 477                dport,
 478                1,           // seq,
 479                1,           // ackseq,
 480                5,           // headerlen,
 481                0,           // reserved,
 482                0,           // flags,
 483                0,           // window,
 484                0,           // checksum,
 485                0);          // urgent
 486
 487 RAW_PACKET rp;
 488 memcpy(&rp, pkt.pkt, sizeof(rp));
 489
 490 if (pkt_free(&pkt))
 491     {
 492     printfd(__FILE__, "pkt_free error!\n");
 493     }
 494 rp.dataLen = -1;
 495 strcpy(rp.iface, "eth0");
 496 return rp;
 497 }
 498 //-----------------------------------------------------------------------------
 499
 500