home *** CD-ROM | disk | FTP | other *** search
/ PC Professionell 2006 May / PCpro_2006_05.ISO / files / free_security / msshared / Shared_Computer_Toolkit_ENU.msi / FileScripts004 < prev    next >
Encoding:
Extensible Markup Language  |  2005-09-02  |  15.4 KB  |  195 lines
  1.  ■<?xml version="1.0" ?>
  3. <package>
  5.     <comment>
  7.         ' *** 
  9.         ' *** ------------------------------------------------------------------------------
  11.         ' *** Filename:        AutoRestart.wsf
  13.         ' *** ------------------------------------------------------------------------------
  15.         ' *** Description:    Command line tool to cause a User Profile to monitor for and
  17.         ' ***             restart a specific program if it is closed.
  19.         ' *** ------------------------------------------------------------------------------
  21.         ' *** Version:        1.0
  23.         ' *** Notes:        
  25.         ' *** ------------------------------------------------------------------------------
  27.         ' *** Copyright (C) Microsoft Corporation 2005, All Rights Reserved
  29.         ' *** ------------------------------------------------------------------------------
  31.         ' ***
  33.     </comment>
  35.     <job>
  37.     <runtime>
  39.             <description>AutoRestart Script</description>
  41.             <named name="enable" required="false" many="false" helpstring="Enable AutoRestart for an account with a specific program" />
  43.             <named name="disable" required="false" many="false" helpstring="Disable AutoRestart for an account" />
  45.             <named name="list" required="false" many="false" helpstring="List AutoRestart command for an account" />
  47.             <example>
  49. Examples:
  51. AutoRestart.wsf /enable Public "C:\Program Files\Internet Explorer\IExplore.exe"
  53. AutoRestart.wsf /disable Public
  55. AutoRestart.wsf /list Public</example>
  57.         </runtime>
  59.     
  61.         <resource id="IncorrectUsage">Incorrect Usage Detected</resource>
  63.         <resource id="ScriptName">AutoRestart.wsf</resource>
  65.         <resource id="EnableSuccess">AutoRestart has been enabled for the user. AutoRestart command:</resource>
  67.         <resource id="DisableSuccess">AutoRestart has been disabled for the user.</resource>
  69.         <resource id="ListSuccess">AutoRestart command for this account:</resource>
  71.         <resource id="ListError">There is no AutoRestart command for this account.</resource>
  73.         <resource id="InvalidUser">Unable to open the registry for this account.</resource>
  75.         <resource id="InvalidUser2">The account is either logged in or does not exist.</resource>
  77.         <resource id="AdminErrorMsg">You have to be an administrator to use this script.</resource>
  79.         <resource id="NoToolkit">The Toolkit path cannot be found. Please reinstall the Toolkit.</resource>
  81.         <resource id="ErrorMessage">Could not modify the user registry successfully.</resource>
  83.         <resource id="CScriptMessage">Restarting script in command-line mode. Run CmdOn.BAT to set command-line mode as the default mode.</resource>
  85.         <resource id="CScriptTitle">Shared Computer Toolkit: Windows Script Mode Detected</resource>
  87.                 
  89.         <?job error="True" debug="False" ?>
  91.     
  93.         <script language="VBScript" src="../include/Common.vbs"></script>
  95.         <script language="VBScript" src="../include/libWSF.vbs"></script>
  97.         <script language="VBScript" src="../include/clsLogging.vbs"></script>
  99.         <script language="VBScript" src="../include/clsRestrictions.vbs"></script>
  101.         <script language="VBScript">
  103.         <![CDATA[
  105.             ' ~~~ 
  107.             ' ~~~ Force variables to be declared 
  109.             ' ~~~ 
  111.             Option Explicit
  113.             
  115.             On Error Goto 0
  119.             ' ~~~ Call the Main 
  121.             Call Main("")
  123.             
  125.             ' ~~~ 
  127.             ' ~~~ Declare variables and constants
  129.             ' ~~~ 
  131.             Dim oRestrict, oLog , bOK, args, sAppDir, sCMD
  133.             
  135.             Const TARGETDIR = "HKLM\Software\Microsoft\Shared Computer Toolkit\TargetDir"
  137.             Const RUNCMD = "bin\AutoRestart.vbs"
  139.             Const RUNKEY = "HKEY_USERS\SSW\Software\Microsoft\Windows\CurrentVersion\Run\SCTRunAutoRestart"
  143.             Set oRestrict = New Restriction
  145.             
  147.             ' ~~~ ------------------------------------------------------------------------------
  149.             ' ~~~ Logging
  151.             ' ~~~ ------------------------------------------------------------------------------
  153.             ' ~~~ Create logging object
  155.             Set oLog = New Logging
  159.             ' ~~~ Initiate logging
  161.             Call oLog.Open(GetRootFolder & "\log\AutoRestart.wsf.log")
  163.             Call oLog.Write("AutoRestart.wsf : Started")
  165.             
  167.             
  169.             ' ~~~ ------------------------------------------------------------------------------
  171.             ' ~~~ Check script usage
  173.             ' ~~~ ------------------------------------------------------------------------------
  175.             bOK = False
  177.             Set args = WScript.Arguments
  179.             
  181.             If Wscript.Arguments.Named.Exists("enable") and args.Count = 3 Then bOK = True
  183.             If Wscript.Arguments.Named.Exists("disable") and args.Count = 2 Then bOK = True
  185.             If Wscript.Arguments.Named.Exists("list") and args.Count = 2 Then bOK = True
  187.             
  189.             If Not(bOK) Then
  191.                 Call oLog.Write(getResource("ScriptName") & " : " & getResource("IncorrectUsage"))
  193.                 Wscript.Echo getResource("IncorrectUsage")
  195.                 WScript.Arguments.Showusage
  197.                 QuitScript()
  199.             End If
  203.             ' ~~~ ------------------------------------------------------------------------------
  205.             ' ~~~ Read the TargetDir for the Toolkit
  207.             ' ~~~ ------------------------------------------------------------------------------
  211.             ' ~~~ Need to check error returns here
  213.             On Error Resume Next
  215.             sAppDir = ""
  217.             sAppDir = oShell.RegRead(TARGETDIR)
  219.             If Err.number <> 0 Then 
  221.                 Call oLog.Write(getResource("ScriptName") & " : " & getResource("NoToolkit"))
  223.                 Wscript.Echo getResource("NoToolkit")
  225.                 WScript.Arguments.Showusage
  227.                 QuitScript()
  229.             End If
  233.             On Error Goto 0
  237.             ' ~~~ ------------------------------------------------------------------------------
  239.             ' ~~~ Load the user's registry hive
  241.             ' ~~~ ------------------------------------------------------------------------------
  243.             
  245.             If NOT oRestrict.LoadUser(args(1)) Then
  247.                 Call oLog.Write(getResource("ScriptName") & " : " & getResource("InvalidUser"))
  249.                 WScript.Echo getResource("InvalidUser")
  251.                 WScript.Echo getResource("InvalidUser2")
  253.                 WScript.Echo
  255.                 QuitScript()
  257.             End If
  261.             ' ~~~ ------------------------------------------------------------------------------
  263.             ' ~~~ Now do enable/disable
  265.             ' ~~~ ------------------------------------------------------------------------------
  269.             ' ~~~ We need to catch error codes below
  271.             On Error Resume Next
  275.             ' ~~~ Enable AutoRestart on the account 
  277.             If Wscript.Arguments.Named.Exists("enable") Then
  279.                 Err.Clear
  281.                 RegWrite RUNKEY, "%windir%\system32\WScript.EXE " & chr(34) & sAppDir & RUNCMD & Chr(34) & " " & Chr(34) & args(2) & Chr(34), "REG_EXPAND_SZ"
  283.                 If err.number = 0 Then 
  285.                     Call oLog.Write("enable : " & getResource("EnableSuccess"))
  287.                     WScript.Echo getResource("EnableSuccess")
  289.                     WScript.Echo "%windir%\system32\WScript.EXE " & chr(34) & sAppDir & RUNCMD & Chr(34) & " " & Chr(34) & args(2) & Chr(34)
  291.                 Else
  293.                     Call oLog.Write("enable : " & getResource("ErrorMessage"))
  295.                     WScript.Echo getResource("ErrorMessage")
  297.                 End If
  299.             End If
  301.             
  303.             ' ~~~ Disable AutoRestart on the account
  305.             If Wscript.Arguments.Named.Exists("disable") Then
  307.                 Err.Clear
  309.                 oShell.RegDelete RUNKEY
  311.                 If err.number = 0 or err.number = -2147024894 Then 
  313.                     Call oLog.Write("enable : " & getResource("DisableSuccess"))
  315.                     WScript.Echo getResource("DisableSuccess")
  317.                 Else
  319.                     Call oLog.Write("enable : " & getResource("ErrorMessage"))
  321.                     WScript.Echo getResource("ErrorMessage")
  323.                 End If
  325.             End If
  329.             If Wscript.Arguments.Named.Exists("list") Then
  331.                 Err.Clear
  333.                 sCMD = oShell.RegRead(RUNKEY)
  335.                 If err.number = 0 Then 
  337.                     Wscript.Echo GetResource("ListSuccess")
  339.                     Wscript.Echo sCMD
  341.                 Else
  343.                     Wscript.Echo GetResource("ListError")
  345.                 End If
  349.             End If
  353.             On Error Goto 0
  357.             ' ~~~ ------------------------------------------------------------------------------
  359.             ' ~~~ Unload the user's registry hive
  361.             ' ~~~ ------------------------------------------------------------------------------
  363.             oRestrict.UnLoadUser
  367.             ' ~~~ Close log
  369.             Call oLog.Close
  371.             
  373.             ' ~~~ Destroy objects
  375.             Set oLog = nothing
  377.             UnLoadObjects()
  379.             
  381.         ]]>
  383.         </script>
  385.     </job>
  387. </package>