Merge "BUG-1568 Ssh Handler for netconf client Mina implementation"
[controller.git] / opendaylight / netconf / netconf-netty-util / src / main / java / org / opendaylight / controller / netconf / nettyutil / handler / ssh / client / SshClient.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
9 package org.opendaylight.controller.netconf.nettyutil.handler.ssh.client;
10
11 import ch.ethz.ssh2.Connection;
12 import ch.ethz.ssh2.Session;
13 import java.io.IOException;
14 import java.util.HashMap;
15 import java.util.Map;
16 import org.opendaylight.controller.netconf.nettyutil.handler.ssh.authentication.AuthenticationHandler;
17 import org.opendaylight.controller.netconf.nettyutil.handler.ssh.virtualsocket.VirtualSocket;
18
19 /**
20  * Wrapper class around GANYMED SSH java library.
21  */
22 class SshClient {
23     private final VirtualSocket socket;
24     private final Map<Integer, SshSession> openSessions = new HashMap<>();
25     private final AuthenticationHandler authenticationHandler;
26     private Connection connection;
27
28     public SshClient(VirtualSocket socket, AuthenticationHandler authenticationHandler) throws IOException {
29         this.socket = socket;
30         this.authenticationHandler = authenticationHandler;
31     }
32
33     public SshSession openSession() throws IOException {
34         if (connection == null) {
35             connect();
36         }
37
38         Session session = connection.openSession();
39         SshSession sshSession = new SshSession(session);
40         openSessions.put(openSessions.size(), sshSession);
41
42         return sshSession;
43     }
44
45     private void connect() throws IOException {
46         connection = new Connection(socket);
47
48         connection.connect();
49         authenticationHandler.authenticate(connection);
50     }
51
52
53     public void close() {
54         for (SshSession session : openSessions.values()){
55             session.close();
56         }
57
58         openSessions.clear();
59
60         if (connection != null) {
61             connection.close();
62         }
63     }
64
65     @Override
66     public String toString() {
67         return "SshClient{" +
68                 "socket=" + socket +
69                 '}';
70     }
71 }

©2013 OpenDaylight, A Linux Foundation Collaborative Project. All Rights Reserved.
OpenDaylight is a registered trademark of The OpenDaylight Project, Inc.
Linux Foundation and OpenDaylight are registered trademarks of the Linux Foundation.
Linux is a registered trademark of Linus Torvalds.