changeset 204:832c61ca1f96

Add tool to check name of running process from pid Reviewed-by: rkennke, vanaltj, omajid Review-thread: http://icedtea.classpath.org/pipermail/thermostat/2012-April/000691.html
author Mario Torre <neugens.limasoftware@gmail.com>
date Wed, 04 Apr 2012 22:11:03 +0200
parents 585a09251bf6
children 38c8665824e5
files common/src/main/java/com/redhat/thermostat/tools/unix/UnixProcessUtilities.java common/src/test/java/com/redhat/thermostat/tools/unix/UnixProcessUtilitiesTest.java
diffstat 2 files changed, 212 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/common/src/main/java/com/redhat/thermostat/tools/unix/UnixProcessUtilities.java	Wed Apr 04 22:11:03 2012 +0200
@@ -0,0 +1,102 @@
+/*
+ * Copyright 2012 Red Hat, Inc.
+ *
+ * This file is part of Thermostat.
+ *
+ * Thermostat is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published
+ * by the Free Software Foundation; either version 2, or (at your
+ * option) any later version.
+ *
+ * Thermostat is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Thermostat; see the file COPYING.  If not see
+ * <http://www.gnu.org/licenses/>.
+ *
+ * Linking this code with other modules is making a combined work
+ * based on this code.  Thus, the terms and conditions of the GNU
+ * General Public License cover the whole combination.
+ *
+ * As a special exception, the copyright holders of this code give
+ * you permission to link this code with independent modules to
+ * produce an executable, regardless of the license terms of these
+ * independent modules, and to copy and distribute the resulting
+ * executable under terms of your choice, provided that you also
+ * meet, for each linked independent module, the terms and conditions
+ * of the license of that module.  An independent module is a module
+ * which is not derived from or based on this code.  If you modify
+ * this code, you may extend this exception to your version of the
+ * library, but you are not obligated to do so.  If you do not wish
+ * to do so, delete this exception statement from your version.
+ */
+
+package com.redhat.thermostat.tools.unix;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+import com.redhat.thermostat.common.utils.LoggedExternalProcess;
+import com.redhat.thermostat.common.utils.LoggingUtils;
+
+public class UnixProcessUtilities {
+
+    private static final Logger logger = LoggingUtils.getLogger(UnixProcessUtilities.class);
+    
+    private static final UnixProcessUtilities instance = new UnixProcessUtilities();
+    public static UnixProcessUtilities getInstance() {
+        return instance;
+    }
+    
+    UnixProcessUtilities() {}
+    
+    /**
+     * Returns the process name of the process with the given pid, or null
+     * if no process is running with the given pid.
+     */
+    public String getProcessName(String pid) {
+        
+        String result = null;
+        
+        List<String> commandLine = new ArrayList<>();
+        commandLine.add("ps");
+        commandLine.add("--no-heading");
+        commandLine.add("-p");
+        commandLine.add(pid);
+        
+        try {
+            Process process = createAndRunProcess(commandLine);
+            BufferedReader reader = getProcessOutput(process);
+            result = reader.readLine();
+            if (result != null) {
+                String [] output = result.split(" ");
+                result = output[output.length - 1];
+            }
+            
+        } catch (IOException e) {
+            logger.log(Level.WARNING, "can't run ps!", e);
+        }
+        
+        return result;
+    }
+    
+    public BufferedReader getProcessOutput(Process process) {
+        InputStream in = process.getInputStream();
+        InputStreamReader isr = new InputStreamReader(in);
+        return new BufferedReader(isr);
+    }
+    
+    public Process createAndRunProcess(List<String> args) throws IOException {
+        LoggedExternalProcess process = new LoggedExternalProcess(args);
+        return process.runAndReturnProcess();
+    }
+}
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/common/src/test/java/com/redhat/thermostat/tools/unix/UnixProcessUtilitiesTest.java	Wed Apr 04 22:11:03 2012 +0200
@@ -0,0 +1,110 @@
+/*
+ * Copyright 2012 Red Hat, Inc.
+ *
+ * This file is part of Thermostat.
+ *
+ * Thermostat is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published
+ * by the Free Software Foundation; either version 2, or (at your
+ * option) any later version.
+ *
+ * Thermostat is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Thermostat; see the file COPYING.  If not see
+ * <http://www.gnu.org/licenses/>.
+ *
+ * Linking this code with other modules is making a combined work
+ * based on this code.  Thus, the terms and conditions of the GNU
+ * General Public License cover the whole combination.
+ *
+ * As a special exception, the copyright holders of this code give
+ * you permission to link this code with independent modules to
+ * produce an executable, regardless of the license terms of these
+ * independent modules, and to copy and distribute the resulting
+ * executable under terms of your choice, provided that you also
+ * meet, for each linked independent module, the terms and conditions
+ * of the license of that module.  An independent module is a module
+ * which is not derived from or based on this code.  If you modify
+ * this code, you may extend this exception to your version of the
+ * library, but you are not obligated to do so.  If you do not wish
+ * to do so, delete this exception statement from your version.
+ */
+
+package com.redhat.thermostat.tools.unix;
+
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.StringReader;
+import java.util.ArrayList;
+import java.util.List;
+
+import junit.framework.Assert;
+
+import org.junit.Before;
+import org.junit.Test;
+
+public class UnixProcessUtilitiesTest {
+
+    private BufferedReader reader;
+    private BufferedReader emptyReader;
+    
+    @Before
+    public void setUp() {
+        
+        String data = "123 fluff";
+        reader = new BufferedReader(new StringReader(data));
+        emptyReader = new BufferedReader(new StringReader(""));
+    }
+    
+    @Test
+    public void getProcessName() {
+
+        final List<String> processArguments = new ArrayList<>();
+        UnixProcessUtilities process = new UnixProcessUtilities() {
+            @Override
+            public Process createAndRunProcess(List<String> args)
+                    throws IOException {
+                processArguments.addAll(args);
+                return null;
+            }
+            
+            public java.io.BufferedReader getProcessOutput(Process process) {
+                return reader;
+            };
+        };
+        
+        String result = process.getProcessName("12345");
+        Assert.assertEquals("fluff", result);
+        Assert.assertTrue(processArguments.contains("12345"));
+        
+        Assert.assertTrue(processArguments.contains("ps"));
+        Assert.assertTrue(processArguments.contains("--no-heading"));
+        Assert.assertTrue(processArguments.contains("-p"));
+    }
+    
+    @Test
+    public void getProcessNameNoOutput() {
+
+        final List<String> processArguments = new ArrayList<>();
+        UnixProcessUtilities process = new UnixProcessUtilities() {
+            @Override
+            public Process createAndRunProcess(List<String> args)
+                    throws IOException {
+                processArguments.addAll(args);
+                return null;
+            }
+            
+            public java.io.BufferedReader getProcessOutput(Process process) {
+                return emptyReader;
+            };
+        };
+        
+        String result = process.getProcessName("12345");
+        Assert.assertNull(result);
+    }    
+}