netconf/callhome-provider/src/main/java/org/opendaylight/netconf/callhome/mount/CallHomeMountSessionManager.java

   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 java.net.InetSocketAddress;
  11 import java.net.SocketAddress;
  12 import java.security.PublicKey;
  13 import java.util.concurrent.ConcurrentHashMap;
  14 import java.util.concurrent.ConcurrentMap;
  15 import org.eclipse.jdt.annotation.Nullable;
  16 import org.opendaylight.netconf.callhome.mount.CallHomeMountSessionContext.CloseCallback;
  17 import org.opendaylight.netconf.callhome.protocol.CallHomeChannelActivator;
  18 import org.opendaylight.netconf.callhome.protocol.CallHomeProtocolSessionContext;
  19 import org.slf4j.Logger;
  20 import org.slf4j.LoggerFactory;
  21
  22 public class CallHomeMountSessionManager implements CallHomeMountSessionContext.CloseCallback {
  23     private static final Logger LOG = LoggerFactory.getLogger(CallHomeMountSessionManager.class);
  24
  25     private final ConcurrentMap<SocketAddress, CallHomeMountSessionContext> contextByAddress =
  26         new ConcurrentHashMap<>();
  27     private final ConcurrentMap<PublicKey, CallHomeMountSessionContext> contextByPublicKey = new ConcurrentHashMap<>();
  28
  29     @Nullable CallHomeMountSessionContext getByAddress(final InetSocketAddress remoteAddr) {
  30         return contextByAddress.get(remoteAddr);
  31     }
  32
  33     CallHomeMountSessionContext createSession(final CallHomeProtocolSessionContext session,
  34             final CallHomeChannelActivator activator, final CloseCallback onCloseHandler) {
  35         final CallHomeMountSessionContext deviceContext = new CallHomeMountSessionContext(session.getSessionId(),
  36             session, activator, devCtxt -> onClosed(devCtxt, onCloseHandler));
  37
  38         final PublicKey remoteKey = session.getRemoteServerKey();
  39         final CallHomeMountSessionContext existing = contextByPublicKey.putIfAbsent(remoteKey, deviceContext);
  40         if (existing != null) {
  41             // Check if the sshkey or certificate of the incoming netconf server is present. If present return null,
  42             // else store the session. The sshkey/certificate is the uniqueness of the callhome sessions not the
  43             // uniqueid/devicename
  44             LOG.error("Server Host Key/Certificate {} is associated with existing session {}, closing session {}",
  45                 remoteKey, existing, session);
  46             session.terminate();
  47             return null;
  48         }
  49
  50         final SocketAddress 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     private void onClosed(final CallHomeMountSessionContext deviceContext, final CloseCallback onCloseHandler) {
  66         onClosed(deviceContext);
  67         onCloseHandler.onClosed(deviceContext);
  68     }
  69 }