3be3103914b392ff75081bfd5fedcd01918ffad7
[openflowplugin.git] / extension / test-extension / src / main / java / org / opendaylight / openflowplugin / extension / test / Test.java
1 /**
2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.openflowplugin.extension.test;
9
10 import com.google.common.util.concurrent.Futures;
11 import com.google.common.util.concurrent.ListenableFuture;
12 import java.math.BigInteger;
13 import java.util.ArrayList;
14 import java.util.List;
15 import org.opendaylight.infrautils.utils.concurrent.JdkFutures;
16 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.DecNwTtlCaseBuilder;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.dec.nw.ttl._case.DecNwTtl;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.dec.nw.ttl._case.DecNwTtlBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.Action;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionKey;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.AddFlowInput;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.AddFlowInputBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.SalFlowService;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.FlowCookie;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.FlowModFlags;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.InstructionsBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.instruction.ApplyActionsCaseBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.instruction.apply.actions._case.ApplyActionsBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.list.Instruction;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.list.InstructionBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.list.InstructionKey;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.EtherType;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.ethernet.match.fields.EthernetTypeBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.EthernetMatchBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.layer._3.match.Ipv4Match;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.layer._3.match.Ipv4MatchBuilder;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowjava.nx.match.rev140421.NxmNxReg0;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.action.rev140714.dst.choice.grouping.dst.choice.DstNxRegCaseBuilder;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.action.rev140714.nodes.node.table.flow.instructions.instruction.instruction.apply.actions._case.apply.actions.action.action.NxActionRegLoadNodesNodeTableFlowApplyActionsCaseBuilder;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.action.rev140714.nx.action.reg.load.grouping.NxRegLoadBuilder;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.nicira.action.rev140714.nx.action.reg.load.grouping.nx.reg.load.DstBuilder;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.test.rev130819.TestFlowInput;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.test.rev130819.TestFlowOutput;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.test.rev130819.TestService;
53 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
54 import org.opendaylight.yangtools.yang.common.RpcResult;
55 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
56 import org.slf4j.Logger;
57 import org.slf4j.LoggerFactory;
58
59 /**
60  * Implementation of TestService.
61  *
62  * @author msunal
63  */
64 public class Test implements TestService {
65     private static final Logger LOG = LoggerFactory.getLogger(Test.class);
66
67     private SalFlowService flowService;
68
69
70     @Override
71     public ListenableFuture<RpcResult<TestFlowOutput>> testFlow(TestFlowInput input) {
72         AddFlowInputBuilder flow = new AddFlowInputBuilder();
73         flow.setPriority(2);
74         flow.setMatch(createMatchBld().build());
75         flow.setInstructions(createDecNwTtlInstructionsBld().build());
76         flow.setBarrier(Boolean.FALSE);
77         BigInteger value = BigInteger.valueOf(10L);
78         flow.setCookie(new FlowCookie(value));
79         flow.setCookieMask(new FlowCookie(value));
80         flow.setHardTimeout(0);
81         flow.setIdleTimeout(0);
82         flow.setInstallHw(false);
83         flow.setStrict(false);
84         flow.setContainerName(null);
85         flow.setFlags(new FlowModFlags(false, false, false, false, true));
86         flow.setTableId((short) 0);
87
88         flow.setFlowName("NiciraFLOW");
89
90         // Construct the flow instance id
91         final InstanceIdentifier<Node> flowInstanceId = InstanceIdentifier
92                 .builder(Nodes.class) // File under nodes
93                 // A particular node identified by nodeKey
94                 .child(Node.class, new NodeKey(new NodeId("openflow:1"))).build();
95         flow.setNode(new NodeRef(flowInstanceId));
96
97         pushFlowViaRpc(flow.build());
98
99         return Futures.immediateFuture(RpcResultBuilder.<TestFlowOutput>status(true).build());
100     }
101
102     private void pushFlowViaRpc(AddFlowInput addFlowInput) {
103         if (flowService != null) {
104             JdkFutures.addErrorLogging(flowService.addFlow(addFlowInput), LOG, "addFlow");
105         }
106     }
107
108     public void setFlowService(SalFlowService flowService) {
109         this.flowService = flowService;
110     }
111
112     private static MatchBuilder createMatchBld() {
113         MatchBuilder match = new MatchBuilder();
114         Ipv4MatchBuilder ipv4Match = new Ipv4MatchBuilder();
115         Ipv4Prefix prefix = new Ipv4Prefix("10.0.0.1/24");
116         ipv4Match.setIpv4Destination(prefix);
117         Ipv4Match i4m = ipv4Match.build();
118         match.setLayer3Match(i4m);
119
120         EthernetMatchBuilder eth = new EthernetMatchBuilder();
121         EthernetTypeBuilder ethTypeBuilder = new EthernetTypeBuilder();
122         ethTypeBuilder.setType(new EtherType(0x0800L));
123         eth.setEthernetType(ethTypeBuilder.build());
124         match.setEthernetMatch(eth.build());
125
126 //        AugmentTuple<Match> extAugmentWrapper = createNxMatchAugment();
127 //        match.addAugmentation(extAugmentWrapper.getAugmentationClass(), extAugmentWrapper.getAugmentationObject());
128
129         return match;
130     }
131
132     private static InstructionsBuilder createDecNwTtlInstructionsBld() {
133         // Add our drop action to a list
134         List<Action> actionList = new ArrayList<>();
135         actionList.add(createOFAction(0).build());
136         actionList.add(createNxActionBld(1).build());
137
138         // Create an Apply Action
139         ApplyActionsBuilder aab = new ApplyActionsBuilder();
140         aab.setAction(actionList);
141
142         // Wrap our Apply Action in an Instruction
143         InstructionBuilder ib = new InstructionBuilder();
144         ib.setInstruction(new ApplyActionsCaseBuilder().setApplyActions(aab.build()).build());
145         ib.setKey(new InstructionKey(0));
146         ib.setOrder(0);
147
148         // Put our Instruction in a list of Instructions
149         InstructionsBuilder isb = new InstructionsBuilder();
150         List<Instruction> instructions = new ArrayList<>();
151         instructions.add(ib.build());
152         ib.setKey(new InstructionKey(0));
153         isb.setInstruction(instructions);
154         return isb;
155     }
156
157     private static ActionBuilder createOFAction(int actionKeyVal) {
158         DecNwTtlBuilder ta = new DecNwTtlBuilder();
159         DecNwTtl decNwTtl = ta.build();
160         ActionBuilder ab = new ActionBuilder();
161         ab.setAction(new DecNwTtlCaseBuilder().setDecNwTtl(decNwTtl).build());
162         ab.setKey(new ActionKey(actionKeyVal));
163         return ab;
164     }
165
166     private static ActionBuilder createNxActionBld(int actionKeyVal) {
167         // vendor part
168         DstNxRegCaseBuilder nxRegCaseBld = new DstNxRegCaseBuilder().setNxReg(NxmNxReg0.class);
169         DstBuilder dstBld = new DstBuilder()
170             .setDstChoice(nxRegCaseBld.build())
171             .setStart(0)
172             .setEnd(5);
173         NxRegLoadBuilder nxRegLoadBuilder = new NxRegLoadBuilder();
174         nxRegLoadBuilder.setDst(dstBld.build());
175         nxRegLoadBuilder.setValue(BigInteger.valueOf(55L));
176         NxActionRegLoadNodesNodeTableFlowApplyActionsCaseBuilder topNxActionCaseBld =
177                 new NxActionRegLoadNodesNodeTableFlowApplyActionsCaseBuilder();
178         topNxActionCaseBld.setNxRegLoad(nxRegLoadBuilder.build());
179
180         // base part
181         ActionBuilder abExt = new ActionBuilder();
182         abExt.setKey(new ActionKey(actionKeyVal));
183         abExt.setAction(topNxActionCaseBld.build());
184         return abExt;
185     }
186 }