apps/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 // Non-final for testing
  23 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.getSessionId(),
  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 or certificate of the incoming netconf server is present. If present return null,
  43             // else store the session. The sshkey/certificate is the uniqueness of the callhome sessions not the
  44             // uniqueid/devicename
  45             LOG.error("Server Host Key/Certificate {} is associated with existing session {}, closing session {}",
  46                 remoteKey, existing, session);
  47             session.terminate();
  48             return null;
  49         }
  50
  51         final SocketAddress remoteAddress = session.getRemoteAddress();
  52         final CallHomeMountSessionContext prev = contextByAddress.put(remoteAddress, deviceContext);
  53         if (prev != null) {
  54             LOG.warn("Remote {} replaced context {} with {}", remoteAddress, prev, deviceContext);
  55         }
  56         return deviceContext;
  57     }
  58
  59     @Override
  60     public void onClosed(final CallHomeMountSessionContext deviceContext) {
  61         final CallHomeProtocolSessionContext session = deviceContext.getProtocol();
  62         contextByAddress.remove(session.getRemoteAddress(), deviceContext);
  63         contextByPublicKey.remove(session.getRemoteServerKey(), deviceContext);
  64     }
  65
  66     private void onClosed(final CallHomeMountSessionContext deviceContext, final CloseCallback onCloseHandler) {
  67         onClosed(deviceContext);
  68         onCloseHandler.onClosed(deviceContext);
  69     }
  70 }