summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorClaire McQuin <claire@chef.io>2015-08-26 11:08:03 -0700
committerClaire McQuin <claire@chef.io>2015-08-31 13:23:48 -0700
commit5d4d979768029a6e396608ab683014946aa1148d (patch)
treebfff236163d16252bdd09cdd298499cb5a247a74
parentd41c83b2543aca51aaef0efe3c1ffe756fc46da0 (diff)
downloadchef-mcquin/monkeypatch-win32-registry-write.tar.gz
Patch Win32::Registry#write on Ruby 2.1, resolves encoding errorsmcquin/monkeypatch-win32-registry-write
-rw-r--r--lib/chef/win32/unicode.rb27
1 files changed, 27 insertions, 0 deletions
diff --git a/lib/chef/win32/unicode.rb b/lib/chef/win32/unicode.rb
index e7399d5255..ea10dc71d0 100644
--- a/lib/chef/win32/unicode.rb
+++ b/lib/chef/win32/unicode.rb
@@ -53,3 +53,30 @@ class String
Chef::ReservedNames::Win32::Unicode.utf8_to_wide(self)
end
end
+
+# https://bugs.ruby-lang.org/issues/11439
+if RUBY_VERSION =~ /^2\.1/
+ module Win32
+ class Registry
+ def write(name, type, data)
+ case type
+ when REG_SZ, REG_EXPAND_SZ
+ data = data.to_s.encode(WCHAR) + WCHAR_NUL
+ when REG_MULTI_SZ
+ data = data.to_a.map {|s| s.encode(WCHAR)}.join(WCHAR_NUL) << WCHAR_NUL << WCHAR_NUL
+ when REG_BINARY
+ data = data.to_s
+ when REG_DWORD
+ data = API.packdw(data.to_i)
+ when REG_DWORD_BIG_ENDIAN
+ data = [data.to_i].pack('N')
+ when REG_QWORD
+ data = API.packqw(data.to_i)
+ else
+ raise TypeError, "Unsupported type #{type}"
+ end
+ API.SetValue(@hkey, name, type, data, data.bytesize)
+ end
+ end
+ end
+end \ No newline at end of file