Dim WShell, FSO, Logger
CreateGlobalObjects()
Main()
''''''''''''''''''''''''''
''' Global Functions
''''''''''''''''''''''''''
Sub CreateGlobalObjects()
Set WShell = CreateObject("wscript.shell")
HandleError("Cannot create object 'wscript.shell' (CreateGlobalObjects).")
Set FSO = CreateObject("Scripting.FileSystemObject")
HandleError("Cannot create object 'Scripting.FileSystemObject'(CreateGlobalObjects).")
Set Logger = new ScriptLogger
HandleError("Cannot create object 'ScriptLogger'(CreateGlobalObjects).")
End Sub
Sub ScriptExit()
Wscript.Quit 0
End Sub
Public Function GetSQLServiceName(sInstance)
If sInstance = SQL_DEFAULT Then
GetSQLServiceName = SQL_DEFAULT
Else
GetSQLServiceName = "MSSQL$" & sInstance
End If
End Function
Function EscapeWQLString (ByVal strValue)
ON ERROR RESUME NEXT
Err.Clear
EscapeWQLString = Replace(Replace(strValue, "\", "\\"), "'", "\'")
End Function
Sub ReturnEmptyPropertyBag(opsMgrAPI)
Dim propertyBag
Set propertyBag = opsMgrAPI.scriptAPI.CreateTypedPropertyBag(2)
opsMgrAPI.scriptAPI.AddItem(propertyBag)
opsMgrAPI.scriptAPI.ReturnItems()
End Sub
'The function returns service or "Unknown" state
'Input:
' server - compute name
' service - system service name
'Output:
' service state or "Unknown" state
Function GetServiceState( sTargetComputer, sServiceName)
On Error Resume Next
Dim sNamespace, sQuery, oWMI, objClasses, sState
sNamespace = "winmgmts://" & sTargetComputer & "/root/cimv2"
sQuery = "SELECT State FROM Win32_Service where Name = """ & EscapeWQLString(sServiceName) & """"
Set oWMI = GetObject(sNamespace)
Set objClasses = oWMI.ExecQuery(sQuery)
if objClasses.Count >= 1 Then
sState = GetFirstItemFromWMIQuery(objClasses).Properties_.Item("State")
End If
If Err.number <> 0 Or objClasses.Count = 0 Then
sState = "Unknown"
End If
Public Sub Save()
m_lNumber = Err.Number
m_sSource = Err.Source
m_sDescription = Err.Description
m_sHelpContext = Err.HelpContext
m_sHelpFile = Err.HelpFile
End Sub
Public Sub Raise()
Err.Raise m_lNumber, m_sSource, m_sDescription, m_sHelpFile, m_sHelpContext
End Sub
Public Sub Clear()
m_lNumber = 0
m_sSource = ""
m_sDescription = ""
m_sHelpContext = ""
m_sHelpFile = ""
End Sub
Public Default Property Get Number()
Number = m_lNumber
End Property
Public Property Get Source()
Source = m_sSource
End Property
Public Property Get Description()
Description = m_sDescription
End Property
Public Property Get HelpContext()
HelpContext = m_sHelpContext
End Property
Public Property Get HelpFile()
HelpFile = m_sHelpFile
End Property
End Class
Class ArrayList
Private m_itemArray
Private Sub Class_Initialize()
Me.Clear
End Sub
Private Sub AddItemToArray(ByRef itemArray, ByVal item)
ReDim Preserve itemArray(UBound(itemArray) + 1)
itemArray(UBound(itemArray)) = item
End Sub
Public Sub Clear()
m_itemArray = Array()
End Sub
Public Sub Add(item)
AddItemToArray m_itemArray, item
End Sub
Public Sub RemoveAt(index)
If index < 0 Or index > UBound(m_itemArray) Then
Exit Sub
End If
Dim newArr: newArr = Array()
Dim i
For i = 0 To UBound(m_itemArray)
If i <> index Then
Call AddItemToArray(newArr, m_itemArray(i))
End If
Next
m_itemArray = newArr
End Sub
Public Property Get Count()
Count = UBound(m_itemArray) + 1
End Property
Public Property Get ItemsArray()
ItemsArray = m_itemArray
End Property
Public Property Get IsEmpty()
IsEmpty = UBound(m_itemArray) < 0
End Property
End Class
''''''''''''''''''''''''''''''''''''
''' ScriptLogger
''''''''''''''''''''''''''''''''''''
Class ScriptLogger
Dim sourceLogEvent
Private Sub Class_Initialize()
sourceLogEvent = "Management Group: " + ManagementGroupName + ". Script: " + WScript.ScriptName + ". Version: " + MANAGEMENT_PACK_VERSION
End Sub
Private Sub Class_Terminate()
End Sub
Public Property Get ErrorEventType
ErrorEventType = 1
End Property
Public Property Get WarningEventType
WarningEventType = 2
End Property
Public Property Get InfoEventType
InfoEventType = 4
End Property
Private Function LogEvent (message, eventType)
On Error Resume Next
Dim oAPI
Set oAPI = CreateObject("MOM.ScriptAPI")
Call oAPI.LogScriptEvent(sourceLogEvent, SCRIPT_EVENT_ID, eventType, message)
End Function
Public Function LogDebug(message)
if DEBUG_MODE Then
WScript.StdOut.WriteLine message
LogEvent message, Me.InfoEventType
End If
End Function
Public Function LogError(message)
if DEBUG_MODE Then
WScript.StdOut.WriteLine message
End If
LogEvent message, Me.ErrorEventType
End Function
Public Function LogWarning(message)
if DEBUG_MODE Then
WScript.StdOut.WriteLine message
End If
LogEvent message, Me.WarningEventType
End Function
Public Function LogFormattedError(customMessage)
If Err.number <> 0 Then
Me.LogError FormatErrorMessage(customMessage, "")
End If
End Function
Private Function ScriptInfo()
Dim commandLineInfo : commandLineInfo = WScript.ScriptFullName
Dim argument
For Each argument In WScript.Arguments
commandLineInfo = commandLineInfo & " """ & argument & """"
Next
ScriptInfo = commandLineInfo
End Function
End Class
Function FormatErrorMessage(customMessage, instanceName)
FormatErrorMessage = customMessage
If Err.number <> 0 Then
Dim msg
msg =_
" Error Number: " & CStr(Err.number) & VbCrLf & _
" Description: " & Err.Description
If Not IsEmpty(instanceName) And instanceName <> "" Then
msg = msg & VbCrLf & " Instance: " & instanceName
End If
If customMessage <> "" Then
msg = customMessage & VbCrLf & msg & VbCrLf
End If
FormatErrorMessage = msg
End If
End Function
Function FormatDbErrorMessage(message, instanceName, dbName)
FormatDbErrorMessage = message & VbCrLf & _
" Instance: " & instanceName & VbCrLf & _
" Database: " & dbName
End Function
Function ThrowScriptErrorNoAbort(ByVal sMessage, ByVal oErr)
On Error Resume Next
Dim errorText: errorText = sMessage & ": " & oErr.Description
GlobalErrorList.Add errorText
End Function
Function ThrowScriptError(Byval sMessage, ByVal oErr)
On Error Resume Next
ThrowScriptErrorNoAbort sMessage, oErr
GlobalErrorListToEventLog()
Quit()
End Function
Sub HandleError(customMessage)
Dim localLogger
If Err.number <> 0 Then
Set localLogger = new ScriptLogger
Call localLogger.LogFormattedError(customMessage)
Call Wscript.Quit(0)
End If
End Sub
Function HandleErrorContinue(customMessage)
HandleErrorContinue = True
If Err.number <> 0 Then
HandleErrorContinue = False
Call GlobalErrorList.Add(FormatErrorMessage(customMessage, ""))
Call Err.Clear()
End If
End Function
Function HandleErrorContinueEx(customMessage, instanceName)
HandleErrorContinueEx = True
If Err.number <> 0 Then
HandleErrorContinueEx = False
Call GlobalErrorList.Add(FormatErrorMessage(customMessage, instanceName))
Call Err.Clear()
End If
End Function
Function HandleSqlErrorContinue(adoConnection, customMessage, instanceName)
HandleSqlErrorContinue = True
If Err.Number <> 0 Then
HandleSqlErrorContinue = False
Dim sqlErr
Dim e: Set e = new Error
e.Save
On Error Resume Next
If adoConnection.Errors.Count > 0 Then
Set sqlErr = adoConnection.Errors(0)
adoConnection.Errors.Clear
Call Err.Raise(sqlErr.Number, sqlErr.Source, sqlErr.Description)
Else
Call e.Raise()
End If
Call HandleErrorContinueEx(customMessage, instanceName)
End If
End Function
Function GetGlobalErrorListEventString()
GetGlobalErrorListEventString = ""
If Not GlobalErrorList.IsEmpty Then
GetGlobalErrorListEventString = "The next errors occured:"& vbNewLine & Join(GlobalErrorList.ItemsArray, vbNewLine & vbNewLine)
End If
End Function
Function GlobalErrorListToEventLog()
On Error Resume Next
If Not GlobalErrorList.IsEmpty Then
Dim localLogger: Set localLogger = New ScriptLogger
localLogger.LogWarning GetGlobalErrorListEventString()
End If
End Function
Function Quit()
WScript.Quit()
End Function
'#Include File:WMI.vbs
Function EscapeWQLString (ByVal strValue)
On Error Resume Next
Err.Clear
EscapeWQLString = Replace(Replace(strValue, "\", "\\"), "'", "\'")
End Function
Function ConnectToWMI(ComputerName, strNamespace)
Set ConnectToWMI = Nothing
Set ConnectToWMI = GetObject("winmgmts:{impersonationLevel=impersonate}!\\" & ComputerName & "\" & strNamespace)
End Function
Function WMIGetProperty(oWmi, sPropName, nCIMType, ErrAction)
Dim sValue, oWmiProp
If Not IsValidObject(oWmi) Then
If (ErrAction And ErrAction_ThrowError) = ErrAction_ThrowError Then _
ThrowScriptErrorNoAbort "Accessing property on invalid WMI object.", Err
If (ErrAction And ErrAction_Abort) = ErrAction_Abort Then _
Quit()
WMIGetProperty = ""
Exit Function
End If
On Error Resume Next
Set oWmiProp = oWmi.Properties_.Item(sPropName)
If Err.Number <> 0 Then
If (ErrAction And ErrAction_ThrowError) = ErrAction_ThrowError Then _
ThrowScriptErrorNoAbort "An error occurred while accessing WMI property: '" & sPropName & "'.", Err
If (ErrAction And ErrAction_Abort) = ErrAction_Abort Then _
Quit()
End If
On Error GoTo 0
If IsValidObject(oWmiProp) Then
sValue = oWmiProp.Value
If IsNull(sValue) Then
'
' If value is null, return blank to avoid any issues
'
WMIGetProperty = ""
Else
Select Case (oWmiProp.CIMType)
Case wbemCimtypeString, wbemCimtypeSint16, wbemCimtypeSint32, wbemCimtypeReal32, wbemCimtypeReal64, wbemCimtypeSint8, wbemCimtypeUint8, wbemCimtypeUint16, wbemCimtypeUint32, wbemCimtypeSint64, wbemCimtypeUint64:
If Not oWmiProp.IsArray Then
WMIGetProperty = Trim(CStr(sValue))
Else
WMIGetProperty = Join(sValue, ", ")
End If
Case wbemCimtypeBoolean:
If sValue = 1 Or UCase(sValue) = "TRUE" Then
WMIGetProperty = "True"
Else
WMIGetProperty = "False"
End If
Case wbemCimtypeDatetime:
Dim sTmpStrDate
'
' First attempt to convert the whole wmi date string
'
sTmpStrDate = Mid(sValue, 5, 2) & "/" & _
Mid(sValue, 7, 2) & "/" & _
Left(sValue, 4) & " " & _
Mid (sValue, 9, 2) & ":" & _
Mid(sValue, 11, 2) & ":" & _
Mid(sValue, 13, 2)
If IsDate(sTmpStrDate) Then
WMIGetProperty = CDate(sTmpStrDate)
Else
'
' Second, attempt just to convert the YYYYMMDD
'
sTmpStrDate = Mid(sValue, 5, 2) & "/" & _
Mid(sValue, 7, 2) & "/" & _
Left(sValue, 4)
If IsDate(sTmpStrDate) Then
WMIGetProperty = CDate(sTmpStrDate)
Else
'
' Nothing works - return passed in string
'
WMIGetProperty = sValue
End If
End If
Case Else:
WMIGetProperty = ""
End Select
End If
Else
If (ErrAction And ErrAction_ThrowError) = ErrAction_ThrowError Then _
ThrowScriptErrorNoAbort "An error occurred while accessing WMI property: '" & sPropName & "'.", Err
If (ErrAction And ErrAction_Abort) = ErrAction_Abort Then _
Quit()
WMIGetProperty = ""
End If
If (ErrAction And ErrAction_Trace) = ErrAction_Trace Then _
WScript.Echo " + " & sPropName & " :: '" & WMIGetProperty & "'"
End Function
Function WMIExecQuery(ByVal sNamespace, ByVal sQuery)
'
' WMIExecQuery :: Executes the WMI query and returns the result set.
'
'
Dim oWMI, oQuery, nInstanceCount
Dim e
Set e = New Error
On Error Resume Next
Set oWMI = GetObject(sNamespace)
e.Save
On Error GoTo 0
If IsEmpty(oWMI) Then
ThrowScriptErrorNoAbort "Unable to open WMI Namespace '" & sNamespace & "'. Check to see if the WMI service is enabled and running, and ensure this WMI namespace exists.", e
ThrowEmptyDiscoveryData
End If
On Error Resume Next
Set oQuery = oWMI.ExecQuery(sQuery)
e.Save
On Error GoTo 0
If IsEmpty(oQuery) Or e.Number <> 0 Then
ThrowScriptError "The Query '" & sQuery & "' returned an invalid result set. Please check to see if this is a valid WMI Query.", e
End If
'Determine if we queried a valid WMI class - Count will return 0 or empty
On Error Resume Next
nInstanceCount = oQuery.Count
e.Save
On Error GoTo 0
If e.Number <> 0 Then
ThrowScriptError "The Query '" & sQuery & "' did not return any valid instances. Please check to see if this is a valid WMI Query.", e
End If
Set WMIExecQuery = oQuery
End Function
Function GetFirstItemFromWMIQuery(ByRef oQuery)
ON ERROR RESUME NEXT
Err.Clear
Dim oResult: Set oResult = Nothing
Set oResult = oQuery.ItemIndex(0)
if Err.number <> 0 then
Err.Clear
Dim oObject
For Each oObject in oQuery
Set oResult = oObject
Exit For
Next
end if
Set GetFirstItemFromWMIQuery = oResult
End Function
'#Include File:Registry.vbs
Class Registry
Public HKEY_CLASSES_ROOT
Public HKEY_CURRENT_USER
Public HKEY_LOCAL_MACHINE
Public HKEY_USERS
Public HKEY_CURRENT_CONFIG
Public HKEY_DYN_DATA
Public ERROR_ACCESS_DENIED
Public ERROR_KEY_NOT_FOUND
Public ERROR_VALUE_NOT_FOUND
Public SUCCESS
Public Sub Connect(ByVal sHostName)
Set m_oReg = GetObject("winmgmts://" & sHostName & "/root/default:StdRegProv")
End Sub
Public Property Get Hive()
Hive = m_lHive
End Property
Public Property Let Hive(ByVal lHive)
m_lHive = lHive
End Property
Public Function ReadDWORDValue(ByVal sKeyPath, ByVal sValueName, ByRef lResult)
Dim lValue
lResult = m_oReg.GetDWORDValue(m_lHive, sKeyPath, sValueName, lValue)
ReadDWORDValue = lValue
End Function
Public Function ReadStringValue(ByVal sKeyPath, ByVal sValueName, ByRef lResult)
Dim sValue
lResult = m_oReg.GetStringValue(m_lHive, sKeyPath, sValueName, sValue)
ReadStringValue = sValue
End Function
Public Function ReadMultiStringValue(ByVal sKeyPath, ByVal sValueName, ByRef lResult)
Dim aValues
lResult = m_oReg.GetMultiStringValue(m_lHive, sKeyPath, sValueName, aValues)
ReadMultiStringValue = aValues
End Function
Public Function EnumKeys(ByVal sKeyPath, ByRef lResult)
Dim aSubKeys
lResult = m_oReg.EnumKey(m_lHive, sKeyPath, aSubKeys)
EnumKeys = aSubKeys
End Function
Public Function EnumValues(ByVal sKeyPath, ByRef lResult)
Dim aNames, aTypes
lResult = m_oReg.EnumValues(m_lHive, sKeyPath, aNames, aTypes)
EnumValues = aNames
End Function
Public Function CreateKey(ByVal sKeyPath)
CreateKey = m_oReg.CreateKey(m_lHive, sKeyPath)
End Function
Public Function WriteStringValue(ByVal sKeyPath, ByVal sValueName, ByVal sValue)
WriteStringValue = m_oReg.SetStringValue(m_lHive, sKeyPath, sValueName, sValue)
End Function
Public Function DeleteValue(ByVal sKeyPath, ByVal sValueName)
DeleteValue = m_oReg.DeleteValue(m_lHive, sKeyPath, sValueName)
End Function
Public Function ReadBinaryValue(ByVal sKeyPath, ByVal sValueName, ByRef lResult)
Dim aData
lResult = m_oReg.GetBinaryValue(m_lHive, sKeyPath, sValueName, aData)
ReadBinaryValue = aData
End Function
End Class
'#Include File:ConnectionString.vbs
Function ArraysAreIntersected(ByRef arrA, ByRef arrB)
Dim itemA, itemB
ArraysAreIntersected = False
For Each itemA in arrA
For Each itemB in arrB
If itemA = itemB Then
ArraysAreIntersected = True
Exit Function
End If
Next
Next
End Function
Function ArrayContains(ByRef arr, ByRef searchItem)
ArrayContains = False
Dim item
For Each item In arr
If item = searchItem Then
ArrayContains = True
Exit Function
End If
Next
End Function
Function GetDriverVersionObj(oVersion)
Dim tpName, vVersion
tpName = TypeName(oVersion)
Select Case tpName
Case "DriverVersion"
Set vVersion = oVersion
Case "String"
Set vVersion = (new DriverVersion)(oVersion)
Case Else
Call Err.Raise(7100, "DriverVersion", "Parameter is not an instance of type DriverVersion.")
End Select
Set GetDriverVersionObj = vVersion
End Function
Class DriverVersion
Private m_major
Private m_minor
Private m_build
Private m_revision
Public Default Function Init(versionStr)
Parse(versionStr)
Set Init = Me
End Function
Private Sub Parse(vStr)
Dim parts, versionRegex, oRegex, matches
Set oRegex = New RegExp
oRegex.Pattern = "^(\d{1,8})(.\d{1,8}){0,3}$"
Set matches = oRegex.Execute(vStr)
If matches.Count = 0 Then
Call Err.Raise(7100, "DriverVersion", "Invalid version string.")
End If
parts = Split(vStr, ".")
m_major = CLng(parts(0))
m_minor = 0
m_build = 0
m_revision = 0
IF UBound(parts) >= 1 Then
m_minor = CLng(parts(1))
If UBound(parts) >= 2 Then
m_build = CLng(parts(2))
If UBound(parts) >= 3 Then
m_revision = CLng(parts(3))
End If
End If
End If
End Sub
Public Function CompareTo(v)
Dim d
d = m_major - v.Major
If d <> 0 Then
CompareTo = d
Exit Function
End If
d = m_minor - v.Minor
If d <> 0 Then
CompareTo = d
Exit Function
End If
d = m_build - v.Build
If d <> 0 Then
CompareTo = d
Exit Function
End If
d = m_revision - v.Revision
CompareTo = d
End Function
Public Function ToString()
ToString = "" & m_major & "." & m_minor & "." & m_build & "." & m_revision
End Function
Public Property Get Major()
Major = m_major
End Property
Public Property Get Minor()
Minor = m_minor
End Property
Public Property Get Build()
Build = m_build
End Property
Public Property Get Revision()
Revision = m_revision
End Property
End Class
Class VersionRange
Private m_startVersion
Private m_endVersion
Public Default Function Init(sStartVersion, sEndVersion)
Set m_startVersion = GetDriverVersionObj(sStartVersion)
Set m_endVersion = GetDriverVersionObj(sEndVersion)
Set Init = Me
End Function
Public Property Get StartVersion()
Set StartVersion = m_startVersion
End Property
Public Property Get EndVersion()
Set EndVersion = m_endVersion
End Property
End Class
Class DriverItem
Private m_name
Private m_nameVersion
Private m_driverVersion
Private m_parseObject
Public Default Function Init(sName, oNameVersion, oDriverVersion, ByRef oParseObject)
m_Name = sName
Set m_nameVersion = GetDriverVersionObj(oNameVersion)
Set m_driverVersion = GetDriverVersionObj(oDriverVersion)
Set m_parseObject = oParseObject
Set Init = Me
End Function
Public Property Get Name()
Name = m_name
End Property
Public Property Get NameVersion()
Set NameVersion = m_nameVersion
End Property
Public Property Get DriverVersion()
Set DriverVersion = m_driverVersion
End Property
Public Property Get ParseObject()
Set ParseObject = m_parseObject
End Property
End Class
Public Default Function Init(sNameRegex, sVersionRegex, sNameMinVersion, sNameMaxVersion, oVersionRangeArray)
m_computerId = "."
m_nameRegexStr = sNameRegex
Set m_nameRegex = New RegExp
m_nameRegex.Pattern = m_nameRegexStr
m_versionRegexStr = sVersionRegex
If Not IsNull(m_versionRegexStr) Then
If Len(m_versionRegexStr) > 0 Then
Set m_versionRegex = New RegExp
m_versionRegex.Pattern = m_versionRegexStr
m_versionRegex.Global = True
End If
End If
Set m_nameMinVersion = GetDriverVersionObj(sNameMinVersion)
Set m_nameMaxVersion = GetDriverVersionObj(sNameMaxVersion)
m_driverVersionArr = oVersionRangeArray
DriverCollection = Array()
IsNativeClient = False
Set Init = Me
End Function
Private Function CheckVersion(oVersion, vMinVersion, vMaxVersion)
Dim vVersion: Set vVersion = GetDriverVersionObj(oVersion)
CheckVersion = vVersion.CompareTo(vMinVersion) >= 0 And (vVersion.ToString() = MAX_DRIVER_VERSION_STR Or vMinVersion.CompareTo(vMaxVersion) = 0 Or vVersion.CompareTo(vMaxVersion) < 0)
End Function
Public Function MatchName(sDriverName)
Dim matches
Set matches = m_nameRegex.Execute(sDriverName)
If matches.Count > 0 Then
MatchName = True
Exit Function
End If
MatchName = False
End Function
Public Function GetNameVersion(sDriverName)
Dim sVersion, matches
If Not IsNull(m_versionRegexStr) Then
Set matches = m_versionRegex.Execute(sDriverName)
If matches.Count > 0 Then
sVersion = matches(0).Value
End If
End If
If Not IsNull(sVersion) Then
Set GetNameVersion = GetDriverVersionObj(sVersion)
Else
Set GetNameVersion = GetDriverVersionObj("0")
End If
End Function
Public Function CheckNameVersion(oVersion)
CheckNameVersion = CheckVersion(oVersion, m_nameMinVersion, m_nameMaxVersion)
End Function
Public Function CheckDriverVersion(oVersion)
CheckDriverVersion = False
Dim driverVersionItem
For Each driverVersionItem In m_driverVersionArr
If CheckVersion(oVersion, driverVersionItem.StartVersion, driverVersionItem.EndVersion) Then
CheckDriverVersion = True
Exit For
End If
Next
End Function
Public Function GetDriverVersion(sDriverName)
Dim computerId, driverPath, sDllVersion, objFSO, oRegistry
Dim HKEY_LOCAL_MACHINE: HKEY_LOCAL_MACHINE = &H80000002
Set oRegistry = GetObject("winmgmts:\\" & m_computerId & "\root\default:StdRegProv")
oRegistry.GetStringValue HKEY_LOCAL_MACHINE, "SOFTWARE\ODBC\ODBCINST.INI\" & sDriverName, "Driver", driverPath
Set objFSO = CreateObject("Scripting.FileSystemObject")
sDllVersion = objFSO.GetFileVersion(driverPath)
Set GetDriverVersion = (new DriverVersion)(sDllVersion)
End Function
Public Sub AddDriver(ByRef oDriver)
ReDim Preserve DriverCollection(UBound(DriverCollection) + 1)
Set DriverCollection(UBound(DriverCollection)) = oDriver
End Sub
Public Sub ResetState()
DriverCollection = Array()
End Sub
Public Property Get DriverNameRegex()
DriverNameRegex = m_nameRegexStr
End Property
Public Property Get NameMinVersion()
Set NameMinVersion= m_nameMinVersion
End Property
Public Property Get NameMaxVersion()
Set NameMaxVersion = m_nameMaxVersion
End Property
Public Property Get VersionRangeCollection()
VersionRangeCollection = m_driverVersionArr
End Property
Public Property Get CollectionIsEmpty()
CollectionIsEmpty = UBound(DriverCollection) < 0
End Property
End Class
Private Sub Class_Initialize()
Set m_selectorRules(0) = (new DriverSelectorRule)("^SQL\sServer\sNative\sClient\s\d{1,8}(\.\d{1,8})?$","\d{1,8}(\.\d{1,8})?$", "11.0", "11.0",_
Array(_
(new VersionRange)("2011.110.6020.0", MAX_DRIVER_VERSION_STR),_
(new VersionRange)("2011.110.5592.0", "2011.110.6000.0"),_
(new VersionRange)("2011.110.5347.0", "2011.110.5522.0")))
m_selectorRules(0).IsNativeClient = True
Set m_selectorRules(1) = (new DriverSelectorRule)("^ODBC\sDriver\s\d{1,8}(\.\d{1,8})?\sfor\sSQL\sServer$", "\d{1,8}(\.\d{1,8})?(?=\sfor\sSQL\sServer$)", "11.0", MAX_DRIVER_VERSION_STR,_
Array(_
(new VersionRange)("2014.120.4219.0", MAX_DRIVER_VERSION_STR),_
(new VersionRange)("2014.120.2546.0", "2014.120.4000.0")))
Set m_selectorRules(2) = (new DriverSelectorRule)("^SQL\sServer\sNative\sClient\s\d{1,8}(\.\d{1,8})?$", "\d{1,8}(\.\d{1,8})?$","0", MAX_DRIVER_VERSION_STR, Array((new VersionRange)("0", MAX_DRIVER_VERSION_STR)))
m_selectorRules(2).IsNativeClient = True
Set m_selectorRules(3) = (new DriverSelectorRule)("^ODBC\sDriver\s\d{1,8}(\.\d{1,8})?\sfor\sSQL\sServer$", "\d{1,8}(\.\d{1,8})?(?=\sfor\sSQL\sServer$)", "0", MAX_DRIVER_VERSION_STR, Array((new VersionRange)("0", MAX_DRIVER_VERSION_STR)))
' Select most appropriate driver from filtered installed system (ODBC) drivers.
' select all available drivers
' Filter by driver name and version
' Intersected: check client and server have shared protocols
' tcp and np (Named Pipes) can work by network, sm can work only locally
' driver priorities: sm, tcp; np not used
'
' For native client need common client and server protocols
'
' Select latest namespace with highest version
' Select client protocols
' Select server protocols (enabled)
' Intersect client and server protocols
' Select odbc drivers related to inresected protocols
' We have 3 possible drivers:
' - ODBC (all protocols enabled)
' - Native Client (tcp, sm, np protocols)
' only native client can choose protocols through registry settings
' - Microsoft SQL Server (all protocols enabled)
Public Sub ProcessDrivers(enabledServerProtocols)
Dim ri, i, isSelected, currentSelect, oError
Set oError = New Error
Call ResetState()
On Error Resume Next
Dim client: Set client = GetClientParameters()
Dim ncProtocolsAreIntersected: ncProtocolsAreIntersected = ArraysAreIntersected(enabledServerProtocols, client.Protocols)
Call ProcessSystemOdbcDrivers(ncProtocolsAreIntersected)
oError.Save
If oError.Number <> 0 Then
Call AddError(oError)
m_selectedDriverName = m_defaultDriverName
m_processed = true
Exit Sub
End If
On Error GoTo 0
Dim driver: Set driver = SelectFreshDriver(m_selectorRules)
If Not driver Is Nothing Then
If driver.ParseObject.IsNativeClient Then
client.IsTcpProtocolEnabled = ArrayContains(client.Protocols, "tcp")
client.IsSharedMemoryProtocolEnabled = ArrayContains(client.Protocols, "sm")
End If
End If
If Not driver Is Nothing Then
client.DriverName = driver.Name
End If
' for compatibility fill deprecated
m_selectedDriverName = client.DriverName
m_ncli_ForceProtocolEncryption = client.IsForceProtocolEncryption
m_ncli_TrustServerCertificate = client.IsTrustServerCertificate
m_ncli_tcpProtocolEnabled = client.IsTcpProtocolEnabled
m_ncli_smProtocolEnabled = client.IsSharedMemoryProtocolEnabled
m_processed = True
End Sub
Private Function SelectFreshDriver(rules)
Set SelectFreshDriver = Nothing
Dim selected: Set selected = Nothing
Dim rule: Set selected = Nothing
Dim driver: Set selected = Nothing
For Each rule In rules
For Each driver In rule.DriverCollection
' select first if not selected
If selected Is Nothing Then
Set selected = driver
Else
' select fresh, compare driver with selected by name and
' version
if driver.NameVersion.CompareTo(selected.NameVersion) >= 0 And _
driver.DriverVersion.CompareTo(selected.DriverVersion) >= 0 Then
Set selected = driver
End If
End If
Next
' all rules have order by priority, if rule contains drivers, then
' selected with high priority
if Not selected Is Nothing Then
Exit For
End If
Next
Set SelectFreshDriver = selected
End Function
Private Function GetClientParameters()
Set GetClientParameters = Nothing
Dim params: Set params = (New ConnectionParams)(".")
params.Namespace = GetNsNameWithHighestVersion()
Call GetNativeClientSettings(params.Namespace, params)
params.Protocols = GetEnabledNativeClientProtocols(params.Namespace)
Set GetClientParameters = params
End Function
Private Sub ProcessOdbcDriver(driverName, protocolsAreIntersected)
Dim isInstalled, oRegistry, oNameVersion, oDriverVersion
Set oRegistry = GetObject("winmgmts:\\.\root\default:StdRegProv")
oRegistry.GetStringValue HKEY_LOCAL_MACHINE, "SOFTWARE\ODBC\ODBCINST.INI\ODBC Drivers", driverName, isInstalled
Dim selectorRule
For Each selectorRule In m_selectorRules
If (isInstalled <> "Installed") Then
Exit For
End If
If selectorRule.MatchName(driverName) Then
If (Not selectorRule.IsNativeClient) Or (selectorRule.IsNativeClient And protocolsAreIntersected) Then
Set oNameVersion = selectorRule.GetNameVersion(driverName)
If (selectorRule.CheckNameVersion(oNameVersion)) Then
Set oDriverVersion = selectorRule.GetDriverVersion(driverName)
If (selectorRule.CheckDriverVersion(oDriverVersion)) Then
selectorRule.AddDriver((new DriverItem)(driverName, oNameVersion, oDriverVersion, selectorRule))
Exit For
End If
End If
End If
End If
Next
End Sub
' Get available drivers in the system and fill rules drivers
' TODO: rename method
Private Sub ProcessSystemOdbcDrivers(protocolsAreIntersected)
Dim oRegistry, driverNames, paramValueTypes, i, oError
Set oError = New Error
Set oRegistry = GetObject("winmgmts:\\" & m_computerId & "\root\default:StdRegProv")
Call oRegistry.EnumValues(HKEY_LOCAL_MACHINE, "SOFTWARE\ODBC\ODBCINST.INI\ODBC Drivers", driverNames, paramValueTypes)
On Error Resume Next
Dim driverName
For Each driverName In driverNames
oError.Clear
Call ProcessOdbcDriver(driverName, protocolsAreIntersected)
oError.Save
If oError.Number <> 0 Then
Call AddError(oError)
End If
Next
End Sub
Private Function GetNsNameWithHighestVersion()
Dim rootNs: rootNs = "winmgmts:{impersonationLevel=impersonate}!\\.\root\Microsoft\SqlServer"
Dim oWMI: Set oWMI = GetObject(rootNs)
Dim namespaces: Set namespaces = oWMI.ExecQuery("SELECT Name FROM __NAMESPACE WHERE Name LIKE 'ComputerManagement%'")
Dim currentName: currentName = ""
Dim objItem
For Each objItem In namespaces
If objItem.Name > currentName Then
currentName = objItem.Name
End If
Next
GetNsNameWithHighestVersion = rootNs & "\" & currentName
End Function
Private Sub GetNativeClientSettings(namespaceFullName, params)
On Error Resume Next
Err.Clear
Dim sNamespace, oWMI, objClasses, sState, sTargetComputer
Set oWMI = GetObject(namespaceFullName)
Set objClasses = Nothing
' query can fail, because native client not supported
Set objClasses = oWMI.ExecQuery("SELECT FlagName, FlagValue FROM ClientSettingsGeneralFlag")
' do nothing, its ok, just skip this settings
If (Err.Number <> 0) Or (objClasses Is Nothing) Then
Exit Sub
End If
Dim objItem
For Each objItem in objClasses
Select Case objItem.FlagName
Case "Force protocol encryption"
'm_ncli_ForceProtocolEncryption = objItem.FlagValue
params.IsForceProtocolEncryption = objItem.FlagValue
Case "Trust Server Certificate"
'm_ncli_TrustServerCertificate = objItem.FlagValue
params.IsTrustServerCertificate = objItem.FlagValue
End Select
Next
End Sub
Private Function GetEnabledNativeClientProtocols(namespaceFullName)
On Error Resume Next
Err.Clear
Dim oWMI: Set oWMI = GetObject(namespaceFullName)
Dim oQuery: Set oQuery = oWMI.ExecQuery("SELECT ProtocolName, ProtocolOrder FROM ClientNetworkProtocol")
If (oQuery.Count > 0) And (Err.Number Is 0) Then
Dim protocolsArr: Set protocolsArr = New ArrayList
Dim protocolItem
For Each protocolItem In oQuery
If protocolItem.ProtocolOrder > 0 Then
protocolsArr.Add LCase(protocolItem.ProtocolName)
End If
Next
GetEnabledNativeClientProtocols = protocolsArr.ItemsArray
Else
GetEnabledNativeClientProtocols = Array()
End If
End Function
Private Sub AddError(oError)
Dim newSize
newSize = UBound(ErrorCollection) + 1
ReDim Preserve ErrorCollection(newSize)
ErrorCollection(newSize) = "[" & (oError.Number and 65535) & "][" & oError.Source & "] " & oError.Description
End Sub
Private Sub ResetState()
Dim pr
ErrorCollection = Array()
For Each pr In m_selectorRules
pr.ResetState()
Next
End Sub
Private Sub ThrowIfNotProcessed()
If Not m_processed Then
Call Err.Raise(SCRIPT_EVENT_ID, "", "Drivers are not processed. Call 'ProcessDrivers' first.")
End If
End Sub
Public Property Get HasErrors()
HasErrors = UBound(ErrorCollection) >= 0
End Property
Public Property Get DriverName()
ThrowIfNotProcessed
DriverName = m_selectedDriverName
End Property
Public Property Get UseFqdn()
ThrowIfNotProcessed
UseFqdn = m_ncli_ForceProtocolEncryption And Not m_ncli_TrustServerCertificate
End Property
Public Property Get ClientTcpProtocolEnabled()
ThrowIfNotProcessed
ClientTcpProtocolEnabled = m_ncli_tcpProtocolEnabled
End Property
Public Property Get ClientSharedMemoryProtocolEnabled()
ThrowIfNotProcessed
ClientSharedMemoryProtocolEnabled = m_ncli_smProtocolEnabled
End Property
End Class
Class ConnectionParams
Public ComputerName
Public DriverName
Public Namespace
Public IsForceProtocolEncryption
Public IsTrustServerCertificate
Public IsTcpProtocolEnabled
Public IsSharedMemoryProtocolEnabled
Public Protocols
Public Default Function Init(sComputerName)
ComputerName = sComputerName
DriverName = DriverSelector_DEFAULT_DRIVER_NAME
Namespace = ""
IsForceProtocolEncryption = False
IsTrustServerCertificate = False
IsTcpProtocolEnabled = True
IsSharedMemoryProtocolEnabled = True
Set Init = Me
End Function
End Class
Class UniqueCollection
Private m_dict
Public Default Function Init()
Set m_dict = CreateObject("Scripting.Dictionary")
Set Init = Me
End Function
Public Sub PutItem(item)
If Not m_dict.Exists(item) Then
m_dict.add item, ""
End If
End Sub
Public Function Exists(item)
Exists = m_dict.Exists(item)
End Function
Public Function GetItems()
GetItems = m_dict.Keys()
End Function
Public Sub Clear()
m_dict.RemoveAll()
End Sub
End Class
Class SqlServerTcpIpSettings
Private m_listenAllIPs
Private m_ipSettings
Public Default Function Init(listenAllIps, ipSettings)
m_listenAllIPs = listenAllIps
Set m_ipSettings = ipSettings
Set Init = Me
End Function
Public Property Get ListenAllIPs()
ListenAllIPs = m_listenAllIPs
End Property
Public Property Get IpSettings()
Set IpSettings = m_ipSettings
End Property
End Class
Function EscapeConnStringValue (ByVal strValue)
On Error Resume Next
Err.Clear
EscapeConnStringValue = "{" + Replace(strValue, "}", "}}") + "}"
End Function
Function EscapeWQLString (ByVal strValue)
ON ERROR RESUME NEXT
Err.Clear
EscapeWQLString = Replace(Replace(strValue, "\", "\\"), "'", "\'")
End Function
For i = 0 to Ubound(aReserved)
If vEncDec = "enc" Then
vString = Replace(vString,aReserved(i,0),aReserved(i,1))
End If
If vEncDec = "dec" Then
vString = Replace(vString,aReserved(i,1),aReserved(i,0))
End If
Next
furlEncode = vString
End Function
Class UtcDateOperations
Private m_wbemDate
Private Sub Class_Initialize()
Set m_wbemDate = CreateObject("WbemScripting.SWbemDateTime")
End Sub
Public Function DateToIso8601UtcString(objDate)
DateToIso8601UtcString = _
DatePart("yyyy", objDate) & "-" &_
Right("0" & DatePart("m", objDate), 2) & "-" &_
Right("0" & DatePart("d", objDate), 2) &_
"T" &_
Right("0" & DatePart("h", objDate), 2) & ":" &_
Right("0" & DatePart("n", objDate), 2) & ":" &_
Right("0" & DatePart("s", objDate), 2) &_
"Z"
End Function
Public Function Iso8601UtcStringToDate(sDate)
Dim oRegex: Set oRegex = New RegExp
oRegex.Pattern = "^(\d{4})-(\d\d)-(\d\d)(?:T|\s)(\d\d):(\d\d):(\d\d)Z$"
Dim matches: Set matches = oRegex.Execute(sDate)
If matches.Count = 0 Then
Err.Raise 4000, "", "Input date string is in invalid format"
End If
Dim year: year = CInt(matches(0).SubMatches(0))
Dim month: month = CInt(matches(0).SubMatches(1))
Dim day: day = CInt(matches(0).SubMatches(2))
Dim hour: hour = CInt(matches(0).SubMatches(3))
Dim minute: minute = CInt(matches(0).SubMatches(4))
Dim second: second = CInt(matches(0).SubMatches(5))
Public Function GetUtcNow()
m_wbemDate.SetVarDate(Now())
GetUtcNow = CDate(m_wbemDate.GetVarDate(false))
End function
Public Function CheckSecondsElapsed(sDate, nSeconds)
Dim utcNow: utcNow = Me.GetUtcNow
Dim utcDate: utcDate = Iso8601UtcStringToDate(sDate)
CheckSecondsElapsed = DateDiff("s",utcDate,utcNow) > nSeconds
End function
End Class
Function GetCacheKey(stateMpPrefix, sKeyName)
Dim oApi: Set oApi = CreateObject("MOM.ScriptAPI")
Dim regKey: regKey = oAPI.GetScriptStateKeyPath(ManagementGroupID)
regKey = regKey & "\" & stateMpPrefix & "\" & sKeyName
GetCacheKey = regKey
End Function
Function GetStringValueFromCache(sKeyName, sValueName, cacheExpirationTime)
Dim stateMpPrefix: stateMpPrefix = "SQL2014MP"
Dim sDateValueName: sDateValueName = sValueName & "_CreationTime"
Dim udo: Set udo = New UtcDateOperations
Dim oReg: Set oReg = New Registry
Dim regKey: regKey = GetCacheKey(stateMpPrefix, sKeyName)
Dim lErrCode
Dim sDate: sDate = oReg.ReadStringValue(regKey, sDateValueName, lErrCode)
If lErrCode <> oReg.SUCCESS Then
Exit Function
End If
Dim sValue: sValue = oReg.ReadStringValue(regKey, sValueName, lErrCode)
If lErrCode <> oReg.SUCCESS Then
Exit Function
End If
If udo.CheckSecondsElapsed(sDate, cacheExpirationTime) Then
Exit Function
End If
GetStringValueFromCache = sValue
End Function
Function PutStringValueToCache(sKeyName, sValueName, sValue)
Dim stateMpPrefix: stateMpPrefix = "SQL2014MP"
Dim sDateValueName: sDateValueName = sValueName & "_CreationTime"
Dim udo: Set udo = New UtcDateOperations
Dim oReg: Set oReg = New Registry
Dim regKey: regKey = GetCacheKey(stateMpPrefix, sKeyName)
Call oReg.WriteStringValue(regKey, sValueName, sValue)
Call oReg.WriteStringValue(regKey, sDateValueName, udo.DateToIso8601UtcString(dUtcNow))
End Function
Function GetWMISingleValue(wmiProvider, query, propertyName)
Dim oQuery: Set oQuery = wmiProvider.ExecQuery(query)
If oQuery.Count = 0 Then
Err.Raise 4000, "", "Query '" & query & "' didn't return any objects"
End If
Dim colSettings: Set colSettings = GetFirstItemFromWMIQuery(oQuery)
GetWMISingleValue = colSettings.Properties_.Item(propertyName).Value
End Function
Class HostNameData
Public HostName
Public IsClustered
Public Default Function Init(sHostName, bIsClustered)
Me.HostName = sHostName
Me.IsClustered = bIsClustered
Set Init = Me
End function
End Class
Function GetSqlServerHostName(strDNSComputerName, instanceName, namespace)
Set GetSqlServerHostName = Nothing
Dim serviceName: serviceName = GetSQLServiceName(instanceName)
Dim escapedServiceName: escapedServiceName = EscapeWQLString(serviceName)
Dim wmiProvider: Set wmiProvider = ConnectToWMI(strDNSComputerName, "ROOT\Microsoft\SqlServer\" + namespace)
Dim isClustered: isClustered = GetWmiSingleValue(wmiProvider, "SELECT PropertyNumValue FROM SqlServiceAdvancedProperty WHERE PropertyName = 'CLUSTERED' AND SqlServiceType = 1 AND ServiceName = '" & escapedServiceName & "'", "PropertyNumValue")
Dim hostName
If isClustered = 0 Then
hostName = GetWmiSingleValue(wmiProvider, "SELECT HostName FROM SqlService WHERE SQLServiceType = 1 AND ServiceName = '" & escapedServiceName & "'", "HostName")
Else
hostName = GetWmiSingleValue(wmiProvider, "SELECT PropertyStrValue FROM SqlServiceAdvancedProperty WHERE PropertyName = 'VSNAME' AND SqlServiceType = 1 AND ServiceName = '" & escapedServiceName & "'", "PropertyStrValue")
End If
Set GetSqlServerHostName = (New HostNameData)(hostName, CBool(isClustered))
End Function
Function EscapeCacheValueName(name)
EscapeCacheValueName = Replace(name, "_", "__")
End Function
Function GetSqlServerHostNameEx(strDNSComputerName, instanceName, namespace)
Set GetSqlServerHostNameEx = Nothing
Dim cacheExpirationTime: cacheExpirationTime = 7200
Dim hostValueName: hostValueName = EscapeCacheValueName(strDNSComputerName)
Dim isClusteredValueName: isClusteredValueName = hostValueName & "_IsClustered"
Dim isClusteredStr
On Error Resume Next
Call Err.Clear()
Dim hostName: hostName = GetStringValueFromCache("SqlHostNames", hostValueName, cacheExpirationTime)
If Err.Number = 0 And Not IsEmpty(hostName) Then
isClusteredStr = GetStringValueFromCache("SqlHostNames", isClusteredValueName, cacheExpirationTime)
If Err.Number = 0 And Not IsEmpty(isClusteredStr) Then
If isClusteredStr = "0" Or isClusteredStr = "1" Then
Set GetSqlServerHostNameEx = (New HostNameData)(hostName, CBool(isClusteredStr))
Exit Function
End If
End If
End If
Call Err.Clear()
On Error GoTo 0
Dim hostNameData: Set hostNameData = GetSqlServerHostName(strDNSComputerName, instanceName, namespace)
If hostNameData.IsClustered Then
isClusteredStr = "1"
Else
isClusteredStr = "0"
End If
On Error Resume Next
Call PutStringValueToCache("SqlHostNames", hostValueName, hostNameData.HostName)
Call PutStringValueToCache("SqlHostNames", isClusteredValueName, isClusteredStr)
Call Err.Clear()
Set GetSqlServerHostNameEx = hostNameData
End function
Function GetLocalHostName()
Dim wshShell: Set wshShell = CreateObject( "WScript.Shell" )
GetLocalHostName = wshShell.ExpandEnvironmentStrings("%COMPUTERNAME%")
End Function
Function DelimitSqlIdentifier(identifier)
DelimitSqlIdentifier = "[" + Replace(identifier, "]", "]]") + "]"
End Function
Function SqlTcpPortIsEmpty(tcpPort)
SqlTcpPortIsEmpty = (IsEmpty(tcpPort) Or tcpPort = "" Or tcpPort = "0")
End Function
Function GetDataSource(server, tcpPort)
Dim dataSource : dataSource = server
If Not SqlTcpPortIsEmpty(tcpPort) Then
Dim nameParts : nameParts = Split(dataSource, "\")
dataSource = nameParts(0) & "," & tcpPort
End If
GetDataSource = dataSource
End Function
Function BuildDataSourceFromParts(computerName, instanceName, tcpPort)
Dim dataSource : dataSource = computerName
If instanceName <> "MSSQLSERVER" Then
dataSource = computerName & "\" & instanceName
End If
BuildDataSourceFromParts = GetDataSource(dataSource, tcpPort)
End Function
Function GetConnectionString(driverName, dataSource, databaseName)
GetConnectionString = "Driver=" & EscapeConnStringValue(driverName) & ";Server=" & EscapeConnStringValue(dataSource) & ";Database=" & EscapeConnStringValue(databaseName) & ";Trusted_Connection=yes;"
End Function
Function GetEnabledSqlServerProtocols(namespaceName, computerName, instanceName)
Dim oWMI: Set oWMI = GetObject("winmgmts:\\" & computerName & "\root\Microsoft\SqlServer\" & namespaceName)
Dim oQuery: Set oQuery = oWMI.ExecQuery("SELECT ProtocolName, Enabled FROM ServerNetworkProtocol WHERE InstanceName = '"& EscapeWQLString(instanceName) &"'")
If oQuery.Count > 0 Then
Dim protocolsArr: Set protocolsArr = New ArrayList
Dim protocolItem
For Each protocolItem In oQuery
If protocolItem.Enabled Then
protocolsArr.Add LCase(protocolItem.ProtocolName)
End If
Next
GetEnabledSqlServerProtocols = protocolsArr.ItemsArray
Else
GetEnabledSqlServerProtocols = Array()
End If
End Function
Function GetSqlServerTcpIpSettings(instanceName, computerName)
ON ERROR RESUME NEXT
Dim oWMI, oQuery, tcpItem
Dim i, j
Dim uc: Set uc = (new UniqueCollection)()
Dim isListenAll: isListenAll = False
Dim ipSettings: Set ipSettings = CreateObject("Scripting.Dictionary")
Set oWMI = GetObject("winmgmts:\\" & computerName & "\root\Microsoft\SqlServer\"& SQL_WMI_NAMESPACE)
Set oQuery = oWMI.ExecQuery("SELECT * FROM ServerNetworkProtocolProperty WHERE ProtocolName = 'Tcp' AND InstanceName = '"& EscapeWQLString(instanceName) &"' AND PropertyName = 'ListenOnAllIPs'")
If oQuery.Count >0 Then
Dim isListenAllObj: Set isListenAllObj = GetFirstItemFromWMIQuery(oQuery)
If isListenAllObj.PropertyNumVal = 1 Then
isListenAll = True
End If
End If
Dim tcpPorts, tcpPort, processedPort
If isListenAll Then
Set oQuery = oWMI.ExecQuery("SELECT * FROM ServerNetworkProtocolProperty WHERE ProtocolName = 'Tcp' AND InstanceName = '"& EscapeWQLString(instanceName) &"' AND IPAddressName = 'IPAll' AND (PropertyName = 'TcpPort' OR PropertyName = 'TcpDynamicPorts') AND PropertyStrVal <> ''")
If oQuery.Count > 0 Then
For Each tcpItem In oQuery
tcpPorts = Split(tcpItem.PropertyStrVal,",")
For Each tcpPort In tcpPorts
processedPort = Trim(tcpPort)
If Not SqlTcpPortIsEmpty(processedPort) Then
uc.PutItem(processedPort)
End If
Next
Next
ipSettings.add "IPAll", uc.GetItems()
End If
Else
Set oQuery = oWMI.ExecQuery("SELECT * FROM ServerNetworkProtocolProperty WHERE ProtocolName = 'Tcp' AND InstanceName = '"& EscapeWQLString(instanceName) &"' AND IPAddressName <> '' AND PropertyName = 'Enabled' AND PropertyNumVal = 1")
Dim ipItem
If oQuery.Count > 0 Then
For Each ipItem In oQuery
Dim ipAddressName : ipAddressName = ipItem.IPAddressName
Dim oQuery2 : Set oQuery2 = oWMI.ExecQuery("SELECT * FROM ServerNetworkProtocolProperty WHERE ProtocolName = 'Tcp' AND InstanceName = '" & EscapeWQLString(instanceName) & "' AND IPAddressName = '" & EscapeWQLString(ipAddressName) & "' AND PropertyName = 'IpAddress' AND PropertyStrVal != ''")
If oQuery2.Count > 0 Then
Dim ipAddress : ipAddress = GetFirstItemFromWMIQuery(oQuery2).PropertyStrVal
Dim oQuery3: Set oQuery3 = oWMI.ExecQuery("SELECT * FROM ServerNetworkProtocolProperty WHERE ProtocolName = 'Tcp' AND InstanceName = '"& EscapeWQLString(instanceName) &"' AND IPAddressName = '"& EscapeWQLString(ipAddressName) &"' AND (PropertyName = 'TcpPort' OR PropertyName = 'TcpDynamicPorts') AND PropertyStrVal <> ''")
If oQuery3.Count > 0 Then
uc.Clear()
Dim portItem
For Each portItem In oQuery3
tcpPorts = Split(portItem.PropertyStrVal,",")
For Each tcpPort In tcpPorts
processedPort = Trim(tcpPort)
If Not SqlTcpPortIsEmpty(processedPort) Then
uc.PutItem(processedPort)
End If
Next
ipSettings.add ipAddress, uc.GetItems()
Next
End If
End If
Next
End If
End If
Set GetSqlServerTcpIpSettings = (new SqlServerTcpIpSettings)(isListenAll, ipSettings)
End Function
Function GetTcpPortsString(ByVal inputDataSource)
On Error Resume Next
Dim computerName, instanceName, pathArray
pathArray = Split(inputDataSource, "\")
computerName = pathArray(0)
instanceName = "MSSQLSERVER"
If (UBound(pathArray) > 0) Then
instanceName = pathArray(1)
End If
Dim tcpIpSettings: Set tcpIpSettings = GetSqlServerTcpIpSettings(instanceName, computerName)
Dim upc : Set upc = (new UniqueCollection)()
Dim ip, port
For Each ip In tcpIpSettings.IpSettings.Keys
For Each port In tcpIpSettings.IpSettings.Item(ip)
upc.PutItem port
Next
Next
GetTcpPortsString = Join(upc.GetItems(),", ")
End Function
Public Sub SqlTestDestination(dbConnection, hostName, instanceName, isADODB)
Dim queryServerName, queryInstanceName
Dim destinationTestQuery: destinationTestQuery = "select CAST(SERVERPROPERTY('MachineName') AS nvarchar(128)) as ServerName, @@servicename as InstanceName"
On Error Goto 0
If 0 = Err.number Then
Dim queryResult
If isADODB Then
Set queryResult = dbConnection.ExecuteQueryTE(destinationTestQuery)
Else
Set queryResult = dbConnection.Execute(destinationTestQuery)
End If
If Not queryResult.EOF Then
queryServerName = UCase(queryResult("ServerName").Value)
queryInstanceName = UCase(queryResult("InstanceName").Value)
If (UCase(hostName) = queryServerName) And (UCase(instanceName) = queryInstanceName) Then
Exit Sub
End If
End If
dbConnection.Close()
Err.Raise 16389, "", "Connection target check failed: connected to " & hostName & "\" & instanceName & ", but got " & queryServerName & "\" & queryInstanceName & "."
End If
End Sub
Sub TryToConnectAndValidate(connectionObj, connectionString, timeout, hostName, instanceName, isADODB)
On Error GoTo 0
If isADODB Then
connectionObj.Open connectionString, "", timeout
Else
if (connectionObj.Mode <> 0) then
connectionObj.Close()
end if
connectionObj.ConnectionTimeout = timeout
connectionObj.Open connectionString
End If
SqlTestDestination connectionObj, hostName, instanceName, isADODB
End Sub
Function FormatConnectionErrorMessage(dataSource, lastError)
FormatConnectionErrorMessage = "Connection to data source '" & dataSource & "' failed: " & lastError.Description
End Function
Function SmartConnectWithoutSQLADODB(ByVal inputDataSource, ByVal tcpPort, ByVal machineName, ByVal instanceName, ByVal databaseName)
On Error Resume Next
Set SmartConnectWithoutSQLADODB = Nothing
Dim dbMasterConnection, dataSource, connectionString, errorMessage
Dim targetName : targetName = inputDataSource
Dim lastError : Set lastError = new Error
Dim errorMessageList : Set errorMessageList = New ArrayList
Dim ds: Set ds = New DriverSelector
Dim netBiosHostNameData: Set netBiosHostNameData = GetSqlServerHostNameEx(machineName, instanceName, SQL_WMI_NAMESPACE)
lastError.Save
If lastError.Number <> 0 Then
GlobalErrorList.Add "Cannot get target instance machine's NetBios host name." &_
"Computer name: " & machineName & vbNewLine &_
"Error number: " & lastError.Number & vbNewLine &_
"Error description:" & lastError.Description
Set SmartConnectWithoutSQLADODB = Nothing
Exit Function
End If
Dim netBiosHostName: netBiosHostName = netBiosHostNameData.HostName
Dim dnsHostName: dnsHostName = Split(machineName, ".")(0)
Dim enabledServerProtocols: enabledServerProtocols = GetEnabledSqlServerProtocols(SQL_WMI_NAMESPACE, machineName, instanceName)
If Not HandleErrorContinueEx("Cannot get a list of enabled Sql Server protocols", instanceName) Then
Exit Function
End If
ds.ProcessDrivers(enabledServerProtocols)
Dim selectedDriverName: selectedDriverName = ds.DriverName
Dim useFqdn: useFqdn = ds.UseFqdn
Dim hasErrors: hasErrors = ds.HasErrors
Set dbMasterConnection = CreateObject("ADODB.Connection")
Dim connStr: connStr = inputDataSource
Call Err.Clear()
' Sql Server Shared Memory protocol require usage of host's NetBios name.
' Shared Memory usually the first in the driver's priority list.
' Rebuild data source string in the case of standalone Sql Server instance, NetBios host name differs
' from DNS host name and enabled Shared Memory on Client and Server
If netBiosHostName <> dnsHostName And Not netBiosHostNameData.IsClustered And ArrayContains(enabledServerProtocols, "sm") And ds.ClientSharedMemoryProtocolEnabled Then
Dim localHostName: localHostName = GetLocalHostName()
If Not HandleErrorContinue("Cannot get local machine's NetBios name") Then
Exit Function
End If
If netBiosHostName = localHostName Then
connStr = "lpc:" & BuildDataSourceFromParts(netBiosHostName, instanceName, "")
End If
End If
'Connect using Sql Browser
dataSource = GetDataSource(connStr, "")
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate dbMasterConnection, connectionString, 15, netBiosHostName, instanceName, False
lastError.Save
If lastError.Number = 0 Then
Set SmartConnectWithoutSQLADODB = dbMasterConnection
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
'Try to connect through tcp/ip protocol only if it is enabled
If ArrayContains(enabledServerProtocols, "tcp") And ds.ClientTcpProtocolEnabled Then
Dim dsComputerName: dsComputerName = Split(inputDataSource, "\")(0)
Dim tcpIpSettings : Set tcpIpSettings = GetSqlServerTcpIpSettings(instanceName, dsComputerName)
If useFqdn Then
targetName = machineName
Else
targetName = dsComputerName
End if
Dim ip, port
'Use ports configured for all interfaces
If tcpIpSettings.ListenAllIPs Then
For Each port In tcpIpSettings.IpSettings.item("IPAll")
dataSource = GetDataSource(targetName, port)
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate dbMasterConnection, connectionString, 10, netBiosHostName, instanceName, False
lastError.Save
If lastError.Number = 0 Then
Set SmartConnectWithoutSQLADODB = dbMasterConnection
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
Next
Else
'Use one port from each interface
Dim upc : Set upc = (new UniqueCollection)()
For Each ip In tcpIpSettings.IpSettings.Keys
port = tcpIpSettings.IpSettings.Item(ip)(0)
If Not upc.Exists(port) Then
upc.PutItem port
dataSource = GetDataSource(targetName, port)
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate dbMasterConnection, connectionString, 10, netBiosHostName, instanceName, False
lastError.Save
If lastError.Number = 0 Then
Set SmartConnectWithoutSQLADODB = dbMasterConnection
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
End If
Next
'Go through all interfaces and ports
If Not useFqdn Then
For Each ip In tcpIpSettings.IpSettings.Keys
For Each port In tcpIpSettings.IpSettings.Item(ip)
dataSource = GetDataSource(ip, port)
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate dbMasterConnection, connectionString, 10, netBiosHostName, instanceName, False
lastError.Save
If lastError.Number = 0 Then
Set SmartConnectWithoutSQLADODB = dbMasterConnection
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
Next
Next
End If
End If
End If
GlobalErrorList.Add "Cannot connect to the target Sql Server instance. Connection log:" & vbNewLine & Join(errorMessageList.ItemsArray, vbNewLine)
End Function
Function SmartConnect(cnADOConnection, inputDataSource, tcpPort, machineName, instanceName, databaseName)
On Error Resume Next
SmartConnect = False
Dim dataSource, connectionString, errorMessage
Dim targetName : targetName = inputDataSource
Dim lastError : Set lastError = new Error
Dim errorMessageList : Set errorMessageList = New ArrayList
Dim ds: Set ds = New DriverSelector
Dim netBiosHostNameData: Set netBiosHostNameData = GetSqlServerHostNameEx(machineName, instanceName, SQL_WMI_NAMESPACE)
lastError.Save
If lastError.Number <> 0 Then
GlobalErrorList.Add "Cannot get target instance machine's NetBios host name." &_
"Computer name: " & machineName & vbNewLine &_
"Error number: " & lastError.Number & vbNewLine &_
"Error description:" & lastError.Description
Exit Function
End If
Dim netBiosHostName: netBiosHostName = netBiosHostNameData.HostName
Dim dnsHostName: dnsHostName = Split(machineName, ".")(0)
Dim enabledServerProtocols: enabledServerProtocols = GetEnabledSqlServerProtocols(SQL_WMI_NAMESPACE, netBiosHostName, instanceName)
If Not HandleErrorContinueEx("Cannot get a list of enabled Sql Server protocols", instanceName) Then
Exit Function
End If
ds.ProcessDrivers(enabledServerProtocols)
Dim selectedDriverName: selectedDriverName = ds.DriverName
Dim useFqdn: useFqdn = ds.UseFqdn
Dim hasErrors: hasErrors = ds.HasErrors
Dim connStr: connStr = inputDataSource
Call Err.Clear()
' Sql Server Shared Memory protocol require usage of host's NetBios name.
' Shared Memory usually the first in the driver's priority list.
' Rebuild data source string in the case of standalone Sql Server instance, NetBios host name differs
' from DNS host name and enabled Shared Memory on Client and Server
If netBiosHostName <> dnsHostName And Not netBiosHostNameData.IsClustered And ArrayContains(enabledServerProtocols, "sm") And ds.ClientSharedMemoryProtocolEnabled Then
Dim localHostName: localHostName = GetLocalHostName()
If Not HandleErrorContinue("Cannot get local machine's NetBios name") Then
Exit Function
End If
If netBiosHostName = localHostName Then
connStr = "lpc:" & BuildDataSourceFromParts(netBiosHostName, instanceName, "")
End If
End If
'Connect using Sql Browser
dataSource = GetDataSource(connStr, "")
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate cnADOConnection, connectionString, 15, netBiosHostName, instanceName, True
lastError.Save
If lastError.Number = 0 Then
SmartConnect = True
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
'Try to connect through tcp/ip protocol only if it is enabled
If ArrayContains(enabledServerProtocols, "tcp") And ds.ClientTcpProtocolEnabled Then
Dim dsComputerName: dsComputerName = Split(inputDataSource, "\")(0)
Dim tcpIpSettings : Set tcpIpSettings = GetSqlServerTcpIpSettings(instanceName, dsComputerName)
If useFqdn Then
targetName = machineName
Else
targetName = dsComputerName
End if
Dim ip, port
'Use ports configured for all interfaces
If tcpIpSettings.ListenAllIPs Then
For Each port In tcpIpSettings.IpSettings.item("IPAll")
dataSource = GetDataSource(targetName, port)
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate cnADOConnection, connectionString, 10, netBiosHostName, instanceName, True
lastError.Save
If lastError.Number = 0 Then
SmartConnect = True
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
Next
Else
'Use one port from each interface
Dim upc : Set upc = (new UniqueCollection)()
For Each ip In tcpIpSettings.IpSettings.Keys
port = tcpIpSettings.IpSettings.Item(ip)(0)
If Not upc.Exists(port) Then
upc.PutItem port
dataSource = GetDataSource(targetName, port)
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate cnADOConnection, connectionString, 10, netBiosHostName, instanceName, True
lastError.Save
If lastError.Number = 0 Then
SmartConnect = True
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
End If
Next
'Go through all interfaces and ports
If Not useFqdn Then
For Each ip In tcpIpSettings.IpSettings.Keys
For Each port In tcpIpSettings.IpSettings.Item(ip)
dataSource = GetDataSource(ip, port)
connectionString = GetConnectionString(selectedDriverName, dataSource, databaseName)
lastError.Clear
TryToConnectAndValidate cnADOConnection, connectionString, 10, netBiosHostName, instanceName, True
lastError.Save
If lastError.Number = 0 Then
SmartConnect = True
Exit Function
Else
errorMessageList.Add FormatConnectionErrorMessage(dataSource, lastError)
End If
Next
Next
End If
End If
End If
GlobalErrorList.Add "Cannot connect to the target Sql Server instance. Connection log:" & vbNewLine & Join(errorMessageList.ItemsArray, vbNewLine)
End Function
'#Include File:Common-SecondPart.vbs
''''''''''''''''''''''''''''''''''''
''' Class OpsMgrAPI
''''''''''''''''''''''''''''''''''''
Class OpsMgrAPI
Dim scriptAPI
Private Sub Class_Initialize()
On Error Resume Next
Set scriptAPI = CreateObject("MOM.ScriptAPI")
HandleError("Cannot create object 'MOM.ScriptAPI' (OpsMgrAPI.Class_Initialize).")
End Sub
End Class
'#Include File:SQLADODB.vbs
''''''''''''''''''''''''''''''''''''
''' Class ADODB
''''''''''''''''''''''''''''''''''''
Class ADODB
Dim ADOConnection
Private Sub Class_Initialize()
On Error Resume Next
Set ADOConnection = CreateObject("ADODB.Connection")
HandleError("Cannot create object 'ADODB.Connection' (ADODB.Class_Initialize).")
End Sub
Public Function Open(connectionString, provider, connectionTimeout)
On Error Resume Next
Open = false
if connectionString = "" Then
Err.Raise SCRIPT_EVENT_ID, "ADODB.Open()", "Argument 'connectionString' cannot be empty." , "", 0
End If
If (ADOConnection.State <> 0) Then
ADOConnection.Close()
End If
ADOConnection.ConnectionString = connectionString
If provider <> "" Then
ADOConnection.Provider = provider
End If
if connectionTimeout <= 0 Then
ADOConnection.ConnectionTimeout = 30
Else
ADOConnection.ConnectionTimeout = connectionTimeout
End If
HandleError("Cannot initialize ADODB connection (ADODB.Open).")
On Error GoTo 0
ADOConnection.Open()
End Function
Public Function ExecuteQuery(query)
On Error Resume Next
Set ExecuteQuery = Nothing
Set ExecuteQuery = ADOConnection.Execute(query)
End Function
Public Function ExecuteQueryTE(query)
On Error GoTo 0
Set ExecuteQueryTE = ADOConnection.Execute(query)
End Function
' Params argument should be a single parameter or a single-dimensional Array
' Parameter places in query should be marked as ?
' Order must be kept
'
' If you want to use named ones, you should start
' your query with something like "declare @paramname int = ?"
' and then use @paramname in the query text.
'
' int, bigint and NVarChar are supported for now, additional support can be added in the AddParam sub
Public Function ExecuteQueryWithParams(query, params)
On Error Resume Next
Set ExecuteQueryWithParams = Nothing
' Create a new Command object
Dim Cmd
Set Cmd = CreateObject("ADODB.Command")
' Specify the connection
Cmd.ActiveConnection = ADOConnection
' Specify command type and text
Cmd.CommandText = query
Cmd.CommandType = 1 ' adCmdText
' Create a new parameter
Dim i
If IsArray(params) Then
For i = 0 To UBound(params)
AddParam Cmd, params(i)
Next
Else
AddParam Cmd, params
End If
Set ExecuteQueryWithParams = Cmd.Execute
End Function
Sub AddParam(cmd, value)
Dim Parameter
Select Case VarType(value)
Case 2 ' int
Set Parameter = cmd.CreateParameter(, 3, 1, , value) ' , adInteger, adParamInput
Case 3 ' long
Set Parameter = cmd.CreateParameter(, 20, 1, , value) ' , adBigInt, adParamInput
Case 8 ' string
Set Parameter = cmd.CreateParameter(, 202, 1, Max(Len(value), 1), value) ' , adVarWChar, adParamInput
Case else
HandleError("Unknown parameter type: " & VarType(value))
End Select
cmd.Parameters.Append Parameter
End Sub
Function Max(a,b)
Max = a
If b > a then Max = b
End Function
Public Function Close()
On Error Resume Next
if Not IsNull(ADOConnection) Then
ADOConnection.Close()
HandleError("Cannot close ADODB connection (ADODB.Close).")
End If
End Function
Public Function HandleOpenConnectionErrorContinue(database, serverName, sqlInstanceName)
HandleOpenConnectionErrorContinue = true
if Err.number <> 0 Then
HandleOpenConnectionErrorContinue = false
Dim oError : Set oError = new Error
oError.Save()
Dim instanceIsRunning : instanceIsRunning = IsServiceRunning(sqlInstanceName, serverName)
On Error Resume Next
oError.Raise()
Dim errorStr
if ((Err.number and 65535) = 16389 or (Err.number and 65535) = 3661) and instanceIsRunning Then
Call LogError("Cannot login to database [" & serverName & "][" & sqlInstanceName & ":" & database & "]: " & oError.Description)
Err.Clear
ElseIf (instanceIsRunning) Then
Call LogFormattedError("Cannot open ADODB connection. (Connection string: '" & ADOConnection.ConnectionString & "'.). Error Description: " & oError.Description, sqlInstanceName)
Err.Clear
Else
Err.Clear
End If
On Error Goto 0
End If
End Function
Public Function HandleExecutionQueryErrorContinue(query, serverName, sqlInstanceName)
HandleExecutionQueryErrorContinue = true
if Err.number <> 0 Then
HandleExecutionQueryErrorContinue = false
Dim oError : Set oError = new Error
oError.Save()
Dim instanceIsRunning : instanceIsRunning = IsServiceRunning(sqlInstanceName, serverName)
On Error Resume Next
oError.Raise()
if ((Err.number and 65535) = 16389 or (Err.number and 65535) = 3661) and instanceIsRunning Then
Call LogError("Cannot execute query [" & serverName & "][" & sqlInstanceName & ":" & ADOConnection.DefaultDatabase & "]: " & oError.Description)
Err.Clear
ElseIf (instanceIsRunning) Then
Call LogFormattedError("Cannot execute query: " & FormatLogQuery(query), sqlInstanceName)
Err.Clear
Else
Err.Clear
End If
On Error Goto 0
End If
End Function
Public Function IsServiceRunning(sInstance, serverName)
Dim sServiceName : sServiceName = GetSQLServiceName(sInstance)
Dim oService, sObjectString
sObjectString = "winmgmts:{impersonationLevel=impersonate}!\\" & serverName & "\root\cimv2"
On Error Resume Next
Err.Clear
'We want to do our own error handling here. No WMIGetObject().
Set oService = GetObject(sObjectString & ":Win32_Service.Name='" & EscapeWQLString(sServiceName) & "'")
If Err.Number <> 0 Then
IsServiceRunning = false
Else
If oService.State = "Running" Then
IsServiceRunning = true
Else
IsServiceRunning = false
End If
End If
Set oService = Nothing
On Error GoTo 0
End Function
Private Function FormatLogQuery(query)
Dim regex: Set regex = New RegExp
regex.Pattern = "\s{2,}"
regex.Global = True
FormatLogQuery = regex.Replace(query," ")
End function
Private Sub LogError(message)
On Error Resume Next
GlobalErrorList.Add message
End Sub
Private Sub LogFormattedError(message, instanceName)
Dim formattedMessage: formattedMessage = FormatErrorMessage(message, instanceName)
Call GlobalErrorList.Add(formattedMessage)
End Sub
End Class
'#Include File:GetDiskVolumeInformation.vbs
'''''''''''''''''''''''''''''''''''''
''' Assembling Volumes Collection '''
'''''''''''''''''''''''''''''''''''''
Const WMI_LINK = "winmgmts:{impersonationLevel=impersonate}!\\{0}\root\cimv2"
Const MOUNT_POINTS_QUERY = "select Directory, Volume from Win32_Mountpoint"
Const VOLUME_QUERY = "select DeviceID, Name, Caption, Label, FileSystem, Capacity, FreeSpace from Win32_Volume"
Const WMI_ERROR = "Can't get WMI object on "
Const MOUNT_POINT_ERROR = "Can't select Mount Point data from WMI object on "
Const DISK_VOLUME_ERROR = "Can't select Disk Volume data from WMI object on "
Const FIND_BY_DEVICEID_ERROR = "Couldn't find volume by DeviceID "
'Main class which is supposed to be used in external scripts
Class DiskVolumeInformation
Public MountPoints
Public Volumes
Public Default Function Initialize(serverName)
Set MountPoints = SortMountPoints(GetMountPoints(serverName))
Set Volumes = GetVolumes(serverName, MountPoints)
Set Initialize = Me
End Function
End Class
Class MountPoint
Public Name
Public Caption
Public DeviceID
End Class
Function GetMountPoints(serverName)
On Error Resume Next
Dim MountPoints : Set MountPoints = CreateObject("Scripting.Dictionary")
Dim objWMIService : Set objWMIService = GetObject(Replace(WMI_LINK, "{0}", serverName))
HandleError(WMI_ERROR & serverName)
Dim colItems : Set colItems = objWMIService.ExecQuery(MOUNT_POINTS_QUERY)
HandleError(MOUNT_POINT_ERROR & serverName)
Dim objItem
For Each objItem In colItems
Dim point : set point = new MountPoint
Dim dirStartIndex : dirStartIndex = InStr(objItem.Directory, """") + 1
Dim dirEndIndex : dirEndIndex = InStr(dirStartIndex, objItem.Directory, """")
Dim Directory
Directory = LCase(Replace(Mid(objItem.Directory, dirStartIndex, dirEndIndex - dirStartIndex), "\\", "\"))
point.Caption = Directory
point.Name = point.DeviceID & point.Caption
MountPoints.Add point.Name, point
Next
Set GetMountPoints = MountPoints
End Function
Class DiskVolume
Public FileSystem
Public Capacity
Public FreeSpace
Public Name
Public DeviceID
Public Label
Public Paths()
Private pSize
Public Function AddPath(path)
pSize = pSize + 1
ReDim Preserve Paths(pSize)
Paths(pSize) = path
End Function
Private Sub Class_Initialize
pSize = -1
End Sub
End Class
Function GetVolumes(serverName, aMountPoints)
Dim Volumes : Set Volumes = CreateObject("Scripting.Dictionary")
Dim objWMIService : Set objWMIService = GetObject(Replace(WMI_LINK, "{0}", serverName))
HandleError(WMI_ERROR & serverName)
Dim colItems : Set colItems = objWMIService.ExecQuery(VOLUME_QUERY)
HandleError(DISK_VOLUME_ERROR & serverName)
Dim objItem
For Each objItem In colItems
Dim vol : set vol = new DiskVolume
vol.Capacity = objItem.Capacity
vol.FileSystem = objItem.FileSystem
vol.FreeSpace = objItem.FreeSpace
vol.Name = objItem.Name
vol.AddPath objItem.Caption
vol.DeviceID = objItem.DeviceID
vol.Label = objItem.Label
Volumes.Add vol.DeviceID, vol
Next
Function SearchVolumeCaption(aVolumes, aCaption)
Dim k
For Each k In aVolumes.Keys
Dim p
For p = 0 To UBound(aVolumes.Item(k).Paths)
If aVolumes.Item(k).Paths(p) = aCaption Then
SearchVolumeCaption = True
Exit Function
End If
Next
Next
SearchVolumeCaption = false
End Function
Function SearchVolumeDeviceID(aVolumes, aDeviceID)
Dim k
For Each k In aVolumes.Keys
If aVolumes.Item(k).DeviceID = aDeviceID Then
Set SearchVolumeDeviceID = aVolumes.Item(k)
Exit Function
End If
Next
Set SearchVolumeDeviceID = Nothing
End Function
Function MergeMountPointsIntoVolumeList(Volumes, MountPoints)
Dim key
For Each key in MountPoints.Keys
Dim Caption : Caption = MountPoints.Item(key).Caption
'Adding backslash in the end of the path is necessary for
'matching the path stored in volumes class, also it prevents
'invalid equality among folders with long composite names like
'"C:\Program Files" and "C:\Program"
If Right(Caption, 1) <> "\" Then
Caption = Caption & "\"
End If
'Search for Mount Points which are not listed among Volumes
Dim found : found = SearchVolumeCaption(Volumes, Caption)
If Not found Then
'Search for the equivalent Volume
Dim equalVolume : Set equalVolume = SearchVolumeDeviceID(Volumes, MountPoints.Item(key).DeviceID)
HandleError(FIND_BY_DEVICEID_ERROR & MountPoints.Item(key).DeviceID)
If Not equalVolume Is Nothing And Not IsNull(equalVolume) And Not IsEmpty(equalVolume) Then
equalVolume.AddPath Caption
End If
End If
Next
End Function
''''''''''''''''''''''''''''''''''''''''''''''''''''''
''' Sorting Mount Points descending by path length '''
''''''''''''''''''''''''''''''''''''''''''''''''''''''
Function SortMountPoints(aMountPoints)
'Convert dictionary to array
Dim MPArray()
Dim i : i = 0
Dim key
For Each key in aMountPoints.Keys
ReDim Preserve MPArray(i)
Set MPArray(i) = aMountPoints.Item(key)
i = i + 1
Next
'Sorting array
Dim SortedArray : SortedArray = SortArray(MPArray)
'Converting back to dictionary
Set SortMountPoints = CreateObject("Scripting.Dictionary")
Dim a
For Each a in SortedArray
SortMountPoints.Add a.Caption, a
Next
End Function
Function SortArray(arr)
If UBound(arr) < 1 Then
SortArray = arr
Exit Function
End If
Dim pivotIndex : pivotIndex = UBound(arr) 'Select last element as pivot
Dim pivot : Set pivot = arr(pivotIndex)
ReDim Preserve arr(UBound(arr) - 1) 'Remove pivot element
Dim longerList()
Dim shorterList()
Dim longerListSize : longerListSize = 0
Dim shorterListSize : shorterListSize = 0
'Put the elements with longer path than pivot to longerList
'and the elements with shorter path to shorterList
Dim a
For Each a in arr
If Len(a.Caption) > Len(pivot.Caption) Then
ReDim Preserve longerList(longerListSize)
Set longerList(longerListSize) = a
longerListSize = longerListSize + 1
Else
ReDim Preserve shorterList(shorterListSize)
Set shorterList(shorterListSize) = a
shorterListSize = shorterListSize + 1
End If
Next
Dim sortedLess, sortedGreater
If longerListSize > 0 Then
sortedLess = SortArray(longerList)
Else
sortedLess = longerList
End If
If shorterListSize > 0 Then
sortedGreater = SortArray(shorterList)
Else
sortedGreater = shorterList
End If
SortArray = MergeArrays(sortedLess, longerListSize, pivot, sortedGreater, shorterListSize)
End Function
Function MergeArrays(lessArray, longerListSize, pivotNode, greaterArray, shorterListSize)
Dim arraySize : arraySize = longerListSize + shorterListSize
Dim ResultArray()
ReDim ResultArray(arraySize)
Dim i
For i = 0 To UBound(ResultArray)
If i < longerListSize Then
Set ResultArray(i) = lessArray(i)
ElseIf i > longerListSize Then
Set ResultArray(i) = greaterArray(i - longerListSize - 1)
Else
Set ResultArray(i) = pivotNode
End If
Next
MergeArrays = ResultArray
End Function
'#Include File:SQLFreeSpaceCalculator.vbs
''''''''''''''''''''''''''''''''''''
''' Class DBSpaceCalculator
''''''''''''''''''''''''''''''''''''
Class DBSpaceCalculator
Public FreeSpaceMB
Public FreeSpacePercent
Public FreeSpaceAutoGrowMB
Public FreeSpaceAutoGrowPercent
Public SizeMB
Public AllocatedFreeSpaceMB
Public AllocatedFreeSpacePercent
Public DiskFreeSpaceMB
Public DiskFreeSpacePercent
Public UsedSpaceMB
Public UsedSpacePercent
Public TotalSpaceMB
Public IsTotalAutoGrow
Private VolumeGrowthArray
Private DiskVolumesInfo
Public AzureMaxFileSizeMB
Public IsError
Public ErrorMessage
Private ReExpProvider
Private FSOProvider
Public LastShared
Public NoSizeAllocationInfo
public LastTotalSize
Private Sub Class_Initialize()
Set ReExpProvider = new regexp
ReExpProvider.IgnoreCase = true
Public Default Function Init(diskVolumesInformation)
Set DiskVolumesInfo = diskVolumesInformation
Dim key
For Each key in DiskVolumesInfo.Volumes.Keys
VolumeGrowthArray.Add DiskVolumesInfo.Volumes.Item(key).DeviceID, 0
Next
Set Init = Me
End Function
Private Function GetVolumeByPath(filePath)
Dim mpKey
For Each mpKey In DiskVolumesInfo.MountPoints.Keys
If InStr(LCase(filePath), LCase(DiskVolumesInfo.MountPoints.Item(mpKey).Caption)) > 0 Then
Set GetVolumeByPath = DiskVolumesInfo.Volumes.Item(DiskVolumesInfo.MountPoints.Item(mpKey).DeviceID)
Exit Function
End If
Next
Set GetVolumeByPath = Nothing
End Function
Public Function CalculateFxFreeSpace(dbFilesSet,takefullSet)
Dim HasFiles : HasFiles = False
Dim GroupUsedSpace : GroupUsedSpace = 0
Dim GroupAvailableSpace : GroupAvailableSpace = 0
Dim DiskArray : Set DiskArray = CreateObject("Scripting.Dictionary")
Do While Not dbFilesSet.EOF
HasFiles = True
Dim filePath : filePath = dbFilesSet("physical_name").Value
Dim fileSize : fileSize = CDbl(dbFilesSet("fileSize").Value)
GroupUsedSpace = GroupUsedSpace + fileSize
Dim GrowthSpaceMaxSize : GrowthSpaceMaxSize = CDbl(1.79769313486232E307)
Dim fileMaxSize : fileMaxSize = CLng(dbFilesSet("fileMaxSize").Value)
if fileMaxSize <> -1 And fileMaxSize <> 0 Then
GrowthSpaceMaxSize = fileMaxSize - fileSize
End If
Dim FreeSpaceDisk : FreeSpaceDisk = GetFreeSpaceDisk(filePath, fileSize)
Dim LogicalDriveGrowth : LogicalDriveGrowth = 0
If GrowthSpaceMaxSize > FreeSpaceDisk Then
LogicalDriveGrowth = FreeSpaceDisk
Else
LogicalDriveGrowth = GrowthSpaceMaxSize
End If
Dim DicKey: DicKey = ""
if Me.LastShared = "" Then
Dim vol : Set vol = GetVolumeByPath(filePath)
DicKey = vol.DeviceID
Else
DicKey = Me.LastShared
End If
if Not DiskArray.Exists(DicKey) Then
DiskArray.Add DicKey, LogicalDriveGrowth
Else
Dim newSize: newSize = DiskArray.Item(DicKey) + LogicalDriveGrowth
If newSize > FreeSpaceDisk Then
newSize = FreeSpaceDisk
End If
DiskArray.Item(DicKey) = newSize
End if
If Not takeFullSet Then
Exit Do
End If
dbFilesSet.MoveNext
Loop
Dim logicalDriveGrowthKey
for each logicalDriveGrowthKey in DiskArray
GroupAvailableSpace = GroupAvailableSpace + DiskArray.Item(logicalDriveGrowthKey)
Next
Me.FreeSpaceMB = GroupAvailableSpace
if Me.FreeSpaceMB < 0 Then
Me.FreeSpaceMB = 0
End If
If HasFiles Then
Me.FreeSpacePercent = GroupAvailableSpace / (GroupUsedSpace + GroupAvailableSpace) * 100
if Me.FreeSpacePercent < 0 Then
Me.FreeSpacePercent = 0
End If
Else
Me.FreeSpacePercent = 100
End If
End Function
Public Function CalculateFdFreeSpace(dbFilesSet)
Dim HasFiles : HasFiles = False
Dim GroupUsedSpace : GroupUsedSpace = 0
Dim GroupAvailableSpace : GroupAvailableSpace = 0
Dim DiskArray : Set DiskArray = CreateObject("Scripting.Dictionary")
Do While Not dbFilesSet.EOF
HasFiles = True
Dim filePath : filePath = dbFilesSet("physical_name").Value
Dim fileSize : fileSize = CDbl(dbFilesSet("fileSize").Value)
GroupUsedSpace = GroupUsedSpace + fileSize
Dim GrowthSpaceMaxSize : GrowthSpaceMaxSize = CDbl(1.79769313486232E307)
Dim fileMaxSize : fileMaxSize = CLng(dbFilesSet("fileMaxSize").Value)
if fileMaxSize <> -1 And fileMaxSize <> 0 Then
GrowthSpaceMaxSize = fileMaxSize - fileSize
End If
Dim FreeSpaceDisk : FreeSpaceDisk = GetFreeSpaceDisk(filePath, fileSize)
If (Me.IsError) Then
Exit Function
End If
Dim vol : Set vol = GetVolumeByPath(filePath)
Dim LogicalDriveGrowth : LogicalDriveGrowth = 0
If GrowthSpaceMaxSize > FreeSpaceDisk Then
LogicalDriveGrowth = FreeSpaceDisk
Else
LogicalDriveGrowth = GrowthSpaceMaxSize
End If
if Not DiskArray.Exists(vol.DeviceID) Then
DiskArray.Add vol.DeviceID, LogicalDriveGrowth
Else
Dim newSize: newSize = DiskArray.Item(vol.DeviceID) + LogicalDriveGrowth
If newSize > FreeSpaceDisk Then
newSize = FreeSpaceDisk
End If
DiskArray.Item(vol.DeviceID) = newSize
End if
dbFilesSet.MoveNext
Loop
Dim logicalDriveGrowthKey
for each logicalDriveGrowthKey in DiskArray
GroupAvailableSpace = GroupAvailableSpace + DiskArray.Item(logicalDriveGrowthKey)
Next
Me.FreeSpaceMB = GroupAvailableSpace
if Me.FreeSpaceMB < 0 Then
Me.FreeSpaceMB = 0
End If
If HasFiles Then
Me.FreeSpacePercent = GroupAvailableSpace / (GroupUsedSpace + GroupAvailableSpace) * 100
if Me.FreeSpacePercent < 0 Then
Me.FreeSpacePercent = 0
End If
Else
Me.FreeSpacePercent = 100
End If
End Function
Private Function GetFreeSpaceDisk(filePath, fileSize)
On Error Resume Next
Dim oMatches
Me.LastShared = ""
'Azure blob
'http://<storage account>.blob.core.windows.net/<container>/<blob>
'https://<storage account>.blob.core.windows.net/<container>/<blob>
ReExpProvider.Pattern = "^(https?:\/\/)?([\da-z\.-]+)\.([a-z\.]{2,6})([\/\w \.-]*)*\/?$"
Set oMatches = ReExpProvider.Execute(filePath)
If Not IsNull(oMatches) And Not oMatches is Nothing And Not IsEmpty(oMatches) Then
If oMatches.Count > 0 Then
GetFreeSpaceDisk = Me.AzureMaxFileSizeMB - fileSize
Me.LastShared = filePath
Exit Function
End If
End If
'UNC = \\<hostname>\<sharename>[\<objectname>]*
ReExpProvider.Pattern = "^\\{2}(?:\?\\)?([^\\\:]+)\\([^\\\:]+)((?:\\[^\\\:]+)+(?:\\)?)+$"
Set oMatches = ReExpProvider.Execute(filePath)
If Not IsNull(oMatches) And Not oMatches is Nothing And Not IsEmpty(oMatches) Then
If oMatches.Count > 0 Then
Dim oMatch
Dim errMessage : errMessage = ""
Set oMatch = oMatches(0)
If Not IsNull(oMatch) And Not oMatch is Nothing And Not IsEmpty(oMatch) Then
If oMatch.SubMatches.Count >= 2 Then
Dim SMBShared
SMBShared = "\\" + oMatch.SubMatches(0) + "\" + oMatch.SubMatches(1)
Me.LastShared = SMBShared
GetFreeSpaceDisk = FSOProvider.getDrive(SMBShared).FreeSpace / 1024 /1024
Me.LastTotalSize = FSOProvider.getDrive(SMBShared).TotalSize / 1024 /1024
if Not IsEmpty(GetFreeSpaceDisk) Then
Exit Function
End If
errMessage = "Cannot get FreeSpace of SMB shared '" & SMBShared & "' ('" & filePath & "')" & vbCrLf
End If
End If
If (len(errMessage) = 0) Then
errMessage = "Cannot get FreeSpace of SMB shared '" & filePath & "'" & vbCrLf
End if
Me.ErrorMessage = Me.ErrorMessage + errMessage
Me.IsError = true
Exit Function
End if
end if
Dim vol : Set vol = GetVolumeByPath(filePath)
If Not vol is Nothing And Not IsNull(vol) And Not IsEmpty(vol) Then
GetFreeSpaceDisk = vol.FreeSpace / 1024 / 1024
Me.LastTotalSize = vol.Capacity / 1024 /1024
Exit Function
End If
Dim diskName : diskName = Left(filePath, 1)
Dim disk : Set disk = FSO.GetDrive(diskName)
HandleError("Cannot get object of disk '" & diskName & "'")
GetFreeSpaceDisk = disk.FreeSpace / 1024 / 1024
Me.LastTotalSize = disk.TotalSize / 1024 /1024
HandleError("Cannot get FreeSpace of disk '" & diskName & "'")
End Function
Public Function CalculateFreeSpace(dbFilesSet)
'On Error Resume Next
Dim GroupAvailableSpace : GroupAvailableSpace = 0
Dim GroupUsedSpace : GroupUsedSpace = 0
Dim GroupAvailableSpaceNoGrowth : GroupAvailableSpaceNoGrowth = 0
Dim GroupUsedSpaceNoGrowth : GroupUsedSpaceNoGrowth = 0
Dim noUsedSpaceInfo: noUsedSpaceInfo = False
Do While Not dbFilesSet.EOF
Dim filePath : filePath = dbFilesSet("physical_name").Value
Dim IsAutoGrow : IsAutoGrow = CBool(dbFilesSet("IsAutoGrow").Value = 1)
IsTotalAutoGrow = CBool(IsTotalAutoGrow Or IsAutoGrow)
if (IsAutoGrow) Then
Dim NumberGrowths : NumberGrowths = 0
Dim GrowthSpaceDisk : GrowthSpaceDisk = 0
Dim GrowthSpaceMaxSize : GrowthSpaceMaxSize = CDbl(1.79769313486232E307)
Dim fileMaxSize : fileMaxSize = CLng(dbFilesSet("fileMaxSize").Value)
Dim fileGrowth : fileGrowth = CLng(dbFilesSet("fileGrowth").Value)
Dim isPercentGrowth : isPercentGrowth = dbFilesSet("isPercentGrowth").Value
if Not isPercentGrowth Then 'FileGrowth not in Percent
fileGrowth = fileGrowth / 128
Else
fileGrowth = fileGrowth / 100
End If
if fileMaxSize <> -1 Then 'FileMaxSize Not Unlimited
'when the FileSize approaches the MaxSize, a large FileGrowth value can cause an error
'if the file tries to grow past the MaxSize value.
'calculate actual growth capacity
if isPercentGrowth Then 'FileGrowth in Percent
NumberGrowths = Fix(Log(fileMaxSize / fileSize) / Log(1 + fileGrowth))
GrowthSpaceMaxSize = (fileSize * ((1 + fileGrowth) ^ NumberGrowths)) - fileSize
Else
GrowthSpaceMaxSize = Fix((fileMaxSize - fileSize) / fileGrowth) * fileGrowth
End If
End If
Dim FreeSpaceDisk : FreeSpaceDisk = GetFreeSpaceDisk(filePath, fileSize)
If (Me.IsError) Then
Exit Function
End If
'when the FileSize approaches the limits of the logical disk, a large FileGrowth
'value can cause an error if the file tries to grow past the physical limit
'calculate actual growth capacity on disk
if isPercentGrowth Then 'FileGrowth in Percent
NumberGrowths = Fix(Log((FreeSpaceDisk + fileSize) / fileSize) / Log(1 + fileGrowth))
GrowthSpaceDisk = (fileSize * ((1 + fileGrowth) ^ NumberGrowths)) - fileSize
Else
GrowthSpaceDisk = Fix(FreeSpaceDisk / fileGrowth) * fileGrowth
End If
'with auto grow the growth allowed for a file with a max size is the minimum
'of either up to the max size or up to the limits of the logical drive.
'With max size unlimited, the growth allowed is not limited except by the limits of the logical drive
Dim GrowthAllowed : GrowthAllowed = GrowthSpaceDisk
if GrowthSpaceMaxSize < GrowthSpaceDisk Then
GrowthAllowed = GrowthSpaceMaxSize
End If
'if there is no access to SMB share space shouldn't take into.
if Not (IsEmpty(FreeSpaceDisk)) Then
'add growth allowed for this file to hosting logical drive up to the free disk space capacity.
Dim LogicalDriveGrowth : LogicalDriveGrowth = 0
if (len(Me.LastShared) > 0) Then
'save if no such SMB shared
if( VolumeGrowthArray.Exists(Me.LastShared)) Then
LogicalDriveGrowth = VolumeGrowthArray.Item(Me.LastShared)
Else
VolumeGrowthArray.Add Me.LastShared, 0
End if
'save result
If (LogicalDriveGrowth + GrowthAllowed > FreeSpaceDisk) Then
VolumeGrowthArray.Item(Me.LastShared) = FreeSpaceDisk
Else
VolumeGrowthArray.Item(Me.LastShared) = LogicalDriveGrowth + GrowthAllowed
End If
'all job done
Me.LastShared = ""
Else
LogicalDriveGrowth = GetLogicalDriveGrowth(filePath)
If LogicalDriveGrowth + GrowthAllowed > FreeSpaceDisk Then
SetLogicalDriveGrowth filePath, FreeSpaceDisk
Else
SetLogicalDriveGrowth filePath, LogicalDriveGrowth + GrowthAllowed
End If
End If
End If
End If
dbFilesSet.MoveNext
Loop
Dim logicalDriveGrowthKey
for each logicalDriveGrowthKey in VolumeGrowthArray
GroupAvailableSpace = GroupAvailableSpace + VolumeGrowthArray.Item(logicalDriveGrowthKey)
Next
If GroupAvailableSpace <> 0 And Me.FreeSpaceMB > 0 Then
Me.FreeSpacePercent = Me.FreeSpaceMB / GroupAvailableSpace * 100
Else
Me.FreeSpaceMB = 0
Me.FreeSpacePercent = 0
End If
If GroupAvailableSpaceNoGrowth <> 0 And Me.FreeSpaceAutoGrowMB > 0 Then
Me.FreeSpaceAutoGrowPercent = Me.FreeSpaceAutoGrowMB / GroupAvailableSpaceNoGrowth * 100
Else
Me.FreeSpaceAutoGrowMB = 0
Me.FreeSpaceAutoGrowPercent = 0
End If
Me.AllocatedFreeSpaceMB = Me.SizeMB - Me.UsedSpaceMB
if Me.AllocatedFreeSpaceMB < 0 Then
Me.AllocatedFreeSpaceMB = 0
End If
Me.DiskFreeSpaceMB = Me.FreeSpaceMB - Me.AllocatedFreeSpaceMB
if Me.DiskFreeSpaceMB < 0 Then
Me.DiskFreeSpaceMB = 0
End If
If TotalSpaceMB <> 0 Then
Me.AllocatedFreeSpacePercent = (Me.AllocatedFreeSpaceMB / TotalSpaceMB) * 100
Me.DiskFreeSpacePercent = (Me.DiskFreeSpaceMB / TotalSpaceMB) * 100
Me.UsedSpacePercent = (Me.UsedSpaceMB / TotalSpaceMB) * 100
Else
Me.AllocatedFreeSpacePercent = 0
Me.DiskFreeSpacePercent = 0
Me.UsedSpacePercent = 0
End If
End Function
Function GetLogicalDriveGrowth(filePath)
Dim vol : Set vol = GetVolumeByPath(filePath)
If Not IsNull(vol) And Not IsEmpty(vol) Then
GetLogicalDriveGrowth = VolumeGrowthArray.Item(vol.DeviceID)
End If
End Function
Function SetLogicalDriveGrowth(filePath, value)
Dim vol : Set vol = GetVolumeByPath(filePath)
If Not IsNull(vol) And Not IsEmpty(vol) Then
VolumeGrowthArray.Item(vol.DeviceID) = value
End If
End Function
Class DBFxContainerSpaceCalculator
'driver free space MB
Public FreeSpaceMB
'Free space percent
Public FreeSpacePercent
'data/delta
Public DataFilePairSizeMB
Public DriveTotalSize
Public DriveFreeSpace
'
Public MountPoint
'
Private DiskVolumesInfo
Private Sub Class_Initialize()
FreeSpaceMB = 0
FreeSpacePercent = 0
End Sub
Public Default Function Init(diskVolumesInformation)
Set DiskVolumesInfo = diskVolumesInformation
Set Init = Me
End Function
Private Function GetVolumeByPath(filePath)
Dim mpKey
For Each mpKey In DiskVolumesInfo.MountPoints.Keys
If InStr(LCase(filePath), LCase(DiskVolumesInfo.MountPoints.Item(mpKey).Caption)) > 0 Then
Set GetVolumeByPath = DiskVolumesInfo.Volumes.Item(DiskVolumesInfo.MountPoints.Item(mpKey).DeviceID)
Exit Function
End If
Next
Set GetVolumeByPath = Nothing
End Function
Private Function FillInLocalProperties(filePath)
On Error Resume Next
Dim vol : Set vol = GetVolumeByPath(filePath)
If Not IsNull(vol) And Not vol is Nothing And Not IsEmpty(vol) Then
Me.MountPoint = vol.Name
Me.DriveTotalSize = vol.Capacity /1024 /1024
Me.DriveFreeSpace = vol.FreeSpace /1024 /1024
Exit Function
End If
Dim diskName : diskName = Left(filePath, 1)
Dim disk : Set disk = FSO.GetDrive(diskName)
HandleError("Cannot get object of disk '" & diskName & "'")
Me.MountPoint = diskName
HandleError("Cannot get mount point of disk '" & diskName & "'")
Me.DriveTotalSize = disk.TotalSize /1024 /1024
HandleError("Cannot get TotalSize of disk '" & diskName & "'")
Me.DriveFreeSpace = disk.FreeSpace /1024 /1024
HandleError("Cannot get FreeSpace of disk '" & diskName & "'")
End Function
Private Function GetTotalPhysicalMemory(computerName)
Set objWMI = GetObject("winmgmts:\\" & computerName & "\root\cimv2")
Set oQuery = objWMI.ExecQuery ("Select * from Win32_ComputerSystem")
If oQuery.Count > 0 Then
Dim computer
Set computer = GetFirstItemFromWMIQuery(oQuery)
Dim TotalPhysicalMemory
TotalPhysicalMemory = computer.TotalPhysicalMemory /1024/1024/1024
If TotalPhysicalMemory > 16 Then
Me.DataFilePairSizeMB = containerDataFileSize128MB + containerDeltaFileSize8MB
End If
End If
End Function
Public Function CalculateFreeSpace(path, number, allocated, serverName)
'get driver size
Dim containerPath : containerPath = path
Dim NumberOfFiles : NumberOfFiles = number
Dim AllocatedSpace : AllocatedSpace = allocated
if (NumberOfFiles + (Me.DriveFreeSpace \ Me.DataFilePairSizeMB)) >= containerMaxPair Then
DriveFreeSpaceHK = (containerMaxPair-NumberOfFiles)*Me.DataFilePairSizeMB
Else
DriveFreeSpaceHK = (Me.DriveFreeSpace \ Me.DataFilePairSizeMB)*Me.DataFilePairSizeMB
End If
Me.FreeSpaceMB = DriveFreeSpaceHK
Me.FreeSpacePercent = ((DriveFreeSpaceHK)*100)/(DriveFreeSpaceHK + AllocatedSpace)
End Function
End Class'#Include File:GetSQLDBFileGroupFreeSpace.vbs
Function Main()
On Error Resume Next
Logger.LogDebug("Start...")
if WScript.Arguments.Count() <> 6 Then
WScript.Quit()
End If
Dim connectionStr : connectionStr = WScript.Arguments(0)
Dim serverName : serverName = WScript.Arguments(1)
Dim instanceName : instanceName = WScript.Arguments(2)
Dim TcpPort : TcpPort = WScript.Arguments(3)
Dim AzureMaxFileSizeMB : AzureMaxFileSizeMB = CLng(WScript.Arguments(4))
Dim Delay : Delay = CLng(WScript.Arguments(5))
Logger.LogDebug("connectionStr = " & connectionStr & "; ServerName = " & serverName & "; instanceName = " & instanceName & "; Azure Max File Size (MB) = " & AzureMaxFileSizeMB & "; Delay = " & Delay)
Function GetAllDbFileGroupFreeSpace(connectionStr, serverName, instanceName, sTcpPort, AzureMaxFileSizeMB, Delay)
On Error Resume Next
Dim opsMgrAPI
Dim tcp : tcp = sTcpPort
Dim dbMasterConnection
Dim listDatabases
Dim FreeSpaceMB, FreeSpacePercent, FreeSpaceAutoGrowMB, FreeSpaceAutoGrowPercent
Dim spaceErrorMessage : spaceErrorMessage = ""
Dim hasItems: hasItems = False
Set opsMgrAPI = new OpsMgrAPI
Set dbMasterConnection = new ADODB
Dim serviceName , state
serviceName = GetSQLServiceName(instanceName)
state = GetServiceState(serverName, serviceName)
if (state <> "Running") And (state <> "Unknown") Then
Call ReturnEmptyPropertyBag(opsMgrAPI)
Exit Function
End If
Dim res : res = SmartConnect(dbMasterConnection, connectionStr, tcp, serverName, instanceName, "master")
if res = False Then
Call ReturnEmptyPropertyBag(opsMgrAPI)
Exit Function
End If
Dim query : query = _
" SET NOCOUNT ON " & vbCrLf & _
" DECLARE @HasAlwaysOn bit " & vbCrLf & _
" SET @HasAlwaysOn = (SELECT TOP 1 CASE WHEN OBJECT_ID('sys.availability_replicas') IS NOT NULL THEN 1 ELSE 0 END AS HasAlwaysOn " & vbCrLf & _
" FROM master.sys.syscolumns columns where name = 'replica_id' and id = OBJECT_ID('sys.databases')) " & vbCrLf & _
" IF @HasAlwaysOn IS NULL OR @HasAlwaysOn = 0 " & vbCrLf & _
" SELECT name, database_id " & vbCrLf & _
" FROM sys.databases " & vbCrLf & _
" WHERE [state] = 0 " & vbCrLf & _
" AND source_database_id is null AND collation_name is not null AND is_read_only = 0 AND is_in_standby = 0 AND user_access != 1 " & vbCrLf & _
" ELSE " & vbCrLf & _
" SELECT d.name, d.database_id " & vbCrLf & _
" FROM sys.databases d " & vbCrLf & _
" OUTER APPLY( " & vbCrLf & _
" SELECT d2.database_id, drs.is_primary_replica AS db_is_primary_replica " & vbCrLf & _
" , CASE WHEN d2.replica_id IS NULL THEN 0 ELSE 1 END AS is_replica " & vbCrLf & _
" , CASE WHEN drs.is_primary_replica = 1 THEN ar.primary_role_allow_connections ELSE ar.secondary_role_allow_connections END AS role_allow_connections " & vbCrLf & _
" , CASE WHEN drs.is_suspended = 0 THEN -1 ELSE suspend_reason END AS db_suspended_state " & vbCrLf & _
" FROM sys.databases as d2 " & vbCrLf & _
" JOIN sys.dm_hadr_database_replica_states drs ON drs.database_id = d2.database_id " & vbCrLf & _
" JOIN sys.availability_replicas ar on d2.replica_id = ar.replica_id " & vbCrLf & _
" WHERE drs.is_local = 1 AND d2.database_id = d.database_id " & vbCrLf & _
" ) df " & vbCrLf & _
" WHERE d.[state] = 0 " & vbCrLf & _
" AND source_database_id is null AND collation_name is not null AND is_read_only = 0 AND is_in_standby = 0 AND user_access != 1 " & vbCrLf & _
" AND (df.database_id IS NULL OR ((df.is_replica = 0) OR (df.is_replica = 1 AND df.role_allow_connections > 1 AND (db_suspended_state <= 0 OR (db_suspended_state = 5 AND db_is_primary_replica = 1))))) "
Set listDatabases = dbMasterConnection.ExecuteQuery(query)
If dbMasterConnection.HandleExecutionQueryErrorContinue(query, serverName, instanceName) Then
If Not listDatabases.EOF Then
Dim diskVolumesInfo : Set diskVolumesInfo = (new DiskVolumeInformation)(serverName)
Dim dbRows: dbRows = listDatabases.GetRows
Dim rowNumber
For rowNumber = LBound(dbRows, 2) To UBound(dbRows, 2)
Dim databaseName : databaseName = dbRows(0,rowNumber)
Dim databaseID : databaseID = dbRows(1,rowNumber)
dbMasterConnection.ADOConnection.DefaultDatabase = databaseName
If HandleSqlErrorContinue(dbMasterConnection.ADOConnection, "Cannot connect to database '" & databaseName & "'", instanceName) Then
query = "SET NOCOUNT ON " & vbCrLf & _
" SELECT fg.name as fileGroupName, " & vbCrLf & _
" fg.data_space_id as fileGroupId, " & vbCrLf & _
" fg.is_read_only as fileGroupReadOnly, " & vbCrLf & _
" fg.type as fileGroupType " & vbCrLf & _
" FROM sys.filegroups fg WHERE (fg.type = 'FG' OR fg.type = 'FX' OR fg.type = 'FD') AND fg.is_read_only = 0 "
Dim listFileGroup : Set listFileGroup = dbMasterConnection.ExecuteQuery(query)
If dbMasterConnection.HandleExecutionQueryErrorContinue(query, serverName, instanceName) Then
Do While Not listFileGroup.EOF
Dim fileGroupId : fileGroupId = listFileGroup("fileGroupId").Value
Dim fileGroupType : fileGroupType = listFileGroup("fileGroupType").Value
Dim propertyBag : Set propertyBag = opsMgrAPI.scriptAPI.CreateTypedPropertyBag(2)
Select Case fileGroupType
Case "FD"
query = "SET NOCOUNT ON " & vbCrLf & _
"SELECT size / 128.0 as fileSize, " & vbCrLf & _
" CASE WHEN max_size = -1 OR max_size = 268435456 THEN -1 ELSE max_size / 128.0 END as fileMaxSize, " & vbCrLf & _
" physical_name " & vbCrLf & _
" FROM sys.database_files WITH (NOLOCK) " & vbCrLf & _
" WHERE type = 2 AND is_read_only = 0 AND data_space_id = ? "
Dim listFdFiles : Set listFdFiles = dbMasterConnection.ExecuteQueryWithParams(query, Array(fileGroupId))
If dbMasterConnection.HandleExecutionQueryErrorContinue(query, serverName, instanceName) Then
Dim spaceFdCalculator : Set spaceFdCalculator = (new DBSpaceCalculator)(diskVolumesInfo)
spaceFdCalculator.AzureMaxFileSizeMB = AzureMaxFileSizeMB
spaceFdCalculator.CalculateFdFreeSpace(listFdFiles)
If Err.Number = 0 And Not spaceFdCalculator.IsError Then
FreeSpaceMB = spaceFdCalculator.FreeSpaceMB
FreeSpacePercent = spaceFdCalculator.FreeSpacePercent
opsMgrAPI.scriptAPI.AddItem(propertyBag)
hasItems = True
Else
Call HandleErrorContinueEx("Error occured during free space calculation for (database:'" & databaseName & "';data_space_id:" & fileGroupId & ")", instanceName)
If spaceFdCalculator.IsError Then
Call GlobalErrorList.Add(FormatDbErrorMessage(spaceFdCalculator.ErrorMessage, instanceName, databaseName))'"Database: " + databaseName +" Error: "+ spaceFdCalculator.ErrorMessage)
End If
End If
End If
Case "FG"
'mdf files
query = "SET NOCOUNT ON " & vbCrLf & _
"SELECT size / 128.0 as fileSize, " & vbCrLf & _
" FILEPROPERTY(name, 'SpaceUsed') / 128.0 as fileUsed, " & vbCrLf & _
" CASE WHEN max_size = -1 OR max_size = 268435456 THEN -1 ELSE max_size / 128.0 END as fileMaxSize, " & vbCrLf & _
" CASE WHEN growth = 0 THEN 0 ELSE 1 END as IsAutoGrow, " & vbCrLf & _
" is_percent_growth as isPercentGrowth, " & vbCrLf & _
" growth as fileGrowth, " & vbCrLf & _
" physical_name " & vbCrLf & _
" FROM sys.database_files WITH (NOLOCK) " & vbCrLf & _
" WHERE type = 0 AND is_read_only = 0 AND data_space_id = ? "
Dim listFiles : Set listFiles = dbMasterConnection.ExecuteQueryWithParams(query, Array(fileGroupId))
If dbMasterConnection.HandleExecutionQueryErrorContinue(query, serverName, instanceName) Then
Dim spaceCalculator : Set spaceCalculator = (new DBSpaceCalculator)(diskVolumesInfo)
spaceCalculator.AzureMaxFileSizeMB = AzureMaxFileSizeMB
spaceCalculator.CalculateFreeSpace(listFiles)
If Err.Number = 0 AND Not spaceCalculator.IsError Then
FreeSpaceMB = spaceCalculator.FreeSpaceMB
FreeSpacePercent = spaceCalculator.FreeSpacePercent
FreeSpaceAutoGrowMB = spaceCalculator.FreeSpaceAutoGrowMB
FreeSpaceAutoGrowPercent = spaceCalculator.FreeSpaceAutoGrowPercent
opsMgrAPI.scriptAPI.AddItem(propertyBag)
hasItems = True
Else
Call HandleErrorContinueEx("Error occured during free space calculation for (database:'" & databaseName & "';data_space_id:" & fileGroupId & ")", instanceName)
If spaceCalculator.IsError Then
Call GlobalErrorList.Add(FormatDbErrorMessage(spaceCalculator.ErrorMessage, instanceName, databaseName))
End If
End If
End If
Case "FX"
query = "SET NOCOUNT ON " & vbCrLf & _
"SELECT size / 128.0 as fileSize, " & vbCrLf & _
" CASE WHEN max_size = -1 OR max_size = 268435456 THEN -1 ELSE max_size / 128.0 END as fileMaxSize, " & vbCrLf & _
" physical_name " & vbCrLf & _
" FROM sys.database_files WITH (NOLOCK) " & vbCrLf & _
" WHERE is_read_only = 0 AND data_space_id = ? "
Dim listFxFiles : Set listFxFiles = dbMasterConnection.ExecuteQueryWithParams(query, fileGroupId)
If dbMasterConnection.HandleExecutionQueryErrorContinue(query, serverName, instanceName) Then
Dim spaceFxCalculator : Set spaceFxCalculator = (new DBSpaceCalculator)(diskVolumesInfo)
spaceFxCalculator.AzureMaxFileSizeMB = AzureMaxFileSizeMB
Call spaceFxCalculator.CalculateFxFreeSpace(listFxFiles, true)
If Err.Number = 0 And Not spaceFxCalculator.IsError Then
FreeSpaceMB = spaceFxCalculator.FreeSpaceMB
FreeSpacePercent = spaceFxCalculator.FreeSpacePercent
opsMgrAPI.scriptAPI.AddItem(propertyBag)
hasItems = True
Else
Call HandleErrorContinueEx("Error occured during free space calculation for (database:'" & databaseName & "';data_space_id:" & fileGroupId & ")", instanceName)
If spaceFxCalculator.IsError Then
Call GlobalErrorList.Add(FormatDbErrorMessage(spaceFxCalculator.ErrorMessage, instanceName, databaseName))
End If
End If
End If
End Select
Logger.LogDebug( "Database = " & databaseName & vbCrLf & _
" FileGroupId = " & fileGroupId & vbCrLf & _
" fileGroupType = " & fileGroupType & vbCrLf & _
" FreeSpaceMB = " & FreeSpaceMB & vbCrLf & _
" FreeSpacePercent = " & FreeSpacePercent & vbCrLf & _
" FreeSpaceAutoGrowMB = " & FreeSpaceAutoGrowMB & vbCrLf & _
" FreeSpaceAutoGrowPercent = " & FreeSpaceAutoGrowPercent )
listFileGroup.MoveNext
Loop
End If
End If
Next
End If
End If
dbMasterConnection.Close()
If hasItems Then
opsMgrAPI.scriptAPI.ReturnItems()
Else
Call ReturnEmptyPropertyBag(opsMgrAPI)
End If
End Function