opendaylight/netconf/netconf-ssh/src/main/java/org/opendaylight/controller/netconf/ssh/NetconfSSHServer.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.controller.netconf.ssh;
   9
  10 import org.opendaylight.controller.netconf.ssh.authentication.AuthProvider;
  11 import org.opendaylight.controller.netconf.ssh.threads.SocketThread;
  12 import org.opendaylight.controller.usermanager.IUserManager;
  13 import org.slf4j.Logger;
  14 import org.slf4j.LoggerFactory;
  15
  16 import javax.annotation.concurrent.ThreadSafe;
  17 import java.io.IOException;
  18 import java.net.InetSocketAddress;
  19 import java.net.ServerSocket;
  20 import java.util.concurrent.atomic.AtomicLong;
  21
  22 @ThreadSafe
  23 public final class NetconfSSHServer implements Runnable {
  24
  25     private ServerSocket ss = null;
  26     private static final Logger logger =  LoggerFactory.getLogger(NetconfSSHServer.class);
  27     private static final AtomicLong sesssionId = new AtomicLong();
  28     private final InetSocketAddress clientAddress;
  29     private final AuthProvider authProvider;
  30     private volatile boolean up = false;
  31
  32     private NetconfSSHServer(int serverPort,InetSocketAddress clientAddress, AuthProvider authProvider) throws IllegalStateException, IOException {
  33
  34         logger.trace("Creating SSH server socket on port {}",serverPort);
  35         this.ss = new ServerSocket(serverPort);
  36         if (!ss.isBound()){
  37             throw new IllegalStateException("Socket can't be bound to requested port :"+serverPort);
  38         }
  39         logger.trace("Server socket created.");
  40         this.clientAddress = clientAddress;
  41         this.authProvider = authProvider;
  42         this.up = true;
  43     }
  44
  45     public static NetconfSSHServer start(int serverPort, InetSocketAddress clientAddress,AuthProvider authProvider) throws IllegalStateException, IOException {
  46         return new NetconfSSHServer(serverPort, clientAddress,authProvider);
  47     }
  48
  49     public void stop() throws IOException {
  50         up = false;
  51         logger.trace("Closing SSH server socket.");
  52         ss.close();
  53         logger.trace("SSH server socket closed.");
  54     }
  55
  56     public void removeUserManagerService(){
  57         this.authProvider.removeUserManagerService();
  58     }
  59
  60     public void addUserManagerService(IUserManager userManagerService){
  61         this.authProvider.addUserManagerService(userManagerService);
  62     }
  63     public boolean isUp(){
  64         return this.up;
  65     }
  66     @Override
  67     public void run() {
  68         while (up) {
  69             logger.trace("Starting new socket thread.");
  70             try {
  71                 SocketThread.start(ss.accept(), clientAddress, sesssionId.incrementAndGet(), authProvider);
  72             }
  73             catch (IOException e) {
  74                 if( up ) {
  75                     logger.error("Exception occurred during socket thread initialization", e);
  76                 }
  77                 else {
  78                     // We're shutting down so an exception is expected as the socket's been closed.
  79                     // Log to debug.
  80                     logger.debug("Shutting down - got expected exception: " + e);
  81                 }
  82             }
  83         }
  84     }
  85 }