Change "bool drop" > "int dropRate"

gilgil1973 · Jul 30, 2014 · 5990b64 · 5990b64
1 parent 4401da8
commit 5990b64
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/include/process/snoopblock.cpp b/include/process/snoopblock.cpp
@@ -7,7 +7,7 @@ REGISTER_METACLASS(SnoopBlock, SnoopProcess)
 // ----------------------------------------------------------------------------
 SnoopBlock::SnoopBlock(void* owner) : SnoopProcess(owner)
 {
-  drop = true;
+  dropRate = 100;
 }
 
 SnoopBlock::~SnoopBlock()
@@ -17,36 +17,37 @@ SnoopBlock::~SnoopBlock()
 
 void SnoopBlock::block(SnoopPacket* packet)
 {
-  if (this->drop)
+  int r = rand() % 100; // 0 <= r <= 99
+  if (r < dropRate)
     packet->drop = true;
 }
 
 void SnoopBlock::load(VXml xml)
 {
   SnoopProcess::load(xml);
 
-  drop = xml.getBool("drop", drop);
+  dropRate = xml.getInt("dropRate", dropRate);
 }
 
 void SnoopBlock::save(VXml xml)
 {
   SnoopProcess::save(xml);
 
-  xml.setBool("drop", drop);
+  xml.setInt("dropRate", dropRate);
 }
 
 #ifdef QT_GUI_LIB
 void SnoopBlock::optionAddWidget(QLayout* layout)
 {
   SnoopProcess::optionAddWidget(layout);
 
-  VOptionable::addCheckBox(layout, "chkDrop", "Drop", drop);
+  VOptionable::addLineEdit(layout, "leDropRate", "Drop Rate(%)", QString::number(dropRate));
 }
 
 void SnoopBlock::optionSaveDlg(QDialog* dialog)
 {
   SnoopProcess::optionSaveDlg(dialog);
 
-  drop = dialog->findChild<QCheckBox*>("chkDrop")->checkState() == Qt::Checked;
+  dropRate = dialog->findChild<QLineEdit*>("leDropRate")->text().toInt();
 }
 #endif // QT_GUI_LIB
diff --git a/include/process/snoopblock.h b/include/process/snoopblock.h
@@ -25,7 +25,7 @@ class SnoopBlock : public SnoopProcess
   virtual ~SnoopBlock();
 
 public:
-  bool drop;
+  int dropRate; // 0%(all allow) ~ 100%(all drop)
 
 public slots:
   void block(SnoopPacket* packet);