pcep/segment-routing/src/main/java/org/opendaylight/protocol/pcep/segment/routing02/SrPcRepMessageParser.java

   1 /*
   2  * Copyright (c) 2014 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
   9 package org.opendaylight.protocol.pcep.segment.routing02;
  10
  11 import com.google.common.collect.Lists;
  12 import io.netty.buffer.ByteBuf;
  13 import java.util.List;
  14 import org.opendaylight.protocol.pcep.impl.message.PCEPReplyMessageParser;
  15 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
  16 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.lsp.setup.type._01.rev140507.Tlvs1;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.lsp.setup.type._01.rev140507.Tlvs2;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.Replies;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.RepliesBuilder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.Result;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.SuccessCase;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.SuccessCaseBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.SuccessBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.success.Paths;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.success.PathsBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
  31
  32 public class SrPcRepMessageParser extends PCEPReplyMessageParser {
  33
  34     public SrPcRepMessageParser(ObjectRegistry registry) {
  35         super(registry);
  36     }
  37
  38     @Override
  39     protected void serializeReply(Replies reply, ByteBuf buffer) {
  40         final Rp rp = reply.getRp();
  41         if(isSegmentRoutingPath(rp)) {
  42             serializeObject(rp, buffer);
  43             if (reply.getResult() != null) {
  44                 if (reply.getResult() instanceof SuccessCase) {
  45                     final SuccessCase s = (SuccessCase) reply.getResult();
  46                     for (final Paths p : s.getSuccess().getPaths()) {
  47                         final Ero ero = p.getEro();
  48                         serializeObject(ero, buffer);
  49                     }
  50                 }
  51             }
  52         } else {
  53             super.serializeReply(reply, buffer);
  54         }
  55     }
  56
  57     @Override
  58     protected Replies getValidReply(List<Object> objects, List<Message> errors) {
  59         if (!(objects.get(0) instanceof Rp)) {
  60             errors.add(createErrorMsg(PCEPErrors.RP_MISSING));
  61             return null;
  62         }
  63         final Rp rp = (Rp) objects.get(0);
  64         if(isSegmentRoutingPath(rp)) {
  65             objects.remove(0);
  66             Result res = null;
  67             if (!objects.isEmpty()) {
  68                 if (objects.get(0) instanceof Ero) {
  69                     final SuccessBuilder builder = new SuccessBuilder();
  70                     final List<Paths> paths = Lists.newArrayList();
  71                     final PathsBuilder pBuilder = new PathsBuilder();
  72                     while (!objects.isEmpty()) {
  73                         final Object object = objects.get(0);
  74                         if(object instanceof Ero) {
  75                             final Ero ero = (Ero) object;
  76                             final PCEPErrors error = SrEroUtil.validateSrEroSubobjects(ero);
  77                             if(error != null) {
  78                                 errors.add(createErrorMsg(error));
  79                                 return null;
  80                             } else {
  81                                 paths.add(pBuilder.setEro(ero).build());
  82                             }
  83                         }
  84                         objects.remove(0);
  85                     }
  86                     builder.setPaths(paths);
  87                     res = new SuccessCaseBuilder().setSuccess(builder.build()).build();
  88                 }
  89             }
  90             return new RepliesBuilder().setRp(rp).setResult(res).build();
  91         }
  92         return super.getValidReply(objects, errors);
  93     }
  94
  95     private boolean isSegmentRoutingPath(final Rp rp) {
  96         if(rp.getTlvs() != null) {
  97             return SrEroUtil.isPst(rp.getTlvs().getAugmentation(Tlvs1.class)) || SrEroUtil.isPst(rp.getTlvs().getAugmentation(Tlvs2.class));
  98         }
  99         return false;
 100     }
 101 }