diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/ClassifierCapability.java b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/ClassifierCapability.java
index 527f3a0..beba81b 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/ClassifierCapability.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/ClassifierCapability.java
@@ -79,13 +79,8 @@ public class ClassifierCapability extends RouterAgentCapability implements
belief(new Threat(flow), true);
}
- //no more reports from classifier then we are done with this link
- if(malicious.size() == 0){
- addBelief(new BenignBelief(new Benign(ip)));
- }
- else{
- setEndState(EndState.FAILED);
- }
+ addBelief(new BenignBelief(new Benign(ip)));
+ setEndState(EndState.SUCCESSFUL);
} else {
setEndState(EndState.FAILED);
}
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java b/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java
index 7bffaeb..ea7db41 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java
@@ -137,9 +137,13 @@ public class BDINetRApp {
NETWORK.addRouter(new Router(new Ip("10.0.0.17"), 5552, "CLASSIFIER", Role.CLASSIFIER.getId(), factory));
NETWORK.addRouter(new Router(new Ip("10.0.0.17"), 5553, "FLOW_EXPORTER", Role.FLOW_EXPORTER.getId(), factory));
- log.info("Total : " + args[0]);
+ int total = 30;
- for(int i = 0; i < Integer.parseInt(args[0]); i++){
+ if(args.length > 0){
+ total = Integer.parseInt(args[0]);
+ }
+
+ for(int i = 0; i < total; i++){
NETWORK.addRouter(new Router(new Ip("10.0.0.17"), 5550, "RATE_LIMITER " + i, Role.RATE_LIMITER.getId(), factory));
NETWORK.addRouter(new Router(new Ip("10.0.0.17"), 5551, "ANOMALY_DETECTION " + i, Role.ANOMALY_DETECTION.getId(), factory));
NETWORK.addRouter(new Router(new Ip("10.0.0.17"), 5552, "CLASSIFIER " + i, Role.CLASSIFIER.getId(), factory));
