Java tutorial
/** * NOTE: This copyright does *not* cover user programs that use Hyperic * program services by normal system calls through the application * program interfaces provided as part of the Hyperic Plug-in Development * Kit or the Hyperic Client Development Kit - this is merely considered * normal use of the program, and does *not* fall under the heading of * "derived work". * * Copyright (C) [2012], VMware, Inc. * This file is part of Hyperic . * * Hyperic is free software; you can redistribute it and/or modify * it under the terms version 2 of the GNU General Public License as * published by the Free Software Foundation. This program 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 this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 * USA. * */ package org.hyperic.hq.product; import java.awt.image.FilteredImageSource; import java.io.BufferedReader; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Map.Entry; import java.util.Set; import java.util.StringTokenizer; import java.util.regex.Matcher; import java.util.regex.Pattern; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.hyperic.sigar.Sigar; import org.hyperic.sigar.SigarProxy; import org.hyperic.sigar.SigarProxyCache; import org.hyperic.util.config.ConfigResponse; public class DetectionUtil { private static Log log = LogFactory.getLog(DetectionUtil.class.getName()); private static Sigar sigarImpl = null; private static SigarProxy sigar = null; private static final String OS_TYPE; private static final boolean IS_WINDOWS; private static final boolean IS_UNIX; static { OS_TYPE = System.getProperty("os.name").toLowerCase(); IS_WINDOWS = OS_TYPE.contains("win"); IS_UNIX = OS_TYPE.contains("nix") || OS_TYPE.contains("nux") || OS_TYPE.contains("sunos") || OS_TYPE.contains("mac os x"); } /** * This method finds all the ports the provided pid listens on and adds them as a list * to the provided ConfigResponse instance * @param pid - the process id for which we want to get the listening ports * @param cf - usually the product config * @param recursive - if true the population of the listening port will use all the child processes of the pid */ public static void populateListeningPorts(long pid, ConfigResponse cf, boolean recursive) { Set<Long> pids = new HashSet<Long>(); pids.add(pid); if (recursive) { pids.addAll(getAllChildPid(pid)); } populateListeningPorts(pids, cf); } /** * This method finds all the ports the provided pids are listening on and adds them as a list * to the provided ConfigResponse instance * @param pids - the ids of the processes we want to get the listening ports for * @param cf - usually the product config */ public static void populateListeningPorts(Set<Long> pids, ConfigResponse cf) { if (pids.isEmpty()) { return; } if (IS_WINDOWS) { populatePortsOnWindows(pids, cf); } else if (IS_UNIX) { populatePortsOnUnix(pids, cf); } } /** * This method finds the listening ports for the provided pids on * Unix platform by using netstat * @param pids * @param cf */ private static void populatePortsOnUnix(Set<Long> pids, ConfigResponse cf) { String cmd = "netstat -lnptu"; //build a string of all the provided pids StringBuilder pidsStr = new StringBuilder(); for (Long pid : pids) { pidsStr.append(',').append(pid); } Set<String> ports = new HashSet<String>(); String line; BufferedReader input = null; try { // Run netstat Process process = Runtime.getRuntime().exec(cmd); input = new BufferedReader(new InputStreamReader(process.getInputStream())); while ((line = input.readLine()) != null) { if (!line.contains("LISTEN")) { continue; } try { line = line.trim(); String pid = line.substring(line.lastIndexOf(" ")); pid = pid.trim(); pid = pid.substring(0, pid.indexOf("/")); //check that the pid that listens on this port is one of the provided pids if (!pids.contains(Long.valueOf(pid))) { continue; } //get the port number line = line.substring(line.indexOf(":")); line = line.substring(0, line.indexOf(" ")); line = line.substring(line.lastIndexOf(":") + 1); line = line.trim(); if (isNumber(line)) { ports.add(line); } } catch (Exception e) { continue; } } } catch (Exception e) { log.warn("Error populating ports for '" + pidsStr + "' ", e); } finally { if (input != null) { try { input.close(); } catch (IOException e) { log.error(e); } } } updatePortsInConfigResponse(cf, ports); } private static void populatePortsOnWindows(Set<Long> pids, ConfigResponse cf) { String cmd = "netstat -ano"; //build a string of all the provided pids StringBuilder pidsStr = new StringBuilder(); for (Long pid : pids) { pidsStr.append(',').append(pid); } Set<String> ports = new HashSet<String>(); String line; BufferedReader input = null; try { // Run netstat Process process = Runtime.getRuntime().exec(cmd); input = new BufferedReader(new InputStreamReader(process.getInputStream())); while ((line = input.readLine()) != null) { if (!(line.trim().startsWith("TCP") || line.startsWith("UDP")) || !line.contains("LISTENING")) { continue; } try { //check that the pid that listens on this port is one of the provided pids long pid = Long.valueOf(line.trim().substring(line.trim().lastIndexOf(" ")).trim()); if (!pids.contains(pid)) { continue; } //get the port number if (line.contains("[::")) { line = line.replaceAll("[::", "").trim(); } line = line.substring(line.indexOf(":") + 1); line = line.substring(0, line.indexOf(" ")); line = line.trim(); if (isNumber(line)) { ports.add(line); } } catch (Exception e) { continue; } } } catch (Exception e) { log.warn("Error populating ports for '" + pidsStr + "' ", e); } finally { if (input != null) { try { input.close(); } catch (IOException e) { log.error(e); } } } updatePortsInConfigResponse(cf, ports); } /** * This method updates the ConfigResponse with the listening ports (if there are any) * @param cf * @param ports */ private static void updatePortsInConfigResponse(ConfigResponse cf, Set<String> ports) { if (!ports.isEmpty()) { StringBuilder portsStr = new StringBuilder(); for (String port : ports) { portsStr.append(',').append(port); } cf.setValue(Collector.LISTEN_PORTS, portsStr.substring(1)); } } /** * This method finds all the childs of the provided process * @param parentPid */ public static Set<Long> getAllChildPid(long parentPid) { Set<Long> childPids = new HashSet<Long>(); if (IS_UNIX) { String cmd = "ps -o pid --no-headers --ppid " + String.valueOf(parentPid); String line; BufferedReader input = null; try { Process process = Runtime.getRuntime().exec(cmd); input = new BufferedReader(new InputStreamReader(process.getInputStream())); while ((line = input.readLine()) != null) { line = line.trim(); if (!line.equals("") && isNumber(line)) { Long childPid = Long.valueOf(line); childPids.addAll(getAllChildPid(childPid)); childPids.add(childPid); } } } catch (Exception e) { } finally { if (input != null) { try { input.close(); } catch (IOException e) { log.error(e); } } } } else if (IS_WINDOWS) { final String cmd = "wmic process get processid,parentprocessid"; String line; BufferedReader input = null; try { Process process = Runtime.getRuntime().exec(cmd); input = new BufferedReader(new InputStreamReader(process.getInputStream())); //[HQ-4176] - don't block this thread if (!input.ready()) { return childPids; } long lPpid = -1; String sPpid = ""; String sCpid = ""; long lCpid = -1; while ((line = input.readLine()) != null) { try { line = line.trim(); sPpid = line.substring(0, line.indexOf(" ")); lPpid = Long.valueOf(sPpid); if (parentPid == lPpid) { sCpid = line.substring(line.indexOf(" ")).trim(); lCpid = Long.valueOf(sCpid); childPids.addAll(getAllChildPid(lCpid)); childPids.add(lCpid); } else { continue; } } catch (Exception e) { continue; } } } catch (Exception e) { log.error(e); } finally { if (input != null) { try { input.close(); } catch (IOException e) { log.error(e); } } } } return childPids; } /** * Returns true if the provided string is a number (not a float number) */ private static boolean isNumber(String value) { try { Long.valueOf(value); return true; } catch (Exception e) { log.error(e); return false; } } protected static SigarProxy getSigar() { if (sigar == null) { int timeout = 10 * 60 * 1000; //10 minutes sigarImpl = new Sigar(); sigar = SigarProxyCache.newInstance(sigarImpl, timeout); } return sigar; } public static String getMacs(String name) throws IOException { //String cmd = "cmd /C powershell Get-VMNetworkAdapter " + name; String cmd = "@echo Name IsManagementOs VMName SwitchName MacAddress Status IPAddresses"; String line; BufferedReader input = null; List<String> macs = new ArrayList<String>(); try { Process process = Runtime.getRuntime().exec(cmd); input = new BufferedReader(new InputStreamReader(process.getInputStream())); int i = -1; while ((line = input.readLine()) != null) { line = line.trim(); if ("".equals(line)) { continue; } if (i == -1) { i = line.indexOf("MacAddress"); } else { macs.add(line.substring(i, line.indexOf(" ", i))); } } StringBuilder sb = new StringBuilder(); for (String mac : macs) { sb.append(',').append(mac); } return sb.toString().substring(1); } catch (Exception e) { log.error(e); return null; } finally { if (input != null) { try { input.close(); } catch (IOException e) { log.error(e); } } } } public static void main(String[] args) throws Throwable { final String in = "Name IsManagementOs VMName SwitchName MacAddress Status IPAddresses"; String line; BufferedReader input = null; List<String> macs = new ArrayList<String>(); Process process = Runtime.getRuntime().exec( "@echo Name IsManagementOs VMName SwitchName MacAddress Status IPAddresses"); try { input = new BufferedReader(new InputStreamReader(process.getInputStream())); int i = -1; while ((line = input.readLine()) != null) { line = line.trim(); System.out.println(line); } } catch (Exception e) { log.error(e); } finally { if (input != null) { try { input.close(); } catch (IOException e) { log.error(e); } } } }//EOM /** * * @param wmiObjName * @param filter a name-value pair. The first '-' sign seperates between the name and the value. The rest which follows are part of the value's name * @param col * @param name * @return * @throws PluginException */ public static Set<String> getWMIObj(String namespace, String wmiObjName, Map<String, String> filters, String col, String name) throws PluginException { if (wmiObjName == null || "".equals(wmiObjName)) { throw new PluginException("object property not specified in the template of " + name); } StringBuilder sb = new StringBuilder().append("wmic /NAMESPACE:\\\\" + namespace + " path ") .append(wmiObjName); if (filters != null && !filters.isEmpty()) { sb.append(" WHERE \""); int num = 0; for (Entry<String, String> filterEntry : filters.entrySet()) { String filterFieldAndVal = filterEntry.getKey(); String operator = filterEntry.getValue(); int i = filterFieldAndVal.indexOf("-"); sb.append(filterFieldAndVal.substring(0, i)).append(" ").append(operator).append(" '") .append(filterFieldAndVal.substring(i + 1, filterFieldAndVal.length())).append("'"); num++; if (num < filters.size()) { sb.append(" and "); } } sb.append("\""); } sb.append(" get"); if (col != null && !"".equals(col)) { sb.append(" " + col); } sb.append(" /format:textvaluelist.xsl"); String cmd = sb.toString(); if (log.isDebugEnabled()) { log.debug("cmd=" + cmd); } BufferedReader input = null; try { Process process = Runtime.getRuntime().exec(cmd); input = new BufferedReader(new InputStreamReader(process.getInputStream())); String line; Set<String> obj = new HashSet<String>(); StringTokenizer st; while ((line = input.readLine()) != null) { line = line.trim(); st = new StringTokenizer(line, "="); while (st.hasMoreElements()) { String k = ((String) st.nextElement()).trim(); String v = ((String) st.nextElement()).trim(); obj.add(v); } } return obj; } catch (IOException e) { throw new PluginException(e); } finally { if (input != null) { try { input.close(); } catch (IOException e) { throw new PluginException(e); } } } } }