vmopt 0.0.2

data/lib/vmopt/utils/registry.rb

@@ -0,0 +1,13 @@
+#encoding: utf-8
+#功能：操作注册表，注册硬件信息
+module Registry
+  class << self
+    def hklm_read(key, value)
+      require 'win32/registry'
+      reg = Win32::Registry::HKEY_LOCAL_MACHINE.open(key)
+      rval = reg[value]
+      reg.close
+      rval
+    end
+  end
+end

data/lib/vmopt/utils/wmi.rb

@@ -0,0 +1,19 @@
+#encoding: utf-8
+#功能：wmi模块，用于查询本地设备信息
+module WMI
+  class << self
+    def connect(uri = wmi_resource_uri)
+      require 'win32ole'
+      WIN32OLE.codepage = WIN32OLE::CP_UTF8
+      WIN32OLE.connect(uri)
+    end
+
+    def wmi_resource_uri( host = '.' )
+      "winmgmts:{impersonationLevel=impersonate}!//#{host}/root/cimv2"
+    end
+
+    def execquery(query)
+      connect().execquery(query)
+    end
+  end
+end

data/lib/vmopt/version.rb

@@ -0,0 +1,3 @@
+module Vmopt
+  VERSION = "0.0.2"
+end

data/lib/vmopt/windows/win_net.rb

@@ -0,0 +1,295 @@
+#encoding: utf-8
+#功能：提供查询网卡详细信息的接口
+#
+$LOAD_PATH.unshift(File.join(__FILE__, '../..', '..'))
+require "vmopt/utils/ip"
+require "vmopt/utils/registry"
+require "vmopt/utils/wmi"
+
+module WinNetError
+  class NoInterfaceError < RuntimeError; end  #没有给网络借口
+  class CMDexecFailedError < RuntimeError; end  #命令执行出错
+  class NotFindFileError < RuntimeError; end   #查不到对应文件
+  class NoSavePathError < RuntimeError;end #没有路径可保存
+  class InvalidParamError < RuntimeError;end #参数错误
+end
+
+class WinNet
+  # The WMI query used to return ip information
+  #
+  # @return [String]
+  #
+  # @api private
+  #WMI_IP_INFO_QUERY = 'SELECT Description, ServiceName, IPAddress, IPConnectionMetric, InterfaceIndex, Index, IPSubnet, MACAddress, MTU, SettingID FROM Win32_NetworkAdapterConfiguration WHERE IPConnectionMetric IS NOT NULL AND IPEnabled = TRUE'
+  WMI_IP_INFO_QUERY = 'SELECT * FROM Win32_NetworkAdapterConfiguration WHERE IPConnectionMetric IS NOT NULL AND IPEnabled = TRUE'
+
+
+  # Mapping fact names to WMI properties of the Win32_NetworkAdapterConfiguration
+  #
+  # @api private
+  WINDOWS_LABEL_WMI_MAP = {
+      :ipaddress => 'IPAddress',
+      :ipaddress6 => 'IPAddress',
+      :macaddress => 'MACAddress',
+      :netmask => 'IPSubnet'
+  }
+  
+  #net connect status map 
+  WINDOWS_CON_STATUS_MAP = {
+    0 => "Disconnected", 
+    1 => "Connecting",
+    2 => "Connected",
+    3 => "Disconnecting",
+    4 => "Hardware not present",
+    5 => "Hardware disabled",
+    6 => "Hardware malfunction",
+    7 => "Media disconnected",
+    8 => "Authenticating",
+    9 => "Authentication succeeded",
+    10 => "Authentication failed",
+    11 => "Invalid address",
+    12 => "Credentials required"
+  }
+
+  def self.to_s
+    'windows'
+  end
+
+  # Windows doesn't display netmask in hex.
+  #
+  # @return [Boolean] false by default
+  #
+  # @api private
+  def self.convert_netmask_from_hex?
+    false
+  end
+
+  # Retrieves a list of unique interfaces names.
+  #
+  # @return [Array<String>]
+  #
+  # @api private
+  def self.interfaces
+    interface_names = []
+
+      WMI.execquery("SELECT * FROM Win32_NetworkAdapter").each do |nic|
+        interface_names << nic.NetConnectionId unless nic.NetConnectionId.nil? or nic.NetConnectionId.empty?
+      end
+
+    interface_names.uniq
+  end
+  
+  # Retrieves netadapter 
+  #
+  # @return [Array<win32ole>]
+  #
+  # @api private
+  def self.network_adapter
+    nics = []
+    WMI.execquery("SELECT * FROM Win32_NetworkAdapter WHERE NetConnectionID IS NOT NULL" ).each do |nic|
+      nics << nic
+    end
+    nics
+
+  end
+  
+  def self.netconnstatus
+    stat={}
+    network_adapter.each do |interface|
+      stat[interface.netConnectionId] = WINDOWS_CON_STATUS_MAP[interface.netConnectionStatus]
+    end
+    stat
+  end
+
+  # Get the value of an interface and label. For example, you may want to find
+  # the MTU for eth0.
+  #
+  # @param [String] interface the name of the interface returned by the {#interfaces} method.
+  # @param [String] label the type of value to return, e.g. ipaddress
+  # @return [String] the value, or nil if not defined
+  #
+  # @api private
+  def self.value_for_interface_and_label(interface, label)
+    wmi_value = WINDOWS_LABEL_WMI_MAP[label.downcase.to_sym]
+    label_value = nil
+    WMI.execquery("SELECT Index FROM Win32_NetworkAdapter WHERE NetConnectionID = '#{interface}'").each do |nic|
+      WMI.execquery("SELECT #{wmi_value} FROM Win32_NetworkAdapterConfiguration WHERE Index = #{nic.Index}").each do |nic_config|
+        case label.downcase.to_sym
+        when :ipaddress
+          nic_config.IPAddress.any? do |addr|
+            label_value = addr if valid_ipv4_address?(addr)
+            label_value
+          end
+        when :ipaddress6
+          nic_config.IPAddress.any? do |addr|
+            label_value = addr if IP::Windows.valid_ipv6_address?(addr)
+            label_value
+          end
+        when :netmask
+          nic_config.IPSubnet.any? do |addr|
+            label_value = addr if IP::Windows.valid_ipv4_address?(addr)
+            label_value
+          end
+        when :macaddress
+          label_value = nic_config.MACAddress
+        end
+      end
+    end
+
+    label_value
+  end
+
+  # Returns an array of partial Win32_NetworkAdapterConfiguration objects.
+  #
+  # @return [Array<WIN32OLE>] objects
+  #
+  # @api private
+  def self.network_adapter_configurations
+    nics = []
+    # WIN32OLE doesn't implement Enumerable
+    WMI.execquery(WMI_IP_INFO_QUERY).each do |nic|
+      nics << nic
+    end
+    nics
+  end
+
+  # Gets a list of active IPv4 network adapter configurations sorted by the
+  # lowest IP connection metric. If two configurations have the same metric,
+  # then the IPv4 specific binding order as specified in the registry will
+  # be used.
+  #
+  # @return [Array<WIN32OLE>]
+  #
+  # @api private
+  def self.get_preferred_ipv4_adapters
+    get_preferred_network_adapters(Bindings4.new)
+  end
+
+  # Gets a list of active IPv6 network adapter configurations sorted by the
+  # lowest IP connection metric. If two configurations have the same metric,
+  # then the IPv6 specific binding order as specified in the registry will
+  # be used.
+  #
+  # @return [Array<WIN32OLE>]
+  #
+  # @api private
+  def self.get_preferred_ipv6_adapters
+    get_preferred_network_adapters(Bindings6.new)
+  end
+
+  # Gets a list of active network adapter configurations sorted by the lowest
+  # IP connection metric. If two configurations have the same metric, then
+  # the adapter binding order as specified in the registry will be used.
+  # Note the order may different for IPv4 vs IPv6 addresses.
+  #
+  # @see http://support.microsoft.com/kb/894564
+  # @return [Array<WIN32OLE>]
+  #
+  # @api private
+  def self.get_preferred_network_adapters(bindings)
+    network_adapter_configurations.select do |nic|
+      bindings.bindings.include?(nic.SettingID)
+    end.sort do |nic_left,nic_right|
+      cmp = nic_left.IPConnectionMetric <=> nic_right.IPConnectionMetric
+      if cmp == 0
+        bindings.bindings[nic_left.SettingID] <=> bindings.bindings[nic_right.SettingID]
+      else
+        cmp
+      end
+    end
+  end
+
+  class Bindings4
+    def initialize
+      @key = 'SYSTEM\CurrentControlSet\Services\Tcpip\Linkage'
+    end
+
+    def bindings
+      require 'vmopt/utils/registry'
+      bindings = {}
+
+      Registry.hklm_read(@key, 'Bind').each_with_index do |entry, index|
+        match_data = entry.match(/\\Device\\(\{.*\})/)
+        unless match_data.nil?
+          bindings[match_data[1]] = index
+        end
+      end
+
+      bindings
+    rescue
+      {}
+    end
+  end
+
+  class Bindings6 < Bindings4
+    def initialize
+      @key = 'SYSTEM\CurrentControlSet\Services\Tcpip6\Linkage'
+    end
+  end
+
+  # Determines if the value passed in is a valid ipv4 address.
+  #
+  # @param [String] ip_address the IPv4 address to validate
+  # @return [Boolean]
+  #
+  # @api private
+  def self.valid_ipv4_address?(ip_address)
+    String(ip_address).scan(/(?:[0-9]{1,3}\.){3}[0-9]{1,3}/).each do |match|
+      # excluding 169.254.x.x in Windows - this is the DHCP APIPA
+      #  meaning that if the node cannot get an ip address from the dhcp server,
+      #  it auto-assigns a private ip address
+      unless match == "127.0.0.1" or match =~ /^169.254.*/
+        return !!match
+      end
+    end
+
+    false
+  end
+
+  # Determines if the value passed in is a valid ipv6 address.
+  #
+  # @param [String] ip_address the IPv6 address to validate
+  # @return [Boolean]
+  #
+  # @api private
+  def self.valid_ipv6_address?(ip_address)
+    String(ip_address).scan(/(?>[0-9,a-f,A-F]*\:{1,2})+[0-9,a-f,A-F]{0,4}/).each do |match|
+      unless match =~ /fe80.*/ or match == "::1"
+        return !!match
+      end
+    end
+
+    false
+  end
+
+end
+
+
+if __FILE__ == $0
+   
+   #查询网络接口名称
+   WinNet.interfaces.each{|i| puts i}
+   
+   #查询网络接口mac地址
+   WinNet.interfaces.each do |inf|
+     puts WinNet.value_for_interface_and_label(inf, :macaddress)
+   end
+   
+   #查询网络接口的属性，分别是描述，服务名，设置id
+   WinNet.network_adapter_configurations.each{|interface| puts  interface.description}
+   WinNet.network_adapter_configurations.each{|interface| puts interface.serviceName}
+   WinNet.network_adapter_configurations.each{|interface| puts interface.settingID}
+   
+   #查询网卡的连接状态
+   WinNet.interfaces.each do|i| 
+     puts "#{i}: "+WinNet.netconnstatus[i]
+   end
+   
+   #打印设备的PNPDeviceID，可用来提交给devcon.exe工具
+   WinNet.network_adapter.each{|interface| puts "#{interface.NetConnectionId} : " + "#{interface.pNPDeviceID}"}
+
+end
+
+
+
+

data/lib/vmopt/windows/win_winutils.rb

@@ -0,0 +1,200 @@
+# encoding: utf-8
+#功能：提供windows窗口的基本操作
+
+=begin rdoc
+类名: 通用操作类
+描述: 封装系统操作调用相关的内容
+=end
+require "win32ole"
+WIN32OLE.codepage = WIN32OLE::CP_UTF8
+require "Win32API"
+require "rautomation"
+require "vmopt/ext/string_ext"
+
+module WinUtils
+
+  class KillProcessFailError < RuntimeError; end  #杀死进程失败的错误
+  class EmptyFileFailError < RuntimeError; end  #清除目录错误
+  class NotFindZipError < RuntimeError; end   #查不到对应的ZIP文件
+  class NotFindFileError < RuntimeError; end   #查不到对应文件
+  class NotSuitableStringError < RuntimeError; end #未按要求的的字符串
+  class NotFindWindowError < RuntimeError; end #查找不到窗口
+  class ActivateFailError < RuntimeError; end  #激活窗口失败
+  class WaitConnectTimeoutError < RuntimeError; end #TCP连接失败异常
+  class NoSavePathError < RuntimeError;end #没有路径可保存
+
+=begin rdoc
+  参数：无
+  作用：检测到除了为简体中文时为中文系统，其余都判断为英文系统.
+  返回值：返回国家域名缩写的小写形式
+=end      
+  def self.os_type
+    WIN32OLE.connect('winmgmts:\\\.').ExecQuery("select * from Win32_OperatingSystem").each do |m|
+      return "cn" if m.OSLanguage == 4 || m.OSLanguage == 2052
+    end
+    "en"
+  end 
+
+=begin rdoc
+  参数:titlename:查找的标题名称,activate_flag = false
+  作用：查找是否有给定标题名称的惟一对象,activate_flag表示是否能够激活窗口。
+  返回值：找到惟一对象则返回true;找不到或找到为非惟一的则返回false
+=end 
+  def self.find_single_active_window?(titlename,activate_flag = false)
+    find_res=0
+    all_rautowindow = RAutomation::Window.windows
+    all_rautowindow.each do  |win|
+      find_res+=1 if  (win.exists? && win.title !=nil && (win.title.include?(titlename) rescue false)) 
+    end
+    return false if find_res.to_s != "1" #找到了有多个标题框的内容，直接返回fasle
+    rautowindow = RAutomation::Window.new(:title=>titlename,:adapter=>"Autoit")
+    if activate_flag
+      return false unless rautowindow.exists? #不存在为假;
+      rautowindow.WinActivate(rautowindow.title) unless rautowindow.WinActive(rautowindow.title)
+      return true if res_status.to_s == "15" 
+      return false
+    else
+      rautowindow.exists?
+    end
+  end 
+
+=begin rdoc
+  参数:要搜索的标题名称和激活窗口标志，激活窗口标志默认为true
+  作用：查找是否有查找到给定的标题名称的对话框，如果有则返回RAutomation对
+  象，否则返回nil
+   返回值：成功找到返回RAutomation的Windows对象
+=end 
+  def self.find_window(titlename,flag = true)
+    retrytime=0
+    begin
+      rautowindow = RAutomation::Window.new(:title=>titlename,:adapter=>"Autoit")
+      retrytime+=1  
+      sleep 1
+    end while !rautowindow.exists? and retrytime <= 10 
+    raise ::WinUtils::NotFindWindowError,"Not found the windows like #{titlename}." if !rautowindow.exists?
+    if flag 
+      res = rautowindow.WinActivate(rautowindow.title) unless rautowindow.WinActive(rautowindow.title)
+      raise ::WinUtils::ActivateFailError,"Cant't Activate the Window #{rautowindow.title}" if res.to_s == "0"
+    end
+    rautowindow
+  end 
+
+
+
+=begin rdoc
+  参数:根据窗口的text来定位title
+  作用：查找是否有查找到给定的窗口的text的对话框，如果有则返回RAutomation对
+  象，否则的返回false
+  返回值：成功找到返回RAutomation的Windows对象
+=end 
+  def self.find_window_by_text(textname)
+    rautowindows = RAutomation::Window.windows
+    rautowindows.map do |win|
+      return win if win.text.include?(textname)
+    end
+    nil
+  end
+  
+=begin rdoc 
+  参数：process_name：进程名;reserve_process_path:期望路径
+  作用：杀掉给定名称的进程，reserve_process_path有值则保留这个环境下的进程
+  返回值：若异常则抛出异常，否则返回真  
+=end  
+  def self.kill_process(process_name,reserve_process_path=nil)
+    raise ArgumentError,"the argument can't be nil." if process_name.nil?
+    begin
+      gbk_reserve_process_path = reserve_process_path.to_gbk unless reserve_process_path.nil?
+      WIN32OLE.connect('winmgmts:\\\.').ExecQuery("SELECT * FROM Win32_Process").each do |item|
+        if item.Caption == process_name.to_gbk
+          if gbk_reserve_process_path == nil
+            item.Terminate
+            next
+          end
+          next if item.ExecutablePath != nil && item.ExecutablePath.gsub("\\","/").downcase == gbk_reserve_process_path.gsub("\\","/").downcase
+          item.Terminate
+        end
+      end
+      return true
+    rescue =>err
+      raise ::WinUtils::KillProcessFailError,"Kill process: #{process_name.to_utf8} failed!err_msg:#{err}"
+    end
+  end
+
+=begin rdoc 
+  参数：process_name：进程名
+  作用：查找给定名称的进程，
+  返回值：查看是否有找到给定文件名的进程，若有则返回真，若无则返回假 
+=end  
+  def self.find_process?(process_name)
+    raise ArgumentError,"the argument can't be nil." if process_name.nil?
+    gbk_process_name = process_name.to_gbk
+    WIN32OLE.connect('winmgmts:\\\.').ExecQuery("SELECT * FROM Win32_Process").each do |item|
+      return true if item.Caption.downcase == gbk_process_name.downcase 
+    end
+    false
+ end
+  
+=begin rdoc
+  参数:准备被清空的目录dir或文件
+  作用：清空目录下的所有文件;但保留dir目录,若dir为文件就会删除
+  返回值：若异常则抛出异常，否则返回真
+=end 
+  def self.empty_dir(dir)
+    raise ArgumentError,"the argument can't be nil." if dir.nil?
+    begin
+      gbk_dest_dir = dir.to_gbk      
+      Find.find(gbk_dest_dir) do |file|
+        next if ! File.exist?(file)
+        #next if File.directory?(file) && (file.downcase == gbk_dest_dir.downcase)
+        if File.directory?(file)
+          FileWinUtils.rm_rf(file)
+        else
+          File.delete(file)
+        end
+      end
+      return true
+    rescue =>err  
+      raise ::WinUtils::EmptyFileFailError,"empty_dir #{dir.to_utf8} failed!err_msg:#{err}."
+    end
+  end   
+
+=begin rdoc
+  参数:给定的目录，需要查找的文件名
+  作用：在给定的目录中查找给定的文件，若查找到时返回这个文件的目路径，否则返回空
+  返回值：返回文件所在路径或为空
+=end 
+  def self.get_filepath(dest_dir,filename)
+    raise ArgumentError,"the argument can't be nil." if dest_dir.nil? || filename.nil?  
+    gbk_dest_dir = dest_dir.to_gbk
+    gbk_filename = filename.to_gbk
+    Find.find(gbk_dest_dir) do |filepath|
+      filepath_arr = filepath.split("/")
+      return (Pathname.new(File.expand_path(filepath)).realpath).to_s.gsub("/","\\\\") if (filepath_arr[-1].downcase == gbk_filename.downcase && !File.directory?(filepath) )
+    end
+    raise ::WinUtils::NotFindFileError,"Not find named :#{filename.to_utf8} file in the dir:#{dest_dir.to_utf8}."
+  end     
+=begin rdoc
+  参数:两个字符串str1,str2
+  作用:去除str1中头部有str2的部分，需要完整匹配
+  返回值：被去除后的字符串
+=end 
+  def self.remove_head_str(str1,str2)
+    str1_size = str1.size
+    str2_size = str2.size
+    raise ::WinUtils::NotSuitableStringError,"the length of str1 need > str2's size." if str1_size<str2_size
+    return "" if str1 == str2
+    i = 1
+    str2_arr_size = str2.split("\n").size
+    result_str=""
+    str1.each_line do |line|
+      if i<=str2_arr_size
+        raise ::WinUtils::NotSuitableStringError,"The relation of str1:#{str1.dump} and str2:#{str2.dump} is not suitable." unless str2.include?(line)
+        i = i+1
+        next
+      end
+      result_str<<line
+    end
+    result_str.strip
+  end     
+  
+end #end of WinUtils.