pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/message/PCEPReportMessageParser.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.impl.message;
   9
  10 import io.netty.buffer.ByteBuf;
  11
  12 import java.util.List;
  13
  14 import org.opendaylight.protocol.pcep.PCEPDeserializerException;
  15 import org.opendaylight.protocol.pcep.PCEPDocumentedException;
  16 import org.opendaylight.protocol.pcep.PCEPErrors;
  17 import org.opendaylight.protocol.pcep.UnknownObject;
  18 import org.opendaylight.protocol.pcep.spi.ObjectHandlerRegistry;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Pcrpt;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcrptBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.object.Lsp;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.PcrptMessageBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.pcrpt.message.Reports;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.pcrpt.message.ReportsBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.pcrpt.message.reports.Path;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.pcrpt.message.reports.PathBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reported.route.object.Rro;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.srp.object.Srp;
  38
  39 import com.google.common.collect.Lists;
  40
  41 /**
  42  * Parser for {@link Pcrpt}
  43  */
  44 public class PCEPReportMessageParser extends AbstractMessageParser {
  45
  46         public static final int TYPE = 10;
  47
  48         public PCEPReportMessageParser(final ObjectHandlerRegistry registry) {
  49                 super(registry);
  50         }
  51
  52         @Override
  53         public void serializeMessage(final Message message, final ByteBuf buffer) {
  54                 if (!(message instanceof Pcrpt)) {
  55                         throw new IllegalArgumentException("Wrong instance of Message. Passed instance of " + message.getClass()
  56                                         + ". Nedded PcrptMessage.");
  57                 }
  58                 final Pcrpt msg = (Pcrpt) message;
  59                 final List<Reports> reports = msg.getPcrptMessage().getReports();
  60                 for (final Reports report : reports) {
  61                         if (report.getSrp() != null) {
  62                                 buffer.writeBytes(serializeObject(report.getSrp()));
  63                         }
  64                         buffer.writeBytes(serializeObject(report.getLsp()));
  65                         final Path p = report.getPath();
  66                         if (p != null) {
  67                                 if (p.getEro() != null) {
  68                                         buffer.writeBytes(serializeObject(p.getEro()));
  69                                 }
  70                                 if (p.getLspa() != null) {
  71                                         buffer.writeBytes(serializeObject(p.getLspa()));
  72                                 }
  73                                 if (p.getBandwidth() != null) {
  74                                         buffer.writeBytes(serializeObject(p.getBandwidth()));
  75                                 }
  76                                 if (p.getMetrics() != null && !p.getMetrics().isEmpty()) {
  77                                         for (final Metrics m : p.getMetrics()) {
  78                                                 buffer.writeBytes(serializeObject(m.getMetric()));
  79                                         }
  80                                 }
  81                                 if (p.getIro() != null) {
  82                                         buffer.writeBytes(serializeObject(p.getIro()));
  83                                 }
  84                                 if (p.getRro() != null) {
  85                                         buffer.writeBytes(serializeObject(p.getRro()));
  86                                 }
  87                         }
  88                 }
  89         }
  90
  91         @Override
  92         public Message parseMessage(final byte[] buffer) throws PCEPDeserializerException {
  93                 if (buffer == null || buffer.length == 0) {
  94                         throw new PCEPDeserializerException("Pcrpt message cannot be empty.");
  95                 }
  96                 try {
  97                         final List<Object> objs = parseObjects(buffer);
  98                         return validate(objs);
  99                 } catch (final PCEPDocumentedException e) {
 100                         return createErrorMsg(e.getError());
 101                 }
 102         }
 103
 104         public Message validate(final List<Object> objects) throws PCEPDeserializerException, PCEPDocumentedException {
 105                 if (objects == null) {
 106                         throw new IllegalArgumentException("Passed list can't be null.");
 107                 }
 108                 final List<Reports> reports = Lists.newArrayList();
 109
 110                 while (!objects.isEmpty()) {
 111                         final Reports report = getValidReports(objects);
 112                         if (reports != null) {
 113                                 reports.add(report);
 114                         }
 115                 }
 116
 117                 if (!objects.isEmpty()) {
 118                         if (objects.get(0) instanceof UnknownObject) {
 119                                 throw new PCEPDocumentedException("Unknown object encountered", ((UnknownObject) objects.get(0)).getError());
 120                         }
 121                         throw new PCEPDeserializerException("Unprocessed Objects: " + objects);
 122                 }
 123                 return new PcrptBuilder().setPcrptMessage(new PcrptMessageBuilder().setReports(reports).build()).build();
 124         }
 125
 126         private Reports getValidReports(final List<Object> objects) throws PCEPDocumentedException {
 127                 final ReportsBuilder builder = new ReportsBuilder();
 128                 if (objects.get(0) instanceof Srp) {
 129                         builder.setSrp((Srp) objects.get(0));
 130                         objects.remove(0);
 131                 }
 132                 if (objects.get(0) instanceof Lsp) {
 133                         builder.setLsp((Lsp) objects.get(0));
 134                         objects.remove(0);
 135                 } else {
 136                         throw new PCEPDocumentedException("LSP object missing", PCEPErrors.LSP_MISSING);
 137                 }
 138                 if (!objects.isEmpty()) {
 139                         final PathBuilder pBuilder = new PathBuilder();
 140                         parsePath(objects, pBuilder);
 141                         builder.setPath(pBuilder.build());
 142                 }
 143                 return builder.build();
 144         }
 145
 146         private void parsePath(final List<Object> objects, final PathBuilder builder) throws PCEPDocumentedException {
 147                 final List<Metrics> pathMetrics = Lists.newArrayList();
 148                 Object obj;
 149                 State state = State.Init;
 150                 while (!objects.isEmpty() && !state.equals(State.End)) {
 151                         obj = objects.get(0);
 152                         switch (state) {
 153                         case Init:
 154                                 state = State.EroIn;
 155                                 if (obj instanceof Ero) {
 156                                         builder.setEro((Ero) obj);
 157                                         break;
 158                                 }
 159                         case EroIn:
 160                                 state = State.LspaIn;
 161                                 if (obj instanceof Lspa) {
 162                                         builder.setLspa((Lspa) obj);
 163                                         break;
 164                                 }
 165                         case LspaIn:
 166                                 state = State.BandwidthIn;
 167                                 if (obj instanceof Bandwidth) {
 168                                         builder.setBandwidth((Bandwidth) obj);
 169                                         break;
 170                                 }
 171                         case BandwidthIn:
 172                                 state = State.MetricIn;
 173                                 if (obj instanceof Metric) {
 174                                         pathMetrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 175                                         state = State.BandwidthIn;
 176                                         break;
 177                                 }
 178                         case MetricIn:
 179                                 state = State.IroIn;
 180                                 if (obj instanceof Iro) {
 181                                         builder.setIro((Iro) obj);
 182                                         break;
 183                                 }
 184                         case IroIn:
 185                                 state = State.RroIn;
 186                                 if (obj instanceof Rro) {
 187                                         builder.setRro((Rro) obj);
 188                                         break;
 189                                 }
 190                         case RroIn:
 191                                 state = State.End;
 192                                 break;
 193                         case End:
 194                                 break;
 195                         default:
 196                                 if (obj instanceof UnknownObject) {
 197                                         throw new PCEPDocumentedException("Unknown object", ((UnknownObject) obj).getError());
 198                                 }
 199                         }
 200                         if (!state.equals(State.End)) {
 201                                 objects.remove(0);
 202                         }
 203                 }
 204                 if (!pathMetrics.isEmpty()) {
 205                         builder.setMetrics(pathMetrics);
 206                 }
 207         }
 208
 209         private enum State {
 210                 Init, EroIn, LspaIn, BandwidthIn, MetricIn, IroIn, RroIn, End
 211         }
 212
 213         @Override
 214         public int getMessageType() {
 215                 return TYPE;
 216         }
 217 }