2 * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.protocol.pcep.impl;
10 import com.google.common.base.Optional;
11 import com.google.common.primitives.UnsignedBytes;
12 import io.netty.channel.Channel;
13 import io.netty.channel.ChannelFuture;
14 import io.netty.channel.ChannelFutureListener;
15 import io.netty.util.concurrent.Promise;
16 import java.net.InetSocketAddress;
17 import java.util.Comparator;
18 import java.util.concurrent.ExecutionException;
19 import org.opendaylight.protocol.framework.AbstractSessionNegotiator;
20 import org.opendaylight.protocol.framework.SessionListenerFactory;
21 import org.opendaylight.protocol.pcep.PCEPSessionListener;
22 import org.opendaylight.protocol.pcep.impl.PCEPPeerRegistry.SessionReference;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
27 public class PCEPSessionNegotiator extends AbstractSessionNegotiator<Message, PCEPSessionImpl> {
29 private static final Logger LOG = LoggerFactory.getLogger(PCEPSessionNegotiator.class);
31 private static final Comparator<byte[]> COMPARATOR = UnsignedBytes.lexicographicalComparator();
33 private final Channel channel;
35 private final Promise<PCEPSessionImpl> promise;
37 private final SessionListenerFactory<PCEPSessionListener> factory;
39 private final AbstractPCEPSessionNegotiatorFactory negFactory;
41 public PCEPSessionNegotiator(final Channel channel, final Promise<PCEPSessionImpl> promise, final SessionListenerFactory<PCEPSessionListener> factory,
42 final AbstractPCEPSessionNegotiatorFactory negFactory) {
43 super(promise, channel);
44 this.channel = channel;
45 this.promise = promise;
46 this.factory = factory;
47 this.negFactory = negFactory;
51 protected void startNegotiation() throws ExecutionException {
52 final Object lock = this;
54 LOG.debug("Bootstrap negotiation for channel {} started", this.channel);
57 * We have a chance to see if there's a client session already
58 * registered for this client.
60 final byte[] clientAddress = ((InetSocketAddress) this.channel.remoteAddress()).getAddress().getAddress();
61 final PCEPPeerRegistry sessionReg = this.negFactory.getSessionRegistry();
64 if (sessionReg.getSessionReference(clientAddress).isPresent()) {
65 final byte[] serverAddress = ((InetSocketAddress) this.channel.localAddress()).getAddress().getAddress();
66 if (COMPARATOR.compare(serverAddress, clientAddress) > 0) {
67 final Optional<SessionReference> sessionRefMaybe = sessionReg.removeSessionReference(clientAddress);
69 if (sessionRefMaybe.isPresent()) {
70 sessionRefMaybe.get().close();
72 } catch (final Exception e) {
73 LOG.error("Unexpected failure to close old session", e);
76 negotiationFailed(new IllegalStateException("A conflicting session for address "
77 + ((InetSocketAddress) this.channel.remoteAddress()).getAddress() + " found."));
82 final Short sessionId = sessionReg.nextSession(clientAddress);
83 final AbstractPCEPSessionNegotiator n = this.negFactory.createNegotiator(this.promise, this.factory.getSessionListener(), this.channel, sessionId);
85 sessionReg.putSessionReference(clientAddress, new SessionReference() {
87 public void close() throws ExecutionException {
89 sessionReg.releaseSession(clientAddress, sessionId);
91 PCEPSessionNegotiator.this.channel.close();
96 public Short getSessionId() {
101 this.channel.closeFuture().addListener(new ChannelFutureListener() {
103 public void operationComplete(final ChannelFuture future) {
104 synchronized (lock) {
105 sessionReg.removeSessionReference(clientAddress);
110 LOG.info("Replacing bootstrap negotiator for channel {}", this.channel);
111 this.channel.pipeline().replace(this, "negotiator", n);
112 n.startNegotiation();
117 protected void handleMessage(final Message msg) {
118 throw new IllegalStateException("Bootstrap negotiator should have been replaced");