vmopt 0.0.2

data/lib/vmopt/power_operation.rb

@@ -0,0 +1,57 @@
+#encoding: utf-8
+module Vmopt
+class PowerOperation
+
+=begin
+参数：无
+功能：关机
+返回值：默认
+=end
+	def shutdown
+		system("shutdown -s -t 00")
+	end
+
+=begin
+参数：无
+功能：重启
+返回值：默认
+=end
+	def reboot
+		system("shutdown -r -t 00")
+	end
+
+
+=begin
+参数：无
+功能：注销
+返回值：默认
+=end
+	def logoff
+		system("shutdown -l")
+	end
+
+
+=begin
+参数：无
+功能：锁定或切换用户
+返回值：默认
+=end
+	def lock_user
+		system("rundll32.exe user32.dll LockWorkStation")
+	end
+
+=begin
+参数：无
+功能：休眠，xp没有休眠的功能，win7有休眠的功能
+返回值：默认
+=end
+	def sleep
+		system("rundll32.exe powrprof.dll SetSuspendState")
+	end
+
+end
+end #end of module vmopt
+
+if __FILE__ == $0
+	PowerOperation.new().lock_user
+end

data/lib/vmopt/serialport_operation.rb

@@ -0,0 +1,62 @@
+#encoding: utf-8
+require "vmopt/utils/wmi"
+require "serialport"
+
+module Vmopt
+class SerialPortOperation
+
+=begin
+参数：无
+功能：查询串口的基本信息
+返回值：默认
+=end
+	def get_serial_port	
+	    data_value={}	
+		colItems = WMI.execquery ("select * from Win32_SerialPort")
+		for objItem in colItems do
+			name = (objItem.Name)[-5,4]
+			str ={"串口名称" => name,
+			      "状态" => objItem.Status
+			     }			
+			data_value["#{name}"] = str
+		end
+		return data_value
+	end
+=begin
+参数：串口号，写入的字符串
+功能：将串口写入字符串
+返回值：默认
+=end
+	def write(strcom,strinput)
+		strinput
+		begin
+		File.open(strcom, 'w+') do |file|
+ 		file.write(strinput)
+ 		end
+		rescue Exception
+			return false
+		end
+		return true
+	end
+=begin
+参数：串口号
+功能：从串口读入字符串
+返回值：读到的com口字符串
+=end
+	def read(strcom)
+		begin
+		stroutput=""
+ 		sp = SerialPort.new "#{strcom}", 9600
+ 	    sp.read_timeout=4000 #定时4秒
+ 	    stroutput = sp.read(50)
+ 	    if stroutput.empty?
+ 	    	return false
+ 	    end
+		rescue Exception
+			return false
+		end
+		return stroutput
+	end
+end
+end #end of module vmopt
+

data/lib/vmopt/system_resource.rb

@@ -0,0 +1,44 @@
+#encoding: utf-8
+#$:.unshift File.join(__FILE__,"..","..")
+require "vmopt/utils/wmi"
+
+module Vmopt
+class SystemResource
+
+	def get_cpu
+		data_value={}
+		colItems = WMI.execquery ("select * from Win32_Processor")
+		for objItem in colItems do
+			str={"CPU数量" => objItem.NumberOfCores,
+			     "CPU主频" => (objItem.MaxClockSpeed/1000.0).round(2),
+			     "CPU使用率" => "#{objItem.LoadPercentage}%"
+		    }
+        	data_value["#{objItem.DeviceID}"]=str
+		end
+		return data_value
+	end
+
+	def get_memory
+		data_value={}
+		colItems = WMI.execquery ("select * from Win32_PhysicalMemory")
+		for objItem in colItems do
+			memsizestr = objItem.Capacity
+			memsizei = memsizestr.to_i/1024/1024  #str转interger		
+		end
+		availMemorys = WMI.execquery ("select * from Win32_PerfRawData_PerfOS_Memory")
+		for availMemory in availMemorys do 
+			availstr = availMemory.AvailableMBytes
+			availi = availstr.to_i    #str转interger
+		end
+		
+		str = {"总物理内存" => memsizei,  
+		 	 "可用内存" => availi,
+			 "内存使用率" => "#{(((memsizei-availi) * 1.0 /memsizei) * 100).round(1)}%"
+    	}
+		data_value["#{objItem.Name}"] = str
+		return data_value
+	end
+
+end
+end #end of module vmopt
+

data/lib/vmopt/utils/ip.rb

@@ -0,0 +1,287 @@
+#encoding: utf-8
+# 功能：提供IP的规范性检查，适合所有平台
+
+module IP
+  # A map of all the different regexes that work for
+  # a given platform or set of platforms.
+  REGEX_MAP = {
+    :linux => {
+      :ipaddress  => /inet (?:addr:)?([0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)/,
+      :ipaddress6 => /inet6 (?:addr: )?((?![fe80|::1])(?>[0-9,a-f,A-F]*\:{1,2})+[0-9,a-f,A-F]{0,4})/,
+      :macaddress => /(?:ether|HWaddr)\s+((\w{1,2}:){5,}\w{1,2})/,
+      :netmask  => /(?:Mask:|netmask )([0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)/,
+      :mtu  => /MTU:(\d+)/
+    },
+    :bsd   => {
+      :aliases  => [:openbsd, :netbsd, :freebsd, :darwin, :"gnu/kfreebsd", :dragonfly],
+      :ipaddress  => /inet\s+([0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)/,
+      :ipaddress6 => /inet6 ((?![fe80|::1])(?>[0-9,a-f,A-F]*\:{1,2})+[0-9,a-f,A-F]{0,4})/,
+      :macaddress => /(?:ether|lladdr)\s+(\w?\w:\w?\w:\w?\w:\w?\w:\w?\w:\w?\w)/,
+      :netmask  => /netmask\s+0x(\w{8})/,
+      :mtu => /mtu\s+(\d+)/
+    },
+    :sunos => {
+      :ipaddress  => /inet\s+([0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)/,
+      :ipaddress6 => /inet6 ((?![fe80|::1])(?>[0-9,a-f,A-F]*\:{1,2})+[0-9,a-f,A-F]{0,4})/,
+      :macaddress => /(?:ether|lladdr)\s+(\w?\w:\w?\w:\w?\w:\w?\w:\w?\w:\w?\w)/,
+      :netmask  => /netmask\s+(\w{8})/,
+      :mtu => /mtu\s+(\d+)/
+    },
+    :"hp-ux" => {
+      :ipaddress  => /\s+inet (\S+)\s.*/,
+      :macaddress => /(\w{1,2}:\w{1,2}:\w{1,2}:\w{1,2}:\w{1,2}:\w{1,2})/,
+      :netmask  => /.*\s+netmask (\S+)\s.*/
+    },
+    :windows => {}
+  }
+
+  # Convert an interface name into purely alphanumeric characters.
+  def self.alphafy(interface)
+    interface.gsub(/[^a-z0-9_]/i, '_')
+  end
+
+  def self.convert_from_hex?(kernel)
+    kernels_to_convert = [:sunos, :openbsd, :netbsd, :freebsd, :darwin, :"hp-ux", :"gnu/kfreebsd", :dragonfly]
+    kernels_to_convert.include?(kernel)
+  end
+
+  def self.supported_platforms
+    REGEX_MAP.inject([]) do |result, tmp|
+      key, map = tmp
+      if map[:aliases]
+        result += map[:aliases]
+      else
+        result << key
+      end
+      result
+    end
+  end
+
+  def self.get_interfaces
+    if Facter.value(:kernel) == 'windows'
+      require 'facter/util/ip/windows'
+      return Vmopt::IP::Windows.interfaces
+    end
+
+    return [] unless output = Vmopt::IP.get_all_interface_output()
+
+    # Our regex appears to be stupid, in that it leaves colons sitting
+    # at the end of interfaces.  So, we have to trim those trailing
+    # characters.  I tried making the regex better but supporting all
+    # platforms with a single regex is probably a bit too much.
+    output.scan(/^\S+/).collect { |i| i.sub(/:$/, '') }.uniq
+  end
+
+  def self.get_all_interface_output
+    case Facter.value(:kernel)
+    when 'Linux', 'OpenBSD', 'NetBSD', 'FreeBSD', 'Darwin', 'GNU/kFreeBSD', 'DragonFly'
+      output = Vmopt::IP.exec_ifconfig(["-a","2>/dev/null"])
+    when 'SunOS'
+      output = Vmopt::IP.exec_ifconfig(["-a"])
+    when 'HP-UX'
+      # (#17487)[https://projects.puppetlabs.com/issues/17487]
+      # Handle NIC bonding where asterisks and virtual NICs are printed.
+      if output = hpux_netstat_in
+        output.gsub!(/\*/, "")                  # delete asterisks.
+        output.gsub!(/^[^\n]*none[^\n]*\n/, "") # delete lines with 'none' instead of IPs.
+        output.sub!(/^[^\n]*\n/, "")            # delete the header line.
+        output
+      end
+    end
+    output
+  end
+
+
+  ##
+  # exec_ifconfig uses the ifconfig command
+  #
+  # @return [String] the output of `ifconfig #{arguments} 2>/dev/null` or nil
+  def self.exec_ifconfig(additional_arguments=[])
+    Facter::Core::Execution.exec("#{self.get_ifconfig} #{additional_arguments.join(' ')}")
+  end
+  ##
+  # get_ifconfig looks up the ifconfig binary
+  #
+  # @return [String] path to the ifconfig binary
+  def self.get_ifconfig
+    common_paths=["/bin/ifconfig","/sbin/ifconfig","/usr/sbin/ifconfig"]
+    common_paths.select{|path| File.executable?(path)}.first
+  end
+  ##
+  # hpux_netstat_in is a delegate method that allows us to stub netstat -in
+  # without stubbing exec.
+  def self.hpux_netstat_in
+    Facter::Core::Execution.exec("/bin/netstat -in")
+  end
+
+  def self.get_infiniband_macaddress(interface)
+    if File.exists?("/sys/class/net/#{interface}/address") then
+      ib_mac_address = `cat /sys/class/net/#{interface}/address`.chomp
+    elsif File.exists?("/sbin/ip") then
+      ip_output = %x{/sbin/ip link show #{interface}}
+      ib_mac_address = ip_output.scan(%r{infiniband\s+((\w{1,2}:){5,}\w{1,2})})
+    else
+      ib_mac_address = "FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF:FF"
+      Facter.debug("ip.rb: nothing under /sys/class/net/#{interface}/address and /sbin/ip not available")
+    end
+    ib_mac_address
+  end
+
+  def self.ifconfig_interface(interface)
+    output = Vmopt::IP.exec_ifconfig([interface,"2>/dev/null"])
+  end
+
+  def self.get_single_interface_output(interface)
+    output = ""
+    case Facter.value(:kernel)
+    when 'OpenBSD', 'NetBSD', 'FreeBSD', 'Darwin', 'GNU/kFreeBSD', 'DragonFly'
+      output = Vmopt::IP.ifconfig_interface(interface)
+    when 'Linux'
+      ifconfig_output = Vmopt::IP.ifconfig_interface(interface)
+      if interface =~ /^ib/ then
+        real_mac_address = get_infiniband_macaddress(interface)
+        output = ifconfig_output.sub(%r{(?:ether|HWaddr)\s+((\w{1,2}:){5,}\w{1,2})}, "HWaddr #{real_mac_address}")
+      else
+        output = ifconfig_output
+      end
+    when 'SunOS'
+      output = Vmopt::IP.exec_ifconfig([interface])
+    when 'HP-UX'
+       mac = ""
+       ifc = hpux_ifconfig_interface(interface)
+       hpux_lanscan.scan(/(\dx\S+).*UP\s+(\w+\d+)/).each {|i| mac = i[0] if i.include?(interface) }
+       mac = mac.sub(/0x(\S+)/,'\1').scan(/../).join(":")
+       output = ifc + "\n" + mac
+    end
+    output
+  end
+
+  def self.hpux_ifconfig_interface(interface)
+    Vmopt::IP.exec_ifconfig([interface])
+  end
+
+  def self.hpux_lanscan
+    Facter::Core::Execution.exec("/usr/sbin/lanscan")
+  end
+
+  def self.get_output_for_interface_and_label(interface, label)
+    return get_single_interface_output(interface) unless Facter.value(:kernel) == 'windows'
+
+    require 'facter/util/ip/windows'
+    output = Vmopt::IP::Windows.value_for_interface_and_label(interface, label)
+    output ? output : ""
+  end
+
+  def self.get_bonding_master(interface)
+    if Facter.value(:kernel) != 'Linux'
+      return nil
+    end
+    # We need ip instead of ifconfig because it will show us
+    # the bonding master device.
+    if not FileTest.executable?("/sbin/ip")
+      return nil
+    end
+    # A bonding interface can never be an alias interface. Alias
+    # interfaces do have a colon in their name and the ip link show
+    # command throws an error message when we pass it an alias
+    # interface.
+    if interface =~ /:/
+      return nil
+    end
+    regex = /SLAVE[,>].* (bond[0-9]+)/
+      ethbond = regex.match(%x{/sbin/ip link show #{interface}})
+    if ethbond
+      device = ethbond[1]
+    else
+      device = nil
+    end
+    device
+  end
+
+  ##
+  # get_interface_value obtains the value of a specific attribute of a specific
+  # interface.
+  #
+  # @param interface [String] the interface identifier, e.g. "eth0" or "bond0"
+  #
+  # @param label [String] the attribute of the interface to obtain a value for,
+  # e.g. "netmask" or "ipaddress"
+  #
+  # @api private
+  #
+  # @return [String] representing the requested value.  An empty array is
+  # returned if the kernel is not supported by the REGEX_MAP constant.
+  def self.get_interface_value(interface, label)
+    if Facter.value(:kernel) == 'windows'
+      require 'facter/util/ip/windows'
+      return Vmopt::IP::Windows.value_for_interface_and_label(interface, label)
+    end
+
+    tmp1 = []
+
+    kernel = Facter.value(:kernel).downcase.to_sym
+
+    # If it's not directly in the map or aliased in the map, then we don't know how to deal with it.
+    unless map = REGEX_MAP[kernel] || REGEX_MAP.values.find { |tmp| tmp[:aliases] and tmp[:aliases].include?(kernel) }
+      return []
+    end
+
+    # Pull the correct regex out of the map.
+    regex = map[label.to_sym]
+
+    # Linux changes the MAC address reported via ifconfig when an ethernet interface
+    # becomes a slave of a bonding device to the master MAC address.
+    # We have to dig a bit to get the original/real MAC address of the interface.
+    bonddev = get_bonding_master(interface)
+    if label == 'macaddress' and bonddev
+      bondinfo = read_proc_net_bonding("/proc/net/bonding/#{bonddev}")
+      re = /^Slave Interface: #{interface}\b.*?\bPermanent HW addr: (([0-9A-F]{2}:?)*)$/im
+      if match = re.match(bondinfo)
+        value = match[1].upcase
+      end
+    else
+      output_int = get_output_for_interface_and_label(interface, label)
+
+      output_int.each_line do |s|
+        if s =~ regex
+          value = $1
+            if label == 'netmask' && convert_from_hex?(kernel)
+              value = value.scan(/../).collect do |byte| byte.to_i(16) end.join('.')
+            end
+          tmp1.push(value)
+        end
+      end
+
+      if tmp1
+        value = tmp1.shift
+      end
+    end
+  end
+
+  ##
+  # read_proc_net_bonding is a seam method for mocking purposes.
+  #
+  # @param path [String] representing the path to read, e.g. "/proc/net/bonding/bond0"
+  #
+  # @api private
+  #
+  # @return [String] modeling the raw file read
+  def self.read_proc_net_bonding(path)
+    File.read(path) if File.exists?(path)
+  end
+  private_class_method :read_proc_net_bonding
+
+  def self.get_network_value(interface)
+    require 'ipaddr'
+
+    ipaddress = get_interface_value(interface, "ipaddress")
+    netmask = get_interface_value(interface, "netmask")
+
+    if ipaddress && netmask
+      ip = IPAddr.new(ipaddress, Socket::AF_INET)
+      subnet = IPAddr.new(netmask, Socket::AF_INET)
+      network = ip.mask(subnet.to_s).to_s
+    end
+  end
+end
+