vrvirtualdesktop 0.0.1

data/lib/Win32/NetResources.rb

@@ -0,0 +1,160 @@
+##################################################################################################
+#
+# $Author: Steve $
+# $Date: 2005/12/16 18:35:42 $
+# $Version: 0.99 $
+# $Change: $
+#
+##################################################################################################
+
+#-------------------------------------------------------------------------------------------------
+# This module mounts SAMBA (Win32) network resources
+module Win32
+
+	module NetResources
+
+		require 'win32ole'
+
+		#-----------------------------------------------------------------------------------------
+		#-----------------------------------------------------------------------------------------
+		# Exceptions
+		class Error < StandardError
+			#-------------------------------------------------------------------------------------
+			# attributesFor: "description"
+			attr_reader :errno, :error, :OLEObject, :OLEMethod
+
+			#-------------------------------------------------------------------------------------
+			# class methodsFor: "initialization"
+			def initialize(p_e)
+				bInOLE = false
+				@errno   = 0
+				@error = ''
+				p_e.to_s.each do |line|
+					case line.chomp
+						when /^\s*OLE error code:([0-9a-hA-H]+) in (\w+)\.(\w+)/
+							@errno = $1.hex
+							@OLEObject = $2
+							@OLEMethod = $3
+							bInOLE = true
+						when /^\s+(.*)/ then @error += $1 + "\n" if bInOLE
+						when /^$/       then bInOLE = false
+					end
+				end
+				@error = @error.chomp
+				super("0x%X" % @errno + ": " + @error)
+			end
+
+			#-------------------------------------------------------------------------------------
+			# methodsFor: "accessing"
+			def to_i
+				return @errno
+			end
+		end
+
+		#-----------------------------------------------------------------------------------------
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "mounting/unmounting"
+		def NetResources.mount(
+				p_strLocal,
+				p_strRemote,
+				p_strUser = nil,
+				p_strPwd  = nil,
+				p_bUpdate = false)		# true if we want to update the local user's profile
+			begin
+				wsh = WIN32OLE.new('WScript.Network')
+				wsh.MapNetworkDrive(p_strLocal.gsub(/\w$/, '\0:'), p_strRemote.gsub(/\//, "\\"),
+									p_bUpdate, p_strUser, p_strPwd)
+				rescue WIN32OLERuntimeError => p_e
+					raise Error, p_e
+			end
+		end
+
+		def NetResources.umount(p_strName, p_bForce = true, p_bUpdate = false)
+			begin
+				wsh = WIN32OLE.new('WScript.Network')
+				wsh.RemoveNetworkDrive(p_strName, p_bForce, p_bUpdate)
+				rescue WIN32OLERuntimeError => p_e
+					raise Error, p_e
+			end
+		end
+
+		def NetResources.mountPrinter(
+				p_strLocal,
+				p_strRemote,
+				p_strUser = nil,
+				p_strPwd  = nil,
+				p_bUpdate = false)		# true if we want to update the local user's profile
+			begin
+				wsh = WIN32OLE.new('WScript.Network')
+				wsh.AddPrinterConnection(p_strLocal, p_strRemote.gsub(/\//, "\\"),
+									p_bUpdate, p_strUser, p_strPwd)
+				rescue WIN32OLERuntimeError => p_e
+					raise Error, p_e
+			end
+		end
+
+		def NetResources.umountPrinter(p_strName, p_bForce = true, p_bUpdate = false)
+			begin
+				wsh = WIN32OLE.new('WScript.Network')
+				wsh.RemovePrinterConnection(p_strName, p_bForce, p_bUpdate)
+				rescue WIN32OLERuntimeError => p_e
+					raise Error, p_e
+			end
+		end
+
+		#-----------------------------------------------------------------------------------------
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "defaulting"
+		def NetResources.setDefaultPrinter(p_strName)
+			begin
+				wsh = WIN32OLE.new('WScript.Network')
+				wsh.SetDefaultPrinter(p_strName)
+				rescue WIN32OLERuntimeError => p_e
+					raise Error, p_e
+			end
+		end
+
+		#-----------------------------------------------------------------------------------------
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "enumerating"
+		def NetResources.mounts
+			begin
+				wsh     = WIN32OLE.new('WScript.Network')
+				vValues = wsh.EnumNetworkDrives()
+				mRet    = {}
+				key     = nil
+				vValues.each do |item|
+					if key.nil? then key = item
+					else
+						mRet[key] = item
+						key = nil
+					end
+				end
+				rescue WIN32OLERuntimeError
+					raise Error, p_e
+				ensure
+					return mRet
+			end
+		end
+
+		def NetResources.printers
+			begin
+				wsh     = WIN32OLE.new('WScript.Network')
+				vValues = wsh.EnumPrinterConnections()
+				mRet    = {}
+				key     = nil
+				vValues.each do |item|
+					if key.nil? then key = item
+					else
+						mRet[key] = item
+						key = nil
+					end
+				end
+				rescue WIN32OLERuntimeError
+					raise Error, p_e
+				ensure
+					return mRet
+			end
+		end
+	end
+end

data/lib/Win32/Process.rb

@@ -0,0 +1,468 @@
+##################################################################################################
+#
+# $Author: Steve $
+# $Date: 2005/12/16 18:35:42 $
+# $Version: 0.99 $
+# $Change: $
+#
+##################################################################################################
+require "Win32/APIMapper"
+
+module Win32
+	require 'Win32API'
+
+	#---------------------------------------------------------------------------------------------
+	#---------------------------------------------------------------------------------------------
+	# This class contains the process info created by Win32::Process
+	class ProcessInfo
+		#-----------------------------------------------------------------------------------------
+		# attributesFor: "process handles"
+		attr_reader :hProcess, :hThread, :pid, :tid # Thread Id
+		attr_writer :hProcess, :hThread, :pid, :tid # Thread Id
+
+		#-----------------------------------------------------------------------------------------
+		# class methodsFor: "Data sizing"
+		def ProcessInfo.sizeof
+			return 16			# 4 longs
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# class methodsFor: "initialization"
+		def initialize(p_hProcess = nil, p_hThread = nil, p_pid = nil, p_tid = nil)
+			@hProcess = p_hProcess
+			@hThread  = p_hThread
+			@pid      = p_pid
+			@tid      = p_tid
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# class methodsFor: "unpacking data"
+		def ProcessInfo.unpack(p_pData)
+			pData = p_pData.unpack("L4")
+			return ProcessInfo.new(pData[0], pData[1], pData[2], pData[3])
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "packing data"
+		def pack
+			return [@hProcess, @hThread, @pid, @tid].pack("L*")
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "printing"
+		def to_s
+			return "hProcess: #{@hProcess}, hThread: #{@hThread}, pid: #{@pid}, thread id: #{@tid}"
+		end
+	end
+
+	#---------------------------------------------------------------------------------------------
+	#---------------------------------------------------------------------------------------------
+	# This class contains the startup info that is passed to Win32::Process.Create
+	# It tells how the process should be created and where it should appear
+	class StartupInfo
+		#-----------------------------------------------------------------------------------------
+		# attributesFor: "window properties"
+		attr_reader :desktop, :title, :x, :y, :width, :length, :xCountChars, :yCountChars,
+					:fillAttribute, :showWindow, :stdin, :stdout, :stderr
+
+		#-----------------------------------------------------------------------------------------
+		# constsFor: "Startup Information Flags"
+		# (from WinBase.h)
+		STARTF_USESHOWWINDOW				= 0x00000001
+		STARTF_USESIZE						= 0x00000002
+		STARTF_USEPOSITION					= 0x00000004
+		STARTF_USECOUNTCHARS				= 0x00000008
+		STARTF_USEFILLATTRIBUTE				= 0x00000010
+		STARTF_RUNFULLSCREEN				= 0x00000020
+		STARTF_FORCEONFEEDBACK				= 0x00000040
+		STARTF_FORCEOFFFEEDBACK				= 0x00000080
+		STARTF_USESTDHANDLES				= 0x00000100
+		STARTF_USEHOTKEY					= 0x00000200
+		FOREGROUND_BLUE						= 0x0001
+		FOREGROUND_GREEN					= 0x0002
+		FOREGROUND_RED						= 0x0004
+		FOREGROUND_INTENSITY				= 0x0008
+		BACKGROUND_BLUE						= 0x0010
+		BACKGROUND_GREEN					= 0x0020
+		BACKGROUND_RED						= 0x0040
+		BACKGROUND_INTENSITY				= 0x0080
+
+		#-----------------------------------------------------------------------------------------
+		# constsFor: "Show Window values"
+		# (from WinUser.h)
+		SW_HIDE								=  0
+		SW_SHOWNORMAL						=  1
+		SW_NORMAL							=  1
+		SW_SHOWMINIMIZED					=  2
+		SW_SHOWMAXIMIZED					=  3
+		SW_MAXIMIZE							=  3
+		SW_SHOWNOACTIVATE					=  4
+		SW_SHOW								=  5
+		SW_MINIMIZE							=  6
+		SW_SHOWMINNOACTIVE					=  7
+		SW_SHOWNA							=  8
+		SW_RESTORE							=  9
+		SW_SHOWDEFAULT						= 10
+		SW_FORCEMINIMIZE					= 11
+		SW_MAX								= 11
+
+		#-----------------------------------------------------------------------------------------
+		# class methodsFor: "Data sizing"
+		def StartupInfo.sizeof
+			return 80 # 16 longs and 2 words
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# class methodsFor: "initialization"
+		def initialize(
+				p_pDesktop = nil, p_strTitle = nil,
+				p_x = nil, p_y = nil, p_width = nil, p_length = nil,
+				p_xCountChars = nil, p_yCountChars = nil,
+				p_fillAttribute = nil,
+				p_showWindow = nil,
+				p_stdin = nil, p_stdout = nil, p_stderr = nil)
+			@desktop           = p_pDesktop
+			@title             = p_strTitle
+			@flags             = 0
+			self.x             = p_x
+			self.y             = p_y
+			self.width         = p_width
+			self.length        = p_length
+			self.xCountChars   = p_xCountChars
+			self.yCountChars   = p_yCountChars
+			self.fillAttribute = p_fillAttribute
+			self.showWindow    = p_showWindow
+			self.stdin         = p_stdin
+			self.stdout        = p_stdout
+			self.stderr        = p_stderr
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# class methodsFor: "unpacking data"
+		def StartupInfo.unpack(p_pData)
+			pData = p_pData.unpack("L12S2L4")
+			return StartupInfo.new(
+						pData[ 1], pData[ 2],
+						pData[ 3], pData[ 4], pData[ 5], pData[ 6],
+						pData[ 7], pData[ 8],
+						pData[ 9],
+						pData[11],
+						pData[14], pData[15], pData[16]
+			)
+		end
+
+		def xx(val)
+			if RUBY_VERSION >= "1.8.2" then
+				if val == nil then
+					0
+				else
+					val
+				end
+			else
+				val
+			end
+		end
+		
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "packing data"
+		def pack
+			return [StartupInfo.sizeof,
+					0,
+					xx(@desktop),
+					xx(@title),
+					xx(@x), xx(@y), xx(@width), xx(@length),
+					xx(@xCountChars), xx(@yCountChars),
+					xx(@fillAttribute),
+					xx(@flags),
+					xx(@showWindow),
+					0, 0,
+					xx(@stdin), xx(@stdout), xx(@stderr)
+					].pack('L12S2L4')
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "accessing"
+		def x=(p_x)
+			condsetf(nil != (@x = p_x), STARTF_USEPOSITION)
+		end
+
+		def y=(p_y)
+			condsetf(nil != (@y = p_y), STARTF_USEPOSITION)
+		end
+
+		def width=(p_width)
+			condsetf(nil != (@width = p_width), STARTF_USESIZE)
+		end
+
+		def length=(p_length)
+			condsetf(nil != (@length = p_length), STARTF_USESIZE)
+		end
+
+		def xCountChars=(p_xCountChars)
+			condsetf(nil != (@xCountChars = p_xCountChars), STARTF_USECOUNTCHARS)
+		end
+
+		def yCountChars=(p_yCountChars)
+			condsetf(nil != (@yCountChars = p_yCountChars), STARTF_USECOUNTCHARS)
+		end
+
+		def fillAttribute=(p_fillAttribute)
+			condsetf(nil != (@fillAttribute = p_fillAttribute), STARTF_USEFILLATTRIBUTE)
+		end
+
+		def showWindow=(p_showWindow)
+			condsetf(nil != (@showWindow = p_showWindow), STARTF_USESHOWWINDOW)
+		end
+
+		def stdin=(p_stdin)
+			condsetf(nil != (@stdin = p_stdin), STARTF_USESTDHANDLES)
+		end
+
+		def stdout=(p_stdout)
+			condsetf(nil != (@stdout = p_stdout), STARTF_USESTDHANDLES)
+		end
+
+		def stderr=(p_stderr)
+			condsetf(nil != (@stderr = p_stderr), STARTF_USESTDHANDLES)
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "printing"
+		def to_s
+			str = "["
+			str += " desktop: #{@desktop}"										if nil != @desktop
+			str += " title: #{@title}"											if nil != @title
+			str += " x: #{@x} y: #{@y}"											if testf(STARTF_USEPOSITION)
+			str += " width: #{@width} length: #{@length}"						if testf(STARTF_USESIZE)
+			str += " xCountChars: #{@xCountChars} yCountChars: #{@yCountChars}"	if testf(STARTF_USECOUNTCHARS)
+			str += " fillAttribute: #{@fillAttribute}"							if testf(STARTF_USEFILLATTRIBUTE)
+			str += " showWindow: #{@showWindow}"								if testf(STARTF_USESHOWWINDOW)
+			str += " stdin: #{@stdin} stdout: #{@stdout} stderr: #{@stderr}"	if testf(STARTF_USESTDHANDLES)
+			str += " ]"
+			return str
+		end
+
+		#-----------------------------------------------------------------------------------------
+		private
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "flag conditions handling"
+		def condsetf(p_condition, p_flag)
+			p_condition ? setf(p_flag) : unsetf(p_flag)
+		end
+	
+		def setf(p_flag)
+			@flags |= p_flag
+		end
+
+		def unsetf(p_flag)
+			@flags &= ~p_flag
+		end
+
+		def testf(p_flag)
+			return p_flag == (@flags & p_flag)
+		end
+
+	end
+
+	#---------------------------------------------------------------------------------------------
+	#---------------------------------------------------------------------------------------------
+	# This class creates a Windows process and can manipulate it.
+	class Process
+
+		#-----------------------------------------------------------------------------------------
+		# constsFor: "Wait Return Codes"
+		# (from WinUser.h)
+		WAIT_INFINITE		= 0xFFFFFFFF
+		WAIT_FAILED			= 0xFFFFFFFF
+		WAIT_OBJECT_0		= 0x00000000
+		WAIT_ABANDONED		= 0x00000080
+		WAIT_IO_COMPLETION	= 0x000000C0
+		WAIT_TIMEOUT		= 0x00000102
+
+		#-----------------------------------------------------------------------------------------
+		include APIMapper
+
+		APIMapper.Initialize(
+			'CreateProcess'		=> {DLL => 'kernel32', NAME => 'CreateProcess',
+									IN => ['P'] * 4 + ['L', 'L'] + ['P'] * 4, OUT => 'L'},
+			'OpenProcess'		=> {DLL => 'kernel32', NAME => 'OpenProcess',
+									IN => ['L', 'L', 'L'], OUT => 'L'},
+			'TerminateProcess'	=> {DLL => 'kernel32', NAME => 'TerminateProcess',
+									IN => ['L', 'L'], OUT => 'L'},
+			'WaitForObject'		=> {DLL => 'kernel32', NAME => 'WaitForSingleObjectEx',
+									IN => ['L', 'L', 'L'], OUT => 'L'},
+			'WaitForInputIdle'	=> {DLL => 'kernel32', NAME => 'WaitForInputIdle',
+									IN => ['L', 'L'], OUT => 'L'}
+		)
+
+		#-----------------------------------------------------------------------------------------
+		# constsFor: "Process Creation Flags"
+		# (from WinBase.h)
+		DEBUG_PROCESS						= 0x00000001
+		DEBUG_ONLY_THIS_PROCESS				= 0x00000002
+		CREATE_SUSPENDED					= 0x00000004
+		DETACHED_PROCESS					= 0x00000008
+		CREATE_NEW_CONSOLE					= 0x00000010
+		NORMAL_PRIORITY_CLASS				= 0x00000020
+		IDLE_PRIORITY_CLASS					= 0x00000040
+		HIGH_PRIORITY_CLASS					= 0x00000080
+		REALTIME_PRIORITY_CLASS				= 0x00000100
+		CREATE_NEW_PROCESS_GROUP			= 0x00000200
+		CREATE_UNICODE_ENVIRONMENT			= 0x00000400
+		CREATE_SEPARATE_WOW_VDM				= 0x00000800
+		CREATE_SHARED_WOW_VDM				= 0x00001000
+		CREATE_FORCED_DOS					= 0x00002000
+		CREATE_DEFAULT_ERROR_MODE			= 0x04000000
+		CREATE_NO_WINDOW					= 0x08000000
+		PROFILE_USER						= 0x10000000
+		PROFILE_KERNEL						= 0x20000000
+		PROFILE_SERVER						= 0x40000000
+
+		#-----------------------------------------------------------------------------------------
+		# constsFor: "Thread Creation Flags"
+		# (from WinBase.h, WinNT.h)
+		THREAD_BASE_PRIORITY_LOWRT			= 15
+		THREAD_BASE_PRIORITY_MAX			= 2
+		THREAD_BASE_PRIORITY_MIN			= -2
+		THREAD_BASE_PRIORITY_IDLE			= -15
+		MAXLONG								= 0x7FFFFFFFFFFFFFFF
+		THREAD_PRIORITY_LOWEST				= THREAD_BASE_PRIORITY_MIN
+		THREAD_PRIORITY_BELOW_NORMAL		= THREAD_PRIORITY_LOWEST + 1
+		THREAD_PRIORITY_NORMAL				= 0
+		THREAD_PRIORITY_HIGHEST				= THREAD_BASE_PRIORITY_MAX
+		THREAD_PRIORITY_ABOVE_NORMAL		= THREAD_PRIORITY_HIGHEST - 1
+		THREAD_PRIORITY_ERROR_RETURN		= MAXLONG
+		THREAD_PRIORITY_TIME_CRITICAL		= THREAD_BASE_PRIORITY_LOWRT
+		THREAD_PRIORITY_IDLE				= THREAD_BASE_PRIORITY_IDLE
+
+		attr_reader :proc
+		#-----------------------------------------------------------------------------------------
+		# class methodsFor: "initialization"
+		def initialize()
+			@proc = nil
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "process creation"
+		def Process.create(
+				p_strAppName,
+				p_strCmdLine,
+				p_processAttrs        = nil,
+				p_threadAttrs         = nil,
+				p_bInheritHandles     = false,
+				p_nCreationFlags      = NORMAL_PRIORITY_CLASS,
+				p_strEnvironment      = nil,
+				p_strCurrentDirectory = nil,
+				p_startupInfo         = nil)
+			process = Process.new()
+			process.create(
+				p_strAppName, p_strCmdLine,
+				p_processAttrs, p_threadAttrs, p_bInheritHandles,
+				p_nCreationFlags, p_strEnvironment, p_strCurrentDirectory, p_startupInfo)
+			return process			
+		end
+
+		def create(
+				p_strAppName,
+				p_strCmdLine,
+				p_processAttrs        = nil,
+				p_threadAttrs         = nil,
+				p_bInheritHandles     = false,
+				p_nCreationFlags      = NORMAL_PRIORITY_CLASS,
+				p_strEnvironment      = nil,
+				p_strCurrentDirectory = nil,
+				p_startupInfo         = nil)
+			self.close() if !@proc.nil?
+			pProcessInfo = " " * ProcessInfo.sizeof
+			ret = WCall('CreateProcess',
+						p_strAppName, p_strCmdLine,
+						p_processAttrs, p_threadAttrs, p_bInheritHandles ? 1 : 0,
+						p_nCreationFlags, p_strEnvironment, p_strCurrentDirectory,
+						p_startupInfo.nil? ? StartupInfo.new().pack : p_startupInfo.pack,
+						pProcessInfo)
+			$TRACE.debug 9, "after CreateProcess ret = #{ret.inspect}"
+			@proc = ProcessInfo.unpack(pProcessInfo)
+			
+			if ret == 0 then
+				raise Win32::Error.new("process '#{p_strAppName}' could not be created because, ")
+			elsif @proc.pid == 0
+				raise Win32::Error.new, "process '#{p_strAppName}' could not be created (unknown reason)", caller
+			end
+		end
+
+		def Process.open(p_processAttrs, p_bInheritHandles, p_pid)
+			process = Process.new()
+			process.open(p_processAttrs, p_bInheritHandles, p_pid)
+		end
+
+		def open(p_processAttrs, p_bInheritHandles, p_pid)
+			self.close() if !@proc.nil?
+			ret = WCall('CreateProcess', p_processAttrs, p_bInheritHandles, p_pid)
+			raise Win32::Error.new, "Cannot Open Process", caller if 0 == ret
+			@proc = ProcessInfo.new(ret)
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "closing"
+		def close
+			return 0 if @proc.nil?
+		end
+		
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "terminating"
+		def terminate(p_nExitCode)
+			raise Win32::Error.new, "Missing Process Info", caller if @proc.nil?
+			ret = WCall('TerminateProcess', @proc.hProcess, p_nExitCode)
+			raise Win32::Error.new, "Error terminating process #{@proc.hProcess}", caller if 0 == ret
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "signalling"
+		def wait(p_nMilliseconds, p_bAlertable = true)
+			raise Win32::Error.new, "Missing Process Info", caller if @proc.nil?
+			ret = WCall('WaitForObject', @proc.hProcess, p_nMilliseconds, p_bAlertable ? 1 : 0)
+			raise Win32::Error.new, "Error while waiting for #{@proc.pid}", caller if ret == WAIT_FAILED
+			return ret
+		end
+
+		def waitForInputIdle(p_nMilliseconds)
+			raise Win32::Error.new, "Missing Process Info", caller if @proc.nil?
+			ret = WCall('WaitForInputIdle', @proc.hProcess, p_nMilliseconds)
+			raise Win32::Error.new, "Error while waiting for #{@proc.pid}", caller if ret == WAIT_FAILED
+			return ret
+		end
+
+		def close_handles
+			if !@proc.nil? then
+				closeHandle = Win32API.new("kernel32", "CloseHandle", ['L'], 'L')
+				closeHandle.Call(@proc.hProcess) if @proc.hProcess != nil
+				closeHandle.Call(@proc.hThread) if @proc.hThread != nil
+			end
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "accessing"
+		def hProcess
+			return  @proc.nil? ? nil : @proc.hProcess
+		end
+
+		def hThread
+			return  @proc.nil? ? nil : @proc.hThread
+		end
+
+		def pid
+			return  @proc.nil? ? nil : @proc.pid
+		end
+
+		def tid
+			return  @proc.nil? ? nil : @proc.tid
+		end
+
+		#-----------------------------------------------------------------------------------------
+		# methodsFor: "printing"
+		def to_s
+			return @proc.nil? ? "nil" : @proc.to_s
+		end
+	end
+end