bindings/ruby/tests/cim_process.rb

# cim_process.rb
#  enumerate/pull/release for CIM_Process
#
#

$:.unshift "../../../bindings/ruby"
$:.unshift "../../../build/bindings/ruby"
$:.unshift "../.libs"

require 'test/unit'
require 'rexml/document'
require 'openwsman/openwsman'
require '_client'

class WsmanTest < Test::Unit::TestCase
  def get_owner process
    client = Client.open
    assert client
    options = Openwsman::ClientOptions.new
    # WMI just needs the Handle
    unless process.name == "Win32_Service"
      options.add_selector( "CSCreationClassName", process.CSCreationClassName )
      options.add_selector( "CSName", process.CSName )
      options.add_selector( "OSCreationClassName", process.OSCreationClassName )
      options.add_selector( "OSName", process.OSName )
      options.add_selector( "CreationClassName", process.CreationClassName )
    end
    options.add_selector( "Handle", process.Handle )
    uri = "http://schemas.dmtf.org/wbem/wscim/1/cim-schema/2/CIM_Process"

    method = "GetOwner"
    result = client.invoke( options, uri, method )

  end

  def test_client
    client = Client.open
    assert client
    options = Openwsman::ClientOptions.new
    assert options
#    options.flags = Openwsman::CLIENTOPTION_DUMP_REQUEST
#    puts "Flags = #{options.flags}"

#
# see http://msdn2.microsoft.com/en-us/library/aa386179.aspx for a list of CIM classes
#   the Win32 classes are derived from
#
    uri = "http://schemas.dmtf.org/wbem/wscim/1/cim-schema/2/CIM_Process"

    result = client.enumerate( options, nil, uri )
    assert result

    results = 0
    faults = 0
    context = nil
 
    printf("%-20s %-10s %-10s  %s\n", "User", "PID", "VSZ", "Command");
    puts "-------------------------------------------------------------"
loop do
    context = result.context
    break unless context
#    puts "Context #{context} retrieved"

    result = client.pull( options, nil, uri, context )
    break unless result

    results += 1
    if result.fault?
      puts "Got fault"
      faults += 1
      break
    end
    
    items = result.body.PullResponse.Items
    node = items.child

#    node = body.child( 0, Openwsman::NS_ENUMERATION, "PullResponse" );
#    node = node.child( 0, Openwsman::NS_ENUMERATION, "Items" );
#    node = node.child( 0, uri, "Win32_Service" );

#    name = node.child( 0, uri, "Name" ).text;
#    state = node.child( 0, uri, "State" ).text;

    if node.name == "Win32_Service"
      caption = node["Caption"]
      handle = node["Handle"]
      virtual_size = node["VirtualSize"]
      proc_id = node["ProcessId"]
      cmd = node["ExecutablePath"]
    else
      caption = node["Caption"]
      handle = node["Handle"]
      virtual_size = 0
      proc_id = handle
      cmd = node["Name"]
    end
    user = ""

    ires = get_owner node
    if ires
      b = ires.body
      output = b.GetOwner_OUTPUT
      output.each do |child|
	text = child.text
	if child.name == "User"
	  user = text
	end
      end if output
    end
    vsz =  (virtual_size.to_s.to_i / (1024 * 1024 ) ).to_f

    printf("%-20s %-10s %-5.0f  %s\n", user, proc_id, vsz, cmd);

end

    client.release( options, uri, context ) if context
    puts "Context released, #{results} results, #{faults} faults"
  end
end