PC Professionell 2006 May

home *** CD-ROM | disk | FTP | other *** search

/ PC Professionell 2006 May / PCpro_2006_05.ISO / files / free_security / msshared / Shared_Computer_Toolkit_ENU.msi / FileScripts010 < prev next >

Wrap

Extensible Markup Language | 2005-09-02 | 34.3 KB | 434 lines

■<?xml version="1.0" ?> <package> <comment> ' *** ' *** ------------------------------------------------------------------------------ ' *** Filename: DiskProtect.wsf ' *** ------------------------------------------------------------------------------ ' *** Description: Command line interface to Disk Protection tool ' *** This file parses command line parameters and calls functions ' *** that are defined in clsDiskProtect.vbs ' *** ------------------------------------------------------------------------------ ' *** Version: 1.0 ' *** Notes: ' *** ------------------------------------------------------------------------------ ' *** Copyright (C) Microsoft Corporation 2005, All Rights Reserved ' *** ------------------------------------------------------------------------------ ' *** </comment> <job> <runtime> <description>Disk Protection Tool</description> <named name="Status" required="false" many="false" helpstring="Displays the status of disk protection" /> <named name="On" required="false" many="false" helpstring="Enables disk protection" /> <named name="Off" required="false" many="false" helpstring="Disables disk protection" /> <named name="Save" required="false" many="false" helpstring="Saves all changes to the protected drive on next restart" /> <named name="Clear" required="false" many="false" helpstring="All changes are cleared with every restart" /> <named name="Retain" required="false" many="false" helpstring="All changes are retained on restart" /> <named name="Once" required="false" many="false" helpstring="Used with the Retain option to keep changes for one restart" /> <named name="Restart" required="false" many="false" helpstring="Used with any option; forces the computer to restart" /> <named name="MU" required="false" many="false" helpstring="Install Microsoft Updates during critical update process" /> <named name="NoMU" required="false" many="false" helpstring="Disable Microsoft Updates during critical update process" /> <named name="AV" required="false" many="false" helpstring="Specifies a script that allows antivirus updates" /> <named name="Other" required="false" many="false" helpstring="Specifies a script that allows other customized updates" /> <named name="Time" required="false" many="false" helpstring="Critical update process time (specify hour between 0 and 23)" /> <named name="Day" required="false" many="false" helpstring="The days of the week to run the critical update process" /> <example>Example: DiskProtect.wsf /On /MU /AV:"full path to script file" /Time:22</example> </runtime> <resource id="CScriptMessage">Restarting script in command-line mode. Run CmdOn.BAT to set command-line mode as the default mode.</resource> <resource id="CScriptTitle">Shared Computer Toolkit: Windows Script Mode Detected</resource> <?job error="True" debug="False" ?> <script language="VBScript" src="../include/Common.vbs"></script> <script language="VBScript" src="../include/libWSF.vbs"></script> <script language="VBScript" src="../include/clsDiskProtect.vbs"></script> <script language="VBScript" src="../include/clsLogging.vbs"></script> <script language="VBScript"> <![CDATA[ ' ~~~ ' ~~~ Force variables to be declared ' ~~~ Option Explicit ' ~~~ ' ~~~ Declare variables and constants ' ~~~ Dim bOK, bResult, bReboot, oDiskProtect, oLog, iWDPcmdVer Dim oOS, colOperatingSystems, iInstallTime, iOverlayTime, iLastBootUpTime, sComputer Dim sUpperFilter,iLocalDateTime Dim bBlackList, bWhiteList Dim arrTempArray, i, arrUpperFilters() Call Main("DiskProtect.hta") ' ~~~ Create the disk protection object Set oDiskProtect = New DiskProtect ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Logging ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Create logging object Set oLog = New Logging ' ~~~ Initiate logging Call oLog.Open(GetRootFolder & "\log\diskprotect.wsf.log") Call oLog.Write("DiskProtection.wsf : Started") ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Check reboot key before trying to run tool ' ~~~ ------------------------------------------------------------------------------ Set colOperatingSystems = oWMIService.ExecQuery _ ("Select * from Win32_OperatingSystem") For Each oOS in colOperatingSystems iLastBootUpTime = oOS.LastBootUpTime iLocalDateTime = oOS.LocalDateTime Next iInstallTime = RegRead(TOOLKITKEY & "SCTInstallTime") iOverlayTime = RegRead(TOOLKITKEY & "SCTOverlayTime") On Error Goto 0 ' ~~~ Check UpperFilter registry key to ensure no other Disk Protection Software is installed ' ~~~ Read and record UpperFilters key in SCT registry oDiskProtect.GetStashUpperFilterKey() On Error Resume Next ' ~~~ Check only if Ewf not in UpperFilter If (InStr(UCase(sUpperFilter),","&"EWF"&",") < 1) or IsNull(sUpperFilter) Then ' ~~~ Set to Ewf if NUll If IsNull(sUpperFilter) Then ReDim Preserve arrUpperFilters(0) arrUpperFilters(0) = "Ewf" ' ~~~ Writes to UpperFilters key oDiskProtect.WriteUpperFilterKey() ' ~~~ Read and record UpperFilters key in SCT registry oDiskProtect.GetStashUpperFilterKey() Call oLog.Write("DiskProtection.wsf : UpperFilter set to Ewf") WScript.Echo WScript.Echo "You must restart this computer before you can run Windows Disk Protection." QuitScript() Else ' ~~~ Black List Check Call oDiskProtect.UpperFilterCheck("BlackList") If bBlackList Then ' ~~~ Warn of Blacklist hit... Exit! Call oLog.Write("DiskProtection.wsf : BlackList hit... Other Disk Protection Software installed. Quit") WScript.Echo WScript.Echo "You must uninstall any other disk protection software before configuring" WScript.Echo "Windows Disk Protection. Windows Disk Protection will not work if you have" WScript.Echo "other disk protection software installed on the same computer." QuitScript() Else ' ~~~ White List Check Call oDiskProtect.UpperFilterCheck("WhiteList") If bWhiteList Then ' ~~~ Proceed with setting UpperFilters arrTempArray = Split(sUpperFilter, ",") For i = LBound(arrTempArray) to UBound(arrTempArray) ReDim Preserve arrUpperFilters(i) arrUpperFilters(i) = arrTempArray(i+1) Next ReDim Preserve arrUpperFilters(i-2) arrUpperFilters(i-2) = "Ewf" ' ~~~ Writes to UpperFilters key oDiskProtect.WriteUpperFilterKey() ' ~~~ Read and record UpperFilters key in SCT registry oDiskProtect.GetStashUpperFilterKey() ' ~~~ Registry modified... prompt for Restart Call oLog.Write("DiskProtection.wsf : UpperFilter set to WhiteList + Ewf") WScript.Echo WScript.Echo "You must restart this computer before you can run Windows Disk Protection." QuitScript() Else ' ~~~ Proceed with setting Gray UpperFilters arrTempArray = Split(sUpperFilter, ",") For i = LBound(arrTempArray) to UBound(arrTempArray) ReDim Preserve arrUpperFilters(i) arrUpperFilters(i) = arrTempArray(i+1) Next ReDim Preserve arrUpperFilters(i-2) arrUpperFilters(i-2) = "Ewf" ' ~~~ Writes to UpperFilters key oDiskProtect.WriteUpperFilterKey() ' ~~~ Read and record UpperFilters key in SCT registry oDiskProtect.GetStashUpperFilterKey() ' ~~~ Registry modified... prompt for Restart Call oLog.Write("DiskProtection.wsf : UpperFilter set to GrayList + Ewf") WScript.Echo WScript.Echo "This computer has multiple UpperFilters configured. Windows Disk Protection" WScript.Echo "might not be compatible with all UpperFilters. Please contact Product Support" WScript.Echo "if Windows Disk Protection causes your computer to malfunction." WScript.Echo WScript.Echo "You must restart this computer before you can run Windows Disk Protection." QuitScript() End If End If End If On Error Goto 0 End If If iInstallTime > iLastBootUpTime Then Call oLog.Write("DiskProtection.wsf : Computer has not been restarted since install. Quit") WScript.Echo WScript.Echo "You must restart this computer before you can run Windows Disk Protection." QuitScript() End If If iOverlayTime > iLastBootUpTime Then Call oLog.Write("DiskProtection.wsf : Computer has not been restarted since install. Quit") WScript.Echo WScript.Echo "Windows Disk Protection has been turned On. The computer must be restarted before you can run Windows Disk Protection." QuitScript() End If ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Check script usage ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Assume usage is ok bOK = True ' ~~~ Simple checks for incorrect usage Select Case WScript.Arguments.Named.Count Case 0 ' ~~~ Can not call with no parameters bOK = False Case 1 ' ~~~ Once can not be used on its own If WScript.Arguments.Named.Exists("Once") Then bOk = False End Select ' ~~~ Call the Validations sub to validate the inputs Call Validations() ' ~~~ If incorrect usage display message and quit If Not(bOK) Then Call oLog.Write("DiskProtection.wsf : Incorrect usage detected") WScript.Arguments.ShowUsage QuitScript() End If ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Now do disk protection ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Default to no reboot required bReboot = False bOK = False ' ~~~ Set the WDP.cmd version # for the DiskProtect class iWDPcmdVer = 2 ' ~~~ Bind to the logging object oDiskProtect.Logging = oLog If Not(oDiskProtect.OverlayCreated) and oDiskProtect.CalcOVSize = 0 and oDiskProtect.ForceOverlaySize = 0 Then WScript.Echo "This computer does not currently support Windows Disk Protection.To resolve this problem refer to Tool kit's documentation." Set oDiskProtect = Nothing QuitScript() End If ' ~~~ Perform disk protection enable action If WScript.Arguments.Named.Exists("On") Then bOK = True If NOT (WScript.Arguments.Named.Exists("Off")) Then Call oLog.Write("DiskProtection.wsf : Enable") If oDiskProtect.GetHibernation() = True Then WScript.Echo "Hibernation is currently Enabled. Hibernation will not work when Windows Disk Protection is turned On." End If Call oDiskProtect.Enable WScript.Echo "Disk Protection has been turned On." bReboot = True Else WScript.Echo "Please enter On or Off. Both options cannot be used at the same time." WScript.Arguments.ShowUsage QuitScript() End If End If ' ~~~ Perform disk protection disable action If WScript.Arguments.Named.Exists("Off") Then bOK = True If NOT (WScript.Arguments.Named.Exists("On")) Then Call oLog.Write("DiskProtection.wsf : Disable") Call oDiskProtect.Disable WScript.Echo "Disk Protection has been turned Off." bReboot = True End If End If ' ~~~ Perform disk protection status action If WScript.Arguments.Named.Exists("Status") Then Call oLog.Write("DiskProtection.wsf : Status") If oDiskProtect.Enabled Then WScript.Echo "Windows Disk Protection is On" Else WScript.Echo "Windows Disk Protection is Off" End If bOK = True End If ' ~~~ Perform disk protection undo changes action If WScript.Arguments.Named.Exists("Clear") Then bOK = True If oDiskProtect.Enabled = True Then Call oLog.Write("DiskProtection.wsf : UndoChanges") Call oDiskProtect.UndoChanges WScript.Echo "Windows Disk Protection is On. All changes will be cleared with every restart" Else WScript.Echo "Restart the computer to clear changes." End If End If ' ~~~ Perform disk protection retain changes action If WScript.Arguments.Named.Exists("Retain") Then bOK = True If oDiskProtect.Enabled = True Then Call oLog.Write("DiskProtection.wsf : RetainChanges") Call oDiskProtect.RetainChanges(WScript.Arguments.Named.Exists("Once")) If WScript.Arguments.Named.Exists("Once") Then WScript.Echo "Windows Disk Protection is On. Changes will be retained for one restart." Else WScript.Echo "Windows Disk Protection is On. Changes will be retained after computer restarts." End If End If End If ' ~~~ Set the scheduled update task time If WScript.Arguments.Named("Time") <> "" and IsNumeric(WScript.Arguments.Named("Time")) and WScript.Arguments.Named("Time") < "24" Then oDiskProtect.CriticalUpdateTime = WScript.Arguments.Named("Time") WScript.Echo "Windows Disk Protection critical update time set." bOK = True End If ' ~~~ Set the scheduled update task day of week If WScript.Arguments.Named("Day") <> "" Then oDiskProtect.CriticalUpdateDay = WScript.Arguments.Named("Day") WScript.Echo "Windows Disk Protection critical update frequency set." bOK = True End If ' ~~~ Set the anti-virus script name If WScript.Arguments.Named("AV") <> "" Then oDiskProtect.AVUpdateScript = WScript.Arguments.Named("AV") WScript.Echo "Windows Disk Protection anti-virus script name set." bOK = True End If ' ~~~ Turn on Microsoft Updates If WScript.Arguments.Named.Exists("MU") Then oDiskProtect.WindowsUpdateScript = True WScript.Echo "Windows Disk Protection Microsoft Updates processing has been turned On." bOK = True End If ' ~~~ Turn off Microsoft Updates If WScript.Arguments.Named.Exists("NOMU") Then oDiskProtect.WindowsUpdateScript = False WScript.Echo "Windows Disk Protection Microsoft Updates processing has been turned Off." bOK = True End If ' ~~~ Set the other script name If WScript.Arguments.Named("Other") <> "" Then oDiskProtect.OtherUpdateScript = WScript.Arguments.Named("Other") WScript.Echo "Windows Disk Protection other script name set." bOK = True End If If WScript.Arguments.Named.Exists("On") Then Call oDiskProtect.EnableCriticalUpdates() If WScript.Arguments.Named.Exists("Off") Then Call oDiskProtect.DisableCriticalUpdates() ' ~~~ Perform disk protection commit action If WScript.Arguments.Named.Exists("Save") Then bOK = True If oDiskProtect.Enabled = True Then Call oDiskProtect.EnableCriticalUpdates() Call oLog.Write("DiskProtection.wsf : Commit") Call oDiskProtect.Commit WScript.Echo "Windows Disk Protection is On. Changes will be saved on the next restart." bReboot = True Else WScript.Echo "Windows Disk Protection is Off. Save requires Windows Disk Protection to be On." End If End If ' ~~~ Force reboot if requested If WScript.Arguments.Named.Exists("Restart") Then Call oLog.Write("DiskProtection.wsf : Restart") Call oDiskProtect.Reboot bOK = True Else If bReboot Then WScript.Echo "Restart the computer for changes to take effect" End If If Not(bOK) Then Call oLog.Write("DiskProtection.wsf : Incorrect usage detected") WScript.Echo "Incorrect usage. See Disk Protection Tool options below." WScript.Echo WScript.Arguments.ShowUsage QuitScript() End If ' *** ' *** -------------------------------------------------------------------------------- ' *** Name: Validations() ' *** -------------------------------------------------------------------------------- ' *** Purpose: Validate the inputs ' *** -------------------------------------------------------------------------------- ' *** Sub Validations() If WScript.Arguments.Named("AV") <> "" Then If NOT oFso.FileExists ( WScript.Arguments.Named("AV") ) Then WScript.Echo "The specified anti-virus script does not exist." WScript.Echo "Enter a valid anti-virus script." QuitScript() End If End If If WScript.Arguments.Named("Other") <> "" Then If NOT oFso.FileExists ( WScript.Arguments.Named("Other") ) Then WScript.Echo "The specified script does not exist." WScript.Echo "Enter a valid script." QuitScript() End If End If End Sub ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Tidy up ' ~~~ ------------------------------------------------------------------------------ ' ~~~ Close log Call oLog.Close ' ~~~ Destroy objects Set oDiskProtect = nothing Set oLog = nothing UnLoadObjects() ]]> </script> </job> </package>