commit 0ff59c27e4578b370524c893328e70a532b86c05
author Michele Santuari <michele.santuari@create-net.org> Fri Jan 15 10:00:32 2016 +0100
committer Gerrit Code Review <gerrit@onlab.us> Sat Feb 06 00:58:01 2016 +0000
tree 9b2ddf1b97fc96f26531ace49a5e09fcec9d5d93
parent 6b045587c95da8475c847643df3c91c5ab272b77

fix on VLAN intent encapsulation ONOS-3467

Change-Id: Ia0a72d13a551fd4183531d35c04ca19d87b85280

core/net/src/main/java/org/onosproject/net/intent/impl/compiler/PathIntentCompiler.java

diff --git a/core/net/src/main/java/org/onosproject/net/intent/impl/compiler/PathIntentCompiler.java b/core/net/src/main/java/org/onosproject/net/intent/impl/compiler/PathIntentCompiler.java
index 7473ce5..93e2d8c 100644
--- a/core/net/src/main/java/org/onosproject/net/intent/impl/compiler/PathIntentCompiler.java
+++ b/core/net/src/main/java/org/onosproject/net/intent/impl/compiler/PathIntentCompiler.java
@@ -172,12 +172,20 @@
         }
         VlanId prevVlanId = vlanId;
 
-        //Tag the traffic with the new VLAN
-        TrafficTreatment treat = DefaultTrafficTreatment.builder()
-                .setVlanId(vlanId)
-                .build();
+        Optional<VlanIdCriterion> vlanCriterion = intent.selector().criteria()
+                .stream().filter(criterion -> criterion.type() == Criterion.Type.VLAN_VID)
+                .map(criterion -> (VlanIdCriterion) criterion)
+                .findAny();
 
-        rules.add(createFlowRule(intent.selector(), treat, srcLink.dst(), link.src(), intent.priority(), true));
+        //Push VLAN if selector does not include VLAN
+        TrafficTreatment.Builder treatBuilder = DefaultTrafficTreatment.builder();
+        if (!vlanCriterion.isPresent()) {
+            treatBuilder.pushVlan();
+        }
+        //Tag the traffic with the new encapsulation VLAN
+        treatBuilder.setVlanId(vlanId);
+        rules.add(createFlowRule(intent.selector(), treatBuilder.build(),
+                                 srcLink.dst(), link.src(), intent.priority(), true));
 
         ConnectPoint prev = link.dst();
 
@@ -211,15 +219,9 @@
                 TrafficSelector egressSelector = DefaultTrafficSelector.builder()
                         .matchInPort(prev.port())
                         .matchVlanId(prevVlanId).build();
-
-                //TODO: think to other cases for egress packet restoration
-                Optional<VlanIdCriterion> vlanCriteria = intent.selector().criteria()
-                        .stream().filter(criteria -> criteria.type() == Criterion.Type.VLAN_VID)
-                        .map(criteria -> (VlanIdCriterion) criteria)
-                        .findAny();
                 TrafficTreatment.Builder egressTreat = DefaultTrafficTreatment.builder(intent.treatment());
-                if (vlanCriteria.isPresent()) {
-                    egressTreat.setVlanId(vlanCriteria.get().vlanId());
+                if (vlanCriterion.isPresent()) {
+                    egressTreat.setVlanId(vlanCriterion.get().vlanId());
                 } else {
                     egressTreat.popVlan();
                 }