Commit | Line | Data |
---|---|---|
eb1bab5b | 1 | /********************************************************************** |
60ae41e1 | 2 | * Copyright (c) 2012, 2014 Ericsson |
ea21cd65 | 3 | * |
eb1bab5b BH |
4 | * All rights reserved. This program and the accompanying materials are |
5 | * made available under the terms of the Eclipse Public License v1.0 which | |
6 | * accompanies this distribution, and is available at | |
7 | * http://www.eclipse.org/legal/epl-v10.html | |
ea21cd65 AM |
8 | * |
9 | * Contributors: | |
eb1bab5b BH |
10 | * Patrick Tasse - Initial API and implementation |
11 | * Bernd Hufmann - Updated using Executor Framework | |
12 | **********************************************************************/ | |
9315aeee | 13 | package org.eclipse.linuxtools.internal.lttng2.ui.views.control.remote; |
eb1bab5b BH |
14 | |
15 | import java.io.BufferedReader; | |
16 | import java.io.IOException; | |
17 | import java.io.InputStreamReader; | |
18 | import java.util.ArrayList; | |
d6fc6e1b | 19 | import java.util.Random; |
eb1bab5b BH |
20 | import java.util.concurrent.Callable; |
21 | import java.util.concurrent.CancellationException; | |
22 | import java.util.concurrent.ExecutorService; | |
23 | import java.util.concurrent.Executors; | |
24 | import java.util.concurrent.FutureTask; | |
25 | import java.util.concurrent.TimeUnit; | |
26 | import java.util.concurrent.TimeoutException; | |
27 | ||
28 | import org.eclipse.core.commands.ExecutionException; | |
29 | import org.eclipse.core.runtime.IProgressMonitor; | |
30 | import org.eclipse.core.runtime.NullProgressMonitor; | |
9315aeee | 31 | import org.eclipse.linuxtools.internal.lttng2.ui.views.control.messages.Messages; |
4bdf5f96 | 32 | import org.eclipse.linuxtools.internal.lttng2.ui.views.control.preferences.ControlPreferences; |
eb1bab5b BH |
33 | import org.eclipse.rse.services.shells.HostShellProcessAdapter; |
34 | import org.eclipse.rse.services.shells.IHostShell; | |
35 | import org.eclipse.rse.services.shells.IShellService; | |
36 | ||
37 | /** | |
eb1bab5b | 38 | * <p> |
ea21cd65 | 39 | * Implementation of remote command execution using RSE's shell service. |
eb1bab5b | 40 | * </p> |
cfdb727a | 41 | * |
dbd4432d BH |
42 | * @author Patrick Tasse |
43 | * @author Bernd Hufmann | |
eb1bab5b BH |
44 | */ |
45 | public class CommandShell implements ICommandShell { | |
46 | ||
47 | // ------------------------------------------------------------------------ | |
48 | // Constants | |
49 | // ------------------------------------------------------------------------ | |
50 | ||
d6fc6e1b | 51 | /** Sub-string to be echo'ed when running command in shell, used to indicate that the command has finished running */ |
77735e82 | 52 | public static final String DONE_MARKUP_STRING = "--RSE:donedonedone:--"; //$NON-NLS-1$ |
ea21cd65 AM |
53 | |
54 | /** Sub-string to be echoed when running a command in shell. */ | |
77735e82 | 55 | public static final String BEGIN_END_TAG = "BEGIN-END-TAG:"; //$NON-NLS-1$ |
ea21cd65 | 56 | |
cfdb727a | 57 | /** Command delimiter for shell */ |
77735e82 | 58 | public static final String CMD_DELIMITER = "\n"; //$NON-NLS-1$ |
eb1bab5b | 59 | |
cfdb727a | 60 | /** Shell "echo" command */ |
77735e82 | 61 | public static final String SHELL_ECHO_CMD = " echo "; //$NON-NLS-1$ |
ea21cd65 | 62 | |
d6fc6e1b | 63 | /** Default command separator */ |
77735e82 | 64 | public static final char CMD_SEPARATOR = ';'; |
eb1bab5b | 65 | |
eb1bab5b BH |
66 | // ------------------------------------------------------------------------ |
67 | // Attributes | |
68 | // ------------------------------------------------------------------------ | |
69 | private IRemoteSystemProxy fProxy = null; | |
70 | private IHostShell fHostShell = null; | |
d6fc6e1b BH |
71 | private BufferedReader fInputBufferReader = null; |
72 | private BufferedReader fErrorBufferReader = null; | |
ea21cd65 | 73 | private final ExecutorService fExecutor = Executors.newFixedThreadPool(1); |
eb1bab5b | 74 | private boolean fIsConnected = false; |
ea21cd65 | 75 | private final Random fRandom = new Random(System.currentTimeMillis()); |
d6fc6e1b | 76 | private int fReturnValue; |
ea21cd65 | 77 | |
eb1bab5b BH |
78 | // ------------------------------------------------------------------------ |
79 | // Constructors | |
80 | // ------------------------------------------------------------------------ | |
ea21cd65 AM |
81 | |
82 | /** | |
83 | * Create a new command shell | |
84 | * | |
85 | * @param proxy | |
86 | * The RSE proxy for this shell | |
87 | */ | |
eb1bab5b BH |
88 | public CommandShell(IRemoteSystemProxy proxy) { |
89 | fProxy = proxy; | |
90 | } | |
91 | ||
92 | // ------------------------------------------------------------------------ | |
93 | // Operations | |
94 | // ------------------------------------------------------------------------ | |
11252342 | 95 | |
eb1bab5b BH |
96 | @Override |
97 | public void connect() throws ExecutionException { | |
98 | IShellService shellService = fProxy.getShellService(); | |
99 | Process p = null; | |
100 | try { | |
101 | fHostShell = shellService.launchShell("", new String[0], new NullProgressMonitor()); //$NON-NLS-1$ | |
102 | p = new HostShellProcessAdapter(fHostShell); | |
103 | } catch (Exception e) { | |
104 | throw new ExecutionException(Messages.TraceControl_CommandShellError, e); | |
105 | } | |
d6fc6e1b BH |
106 | fInputBufferReader = new BufferedReader(new InputStreamReader(p.getInputStream())); |
107 | fErrorBufferReader = new BufferedReader(new InputStreamReader(p.getErrorStream())); | |
eb1bab5b | 108 | fIsConnected = true; |
eb1bab5b BH |
109 | } |
110 | ||
eb1bab5b BH |
111 | @Override |
112 | public void disconnect() { | |
113 | fIsConnected = false; | |
114 | try { | |
d6fc6e1b BH |
115 | fInputBufferReader.close(); |
116 | fErrorBufferReader.close(); | |
eb1bab5b BH |
117 | } catch (IOException e) { |
118 | // ignore | |
119 | } | |
120 | } | |
121 | ||
eb1bab5b BH |
122 | @Override |
123 | public ICommandResult executeCommand(String command, IProgressMonitor monitor) throws ExecutionException { | |
124 | return executeCommand(command, monitor, true); | |
125 | } | |
126 | ||
eb1bab5b BH |
127 | @Override |
128 | public ICommandResult executeCommand(final String command, final IProgressMonitor monitor, final boolean checkReturnValue) throws ExecutionException { | |
129 | if (fIsConnected) { | |
e0838ca1 | 130 | FutureTask<CommandResult> future = new FutureTask<>(new Callable<CommandResult>() { |
eb1bab5b BH |
131 | @Override |
132 | public CommandResult call() throws IOException, CancellationException { | |
e0838ca1 | 133 | final ArrayList<String> result = new ArrayList<>(); |
ea21cd65 | 134 | |
eb1bab5b | 135 | synchronized (fHostShell) { |
d6fc6e1b BH |
136 | // Initialize return value which will be updated in isAliasEchoResult() |
137 | fReturnValue = 0; | |
138 | ||
139 | int startAlias = fRandom.nextInt(); | |
140 | int endAlias = fRandom.nextInt(); | |
141 | fHostShell.writeToShell(formatShellCommand(command, startAlias, endAlias)); | |
ea21cd65 | 142 | |
eb1bab5b | 143 | String nextLine; |
d6fc6e1b BH |
144 | boolean isStartFound = false; |
145 | while ((nextLine = fInputBufferReader.readLine()) != null) { | |
eb1bab5b BH |
146 | |
147 | if (monitor.isCanceled()) { | |
148 | flushInput(); | |
ea21cd65 | 149 | throw new CancellationException(); |
eb1bab5b BH |
150 | } |
151 | ||
ea21cd65 | 152 | // check if line contains echoed start alias |
d6fc6e1b BH |
153 | if (isAliasEchoResult(nextLine, startAlias, true)) { |
154 | isStartFound = true; | |
155 | continue; | |
eb1bab5b | 156 | } |
eb1bab5b | 157 | |
ea21cd65 AM |
158 | // check if line contains is the end mark-up. This will retrieve also |
159 | // the return value of the actual command. | |
d6fc6e1b BH |
160 | if (isAliasEchoResult(nextLine, endAlias, false)) { |
161 | break; | |
eb1bab5b BH |
162 | } |
163 | ||
d6fc6e1b BH |
164 | // Ignore line if |
165 | // 1) start hasn't been found or | |
166 | // 2) line is an echo of the command or | |
167 | // 3) line is an echo of the end mark-up | |
168 | if (!isStartFound || | |
169 | isCommandEcho(nextLine, command) || | |
170 | nextLine.contains(getEchoResult(endAlias))) | |
171 | { | |
172 | continue; | |
eb1bab5b | 173 | } |
d6fc6e1b BH |
174 | |
175 | // Now it's time add to the result | |
176 | result.add(nextLine); | |
eb1bab5b BH |
177 | } |
178 | ||
ea21cd65 | 179 | // Read any left over output |
eb1bab5b | 180 | flushInput(); |
ea21cd65 | 181 | |
d6fc6e1b | 182 | // Read error stream output when command failed. |
ea21cd65 | 183 | if (fReturnValue != 0) { |
d6fc6e1b BH |
184 | while(fErrorBufferReader.ready()) { |
185 | if ((nextLine = fErrorBufferReader.readLine()) != null) { | |
186 | result.add(nextLine); | |
187 | } | |
188 | } | |
189 | } | |
eb1bab5b | 190 | } |
d6fc6e1b | 191 | return new CommandResult(fReturnValue, result.toArray(new String[result.size()])); |
eb1bab5b BH |
192 | } |
193 | }); | |
194 | ||
195 | fExecutor.execute(future); | |
196 | ||
197 | try { | |
4bdf5f96 | 198 | return future.get(ControlPreferences.getInstance().getCommandTimeout(), TimeUnit.SECONDS); |
eb1bab5b BH |
199 | } catch (java.util.concurrent.ExecutionException ex) { |
200 | throw new ExecutionException(Messages.TraceControl_ExecutionFailure, ex); | |
201 | } catch (InterruptedException ex) { | |
202 | throw new ExecutionException(Messages.TraceControl_ExecutionCancelled, ex); | |
203 | } catch (TimeoutException ex) { | |
204 | throw new ExecutionException(Messages.TraceControl_ExecutionTimeout, ex); | |
205 | } | |
206 | } | |
207 | throw new ExecutionException(Messages.TraceControl_ShellNotConnected, null); | |
208 | } | |
cfdb727a | 209 | |
eb1bab5b BH |
210 | // ------------------------------------------------------------------------ |
211 | // Helper methods | |
212 | // ------------------------------------------------------------------------ | |
213 | /** | |
cfdb727a | 214 | * Flushes the buffer reader |
eb1bab5b BH |
215 | * @throws IOException |
216 | */ | |
217 | private void flushInput() throws IOException { | |
218 | char[] cbuf = new char[1]; | |
d6fc6e1b BH |
219 | while (fInputBufferReader.ready()) { |
220 | if (fInputBufferReader.read(cbuf, 0, 1) == -1) { | |
eb1bab5b BH |
221 | break; |
222 | } | |
223 | } | |
224 | } | |
cfdb727a | 225 | |
eb1bab5b | 226 | /** |
d6fc6e1b | 227 | * Format the command to be sent into the shell command with start and end marker strings. |
ea21cd65 | 228 | * The start marker is need to know when the actual command output starts. The end marker |
d6fc6e1b | 229 | * string is needed so we can tell that end of output has been reached. |
ea21cd65 AM |
230 | * |
231 | * @param cmd The actual command | |
d6fc6e1b BH |
232 | * @param startAlias The command alias for start marker |
233 | * @param endAlias The command alias for end marker | |
eb1bab5b BH |
234 | * @return formatted command string |
235 | */ | |
ea21cd65 | 236 | private static String formatShellCommand(String cmd, int startAlias, int endAlias) { |
cfdb727a | 237 | if (cmd == null || cmd.equals("")) { //$NON-NLS-1$ |
eb1bab5b | 238 | return cmd; |
cfdb727a | 239 | } |
eb1bab5b | 240 | StringBuffer formattedCommand = new StringBuffer(); |
ea21cd65 | 241 | // Make multi-line command. |
d6fc6e1b BH |
242 | // Wrap actual command with start marker and end marker to wrap actual command. |
243 | formattedCommand.append(getEchoCmd(startAlias)); | |
244 | formattedCommand.append(CMD_DELIMITER); | |
245 | formattedCommand.append(cmd); | |
246 | formattedCommand.append(CMD_DELIMITER); | |
247 | formattedCommand.append(getEchoCmd(endAlias)); | |
eb1bab5b BH |
248 | formattedCommand.append(CMD_DELIMITER); |
249 | return formattedCommand.toString(); | |
250 | } | |
ea21cd65 | 251 | |
d6fc6e1b BH |
252 | /** |
253 | * Creates a echo command line in the format: echo <start tag> <alias> <end tag> $? | |
ea21cd65 | 254 | * |
d6fc6e1b BH |
255 | * @param alias The command alias integer to be included in the echoed message. |
256 | * @return the echo command line | |
257 | */ | |
ea21cd65 | 258 | private static String getEchoCmd(int alias) { |
d6fc6e1b BH |
259 | return SHELL_ECHO_CMD + getEchoResult(alias) + "$?"; //$NON-NLS-1$ |
260 | } | |
261 | ||
262 | /** | |
ea21cd65 | 263 | * Creates the expected result for a given command alias: |
d6fc6e1b | 264 | * <start tag> <alias> <end tag> $? |
ea21cd65 | 265 | * |
d6fc6e1b BH |
266 | * @param alias The command alias integer to be included in the echoed message. |
267 | * @return the expected echo result | |
268 | */ | |
ea21cd65 | 269 | private static String getEchoResult(int alias) { |
d6fc6e1b BH |
270 | return BEGIN_END_TAG + String.valueOf(alias) + DONE_MARKUP_STRING; |
271 | } | |
ea21cd65 | 272 | |
d6fc6e1b BH |
273 | /** |
274 | * Verifies if given command line contains a command alias echo result. | |
ea21cd65 | 275 | * |
d6fc6e1b BH |
276 | * @param line The output line to test. |
277 | * @param alias The command alias | |
278 | * @param checkReturnValue <code>true</code> to retrieve command result (previous command) <code>false</code> | |
279 | * @return <code>true</code> if output line is a command alias echo result else <code>false</code> | |
280 | */ | |
281 | private boolean isAliasEchoResult(String line, int alias, boolean checkReturnValue) { | |
282 | String expected = getEchoResult(alias); | |
283 | if (line.startsWith(expected)) { | |
284 | if (!checkReturnValue) { | |
285 | try { | |
286 | int k = Integer.valueOf(line.substring(expected.length())); | |
287 | fReturnValue = k; | |
288 | } catch (NumberFormatException e) { | |
289 | // do nothing | |
290 | } | |
291 | } | |
292 | return true; | |
ea21cd65 AM |
293 | } |
294 | int index = line.indexOf(expected); | |
77735e82 BH |
295 | if ((index > 0) && (line.indexOf(SHELL_ECHO_CMD) == -1)) { |
296 | return true; | |
d6fc6e1b | 297 | } |
eb1bab5b | 298 | |
d6fc6e1b BH |
299 | return false; |
300 | } | |
ea21cd65 | 301 | |
d6fc6e1b | 302 | /** |
ea21cd65 AM |
303 | * Verifies if output line is an echo of the given command line. If the |
304 | * output line is longer then the maximum line lengths (e.g. for ssh), the | |
305 | * shell adds a line break character. This method takes this in | |
306 | * consideration by comparing the command line without any whitespaces. | |
307 | * | |
308 | * @param line | |
309 | * The output line to verify | |
310 | * @param cmd | |
311 | * The command executed | |
312 | * @return <code>true</code> if it's an echoed command line else | |
313 | * <code>false</code> | |
d6fc6e1b BH |
314 | */ |
315 | @SuppressWarnings("nls") | |
ea21cd65 | 316 | private static boolean isCommandEcho(String line, String cmd) { |
d6fc6e1b BH |
317 | String s1 = line.replaceAll("\\s",""); |
318 | String s2 = cmd.replaceAll("\\s",""); | |
ea21cd65 | 319 | s2 = s2.replaceAll("(\\*)", "(\\\\*)"); |
d6fc6e1b BH |
320 | String patternStr = ".*(" + s2 +")$"; |
321 | return s1.matches(patternStr); | |
322 | } | |
eb1bab5b | 323 | } |