pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/message/PcinitiateMessageParser.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.Arrays;
  13 import java.util.List;
  14
  15 import org.opendaylight.protocol.pcep.PCEPDeserializerException;
  16 import org.opendaylight.protocol.pcep.PCEPDocumentedException;
  17 import org.opendaylight.protocol.pcep.PCEPErrorMapping;
  18 import org.opendaylight.protocol.pcep.UnknownObject;
  19 import org.opendaylight.protocol.pcep.spi.ObjectHandlerRegistry;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Pcinitiate;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcinitiateBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcinitiateMessage;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.endpoints.object.EndpointsObj;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.object.Lsp;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObjectBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.ErrorsBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcinitiate.message.PcinitiateMessageBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcinitiate.message.pcinitiate.message.Requests;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcinitiate.message.pcinitiate.message.RequestsBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.srp.object.Srp;
  42
  43 import com.google.common.collect.Lists;
  44
  45 /**
  46  * Parser for {@link Pcinitiate}
  47  */
  48 public class PcinitiateMessageParser extends AbstractMessageParser {
  49
  50         public static final int TYPE = 12;
  51
  52         public PcinitiateMessageParser(final ObjectHandlerRegistry registry) {
  53                 super(registry);
  54         }
  55
  56         @Override
  57         public void serializeMessage(final Message message, final ByteBuf buffer) {
  58                 if (!(message instanceof PcinitiateMessage)) {
  59                         throw new IllegalArgumentException("Wrong instance of Message. Passed instance of " + message.getClass()
  60                                         + ". Needed PcinitiateMessage.");
  61                 }
  62                 final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcinitiate.message.PcinitiateMessage init = ((Pcinitiate) message).getPcinitiateMessage();
  63                 for (final Requests req : init.getRequests()) {
  64                         buffer.writeBytes(serializeObject(req.getSrp()));
  65                         buffer.writeBytes(serializeObject(req.getLsp()));
  66                         if (req.getEndpointsObj() != null) {
  67                                 buffer.writeBytes(serializeObject(req.getEndpointsObj()));
  68                         }
  69                         if (req.getEro() != null) {
  70                                 buffer.writeBytes(serializeObject(req.getEro()));
  71                         }
  72                         if (req.getLspa() != null) {
  73                                 buffer.writeBytes(serializeObject(req.getLspa()));
  74                         }
  75                         if (req.getBandwidth() != null) {
  76                                 buffer.writeBytes(serializeObject(req.getBandwidth()));
  77                         }
  78                         if (req.getMetrics() != null && !req.getMetrics().isEmpty()) {
  79                                 for (final Metrics m : req.getMetrics()) {
  80                                         buffer.writeBytes(serializeObject(m.getMetric()));
  81                                 }
  82                         }
  83                         if (req.getIro() != null) {
  84                                 buffer.writeBytes(serializeObject(req.getIro()));
  85                         }
  86                 }
  87         }
  88
  89         @Override
  90         protected Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
  91                 if (objects == null) {
  92                         throw new IllegalArgumentException("Passed list can't be null.");
  93                 }
  94                 final PcinitiateMessageBuilder builder = new PcinitiateMessageBuilder();
  95                 final PCEPErrorMapping maping = PCEPErrorMapping.getInstance();
  96                 final List<Requests> reqs = Lists.newArrayList();
  97                 Requests req = null;
  98                 while (!objects.isEmpty()) {
  99                         try {
 100                                 req = this.getValidRequest(objects);
 101                                 if (req == null) {
 102                                         break;
 103                                 }
 104                         } catch (final PCEPDocumentedException e) {
 105                                 final PcerrMessageBuilder b = new PcerrMessageBuilder();
 106                                 b.setErrors(Arrays.asList(new ErrorsBuilder().setErrorObject(
 107                                                 new ErrorObjectBuilder().setType(maping.getFromErrorsEnum(e.getError()).type).setValue(
 108                                                                 maping.getFromErrorsEnum(e.getError()).value).build()).build()));
 109                                 return new PcerrBuilder().setPcerrMessage(b.build()).build();
 110                         }
 111                         reqs.add(req);
 112                 }
 113                 builder.setRequests(reqs);
 114                 return new PcinitiateBuilder().setPcinitiateMessage(builder.build()).build();
 115         }
 116
 117         private Requests getValidRequest(final List<Object> objects) throws PCEPDocumentedException {
 118                 final RequestsBuilder builder = new RequestsBuilder();
 119                 builder.setSrp((Srp) objects.get(0));
 120                 objects.remove(0);
 121
 122                 builder.setLsp((Lsp) objects.get(0));
 123                 objects.remove(0);
 124
 125                 final List<Metrics> metrics = Lists.newArrayList();
 126
 127                 Object obj;
 128                 State state = State.Init;
 129                 while (!objects.isEmpty() && !state.equals(State.End)) {
 130                         obj = objects.get(0);
 131
 132                         switch (state) {
 133                         case Init:
 134                                 state = State.EndpointsIn;
 135                                 if (obj instanceof EndpointsObj) {
 136                                         builder.setEndpointsObj((EndpointsObj) obj);
 137                                         break;
 138                                 }
 139                         case EndpointsIn:
 140                                 state = State.EroIn;
 141                                 if (obj instanceof Ero) {
 142                                         builder.setEro((Ero) obj);
 143                                         break;
 144                                 }
 145                         case EroIn:
 146                                 state = State.LspaIn;
 147                                 if (obj instanceof Lspa) {
 148                                         builder.setLspa((Lspa) obj);
 149                                         break;
 150                                 }
 151                         case LspaIn:
 152                                 state = State.BandwidthIn;
 153                                 if (obj instanceof Bandwidth) {
 154                                         builder.setBandwidth((Bandwidth) obj);
 155                                         break;
 156                                 }
 157                         case BandwidthIn:
 158                                 state = State.MetricIn;
 159                                 if (obj instanceof Metric) {
 160                                         metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 161                                         state = State.BandwidthIn;
 162                                         break;
 163                                 }
 164                         case MetricIn:
 165                                 state = State.IroIn;
 166                                 if (obj instanceof Iro) {
 167                                         builder.setIro((Iro) obj);
 168                                         break;
 169                                 }
 170                         case IroIn:
 171                                 state = State.End;
 172                                 break;
 173                         case End:
 174                                 break;
 175                         default:
 176                                 throw new PCEPDocumentedException("Unknown object", ((UnknownObject) obj).getError());
 177                         }
 178                         if (!state.equals(State.End)) {
 179                                 objects.remove(0);
 180                         }
 181                 }
 182                 builder.setMetrics(metrics);
 183                 return builder.build();
 184         }
 185
 186         private enum State {
 187                 Init, EndpointsIn, EroIn, LspaIn, BandwidthIn, MetricIn, IroIn, End
 188         }
 189
 190         @Override
 191         public int getMessageType() {
 192                 return TYPE;
 193         }
 194 }