madscience 0.0.1

data/cookbooks/windows/libraries/powershell_out.rb

@@ -0,0 +1,79 @@
+class Chef
+  module Mixin
+    module PowershellOut
+      include Chef::Mixin::ShellOut
+
+      begin
+        include Chef::Mixin::WindowsArchitectureHelper
+      rescue
+        # nothing to do, as the include will happen when windows_architecture_helper.rb
+        # is loaded.  This is for ease of removal of that library when either
+        # powershell_out is core chef or powershell cookbook depends upon version
+        # of chef that has Chef::Mixin::WindowsArchitectureHelper in core chef
+      end
+
+      def powershell_out(*command_args)
+        script = command_args.first
+        options = command_args.last.is_a?(Hash) ? command_args.last : nil
+
+        run_command(script, options)
+      end
+
+      def powershell_out!(*command_args)
+        cmd = powershell_out(*command_args)
+        cmd.error!
+        cmd
+      end
+
+      private
+      def run_command(script, options)
+        if options && options[:architecture]
+          architecture = options[:architecture]
+          options.delete(:architecture)
+        else
+          architecture = node_windows_architecture(node)
+        end
+
+        disable_redirection = wow64_architecture_override_required?(node, architecture)
+
+        if disable_redirection
+          original_redirection_state = disable_wow64_file_redirection(node)
+        end
+
+        command = build_command(script)
+
+        if options
+          cmd = shell_out(command, options)
+        else
+          cmd = shell_out(command)
+        end
+
+        if disable_redirection
+          restore_wow64_file_redirection(node, original_redirection_state)
+        end
+
+        cmd
+      end
+
+      def build_command(script)
+        flags = [
+          # Hides the copyright banner at startup.
+          "-NoLogo",
+          # Does not present an interactive prompt to the user.
+          "-NonInteractive",
+          # Does not load the Windows PowerShell profile.
+          "-NoProfile",
+          # always set the ExecutionPolicy flag
+          # see http://technet.microsoft.com/en-us/library/ee176961.aspx
+          "-ExecutionPolicy RemoteSigned",
+          # Powershell will hang if STDIN is redirected
+          # http://connect.microsoft.com/PowerShell/feedback/details/572313/powershell-exe-can-hang-if-stdin-is-redirected
+          "-InputFormat None"
+        ]
+
+        command = "powershell.exe #{flags.join(' ')} -Command \"#{script}\""
+        command
+      end
+    end
+  end
+end

data/cookbooks/windows/libraries/registry_helper.rb

@@ -0,0 +1,357 @@
+#
+# Author:: Doug MacEachern (<dougm@vmware.com>)
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Author:: Paul Morton (<pmorton@biaprotect.com>)
+# Cookbook Name:: windows
+# Provider:: registry
+#
+# Copyright:: 2010, VMware, Inc.
+# Copyright:: 2011, Opscode, Inc.
+# Copyright:: 2011, Business Intelligence Associates, Inc
+#
+# 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.
+#
+
+if RUBY_PLATFORM =~ /mswin|mingw32|windows/
+  require 'win32/registry'
+  require 'ruby-wmi'
+end
+
+module Windows
+  module RegistryHelper
+
+    @@native_registry_constant = ENV['PROCESSOR_ARCHITEW6432'] == 'AMD64' ? 0x0100 : 0x0200
+
+    def get_hive_name(path)
+      Chef::Log.debug("Resolving registry shortcuts to full names")
+
+      reg_path = path.split("\\")
+      hive_name = reg_path.shift
+
+      hkey = {
+        "HKLM" => "HKEY_LOCAL_MACHINE",
+        "HKCU" => "HKEY_CURRENT_USER",
+        "HKU"  => "HKEY_USERS"
+      }[hive_name] || hive_name
+
+      Chef::Log.debug("Hive resolved to #{hkey}")
+      return hkey
+    end
+
+    def get_hive(path)
+
+      Chef::Log.debug("Getting hive for #{path}")
+      reg_path = path.split("\\")
+      hive_name = reg_path.shift
+
+      hkey = get_hive_name(path)
+
+      hive = {
+        "HKEY_LOCAL_MACHINE" => ::Win32::Registry::HKEY_LOCAL_MACHINE,
+        "HKEY_USERS" => ::Win32::Registry::HKEY_USERS,
+        "HKEY_CURRENT_USER" => ::Win32::Registry::HKEY_CURRENT_USER
+        }[hkey]
+
+      unless hive
+        Chef::Application.fatal!("Unsupported registry hive '#{hive_name}'")
+      end
+
+
+      Chef::Log.debug("Registry hive resolved to #{hkey}")
+      return hive
+    end
+
+    def unload_hive(path)
+      hive = get_hive(path)
+      if hive == ::Win32::Registry::HKEY_USERS
+        reg_path = path.split("\\")
+        priv = Chef::WindowsPrivileged.new
+        begin
+          priv.reg_unload_key(reg_path[1])
+        rescue
+        end
+      end
+    end
+
+    def set_value(mode,path,values,type=nil)
+      hive, reg_path, hive_name, root_key, hive_loaded = get_reg_path_info(path)
+      key_name = reg_path.join("\\")
+
+      Chef::Log.debug("Creating #{path}")
+
+      if !key_exists?(path,true)
+        create_key(path)
+      end
+
+      hive.send(mode, key_name, ::Win32::Registry::KEY_ALL_ACCESS | @@native_registry_constant) do |reg|
+        changed_something = false
+        values.each do |k,val|
+          key = k.to_s #wtf. avoid "can't modify frozen string" in win32/registry.rb
+          cur_val = nil
+          begin
+            cur_val = reg[key]
+          rescue
+            #subkey does not exist (ok)
+          end
+          if cur_val != val
+            Chef::Log.debug("setting #{key}=#{val}")
+            
+            if type.nil?
+              type = :string
+            end
+
+            reg_type = {
+              :binary => ::Win32::Registry::REG_BINARY,
+              :string => ::Win32::Registry::REG_SZ,
+              :multi_string => ::Win32::Registry::REG_MULTI_SZ,
+              :expand_string => ::Win32::Registry::REG_EXPAND_SZ,
+              :dword => ::Win32::Registry::REG_DWORD,
+              :dword_big_endian => ::Win32::Registry::REG_DWORD_BIG_ENDIAN,
+              :qword => ::Win32::Registry::REG_QWORD
+            }[type]
+
+            reg.write(key, reg_type, val)
+
+            ensure_hive_unloaded(hive_loaded)
+
+            changed_something = true
+          end
+        end
+        return changed_something
+      end
+      return false
+    end
+
+    def get_value(path,value)
+      hive, reg_path, hive_name, root_key, hive_loaded = get_reg_path_info(path)
+      key = reg_path.join("\\")
+
+      hive.open(key, ::Win32::Registry::KEY_ALL_ACCESS | @@native_registry_constant) do | reg |
+        begin
+          return reg[value]
+        rescue
+          return nil
+        ensure
+          ensure_hive_unloaded(hive_loaded)
+        end
+      end
+    end
+
+    def get_values(path)
+      hive, reg_path, hive_name, root_key, hive_loaded = get_reg_path_info(path)
+      key = reg_path.join("\\")
+      hive.open(key, ::Win32::Registry::KEY_ALL_ACCESS | @@native_registry_constant) do | reg |
+        values = []
+        begin
+        reg.each_value do |name, type, data|
+          values << [name, type, data]
+        end
+        rescue
+        ensure
+          ensure_hive_unloaded(hive_loaded)
+        end
+        values
+      end
+    end
+
+    def delete_value(path,values)
+      hive, reg_path, hive_name, root_key, hive_loaded = get_reg_path_info(path)
+      key = reg_path.join("\\")
+      Chef::Log.debug("Deleting values in #{path}")
+      hive.open(key, ::Win32::Registry::KEY_ALL_ACCESS | @@native_registry_constant) do | reg |
+        values.each_key { |key|
+          name = key.to_s
+          # Ensure delete operation is idempotent.
+          if value_exists?(path, key)
+            Chef::Log.debug("Deleting value #{name} in #{path}")
+            reg.delete_value(name)
+          else
+            Chef::Log.debug("Value #{name} in #{path} does not exist, skipping.")
+          end
+        }
+      end
+
+    end
+
+    def create_key(path)
+      hive, reg_path, hive_name, root_key, hive_loaded = get_reg_path_info(path)
+      key = reg_path.join("\\")
+      Chef::Log.debug("Creating registry key #{path}")
+      hive.create(key)
+    end
+
+    def value_exists?(path,value)
+      if key_exists?(path,true)
+
+        hive, reg_path, hive_name, root_key , hive_loaded = get_reg_path_info(path)
+        key = reg_path.join("\\")
+
+        Chef::Log.debug("Attempting to open #{key}");
+        Chef::Log.debug("Native Constant #{@@native_registry_constant}")
+        Chef::Log.debug("Hive #{hive}")
+
+        hive.open(key, ::Win32::Registry::KEY_READ | @@native_registry_constant) do | reg |
+          begin
+            rtn_value = reg[value]
+            return true
+          rescue
+            return false
+          ensure
+            ensure_hive_unloaded(hive_loaded)
+          end
+        end
+
+      end
+      return false
+    end
+
+    # TODO: Does not load user registry...
+    def key_exists?(path, load_hive = false)
+      if load_hive
+        hive, reg_path, hive_name, root_key , hive_loaded = get_reg_path_info(path)
+        key = reg_path.join("\\")
+      else
+        hive = get_hive(path)
+        reg_path = path.split("\\")
+        hive_name = reg_path.shift
+        root_key = reg_path[0]
+        key = reg_path.join("\\")
+        hive_loaded = false
+      end
+
+      begin
+        hive.open(key, ::Win32::Registry::Constants::KEY_READ | @@native_registry_constant )
+        return true
+      rescue
+        return false
+      ensure
+        ensure_hive_unloaded(hive_loaded)
+      end
+    end
+
+    def get_user_hive_location(sid)
+      reg_key = "HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList\\#{sid}"
+      Chef::Log.debug("Looking for profile at #{reg_key}")
+      if key_exists?(reg_key)
+        return get_value(reg_key,'ProfileImagePath')
+      else
+        return nil
+      end
+
+    end
+
+    def resolve_user_to_sid(username)
+      begin
+        sid = WMI::Win32_UserAccount.find(:first, :conditions => {:name => username}).sid
+        Chef::Log.debug("Resolved user SID to #{sid}")
+        return sid
+      rescue
+        return nil
+      end
+    end
+
+    def hive_loaded?(path)
+      hive = get_hive(path)
+      reg_path = path.split("\\")
+      hive_name = reg_path.shift
+      user_hive = path[0]
+
+      if is_user_hive?(hive)
+        return key_exists?("#{hive_name}\\#{user_hive}")
+      else
+        return true
+      end
+    end
+
+    def is_user_hive?(hive)
+      if hive == ::Win32::Registry::HKEY_USERS
+        return true
+      else
+        return true
+      end
+    end
+
+    def get_reg_path_info(path)
+      hive = get_hive(path)
+      reg_path = path.split("\\")
+      hive_name = reg_path.shift
+      root_key = reg_path[0]
+      hive_loaded = false
+
+      if is_user_hive?(hive) && !key_exists?("#{hive_name}\\#{root_key}")
+        reg_path, hive_loaded = load_user_hive(hive,reg_path,root_key)
+        root_key = reg_path[0]
+        Chef::Log.debug("Resolved user (#{path}) to (#{reg_path.join('/')})")
+      end
+
+      return hive, reg_path, hive_name, root_key, hive_loaded
+    end
+
+    def load_user_hive(hive,reg_path,user_hive)
+      Chef::Log.debug("Reg Path #{reg_path}")
+      # See if the hive is loaded. Logged in users will have a key that is named their SID
+      # if the user has specified the a path by SID and the user is logged in, this function
+      # should not be executed.
+      if is_user_hive?(hive) && !key_exists?("HKU\\#{user_hive}")
+        Chef::Log.debug("The user is not logged in and has not been specified by SID")
+        sid = resolve_user_to_sid(user_hive)
+        Chef::Log.debug("User SID resolved to (#{sid})")
+        # Now that the user has been resolved to a SID, check and see if the hive exists.
+        # If this exists by SID, the user is logged in and we should use that key.
+        # TODO: Replace the username with the sid and send it back because the username
+        # does not exist as the key location.
+        load_reg = false
+        if key_exists?("HKU\\#{sid}")
+          reg_path[0] = sid #use the active profile (user is logged on)
+          Chef::Log.debug("HKEY_USERS Mapped: #{user_hive} -> #{sid}")
+        else
+          Chef::Log.debug("User is not logged in")
+          load_reg = true
+        end
+
+        # The user is not logged in, so we should load the registry from disk
+        if load_reg
+          profile_path = get_user_hive_location(sid)
+          if profile_path != nil
+            ntuser_dat = "#{profile_path}\\NTUSER.DAT"
+            if ::File.exists?(ntuser_dat)
+              priv = Chef::WindowsPrivileged.new
+              if priv.reg_load_key(sid,ntuser_dat)
+                Chef::Log.debug("RegLoadKey(#{sid}, #{user_hive}, #{ntuser_dat})")
+                reg_path[0] = sid
+              else
+                Chef::Log.debug("Failed RegLoadKey(#{sid}, #{user_hive}, #{ntuser_dat})")
+              end
+            end
+          end
+        end
+      end
+
+      return reg_path, load_reg
+
+    end
+
+    private
+    def ensure_hive_unloaded(hive_loaded=false)
+      if(hive_loaded)
+        Chef::Log.debug("Hive was loaded, we really should unload it")
+        unload_hive(path)
+      end
+    end
+  end
+end
+
+module Registry
+  module_function
+  extend Windows::RegistryHelper
+end

data/cookbooks/windows/libraries/version.rb

@@ -0,0 +1,206 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Cookbook Name:: windows
+# Library:: version
+#
+# Copyright:: 2011, Opscode, Inc.
+#
+# 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.
+#
+
+if RUBY_PLATFORM =~ /mswin|mingw32|windows/
+  require 'ruby-wmi'
+  require 'Win32API'
+end
+
+module Windows
+  class Version
+
+    # http://msdn.microsoft.com/en-us/library/ms724833(v=vs.85).aspx
+
+    # Suite Masks
+    # Microsoft BackOffice components are installed.
+    VER_SUITE_BACKOFFICE = 0x00000004.freeze unless defined?(VER_SUITE_BACKOFFICE)
+    # Windows Server 2003, Web Edition is installed.
+    VER_SUITE_BLADE = 0x00000400.freeze unless defined?(VER_SUITE_BLADE)
+    # Windows Server 2003, Compute Cluster Edition is installed.
+    VER_SUITE_COMPUTE_SERVER = 0x00004000.freeze unless defined?(VER_SUITE_COMPUTE_SERVER)
+    # Windows Server 2008 Datacenter, Windows Server 2003, Datacenter Edition, or Windows 2000 Datacenter Server is installed.
+    VER_SUITE_DATACENTER = 0x00000080.freeze unless defined?(VER_SUITE_DATACENTER)
+    # Windows Server 2008 Enterprise, Windows Server 2003, Enterprise Edition, or Windows 2000 Advanced Server is installed. Refer to the Remarks section for more information about this bit flag.
+    VER_SUITE_ENTERPRISE = 0x00000002.freeze unless defined?(VER_SUITE_ENTERPRISE)
+    # Windows XP Embedded is installed.
+    VER_SUITE_EMBEDDEDNT = 0x00000040.freeze unless defined?(VER_SUITE_EMBEDDEDNT)
+    # Windows Vista Home Premium, Windows Vista Home Basic, or Windows XP Home Edition is installed.
+    VER_SUITE_PERSONAL = 0x00000200.freeze unless defined?(VER_SUITE_PERSONAL)
+    # Remote Desktop is supported, but only one interactive session is supported. This value is set unless the system is running in application server mode.
+    VER_SUITE_SINGLEUSERTS = 0x00000100.freeze unless defined?(VER_SUITE_SINGLEUSERTS)
+    # Microsoft Small Business Server was once installed on the system, but may have been upgraded to another version of Windows. Refer to the Remarks section for more information about this bit flag.
+    VER_SUITE_SMALLBUSINESS = 0x00000001.freeze unless defined?(VER_SUITE_SMALLBUSINESS)
+    # Microsoft Small Business Server is installed with the restrictive client license in force. Refer to the Remarks section for more information about this bit flag.
+    VER_SUITE_SMALLBUSINESS_RESTRICTED = 0x00000020.freeze unless defined?(VER_SUITE_SMALLBUSINESS_RESTRICTED)
+    # Windows Storage Server 2003 R2 or Windows Storage Server 2003is installed.
+    VER_SUITE_STORAGE_SERVER = 0x00002000.freeze unless defined?(VER_SUITE_STORAGE_SERVER)
+    # Terminal Services is installed. This value is always set.
+    # If VER_SUITE_TERMINAL is set but VER_SUITE_SINGLEUSERTS is not set, the system is running in application server mode.
+    VER_SUITE_TERMINAL = 0x00000010.freeze unless defined?(VER_SUITE_TERMINAL)
+    # Windows Home Server is installed.
+    VER_SUITE_WH_SERVER = 0x00008000.freeze unless defined?(VER_SUITE_WH_SERVER)
+
+    # Product Type
+    # The system is a domain controller and the operating system is Windows Server 2012, Windows Server 2008 R2, Windows Server 2008, Windows Server 2003, or Windows 2000 Server.
+    VER_NT_DOMAIN_CONTROLLER = 0x0000002.freeze unless defined?(VER_NT_DOMAIN_CONTROLLER)
+    # The operating system is Windows Server 2012, Windows Server 2008 R2, Windows Server 2008, Windows Server 2003, or Windows 2000 Server.
+    # Note that a server that is also a domain controller is reported as VER_NT_DOMAIN_CONTROLLER, not VER_NT_SERVER.
+    VER_NT_SERVER = 0x0000003.freeze unless defined?(VER_NT_SERVER)
+    # The operating system is Windows 7, Windows Vista, Windows XP Professional, Windows XP Home Edition, or Windows 2000 Professional.
+    VER_NT_WORKSTATION = 0x0000001.freeze unless defined?(VER_NT_WORKSTATION)
+
+    # GetSystemMetrics
+    # The build number if the system is Windows Server 2003 R2; otherwise, 0.
+    SM_SERVERR2 = 89.freeze unless defined?(SM_SERVERR2)
+
+    # http://msdn.microsoft.com/en-us/library/ms724358(v=vs.85).aspx
+    SKU = {
+      0x00000006 => {:ms_const => 'PRODUCT_BUSINESS', :name => 'Business'},
+      0x00000010 => {:ms_const => 'PRODUCT_BUSINESS_N', :name => 'Business N'},
+      0x00000012 => {:ms_const => 'PRODUCT_CLUSTER_SERVER', :name => 'HPC Edition'},
+      0x00000008 => {:ms_const => 'PRODUCT_DATACENTER_SERVER', :name => 'Server Datacenter (full installation)'},
+      0x0000000C => {:ms_const => 'PRODUCT_DATACENTER_SERVER_CORE', :name => 'Server Datacenter (core installation)'},
+      0x00000027 => {:ms_const => 'PRODUCT_DATACENTER_SERVER_CORE_V', :name => 'Server Datacenter without Hyper-V (core installation)'},
+      0x00000025 => {:ms_const => 'PRODUCT_DATACENTER_SERVER_V', :name => 'Server Datacenter without Hyper-V (full installation)'},
+      0x00000004 => {:ms_const => 'PRODUCT_ENTERPRISE', :name => 'Enterprise'},
+      0x00000046 => {:ms_const => 'PRODUCT_ENTERPRISE_E', :name => 'Not supported'},
+      0x0000001B => {:ms_const => 'PRODUCT_ENTERPRISE_N', :name => 'Enterprise N'},
+      0x0000000A => {:ms_const => 'PRODUCT_ENTERPRISE_SERVER', :name => 'Server Enterprise (full installation)'},
+      0x0000000E => {:ms_const => 'PRODUCT_ENTERPRISE_SERVER_CORE', :name => 'Server Enterprise (core installation)'},
+      0x00000029 => {:ms_const => 'PRODUCT_ENTERPRISE_SERVER_CORE_V', :name => 'Server Enterprise without Hyper-V (core installation)'},
+      0x0000000F => {:ms_const => 'PRODUCT_ENTERPRISE_SERVER_IA64', :name => 'Server Enterprise for Itanium-based Systems'},
+      0x00000026 => {:ms_const => 'PRODUCT_ENTERPRISE_SERVER_V', :name => 'Server Enterprise without Hyper-V (full installation)'},
+      0x00000002 => {:ms_const => 'PRODUCT_HOME_BASIC', :name => 'Home Basic'},
+      0x00000043 => {:ms_const => 'PRODUCT_HOME_BASIC_E', :name => 'Not supported'},
+      0x00000005 => {:ms_const => 'PRODUCT_HOME_BASIC_N', :name => 'Home Basic N'},
+      0x00000003 => {:ms_const => 'PRODUCT_HOME_PREMIUM', :name => 'Home Premium'},
+      0x00000044 => {:ms_const => 'PRODUCT_HOME_PREMIUM_E', :name => 'Not supported'},
+      0x0000001A => {:ms_const => 'PRODUCT_HOME_PREMIUM_N', :name => 'Home Premium N'},
+      0x0000002A => {:ms_const => 'PRODUCT_HYPERV', :name => 'Microsoft Hyper-V Server'},
+      0x0000001E => {:ms_const => 'PRODUCT_MEDIUMBUSINESS_SERVER_MANAGEMENT', :name => 'Windows Essential Business Server Management Server'},
+      0x00000020 => {:ms_const => 'PRODUCT_MEDIUMBUSINESS_SERVER_MESSAGING', :name => 'Windows Essential Business Server Messaging Server'},
+      0x0000001F => {:ms_const => 'PRODUCT_MEDIUMBUSINESS_SERVER_SECURITY', :name => 'Windows Essential Business Server Security Server'},
+      0x00000030 => {:ms_const => 'PRODUCT_PROFESSIONAL', :name => 'Professional'},
+      0x00000045 => {:ms_const => 'PRODUCT_PROFESSIONAL_E', :name => 'Not supported'},
+      0x00000031 => {:ms_const => 'PRODUCT_PROFESSIONAL_N', :name => 'Professional N'},
+      0x00000018 => {:ms_const => 'PRODUCT_SERVER_FOR_SMALLBUSINESS', :name => 'Windows Server 2008 for Windows Essential Server Solutions'},
+      0x00000023 => {:ms_const => 'PRODUCT_SERVER_FOR_SMALLBUSINESS_V', :name => 'Windows Server 2008 without Hyper-V for Windows Essential Server Solutions'},
+      0x00000021 => {:ms_const => 'PRODUCT_SERVER_FOUNDATION', :name => 'Server Foundation'},
+      0x00000022 => {:ms_const => 'PRODUCT_HOME_PREMIUM_SERVER', :name => 'Windows Home Server 2011'},
+      0x00000032 => {:ms_const => 'PRODUCT_SB_SOLUTION_SERVER', :name => 'Windows Small Business Server 2011 Essentials'},
+      0x00000013 => {:ms_const => 'PRODUCT_HOME_SERVER', :name => 'Windows Storage Server 2008 R2 Essentials'},
+      0x00000009 => {:ms_const => 'PRODUCT_SMALLBUSINESS_SERVER', :name => 'Windows Small Business Server'},
+      0x00000038 => {:ms_const => 'PRODUCT_SOLUTION_EMBEDDEDSERVER', :name => 'Windows MultiPoint Server'},
+      0x00000007 => {:ms_const => 'PRODUCT_STANDARD_SERVER', :name => 'Server Standard (full installation)'},
+      0x0000000D => {:ms_const => 'PRODUCT_STANDARD_SERVER_CORE', :name => 'Server Standard (core installation)'},
+      0x00000028 => {:ms_const => 'PRODUCT_STANDARD_SERVER_CORE_V', :name => 'Server Standard without Hyper-V (core installation)'},
+      0x00000024 => {:ms_const => 'PRODUCT_STANDARD_SERVER_V', :name => 'Server Standard without Hyper-V (full installation)'},
+      0x0000000B => {:ms_const => 'PRODUCT_STARTER', :name => 'Starter'},
+      0x00000042 => {:ms_const => 'PRODUCT_STARTER_E', :name => 'Not supported'},
+      0x0000002F => {:ms_const => 'PRODUCT_STARTER_N', :name => 'Starter N'},
+      0x00000017 => {:ms_const => 'PRODUCT_STORAGE_ENTERPRISE_SERVER', :name => 'Storage Server Enterprise'},
+      0x00000014 => {:ms_const => 'PRODUCT_STORAGE_EXPRESS_SERVER', :name => 'Storage Server Express'},
+      0x00000015 => {:ms_const => 'PRODUCT_STORAGE_STANDARD_SERVER', :name => 'Storage Server Standard'},
+      0x00000016 => {:ms_const => 'PRODUCT_STORAGE_WORKGROUP_SERVER', :name => 'Storage Server Workgroup'},
+      0x00000000 => {:ms_const => 'PRODUCT_UNDEFINED', :name => 'An unknown product'},
+      0x00000001 => {:ms_const => 'PRODUCT_ULTIMATE', :name => 'Ultimate'},
+      0x00000047 => {:ms_const => 'PRODUCT_ULTIMATE_E', :name => 'Not supported'},
+      0x0000001C => {:ms_const => 'PRODUCT_ULTIMATE_N', :name => 'Ultimate N'},
+      0x00000011 => {:ms_const => 'PRODUCT_WEB_SERVER', :name => 'Web Server (full installation)'},
+      0x0000001D => {:ms_const => 'PRODUCT_WEB_SERVER_CORE', :name => 'Web Server (core installation)'}
+    }.freeze unless defined?(SKU)
+
+    attr_reader :major_version, :minor_version, :build_number, :service_pack_major_version, :service_pack_minor_version
+    attr_reader :version, :product_type, :product_suite, :sku
+
+    def initialize
+      unless RUBY_PLATFORM =~ /mswin|mingw32|windows/
+        raise NotImplementedError, 'only valid on Windows platform'
+      end
+      @version, @product_type, @product_suite, @sku, @service_pack_major_version, @service_pack_minor_version = get_os_info
+      @major_version, @minor_version, @build_number = version.split('.').map{|v| v.to_i }
+    end
+
+    WIN_VERSIONS = {
+      "Windows Server 2012 R2" => {:major => 6, :minor => 3, :callable => lambda{ @product_type != VER_NT_WORKSTATION }},
+      "Windows 8" => {:major => 6, :minor => 2, :callable => lambda{ @product_type == VER_NT_WORKSTATION }},
+      "Windows Server 2012" => {:major => 6, :minor => 2, :callable => lambda{ @product_type != VER_NT_WORKSTATION }},
+      "Windows 7" => {:major => 6, :minor => 1, :callable => lambda{ @product_type == VER_NT_WORKSTATION }},
+      "Windows Server 2008 R2" => {:major => 6, :minor => 1, :callable => lambda{ @product_type != VER_NT_WORKSTATION }},
+      "Windows Server 2008" => {:major => 6, :minor => 0, :callable => lambda{ @product_type != VER_NT_WORKSTATION }},
+      "Windows Vista" => {:major => 6, :minor => 0, :callable => lambda{ @product_type == VER_NT_WORKSTATION }},
+      "Windows Server 2003 R2" => {:major => 5, :minor => 2, :callable => lambda{ Win32API.new('user32', 'GetSystemMetrics', 'I', 'I').call(SM_SERVERR2) != 0 }},
+      "Windows Home Server" => {:major => 5, :minor => 2, :callable => lambda{  (@product_suite & VER_SUITE_WH_SERVER) == VER_SUITE_WH_SERVER }},
+      "Windows Server 2003" => {:major => 5, :minor => 2, :callable => lambda{ Win32API.new('user32', 'GetSystemMetrics', 'I', 'I').call(SM_SERVERR2) == 0 }},
+      "Windows XP" => {:major => 5, :minor => 1},
+      "Windows 2000" => {:major => 5, :minor => 0}
+    }.freeze unless defined?(WIN_VERSIONS)
+
+    marketing_names = Array.new
+
+    # General Windows checks
+    WIN_VERSIONS.each do |k,v|
+      method_name = "#{k.gsub(/\s/, '_').downcase}?"
+      define_method(method_name) do
+        (@major_version == v[:major]) &&
+        (@minor_version == v[:minor]) &&
+        (v[:callable] ? v[:callable].call : true)
+      end
+      marketing_names << [k, method_name]
+    end
+
+    define_method(:marketing_name) do
+      marketing_names.each do |mn|
+        break mn[0] if self.send(mn[1])
+      end
+    end
+
+    # Server Type checks
+    %w{ core full datacenter }.each do |m|
+      define_method("server_#{m}?") do
+        if @sku
+          !(SKU[@sku][:name] =~ /#{m}/i).nil?
+        else
+          false
+        end
+      end
+    end
+
+    private
+    # Win32API call to GetSystemMetrics(SM_SERVERR2)
+    # returns: The build number if the system is Windows Server 2003 R2; otherwise, 0.
+    def sm_serverr2
+      @sm_serverr2 ||= Win32API.new('user32', 'GetSystemMetrics', 'I', 'I').call(SM_SERVERR2)
+    end
+
+    # query WMI Win32_OperatingSystem for required OS info
+    def get_os_info
+      cols = %w{ Version ProductType OSProductSuite OperatingSystemSKU ServicePackMajorVersion ServicePackMinorVersion }
+      os_info = WMI::Win32_OperatingSystem.find(:first)
+      cols.map do |c|
+        begin
+          os_info.send(c)
+        rescue # OperatingSystemSKU doesn't exist in all versions of Windows
+          nil
+        end
+      end
+    end
+  end
+end