pcep/topology-provider/src/main/java/org/opendaylight/bgpcep/pcep/topology/provider/Stateful07TopologySessionListener.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.bgpcep.pcep.topology.provider;
   9
  10 import com.google.common.base.Charsets;
  11 import com.google.common.base.Function;
  12 import com.google.common.base.Optional;
  13 import com.google.common.base.Preconditions;
  14 import com.google.common.util.concurrent.AsyncFunction;
  15 import com.google.common.util.concurrent.Futures;
  16 import com.google.common.util.concurrent.ListenableFuture;
  17 import java.net.InetAddress;
  18 import java.nio.ByteBuffer;
  19 import java.util.ArrayList;
  20 import java.util.Collections;
  21 import java.util.List;
  22 import java.util.concurrent.atomic.AtomicLong;
  23 import org.opendaylight.controller.config.yang.pcep.topology.provider.PeerCapabilities;
  24 import org.opendaylight.protocol.pcep.PCEPSession;
  25 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.PcinitiateBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Srp1;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Srp1Builder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Stateful1;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.pcinitiate.message.PcinitiateMessageBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.pcinitiate.message.pcinitiate.message.Requests;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.pcinitiate.message.pcinitiate.message.RequestsBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Arguments1;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Arguments2;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Arguments3;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.OperationalStatus;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Path1;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Path1Builder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PcrptMessage;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PcupdBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PlspId;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SrpIdNumber;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.StatefulTlv1;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.StatefulTlv1Builder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.Lsp;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcerr.pcerr.message.error.type.StatefulCase;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcerr.pcerr.message.error.type.stateful._case.stateful.Srps;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.Reports;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcupd.message.PcupdMessageBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcupd.message.pcupd.message.UpdatesBuilder;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcupd.message.pcupd.message.updates.PathBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.Srp;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.SrpBuilder;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.stateful.capability.tlv.Stateful;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathNameBuilder;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcerrMessage;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.Tlvs;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.path.setup.type.tlv.PathSetupType;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.AddLspArgs;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.EnsureLspOperationalInput;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.LspId;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.OperationResult;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.PccSyncState;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.RemoveLspArgs;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspArgs;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.PathComputationClientBuilder;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLsp;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLspBuilder;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.StatefulTlvBuilder;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.reported.lsp.Path;
  76 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  77 import org.slf4j.Logger;
  78 import org.slf4j.LoggerFactory;
  79
  80 class Stateful07TopologySessionListener extends AbstractTopologySessionListener<SrpIdNumber, PlspId> {
  81     private static final Logger LOG = LoggerFactory.getLogger(Stateful07TopologySessionListener.class);
  82
  83     private final AtomicLong requestId = new AtomicLong(1L);
  84
  85     /**
  86      * Creates a new stateful topology session listener for given server session manager.
  87      *
  88      * @param serverSessionManager
  89      */
  90     Stateful07TopologySessionListener(final ServerSessionManager serverSessionManager) {
  91         super(serverSessionManager);
  92     }
  93
  94     @Override
  95     protected void onSessionUp(final PCEPSession session, final PathComputationClientBuilder pccBuilder) {
  96         final InetAddress peerAddress = session.getRemoteAddress();
  97
  98         final Tlvs tlvs = session.getRemoteTlvs();
  99         if (tlvs != null && tlvs.getAugmentation(Tlvs1.class) != null) {
 100             final Stateful stateful = tlvs.getAugmentation(Tlvs1.class).getStateful();
 101             if (stateful != null) {
 102                 getSessionListenerState().setPeerCapabilities(getCapabilities(stateful));
 103                 pccBuilder.setReportedLsp(Collections.<ReportedLsp> emptyList());
 104                 pccBuilder.setStateSync(PccSyncState.InitialResync);
 105                 pccBuilder.setStatefulTlv(new StatefulTlvBuilder().addAugmentation(StatefulTlv1.class,
 106                     new StatefulTlv1Builder(tlvs.getAugmentation(Tlvs1.class)).build()).build());
 107             } else {
 108                 LOG.debug("Peer {} does not advertise stateful TLV", peerAddress);
 109             }
 110         } else {
 111             LOG.debug("Peer {} does not advertise stateful TLV", peerAddress);
 112         }
 113     }
 114
 115     private boolean handleErrorMessage(final PcerrMessage message) {
 116         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessage errMsg = message.getPcerrMessage();
 117         if (errMsg.getErrorType() instanceof StatefulCase) {
 118             final StatefulCase stat = (StatefulCase)errMsg.getErrorType();
 119             for (final Srps srps : stat.getStateful().getSrps()) {
 120                 final SrpIdNumber id = srps.getSrp().getOperationId();
 121                 if (id.getValue() != 0) {
 122                     final PCEPRequest req = removeRequest(id);
 123                     if (req != null) {
 124                         req.done(OperationResults.createFailed(errMsg.getErrors()));
 125                     } else {
 126                         LOG.warn("Request ID {} not found in outstanding DB", id);
 127                     }
 128                 }
 129             }
 130         } else {
 131             LOG.warn("Unhandled PCErr message {}.", errMsg);
 132             return true;
 133         }
 134         return false;
 135     }
 136
 137     private boolean isSolicited(final Srp srp, final Lsp lsp, final MessageContext ctx, final ReportedLspBuilder rlb) {
 138         if (srp == null) {
 139             return false;
 140         }
 141         final SrpIdNumber id = srp.getOperationId();
 142         if (id.getValue() == 0) {
 143             return false;
 144         }
 145         switch (lsp.getOperational()) {
 146         case Active:
 147         case Down:
 148         case Up:
 149             final PCEPRequest req = removeRequest(id);
 150             if (req != null) {
 151                 LOG.debug("Request {} resulted in LSP operational state {}", id, lsp.getOperational());
 152                 rlb.setMetadata(req.getMetadata());
 153                 ctx.resolveRequest(req);
 154             } else {
 155                 LOG.warn("Request ID {} not found in outstanding DB", id);
 156             }
 157             break;
 158         case GoingDown:
 159         case GoingUp:
 160             // These are transitive states, so we don't have to do anything, as they will be followed
 161             // up...
 162             break;
 163         default:
 164             break;
 165         }
 166         return true;
 167     }
 168
 169     private boolean manageNextReport(final Reports report, final MessageContext ctx) {
 170         final Lsp lsp = report.getLsp();
 171         final PlspId plspid = lsp.getPlspId();
 172         if (!lsp.isSync() && (lsp.getPlspId() == null || plspid.getValue() == 0)) {
 173             stateSynchronizationAchieved(ctx);
 174             return true;
 175         }
 176         final ReportedLspBuilder rlb = new ReportedLspBuilder();
 177         boolean solicited = false;
 178         final Srp srp = report.getSrp();
 179         solicited = isSolicited(srp, lsp, ctx, rlb);
 180
 181         // if remove flag is set in SRP object, remove the tunnel immediately
 182         if (solicited && srp.getAugmentation(Srp1.class) != null) {
 183             final Srp1 initiatedSrp = srp.getAugmentation(Srp1.class);
 184             if (initiatedSrp.isRemove()) {
 185                 super.removeLsp(ctx, plspid);
 186                 return false;
 187             }
 188         }
 189         rlb.setPath(Collections.singletonList(buildPath(report, srp, lsp)));
 190
 191         String name = lookupLspName(plspid);
 192         if (report.getLsp().getTlvs() != null && report.getLsp().getTlvs().getSymbolicPathName() != null) {
 193             name = Charsets.UTF_8.decode(ByteBuffer.wrap(report.getLsp().getTlvs().getSymbolicPathName().getPathName().getValue())).toString();
 194         }
 195         updateLsp(ctx, plspid, name, rlb, solicited, lsp.isRemove());
 196         LOG.debug("LSP {} updated", lsp);
 197         return true;
 198     }
 199
 200     private Path buildPath(final Reports report, final Srp srp, final Lsp lsp) {
 201         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.reported.lsp.PathBuilder pb = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.reported.lsp.PathBuilder();
 202         if (report.getPath() != null) {
 203             pb.fieldsFrom(report.getPath());
 204         }
 205         // LSP is mandatory (if there is none, parser will throw an exception)
 206         // this is to ensure a path will be created at any rate
 207         final Path1Builder p1Builder = new Path1Builder();
 208         p1Builder.setLsp(report.getLsp());
 209         final PathSetupType pst;
 210         if (srp != null && srp.getTlvs() != null && srp.getTlvs().getPathSetupType() != null) {
 211             pst = srp.getTlvs().getPathSetupType();
 212             p1Builder.setPathSetupType(pst);
 213         } else {
 214             pst = null;
 215         }
 216         pb.addAugmentation(Path1.class, p1Builder.build());
 217         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.Tlvs tlvs = report.getLsp().getTlvs();
 218         if (tlvs != null) {
 219             if (tlvs.getLspIdentifiers() != null) {
 220                 pb.setLspId(tlvs.getLspIdentifiers().getLspId());
 221             } else if (!isDefaultPST(pst)) {
 222                 pb.setLspId(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId(lsp.getPlspId().getValue()));
 223             }
 224         }
 225         return pb.build();
 226     }
 227
 228     @Override
 229     protected boolean onMessage(final MessageContext ctx, final Message message) {
 230         if (message instanceof PcerrMessage) {
 231             return handleErrorMessage((PcerrMessage) message);
 232         }
 233         if (!(message instanceof PcrptMessage)) {
 234             return true;
 235         }
 236         getSessionListenerState().updateLastReceivedRptMsg();
 237         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.PcrptMessage rpt = ((PcrptMessage) message).getPcrptMessage();
 238         for (final Reports report : rpt.getReports()) {
 239             if (!manageNextReport(report, ctx)) {
 240                 return false;
 241             }
 242         }
 243         return false;
 244     }
 245
 246     private SrpIdNumber nextRequest() {
 247         return new SrpIdNumber(this.requestId.getAndIncrement());
 248     }
 249
 250     private class AddFunction implements AsyncFunction<Optional<ReportedLsp>, OperationResult>  {
 251
 252         private final AddLspArgs input;
 253         private final InstanceIdentifier<ReportedLsp> lsp;
 254
 255         public AddFunction(final AddLspArgs input, final InstanceIdentifier<ReportedLsp> lsp) {
 256             this.input = input;
 257             this.lsp = lsp;
 258         }
 259
 260         @Override
 261         public ListenableFuture<OperationResult> apply(final Optional<ReportedLsp> rep) {
 262             if (rep.isPresent()) {
 263                 LOG.debug("Node {} already contains lsp {} at {}", this.input.getNode(), this.input.getName(), this.lsp);
 264                 return OperationResults.createUnsent(PCEPErrors.USED_SYMBOLIC_PATH_NAME).future();
 265             }
 266             if (!getPeerCapabilities().getInstantiation()) {
 267                 return OperationResults.createUnsent(PCEPErrors.CAPABILITY_NOT_SUPPORTED).future();
 268             }
 269
 270             // Build the request
 271             final RequestsBuilder rb = new RequestsBuilder();
 272             final Arguments2 args = this.input.getArguments().getAugmentation(Arguments2.class);
 273             Preconditions.checkArgument(args != null, "Input is missing operational tag.");
 274             final Lsp inputLsp = args.getLsp();
 275             Preconditions.checkArgument(inputLsp != null, "Reported LSP does not contain LSP object.");
 276
 277             rb.fieldsFrom(this.input.getArguments());
 278
 279             final TlvsBuilder tlvsBuilder;
 280             if (inputLsp.getTlvs() != null) {
 281                 tlvsBuilder = new TlvsBuilder(inputLsp.getTlvs());
 282             } else {
 283                 tlvsBuilder = new TlvsBuilder();
 284             }
 285             tlvsBuilder.setSymbolicPathName(
 286                 new SymbolicPathNameBuilder().setPathName(new SymbolicPathName(this.input.getName().getBytes(Charsets.UTF_8))).build());
 287
 288             final SrpBuilder srpBuilder = new SrpBuilder();
 289             srpBuilder.setOperationId(nextRequest());
 290             srpBuilder.setProcessingRule(Boolean.TRUE);
 291             if (!isDefaultPST(args.getPathSetupType())) {
 292                 srpBuilder.setTlvs(
 293                         new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.srp.TlvsBuilder()
 294                             .setPathSetupType(args.getPathSetupType()).build());
 295             }
 296             rb.setSrp(srpBuilder.build());
 297
 298             rb.setLsp(new LspBuilder().setAdministrative(inputLsp.isAdministrative()).setDelegate(inputLsp.isDelegate()).setPlspId(
 299                 new PlspId(0L)).setTlvs(tlvsBuilder.build()).build());
 300
 301             final PcinitiateMessageBuilder ib = new PcinitiateMessageBuilder(MESSAGE_HEADER);
 302             ib.setRequests(Collections.singletonList(rb.build()));
 303
 304             // Send the message
 305             return sendMessage(new PcinitiateBuilder().setPcinitiateMessage(ib.build()).build(), rb.getSrp().getOperationId(),
 306                 this.input.getArguments().getMetadata());
 307         }
 308     }
 309
 310     @Override
 311     public synchronized ListenableFuture<OperationResult> addLsp(final AddLspArgs input) {
 312         Preconditions.checkArgument(input != null && input.getName() != null && input.getNode() != null && input.getArguments() != null, MISSING_XML_TAG);
 313         LOG.trace("AddLspArgs {}", input);
 314         // Make sure there is no such LSP
 315         final InstanceIdentifier<ReportedLsp> lsp = lspIdentifier(input.getName());
 316         final ListenableFuture<Optional<ReportedLsp>> f = readOperationalData(lsp);
 317         if (f == null) {
 318             return OperationResults.createUnsent(PCEPErrors.LSP_INTERNAL_ERROR).future();
 319         }
 320         return Futures.transform(f, new AddFunction(input, lsp));
 321     }
 322
 323     @Override
 324     public synchronized ListenableFuture<OperationResult> removeLsp(final RemoveLspArgs input) {
 325         Preconditions.checkArgument(input != null && input.getName() != null && input.getNode() != null, MISSING_XML_TAG);
 326         LOG.trace("RemoveLspArgs {}", input);
 327         // Make sure the LSP exists, we need it for PLSP-ID
 328         final InstanceIdentifier<ReportedLsp> lsp = lspIdentifier(input.getName());
 329         final ListenableFuture<Optional<ReportedLsp>> f = readOperationalData(lsp);
 330         if (f == null) {
 331             return OperationResults.createUnsent(PCEPErrors.LSP_INTERNAL_ERROR).future();
 332         }
 333         return Futures.transform(f, new AsyncFunction<Optional<ReportedLsp>, OperationResult>() {
 334             @Override
 335             public ListenableFuture<OperationResult> apply(final Optional<ReportedLsp> rep) {
 336                 final Lsp reportedLsp = validateReportedLsp(rep, input);
 337                 if (reportedLsp == null) {
 338                     return OperationResults.createUnsent(PCEPErrors.UNKNOWN_PLSP_ID).future();
 339                 }
 340                 final PcinitiateMessageBuilder ib = new PcinitiateMessageBuilder(MESSAGE_HEADER);
 341                 final Requests rb = buildRequest(rep, reportedLsp);
 342                 ib.setRequests(Collections.singletonList(rb));
 343                 return sendMessage(new PcinitiateBuilder().setPcinitiateMessage(ib.build()).build(), rb.getSrp().getOperationId(), null);
 344             }
 345         });
 346     }
 347
 348     private Requests buildRequest(final Optional<ReportedLsp> rep, final Lsp reportedLsp) {
 349         // Build the request and send it
 350         final RequestsBuilder rb = new RequestsBuilder();
 351         final SrpBuilder srpBuilder = new SrpBuilder().addAugmentation(Srp1.class, new Srp1Builder().setRemove(Boolean.TRUE).build()).setOperationId(nextRequest()).setProcessingRule(Boolean.TRUE);
 352         final Optional<PathSetupType> maybePST = getPST(rep);
 353         if (maybePST.isPresent()) {
 354             srpBuilder.setTlvs(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.srp.TlvsBuilder()
 355                 .setPathSetupType(maybePST.get())
 356                 .build());
 357         }
 358         rb.setSrp(srpBuilder.build());
 359         rb.setLsp(new LspBuilder().setRemove(Boolean.FALSE).setPlspId(reportedLsp.getPlspId()).setDelegate(reportedLsp.isDelegate()).build());
 360         return rb.build();
 361     }
 362
 363     private class UpdateFunction implements AsyncFunction<Optional<ReportedLsp>, OperationResult>  {
 364
 365         private final UpdateLspArgs input;
 366
 367         public UpdateFunction(final UpdateLspArgs input) {
 368             this.input = input;
 369         }
 370
 371         @Override
 372         public ListenableFuture<OperationResult> apply(final Optional<ReportedLsp> rep) {
 373             final Lsp reportedLsp = validateReportedLsp(rep, this.input);
 374             if (reportedLsp == null) {
 375                 return OperationResults.createUnsent(PCEPErrors.UNKNOWN_PLSP_ID).future();
 376             }
 377             // create mandatory objects
 378             final Arguments3 args = this.input.getArguments().getAugmentation(Arguments3.class);
 379             final SrpBuilder srpBuilder = new SrpBuilder();
 380             srpBuilder.setOperationId(nextRequest());
 381             srpBuilder.setProcessingRule(Boolean.TRUE);
 382             if (args != null && args.getPathSetupType() != null) {
 383                 if (!isDefaultPST(args.getPathSetupType())) {
 384                     srpBuilder.setTlvs(
 385                             new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.srp.TlvsBuilder()
 386                                 .setPathSetupType(args.getPathSetupType()).build());
 387                 }
 388             } else {
 389                 final Optional<PathSetupType> maybePST = getPST(rep);
 390                 if (maybePST.isPresent()) {
 391                     srpBuilder.setTlvs(
 392                             new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.srp.TlvsBuilder()
 393                                 .setPathSetupType(maybePST.get()).build());
 394                 }
 395             }
 396             final Srp srp = srpBuilder.build();
 397             final Lsp inputLsp = args.getLsp();
 398             final Lsp lsp = (inputLsp != null) ?
 399                 new LspBuilder().setPlspId(reportedLsp.getPlspId()).setDelegate((inputLsp.isDelegate() != null) ? inputLsp.isDelegate() : false).setTlvs(inputLsp.getTlvs()).setAdministrative((inputLsp.isAdministrative() != null) ? inputLsp.isAdministrative() : false).build()
 400                 : new LspBuilder().setPlspId(reportedLsp.getPlspId()).build();
 401             final Message msg = redelegate(reportedLsp.isDelegate(), srp, lsp, this.input);
 402             return sendMessage(msg, srp.getOperationId(), this.input.getArguments().getMetadata());
 403         }
 404     }
 405
 406     private Message redelegate(final Boolean isDelegate, final Srp srp, final Lsp lsp, final UpdateLspArgs input) {
 407         // the D bit that was reported decides the type of PCE message sent
 408         Preconditions.checkNotNull(isDelegate);
 409         if (isDelegate) {
 410             // we already have delegation, send update
 411             final UpdatesBuilder rb = new UpdatesBuilder();
 412             rb.setSrp(srp);
 413             rb.setLsp(lsp);
 414             final PathBuilder pb = new PathBuilder();
 415             pb.fieldsFrom(input.getArguments());
 416             rb.setPath(pb.build());
 417             final PcupdMessageBuilder ub = new PcupdMessageBuilder(MESSAGE_HEADER);
 418             ub.setUpdates(Collections.singletonList(rb.build()));
 419             return new PcupdBuilder().setPcupdMessage(ub.build()).build();
 420         }
 421         // we want to revoke delegation, different type of message
 422         // is sent because of specification by Siva
 423         // this message is also sent, when input delegate bit is set to 0
 424         // generating an error in PCC
 425         final List<Requests> reqs = new ArrayList<>();
 426         reqs.add(new RequestsBuilder().setSrp(srp).setLsp(lsp).build());
 427         final PcinitiateMessageBuilder ib = new PcinitiateMessageBuilder();
 428         ib.setRequests(reqs);
 429         return new PcinitiateBuilder().setPcinitiateMessage(ib.build()).build();
 430     }
 431
 432     @Override
 433     public synchronized ListenableFuture<OperationResult> updateLsp(final UpdateLspArgs input) {
 434         Preconditions.checkArgument(input != null && input.getName() != null && input.getNode() != null && input.getArguments() != null, MISSING_XML_TAG);
 435         LOG.trace("UpdateLspArgs {}", input);
 436         // Make sure the LSP exists
 437         final InstanceIdentifier<ReportedLsp> lsp = lspIdentifier(input.getName());
 438         final ListenableFuture<Optional<ReportedLsp>> f = readOperationalData(lsp);
 439         if (f == null) {
 440             return OperationResults.createUnsent(PCEPErrors.LSP_INTERNAL_ERROR).future();
 441         }
 442         return Futures.transform(f, new UpdateFunction(input));
 443     }
 444
 445     @Override
 446     public synchronized ListenableFuture<OperationResult> ensureLspOperational(final EnsureLspOperationalInput input) {
 447         Preconditions.checkArgument(input != null && input.getName() != null && input.getNode() != null && input.getArguments() != null, MISSING_XML_TAG);
 448         final OperationalStatus op;
 449         final Arguments1 aa = input.getArguments().getAugmentation(Arguments1.class);
 450         if (aa != null) {
 451             op = aa.getOperational();
 452         } else {
 453             op = null;
 454         }
 455
 456         // Make sure the LSP exists
 457         final InstanceIdentifier<ReportedLsp> lsp = lspIdentifier(input.getName());
 458         LOG.debug("Checking if LSP {} has operational state {}", lsp, op);
 459         final ListenableFuture<Optional<ReportedLsp>> f = readOperationalData(lsp);
 460         if (f == null) {
 461             return OperationResults.createUnsent(PCEPErrors.LSP_INTERNAL_ERROR).future();
 462         }
 463         return listenableFuture(f, input, op);
 464     }
 465
 466     private ListenableFuture<OperationResult> listenableFuture(final ListenableFuture<Optional<ReportedLsp>> f, final EnsureLspOperationalInput input, final OperationalStatus op) {
 467         return Futures.transform(f, new Function<Optional<ReportedLsp>, OperationResult>() {
 468             @Override
 469             public OperationResult apply(final Optional<ReportedLsp> rep) {
 470                 if (!rep.isPresent()) {
 471                     LOG.debug("Node {} does not contain LSP {}", input.getNode(), input.getName());
 472                     return OperationResults.UNSENT;
 473                 }
 474                 // check if at least one of the paths has the same status as requested
 475                 for (final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.reported.lsp.Path p : rep.get().getPath()) {
 476                     final Path1 p1 = p.getAugmentation(Path1.class);
 477                     if (p1 == null) {
 478                         LOG.warn("Node {} LSP {} does not contain data", input.getNode(), input.getName());
 479                         return OperationResults.UNSENT;
 480                     }
 481                     if (op.equals(p1.getLsp().getOperational())) {
 482                         return OperationResults.SUCCESS;
 483                     }
 484                 }
 485                 return OperationResults.UNSENT;
 486             }
 487         });
 488     }
 489
 490     @Override
 491     protected Lsp validateReportedLsp(final Optional<ReportedLsp> rep, final LspId input) {
 492         if (!rep.isPresent()) {
 493             LOG.debug("Node {} does not contain LSP {}", input.getNode(), input.getName());
 494             return null;
 495         }
 496         // it doesn't matter how many lsps there are in the path list, we only need data that is the same in each path
 497         final Path1 ra = rep.get().getPath().get(0).getAugmentation(Path1.class);
 498         Preconditions.checkState(ra != null, "Reported LSP reported null from data-store.");
 499         final Lsp reportedLsp = ra.getLsp();
 500         Preconditions.checkState(reportedLsp != null, "Reported LSP does not contain LSP object.");
 501         return reportedLsp;
 502     }
 503
 504     private Optional<PathSetupType> getPST(final Optional<ReportedLsp> rep) {
 505         if (rep.isPresent()) {
 506             final Path1 path1 = rep.get().getPath().get(0).getAugmentation(Path1.class);
 507             if (path1 != null) {
 508                 final PathSetupType pst = path1.getPathSetupType();
 509                 if (!isDefaultPST(pst)) {
 510                     return Optional.of(pst);
 511                 }
 512             }
 513         }
 514         return Optional.absent();
 515     }
 516
 517     private static boolean isDefaultPST(final PathSetupType pst) {
 518         if (pst != null && pst.getPst() != null && pst.getPst().shortValue() != 0) {
 519             return false;
 520         }
 521         return true;
 522     }
 523
 524     private static PeerCapabilities getCapabilities(final Stateful stateful) {
 525         final PeerCapabilities capa = new PeerCapabilities();
 526         capa.setStateful(true);
 527         if (stateful.isLspUpdateCapability() != null) {
 528             capa.setActive(stateful.isLspUpdateCapability());
 529         }
 530         final Stateful1 stateful1 = stateful.getAugmentation(Stateful1.class);
 531         if (stateful1 != null && stateful1.isInitiation() != null) {
 532             capa.setInstantiation(stateful1.isInitiation());
 533         }
 534         return capa;
 535     }
 536 }