cloudflock 0.6.1 → 0.7.0

data/lib/cloudflock/target/servers/profile.rb

@@ -1,394 +1,385 @@
 require 'cloudflock/remote/ssh'
+require 'socket'
 require 'cpe'
 
-# Public: Provides methods to create a profile for a given host, mapping it to
-# available public cloud offerings, and reference data gathered during this
-# process.
-#
-# Examples
-#
-#   # Generate a profile from a shell already logged in to a remote host
-#   profile = Profile.new(ssh_object)
-#   profile.build
-#
-#   # Only determine the memory and I/O statistics for a host
-#   profile = Profile.new(ssh_object)
-#   profile.determine_memory
-#   profile.determine_io
-class CloudFlock::Target::Servers::Profile
-  # Public: Array containing warnings generated by the info gathering process.
-  attr_reader :warnings
-
-  # Public: Initialize the Profile object.
-  #
-  # shell - An SSH object open to the host to be profiled.
-  #
-  # Raises ArgumentError if passed anything but an SSH object.
-  def initialize(shell)
-    raise ArgumentError unless shell.kind_of? CloudFlock::Remote::SSH
-
-    @shell = shell
-
-    @warnings = []
-    @info = {}
-  end
-
-  # Public: Run all available determinations against the SSH object.
-  #
-  # Returns nothing.
-  def build
-    determine_version
-    determine_arch
-    determine_hostname
-    determine_memory
-    determine_cpu
-    determine_disk
-    determine_ips
-    determine_io
-    determine_web
-    determine_db
-    determine_lib
-    determine_rsync
-    determine_processes
-  end
-
-  # Public: Determine vendor and version of the OS running on the target host,
-  # and create an appropriate CPE object for it, assigning it to @info[:cpe].
-  #
-  # Returns nothing.
-  def determine_version
-    release = @shell.query("CPE", "cat /etc/system-release-cpe")
+module CloudFlock; module Target; module Servers
+  class Profile
+    # Public: List of linux distributions supported by CloudFlock
+    SUPPORTED_DISTROS = %w{Arch CentOS Debian SUSE Ubuntu RedHat Gentoo}
+
+    # Public: Initialize the Profile object.
+    #
+    # shell - An SSH object which is open to the host which will be profiled.
+    #
+    # Raises TypeError if shell is not of type SSH.
+    def initialize(shell)
+      unless shell.is_a?(CloudFlock::Remote::SSH)
+        raise(TypeError, Errstr::NOT_SSH)
+      end
 
-    begin
-      cpe = CPE.parse(release)
-      cpe.version.gsub!(/[^0-9.]/, '')
-      @info[:cpe] = cpe
-      return
-    rescue ArgumentError
-      cpe = CPE.new(part: CPE::OS, product: "linux")
+      @shell = shell
+      @warnings = []
+      @info = {}
     end
 
-    issue = @shell.query("ISSUE", "cat /etc/issue")
-
-    case issue
-    when /Arch/
-      cpe.vendor = "Arch"
-    when /CentOS/
-      cpe.vendor = "CentOS"
-      issue.gsub!(/\.\d.*$/, '')
-    when /Debian/
-      cpe.vendor = "Debian"
-      issue.gsub!(/\.\d.*$/, '')
-    when /This is/
-      cpe.vendor = "Gentoo"
-    when /SUSE/
-      cpe.vendor = "openSUSE"
-    when /Ubuntu/
-      cpe.vendor = "Ubuntu"
-    when /Red/
-      cpe.vendor = "Redhat"
-      issue.gsub!(/\.\d.*$/, '')
-    else
-      cpe.vendor = "Unknown"
+    # Public: Build the profile by calling all methods which begin with
+    # 'determine_'.
+    #
+    # Returns nothing.
+    def build
+      methods.select { |x| x =~ /^determine_/ }.each do |method|
+        self.send(method)
+      end
+      methods.select { |x| x =~ /^warning_/ }.each do |method|
+        self.send(method)
+      end
     end
 
-    cpe.version = version_number(issue)
-
-    @info[:cpe] = cpe
-  end
+    # Public: Allow access to the list of keys in @info.
+    #
+    # Returns an Array of keys in @info.
+    def keys
+      @info.keys
+    end
 
-  # Public: Determine the architecture of the target host and assign it to
-  # @info[:arch].
-  #
-  # Returns nothing.
-  def determine_arch
-    uname = @shell.query("UNAME", "uname -m")
-    case uname
-    when /x86_64/
-      @info[:arch] = 'x86_64'
-    when /i\d86/
-      @info[:arch] = 'i386'
-    else
-      @info[:arch] = "Unknown"
+    # Public: Simplify access to @info.
+    #
+    # key - Object to be used as the key in the @info Hash.
+    #
+    # Returns a value stored in @info.
+    def [](key)
+      @info[key]
     end
-  end
 
-  # Public: Determine the hostname of the target host and assign it to
-  # @info[:hostname]
-  #
-  # Returns nothing.
-  def determine_hostname
-    @info[:hostname] = @shell.query("HOST", "hostname")
-  end
+    # Public: Return server information and warnings as a Hash.
+    #
+    # Returns a Hash.
+    def to_hash
+      @info.merge({warnings: @warnings})
+    end
 
-  # Public: Determine the available and total memory on the target host as
-  # well as historical usage via sar(1), if possible.  Assign these to
-  # @info[:memory] and @info[:memory_hist], respectively.
-  #
-  # Returns nothing.
-  def determine_memory
-    result = {}
-
-    free = %w{free -m |awk '$1 ~ /Mem/ {print $2, $2-$6-$7}; $1 ~ /Swap/
-              {print $3}'|xargs}.join(' ')
-    mem = @shell.query("MEMORY", free)
-    total, used, swap = mem.split(/\s+/)
-
-    result[:total] = total.to_i
-    result[:mem_used] = used.to_i
-    result[:swap_used] = swap.to_i
-    result[:swapping?] = swap.to_i > 0
-    @info[:memory] = result
-
-    # Determine average mem and swap usage
-    result = {}
-    sar_location = @shell.query("SAR", "which sar 2>/dev/null")
-    sar_command = %w{for l in $(find /var/log/ -name 'sa??'); do sar -r -f $l |
-                     grep Average; done | awk '{I+=1; TOT=$2+$3; CACHE+=$5+$6;
-                     FREE+=$2; SWAP+=$9;} END {CACHE=CACHE/I; FREE=FREE/I;
-                     SWAP=SWAP/I; print (TOT-(CACHE+FREE))/TOT*100,
-                     SWAP;}'}.join(' ')
-
-    if sar_location =~ /bin\//
-      sar_usage = @shell.query("HIST_MEM", sar_command)
-
-      if sar_usage =~ /\d \d/
-        hist_mem, hist_swap = sar_usage.split(/ /)
-        result[:mem_used] = hist_mem.to_i
-        result[:swap_used] = hist_swap.to_i
+    private
+
+    # Internal: Determine important statistics relating to the CPU (available
+    # core count, speed).
+    #
+    # Returns nothing.
+    def determine_cpu
+      cpu = @info[:cpu] = {}
+
+      lscpu = @shell.query('LSCPU', 'lscpu')
+      if lscpu.empty?
+        cpuinfo = @shell.query('cat /proc/cpuinfo')
+        count = cpuinfo.lines.select { |l| l =~ /^processor\s*: [0-9]/}
+        speed = cpuinfo.lines.select { |l| l =~ /MHz/ }
+        cpu[:count] = count.size
+        cpu[:speed] = speed[0].to_s.gsub(/.* /, '')
+      else
+        cpu[:count] = lscpu.select { |l| l =~ /CPU\(s\)/ }.gsub(/.* /, '')
+        cpu[:speed] = lscpu.select { |l| l =~ /MHz/ }.gsub(/.* /, '')
       end
     end
 
-    @info[:memory_hist] = result
-  end
-
-  # Public: Determine the number of CPUs present on the target host and the
-  # speed of the processors.  Assign these to @info[:cpu][:count] and
-  # @info[:cpu][:speed] respectively.
-  #
-  # Returns nothing.
-  def determine_cpu
-    count_command = 'cat /proc/cpuinfo|grep "^processor\\s*: [0-9]"|wc -l'
-    speed_command = 'cat /proc/cpuinfo|grep "MHz"|head -1'
-    cpus = {}
-    cpus[:count] = @shell.query("CPU", count_command).to_i
-    cpus[:speed] = @shell.query("MHZ", speed_command).gsub(/.*: /, '').to_i
-    @info[:cpu] = cpus
-  end
-
-  # Public: Determine the amount of disk space in use on the target host and
-  # set that to @info[:disk].
-  #
-  # Returns nothing.
-  def determine_disk
-    # Use a less accurate (tends to inflate) method if du takes too long
-    df_command = "df 2>/dev/null |awk '$1 ~ /\\// {I = I + $3} END {print I}'"
-    disk = @shell.query("DISK_USED_DF", df_command) 
-
-    # Result is returned as KiB used. We need GB used.
-    @info[:disk] = disk.to_f / 1000 ** 2
-  end
-
-  # Public: Determine the number of public and private IP addressess in use on
-  # the target host and assign that to @info[:ip].
-  #
-  # Returns nothing.
-  def determine_ips
-    ips = {:private => [], :public => []}
+    # Internal: Determine the number and size of MySQL databases resident on
+    # the target host.
+    #
+    # Returns nothing.
+    def determine_databases
+      db = @info[:db] = {}
+      mysql_count_cmd = 'find /var/lib/mysql* -maxdepth 0 -type d ' \
+                        '2>/dev/null|wc -l'
+      db[:count] = @shell.query('DB_MYSQL_COUNT', mysql_count_cmd)
+      db[:count] = db[:count].to_i
+
+      mysql_size_cmd = "du -s /var/lib/mysql 2>/dev/null|awk '{print $1}'"
+      db[:size] = @shell.query('DB_MYSQL_SIZE', mysql_size_cmd)
+      db[:size] = db[:size].to_i
+    end
 
-    ip_command = %w{/sbin/ifconfig | grep 'inet addr' | egrep -v ':127' | sed
-                    -e 's/.*addr:\([0-9.]*\) .*/\1/'}.join(' ')
-    ifconfig = @shell.query("IP_CONFIG", ip_command)
+    # Internal: Determine the amount of disk space in use on the target host.
+    #
+    # Returns nothing.
+    def determine_disk
+      df_cmd = "df 2>/dev/null|awk '$1 ~ /\\// {I=I+$3} END {print I}'"
+      disk = @shell.query('DISK_DF', df_cmd)
 
-    ifconfig.each_line do |ip|
-      ip.strip!
-      ips[rfc1918?(ip)] << ip
+      # Result is expected to be in KiB.  Convert to GB.
+      @info[:disk] = disk.to_f / 1000 ** 2
     end
 
-    @info[:ip] = ips
-  end
+    # Internal: Attempt to determine which linux distribution the target host
+    # is running.
+    #
+    # Returns nothing.
+    def determine_distribution
+      # Some distros ship with a file containing the CPE for their platform;
+      # this should be used if at all possible.
+      release = @shell.query('CPE', 'cat /etc/system-release-cpe')
+      begin
+        cpe = CPE.parse(release)
+        cpe.version.gsub!(/[^0-9.]/, '')
+        @info[:cpe] = cpe
+        return
+      rescue ArgumentError
+        cpe = CPE.new(part: CPE::OS, product: 'linux')
+      end
 
-  # Public: Determine amount of historical I/O usage via sysstat and set it to
-  # @info[:io].
-  #
-  # Returns nothing.
-  def determine_io
-    io = {}
+      # Fall back to depending on /etc/issue if available
+      issue = @shell.query('ISSUE', 'cat /etc/issue')
+      cpe.vendor = distro_name(issue)
 
-    iostat = @shell.query("IOSTAT", "iostat -c | sed -n 4p | awk '{print $4}'")
-    io[:wait] = iostat.to_f
+      # If all else fails, resort to looking in release files
+      if cpe.vendor.empty?
+        release_cmd = "grep -h '^ID=' /etc/[A-Za-z]*[_-][rv]e[lr]*|head -1"
+        release = @shell.query("RELEASE", release_cmd)
+        cpe.vendor = distro_name(release)
+      end
 
-    up = @shell.query("UPTIME", "uptime | sed -e 's/.*up\\([^,]*\\),.*/\\1/'")
-    io[:uptime] = up.chomp
+      # Fall back to "Unknown"
+      cpe.vendor = "" if cpe.vendor.empty?
 
-    @info[:io] = io
-  end
+      # Version number will be determined from /etc/issue
+      cpe.version = version_number(issue)
+      @info[:cpe] = cpe
+    end
 
-  # Public: Determine the web server used on the host, and attempt to enumerate
-  # domain names configured on the server for both non-SSL and SSL; set this
-  # information to @info[:web].  Presently only Apache is supported.
-  #
-  # Returns nothing.
-  def determine_web
-    web = {}
-
-    netstat_command = %w{netstat -ntlp | awk '$4 ~ /:80$/ || $4 ~ /:443$/
-                 {sub(/^[^\/]*\//, ""); print $NF}' | head -1}.join(' ')
-    web[:binary] = @shell.query("WEB_NETSTAT", netstat_command)
-
-    unless web[:binary].empty?
-      version_command = "`which #{web[:binary]}` -v | grep version"
-      web[:version] = @shell.query("WEB_VERSION", version_command)
-      web[:version].gsub!(/.*version: /i, '')
-      binary = web[:binary] == "httpd" ? "apachectl" : "apache2ctl"
-      binary << " -S 2>&1"
-
-      web_command = "#{binary} | grep ')' | grep -vi 'default' | wc -l"
-      hosts = @shell.query("WEB_HTTP", web_command)
-      web[:hosts_http] = hosts.to_i
-
-      ssl_command = "#{binary} | grep ':443'|grep -vi 'default' | wc -l"
-      ssl_hosts = @shell.query("WEB_HTTPS", ssl_command)
-      web[:hosts_https] = ssl_hosts.to_i
+    # Internal: Determine the hostname of the target host.
+    #
+    # Returns nothing.
+    def determine_hostname
+      @info[:hostname] = @shell.query('HOST', 'hostname')
     end
 
-    @info[:web] = web
-  end
+    # Internal: Determine the amount of historical IO activity on the target
+    # host using sysstat if available.
+    #
+    # Returns nothing.
+    def determine_io
+      io = @info[:io] = {}
 
-  # Public: Determine the amount of disk usage attributable to databases, as
-  # well as database count, and set this to @info[:database].  Currently
-  # supports MySQL.
-  #
-  # Returns nothing.
-  def determine_db
-    db = {}
-    mysql_count = %w{find /var/lib/mysql/* -maxdepth 0 -type d 2>/dev/null |
-                       wc -l}.join(' ')
-    db[:count] = @shell.query("DB_MYSQL_COUNT", mysql_count)
+      iostat = @shell.query('IOSTAT', "iostat -c|sed -n 4p|awk '#{print $4}'")
+      io[:wait] = iostat.to_f
 
-    mysql_size = "du -s /var/lib/mysql 2>/dev/null | awk '{print $1}'"
-    db[:size] = @shell.query("DB_MYSQL_SIZE", mysql_size)
+      up = @shell.query('UPTIME', "uptime|sed -e 's/.*up\\([^,]*\\),.*/\\1/'")
+      io[:uptime] = up.chomp
+    end
 
-    db[:count] = db[:count].to_i
-    db[:size] = db[:size].to_i
+    # Internal: Determine IPv4 addresses in use by the target host, splitting
+    # them into public and private groups.
+    #
+    # Returns nothing.
+    def determine_ips
+      ips = @info[:ip] = {private: [], public: []}
 
-    @info[:db] = db
-  end
+      ifc_cmd = "/sbin/ifconfig|grep 'inet addr'|grep -v ':127'|sed -e " \
+                "'s/.*addr:\([0-9.]*\) .*/\\1/'"
+      ifconfig = @shell.query('IFCONFIG', ifc_cmd)
 
-  # Public: Gather information about the currently installed libc, ruby, perl,
-  # python and php versions on the host, then set these values to @info[:lib].
-  #
-  # Returns nothing.
-  def determine_lib
-    lib = {}
+      ifconfig.each_line do |ip|
+        ip.strip!
+        ips[rfc1918?(ip)] << ip
+      end
+    end
 
-    libc_command = %w{ls -al `find /lib /usr/lib -name 'libc.so*' | head -1` |
-                     sed 's/.*-> //'}.join(' ')
-    lib[:libc] = @shell.query("LIBC", libc_command)
-    lib[:libc].gsub!(/^.*-|\.so$/, '')
+    # Internal: Determine common libraries installed on the system.
+    #
+    # Returns nothing.
+    def determine_libraries
+      lib = @info[:lib] = {}
 
-    lib[:perl] = @shell.query("PERL", "perl -e 'print $^V;'")
-    lib[:perl] = lib[:perl].gsub(/^v([0-9.]*).*/, '\1')
+      libc_cmd = "ls -la `find /lib /usr/lib -name 'libc.so*'|head -1`|" \
+                 "sed 's/.*-> //'"
+      lib[:libc] = @shell.query('LIBC', libc_cmd)
+      lib[:libc].gsub!(/^.*-|\.so$/, '')
 
-    python_command = "python -c 'import sys; print sys.version' 2>/dev/null"
-    lib[:python] = @shell.query("PYTHON", python_command)
-    lib[:python] = lib[:python].gsub(/^([0-9.]*).*/m, '\1')
+      lib[:perl] = @shell.query('PERL', 'perl -e "print $^V;"')
+      lib[:perl].gsub!(/^v([0-9.]*).*/, '\1')
 
-    ruby_command = "ruby -e 'print RUBY_VERSION' 2>/dev/null"
-    lib[:ruby] = @shell.query("RUBY", ruby_command)
+      python_cmd = 'python -c "import sys; print sys.version" 2>/dev/null'
+      lib[:python] = @shell.query('PYTHON', python_cmd)
+      lib[:python].gsub!(/([0-9.]*).*/m, '\1')
 
-    lib[:php] = @shell.query("PHP_VER", "php -v 2>/dev/null | head -1")
-    lib[:php] = lib[:php].gsub(/^PHP ([0-9.]*).*/, '\1')
+      ruby_cmd = 'ruby -e "print RUBY_VERSION" 2>/dev/null'
+      lib[:ruby] = @shell.query('RUBY', ruby_cmd)
 
-    @info[:lib] = lib
-  end
+      lib[:php] = @shell.query('PHP', 'php -v 2>/dev/null|head -1')
+      lib[:php].gsub!(/^PHP ([0-9.]*).*/, '\1')
+    end
 
-  # Public: Check for the existence of rsync(1) on the host.  Set @info[:rsync]
-  # accordingly.
-  #
-  # Returns nothing.
-  def determine_rsync
-    rsync_command = %w{which rsync 2>/dev/null || ([ -f
-                       /root/.rackspace/rsync ] && printf
-                       '/root/.rackspace/rsync') || printf
-                       'NONE'}.join(' ')
+    # Internal: Determine the total amount of memory on the target host, the
+    # amount of memory in use, and the amount of swap space being used.
+    #
+    # Returns nothing.
+    def determine_memory
+      result = @info[:memory] = {}
+
+      free_cmd = "free -m|awk '$1 ~ /Mem/ {print $2, $2-$6-$7}; $1 ~ /Swap/ " \
+                 "{print $3}'|xargs"
+      mem = @shell.query('MEMORY', free_cmd)
+      total, used, swap = mem.split(/\s+/)
+
+      result[:total] = total.to_i
+      result[:mem_used] = used.to_i
+      result[:swap_used] = swap.to_i
+      result[:swapping?] = swqp.to_i > 0
+    end
 
-    rsync = @shell.query("RSYNC", rsync_command)
+    # Internal: If the sysstat suite is installed on the target host, determine
+    # the average amount of memory and swap used over whatever historical
+    # period sar is able to represent.
+    #
+    # Returns nothing.
+    def determine_memory_history
+      result = @info[:memory_hist] = {}
+      sar_cmd = "for l in $(find /var/log -name 'sa??');do sar -r -f $l|" \
+                "grep Average;done|awk '{I+=1;TOT=$2+$3;CACHE+=$5+$6;"    \
+                "FREE+=$2;SWAP+=$9;} END {CACHE=CACHE/I;FREE=FREE/I;"     \
+                "SWAP=SWAP/I;print (TOT-(CACHE+FREE))/TOT*100,SWAP;}'"
+
+      sar_location = @shell.query('SAR_LOCATION', 'which sar 2>/dev/null')
+      if sar_location =~ /bin\//
+        sar_usage = @shell.query('SAR', sar_cmd)
+
+        if sar_usage =~/\d \d/
+          mem, swap = sar_usage.split(/ /)
+          result[:mem_used] = mem
+          result[:swap_used] = swap
+        end
+      end
+    end
 
-    @info[:rsync] = (rsync =~ /NONE/).nil? ? rsync : false
-  end
+    # Internal: Gather a list of all listening ports on the target host.
+    #
+    # Returns nothing.
+    def determine_ports
+      ports = @info[:ports] = {}
+
+      netstat = @shell.query('NETSTAT', "netstat -ntlp|awk '{print $4, $NF}'")
+      netstat.lines.each do |line|
+        net, process = line.split(/\s+/, 2)
+        process = process.split(/\//, 2)[1]
+        net = net.gsub(/([0-9.:]+):([0-9]+)/, '\1 \2')
+        net, port = net.split(/ /, 2)
+
+        ports[net] ||= {}
+        ports[net][port] = process
+      end
+    end
 
-  # Public: Check the process listing and store it in 
-  #
-  # Returns nothing.
-  def determine_processes
-    ps_command = 'ps aux'
+    # Internal: Gather a list of running processes on the target host.
+    #
+    # Returns nothing.
+    def determine_processes
+      procs = @shell.query('PROCESSES', 'ps aux')
+      @info[:processes] = procs.gsub(/\r/, '').split(/\n/)
+    end
 
-    processes = @shell.query("PS_LIST", ps_command)
-    @info[:processes] = processes.gsub(/\r/, '').split(/\n/)
+    # Internal: Locate rsync on the target host.
+    #
+    # Returns nothing.
+    def determine_rsync
+      rsync = @shell.query('RSYNC', 'which rsync 2>/dev/null')
+
+      if rsync.empty?
+        rsync_cmd = '[ -f /root/.cloudflock/rsync ] && printf ' \
+                    '"/root/.rackspace/rsync"'
+        rsync = @shell.query('LOCAL_RSYNC', rsync_cmd)
+        rsync = nil if rsync.empty?
+      end
 
-    unless @info[:processes].grep(/psa/i).empty?
-      @warnings << "Server likely to be running Plesk"
+      @info[:rsync] = rsync
     end
-    unless @info[:processes].grep(/cpanellogd/i).empty?
-      @warnings << "Server likely to be running cPanel"
+
+    # Internal: Determine the architecture of the target host.
+    #
+    # Returns nothing.
+    def determine_system_architecture
+      @info[:arch] = @shell.query('UNAME', 'uname -m')
+      @info[:arch].gsub!(/i\d86/, 'i386')
     end
-  end
 
-  # Public: Simplify access to @info.
-  #
-  # key - Key to check in @info.
-  #
-  # Return value contained in @info by key.
-  def [](key)
-    @info[key]
-  end
+    # Internal: Determine which web server, if any, is running on the target
+    # host.  If the web server is supported, discover how many HTTP/HTTPS
+    # domains are configured on the server.
+    #
+    # Returns nothing.
+    def determine_web
+      web = @info[:web] = {}
+      netstat_cmd = 'netstat -ntlp|awk \'$4 ~ /:80$/ || $4 ~ /:443$/ ' \
+                    '{sub (/^[^\/]*\//, ""); print $NF}\'|head -1'
+      web[:binary] = @shell.query('WEB_NETSTAT', netstat_command)
+
+      unless web[:binary].empty?
+        if web[:binary] == 'httpd' || web[:binary] == 'apache2'
+          version_cmd = "`which #{web[:binary]}` -v|grep version"
+          web[:version] = @shell.query('WEB_VERSION', version_cmd)
+          web[:version].gsub!(/.*version: /i, '')
+
+          ctl_cmd = web[:binary] == 'httpd' ? 'apachectl' : 'apache2ctl'
+          ctl_cmd << ' -S 2>&1'
+
+          web_cmd = "#{ctl_cmd}|grep -vi 'default'|wc -l"
+          hosts = @shell.query('WEB_HOSTS', web_cmd)
+
+          ssl = hosts.lines.select { |line| line =~ /:443([^\d]|$)/ }
+          http = hosts.lines - ssl
+          web[:hosts_http] = http.length
+          web[:hosts_https] = https.length
+        end
+      end
+    end
 
-  # Public: Allow access to the list of keys extant in @info.
-  #
-  # Return an Array of keys present in @info.
-  def keys
-    @info.keys
-  end
+    # Internal: Check for signs of running Plesk.
+    #
+    # Returns nothing.
+    def warning_plesk
+      unless @info[:processes].to_a.grep(/psa/i).empty?
+        @warnings << "Server likely to be running Plesk"
+      end
+    end
 
-  # Public: Return server information and warnings as a Hash.  Useful for
-  # calling Hash#merge.
-  #
-  # Return info Hash.
-  def to_hash
-    @info.merge({warnings: @warnings})
-  end
+    # Internal: Check for signs of running cPanel.
+    #
+    # Returns nothing.
+    def warning_webmin
+      unless @info[:processes].to_a.grep(/cpanel/i).empty?
+        @warnings << "Server likely to be running cPanel"
+      end
+    end
 
-  # Internal: Deconstruct the version String provided in order to strip out
-  # extraneous data before and after the version number.  Version number
-  # strings are defined as beginning with a digit, ending with a digit, and
-  # containing nothing but digits and at most one decimal point.
-  #
-  # version - A String containing version number of the OS on the target host.
-  #
-  # Returns a String containing the parsed version string.
-  def version_number(version)
-    if version =~ /\d/
-      version.gsub(/^[^\d]*/, '').gsub(/[^\d]*$/, '').gsub(/(\d*\.\d*).*/, '\1')
-    else
-      "-"
+    # Internal: Search for names of supported Linux distributions in a string
+    # which may contain the name of the distribution currently installed on
+    # the target host.
+    #
+    # Returns a String.
+    def distro_name(str)
+      SUPPORTED_DISTROS.select do |distro|
+        Regexp.new(distro, Regexp::IGNORECASE).match(str)
+      end[0].to_s
     end
-  end
 
-  # Internal: Determine whether a given IP resides in a block designated as
-  # private by RFC1918.
-  #
-  # ip - A String containing an IP address.
-  #
-  # Returns :private or :public Symbol based on whether the IP is private.
-  def rfc1918?(ip)
-    octets = ip.split /\./
-    if octets[0] == "10" || (octets[0] == "192" && octets[1] == "168")
-      return :private
-    elsif octets[0] == "172" && octets[1].to_i >=16 && octets[1].to_i <= 31
-      return :private
+    # Internal: Inspect a String which may contain information regarding the
+    # version of the Linux distribution running on the target host.
+    #
+    # Returns nothing.
+    def version_number(str)
+      if str =~ /\d/
+        str.gsub(/^[^\d]*/, '').gsub(/[^\d]*$/, '').gsub(/(\d*\.\d*).*/, '\1')
+      else
+        '-'
+      end
     end
 
-    :public
+    # Internal: Determine if a v4 IP address belongs to a private (RFC 1918)
+    # network.
+    #
+    # ip - String containing an IP.
+    #
+    # Returns either the symbol :public or :private.
+    def rfc1918?(ip)
+      return :private if Addrinfo.ip(ip).ipv4_private?
+
+      :public
+    end
   end
-end
+end; end; end