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.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Pcrpt;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PcrptBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.Lsp;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.PcrptMessageBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.Reports;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.ReportsBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.reports.Path;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.reports.PathBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.Srp;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reported.route.object.Rro;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
  41
  42 /**
  43  * Parser for {@link Pcrpt}
  44  */
  45 public class Stateful07PCReportMessageParser extends AbstractMessageParser {
  46
  47     public static final int TYPE = 10;
  48
  49     public Stateful07PCReportMessageParser(final ObjectRegistry registry) {
  50         super(registry);
  51     }
  52
  53     @Override
  54     public void serializeMessage(final Message message, final ByteBuf out) {
  55         Preconditions.checkArgument(message instanceof Pcrpt, "Wrong instance of Message. Passed instance of %s. Need Pcrpt.", message.getClass());
  56         final Pcrpt msg = (Pcrpt) message;
  57         final List<Reports> reports = msg.getPcrptMessage().getReports();
  58         final ByteBuf buffer = Unpooled.buffer();
  59         for (final Reports report : reports) {
  60             serializeReport(report, buffer);
  61         }
  62         MessageUtil.formatMessage(TYPE, buffer, out);
  63     }
  64
  65     protected void serializeReport(final Reports report, final ByteBuf buffer) {
  66         if (report.getSrp() != null) {
  67             serializeObject(report.getSrp(), buffer);
  68         }
  69         serializeObject(report.getLsp(), buffer);
  70         final Path p = report.getPath();
  71         if (p != null) {
  72             serializeObject(p.getEro(), buffer);
  73             serializeObject(p.getLspa(), buffer);
  74             serializeObject(p.getBandwidth(), buffer);
  75             if (p.getMetrics() != null) {
  76                 for (final Metrics m : p.getMetrics()) {
  77                     serializeObject(m.getMetric(), buffer);
  78                 }
  79             }
  80             serializeObject(p.getIro(), buffer);
  81             serializeObject(p.getRro(), buffer);
  82         }
  83     }
  84
  85     @Override
  86     public Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
  87         Preconditions.checkArgument(objects != null, "Passed list can't be null.");
  88         if (objects.isEmpty()) {
  89             throw new PCEPDeserializerException("Pcrpt message cannot be empty.");
  90         }
  91
  92         final List<Reports> reports = Lists.newArrayList();
  93
  94         while (!objects.isEmpty()) {
  95             final Reports report = getValidReports(objects, errors);
  96             if (report != null) {
  97                 reports.add(report);
  98             }
  99         }
 100         return new PcrptBuilder().setPcrptMessage(new PcrptMessageBuilder().setReports(reports).build()).build();
 101     }
 102
 103     protected Reports getValidReports(final List<Object> objects, final List<Message> errors) {
 104         boolean isValid = true;
 105         final ReportsBuilder builder = new ReportsBuilder();
 106         if (objects.get(0) instanceof Srp) {
 107             builder.setSrp((Srp) objects.get(0));
 108             objects.remove(0);
 109         }
 110         if (objects.get(0) instanceof Lsp) {
 111             builder.setLsp((Lsp) objects.get(0));
 112             objects.remove(0);
 113         } else {
 114             errors.add(createErrorMsg(PCEPErrors.LSP_MISSING, Optional.<Rp>absent()));
 115             isValid = false;
 116         }
 117         if (!objects.isEmpty()) {
 118             final PathBuilder pBuilder = new PathBuilder();
 119             if (objects.get(0) instanceof Ero) {
 120                 pBuilder.setEro((Ero) objects.get(0));
 121                 objects.remove(0);
 122             } else {
 123                 errors.add(createErrorMsg(PCEPErrors.ERO_MISSING, Optional.<Rp>absent()));
 124                 isValid = false;
 125             }
 126             parsePath(objects, pBuilder);
 127             builder.setPath(pBuilder.build());
 128         }
 129         if(isValid) {
 130             return builder.build();
 131         }
 132         return null;
 133     }
 134
 135     private void parsePath(final List<Object> objects, final PathBuilder builder) {
 136         final List<Metrics> pathMetrics = Lists.newArrayList();
 137         Object obj;
 138         State state = State.INIT;
 139         while (!objects.isEmpty() && !state.equals(State.END)) {
 140             obj = objects.get(0);
 141             state = insertObject(state, obj, builder, pathMetrics);
 142             if (!state.equals(State.END)) {
 143                 objects.remove(0);
 144             }
 145         }
 146         if (!pathMetrics.isEmpty()) {
 147             builder.setMetrics(pathMetrics);
 148         }
 149     }
 150
 151     private State insertObject(final State state, final Object obj, final PathBuilder builder, final List<Metrics> pathMetrics) {
 152         switch (state) {
 153         case INIT:
 154             if (obj instanceof Lspa) {
 155                 builder.setLspa((Lspa) obj);
 156                 return State.LSPA_IN;
 157             }
 158         case LSPA_IN:
 159             if (obj instanceof Bandwidth) {
 160                 builder.setBandwidth((Bandwidth) obj);
 161                 return State.BANDWIDTH_IN;
 162             }
 163         case BANDWIDTH_IN:
 164             if (obj instanceof Metric) {
 165                 pathMetrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 166                 return State.BANDWIDTH_IN;
 167             }
 168         case METRIC_IN:
 169             if (obj instanceof Iro) {
 170                 builder.setIro((Iro) obj);
 171                 return State.IRO_IN;
 172             }
 173         case IRO_IN:
 174             if (obj instanceof Rro) {
 175                 builder.setRro((Rro) obj);
 176                 return State.RRO_IN;
 177             }
 178         case RRO_IN:
 179         case END:
 180             return State.END;
 181         default:
 182             return state;
 183         }
 184     }
 185
 186     private enum State {
 187         INIT, LSPA_IN, BANDWIDTH_IN, METRIC_IN, IRO_IN, RRO_IN, END
 188     }
 189 }