Code clean up
[bgpcep.git] / pcep / ietf-stateful07 / src / main / java / org / opendaylight / protocol / pcep / ietf / stateful07 / Stateful07PCReportMessageParser.java
1 /*
2  * Copyright (c) 2013 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.stateful07;
9
10 import com.google.common.base.Optional;
11 import com.google.common.base.Preconditions;
12 import com.google.common.collect.Lists;
13 import io.netty.buffer.ByteBuf;
14 import io.netty.buffer.Unpooled;
15 import java.util.List;
16 import org.opendaylight.protocol.pcep.spi.AbstractMessageParser;
17 import org.opendaylight.protocol.pcep.spi.MessageUtil;
18 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
19 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
20 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
21 import org.opendaylight.protocol.pcep.spi.PSTUtil;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Pcrpt;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PcrptBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.Lsp;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.PcrptMessageBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.Reports;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.ReportsBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.reports.Path;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.reports.PathBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.Srp;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.srp.Tlvs;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
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 {@link Pcrpt}
46  */
47 public class Stateful07PCReportMessageParser extends AbstractMessageParser {
48
49     public static final int TYPE = 10;
50
51     public Stateful07PCReportMessageParser(final ObjectRegistry registry) {
52         super(registry);
53     }
54
55     @Override
56     public void serializeMessage(final Message message, final ByteBuf out) {
57         Preconditions.checkArgument(message instanceof Pcrpt, "Wrong instance of Message. Passed instance of %s. Need Pcrpt.", message.getClass());
58         final Pcrpt msg = (Pcrpt) message;
59         final List<Reports> reports = msg.getPcrptMessage().getReports();
60         final ByteBuf buffer = Unpooled.buffer();
61         for (final Reports report : reports) {
62             serializeReport(report, buffer);
63         }
64         MessageUtil.formatMessage(TYPE, buffer, out);
65     }
66
67     private void serializeReport(final Reports report, final ByteBuf buffer) {
68         if (report.getSrp() != null) {
69             serializeObject(report.getSrp(), buffer);
70         }
71         serializeObject(report.getLsp(), buffer);
72         final Path p = report.getPath();
73         if (p != null) {
74             serializeObject(p.getEro(), buffer);
75             serializeObject(p.getLspa(), buffer);
76             serializeObject(p.getBandwidth(), buffer);
77             serializeObject(p.getReoptimizationBandwidth(), buffer);
78             if (p.getMetrics() != null) {
79                 for (final Metrics m : p.getMetrics()) {
80                     serializeObject(m.getMetric(), buffer);
81                 }
82             }
83             serializeObject(p.getIro(), buffer);
84             serializeObject(p.getRro(), buffer);
85         }
86     }
87
88     @Override
89     public Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
90         Preconditions.checkArgument(objects != null, "Passed list can't be null.");
91         if (objects.isEmpty()) {
92             throw new PCEPDeserializerException("Pcrpt message cannot be empty.");
93         }
94
95         final List<Reports> reports = Lists.newArrayList();
96
97         while (!objects.isEmpty()) {
98             final Reports report = getValidReports(objects, errors);
99             if (report != null) {
100                 reports.add(report);
101             }
102         }
103         return new PcrptBuilder().setPcrptMessage(new PcrptMessageBuilder().setReports(reports).build()).build();
104     }
105
106     protected Reports getValidReports(final List<Object> objects, final List<Message> errors) {
107         final ReportsBuilder builder = new ReportsBuilder();
108
109         boolean lspViaSR = false;
110         Object object = objects.remove(0);
111         if (object instanceof Srp) {
112             final Srp srp = (Srp) object;
113             final Tlvs tlvs = srp.getTlvs();
114             if (tlvs != null) {
115                 lspViaSR = PSTUtil.isDefaultPST(tlvs.getPathSetupType());
116             }
117             builder.setSrp(srp);
118             if (objects.isEmpty()) {
119                 object = null;
120             } else {
121                 object = objects.remove(0);
122             }
123         }
124
125         if (validateLsp(object, lspViaSR, errors, builder)) {
126             if (!objects.isEmpty()) {
127                 if (!validatePath(objects, errors, builder)) {
128                     return null;
129                 }
130             }
131
132             return builder.build();
133         }
134         return null;
135     }
136
137     private boolean validateLsp(final Object object, final boolean lspViaSR, final List<Message> errors, final ReportsBuilder builder) {
138         if (object instanceof Lsp) {
139             final Lsp lsp = (Lsp) object;
140             final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.Tlvs tlvs = lsp.getTlvs();
141             if (!lspViaSR && lsp.getPlspId().getValue() != 0 && (tlvs == null || tlvs.getLspIdentifiers() == null)) {
142                 final Message errorMsg = createErrorMsg(PCEPErrors.LSP_IDENTIFIERS_TLV_MISSING, Optional.absent());
143                 errors.add(errorMsg);
144                 return false;
145             } else {
146                 builder.setLsp(lsp);
147             }
148         } else {
149             errors.add(createErrorMsg(PCEPErrors.LSP_MISSING, Optional.absent()));
150             return false;
151         }
152         return true;
153     }
154
155     private boolean validatePath(final List<Object> objects, final List<Message> errors, final ReportsBuilder builder) {
156         final PathBuilder pBuilder = new PathBuilder();
157         Object object = objects.remove(0);
158         if (object instanceof Ero) {
159             pBuilder.setEro((Ero) object);
160         } else {
161             errors.add(createErrorMsg(PCEPErrors.ERO_MISSING, Optional.absent()));
162             return false;
163         }
164         parsePath(objects, pBuilder);
165         builder.setPath(pBuilder.build());
166         return true;
167     }
168
169     private void parsePath(final List<Object> objects, final PathBuilder builder) {
170         final List<Metrics> pathMetrics = Lists.newArrayList();
171         Object obj;
172         State state = State.INIT;
173         while (!objects.isEmpty() && !state.equals(State.END)) {
174             obj = objects.get(0);
175             state = insertObject(state, obj, builder, pathMetrics);
176             if (!state.equals(State.END)) {
177                 objects.remove(0);
178             }
179         }
180         if (!pathMetrics.isEmpty()) {
181             builder.setMetrics(pathMetrics);
182         }
183     }
184
185     private State insertObject(final State state, final Object obj, final PathBuilder builder, final List<Metrics> pathMetrics) {
186         switch (state) {
187         case INIT:
188             if (obj instanceof Lspa) {
189                 builder.setLspa((Lspa) obj);
190                 return State.LSPA_IN;
191             }
192         case LSPA_IN:
193             if (obj instanceof Bandwidth) {
194                 builder.setBandwidth((Bandwidth) obj);
195                 return State.LSPA_IN;
196             }
197             if (obj instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reoptimization.bandwidth.object.ReoptimizationBandwidth) {
198                 builder.setReoptimizationBandwidth((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reoptimization.bandwidth.object.ReoptimizationBandwidth) obj);
199                 return State.LSPA_IN;
200             }
201         case BANDWIDTH_IN:
202             if (obj instanceof Metric) {
203                 pathMetrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
204                 return State.BANDWIDTH_IN;
205             }
206         case METRIC_IN:
207             if (obj instanceof Iro) {
208                 builder.setIro((Iro) obj);
209                 return State.IRO_IN;
210             }
211         case IRO_IN:
212             if (obj instanceof Rro) {
213                 builder.setRro((Rro) obj);
214                 return State.RRO_IN;
215             }
216         case RRO_IN:
217         case END:
218             return State.END;
219         default:
220             return state;
221         }
222     }
223
224     private enum State {
225         INIT, LSPA_IN, BANDWIDTH_IN, METRIC_IN, IRO_IN, RRO_IN, END
226     }
227 }