Added encrypt wrapper

1 files changed, 48 insertions, 0 deletions

diff --git a/lib/chef/win32/crypto.rb b/lib/chef/win32/crypto.rb
new file mode 100644
index 0000000000..5cb0a25ee7
--- /dev/null
+++ b/lib/chef/win32/crypto.rb
@@ -0,0 +1,48 @@
+#

+# Author:: Jay Mundrawala (<jdm@chef.io>)

+# Copyright:: Copyright 2015 Chef Software, 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/win32/error'

+require 'chef/win32/api/memory'

+require 'chef/win32/api/crypto'

+

+class Chef

+  module ReservedNames::Win32

+    class Crypto

+      include Chef::ReservedNames::Win32::API::Crypto

+      extend Chef::ReservedNames::Win32::API::Crypto

+

+      def self.encrypt(str, &block)

+        data_blob = CRYPT_INTEGER_BLOB.new

+        unless CryptProtectData(CRYPT_INTEGER_BLOB.new(str.to_wstring), nil, nil, nil, nil, 0, data_blob)

+          Chef::ReservedNames::Win32::Error.raise!

+        end

+        bytes = data_blob[:pbData].get_bytes(0, data_blob[:cbData])

+        if block

+          block.call(bytes)

+        else

+          bytes

+        end

+      ensure

+        unless data_blob[:pbData].null?

+          Chef::ReservedNames::Win32::Memory.local_free(data_blob[:pbData])

+        end

+      end

+

+    end

+  end

+end