Module: PDK::Util::Windows::Process

Extended by:
FFI::Library, String
Defined in:
lib/pdk/util/windows/process.rb

Class Method Summary collapse

Methods included from String

wide_string

Class Method Details

.environment_hash ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
 
# File 'lib/pdk/util/windows/process.rb', line 8

def environment_hash
  env_ptr = GetEnvironmentStringsW()

  contains_unicode_replacement = ->(string) do
    return false unless string.include?("\uFFFD")

    PDK.logger.warning _(
      'Discarding environment variable %{string} which contains invalid ' \
      'bytes',
    ) % { string: string }
    true
  end

  # pass :invalid => :replace to the Ruby String#encode to use replacement
  # characters
  pairs = env_ptr.read_arbitrary_wide_string_up_to(65_534, :double_null, invalid: :replace)
                 .split(?\x00)
                 .reject { |env_str| env_str.nil? || env_str.empty? || env_str[0] == '=' }
                 .reject { |env_str| contains_unicode_replacement.call(env_str) }
                 .map { |env_pair| env_pair.split('=', 2) }
  Hash[pairs]
ensure
  if env_ptr && !env_ptr.null?
    if FreeEnvironmentStringsW(env_ptr) == PDK::Util::Windows::WIN32_FALSE
      PDK.logger.debug _('FreeEnvironmentStringsW memory leak')
    end
  end
end

.set_environment_variable(name, val) ⇒ Object

Raises:

  • (ArgumentError) 


38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/pdk/util/windows/process.rb', line 38

def set_environment_variable(name, val)
  raise ArgumentError, _('Environment variable name must not be nil or empty') if name.nil? || name.empty?

  FFI::MemoryPointer.from_string_to_wide_string(name) do |name_ptr|
    if val.nil?
      if SetEnvironmentVariableW(name_ptr, FFI::MemoryPointer::NULL) == PDK::Util::Windows::WIN32_FALSE
        raise _('Failed to remove environment variable: %{name}') % { name: name }
      end
    else
      FFI::MemoryPointer.from_string_to_wide_string(val) do |val_ptr|
        if SetEnvironmentVariableW(name_ptr, val_ptr) == PDK::Util::Windows::WIN32_FALSE
          raise _('Failed to set environment variaible: %{name}') % { name: name }
        end
      end
    end
  end
end