mbailey-chef 0.9.12.1

data/lib/chef/util/windows/net_use.rb

@@ -0,0 +1,121 @@
+#
+# Author:: Doug MacEachern (<dougm@vmware.com>)
+# Copyright:: Copyright (c) 2010 VMware, 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.
+#
+
+#the Win32 Volume APIs do not support mapping network drives. not supported by WMI either.
+#see also: WNetAddConnection2 and WNetAddConnection3
+#see also cmd.exe: net use /?
+
+require 'chef/util/windows'
+
+class Chef::Util::Windows::NetUse < Chef::Util::Windows
+
+  private
+
+  USE_NOFORCE = 0
+  USE_FORCE = 1
+  USE_LOTS_OF_FORCE = 2 #every windows API should support this flag
+
+  USE_INFO_2 = [
+    [:local, nil],
+    [:remote, nil],
+    [:password, nil],
+    [:status, 0],
+    [:asg_type, 0],
+    [:refcount, 0],
+    [:usecount, 0],
+    [:username, nil],
+    [:domainname, nil]
+  ]
+
+  USE_INFO_2_TEMPLATE =
+    USE_INFO_2.collect { |field| field[1].class == Fixnum ? 'i' : 'L' }.join
+
+  SIZEOF_USE_INFO_2 = #sizeof(USE_INFO_2)
+    USE_INFO_2.inject(0){|sum,item|
+    sum + (item[1].class == Fixnum ? 4 : PTR_SIZE)
+  }
+
+  def use_info_2(args)
+    USE_INFO_2.collect { |field|
+      args.include?(field[0]) ? args[field[0]] : field[1]
+    }
+  end
+
+  def use_info_2_pack(use)
+    use.collect { |v|
+      v.class == Fixnum ? v : str_to_ptr(multi_to_wide(v))
+    }.pack(USE_INFO_2_TEMPLATE)
+  end
+
+  def use_info_2_unpack(buffer)
+    use = Hash.new
+    USE_INFO_2.each_with_index do |field,offset|
+      use[field[0]] = field[1].class == Fixnum ?
+      dword_to_i(buffer, offset) : lpwstr_to_s(buffer, offset)
+    end
+    use
+  end
+
+  public
+
+  def initialize(localname)
+    @localname = localname
+    @name = multi_to_wide(localname)
+  end
+
+  def add(args)
+    if args.class == String
+      remote = args
+      args = Hash.new
+      args[:remote] = remote
+    end
+    args[:local] ||= @localname
+    use = use_info_2(args)
+    buffer = use_info_2_pack(use)
+    rc = NetUseAdd.call(nil, 2, buffer, nil)
+    if rc != NERR_Success
+      raise ArgumentError, get_last_error(rc)
+    end
+  end
+
+  def get_info
+    ptr  = 0.chr * PTR_SIZE
+    rc = NetUseGetInfo.call(nil, @name, 2, ptr)
+
+    if rc != NERR_Success
+      raise ArgumentError, get_last_error(rc)
+    end
+
+    ptr = ptr.unpack('L')[0]
+    buffer = 0.chr * SIZEOF_USE_INFO_2
+    memcpy(buffer, ptr, buffer.size)
+    NetApiBufferFree(ptr)
+    use_info_2_unpack(buffer)
+  end
+
+  def device
+    get_info()[:remote]
+  end
+  #XXX should we use some FORCE here?
+  def delete
+    rc = NetUseDel.call(nil, @name, USE_NOFORCE)
+    if rc != NERR_Success
+      raise ArgumentError, get_last_error(rc)
+    end
+  end
+end

data/lib/chef/util/windows/net_user.rb

@@ -0,0 +1,198 @@
+#
+# Author:: Doug MacEachern (<dougm@vmware.com>)
+# Copyright:: Copyright (c) 2010 VMware, 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 'chef/util/windows'
+
+#wrapper around a subset of the NetUser* APIs.
+#nothing Chef specific, but not complete enough to be its own gem, so util for now.
+class Chef::Util::Windows::NetUser < Chef::Util::Windows
+
+  private
+
+  LogonUser = Windows::API.new('LogonUser', 'SSSLLP', 'I', 'advapi32')
+
+  DOMAIN_GROUP_RID_USERS = 0x00000201
+
+  UF_SCRIPT              = 0x000001
+  UF_ACCOUNTDISABLE      = 0x000002
+  UF_PASSWD_CANT_CHANGE  = 0x000040
+  UF_NORMAL_ACCOUNT      = 0x000200
+  UF_DONT_EXPIRE_PASSWD  = 0x010000
+
+  #[:symbol_name, default_val]
+  #default_val duals as field type
+  #array index duals as structure offset
+  USER_INFO_3 = [
+    [:name, nil],
+    [:password, nil],
+    [:password_age, 0],
+    [:priv, 0], #"The NetUserAdd and NetUserSetInfo functions ignore this member"
+    [:home_dir, nil],
+    [:comment, nil],
+    [:flags, UF_SCRIPT|UF_DONT_EXPIRE_PASSWD|UF_NORMAL_ACCOUNT],
+    [:script_path, nil],
+    [:auth_flags, 0],
+    [:full_name, nil],
+    [:user_comment, nil],
+    [:parms, nil],
+    [:workstations, nil],
+    [:last_logon, 0],
+    [:last_logoff, 0],
+    [:acct_expires, -1],
+    [:max_storage, -1],
+    [:units_per_week, 0],
+    [:logon_hours, nil],
+    [:bad_pw_count, 0],
+    [:num_logons, 0],
+    [:logon_server, nil],
+    [:country_code, 0],
+    [:code_page, 0],
+    [:user_id, 0],
+    [:primary_group_id, DOMAIN_GROUP_RID_USERS],
+    [:profile, nil],
+    [:home_dir_drive, nil],
+    [:password_expired, 0]
+  ]
+
+  USER_INFO_3_TEMPLATE =
+    USER_INFO_3.collect { |field| field[1].class == Fixnum ? 'i' : 'L' }.join
+
+  SIZEOF_USER_INFO_3 = #sizeof(USER_INFO_3)
+    USER_INFO_3.inject(0){|sum,item|
+      sum + (item[1].class == Fixnum ? 4 : PTR_SIZE)
+    }
+
+  def user_info_3(args)
+    USER_INFO_3.collect { |field|
+      args.include?(field[0]) ? args[field[0]] : field[1]
+    }
+  end
+
+  def user_info_3_pack(user)
+    user.collect { |v|
+      v.class == Fixnum ? v : str_to_ptr(multi_to_wide(v))
+    }.pack(USER_INFO_3_TEMPLATE)
+  end
+
+  def user_info_3_unpack(buffer)
+    user = Hash.new
+    USER_INFO_3.each_with_index do |field,offset|
+      user[field[0]] = field[1].class == Fixnum ?
+        dword_to_i(buffer, offset) : lpwstr_to_s(buffer, offset)
+    end
+    user
+  end
+
+  def set_info(args)
+    user = user_info_3(args)
+    buffer = user_info_3_pack(user)
+    rc = NetUserSetInfo.call(nil, @name, 3, buffer, nil)
+    if rc != NERR_Success
+      raise ArgumentError, get_last_error(rc)
+    end
+  end
+
+  public
+
+  def initialize(username)
+    @username = username
+    @name = multi_to_wide(username)
+  end
+
+  LOGON32_PROVIDER_DEFAULT = 0
+  LOGON32_LOGON_NETWORK = 3
+  #XXX for an extra painful alternative, see: http://support.microsoft.com/kb/180548
+  def validate_credentials(passwd)
+    token = 0.chr * PTR_SIZE
+    res = LogonUser.call(@username, nil, passwd,
+                         LOGON32_LOGON_NETWORK, LOGON32_PROVIDER_DEFAULT, token)
+    if res == 0
+      return false
+    end
+    ::Windows::Handle::CloseHandle.call(token.unpack('L')[0])
+    return true
+  end
+
+  def get_info
+    ptr  = 0.chr * PTR_SIZE
+    rc = NetUserGetInfo.call(nil, @name, 3, ptr)
+
+    if rc != NERR_Success
+      raise ArgumentError, get_last_error(rc)
+    end
+
+    ptr = ptr.unpack('L')[0]
+    buffer = 0.chr * SIZEOF_USER_INFO_3
+    memcpy(buffer, ptr, buffer.size)
+    NetApiBufferFree(ptr)
+    user_info_3_unpack(buffer)
+  end
+
+  def add(args)
+    user = user_info_3(args)
+    buffer = user_info_3_pack(user)
+
+    rc = NetUserAdd.call(nil, 3, buffer, rc)
+    if rc != NERR_Success
+      raise ArgumentError, get_last_error(rc)
+    end
+
+    #usri3_primary_group_id:
+    #"When you call the NetUserAdd function, this member must be DOMAIN_GROUP_RID_USERS"
+    NetLocalGroupAddMembers(nil, multi_to_wide("Users"), 3, buffer[0,PTR_SIZE], 1)
+  end
+
+  def user_modify(&proc)
+    user = get_info
+    user[:last_logon] = user[:units_per_week] = 0 #ignored as per USER_INFO_3 doc
+    user[:logon_hours] = nil #PBYTE field; \0 == no changes
+    proc.call(user)
+    set_info(user)
+  end
+
+  def update(args)
+    user_modify do |user|
+      args.each do |key,val|
+        user[key] = val
+      end
+    end
+  end
+
+  def delete
+    rc = NetUserDel.call(nil, @name)
+    if rc != NERR_Success
+      raise ArgumentError, get_last_error(rc)
+    end
+  end
+
+  def disable_account
+    user_modify do |user|
+      user[:flags] |= UF_ACCOUNTDISABLE
+    end
+  end
+
+  def enable_account
+    user_modify do |user|
+      user[:flags] &= ~UF_ACCOUNTDISABLE
+    end
+  end
+
+  def check_enabled
+    (get_info()[:flags] & UF_ACCOUNTDISABLE) != 0
+  end
+end

data/lib/chef/util/windows/volume.rb

@@ -0,0 +1,59 @@
+#
+# Author:: Doug MacEachern (<dougm@vmware.com>)
+# Copyright:: Copyright (c) 2010 VMware, 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.
+#
+
+#simple wrapper around Volume APIs. might be possible with WMI, but possibly more complex.
+
+require 'chef/util/windows'
+require 'windows/volume'
+
+class Chef::Util::Windows::Volume < Chef::Util::Windows
+
+  private
+  include Windows::Volume
+  #XXX not defined in the current windows-pr release
+  DeleteVolumeMountPoint =
+    Windows::API.new('DeleteVolumeMountPoint', 'S', 'B') unless defined? DeleteVolumeMountPoint
+
+  public
+
+  def initialize(name)
+    name += "\\" unless name =~ /\\$/ #trailing slash required
+    @name = name
+  end
+  
+  def device
+    buffer = 0.chr * 256
+    if GetVolumeNameForVolumeMountPoint(@name, buffer, buffer.size)
+      return buffer[0,buffer.size].unpack("Z*")[0]
+    else
+      raise ArgumentError, get_last_error
+    end
+  end
+
+  def delete
+    unless DeleteVolumeMountPoint.call(@name)
+      raise ArgumentError, get_last_error
+    end
+  end
+
+  def add(device)
+    unless SetVolumeMountPoint(@name, device)
+      raise ArgumentError, get_last_error
+    end
+  end
+end

data/lib/chef/util/windows.rb

@@ -0,0 +1,56 @@
+#
+# Author:: Doug MacEachern (<dougm@vmware.com>)
+# Copyright:: Copyright (c) 2010 VMware, 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.
+#
+#requires: gem install windows-pr
+require 'windows/api'
+require 'windows/error'
+require 'windows/handle'
+require 'windows/unicode'
+require 'windows/msvcrt/buffer'
+require 'windows/msvcrt/string'
+require 'windows/network/management'
+
+class Chef
+  class Util
+    class Windows
+      protected
+
+      include ::Windows::Error
+      include ::Windows::Unicode
+      include ::Windows::MSVCRT::Buffer
+      include ::Windows::MSVCRT::String
+      include ::Windows::Network::Management
+
+      PTR_SIZE = 4 #XXX 64-bit
+
+      def lpwstr_to_s(buffer, offset)
+        str = 0.chr * (256 * 2) #XXX unhardcode this length (*2 for WCHAR)
+        wcscpy str, buffer[offset*PTR_SIZE,PTR_SIZE].unpack('L')[0]
+        wide_to_multi str
+      end
+
+      def dword_to_i(buffer, offset)
+        buffer[offset*PTR_SIZE,PTR_SIZE].unpack('i')[0] || 0
+      end
+
+      #return pointer for use with pack('L')
+      def str_to_ptr(v)
+        [v].pack('p*').unpack('L')[0]
+      end
+    end
+  end
+end

data/lib/chef/version.rb

@@ -0,0 +1,21 @@
+#
+# Author:: Daniel DeLeo (<dan@opscode.com>)
+# Copyright:: Copyright (c) 2010 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.
+
+class Chef
+  CHEF_ROOT = File.dirname(File.expand_path(File.dirname(__FILE__)))
+  VERSION = '0.9.12.1'
+end