ohai 0.3.4

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,241 @@
+#
+# 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'
+
+begin
+  require 'yajl/json_gem'
+rescue LoadError
+  begin
+    require 'json'
+  rescue LoadError
+    begin
+      require 'json/pure'
+    rescue LoadError
+      STDERR.puts "No valid JSON library detected, please install one of 'yajl-ruby', 'json' or 'json_pure'."
+      exit -2
+    end
+  end
+end
+
+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 = ""
+      Ohai::Log.level(Ohai::Config.log_level) 
+    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

data/spec/ohai/mixin/command_spec.rb

@@ -0,0 +1,37 @@
+#
+# Author:: Diego Algorta (diego@oboxodo.com)
+# Copyright:: Copyright (c) 2009 Diego Algorta
+# 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 File.join(File.dirname(__FILE__), '..', '..', '/spec_helper.rb')
+
+describe Ohai::Mixin::Command, "popen4" do
+
+  it "should default all commands to be run in the POSIX standard C locale" do
+    Ohai::Mixin::Command.popen4("echo $LC_ALL") do |pid, stdin, stdout, stderr|
+      stdin.close
+      stdout.read.strip.should == "C"
+    end
+  end
+
+  it "should respect locale when specified explicitly" do
+    Ohai::Mixin::Command.popen4("echo $LC_ALL", :environment => {"LC_ALL" => "es"}) do |pid, stdin, stdout, stderr|
+      stdin.close
+      stdout.read.strip.should == "es"
+    end
+  end
+
+end

data/spec/ohai/mixin/from_file_spec.rb

@@ -0,0 +1,53 @@
+#
+# 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 File.join(File.dirname(__FILE__), '..', '..', '/spec_helper.rb')
+
+describe Ohai::System, "from_file" do
+  before(:each) do
+    @ohai = Ohai::System.new
+    File.stub!(:exists?).and_return(true)
+    File.stub!(:readable?).and_return(true)
+    IO.stub!(:read).and_return("king 'herod'")
+  end
+  
+  it "should check to see that the file exists" do
+    File.should_receive(:exists?).and_return(true)
+    @ohai.from_file("/tmp/foo")
+  end
+  
+  it "should check to see that the file is readable" do
+    File.should_receive(:readable?).and_return(true)
+    @ohai.from_file("/tmp/foo")
+  end
+  
+  it "should actually read the file" do
+    IO.should_receive(:read).and_return("king 'herod'")
+    @ohai.from_file("/tmp/foo")
+  end
+  
+  it "should call instance_eval with the contents of the file, file name, and line 1" do
+    @ohai.should_receive(:instance_eval).with("king 'herod'", "/tmp/foo", 1)
+    @ohai.from_file("/tmp/foo")
+  end
+  
+  it "should raise an IOError if it cannot read the file" do
+    File.stub!(:exists?).and_return(false)
+    lambda { @ohai.from_file("/tmp/foo") }.should raise_error(IOError)
+  end
+end