1 files changed, 107 insertions, 0 deletions

diff --git a/plugins/org.yocto.sdk.remotetools/src/org/yocto/sdk/remotetools/actions/SystemtapModel.java b/plugins/org.yocto.sdk.remotetools/src/org/yocto/sdk/remotetools/actions/SystemtapModel.java
new file mode 100644
index 0000000..60d7d76
--- /dev/null
+++ b/plugins/org.yocto.sdk.remotetools/src/org/yocto/sdk/remotetools/actions/SystemtapModel.java
@@ -0,0 +1,107 @@
+/*******************************************************************************
+ * Copyright (c) 2010 Intel Corporation.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ * Intel - initial API and implementation
+ *******************************************************************************/
+package org.yocto.sdk.remotetools.actions;
+
+import java.io.File;
+import java.lang.reflect.InvocationTargetException;
+
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.core.runtime.NullProgressMonitor;
+import org.eclipse.core.runtime.OperationCanceledException;
+import org.eclipse.jface.dialogs.ProgressMonitorDialog;
+import org.eclipse.jface.operation.IRunnableWithProgress;
+import org.eclipse.rse.subsystems.terminals.core.ITerminalServiceSubSystem;
+import org.eclipse.rse.ui.SystemBasePlugin;
+import org.eclipse.swt.widgets.Display;
+import org.eclipse.ui.console.ConsolePlugin;
+import org.eclipse.ui.console.IConsole;
+import org.eclipse.ui.console.IConsoleManager;
+import org.eclipse.ui.console.MessageConsole;
+import org.yocto.remote.utils.ShellSession;
+import org.yocto.remote.utils.RemoteHelper;
+import org.yocto.remote.utils.CommonHelper;
+
+public class SystemtapModel extends BaseModel {
+        protected static final String DEFAULT_INIT_SCRIPT = "oe-init-build-env";
+        protected static final String SYSTEMTAP_CONSOLE = "Systemtap Console";
+
+        private static final String TASK_NAME = "systemtap command";
+
+        protected MessageConsole sessionConsole;
+        private String metadata_location;
+        private String builddir_location;
+        private String remote_host;
+        private String user_id;
+        private String systemtap_script;
+        private String systemtap_args;
+
+        Display display;
+        
+        public SystemtapModel(String metadata_location, String builddir_location, String remote_host, String user_id, String systemtap_script, String systemtap_args, Display display) {
+                super(null, TASK_NAME, "", "");
+                this.metadata_location = metadata_location;
+                this.builddir_location = builddir_location;
+                this.remote_host = remote_host;
+                this.user_id = user_id;
+                this.systemtap_script = systemtap_script;
+                this.systemtap_args = systemtap_args;
+                this.display = display;
+        }
+        
+        @Override
+        public void preProcess(IProgressMonitor monitor) 
+                        throws InvocationTargetException, InterruptedException {
+                final ITerminalServiceSubSystem terminalSubSystem = RemoteHelper.getTerminalSubSystem(host);
+                if (!terminalSubSystem.isConnected()) {
+                        try {
+                                ProgressMonitorDialog dialog = new ProgressMonitorDialog(null);
+                                dialog.run(true, true, new IRunnableWithProgress(){
+                                        @Override
+                                        public void run(IProgressMonitor monitor) {
+                                                monitor.beginTask("Connecting to remote target ...", 100);
+                                                try {
+                                                        terminalSubSystem.connect(new NullProgressMonitor(), false);
+                                                        monitor.done();
+                                                } catch (Exception e) {
+                                                        CommonHelper.showErrorDialog("Connection failure", null, e.getMessage());
+                                                        monitor.done();
+                                                }
+                                        }
+                                });
+                        } catch (OperationCanceledException e) {
+                                // user canceled, return silently
+                        } catch (Exception e) {
+                                SystemBasePlugin.logError(e.getLocalizedMessage(), e);
+                        }
+                }
+                }
+        
+                protected String changeTerm = "export TERM=vt100;";
+    
+        @Override
+        public void process(IProgressMonitor monitor) 
+                        throws InvocationTargetException, InterruptedException {
+                try {
+                        ShellSession shell = new ShellSession(ShellSession.SHELL_TYPE_BASH, 
+                                                                                                new File(this.metadata_location),new File(this.builddir_location),
+                                                                                                DEFAULT_INIT_SCRIPT, sessionConsole.newOutputStream());
+                        boolean acceptedKey = shell.ensureKnownHostKey(user_id, remote_host);
+                        if (acceptedKey) {
+                                String crosstapCmd = "crosstap " + user_id + "@" + remote_host + " " + systemtap_script;
+                                if (systemtap_args != null)
+                                        crosstapCmd = crosstapCmd + " " + systemtap_args;
+                                shell.execute(crosstapCmd);
+                        }
+                } catch (Exception e) {
+                        throw new InvocationTargetException(e,e.getMessage());
+                }
+        }
+}