'ValidateRestoreFolderDSS
productname  = Session.Property("ProductDisplay")
product_version  = Session.Property("ProductVersion")
restore_folder = Session.Property("RESTORE_FOLDER")
DSS_installFolder = Session.Property("INSTALLDIR")

'Open global log file for append, in Unicode format
Set fso = CreateObject("Scripting.FileSystemObject")
Set GlobalLogFile = fso.OpenTextFile(DSS_installFolder + "ValidateBackup.log", 8, True, -1)
GlobalLogFile.WriteLine(CStr(Now()) + " --- Validate installation directory starting")



' Size of DSS backup folder
RestoreFolderSizeMB = 0
' Free space partition of DSS
GetPartitionFreeSpaceMB = 0

' Free space of SQL database partition
GetPartitionFreeSpaceDbLocationMB = 0
' Size of database (.bak file)
DataBaseBackupSize = 0


Public SQLServer
Public SQLDBName
Public SQLConnection
Public SQLConnectionString



drive = Mid(restore_folder, 1, 2)


If drive = "\\" Then
	description = "Restoring from a UNC path is not supported." + chr(13) + "Please copy your backup folder locally and try again."
	EventLoggerErrorAndMessage(description)
	Session.Property("VALID_RESTORE_FOLDER") = Empty
End If


strComputer = "." ' Local Computer
Set objWMI = GetObject("winmgmts:\\" & strComputer & "\root\cimv2")
Set colDrives = objWMI.ExecQuery("select * from Win32_MappedLogicalDisk")

For Each objDrive In colDrives
  If drive = objDrive.DeviceID Then
	description = "Restoring from a mapped network drive is not supported." + chr(13) + "Please copy your backup folder locally and try again."
    EventLoggerErrorAndMessage(description)
    Session.Property("VALID_RESTORE_FOLDER") = Empty
  End If
Next

If Not BackupPackageIsValid Then
	description = "Backup Package is not vald! Cannot proceed."
	EventLoggerErrorAndMessage(description)
	Session.Property("VALID_RESTORE_FOLDER") = Empty
Else
	backup_id = restore_folder + "backup.txt"
	import_data = ""
	mssql_version_from_backup = ""
	raw_data_string = ""
	pattern = "mssql_version"
	If fso.FileExists(backup_id) Then
		Set backup_file = fso.OpenTextFile(backup_id, 1, False, 0)
		backup_version = backup_file.ReadLine
		backup_product = "unknown"
		If InStr(backup_version,"-") > 0 Then
			backup_product = Left(backup_version, InStr(backup_version,"-")-1)
			backup_version = Mid(backup_version, InStr(backup_version,"-")+1)
		End If
		If Not backup_file.AtEndOfStream Then
			Session.Property("RESTOREFQDN") = backup_file.ReadLine
			Session.Property("RESTOREIPADDRESS") = backup_file.ReadLine
		End If
		If Not backup_file.AtEndOfStream Then
			raw_data_string = backup_file.ReadLine
			separated_data_string = Split(raw_data_string, "=")
			If Trim(separated_data_string(0)) = pattern Then
				mssql_version_from_backup = Trim(separated_data_string(1))
			Else
				import_data = raw_data_string
				Session.Property("IMPORT_DATA") = import_data
				arr = Split(import_data, ".")
				Session.Property("IMPORT_DATA_SHORT") = arr(0) & "." & arr(1)
				backup_version = product_version
			End If
		Else
			Session.Property("IMPORT_DATA") = Empty
			Session.Property("IMPORT_DATA_SHORT") = Empty
		End If
		backup_file.Close()
		product_build = Mid(product_version, InStrRev(product_version, ".") + 1)
		backup_build = Mid(backup_version, InStrRev(backup_version, ".") + 1)
		backup_release = Mid(backup_version, 1, InStrRev(backup_version, ".") - 1)
		If backup_product <> "DSS" Then
			description = "Cannot verify folder located at " + restore_folder + chr(13) + "Cannot proceed."
			EventLoggerErrorAndMessage(description)
			Session.Property("VALID_RESTORE_FOLDER") = Empty
		Elseif Not ComparePV(product_version, backup_version) = 0 Then
			description = "The backup located at " + restore_folder + " is from a different release (" + backup_release + ")" + chr(13) + "Cannot proceed."
			EventLoggerErrorAndMessage(description)
			Session.Property("VALID_RESTORE_FOLDER") = Empty
		Elseif mssql_version_from_backup <> "" And CStr(GetMSSQLVersion) <> mssql_version_from_backup Then
			description = "The database backup version located at " + restore_folder + " is different from the database version you are trying to restore." + "Cannot proceed."
			EventLoggerErrorAndMessage(description)
			Session.Property("VALID_RESTORE_FOLDER") = Empty
		Elseif product_build <> backup_build And import_data = "" Then
			y = MsgBox("The backup located at " + restore_folder + " is from the same release but from a different build (" + backup_build + ")" + chr(13) + "Proceed?", 256+48+4, productname)
			If y <> 6 Then
			Session.Property("VALID_RESTORE_FOLDER") = Empty
		End If
	End If
Else
	description = "Cannot verify folder located at " + restore_folder + chr(13) + "Cannot proceed."
	EventLoggerErrorAndMessage(description)
	Session.Property("VALID_RESTORE_FOLDER") = Empty
End If
End If

If BackupPackageIsValid Then
	' Calculate free disk space where DSS installed before restoring system backup
	CalculateFolderSize restore_folder
	CalculatePartitionFreeSpace

	GlobalLogFile.WriteLine(CStr(Now()) + " --- DLP system backup folder size = "  & RestoreFolderSizeMB & " MB")
	GlobalLogFile.WriteLine(CStr(Now()) + " --- DLP system available space = "  & GetPartitionFreeSpaceMB & " MB")
	If RestoreFolderSizeMB > GetPartitionFreeSpaceMB Then
		description = "There is not enough disk space to restore the Forcepoint DLP system. You must have at least " & CInt(RestoreFolderSizeMB) & " MB free."
		EventLoggerErrorAndMessage(description)
		Session.Property("VALID_RESTORE_FOLDER") = Empty
	End If

	 ' Calculate free disk space where Database located before restoring Database backup
	GetDBPartitionFreeSpace
	GlobalLogFile.WriteLine(CStr(Now()) + " --- Forcepoint database backup size = "  & CInt(RestoreFolderSizeMB) & " MB")
	GlobalLogFile.WriteLine(CStr(Now()) + " --- Forcepoint database partition available space = "  & GetPartitionFreeSpaceDbLocationMB & " MB")

	If DataBaseBackupSize > GetPartitionFreeSpaceDbLocationMB Then
		description = "There is not enough disk space to restore the Forcepoint database. You must have at least " & CInt(RestoreFolderSizeMB) & " MB free."
		EventLoggerErrorAndMessage(description)
		Session.Property("VALID_RESTORE_FOLDER") = Empty
	End If
End if

Set fso = Nothing
GlobalLogFile.Close

Function ComparePV(x, y)
  a = Split(x, ".")
  b = Split(y, ".")
  For k = 0 To UBound(a) - 1
    If CLng(a(k)) > CLng(b(k)) Then
      ComparePV = 1
      Exit For
    ElseIf CLng(a(k)) < CLng(b(k)) Then
      ComparePV = 2
      Exit For
    Else
      If k = 2 Then
        ComparePV = 0
      End If
    End If
  Next
End Function

Function GetMSSQLVersion()
	Set shell = CreateObject("WScript.Shell")
	On Error Resume Next
	shell.RegRead("HKEY_LOCAL_MACHINE\SOFTWARE\Wow6432Node\Websense\Data Security\INSTALLDIR")
	If Err.Number = 0 Then
		Wow6432Node = "Wow6432Node\"
	Else
		Wow6432Node = ""
	End If

	Set SQLConnection = CreateObject("ADODB.Connection")
	SQLDBName = shell.RegRead("HKEY_LOCAL_MACHINE\SOFTWARE\" & Wow6432Node & "Websense\Data Security\DBNAME")
	SQLServer = Session.Property("WISESQLSERVERNAME")
	SQLConnectionString = "Provider=sqloledb;Data Source="& SQLServer & ";" &_
						"Initial Catalog=" & SQLDBName


	If Session.Property("WISESQLAUTH") = "NT" Then
		SQLConnectionString = SQLConnectionString & ";Integrated Security=SSPI"
	End If

	SQLUser = Session.Property("WISESQLUSER")
	SQLPass = Session.Property("WISESQLPASS")
	On Error Resume Next
	SQLConnection.Open SQLConnectionString, SQLUser, SQLPass
	If Err.Number <> 0 Then
	WScript.Echo "Failed connecting to SQL"
'      GlobalLogFile.WriteLine(CStr(Now()) + " --- Cannot Connection to SQL-Server using connection string " + SQLConnectionString + _
'				". Error: " + Err.Description )
	End If
	On Error Goto 0
	query = "SELECT @@MICROSOFTVERSION  / POWER(2,24)"
	Set recordSet = SQLConnection.Execute(query)
	GetMSSQLVersion = recordSet(0)

End Function

Function BackupPackageIsValid()
	status = FALSE
	backup_files_list_fn = "dss_backup_files_list.txt"
	Set fso = CreateObject("Scripting.FileSystemObject")
	backup_files_list_path = restore_folder & backup_files_list_fn
	If fso.FileExists(backup_files_list_path) Then
		Set f = fso.OpenTextFile(backup_files_list_path)
		If BackupPackageFilesAreValid(f) Then
			status = TRUE
		End If
		f.Close

	End If
	BackupPackageIsValid = status
End Function

Function BackupPackageFilesAreValid(f)
	Set fso = CreateObject("Scripting.FileSystemObject")
	files_status = TRUE
	Do Until f.AtEndOfStream OR files_status = FALSE
		raw_text_arr = split(f.ReadLine, "")
		file_path = restore_folder & Trim(raw_text_arr(0))
		file_size = Trim(raw_text_arr(1))
		If (fso.FileExists(file_path)) Then
			Set file_obj = fso.GetFile(file_path)
			If file_obj.size <> CLng(file_size) Then
				files_status = FALSE
			End If
		Else
			files_status = FALSE
		End If
	Loop
	BackupPackageFilesAreValid = files_status
End Function

Function CalculateFolderSize (sInstallFolder)

dataBaseFullPath = restore_folder & "MngDB\wbsn-data-security.bak"

If FSO.FolderExists(sInstallFolder ) Then
	Set objFolder = FSO.GetFolder(sInstallFolder)
	If FSO.FileExists(dataBaseFullPath) Then
		Set objFile = FSO.GetFile(dataBaseFullPath)
		DataBaseBackupSize = objFile.Size /  (1024 * 1024)
	End If
	RestoreFolderSizeMB = RestoreFolderSizeMB + (objFolder.Size /  (1024 * 1024)) - (objFile.Size /  (1024 * 1024))

Else
		GlobalLogFile.WriteLine(CStr(Now()) + "--- Restore folder not found")
End If


End Function


Function CalculatePartitionFreeSpace

' Calculate free space on the destination backup partition
Set FSO = CreateObject("Scripting.FileSystemObject")
Set Partition = FSO.GetDrive(Left(DSS_installFolder,2))
GetPartitionFreeSpaceMB = Partition.FreeSpace / (1024 * 1024)
End Function

Function GetDBPartitionFreeSpace

' Connection to the SQL server is already opened
On Error Resume Next
On Error Goto 0
query = "SELECT physical_name AS current_file_location FROM sys.master_files Where name='" & SQLDBName & "'"
Set recordSet = SQLConnection.Execute(query)
dbLocation = recordSet(0)

Set DbPartition =  FSO.GetDrive(Left(dbLocation,2))
GetPartitionFreeSpaceDbLocationMB = DbPartition.FreeSpace / (1024 * 1024)

End Function

Sub EventLoggerErrorAndMessage(description)
    Set shell = CreateObject("WScript.Shell")
	cmd = "EventCreate /T ERROR /L Application /ID 264 /SO ""Forcepoint AP-DATA"" /D ""BACKUP FAILED - " & description & """"
	rc = shell.Run(cmd, 0, True)
	MsgBox description, 16, productname
End Sub