use of org.onosproject.segmentrouting.policy.api.Policy.PolicyType in project trellis-control by opennetworkinglab.
the class PolicyManager method installTrafficMatchToDevice.
// Orchestrate traffic match installation according to the type
private void installTrafficMatchToDevice(DeviceId deviceId, TrafficMatch trafficMatch) {
if (log.isDebugEnabled()) {
log.debug("Installing traffic match {} associated to policy {}", trafficMatch.trafficMatchId(), trafficMatch.policyId());
}
TrafficMatchKey trafficMatchKey = new TrafficMatchKey(deviceId, trafficMatch.trafficMatchId());
Operation oldTrafficOperation = Versioned.valueOrNull(operations.get(trafficMatchKey.toString()));
if (oldTrafficOperation != null && oldTrafficOperation.isInstall()) {
if (trafficMatch.equals(oldTrafficOperation.trafficMatch().orElse(null))) {
if (log.isDebugEnabled()) {
log.debug("There is already an install operation for traffic match {} associated to policy {} " + "for device {}", trafficMatch.trafficMatchId(), trafficMatch.policyId(), deviceId);
}
// If we add or submit a trafficMatch multiple times
// We skip the installation and update the state directly
updateTrafficMatch(trafficMatch, true);
return;
} else {
if (log.isDebugEnabled()) {
log.debug("Starts updating traffic match {} associated to policy {} " + "for device {}", trafficMatch.trafficMatchId(), trafficMatch.policyId(), deviceId);
}
}
}
// For the DROP policy we need to set an ACL drop in the fwd objective. The other
// policies require to retrieve the next Id and sets the next step.
PolicyKey policyKey = new PolicyKey(deviceId, trafficMatch.policyId());
Operation policyOperation = Versioned.valueOrNull(operations.get(policyKey.toString()));
if (policyOperation == null || !policyOperation.isDone() || !policyOperation.isInstall() || policyOperation.policy().isEmpty() || (policyOperation.policy().get().policyType() == PolicyType.REDIRECT && policyOperation.objectiveOperation() == null)) {
log.info("Deferring traffic match {} installation on device {}. Policy {} not yet installed", trafficMatch.trafficMatchId(), deviceId, trafficMatch.policyId());
return;
}
// Updates the store and then send the versatile fwd objective to the pipeliner
Operation newTrafficOperation = Operation.builder().isInstall(true).trafficMatch(trafficMatch).build();
operations.put(trafficMatchKey.toString(), newTrafficOperation);
Policy policy = policyOperation.policy().get();
ForwardingObjective.Builder builder = trafficMatchFwdObjective(trafficMatch, policy.policyType());
if (policy.policyType() == PolicyType.DROP) {
// Firstly builds the fwd objective with the wipeDeferred action.
TrafficTreatment dropTreatment = DefaultTrafficTreatment.builder().wipeDeferred().build();
builder.withTreatment(dropTreatment);
} else if (policy.policyType() == PolicyType.REDIRECT) {
// Here we need to set only the next step
builder.nextStep(policyOperation.objectiveOperation().id());
}
// Once, the fwd objective has completed its execution, we update the policiesOps map
CompletableFuture<Objective> addNewFuture = new CompletableFuture<>();
CompletableFuture<Objective> removeOldFuture = new CompletableFuture<>();
if (log.isDebugEnabled()) {
log.debug("Installing forwarding objective for dev: {}", deviceId);
}
ObjectiveContext addNewContext = new DefaultObjectiveContext((objective) -> {
if (log.isDebugEnabled()) {
log.debug("Forwarding objective for policy {} installed", trafficMatch.policyId());
}
addNewFuture.complete(objective);
}, (objective, error) -> {
log.warn("Failed to install forwarding objective for policy {}: {}", trafficMatch.policyId(), error);
addNewFuture.complete(null);
});
ObjectiveContext removeOldContext = new DefaultObjectiveContext((objective) -> {
if (log.isDebugEnabled()) {
log.debug("Old forwarding objective for policy {} removed, update finished", trafficMatch.policyId());
}
removeOldFuture.complete(objective);
}, (objective, error) -> {
log.warn("Failed to remove old forwarding objective for policy {}: {}", trafficMatch.policyId(), error);
removeOldFuture.complete(null);
});
// Context is not serializable
ForwardingObjective serializableObjective = builder.add();
flowObjectiveService.forward(deviceId, builder.add(addNewContext));
addNewFuture.whenComplete((objective, ex) -> {
if (ex != null) {
log.error("Exception installing forwarding objective", ex);
} else if (objective != null) {
// base on priority, selector and metadata change
if (oldTrafficOperation != null && oldTrafficOperation.objectiveOperation() != null && oldTrafficOperation.isInstall() && (oldTrafficOperation.objectiveOperation().priority() != serializableObjective.priority() || !((ForwardingObjective) oldTrafficOperation.objectiveOperation()).selector().equals(serializableObjective.selector()) || !((ForwardingObjective) oldTrafficOperation.objectiveOperation()).meta().equals(serializableObjective.meta()))) {
ForwardingObjective oldFwdObj = (ForwardingObjective) oldTrafficOperation.objectiveOperation();
ForwardingObjective.Builder oldBuilder = DefaultForwardingObjective.builder(oldFwdObj);
flowObjectiveService.forward(deviceId, oldBuilder.remove(removeOldContext));
} else {
operations.computeIfPresent(trafficMatchKey.toString(), (k, v) -> {
if (!v.isDone() && v.isInstall()) {
v.isDone(true);
v.objectiveOperation(serializableObjective);
}
return v;
});
}
}
});
removeOldFuture.whenComplete((objective, ex) -> {
if (ex != null) {
log.error("Exception removing old forwarding objective", ex);
} else if (objective != null) {
operations.computeIfPresent(trafficMatchKey.toString(), (k, v) -> {
if (!v.isDone() && v.isInstall()) {
v.isDone(true);
v.objectiveOperation(serializableObjective);
}
return v;
});
}
});
}
use of org.onosproject.segmentrouting.policy.api.Policy.PolicyType in project trellis-control by opennetworkinglab.
the class PolicyWebResource method getPolicies.
/**
* Get all Policies.
*
* @return 200 OK will a collection of Policies
*/
@GET
@Produces(MediaType.APPLICATION_JSON)
public Response getPolicies() {
PolicyService policyService = get(PolicyService.class);
ObjectNode root = mapper().createObjectNode();
ArrayNode policiesArr = root.putArray(POLICY);
// Create a filter set contains all PolicyType
Set<PolicyType> policyTypes = Set.of(PolicyType.values());
for (PolicyData policyData : policyService.policies(policyTypes)) {
Policy policy = policyData.policy();
switch(policy.policyType()) {
case DROP:
policiesArr.add(codec(DropPolicy.class).encode((DropPolicy) policy, this));
break;
case REDIRECT:
policiesArr.add(codec(RedirectPolicy.class).encode((RedirectPolicy) policy, this));
break;
default:
continue;
}
}
return Response.ok(root).build();
}
use of org.onosproject.segmentrouting.policy.api.Policy.PolicyType in project trellis-control by opennetworkinglab.
the class PolicyWebResource method getDropPolicies.
/**
* Get all Drop Policies.
*
* @return 200 OK will a collection of Dop Policies
*/
@GET
@Produces(MediaType.APPLICATION_JSON)
@Path("drop")
public Response getDropPolicies() {
PolicyService policyService = get(PolicyService.class);
ObjectNode root = mapper().createObjectNode();
ArrayNode policiesArr = root.putArray(POLICY);
Set<PolicyType> policyTypes = Set.of(PolicyType.DROP);
for (PolicyData policyData : policyService.policies(policyTypes)) {
Policy policy = policyData.policy();
policiesArr.add(codec(DropPolicy.class).encode((DropPolicy) policy, this));
}
return Response.ok(root).build();
}
use of org.onosproject.segmentrouting.policy.api.Policy.PolicyType in project trellis-control by opennetworkinglab.
the class PolicyWebResource method getRedirectPolicies.
/**
* Get all Redirect Policies.
*
* @return 200 OK will a collection of Redirect Policies
*/
@GET
@Produces(MediaType.APPLICATION_JSON)
@Path("redirect")
public Response getRedirectPolicies() {
PolicyService policyService = get(PolicyService.class);
ObjectNode root = mapper().createObjectNode();
ArrayNode policiesArr = root.putArray(POLICY);
Set<PolicyType> policyTypes = Set.of(PolicyType.REDIRECT);
for (PolicyData policyData : policyService.policies(policyTypes)) {
Policy policy = policyData.policy();
policiesArr.add(codec(RedirectPolicy.class).encode((RedirectPolicy) policy, this));
}
return Response.ok(root).build();
}
Aggregations