command who
[saartuer.git] / JavaTuer / src / de / hacksaar / javatuer / SshClient.java
1 package de.hacksaar.javatuer;
2
3 import com.jcraft.jsch.*;
4
5 import java.io.IOException;
6 import java.io.InputStream;
7 import java.io.OutputStream;
8
9 public class SshClient {
10         private static final String TAG = "SshClient";
11         private final String host;
12         private final String username;
13         private final int port;
14         private final JSch jSch;
15         private final TuerLogging log;
16         private Session session;
17         private Channel channel;
18
19         public SshClient(String host, String username, int port, TuerLogging log) {
20                 this.host = host;
21                 this.username = username;
22                 this.port = port;
23                 this.log = log;
24                 JSch.setLogger(log);
25                 jSch = new JSch();
26         }
27
28         public void addPrivateKey(String keyFilePath) throws JSchException {
29                 jSch.addIdentity(keyFilePath);
30         }
31
32         public void addPrivateKey(String keyFilePath, String passphrase) throws JSchException {
33                 jSch.addIdentity(keyFilePath, passphrase);
34         }
35
36         public void disconnect() {
37                 channel.disconnect();
38                 channel = null;
39                 session.disconnect();
40                 session = null;
41         }
42
43         public String getHost() {
44                 return host;
45         }
46
47         public InputStream getInputStream() throws IOException {
48                 if (channel == null) {
49                         return null;
50                 }
51                 return channel.getInputStream();
52         }
53
54         public OutputStream getOutputStream() throws IOException {
55                 if (channel == null) {
56                         return null;
57                 }
58                 return channel.getOutputStream();
59         }
60
61         public int getPort() {
62                 return port;
63         }
64
65         public String getUsername() {
66                 return username;
67         }
68
69         public boolean isConnected() {
70                 return channel != null && channel.isConnected() && session != null && session.isConnected();
71         }
72
73         public void login(final String password) throws JSchException {
74                 login(new UserInfo() {
75
76                         @Override
77                         public String getPassphrase() {
78                                 return null;
79                         }
80
81                         @Override
82                         public String getPassword() {
83                                 return password;
84                         }
85
86                         @Override
87                         public boolean promptPassphrase(String s) {
88                                 return false;
89                         }
90
91                         @Override
92                         public boolean promptPassword(String s) {
93                                 return true;
94                         }
95
96                         @Override
97                         public boolean promptYesNo(String s) {
98                                 return false;
99                         }
100
101                         @Override
102                         public void showMessage(String s) {
103                                 log.info(TAG, s);
104                         }
105                 });
106         }
107
108         public void login() throws JSchException {
109                 login(new UserInfo() {
110
111                         @Override
112                         public String getPassphrase() {
113                                 return null;
114                         }
115
116                         @Override
117                         public String getPassword() {
118                                 return null;
119                         }
120
121                         @Override
122                         public boolean promptPassphrase(String s) {
123                                 return false;
124                         }
125
126                         @Override
127                         public boolean promptPassword(String s) {
128                                 return false;
129                         }
130
131                         @Override
132                         public boolean promptYesNo(String s) {
133                                 return false;
134                         }
135
136                         @Override
137                         public void showMessage(String s) {
138                                 log.info(TAG, s);
139                         }
140                 });
141         }
142
143         public void login(UserInfo userInfo) throws JSchException {
144                 log.debug(TAG, "Creating session");
145                 if (session == null) {
146                         session = jSch.getSession(username, host, port);
147                 }
148                 if (!session.isConnected()) {
149                         session.setUserInfo(userInfo);
150                         log.debug(TAG, "Connecting");
151                         session.connect();
152                         log.debug(TAG, "Opening channel");
153                         channel = session.openChannel("shell");
154                         channel.connect();
155                 }
156         }
157
158 }