hawatel_ps 0.1.1

data/spec/windows/factories/proc_fetch_factorie.rb

@@ -0,0 +1,76 @@
+FactoryGirl.define do
+  factory :win32_data, class:Hash do
+    proc_attrs  {{
+        :availablevirtualsize => "3817041920",
+        :caption => "rubymine.exe",
+        :childs => [],
+        :commandline => "\"C:\\Program Files (x86)\\JetBrains\\RubyMine 8.0.3\\bin\\rubymine.exe\" ",
+        :cpupercent => "19.47",
+        :creationclassname => "Win32_Process",
+        :creationdate => "20160307091252.612549+060",
+        :cscreationclassname => "Win32_ComputerSystem",
+        :csname => "TEST",
+        :description => "rubymine.exe",
+        :domain => "CORP",
+        :executablepath => "C:\\Program Files (x86)\\JetBrains\\RubyMine 8.0.3\\bin\\rubymine.exe",
+        :handle => "9336",
+        :handlecount => "799",
+        :kernelmodetime => "3058906250",
+        :maximumworkingsetsize => "1380",
+        :memorypercent => "5.09",
+        :minimumworkingsetsize => "200",
+        :name => "rubymine.exe",
+        :oscreationclassname => "Win32_OperatingSystem",
+        :osname => "Microsoft Windows|C:\\WINDOWS|\\Device\\Harddisk0\\Partition2",
+        :otheroperationcount => "3662530",
+        :othertransfercount => "65362876",
+        :pagefaults => "404989",
+        :pagefileusage => "480264",
+        :parentprocessid => "9336",
+        :peakpagefileusage => "481136",
+        :peakvirtualsize => "997543936",
+        :peakworkingsetsize => "428776",
+        :priority => "8",
+        :privatepagecount => "491790336",
+        :processid => "9336",
+        :quotanonpagedpoolusage => "57",
+        :quotapagedpoolusage => "248",
+        :quotapeaknonpagedpoolusage => "92",
+        :quotapeakpagedpoolusage => "273",
+        :readoperationcount => "304603",
+        :readtransfercount => "306717431",
+        :sessionid => "9",
+        :threadcount => "56",
+        :user => "asmith",
+        :usermodetime => "94399218750",
+        :sid => "S-1-2-10-0000000000-1111111111-2222222222-1000",
+        :virtualsize => "982863872",
+        :windowsversion => "10.0.10586",
+        :workingsetsize => "437301248",
+        :writeoperationcount => "80408",
+        :writetransfercount => "450486611"}}
+
+    users {{
+        :accounttype => 512,
+        :caption => "CORP\\asmith",
+        :description => "Account for administering the computer/domain",
+        :disabled => "false",
+        :domain => "CORP",
+        'domain' => "CORP",
+        :fullname => "",
+        :installdate => nil,
+        :localaccount => "true",
+        :lockout => "false",
+        :name => "asmith",
+        'name' => "asmith",
+        :passwordchangeable => "true",
+        :passwordexpires => "false",
+        :passwordrequired => "true",
+        :sid => "S-1-2-10-0000000000-1111111111-2222222222-1000",
+        'sid' => "S-1-2-10-0000000000-1111111111-2222222222-1000",
+        :sidtype => 1,
+        :status => "OK"
+    }}
+    initialize_with { attributes }
+  end
+end

data/spec/windows/tdd/proc_control_spec.rb

@@ -0,0 +1,36 @@
+require 'spec_helper'
+
+describe HawatelPS::Windows::ProcControl do
+  let(:win32_data) { FactoryGirl.build(:win32_data) }
+  let(:wmi_obj) { double 'WIN32OLE', :ole_respond_to? => 1, :Terminate => 0}
+
+  before do
+    allow(HawatelPS::Windows::ProcFetch).to receive(:get_process).and_return(Array.new([win32_data[:proc_attrs]]))
+  end
+
+  it '#state process is running' do
+    allow(Process).to receive(:kill).and_return(1)
+    process = HawatelPS::Windows::ProcTable.proc_table
+    expect(process[0].status).to eq("running")
+  end
+
+  it '#state process is not running' do
+    allow(Process).to receive(:kill).and_raise(Errno::ESRCH)
+    process = HawatelPS::Windows::ProcTable.proc_table
+    expect(process[0].status).to eq("not running")
+  end
+
+  it '#terminate process successful' do
+    allow(@proc_attrs).to receive(:Terminate).and_return(0)
+    allow(@proc_attrs).to receive(:ole_respond_to?).and_return(1)
+    process = HawatelPS::Windows::ProcTable.proc_table
+    expect(process[0].terminate).to eq(0)
+  end
+
+  it '#terminate process no successful' do
+    allow(@proc_attrs).to receive(:Terminate).and_raise(WIN32OLERuntimeError)
+    allow(@proc_attrs).to receive(:ole_respond_to?).and_return(1)
+    process = HawatelPS::Windows::ProcTable.proc_table
+    expect(process[0].terminate).to eq(1)
+  end
+end

data/spec/windows/tdd/proc_fetch_spec.rb

@@ -0,0 +1,73 @@
+require 'spec_helper'
+require 'ostruct'
+
+describe HawatelPS::Windows::ProcFetch do
+  let(:win32_data) { FactoryGirl.build(:win32_data) }
+  let(:wmi_cli) { double 'HawatelPS::Windows::WmiCli' }
+  let(:native_wmi_obj) { double 'WIN32OLE'}
+  let(:wmi_object) { double 'HawatelPS::Windows::WmiCli::Instance',
+                            :properties => win32_data[:proc_attrs],
+                            :wmi_ole_object => native_wmi_obj}
+  let(:user_sid) { prepare_owner_sid_as_struct(win32_data[:users]) }
+  let(:user_struct) { prepare_user_info_as_struct(win32_data[:users]) }
+  let(:system_info_struct) { OpenStruct.new(:properties => {:totalvisiblememorysize => "8388608"})}
+  let(:system_idle_struct) { OpenStruct.new(:properties => {:kernelmodetime => "500592031250"})}
+
+  before do
+  allow(HawatelPS::Windows::WmiCli).to receive(:new).and_return(wmi_cli)
+  win32_data[:proc_attrs][:wmi_object] = native_wmi_obj
+
+  allow(wmi_cli).to receive(:query).with('SELECT * FROM Win32_Process').and_return(Array.new([wmi_object]))
+  allow(wmi_cli).to receive(:query).with('SELECT * FROM Win32_UserAccount').and_return(Array.new([user_struct]))
+  allow(wmi_cli).to receive(:query).with('SELECT * FROM Win32_OperatingSystem').and_return(Array.new([system_info_struct]))
+  allow(wmi_cli).to receive(:query).with("SELECT KernelModeTime FROM Win32_Process WHERE ProcessId = '0'").and_return(Array.new([system_idle_struct]))
+
+  allow(wmi_object).to receive(:execMethod).with('GetAvailableVirtualSize').and_return(
+      OpenStruct.new(:AvailableVirtualSize => win32_data[:proc_attrs][:availablevirtualsize]))
+  end
+
+  it '#get_process' do
+    allow(wmi_object).to receive(:execMethod).with('GetOwnerSid').and_return(user_sid)
+
+    proc_table = HawatelPS::Windows::ProcFetch.get_process
+
+    expect(proc_table[0]).to eq(win32_data[:proc_attrs])
+  end
+
+  it '#get_owner_sid raise error' do
+    allow(wmi_object).to receive(:execMethod).with('GetOwnerSid').and_raise(HawatelPS::Windows::WmiCliException)
+    proc_table = HawatelPS::Windows::ProcFetch.get_process
+    expect(proc_table[0][:sid]).to be_nil
+  end
+
+  it '#get_avail_virtual_size raise error' do
+    allow(wmi_object).to receive(:execMethod).with('GetOwnerSid').and_return(user_sid)
+    allow(wmi_object).to receive(:execMethod).with('GetAvailableVirtualSize').and_raise(HawatelPS::Windows::WmiCliException)
+    proc_table = HawatelPS::Windows::ProcFetch.get_process
+    expect(proc_table[0][:availablevirtualsize]).to be_nil
+  end
+
+  it '#cpu_percent return 0' do
+    allow(wmi_object).to receive(:execMethod).with('GetOwnerSid').and_return(user_sid)
+    allow(HawatelPS::Windows::ProcFetch).to receive(:cpu_time).and_return(0)
+
+    proc_table = HawatelPS::Windows::ProcFetch.get_process
+
+    expect(proc_table[0][:cpupercent]).to eq("0.0")
+  end
+
+end
+
+private
+
+def prepare_user_info_as_struct(user_info)
+  user_struct = OpenStruct.new
+  user_struct.properties = user_info
+  return user_struct
+end
+
+def prepare_owner_sid_as_struct(user_info)
+  sid_struct = OpenStruct.new
+  sid_struct.Sid = user_info[:sid]
+  return sid_struct
+end

data/spec/windows/tdd/proc_table_spec.rb

@@ -0,0 +1,71 @@
+require 'spec_helper'
+
+describe HawatelPS::Windows::ProcTable do
+  let(:win32_data) { FactoryGirl.build(:win32_data) }
+
+  before do
+    allow(HawatelPS::Windows::ProcFetch).to receive(:get_process).and_return(Array.new([win32_data[:proc_attrs]]))
+  end
+
+
+  it "return all process instances" do
+    process = HawatelPS::Windows::ProcTable.proc_table
+    expect(process[0].processid).to match(/^[0-9]+$/)
+  end
+
+  it "return proc attributes" do
+    process = HawatelPS::Windows::ProcTable.proc_table
+    expect(process[0].each).to be_a(Enumerator)
+  end
+
+  it "search process by name" do
+    process = HawatelPS::Windows::ProcTable.search_by_name('rubymine.exe')
+    expect(process[0].processid).to match(/^[0-9]+$/)
+  end
+
+  it "search process by name using regex" do
+    process = HawatelPS::Windows::ProcTable.search_by_name('/^ruby/')
+    expect(process[0].processid).to match(/^[0-9]*$/)
+  end
+
+  it "search process by name using regex with raise error" do
+    process = HawatelPS::Windows::ProcTable.search_by_name('/^jasdfsdfva/')
+    expect { process[0].pid }.to raise_error(NoMethodError)
+  end
+
+  it "search process by pid" do
+    process = HawatelPS::Windows::ProcTable.search_by_pid(9336)
+    expect(process.processid).to eq('9336')
+  end
+
+  it "search process by condition == " do
+    process = HawatelPS::Windows::ProcTable.search_by_condition(:attr => 'processid', :oper => '==', :value => '9336')
+    expect(process[0].processid).to eq('9336')
+  end
+
+  it "search process by condition <= " do
+    process = HawatelPS::Windows::ProcTable.search_by_condition(:attr => 'workingsetsize', :oper => '<=', :value => '437301248')
+    expect(process[0].processid).to match(/^[0-9]*$/)
+  end
+
+  it "search process by condition >= " do
+    process = HawatelPS::Windows::ProcTable.search_by_condition(:attr => 'workingsetsize', :oper => '>=', :value => '200000')
+    expect(process[0].processid).to match(/^[0-9]*$/)
+  end
+
+  it "search process by condition != " do
+    process = HawatelPS::Windows::ProcTable.search_by_condition(:attr => 'workingsetsize', :oper => '!=', :value => '200000')
+    expect(process[0].processid).to match(/^[0-9]*$/)
+  end
+
+  it "search process by condition > " do
+    process = HawatelPS::Windows::ProcTable.search_by_condition(:attr => 'workingsetsize', :oper => '>', :value => '200000')
+    expect(process[0].processid).to match(/^[0-9]*$/)
+  end
+
+  it "search process by condition < " do
+    process = HawatelPS::Windows::ProcTable.search_by_condition(:attr => 'workingsetsize', :oper => '<', :value => '537301248')
+    expect(process[0].processid).to match(/^[0-9]*$/)
+  end
+
+end

data/spec/windows/tdd/wmi_spec.rb

@@ -0,0 +1,59 @@
+require 'spec_helper'
+require 'ostruct'
+
+describe HawatelPS::Windows::WmiCli do
+  let(:win32_data) { FactoryGirl.build(:win32_data) }
+  let(:wmi_ole) { double 'WIN32OLE', :properties_ => Array.new([OpenStruct.new(:name => 'name', :value => 'rubymine')])}
+  let(:instance) { double 'HawatelPS::Windows::WmiCli::Instance' }
+
+  before do
+    allow(WIN32OLE).to receive(:new).with('WbemScripting.SWbemLocator').and_return(wmi_ole)
+  end
+
+  it 'query results success' do
+    allow_connect_server
+    allow(wmi_ole).to receive(:execMethod_).with('GetOwner')
+                          .and_return(OpenStruct.new(:User => 'asmith', :Domain => 'test'))
+
+    obj = HawatelPS::Windows::WmiCli.new.query('SELECT * FROM Win32_Process')[0]
+    expect(obj.class).to eq(HawatelPS::Windows::WmiCli::Instance)
+    expect(obj.wmi_ole_object).to eq(wmi_ole)
+    expect(obj.properties[:name]).to eq('rubymine')
+
+    owner = obj.execMethod('GetOwner')
+    expect(owner.User).to eq('asmith')
+    expect(owner.Domain).to eq('test')
+  end
+
+  it 'query raise exception' do
+    allow(wmi_ole).to receive(:ConnectServer).and_return(wmi_ole)
+    allow(wmi_ole).to receive(:ExecQuery).with('SELECT * FROM Win32_ERROR').and_raise(WIN32OLERuntimeError)
+
+    expect{HawatelPS::Windows::WmiCli.new.query('SELECT * FROM Win32_ERROR')}
+        .to raise_error(HawatelPS::Windows::WmiCliException)
+  end
+
+  it 'ConnectServer raise exception' do
+    allow(wmi_ole).to receive(:ConnectServer).and_raise(WIN32OLERuntimeError)
+
+    expect{HawatelPS::Windows::WmiCli.new.query('SELECT * FROM Win32_Process')}
+        .to raise_error(HawatelPS::Windows::WmiCliException)
+  end
+
+  it '#execMethod raise exception' do
+    allow_connect_server
+    allow(wmi_ole).to receive(:execMethod_).with('GetOwner').and_raise(WIN32OLERuntimeError)
+
+    obj = HawatelPS::Windows::WmiCli.new.query('SELECT * FROM Win32_Process')[0]
+    expect { obj.execMethod('GetOwner') }.to raise_error(HawatelPS::Windows::WmiCliException)
+  end
+
+  private
+
+  def allow_connect_server
+    allow(wmi_ole).to receive(:ExecQuery).with('SELECT * FROM Win32_Process').and_return(Array.new([wmi_ole]))
+    allow(wmi_ole).to receive(:ConnectServer).and_return(wmi_ole)
+    allow(wmi_ole).to receive(:ole_respond_to?).and_return(true)
+  end
+
+end

metadata

@@ -0,0 +1,181 @@
+--- !ruby/object:Gem::Specification
+name: hawatel_ps
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Przemyslaw Mantaj
+- Daniel Iwaniuk
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-03-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: factory_girl
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: |-
+  HawatelPS (hawatel_ps) is a Ruby gem for retrieving information about running processes. It is easy to use and you can get useful information about a process.
+                            You can terminate, suspend, resume and check status of the process on Linxu platform. On Windows platform you can terminate and check state of the process.
+email:
+- przemyslaw.mantaj@hawatel.com
+- daniel.iwaniuk@hawatel.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- ".yardopts"
+- CONTRIBUTING.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- hawatel_ps.gemspec
+- lib/hawatel_ps.rb
+- lib/hawatel_ps/linux.rb
+- lib/hawatel_ps/linux/proc_control.rb
+- lib/hawatel_ps/linux/proc_fetch.rb
+- lib/hawatel_ps/linux/proc_info.rb
+- lib/hawatel_ps/linux/proc_table.rb
+- lib/hawatel_ps/shared/hawatelps_exception.rb
+- lib/hawatel_ps/version.rb
+- lib/hawatel_ps/windows.rb
+- lib/hawatel_ps/windows/proc_control.rb
+- lib/hawatel_ps/windows/proc_fetch.rb
+- lib/hawatel_ps/windows/proc_info.rb
+- lib/hawatel_ps/windows/proc_table.rb
+- lib/hawatel_ps/windows/wmi/wmi_cli.rb
+- lib/hawatel_ps/windows/wmi/wmi_exception.rb
+- lib/hawatel_ps/windows/wmi/wmi_instance.rb
+- spec/hawatel_ps_spec.rb
+- spec/linux/bdd/proc_spec.rb
+- spec/linux/factories/etc/passwd
+- spec/linux/factories/proc/1761/cmdline
+- spec/linux/factories/proc/1761/environ
+- spec/linux/factories/proc/1761/io
+- spec/linux/factories/proc/1761/limits
+- spec/linux/factories/proc/1761/stat
+- spec/linux/factories/proc/1761/status
+- spec/linux/factories/proc/meminfo
+- spec/linux/factories/proc/net/tcp
+- spec/linux/factories/proc/net/udp
+- spec/linux/factories/proc/uptime
+- spec/linux/support/stub_dir.rb
+- spec/linux/support/stub_file.rb
+- spec/linux/tdd/proc_fetch_spec.rb
+- spec/linux/tdd/proc_table_spec.rb
+- spec/shared/hawatelps_exception_spec.rb
+- spec/spec_helper.rb
+- spec/windows/bdd/proc_spec.rb
+- spec/windows/factories/proc_fetch_factorie.rb
+- spec/windows/tdd/proc_control_spec.rb
+- spec/windows/tdd/proc_fetch_spec.rb
+- spec/windows/tdd/proc_table_spec.rb
+- spec/windows/tdd/wmi_spec.rb
+homepage: http://github.com/hawatel/hawatel_ps
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '1.9'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Ruby gem for retrieving information about running processes
+test_files:
+- spec/hawatel_ps_spec.rb
+- spec/linux/bdd/proc_spec.rb
+- spec/linux/factories/etc/passwd
+- spec/linux/factories/proc/1761/cmdline
+- spec/linux/factories/proc/1761/environ
+- spec/linux/factories/proc/1761/io
+- spec/linux/factories/proc/1761/limits
+- spec/linux/factories/proc/1761/stat
+- spec/linux/factories/proc/1761/status
+- spec/linux/factories/proc/meminfo
+- spec/linux/factories/proc/net/tcp
+- spec/linux/factories/proc/net/udp
+- spec/linux/factories/proc/uptime
+- spec/linux/support/stub_dir.rb
+- spec/linux/support/stub_file.rb
+- spec/linux/tdd/proc_fetch_spec.rb
+- spec/linux/tdd/proc_table_spec.rb
+- spec/shared/hawatelps_exception_spec.rb
+- spec/spec_helper.rb
+- spec/windows/bdd/proc_spec.rb
+- spec/windows/factories/proc_fetch_factorie.rb
+- spec/windows/tdd/proc_control_spec.rb
+- spec/windows/tdd/proc_fetch_spec.rb
+- spec/windows/tdd/proc_table_spec.rb
+- spec/windows/tdd/wmi_spec.rb
+has_rdoc: