09ce406ad69a1fbd0256e2ff28b53a72820927c9
[bgpcep.git] / pcep / ietf-stateful02 / src / main / java / org / opendaylight / protocol / pcep / ietf / stateful02 / Stateful02PCRequestMessageParser.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.protocol.pcep.ietf.stateful02;
9
10 import com.google.common.collect.Lists;
11
12 import io.netty.buffer.ByteBuf;
13
14 import java.util.List;
15
16 import org.opendaylight.protocol.pcep.impl.message.PCEPRequestMessageParser;
17 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
18 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.P2p1;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.P2p1Builder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.Lsp;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.BandwidthBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.classtype.object.ClassType;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.exclude.route.object.Xro;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.load.balancing.object.LoadBalancing;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.of.object.Of;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.SegmentComputation;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.SegmentComputationBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.P2p;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.P2pBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.p2p.ReportedRoute;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.p2p.ReportedRouteBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reported.route.object.Rro;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
43
44 /**
45  * Parser for Pcreq
46  */
47 public final class Stateful02PCRequestMessageParser extends PCEPRequestMessageParser {
48
49     public Stateful02PCRequestMessageParser(final ObjectRegistry registry) {
50         super(registry);
51     }
52
53     @Override
54     protected void serializeP2P(final ByteBuf buffer, final P2p p2p) {
55         if (p2p.getEndpointsObj() != null) {
56             serializeObject(p2p.getEndpointsObj(), buffer);
57         }
58         if (p2p.getAugmentation(P2p1.class) != null && p2p.getAugmentation(P2p1.class).getLsp() != null) {
59             serializeObject(p2p.getAugmentation(P2p1.class).getLsp(), buffer);
60         }
61         if (p2p.getReportedRoute() != null) {
62             final ReportedRoute rr = p2p.getReportedRoute();
63             if (rr.getRro() != null) {
64                 serializeObject(rr.getRro(), buffer);
65             }
66             if (rr.getBandwidth() != null) {
67                 serializeObject(rr.getBandwidth(), buffer);
68             }
69         }
70         if (p2p.getLoadBalancing() != null) {
71             serializeObject(p2p.getLoadBalancing(), buffer);
72         }
73         if (p2p.getLspa() != null) {
74             serializeObject(p2p.getLspa(), buffer);
75         }
76         if (p2p.getBandwidth() != null) {
77             serializeObject(p2p.getBandwidth(), buffer);
78         }
79         if (p2p.getMetrics() != null) {
80             for (final Metrics m : p2p.getMetrics()) {
81                 serializeObject(m.getMetric(), buffer);
82             }
83         }
84         if (p2p.getIro() != null) {
85             serializeObject(p2p.getIro(), buffer);
86         }
87         if (p2p.getRro() != null) {
88             serializeObject(p2p.getRro(), buffer);
89         }
90         if (p2p.getXro() != null) {
91             serializeObject(p2p.getXro(), buffer);
92         }
93         if (p2p.getOf() != null) {
94             serializeObject(p2p.getOf(), buffer);
95         }
96         if (p2p.getClassType() != null) {
97             serializeObject(p2p.getClassType(), buffer);
98         }
99     }
100
101     @Override
102     protected SegmentComputation getSegmentComputation(final P2pBuilder builder, final List<Object> objects, final List<Message> errors,
103             final Rp rp) {
104         final List<Metrics> metrics = Lists.newArrayList();
105
106         State state = State.Init;
107         while (!objects.isEmpty() && state != State.End) {
108             Object obj = objects.get(0);
109
110             switch (state) {
111             case Init:
112                 state = State.LspIn;
113                 if (obj instanceof Lsp) {
114                     builder.addAugmentation(P2p1.class, new P2p1Builder().setLsp((Lsp) obj).build());
115                     break;
116                 }
117             case LspIn:
118                 state = State.ReportedIn;
119                 if (obj instanceof Rro) {
120                     final ReportedRouteBuilder rrBuilder = new ReportedRouteBuilder();
121                     rrBuilder.setRro((Rro) obj);
122                     objects.remove(0);
123                     obj = objects.get(0);
124                     if (obj instanceof Bandwidth) {
125                         rrBuilder.setBandwidth((Bandwidth) obj);
126                     }
127                     break;
128                 }
129             case ReportedIn:
130                 state = State.LoadBIn;
131                 if (obj instanceof LoadBalancing) {
132                     builder.setLoadBalancing((LoadBalancing) obj);
133                     break;
134                 }
135             case LoadBIn:
136                 state = State.LspaIn;
137                 if (obj instanceof Lspa) {
138                     builder.setLspa((Lspa) obj);
139                     break;
140                 }
141             case LspaIn:
142                 state = State.BandwidthIn;
143                 if (obj instanceof Bandwidth) {
144                     builder.setBandwidth((Bandwidth) obj);
145                     break;
146                 }
147             case BandwidthIn:
148                 state = State.MetricIn;
149                 if (obj instanceof Metric) {
150                     metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
151                     state = State.BandwidthIn;
152                     break;
153                 }
154             case MetricIn:
155                 state = State.IroIn;
156                 if (obj instanceof Iro) {
157                     builder.setIro((Iro) obj);
158                     break;
159                 }
160             case IroIn:
161                 state = State.RroIn;
162                 if (obj instanceof Rro) {
163                     builder.setRro((Rro) obj);
164                     break;
165                 }
166             case RroIn:
167                 state = State.XroIn;
168                 if (obj instanceof Xro) {
169                     builder.setXro((Xro) obj);
170                     break;
171                 }
172             case XroIn:
173                 state = State.OfIn;
174                 if (obj instanceof Of) {
175                     builder.setOf((Of) obj);
176                     break;
177                 }
178             case OfIn:
179                 state = State.CtIn;
180                 if (obj instanceof ClassType) {
181                     final ClassType classType = (ClassType) obj;
182                     if (!classType.isProcessingRule()) {
183                         errors.add(createErrorMsg(PCEPErrors.P_FLAG_NOT_SET, rp));
184                     } else {
185                         builder.setClassType(classType);
186                     }
187                     break;
188                 }
189             case CtIn:
190                 state = State.End;
191                 break;
192             case End:
193                 break;
194             }
195             if (!state.equals(State.End)) {
196                 objects.remove(0);
197             }
198         }
199         if (!metrics.isEmpty()) {
200             builder.setMetrics(metrics);
201         }
202
203         if (rp.isReoptimization()
204                 && builder.getBandwidth() != null
205                 && builder.getReportedRoute().getBandwidth().getBandwidth() != new BandwidthBuilder().setBandwidth(
206                         new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.concepts.rev131125.Bandwidth(new byte[] { 0 })).build()
207                         && builder.getReportedRoute().getRro() == null) {
208             errors.add(createErrorMsg(PCEPErrors.RRO_MISSING, rp));
209             return null;
210         }
211         return new SegmentComputationBuilder().setP2p(builder.build()).build();
212     }
213
214     private enum State {
215         Init, LspIn, ReportedIn, LoadBIn, LspaIn, BandwidthIn, MetricIn, IroIn, RroIn, XroIn, OfIn, CtIn, End
216     }
217 }