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         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             if (p.getLspa() != null) {
  74                 serializeObject(p.getLspa(), buffer);
  75             }
  76             if (p.getBandwidth() != null) {
  77                 serializeObject(p.getBandwidth(), buffer);
  78             }
  79             if (p.getMetrics() != null && !p.getMetrics().isEmpty()) {
  80                 for (final Metrics m : p.getMetrics()) {
  81                     serializeObject(m.getMetric(), buffer);
  82                 }
  83             }
  84             if (p.getIro() != null) {
  85                 serializeObject(p.getIro(), buffer);
  86             }
  87             if (p.getRro() != null) {
  88                 serializeObject(p.getRro(), buffer);
  89             }
  90         }
  91     }
  92
  93     @Override
  94     public Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
  95         if (objects == null) {
  96             throw new IllegalArgumentException("Passed list can't be null.");
  97         }
  98         if (objects.isEmpty()) {
  99             throw new PCEPDeserializerException("Pcrpt message cannot be empty.");
 100         }
 101
 102         final List<Reports> reports = Lists.newArrayList();
 103
 104         while (!objects.isEmpty()) {
 105             final Reports report = getValidReports(objects, errors);
 106             if (report != null) {
 107                 reports.add(report);
 108             }
 109         }
 110         return new PcrptBuilder().setPcrptMessage(new PcrptMessageBuilder().setReports(reports).build()).build();
 111     }
 112
 113     protected Reports getValidReports(final List<Object> objects, final List<Message> errors) {
 114         boolean isValid = true;
 115         final ReportsBuilder builder = new ReportsBuilder();
 116         if (objects.get(0) instanceof Srp) {
 117             builder.setSrp((Srp) objects.get(0));
 118             objects.remove(0);
 119         }
 120         if (objects.get(0) instanceof Lsp) {
 121             builder.setLsp((Lsp) objects.get(0));
 122             objects.remove(0);
 123         } else {
 124             errors.add(createErrorMsg(PCEPErrors.LSP_MISSING, Optional.<Rp>absent()));
 125             isValid = false;
 126         }
 127         if (!objects.isEmpty()) {
 128             final PathBuilder pBuilder = new PathBuilder();
 129             if (objects.get(0) instanceof Ero) {
 130                 pBuilder.setEro((Ero) objects.get(0));
 131                 objects.remove(0);
 132             } else {
 133                 errors.add(createErrorMsg(PCEPErrors.ERO_MISSING, Optional.<Rp>absent()));
 134                 isValid = false;
 135             }
 136             parsePath(objects, pBuilder);
 137             builder.setPath(pBuilder.build());
 138         }
 139         if(isValid) {
 140             return builder.build();
 141         }
 142         return null;
 143     }
 144
 145     private void parsePath(final List<Object> objects, final PathBuilder builder) {
 146         final List<Metrics> pathMetrics = Lists.newArrayList();
 147         Object obj;
 148         State state = State.Init;
 149         while (!objects.isEmpty() && !state.equals(State.End)) {
 150             obj = objects.get(0);
 151             switch (state) {
 152             case Init:
 153                 state = State.LspaIn;
 154                 if (obj instanceof Lspa) {
 155                     builder.setLspa((Lspa) obj);
 156                     break;
 157                 }
 158             case LspaIn:
 159                 state = State.BandwidthIn;
 160                 if (obj instanceof Bandwidth) {
 161                     builder.setBandwidth((Bandwidth) obj);
 162                     break;
 163                 }
 164             case BandwidthIn:
 165                 state = State.MetricIn;
 166                 if (obj instanceof Metric) {
 167                     pathMetrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 168                     state = State.BandwidthIn;
 169                     break;
 170                 }
 171             case MetricIn:
 172                 state = State.IroIn;
 173                 if (obj instanceof Iro) {
 174                     builder.setIro((Iro) obj);
 175                     break;
 176                 }
 177             case IroIn:
 178                 state = State.RroIn;
 179                 if (obj instanceof Rro) {
 180                     builder.setRro((Rro) obj);
 181                     break;
 182                 }
 183             case RroIn:
 184                 state = State.End;
 185                 break;
 186             case End:
 187                 break;
 188             }
 189             if (!state.equals(State.End)) {
 190                 objects.remove(0);
 191             }
 192         }
 193         if (!pathMetrics.isEmpty()) {
 194             builder.setMetrics(pathMetrics);
 195         }
 196     }
 197
 198     private enum State {
 199         Init, LspaIn, BandwidthIn, MetricIn, IroIn, RroIn, End
 200     }
 201 }