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         public Message parseMessage(final byte[] buffer) throws PCEPDeserializerException, PCEPDocumentedException {
  91                 if (buffer == null || buffer.length == 0) {
  92                         throw new PCEPDeserializerException("Initiate message cannot be empty.");
  93                 }
  94                 final List<Object> objs = parseObjects(buffer);
  95                 return validate(objs);
  96         }
  97
  98         public Message validate(final List<Object> objects) throws PCEPDeserializerException {
  99                 if (objects == null) {
 100                         throw new IllegalArgumentException("Passed list can't be null.");
 101                 }
 102                 final PcinitiateMessageBuilder builder = new PcinitiateMessageBuilder();
 103                 final PCEPErrorMapping maping = PCEPErrorMapping.getInstance();
 104                 final List<Requests> reqs = Lists.newArrayList();
 105                 Requests req = null;
 106                 while (!objects.isEmpty()) {
 107                         try {
 108                                 req = this.getValidRequest(objects);
 109                                 if (req == null) {
 110                                         break;
 111                                 }
 112                         } catch (final PCEPDocumentedException e) {
 113                                 final PcerrMessageBuilder b = new PcerrMessageBuilder();
 114                                 b.setErrors(Arrays.asList(new ErrorsBuilder().setErrorObject(
 115                                                 new ErrorObjectBuilder().setType(maping.getFromErrorsEnum(e.getError()).type).setValue(
 116                                                                 maping.getFromErrorsEnum(e.getError()).value).build()).build()));
 117                                 return new PcerrBuilder().setPcerrMessage(b.build()).build();
 118                         }
 119                         reqs.add(req);
 120                 }
 121                 builder.setRequests(reqs);
 122                 return new PcinitiateBuilder().setPcinitiateMessage(builder.build()).build();
 123         }
 124
 125         private Requests getValidRequest(final List<Object> objects) throws PCEPDocumentedException {
 126                 final RequestsBuilder builder = new RequestsBuilder();
 127                 builder.setSrp((Srp) objects.get(0));
 128                 objects.remove(0);
 129
 130                 builder.setLsp((Lsp) objects.get(0));
 131                 objects.remove(0);
 132
 133                 final List<Metrics> metrics = Lists.newArrayList();
 134
 135                 Object obj;
 136                 State state = State.Init;
 137                 while (!objects.isEmpty() && !state.equals(State.End)) {
 138                         obj = objects.get(0);
 139
 140                         switch (state) {
 141                         case Init:
 142                                 state = State.EndpointsIn;
 143                                 if (obj instanceof EndpointsObj) {
 144                                         builder.setEndpointsObj((EndpointsObj) obj);
 145                                         break;
 146                                 }
 147                         case EndpointsIn:
 148                                 state = State.EroIn;
 149                                 if (obj instanceof Ero) {
 150                                         builder.setEro((Ero) obj);
 151                                         break;
 152                                 }
 153                         case EroIn:
 154                                 state = State.LspaIn;
 155                                 if (obj instanceof Lspa) {
 156                                         builder.setLspa((Lspa) obj);
 157                                         break;
 158                                 }
 159                         case LspaIn:
 160                                 state = State.BandwidthIn;
 161                                 if (obj instanceof Bandwidth) {
 162                                         builder.setBandwidth((Bandwidth) obj);
 163                                         break;
 164                                 }
 165                         case BandwidthIn:
 166                                 state = State.MetricIn;
 167                                 if (obj instanceof Metric) {
 168                                         metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 169                                         state = State.BandwidthIn;
 170                                         break;
 171                                 }
 172                         case MetricIn:
 173                                 state = State.IroIn;
 174                                 if (obj instanceof Iro) {
 175                                         builder.setIro((Iro) obj);
 176                                         break;
 177                                 }
 178                         case IroIn:
 179                                 state = State.End;
 180                                 break;
 181                         case End:
 182                                 break;
 183                         default:
 184                                 throw new PCEPDocumentedException("Unknown object", ((UnknownObject) obj).getError());
 185                         }
 186                         if (!state.equals(State.End)) {
 187                                 objects.remove(0);
 188                         }
 189                 }
 190                 builder.setMetrics(metrics);
 191                 return builder.build();
 192         }
 193
 194         private enum State {
 195                 Init, EndpointsIn, EroIn, LspaIn, BandwidthIn, MetricIn, IroIn, End
 196         }
 197
 198         @Override
 199         public int getMessageType() {
 200                 return TYPE;
 201         }
 202 }