Details
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/AnomalyDetectionCapability.java b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/AnomalyDetectionCapability.java
index a894596..c2ce25f 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/AnomalyDetectionCapability.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/AnomalyDetectionCapability.java
@@ -21,7 +21,6 @@
//----------------------------------------------------------------------------
package br.ufrgs.inf.bdinetr.agent;
-import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
@@ -231,7 +230,7 @@ public class AnomalyDetectionCapability extends RouterAgentCapability implements
@Override
public void generateGoals(GoalUpdateSet goalUpdateSet) {
// Anomalous(ip) AND not Restricted(ip) --> goal(Restricted(ip)) AND
- // goal(?Benign(ip)
+ // goal(?belief(Anomalous(ip))
Set<Belief<?, ?>> anomalousIpBeliefs = getBeliefBase()
.getBeliefsByType(Anomalous.class);
for (Belief<?, ?> belief : anomalousIpBeliefs) {
@@ -275,10 +274,8 @@ public class AnomalyDetectionCapability extends RouterAgentCapability implements
}
@Override
- protected Set<Class<?>> getLowPriorityGoal() {
- Set<Class<?>> lowPriorityGoals = new HashSet<>();
- lowPriorityGoals.add(Benign.class);
- return lowPriorityGoals;
+ protected Class<?> getLowPriorityGoal() {
+ return Benign.class;
}
@Override
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 f694ff6..b215b89 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/ClassifierCapability.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/ClassifierCapability.java
@@ -158,10 +158,10 @@ public class ClassifierCapability extends RouterAgentCapability implements
GoalTemplateFactory.goalOfType(FlowExport.class), this,
Role.FLOW_EXPORTER, false);
- this.analyseIpFlows = new DefaultPlan(
+ analyseIpFlows = new DefaultPlan(
GoalTemplateFactory.hasBeliefOfType(Benign.class),
AnalyseIPFlows.class);
- this.limitFlowRate = new DefaultPlan(
+ limitFlowRate = new DefaultPlan(
GoalTemplateFactory.hasNoBeliefOfType(Threat.class),
LimitFlowRatePlan.class) {
public boolean isContextApplicable(Goal goal) {
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/LinkMonitorCapability.java b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/LinkMonitorCapability.java
index 2f343c4..42b0be1 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/LinkMonitorCapability.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/LinkMonitorCapability.java
@@ -21,7 +21,6 @@
//----------------------------------------------------------------------------
package br.ufrgs.inf.bdinetr.agent;
-import java.util.HashSet;
import java.util.Observable;
import java.util.Observer;
import java.util.Set;
@@ -55,8 +54,9 @@ import br.ufrgs.inf.bdinetr.domain.predicate.OverUsage;
/**
* @author Ingrid Nunes
*/
-public class LinkMonitorCapability extends RouterAgentCapability
- implements BeliefRevisionStrategy, OptionGenerationFunction, GoalListener, Observer {
+public class LinkMonitorCapability extends RouterAgentCapability implements
+ BeliefRevisionStrategy, OptionGenerationFunction, GoalListener,
+ Observer {
public class LimitLinkRate extends BeliefGoalPlanBody {
private static final long serialVersionUID = -3493377510830902961L;
@@ -67,13 +67,15 @@ public class LinkMonitorCapability extends RouterAgentCapability
@Override
public void execute() {
if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new BeliefValueGoal<>(new LinkRateLimited(link), true));
+ dispatchSubgoalAndListen(new BeliefValueGoal<>(
+ new LinkRateLimited(link), true));
this.subgoalDispatched = true;
} else {
GoalEvent event = getGoalEvent();
if (event != null) {
if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
+ addBelief(((BeliefGoal<?>) event.getGoal())
+ .getOutputBelief());
belief(new AttackPrevented(link), true);
} else {
setEndState(EndState.FAILED);
@@ -102,13 +104,15 @@ public class LinkMonitorCapability extends RouterAgentCapability
@Override
public void execute() {
if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new BeliefValueGoal<>(new LinkRateLimited(link), false));
+ dispatchSubgoalAndListen(new BeliefValueGoal<>(
+ new LinkRateLimited(link), false));
this.subgoalDispatched = true;
} else {
GoalEvent event = getGoalEvent();
if (event != null) {
if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
+ addBelief(((BeliefGoal<?>) event.getGoal())
+ .getOutputBelief());
belief(new AttackPrevented(link), false);
belief(new AnomalousUsage(link), null);
} else {
@@ -138,31 +142,39 @@ public class LinkMonitorCapability extends RouterAgentCapability
@bdi4jade.annotation.TransientBelief
private LinkMonitor role;
- public LinkMonitorCapability(LinkMonitor linkMonitor, GoalRequestPlan beliefGoalRequestPlan) {
+ public LinkMonitorCapability(LinkMonitor linkMonitor,
+ GoalRequestPlan beliefGoalRequestPlan) {
this.role = linkMonitor;
role.addObserver(this);
setBeliefRevisionStrategy(this);
setOptionGenerationFunction(this);
- beliefGoalRequestPlan.addGoalTemplate(GoalTemplateFactory.hasBeliefOfTypeWithValue(LinkRateLimited.class, true),
- this, Role.RATE_LIMITER, false);
+ beliefGoalRequestPlan.addGoalTemplate(GoalTemplateFactory
+ .hasBeliefOfTypeWithValue(LinkRateLimited.class, true), this,
+ Role.RATE_LIMITER, false);
+ beliefGoalRequestPlan.addGoalTemplate(GoalTemplateFactory
+ .hasBeliefOfTypeWithValue(LinkRateLimited.class, false), this,
+ Role.RATE_LIMITER, false);
beliefGoalRequestPlan.addGoalTemplate(
- GoalTemplateFactory.hasBeliefOfTypeWithValue(LinkRateLimited.class, false), this, Role.RATE_LIMITER,
- false);
- beliefGoalRequestPlan.addGoalTemplate(GoalTemplateFactory.hasBeliefOfType(AnomalousUsage.class), this,
- Role.ANOMALY_DETECTION, true);
+ GoalTemplateFactory.hasBeliefOfType(AnomalousUsage.class),
+ this, Role.ANOMALY_DETECTION, true);
this.limitLinkRate = new DefaultPlan(
- GoalTemplateFactory.hasBeliefOfTypeWithValue(AttackPrevented.class, Boolean.TRUE), LimitLinkRate.class);
+ GoalTemplateFactory.hasBeliefOfTypeWithValue(
+ AttackPrevented.class, Boolean.TRUE),
+ LimitLinkRate.class);
this.restoreLinkRate = new DefaultPlan(
- GoalTemplateFactory.hasBeliefOfTypeWithValue(AttackPrevented.class, Boolean.FALSE),
+ GoalTemplateFactory.hasBeliefOfTypeWithValue(
+ AttackPrevented.class, Boolean.FALSE),
RestoreLinkRate.class) {
@Override
public boolean isContextApplicable(Goal goal) {
BeliefGoal<AttackPrevented> bg = (BeliefGoal<AttackPrevented>) goal;
PredicateBelief<LinkRateLimited> linkRateLimited = (PredicateBelief<LinkRateLimited>) getBeliefBase()
- .getBelief(new LinkRateLimited(bg.getBeliefName().getVariable()));
+ .getBelief(
+ new LinkRateLimited(bg.getBeliefName()
+ .getVariable()));
return (linkRateLimited != null && linkRateLimited.getValue());
}
};
@@ -170,7 +182,8 @@ public class LinkMonitorCapability extends RouterAgentCapability
@Override
public void generateGoals(GoalUpdateSet goalUpdateSet) {
- Set<Belief<?, ?>> overUsageBeliefs = getBeliefBase().getBeliefsByType(OverUsage.class);
+ Set<Belief<?, ?>> overUsageBeliefs = getBeliefBase().getBeliefsByType(
+ OverUsage.class);
for (Belief<?, ?> belief : overUsageBeliefs) {
PredicateBelief<OverUsage> overUsage = (PredicateBelief<OverUsage>) belief;
Link link = overUsage.getName().getVariable();
@@ -186,7 +199,8 @@ public class LinkMonitorCapability extends RouterAgentCapability
PredicateBelief<AttackPrevented> attackPrevented = (PredicateBelief<AttackPrevented>) getBeliefBase()
.getBelief(new AttackPrevented(link));
if ((anomalousUsage == null || anomalousUsage.getValue())
- && (attackPrevented == null || !attackPrevented.getValue())) {
+ && (attackPrevented == null || !attackPrevented
+ .getValue())) {
// OverUsage(l) AND !(not AnomalousUsage(l)) AND
// !(AttackPrevented(l)) --> AttackPrevented(l)
goal(goalUpdateSet, new AttackPrevented(link), Boolean.TRUE);
@@ -194,26 +208,28 @@ public class LinkMonitorCapability extends RouterAgentCapability
}
}
- Set<Belief<?, ?>> attackPreventedBeliefs = getBeliefBase().getBeliefsByType(AttackPrevented.class);
+ Set<Belief<?, ?>> attackPreventedBeliefs = getBeliefBase()
+ .getBeliefsByType(AttackPrevented.class);
for (Belief<?, ?> belief : attackPreventedBeliefs) {
PredicateBelief<AttackPrevented> attackPrevented = (PredicateBelief<AttackPrevented>) belief;
if (attackPrevented.getValue()) {
PredicateBelief<AnomalousUsage> anomalousUsage = (PredicateBelief<AnomalousUsage>) getBeliefBase()
- .getBelief(new AnomalousUsage(attackPrevented.getName().getVariable()));
+ .getBelief(
+ new AnomalousUsage(attackPrevented.getName()
+ .getVariable()));
// AttackPrevented(l) AND not AnomalousUsage(l) --> not
// AttackPrevented(l)
if (anomalousUsage != null && !anomalousUsage.getValue()) {
- goal(goalUpdateSet, attackPrevented.getName(), Boolean.FALSE);
+ goal(goalUpdateSet, attackPrevented.getName(),
+ Boolean.FALSE);
}
}
}
}
@Override
- protected Set<Class<?>> getLowPriorityGoal() {
- Set<Class<?>> lowPriorityGoals = new HashSet<>();
- lowPriorityGoals.add(AnomalousUsage.class);
- return lowPriorityGoals;
+ protected Class<?> getLowPriorityGoal() {
+ return AnomalousUsage.class;
}
@Override
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/RouterAgentCapability.java b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/RouterAgentCapability.java
index 27ae426..7c341b1 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/agent/RouterAgentCapability.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/agent/RouterAgentCapability.java
@@ -101,7 +101,7 @@ public abstract class RouterAgentCapability extends Capability implements Delibe
}
}
- protected Set<Class<?>> getLowPriorityGoal() {
+ protected Class<?> getLowPriorityGoal() {
return null;
}
@@ -144,11 +144,7 @@ public abstract class RouterAgentCapability extends Capability implements Delibe
if (GoalStatus.WAITING.equals(goalDesc.getStatus())) {
if (goalDesc.getGoal() instanceof BeliefGoal) {
BeliefGoal<?> bg = (BeliefGoal<?>) goalDesc.getGoal();
- for (Class<?> goalClass : getLowPriorityGoal()) {
- if (bg.getBeliefName().getClass().equals(goalClass)) {
- return true;
- }
- }
+ return bg.getBeliefName().getClass().equals(getLowPriorityGoal());
}
}
return false;
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java b/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java
index 5d71547..6852e29 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/BDINetRApp.java
@@ -21,6 +21,18 @@
//----------------------------------------------------------------------------
package br.ufrgs.inf.bdinetr;
+import jade.BootProfileImpl;
+import jade.core.Profile;
+import jade.core.ProfileImpl;
+import jade.core.Specifier;
+import jade.core.event.NotificationService;
+import jade.core.messaging.TopicManagementHelper;
+import jade.core.messaging.TopicManagementService;
+import jade.tools.rma.rma;
+import jade.wrapper.AgentContainer;
+import jade.wrapper.AgentController;
+import jade.wrapper.PlatformController;
+
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
@@ -48,17 +60,6 @@ import br.ufrgs.inf.bdinetr.domain.Role;
import br.ufrgs.inf.bdinetr.domain.Router;
import br.ufrgs.inf.bdinetr.domain.dummy.DummyRouterComponentFactory;
import br.ufrgs.inf.bdinetr.domain.omnet.OMNeTRouterComponentFactory;
-import jade.BootProfileImpl;
-import jade.core.Profile;
-import jade.core.ProfileImpl;
-import jade.core.Specifier;
-import jade.core.event.NotificationService;
-import jade.core.messaging.TopicManagementHelper;
-import jade.core.messaging.TopicManagementService;
-import jade.tools.rma.rma;
-import jade.wrapper.AgentContainer;
-import jade.wrapper.AgentController;
-import jade.wrapper.PlatformController;
/**
* @author Ingrid Nunes
@@ -66,12 +67,12 @@ import jade.wrapper.PlatformController;
public class BDINetRApp {
enum SimulationType {
- OMNeT, SINGLE_AGENT, SIMPLE_NETWORK, COMPLEX_NETWORK, PALLIATIVE_AGENT
+ OMNeT, SINGLE_AGENT, SIMPLE_NETWORK, COMPLEX_NETWORK
};
private static final Set<Link> AFFECTED_LINKS;
private static final Network NETWORK;
- private static final SimulationType SIMULATION_TYPE = SimulationType.PALLIATIVE_AGENT;
+ private static final SimulationType SIMULATION_TYPE = SimulationType.SINGLE_AGENT;
static {
PropertyConfigurator.configure(BDINetRApp.class
@@ -135,14 +136,6 @@ public class BDINetRApp {
AFFECTED_LINKS.add(new Link("AFFECTED_LINK"));
break;
- case PALLIATIVE_AGENT:
- factory = new DummyRouterComponentFactory();
- NETWORK.addRouterPalliative(new Router(new Ip("PALLIATIVE_AGENT"), Role.LINK_MONITOR
- .getId() | Role.RATE_LIMITER.getId() | Role.ANOMALY_DETECTION.getId() | Role.CLASSIFIER
- .getId() | Role.FLOW_EXPORTER
- .getId(), factory));
- AFFECTED_LINKS.add(new Link("AFFECTED_LINK"));
- break;
}
Set<Router> routers = NETWORK.getRouters();
@@ -242,7 +235,8 @@ public class BDINetRApp {
for (Link link : AFFECTED_LINKS) {
for (Router router : NETWORK.getRouters()) {
if (router.hasRole(Role.LINK_MONITOR)) {
- LinkMonitor lm = (LinkMonitor) router.getRole(Role.LINK_MONITOR);
+ LinkMonitor lm = (LinkMonitor) router
+ .getRole(Role.LINK_MONITOR);
lm.setOverUsage(link, true);
}
}
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyAnomalyDetection.java b/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyAnomalyDetection.java
index af85356..bb1dfb5 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyAnomalyDetection.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyAnomalyDetection.java
@@ -32,31 +32,25 @@ import br.ufrgs.inf.bdinetr.domain.Router;
/**
* @author Ingrid Nunes
*/
-public class DummyAnomalyDetection extends AbstractRouterComponent implements AnomalyDetection {
+public class DummyAnomalyDetection extends AbstractRouterComponent implements
+ AnomalyDetection {
private boolean first;
- private boolean mode;
public DummyAnomalyDetection(Router router) {
super(router);
this.first = true;
- this.mode = true;
}
@Override
public Set<Ip> detectIntrusion(Link link) {
Set<Ip> intrusions = new HashSet<>();
if (link.getId().equals("AFFECTED_LINK")) {
- if (mode) {
+ if (first) {
intrusions.add(new Ip("victim1"));
- intrusions.add(new Ip("victim2"));
+ first = false;
} else {
- if (first) {
- intrusions.add(new Ip("victim1"));
- first = false;
- } else {
- intrusions.add(new Ip("victim2"));
- }
+ intrusions.add(new Ip("victim2"));
}
}
return intrusions;
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyLinkMonitor.java b/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyLinkMonitor.java
index 50da416..afb3afc 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyLinkMonitor.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/domain/dummy/DummyLinkMonitor.java
@@ -44,7 +44,7 @@ public class DummyLinkMonitor extends AbstractRouterComponent implements
public DummyLinkMonitor(Router router) {
super(router);
this.overUsageLinks = new HashMap<>();
- this.first = false;
+ this.first = true;
}
@Override
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/Network.java b/network-resilience/src/br/ufrgs/inf/bdinetr/Network.java
index 16ac1f9..7a215d5 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/Network.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/Network.java
@@ -28,15 +28,13 @@ import java.util.Set;
import br.ufrgs.inf.bdinetr.agent.RouterAgent;
import br.ufrgs.inf.bdinetr.domain.Router;
-import br.ufrgs.inf.bdinetr.palliative.BDI2DoSAgent;
-import jade.core.Agent;
/**
* @author Ingrid Nunes
*/
public class Network {
- private final Map<Router, Agent> routerAgents;
+ private final Map<Router, RouterAgent> routerAgents;
public Network() {
this.routerAgents = new HashMap<>();
@@ -46,15 +44,11 @@ public class Network {
this.routerAgents.put(router, new RouterAgent(router));
}
- public void addRouterPalliative(Router router) {
- this.routerAgents.put(router, new BDI2DoSAgent(router));
- }
-
- public Agent getAgent(Router router) {
+ public RouterAgent getAgent(Router router) {
return routerAgents.get(router);
}
- public Collection<Agent> getRouterAgents() {
+ public Collection<RouterAgent> getRouterAgents() {
return routerAgents.values();
}
diff --git a/network-resilience/src/br/ufrgs/inf/bdinetr/palliative/BDI2DoSCapability.java b/network-resilience/src/br/ufrgs/inf/bdinetr/palliative/BDI2DoSCapability.java
index a04206f..adcd1bf 100644
--- a/network-resilience/src/br/ufrgs/inf/bdinetr/palliative/BDI2DoSCapability.java
+++ b/network-resilience/src/br/ufrgs/inf/bdinetr/palliative/BDI2DoSCapability.java
@@ -1,346 +1,19 @@
package br.ufrgs.inf.bdinetr.palliative;
-import java.util.HashSet;
-import java.util.Iterator;
-import java.util.Observable;
-import java.util.Observer;
-import java.util.Set;
-
-import bdi4jade.annotation.Parameter;
-import bdi4jade.annotation.Parameter.Direction;
import bdi4jade.belief.Belief;
-import bdi4jade.belief.BeliefSet;
-import bdi4jade.belief.PredicateBelief;
import bdi4jade.belief.TransientBelief;
-import bdi4jade.belief.TransientBeliefSet;
-import bdi4jade.core.GoalUpdateSet;
-import bdi4jade.event.GoalEvent;
-import bdi4jade.event.GoalListener;
-import bdi4jade.extension.palliative.PlanRequiredResource;
-import bdi4jade.extension.palliative.Resource;
-import bdi4jade.extension.palliative.ResourcePreferences;
-import bdi4jade.extension.palliative.goal.ConstrainedGoal;
-import bdi4jade.extension.palliative.goal.ObjectiveFunction;
-import bdi4jade.extension.palliative.graph.AlternativeCauseSet;
-import bdi4jade.extension.palliative.graph.CauseEffectKnowledgeModel;
-import bdi4jade.extension.palliative.graph.CauseEffectRelationship;
-import bdi4jade.extension.palliative.logics.Fact;
-import bdi4jade.extension.palliative.logics.MathExpression;
-import bdi4jade.extension.palliative.logics.MathExpression.Operator;
-import bdi4jade.extension.palliative.logics.UnaryPredicate;
-import bdi4jade.extension.palliative.reasoning.PalliativeOptionGenerationFunction;
-import bdi4jade.extension.palliative.reasoning.PalliativePlanSelectionStrategy;
-import bdi4jade.goal.BeliefGoal;
-import bdi4jade.goal.BeliefValueGoal;
-import bdi4jade.goal.Goal;
-import bdi4jade.goal.GoalStatus;
-import bdi4jade.goal.GoalTemplate;
-import bdi4jade.goal.GoalTemplateFactory;
-import bdi4jade.goal.PredicateGoal;
-import bdi4jade.plan.DefaultPlan;
-import bdi4jade.plan.Plan;
-import bdi4jade.plan.Plan.EndState;
-import bdi4jade.plan.planbody.BeliefGoalPlanBody;
-import bdi4jade.reasoning.BeliefRevisionStrategy;
+import bdi4jade.extension.palliative.PalliativeCapability;
import br.ufrgs.inf.bdinetr.agent.FlowExporterCapability;
import br.ufrgs.inf.bdinetr.agent.RateLimiterCapability;
-import br.ufrgs.inf.bdinetr.agent.RouterAgentCapability;
-import br.ufrgs.inf.bdinetr.agent.belief.AnomalousUsageBelief;
-import br.ufrgs.inf.bdinetr.agent.belief.BenignBelief;
import br.ufrgs.inf.bdinetr.domain.AnomalyDetection;
import br.ufrgs.inf.bdinetr.domain.Classifier;
-import br.ufrgs.inf.bdinetr.domain.Flow;
import br.ufrgs.inf.bdinetr.domain.FlowExporter;
-import br.ufrgs.inf.bdinetr.domain.Ip;
-import br.ufrgs.inf.bdinetr.domain.Link;
import br.ufrgs.inf.bdinetr.domain.LinkMonitor;
import br.ufrgs.inf.bdinetr.domain.RateLimiter;
import br.ufrgs.inf.bdinetr.domain.Role;
import br.ufrgs.inf.bdinetr.domain.Router;
-import br.ufrgs.inf.bdinetr.domain.predicate.Anomalous;
-import br.ufrgs.inf.bdinetr.domain.predicate.AnomalousUsage;
-import br.ufrgs.inf.bdinetr.domain.predicate.AttackPrevented;
-import br.ufrgs.inf.bdinetr.domain.predicate.Benign;
-import br.ufrgs.inf.bdinetr.domain.predicate.FlowExport;
-import br.ufrgs.inf.bdinetr.domain.predicate.FlowRateLimited;
-import br.ufrgs.inf.bdinetr.domain.predicate.IpRateLimited;
-import br.ufrgs.inf.bdinetr.domain.predicate.LinkRateLimited;
-import br.ufrgs.inf.bdinetr.domain.predicate.OverUsage;
-import br.ufrgs.inf.bdinetr.domain.predicate.OverUsageCause;
-import br.ufrgs.inf.bdinetr.domain.predicate.Restricted;
-import br.ufrgs.inf.bdinetr.domain.predicate.Threat;
-
-public class BDI2DoSCapability extends RouterAgentCapability implements BeliefRevisionStrategy, GoalListener, Observer {
-
- public class OverUsageGoal extends ConstrainedGoal {
-
- public OverUsageGoal(Link link, Boolean value) {
- super(new PredicateGoal<OverUsage>(new OverUsage(link), false));
-
- this.addOperationConstraint(new MathExpression<Resource>(Resources.VULNERABILITY, Operator.LESS_THAN, 0.5));
-
- this.addObjectiveFunction(Resources.TIME, ObjectiveFunction.MINIMIZE);
- this.addObjectiveFunction(Resources.NETWORK_AVAILABILITY, ObjectiveFunction.MAXIMIZE);
- this.addObjectiveFunction(Resources.VULNERABILITY, ObjectiveFunction.MINIMIZE);
- }
-
- }
-
- public class LimitLinkRate extends BeliefGoalPlanBody {
- private static final long serialVersionUID = -3493377510830902961L;
-
- private Link link;
- private boolean subgoalDispatched;
-
- @Override
- public void execute() {
- if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new BeliefValueGoal<>(new LinkRateLimited(link), true));
- this.subgoalDispatched = true;
- } else {
- GoalEvent event = getGoalEvent();
- if (event != null) {
- if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
- belief(new AttackPrevented(link), true);
- } else {
- setEndState(EndState.FAILED);
- }
- }
- }
- }
-
- @Override
- protected void init() {
- this.subgoalDispatched = false;
- }
-
- @Parameter(direction = Direction.IN)
- public void setBeliefName(UnaryPredicate<Link> attackPrevented) {
- this.link = attackPrevented.getVariable();
- }
- }
-
- public class RestoreLinkRate extends BeliefGoalPlanBody {
- private static final long serialVersionUID = -3493377510830902961L;
-
- private Link link;
- private boolean subgoalDispatched;
-
- @Override
- public void execute() {
- if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new BeliefValueGoal<>(new LinkRateLimited(link), false));
- this.subgoalDispatched = true;
- } else {
- GoalEvent event = getGoalEvent();
- if (event != null) {
- if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
- belief(new AttackPrevented(link), false);
- belief(new AnomalousUsage(link), null);
- } else {
- setEndState(EndState.FAILED);
- }
- }
- }
- }
-
- @Override
- protected void init() {
- this.subgoalDispatched = false;
- }
-
- @Parameter(direction = Direction.IN)
- public void setBeliefName(AttackPrevented attackPrevented) {
- this.link = attackPrevented.getVariable();
- }
- }
-
- public class AnalyseLinkStatistics extends BeliefGoalPlanBody {
- private static final long serialVersionUID = -3493377510830902961L;
-
- private Link link;
-
- @Override
- public void execute() {
- Set<Ip> outliers = getAnomalyDetection().detectIntrusion(link);
- BeliefSet<OverUsageCause, Ip> overUsageCause = new TransientBeliefSet<OverUsageCause, Ip>(
- new OverUsageCause(link));
- if (outliers != null && !outliers.isEmpty()) {
- for (Ip outlier : outliers) {
- belief(new Anomalous(outlier), true);
- belief(new Benign(outlier), null);
- overUsageCause.addValue(outlier);
- }
- }
- addBelief(overUsageCause);
- addBelief(new AnomalousUsageBelief(new AnomalousUsage(link)));
- }
-
- @Parameter(direction = Direction.IN)
- public void setBeliefName(AnomalousUsage anomalousUsage) {
- this.link = anomalousUsage.getVariable();
- }
- }
-
- public class LimitIPRatePlan extends BeliefGoalPlanBody {
- private static final long serialVersionUID = -3493377510830902961L;
-
- private Ip ip;
- private boolean subgoalDispatched;
-
- @Override
- public void execute() {
- if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new BeliefValueGoal<>(new IpRateLimited(ip), true));
- this.subgoalDispatched = true;
- } else {
- GoalEvent event = getGoalEvent();
- if (event != null) {
- if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
- belief(new Restricted(ip), true);
-
- Set<Belief<?, ?>> overUsageCauseBeliefs = getBeliefBase()
- .getBeliefsByType(OverUsageCause.class);
- Iterator<Belief<?, ?>> it = overUsageCauseBeliefs.iterator();
- while (it.hasNext()) {
- BeliefSet<OverUsageCause, Ip> overUsageCause = (BeliefSet<OverUsageCause, Ip>) it.next();
- if (overUsageCause.hasValue(ip)) {
- // ip in OverUsageCause(l) --> remove
- overUsageCause.removeValue(ip);
- }
- }
- belief(new Anomalous(ip), false);
- } else {
- setEndState(EndState.FAILED);
- }
- }
- }
- }
-
- @Override
- protected void init() {
- this.subgoalDispatched = false;
- }
-
- @Parameter(direction = Direction.IN)
- public void setBeliefName(UnaryPredicate<Ip> restricted) {
- this.ip = restricted.getVariable();
- }
- }
-
- public class RestoreIPRatePlan extends BeliefGoalPlanBody {
- private static final long serialVersionUID = -3493377510830902961L;
-
- private Ip ip;
- private boolean subgoalDispatched;
-
- @Override
- public void execute() {
- if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new BeliefValueGoal<>(new IpRateLimited(ip), false));
- this.subgoalDispatched = true;
- } else {
- GoalEvent event = getGoalEvent();
- if (event != null) {
- if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
- belief(new Restricted(ip), false);
- belief(new Anomalous(ip), null);
- } else {
- setEndState(EndState.FAILED);
- }
- }
- }
- }
-
- @Override
- protected void init() {
- this.subgoalDispatched = false;
- }
-
- @Parameter(direction = Direction.IN)
- public void setBeliefName(Restricted restricted) {
- this.ip = restricted.getVariable();
- }
- }
-
- public class AnalyseIPFlows extends BeliefGoalPlanBody {
- private static final long serialVersionUID = -3493377510830902961L;
-
- private Ip ip;
- private boolean subgoalDispatched;
-
- @Override
- public void execute() {
- if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new FlowExport(ip));
- this.subgoalDispatched = true;
- } else {
- GoalEvent event = getGoalEvent();
- if (event != null) {
- if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- Set<Flow> malicious = getClassifier().classifyFlows(ip);
- for (Flow flow : malicious) {
- belief(new Threat(flow), true);
- }
- addBelief(new BenignBelief(new Benign(ip)));
- } else {
- setEndState(EndState.FAILED);
- }
- } else {
- block();
- }
- }
- }
-
- @Override
- public void init() {
- this.subgoalDispatched = false;
- }
-
- @Parameter(direction = Direction.IN)
- public void setBeliefName(Benign benign) {
- this.ip = benign.getVariable();
- }
- }
- public class LimitFlowRatePlan extends BeliefGoalPlanBody {
- private static final long serialVersionUID = -3493377510830902961L;
-
- private Flow flow;
- private boolean subgoalDispatched;
-
- @Override
- public void execute() {
- if (!subgoalDispatched) {
- dispatchSubgoalAndListen(new BeliefValueGoal<>(new FlowRateLimited(flow), true));
- this.subgoalDispatched = true;
- } else {
- GoalEvent event = getGoalEvent();
- if (event != null) {
- if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
- belief(new Threat(flow), false);
- } else {
- setEndState(EndState.FAILED);
- }
- }
- }
- }
-
- @Override
- protected void init() {
- this.subgoalDispatched = false;
- }
-
- @Parameter(direction = Direction.IN)
- public void setBeliefName(Threat threat) {
- this.flow = threat.getVariable();
- }
- }
+public class BDI2DoSCapability extends PalliativeCapability {
public static final String ROUTER_BELIEF = "router";
private static final long serialVersionUID = 4633237702870865396L;
@@ -348,127 +21,11 @@ public class BDI2DoSCapability extends RouterAgentCapability implements BeliefRe
@bdi4jade.annotation.Belief
private final Belief<String, Router> router;
- @bdi4jade.annotation.Belief
- protected CauseEffectKnowledgeModel causeEffectKnowledgeModel = new CauseEffectKnowledgeModel();
-
- @bdi4jade.annotation.Belief
- protected ResourcePreferences preferences = new ResourcePreferences();
-
- @bdi4jade.annotation.Plan
- private Plan limitLinkRate;
- @bdi4jade.annotation.Plan
- private Plan restoreLinkRate;
-
- @bdi4jade.annotation.Plan
- private Plan analyseLinkStatistics;
- @bdi4jade.annotation.Plan
- private Plan limitIpRate;
- @bdi4jade.annotation.Plan
- private Plan restoreIpRate;
-
- @bdi4jade.annotation.Plan
- private Plan analyseIpFlows;
- @bdi4jade.annotation.Plan
- private Plan limitFlowRate;
-
public BDI2DoSCapability(Router router) {
this.router = new TransientBelief<>(ROUTER_BELIEF, router);
- getLinkMonitor().addObserver(this);
-
- setBeliefRevisionStrategy(this);
- setOptionGenerationFunction(new MyGoalGenerationFunction());
- setPlanSelectionStrategy(new PalliativePlanSelectionStrategy(this));
addPartCapability(new RateLimiterCapability(getRateLimiter()));
addPartCapability(new FlowExporterCapability(getFlowExporter()));
-
- // Cause-Effect Knowledge Model
- CauseEffectRelationship cer = new CauseEffectRelationship(
- new Fact(new OverUsage(new Link("AFFECTED_LINK")), true));
- cer.addOptionalCause(new Fact(new AnomalousUsage(new Link("AFFECTED_LINK")), true));
- AlternativeCauseSet alternatives = new AlternativeCauseSet(1, 2);
- alternatives.addAlternativeCause(new Fact(new Anomalous(new Ip("victim1")), true));
- alternatives.addAlternativeCause(new Fact(new Anomalous(new Ip("victim2")), true));
- cer.addAlternativeCauseSet(alternatives);
- causeEffectKnowledgeModel.addCauseEffectRelationship(cer);
-
- cer = new CauseEffectRelationship(new Fact(new Anomalous(new Ip("victim1")), true));
- cer.addOptionalCause(new Fact(new Benign(new Ip("victim1")), false));
- alternatives = new AlternativeCauseSet(1, 2);
- alternatives.addAlternativeCause(
- new Fact(new Threat(new Flow(new Ip("DDoS1"), 80, new Ip("victim1"), 80, "http")), true));
- alternatives.addAlternativeCause(
- new Fact(new Threat(new Flow(new Ip("DDoS2"), 80, new Ip("victim1"), 80, "http")), true));
- cer.addAlternativeCauseSet(alternatives);
- causeEffectKnowledgeModel.addCauseEffectRelationship(cer);
-
- cer = new CauseEffectRelationship(new Fact(new Anomalous(new Ip("victim2")), true));
- cer.addOptionalCause(new Fact(new Benign(new Ip("victim2")), false));
- alternatives = new AlternativeCauseSet(1, 1);
- alternatives.addAlternativeCause(
- new Fact(new Threat(new Flow(new Ip("DDoS3"), 80, new Ip("victim2"), 80, "http")), true));
- cer.addAlternativeCauseSet(alternatives);
- causeEffectKnowledgeModel.addCauseEffectRelationship(cer);
-
- // Resource Preferences
- preferences.getValue().put(Resources.TIME, 0.75);
- preferences.getValue().put(Resources.NETWORK_AVAILABILITY, 0.25);
-
- // Plans
-
- this.limitLinkRate = new DefaultPlan(
- new GoalTemplate[] { GoalTemplateFactory.hasBeliefOfTypeWithValue(AttackPrevented.class, Boolean.TRUE),
- GoalTemplateFactory.hasBeliefOfTypeWithValue(OverUsage.class, Boolean.FALSE) },
- LimitLinkRate.class);
- PlanRequiredResource prr = new PlanRequiredResource();
- prr.setRequiredResource(Resources.TIME, 5.0);
- prr.setRequiredResource(Resources.NETWORK_AVAILABILITY, 0.6);
- prr.setRequiredResource(Resources.VULNERABILITY, 0.3);
- this.limitLinkRate.putMetadata(PlanRequiredResource.METADATA_NAME, prr);
- this.restoreLinkRate = new DefaultPlan(
- GoalTemplateFactory.hasBeliefOfTypeWithValue(AttackPrevented.class, Boolean.FALSE),
- RestoreLinkRate.class) {
- @Override
- public boolean isContextApplicable(Goal goal) {
- BeliefGoal<AttackPrevented> bg = (BeliefGoal<AttackPrevented>) goal;
- PredicateBelief<LinkRateLimited> linkRateLimited = (PredicateBelief<LinkRateLimited>) getBeliefBase()
- .getBelief(new LinkRateLimited(bg.getBeliefName().getVariable()));
- return (linkRateLimited != null && linkRateLimited.getValue());
- }
- };
-
- this.analyseLinkStatistics = new DefaultPlan(GoalTemplateFactory.hasBeliefOfType(AnomalousUsage.class),
- AnalyseLinkStatistics.class);
- this.limitIpRate = new DefaultPlan(
- new GoalTemplate[] { GoalTemplateFactory.hasBeliefOfTypeWithValue(Restricted.class, Boolean.TRUE),
- GoalTemplateFactory.hasBeliefOfTypeWithValue(Anomalous.class, Boolean.FALSE) },
- LimitIPRatePlan.class);
- this.restoreIpRate = new DefaultPlan(
- GoalTemplateFactory.hasBeliefOfTypeWithValue(Restricted.class, Boolean.FALSE),
- RestoreIPRatePlan.class) {
- @Override
- public boolean isContextApplicable(Goal goal) {
- BeliefGoal<Restricted> bg = (BeliefGoal<Restricted>) goal;
- PredicateBelief<IpRateLimited> ipRateLimited = (PredicateBelief<IpRateLimited>) getBeliefBase()
- .getBelief(new IpRateLimited(bg.getBeliefName().getVariable()));
- return (ipRateLimited != null && ipRateLimited.getValue());
- }
- };
-
- this.analyseIpFlows = new DefaultPlan(GoalTemplateFactory.hasBeliefOfType(Benign.class)
-
- , AnalyseIPFlows.class);
- this.limitFlowRate = new DefaultPlan(
- new GoalTemplate[] { GoalTemplateFactory.hasBeliefOfTypeWithValue(Threat.class, Boolean.FALSE) },
- LimitFlowRatePlan.class) {
- public boolean isContextApplicable(Goal goal) {
- BeliefGoal<Threat> bg = (BeliefGoal<Threat>) goal;
- PredicateBelief<Threat> threat = (PredicateBelief<Threat>) getBeliefBase()
- .getBelief(bg.getBeliefName());
- return (threat != null && threat.getValue());
- };
- };
-
}
private LinkMonitor getLinkMonitor() {
@@ -491,137 +48,4 @@ public class BDI2DoSCapability extends RouterAgentCapability implements BeliefRe
return (AnomalyDetection) router.getValue().getRole(Role.ANOMALY_DETECTION);
}
- @Override
- public Role getRole() {
- // FIXME
- return Role.LINK_MONITOR;
- }
-
- @Override
- protected Set<Class<?>> getLowPriorityGoal() {
- Set<Class<?>> lowPriorityGoals = new HashSet<>();
- lowPriorityGoals.add(AnomalousUsage.class);
- lowPriorityGoals.add(Benign.class);
- return lowPriorityGoals;
- }
-
- class MyGoalGenerationFunction extends PalliativeOptionGenerationFunction {
-
- private Boolean added = false;
-
- public MyGoalGenerationFunction() {
- super(BDI2DoSCapability.this);
- }
-
- @Override
- public void generateGoals(GoalUpdateSet goalUpdateSet) {
- super.generateGoals(goalUpdateSet);
-
- // Link Monitor
- Set<Belief<?, ?>> overUsageBeliefs = getBeliefBase().getBeliefsByType(OverUsage.class);
- for (Belief<?, ?> belief : overUsageBeliefs) {
- PredicateBelief<OverUsage> overUsage = (PredicateBelief<OverUsage>) belief;
- Link link = overUsage.getName().getVariable();
- if (overUsage.getValue()) {
- PredicateBelief<AnomalousUsage> anomalousUsage = (PredicateBelief<AnomalousUsage>) getBeliefBase()
- .getBelief(new AnomalousUsage(link));
- PredicateBelief<AttackPrevented> attackPrevented = (PredicateBelief<AttackPrevented>) getBeliefBase()
- .getBelief(new AttackPrevented(link));
- if (anomalousUsage == null || anomalousUsage.getValue()) {
- // OverUsage(l) AND !(not AnomalousUsage(l)) -->
- // ~OverUsage(l)
- if (!added) {
- goalUpdateSet.generateGoal(new OverUsageGoal(link, Boolean.FALSE), BDI2DoSCapability.this,
- null);
- log.info("OverUsageGoal added.");
- added = true;
- }
- }
- }
- }
-
- Set<Belief<?, ?>> attackPreventedBeliefs = getBeliefBase().getBeliefsByType(AttackPrevented.class);
- for (Belief<?, ?> belief : attackPreventedBeliefs) {
- PredicateBelief<AttackPrevented> attackPrevented = (PredicateBelief<AttackPrevented>) belief;
- if (attackPrevented.getValue()) {
- PredicateBelief<AnomalousUsage> anomalousUsage = (PredicateBelief<AnomalousUsage>) getBeliefBase()
- .getBelief(new AnomalousUsage(attackPrevented.getName().getVariable()));
- // AttackPrevented(l) AND not AnomalousUsage(l) --> not
- // AttackPrevented(l)
- if (anomalousUsage != null && !anomalousUsage.getValue()) {
- goal(goalUpdateSet, attackPrevented.getName(), Boolean.FALSE);
- }
- }
- }
-
- Set<Belief<?, ?>> restrictedBeliefs = getBeliefBase().getBeliefsByType(Restricted.class);
- for (Belief<?, ?> belief : restrictedBeliefs) {
- PredicateBelief<Restricted> restricted = (PredicateBelief<Restricted>) belief;
- if (restricted.getValue()) {
- PredicateBelief<Benign> benign = (PredicateBelief<Benign>) getBeliefBase()
- .getBelief(new Benign(restricted.getName().getVariable()));
- if (benign != null && benign.getValue()) {
- // Restricted(l) AND Benign(l) --> not
- // Restricted(l)
- goal(goalUpdateSet, restricted.getName(), false);
- }
- }
- }
-
- }
- }
-
- @Override
- public void goalPerformed(GoalEvent event) {
- if (GoalStatus.ACHIEVED.equals(event.getStatus())) {
- addBelief(((BeliefGoal<?>) event.getGoal()).getOutputBelief());
- }
- }
-
- @Override
- public void reviewBeliefs() {
- for (Link link : getLinkMonitor().getLinks()) {
- OverUsage overUsage = new OverUsage(link);
- if (getLinkMonitor().isOverUsage(link)) {
- PredicateBelief<OverUsage> overUsageBelief = (PredicateBelief<OverUsage>) getBeliefBase()
- .getBelief(overUsage);
- if (overUsageBelief == null || !overUsageBelief.getValue()) {
- belief(overUsage, true);
- belief(new AnomalousUsage(link), null);
- }
- } else {
- belief(overUsage, false);
- getLinkMonitor().removeLink(link);
- }
- }
-
- Set<Belief<?, ?>> overUsageCauseBeliefs = getBeliefBase().getBeliefsByType(OverUsageCause.class);
- Iterator<Belief<?, ?>> it = overUsageCauseBeliefs.iterator();
- while (it.hasNext()) {
- BeliefSet<OverUsageCause, Ip> overUsageCause = (BeliefSet<OverUsageCause, Ip>) it.next();
- if (overUsageCause.getValue().isEmpty()) {
- removeBelief(overUsageCause);
- }
- PredicateBelief<AnomalousUsage> anomalousUsage = (PredicateBelief<AnomalousUsage>) getBeliefBase()
- .getBelief(new AnomalousUsage(overUsageCause.getName().getVariable()));
- if (anomalousUsage != null && (anomalousUsage.getValue() == null || !anomalousUsage.getValue())) {
- removeBelief(anomalousUsage);
- }
- }
-
- Set<Belief<?, ?>> benignBeliefs = getBeliefBase().getBeliefsByType(Benign.class);
- Iterator<Belief<?, ?>> it2 = benignBeliefs.iterator();
- while (it2.hasNext()) {
- PredicateBelief<Benign> benignBelief = (PredicateBelief<Benign>) it2.next();
- if (benignBelief.getValue() == null || benignBelief.getValue()) {
- removeBelief(benignBelief);
- }
- }
- }
-
- @Override
- public void update(Observable o, Object arg) {
- getMyAgent().restart();
- }
-
}
