rightscale-ohai 0.3.3.1

data/lib/ohai/plugins/windows/hostname.rb

@@ -0,0 +1,25 @@
+#
+# Author:: James Gartrell (<jgartrel@gmail.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'ruby-wmi'
+require 'socket'
+
+host = WMI::Win32_ComputerSystem.find(:first)
+hostname "#{host.Name}"
+#hostname "#{Socket.gethostname}"
+fqdn "#{Socket.gethostbyname(Socket.gethostname).first}"

data/lib/ohai/plugins/windows/kernel.rb

@@ -0,0 +1,75 @@
+#
+# Author:: James Gartrell (<jgartrel@gmail.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'ruby-wmi'
+
+def machine_lookup(sys_type)
+  return "i386" if sys_type.eql?("X86-based PC")
+  return "x86_64" if sys_type.eql?("x64-based PC")
+  sys_type
+end
+
+def os_lookup(sys_type)
+  return "Unknown" if sys_type.to_s.eql?("0")
+  return "Other" if sys_type.to_s.eql?("1")
+  return "MSDOS" if sys_type.to_s.eql?("14")
+  return "WIN3x" if sys_type.to_s.eql?("15")
+  return "WIN95" if sys_type.to_s.eql?("16")
+  return "WIN98" if sys_type.to_s.eql?("17")
+  return "WINNT" if sys_type.to_s.eql?("18")
+  return "WINCE" if sys_type.to_s.eql?("19")
+  return nil
+end
+
+host = WMI::Win32_OperatingSystem.find(:first)
+kernel[:os_info] = Mash.new
+host.properties_.each do |p|
+  kernel[:os_info][p.name.wmi_underscore.to_sym] = host[p.name]
+end
+
+kernel[:name] = "#{kernel[:os_info][:caption]}"
+kernel[:release] = "#{kernel[:os_info][:version]}"
+kernel[:version] = "#{kernel[:os_info][:version]} #{kernel[:os_info][:csd_version]} Build #{kernel[:os_info][:build_number]}"
+kernel[:os] = os_lookup(kernel[:os_info][:os_type]) || languages[:ruby][:host_os]
+
+host = WMI::Win32_ComputerSystem.find(:first)
+kernel[:cs_info] = Mash.new
+host.properties_.each do |p|
+  kernel[:cs_info][p.name.wmi_underscore.to_sym] = host[p.name]
+end
+
+kernel[:machine] = machine_lookup("#{kernel[:cs_info][:system_type]}")
+
+kext = Mash.new
+pnp_drivers = Mash.new
+
+drivers = WMI::Win32_PnPSignedDriver.find(:all)
+drivers.each do |driver|
+  pnp_drivers[driver.DeviceID] = Mash.new
+  driver.properties_.each do |p|
+    pnp_drivers[driver.DeviceID][p.name.wmi_underscore.to_sym] = driver[p.name]
+  end
+  if driver.DeviceName
+    kext[driver.DeviceName] = pnp_drivers[driver.DeviceID]
+    kext[driver.DeviceName][:version] = pnp_drivers[driver.DeviceID][:driver_version]
+    kext[driver.DeviceName][:date] = pnp_drivers[driver.DeviceID][:driver_date] ? pnp_drivers[driver.DeviceID][:driver_date].to_s[0..7] : nil
+  end 
+end
+
+kernel[:pnp_drivers] = pnp_drivers
+kernel[:modules] = kext

data/lib/ohai/plugins/windows/network.rb

@@ -0,0 +1,114 @@
+#
+# Author:: James Gartrell (<jgartrel@gmail.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'ruby-wmi'
+
+def encaps_lookup(encap)
+  return "Ethernet" if encap.eql?("Ethernet 802.3")
+  encap
+end
+
+def derive_bcast(ipaddr, ipmask, zero_bcast = false)
+  begin
+    ipaddr_int = ipaddr.split(".").collect{ |x| x.to_i}.pack("C4").unpack("N").first
+    ipmask_int = ipmask.split(".").collect{ |x| x.to_i}.pack("C4").unpack("N").first
+    if zero_bcast
+      bcast_int = ipaddr_int & ipmask_int
+    else
+      bcast_int = ipaddr_int | 2 ** 32 - ipmask_int - 1  
+    end  
+    bcast = [bcast_int].pack("N").unpack("C4").join(".")                                     
+    return bcast
+  rescue
+    return nil
+  end
+end
+
+iface = Mash.new
+iface_config = Mash.new
+iface_instance = Mash.new
+
+adapters = WMI::Win32_NetworkAdapterConfiguration.find(:all)
+adapters.each do |adapter|
+    i = adapter.Index
+    iface_config[i] = Mash.new
+    adapter.properties_.each do |p|
+      iface_config[i][p.name.wmi_underscore.to_sym] = adapter[p.name]
+    end
+end
+
+adapters = WMI::Win32_NetworkAdapter.find(:all)
+adapters.each do |adapter|
+    i = adapter.Index
+    iface_instance[i] = Mash.new
+     adapter.properties_.each do |p|
+      iface_instance[i][p.name.wmi_underscore.to_sym] = adapter[p.name]
+    end
+end
+
+iface_instance.keys.each do |i|
+  if iface_config[i][:ip_enabled] and iface_instance[i][:net_connection_id] and iface_instance[i][:interface_index]
+    cint = sprintf("0x%X", iface_instance[i][:interface_index])
+    iface[cint] = Mash.new
+    iface[cint][:configuration] = iface_config[i]
+    iface[cint][:instance] = iface_instance[i]
+
+    iface[cint][:counters] = Mash.new
+    iface[cint][:addresses] = Mash.new
+    iface[cint][:configuration][:ip_address].each_index do |i|
+      begin
+         if iface[cint][:configuration][:ip_address][i] =~ /./
+           iface[cint][:addresses][iface[cint][:configuration][:ip_address][i]] = {
+             "family"    => "inet",
+             "netmask"   => iface[cint][:configuration][:ip_subnet][i],
+             "broadcast" => derive_bcast( iface[cint][:configuration][:ip_address][i],
+                                          iface[cint][:configuration][:ip_subnet][i],
+                                          iface[cint][:configuration][:ip_use_zero_broadcast]
+             )
+           }
+         end
+      rescue
+      end
+    end
+    iface[cint][:configuration][:mac_address].each do |mac_addr|
+      iface[cint][:addresses][mac_addr] = {
+        "family"    => "lladdr"
+      }
+    end
+    iface[cint][:mtu] = iface[cint][:configuration][:mtu]
+    iface[cint][:type] = iface[cint][:instance][:adapter_type]
+    iface[cint][:arp] = {}
+    iface[cint][:encapsulation] = encaps_lookup(iface[cint][:instance][:adapter_type])
+  end
+end
+
+cint=nil
+from("arp /a").split("\n").each do |line|
+  if line == ""
+    cint = nil
+  end
+  if line =~ /^Interface:\s+(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})\s+[-]+\s+(0x\d+)/
+    cint = $2
+  end
+  next unless iface[cint]
+  if line =~ /^\s+(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})\s+([a-fA-F0-9\:-]+)/
+    iface[cint][:arp][$1] = $2.gsub("-",":").downcase
+  end
+end
+
+network["interfaces"] = iface

data/lib/ohai/plugins/windows/platform.rb

@@ -0,0 +1,26 @@
+#
+# Author:: James Gartrell (<jgartrel@gmail.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+# After long discussion in IRC the "powers that be" have come to a concensus
+# that there is no other Windows platforms exist that were not based on the
+# Windows_NT kernel, so we herby decree that "windows" will refer to all
+# platforms built upon the Windows_NT kernel and have access to win32 or win64
+# subsystems.
+platform os
+platform_version kernel['release']
+

data/lib/ohai/system.rb

@@ -0,0 +1,226 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'rubygems'
+require 'extlib'
+require 'ohai/log'
+require 'ohai/mixin/from_file'
+require 'ohai/mixin/command'
+require 'ohai/mixin/string'
+require 'json'
+
+module Ohai
+  class System
+    attr_accessor :data, :seen_plugins
+    
+    include Ohai::Mixin::FromFile
+    include Ohai::Mixin::Command
+    
+    def initialize
+      @data = Mash.new
+      @seen_plugins = Hash.new
+      @providers = Mash.new
+      @plugin_path = ""
+    end
+    
+    def [](key)
+      @data[key]
+    end
+    
+    def []=(key, value)
+      @data[key] = value
+    end
+    
+    def each(&block)
+      @data.each do |key, value|
+        block.call(key, value)
+      end
+    end
+    
+    def attribute?(name)
+      @data.has_key?(name) 
+    end
+    
+    def set(name, *value)
+      set_attribute(name, *value)
+    end
+    
+    def from(cmd)
+      status, stdout, stderr = run_command(:command => cmd)
+      return "" if stdout.nil?
+      stdout.chomp!.strip
+    end
+    
+    def provides(*paths)
+      paths.each do |path|
+        parts = path.split('/')
+        h = @providers
+        unless parts.length == 0
+          parts.shift if parts[0].length == 0
+          parts.each do |part|
+            h[part] ||= Mash.new
+            h = h[part]
+          end
+        end
+        h[:_providers] ||= []
+        h[:_providers] << @plugin_path
+      end
+    end
+    
+    # Set the value equal to the stdout of the command, plus run through a regex - the first piece of match data is the value.
+    def from_with_regex(cmd, *regex_list)
+      regex_list.flatten.each do |regex|
+        status, stdout, stderr = run_command(:command => cmd)
+        return "" if stdout.nil?
+        stdout.chomp!.strip
+        md = stdout.match(regex)
+        return md[1]
+      end
+    end
+    
+    def set_attribute(name, *value)
+      @data[name] = *value
+      @data[name]
+    end
+    
+    def get_attribute(name)
+      @data[name]
+    end
+    
+    def all_plugins
+      require_plugin('os')
+      
+      Ohai::Config[:plugin_path].each do |path|
+        [ 
+          Dir[File.join(path, '*')], 
+          Dir[File.join(path, @data[:os], '**', '*')] 
+        ].flatten.each do |file|
+          file_regex = Regexp.new("#{path}#{File::SEPARATOR}(.+).rb$")
+          md = file_regex.match(file)
+          if md
+            plugin_name = md[1].gsub(File::SEPARATOR, "::")
+            require_plugin(plugin_name) unless @seen_plugins.has_key?(plugin_name)
+          end
+        end
+      end
+    end
+
+    def collect_providers(providers)
+      refreshments = []
+      if providers.is_a?(Mash)
+        providers.keys.each do |provider|
+          if provider.eql?("_providers")
+            refreshments << providers[provider]
+          else
+            refreshments << collect_providers(providers[provider])
+          end
+        end
+      else
+        refreshments << providers
+      end
+      refreshments.flatten.uniq
+    end
+    
+    def refresh_plugins(path = '/')
+      parts = path.split('/')
+      if parts.length == 0
+        h = @providers
+      else
+        parts.shift if parts[0].length == 0
+        h = @providers
+        parts.each do |part|
+          break unless h.has_key?(part)
+          h = h[part]
+        end
+      end
+
+      refreshments = collect_providers(h)
+      Ohai::Log.debug("Refreshing plugins: #{refreshments.join(", ")}")
+      
+      refreshments.each do |r|
+        @seen_plugins.delete(r) if @seen_plugins.has_key?(r)
+      end
+      refreshments.each do |r|
+        require_plugin(r) unless @seen_plugins.has_key?(r)        
+      end
+    end
+    
+    def require_plugin(plugin_name, force=false)
+      unless force
+        return true if @seen_plugins[plugin_name]
+      end
+      
+      @plugin_path = plugin_name
+      
+      filename = "#{plugin_name.gsub("::", File::SEPARATOR)}.rb"
+            
+      Ohai::Config[:plugin_path].each do |path|
+        check_path = File.expand_path(File.join(path, filename))
+        begin
+          @seen_plugins[plugin_name] = true
+          Ohai::Log.debug("Loading plugin #{plugin_name}")
+          from_file(check_path)
+          return true
+        rescue IOError => e
+          Ohai::Log.debug("No #{plugin_name} at #{check_path}")
+        rescue Exception,Errno::ENOENT => e
+          Ohai::Log.debug("Plugin #{plugin_name} threw exception #{e.inspect}")
+        end
+      end
+    end
+    
+    # Sneaky!  Lets us stub out require_plugin when testing plugins, but still
+    # call the real require_plugin to kick the whole thing off.
+    alias :_require_plugin :require_plugin
+    
+    # Serialize this object as a hash 
+    def to_json(*a)
+      output = @data.clone
+      output["json_class"] = self.class.name
+      output.to_json(*a)
+    end
+    
+    # Pretty Print this object as JSON 
+    def json_pretty_print
+      JSON.pretty_generate(@data)
+    end
+    
+    def attributes_print(a)
+      JSON.pretty_generate(@data[a])
+    end
+    # Create an Ohai::System from JSON
+    def self.json_create(o)
+      ohai = new
+      o.each do |key, value|
+        ohai.data[key] = value unless key == "json_class"
+      end
+      ohai
+    end
+    
+    def method_missing(name, *args)
+      return get_attribute(name) if args.length == 0 
+      
+      set_attribute(name, *args)
+    end
+    
+    private
+      def load_plugin_file
+        
+      end
+  end
+end