import io.netty.buffer.ByteBuf;
-import org.opendaylight.protocol.pcep.PCEPDeserializerException;
-import org.opendaylight.protocol.pcep.spi.AbstractMessageParser;
-import org.opendaylight.protocol.pcep.spi.HandlerRegistry;
+import java.util.List;
+
+import org.opendaylight.protocol.pcep.spi.ObjectHandlerRegistry;
+import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
+import org.opendaylight.protocol.pcep.spi.PCEPErrors;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Pcrep;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcrepBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcrepMessage;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.of.object.Of;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.PcrepMessageBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.Replies;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.RepliesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.Result;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.FailureCase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.FailureCaseBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.SuccessCase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.SuccessCaseBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.failure._case.NoPath;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.SuccessBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.success.Paths;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.success.PathsBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
+
+import com.google.common.collect.Lists;
/**
- * Parser for {@link PcrepMessage}
+ * Parser for {@link Pcrep}
*/
-// FIXME: finish
public class PCEPReplyMessageParser extends AbstractMessageParser {
-
- private final int TYPE = 4;
- public PCEPReplyMessageParser(HandlerRegistry registry) {
+ public static final int TYPE = 4;
+
+ public PCEPReplyMessageParser(final ObjectHandlerRegistry registry) {
super(registry);
}
@Override
- public void serializeMessage(Message message, ByteBuf buffer) {
- if (!(message instanceof PcrepMessage))
+ public void serializeMessage(final Message message, final ByteBuf buffer) {
+ if (!(message instanceof Pcrep)) {
throw new IllegalArgumentException("Wrong instance of Message. Passed instance of " + message.getClass()
+ ". Nedded PcrepMessage.");
-
+ }
+ final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.PcrepMessage repMsg = ((Pcrep) message).getPcrepMessage();
+ if (repMsg.getReplies() == null || repMsg.getReplies().isEmpty()) {
+ throw new IllegalArgumentException("Replies cannot be null or empty.");
+ }
+ for (final Replies reply : repMsg.getReplies()) {
+ if (reply.getRp() == null) {
+ throw new IllegalArgumentException("Reply must contain RP object.");
+ }
+ buffer.writeBytes(serializeObject(reply.getRp()));
+ if (reply.getResult() != null) {
+ if (reply.getResult() instanceof FailureCase) {
+ final FailureCase f = ((FailureCase) reply.getResult());
+ buffer.writeBytes(serializeObject(f.getNoPath()));
+ if (f.getLspa() != null) {
+ buffer.writeBytes(serializeObject(f.getLspa()));
+ }
+ if (f.getBandwidth() != null) {
+ buffer.writeBytes(serializeObject(f.getBandwidth()));
+ }
+ if (f.getMetrics() != null && !f.getMetrics().isEmpty()) {
+ for (final Metrics m : f.getMetrics()) {
+ buffer.writeBytes(serializeObject(m.getMetric()));
+ }
+ }
+ if (f.getIro() != null) {
+ buffer.writeBytes(serializeObject(f.getIro()));
+ }
+ } else {
+ final SuccessCase s = (SuccessCase) reply.getResult();
+ for (final Paths p : s.getSuccess().getPaths()) {
+ buffer.writeBytes(serializeObject(p.getEro()));
+ if (p.getLspa() != null) {
+ buffer.writeBytes(serializeObject(p.getLspa()));
+ }
+ if (p.getOf() != null) {
+ buffer.writeBytes(serializeObject(p.getOf()));
+ }
+ if (p.getBandwidth() != null) {
+ buffer.writeBytes(serializeObject(p.getBandwidth()));
+ }
+ if (p.getMetrics() != null && !p.getMetrics().isEmpty()) {
+ for (final Metrics m : p.getMetrics()) {
+ buffer.writeBytes(serializeObject(m.getMetric()));
+ }
+ }
+ if (p.getIro() != null) {
+ buffer.writeBytes(serializeObject(p.getIro()));
+ }
+ }
+ }
+ }
+ }
}
@Override
- public PcrepMessage parseMessage(byte[] buffer) throws PCEPDeserializerException {
- return null;
+ protected Pcrep validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
+ if (objects == null) {
+ throw new IllegalArgumentException("Passed list can't be null.");
+ }
+ if (objects.isEmpty()) {
+ throw new PCEPDeserializerException("Pcrep message cannot be empty.");
+ }
+ final List<Replies> replies = Lists.newArrayList();
+ while (!objects.isEmpty()) {
+ final Replies r = this.getValidReply(objects, errors);
+ if (r != null) {
+ replies.add(r);
+ }
+ }
+ if (!objects.isEmpty()) {
+ throw new PCEPDeserializerException("Unprocessed Objects: " + objects);
+ }
+ return new PcrepBuilder().setPcrepMessage(new PcrepMessageBuilder().setReplies(replies).build()).build();
+ }
+
+ private Replies getValidReply(final List<Object> objects, final List<Message> errors) {
+ if (!(objects.get(0) instanceof Rp)) {
+ errors.add(createErrorMsg(PCEPErrors.RP_MISSING));
+ return null;
+ }
+ final Rp rp = (Rp) objects.get(0);
+ objects.remove(0);
+ Result res = null;
+ if (!objects.isEmpty()) {
+ if (objects.get(0) instanceof NoPath) {
+ final NoPath noPath = (NoPath) objects.get(0);
+ objects.remove(0);
+ final FailureCaseBuilder builder = new FailureCaseBuilder();
+ builder.setNoPath(noPath);
+ while (!objects.isEmpty()) {
+ this.parseAttributes(builder, objects);
+ }
+ res = builder.build();
+ } else if (objects.get(0) instanceof Ero) {
+ final Ero ero = (Ero) objects.get(0);
+ objects.remove(0);
+ final SuccessBuilder builder = new SuccessBuilder();
+ final List<Paths> paths = Lists.newArrayList();
+ final PathsBuilder pBuilder = new PathsBuilder();
+ pBuilder.setEro(ero);
+ while (!objects.isEmpty()) {
+ this.parsePath(pBuilder, objects);
+ paths.add(pBuilder.build());
+ }
+ builder.setPaths(paths);
+ res = new SuccessCaseBuilder().setSuccess(builder.build()).build();
+ }
+ }
+ return new RepliesBuilder().setRp(rp).setResult(res).build();
+ }
+
+ private void parseAttributes(final FailureCaseBuilder builder, final List<Object> objects) {
+ final List<Metrics> pathMetrics = Lists.newArrayList();
+
+ Object obj;
+ State state = State.Init;
+ while (!objects.isEmpty() && !state.equals(State.End)) {
+ obj = objects.get(0);
+
+ switch (state) {
+ case Init:
+ state = State.LspaIn;
+ if (obj instanceof Lspa) {
+ builder.setLspa((Lspa) obj);
+ break;
+ }
+ case LspaIn:
+ state = State.BandwidthIn;
+ if (obj instanceof Bandwidth) {
+ builder.setBandwidth((Bandwidth) obj);
+ break;
+ }
+ case BandwidthIn:
+ state = State.MetricIn;
+ if (obj instanceof Metric) {
+ pathMetrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
+ state = State.MetricIn;
+ break;
+ }
+ case MetricIn:
+ state = State.IroIn;
+ if (obj instanceof Iro) {
+ builder.setIro((Iro) obj);
+ break;
+ }
+ case IroIn:
+ state = State.End;
+ break;
+ case End:
+ break;
+ }
+ if (!state.equals(State.End)) {
+ objects.remove(0);
+ }
+ }
+ builder.setMetrics(pathMetrics);
+ }
+
+ private void parsePath(final PathsBuilder builder, final List<Object> objects) {
+ final List<Metrics> pathMetrics = Lists.newArrayList();
+
+ Object obj;
+ State state = State.Init;
+ while (!objects.isEmpty() && !state.equals(State.End)) {
+ obj = objects.get(0);
+
+ switch (state) {
+ case Init:
+ state = State.LspaIn;
+ if (obj instanceof Lspa) {
+ builder.setLspa((Lspa) obj);
+ break;
+ }
+ case LspaIn:
+ state = State.OfIn;
+ if (obj instanceof Of) {
+ builder.setOf((Of) obj);
+ break;
+ }
+ case OfIn:
+ state = State.BandwidthIn;
+ if (obj instanceof Bandwidth) {
+ builder.setBandwidth((Bandwidth) obj);
+ break;
+ }
+ case BandwidthIn:
+ state = State.MetricIn;
+ if (obj instanceof Metric) {
+ pathMetrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
+ state = State.BandwidthIn;
+ break;
+ }
+ case MetricIn:
+ state = State.IroIn;
+ if (obj instanceof Iro) {
+ builder.setIro((Iro) obj);
+ break;
+ }
+ case IroIn:
+ state = State.End;
+ break;
+ case End:
+ break;
+ }
+ if (!state.equals(State.End)) {
+ objects.remove(0);
+ }
+ }
+ builder.setMetrics(pathMetrics);
+ }
+
+ private enum State {
+ Init, LspaIn, OfIn, BandwidthIn, MetricIn, IroIn, End
}
@Override
Code Review / bgpcep.git / blobdiff