git.opendaylight Code Review - netconf.git/blob

   1 /*
   2  * Copyright (c) 2016 Brocade Communication Systems 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.netconf.callhome.mount;
   9
  10 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
  11 import java.net.InetSocketAddress;
  12 import java.net.SocketAddress;
  13 import java.security.PublicKey;
  14 import java.util.concurrent.ConcurrentHashMap;
  15 import java.util.concurrent.ConcurrentMap;
  16 import org.eclipse.jdt.annotation.Nullable;
  17 import org.opendaylight.netconf.callhome.mount.CallHomeMountSessionContext.CloseCallback;
  18 import org.opendaylight.netconf.callhome.protocol.CallHomeChannelActivator;
  19 import org.opendaylight.netconf.callhome.protocol.CallHomeProtocolSessionContext;
  20 import org.slf4j.Logger;
  21 import org.slf4j.LoggerFactory;
  22
  23 public class CallHomeMountSessionManager implements CallHomeMountSessionContext.CloseCallback {
  24     private static final Logger LOG = LoggerFactory.getLogger(CallHomeMountSessionManager.class);
  25
  26     private final ConcurrentMap<SocketAddress, CallHomeMountSessionContext> contextByAddress =
  27         new ConcurrentHashMap<>();
  28     private final ConcurrentMap<PublicKey, CallHomeMountSessionContext> contextByPublicKey = new ConcurrentHashMap<>();
  29
  30     @Nullable CallHomeMountSessionContext getByAddress(final InetSocketAddress remoteAddr) {
  31         return contextByAddress.get(remoteAddr);
  32     }
  33
  34     CallHomeMountSessionContext createSession(final CallHomeProtocolSessionContext session,
  35             final CallHomeChannelActivator activator, final CloseCallback onCloseHandler) {
  36         final CallHomeMountSessionContext deviceContext = new CallHomeMountSessionContext(session.getSessionName(),
  37             session, activator, devCtxt -> onClosed(devCtxt, onCloseHandler));
  38
  39         final PublicKey remoteKey = session.getRemoteServerKey();
  40         final CallHomeMountSessionContext existing = contextByPublicKey.putIfAbsent(remoteKey, deviceContext);
  41         if (existing != null) {
  42             // Check if the sshkey of the incoming netconf server is present. If present return null, else store the
  43             // session. The sshkey is the uniqueness of the callhome sessions not the uniqueid/devicename.
  44             LOG.error("SSH Host Key {} is associated with existing session {}, closing session {}", remoteKey, existing,
  45                 session);
  46             session.terminate();
  47             return null;
  48         }
  49
  50         final InetSocketAddress remoteAddress = session.getRemoteAddress();
  51         final CallHomeMountSessionContext prev = contextByAddress.put(remoteAddress, deviceContext);
  52         if (prev != null) {
  53             LOG.warn("Remote {} replaced context {} with {}", remoteAddress, prev, deviceContext);
  54         }
  55         return deviceContext;
  56     }
  57
  58     @Override
  59     public void onClosed(final CallHomeMountSessionContext deviceContext) {
  60         final CallHomeProtocolSessionContext session = deviceContext.getProtocol();
  61         contextByAddress.remove(session.getRemoteAddress(), deviceContext);
  62         contextByPublicKey.remove(session.getRemoteServerKey(), deviceContext);
  63     }
  64
  65     @SuppressFBWarnings(value = "UPM_UNCALLED_PRIVATE_METHOD",
  66             justification = "https://github.com/spotbugs/spotbugs/issues/811")
  67     private void onClosed(final CallHomeMountSessionContext deviceContext, final CloseCallback onCloseHandler) {
  68         onClosed(deviceContext);
  69         onCloseHandler.onClosed(deviceContext);
  70     }
  71 }