#Region ;**** Directives created by AutoIt3Wrapper_GUI ****
#AutoIt3Wrapper_UseUpx=y
#AutoIt3Wrapper_Res_Comment=Update Application via Internet
#AutoIt3Wrapper_Res_Description=Update demo
#AutoIt3Wrapper_Res_Fileversion=1.0.1.0
#AutoIt3Wrapper_Res_LegalCopyright=GreenCan
#EndRegion ;**** Directives created by AutoIt3Wrapper_GUI ****

#Include <Misc.au3>
#include <Date.au3>
#include <StringConstants.au3>
#include <File.au3>

; Update a script (compiled or not) by downloading the updated version from the Internet
;  script version must have format x.x.x.x !
;	as per format above (#AutoIt3Wrapper_Res_Fileversion=1.0.1.0)
;   If you use a different version number, you will have to change the logic in function CheckUpdate
;	You have to compile the script using wrapper functions, because the version number is fetched with FileGetVersion()

; Globals
Global $_CRC32_CodeBuffer, $_CRC32_CodeBufferMemory ; CRC checksum

#Region version
Global $sVersion = FileGetVersion(@ScriptName)
Local $sHeaderLine
; get version for non-compiled script (get the script version (#AutoIt3Wrapper_Res_Fileversion=1.0.1.0)
If Not @Compiled Then
	For $i = 2 to 10
		$sHeaderLine = FileReadLine(@ScriptDir & "\" & @ScriptName, $i)
		If StringLeft($sHeaderLine, 10) = "#EndRegion" Then ExitLoop ; not found
		If StringInStr($sHeaderLine, "Res_Fileversion") > 0 Then
			$sVersion = StringTrimLeft($sHeaderLine, 32)
			ExitLoop
		EndIf
	Next
EndIf
#EndRegion version

MsgBox(0,"Update demo", "This is demo version " & $sVersion )

; check for updates
CheckUpdate("UpdateTest" & (@Compiled ? ".exe" : ".au3"), $sVersion, "http://users.telenet.be/GreenCan/AutoIt/Updates/AppsUpdates.txt")
If @error  = 1 Then MsgBox(0,"Update demo", "No newer version available than " & $sVersion )
If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") :" & " @error "  & @error & @CRLF)


#Region update
; #FUNCTION# ================================================================================
; Name...........: CheckUpdate
; Description ...: Check version of the script or executable on Internet download site and Updates it if newer, if user agrees
; Syntax.........: CheckUpdate($sFileToUpdate, $sCurrentVersion, $sUpdateINI[, $InetForceReload = 1])
; Parameters ....: $sFileToUpdate - Script of Compiled script to Update
;				   $sCurrentVersion - Current version of the script ( as set in #AutoIt3Wrapper_Res_Fileversion consisiting of 4 version levels x.x.x.x)
;				   $sUpdateINI - url to INI file containing the update information
;				   $InetForceReload - force (uncached) download of the INI file
; Return values .: Success - exit the current script and restart new version
;                  Failure - 1, sets @error
;                  |1 - No newer version available
;                  |2 - User refused Update
;                  |3 - Update file not available
;                  |4 - Version info not available in Update file
;                  |5 - Download failure (and error message displayed)
;                  |6 - File size error, update aborted (and error message displayed)
;                  |7 - CRC Checksum error, update aborted (and error message displayed)
;
; 					Note: In case of error, just continue with current script...
;
; Author ........: GreenCan
; Modified.......:
; Remarks .......: 		The script version has format x.x.x.x !
;						In case the script (or exe) has been renamed (and does not correspond to the original file name),
;						The orginal file will not be renamed
;						If the file has it original file name, then the original file will be renamed to the filename_version.extension
;						example:
;							1. UpdateTest.au3 version 1.0.1.0 will be renamed to UpdateTest_1.0.1.0.au3
;								and UpdateTest.au3 version 1.0.1.2 will take its place
;							2. WhathEverNewName.au3 (UpdateTest.au3 version 1.0.1.0) will not be renamed and remane WhathEverNewName.au3
;								and UpdateTest.au3 version 1.0.1.2 will be started
;
;						ini file for this example
;						[UpdateTest.au3]
;						version=1.0.1.1
;						date=2014/06/17 10:00
;						Filesize=16720
;						CRC=FA7B28EB
;						download=http://users.telenet.be/GreenCan/AutoIt/Updates/UpdateTest_1.0.1.1.au3
;
;						[UpdateTest.exe]
;						version=1.0.1.1
;						date=2014/06/17 10:00
;						Filesize=473088
;						CRC=BC2B80BE
;						download=http://users.telenet.be/GreenCan/AutoIt/Updates/UpdateTest_1.0.1.1.exe
;
; Related .......:
; Link ..........:
; Example .......:
; ===========================================================================================
Func CheckUpdate($sFileToUpdate, $sCurrentVersion, $sUpdateINI, $InetForceReload = 1)
	Local $sINI_Data, $sNewVersion, $sDate, $sURL, $sDestinationFile, $szDrive, $szDir, $szFName, $szExt, $UpdateScript, $sTempFile
	Local $procwatchPID, $aVersion, $iNewVersion, $iCurrentVersion, $iBufferSize = 0x80000, $iCRC, $iCRC32 = 0, $sData, $FileSize, $i_FileSize, $hFile
	$sINI_Data = InetRead($sUpdateINI, $InetForceReload) ; get the ini file
	If Not @error Then
		If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") :" & " Update file found" & @CRLF)
		$sINI_Data = BinaryToString($sINI_Data)
		; read the ini file in memory
		$sNewVersion = IniMemoryRead($sINI_Data, $sFileToUpdate, "version", "")
		If $sNewVersion <> "" Then
			$sDate = IniMemoryRead($sINI_Data, $sFileToUpdate, "date", "")
			$sDate = _DateTimeFormat($sDate, 2) & " " & _DateTimeFormat($sDate, 4) ; set current time in Locale format
			$sURL = IniMemoryRead($sINI_Data, $sFileToUpdate, "download", "")
			$iFilesize = IniMemoryRead($sINI_Data, $sFileToUpdate, "Filesize", 0)
			$iCRC = IniMemoryRead($sINI_Data, $sFileToUpdate, "CRC", 0)

			If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") :" & " Application version: Current: " & $sCurrentVersion & " - New: " & $sNewVersion & "<" & @CRLF)

			; convert version x.x.x.x to a number where each x can go up to 999, so the max number can be 999 999 999 999
			; so version 1.0.1.29 will be converted to 1000001029  (001.000.001.029)
			$aVersion = StringSplit($sNewVersion, ".")
			$iNewVersion = $aVersion[4]  + ($aVersion[3] * 1000) + ($aVersion[2] * 1000000) + ($aVersion[1] * 1000000000)

			; do the same for current version
			$aVersion = StringSplit($sCurrentVersion, ".")
			$iCurrentVersion = $aVersion[4]  + ($aVersion[3] * 1000) + ($aVersion[2] * 1000000) + ($aVersion[1] * 1000000000)
			; Update only to a newer version
			If $sNewVersion > $sCurrentVersion Then
				If MsgBox(36, "Update " & $sFileToUpdate & " ver " & $sCurrentVersion, "A new version of " & $sFileToUpdate & " is available since " & $sDate & "." & @CRLF & "Download version " & $sNewVersion & " now? ") = 6 Then ;  No is Default
					;$sDestinationFile = HIER $sFileToUpdate
					_PathSplit($sFileToUpdate, $szDrive, $szDir, $szFName, $szExt)

					If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") : Downloading " & @ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt & @CRLF)
					InetgetProgress($sURL, @ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt )
					If @error Then
						MsgBox(16, "Error", "Download failure " & @CR & _
						$sURL & @CR & "Please retry later.")
						If FileExists(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt ) Then _
							FileDelete(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt) ; delete the file because it may be corrupt
						Return SetError(5, 0, 0)
					Else
						; checksum verification
						$i_FileSize = FileGetSize(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt)
						If $i_FileSize <> $iFilesize Then
							MsgBox(16, "Error", "Download failure, File size error!" & @CR & _
							$sURL & @CR & "Please retry later.")
							If FileExists(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt ) Then _
								FileDelete(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt) ; delete the file because it may be corrupt
							Return SetError(6, 0, 0)
						Else
							$hFile = FileOpen(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt, 16)

							For $i = 1 To Ceiling($i_FileSize / $iBufferSize)
								$sData = FileRead($hFile, $iBufferSize)
								$iCRC32 = _CRC32($sData, BitNot($iCRC32))
							Next
							FileClose($hFile)
							If Hex($iCRC32, 8) <> $iCRC Then
								MsgBox(16, "Error", "Download failure, CRC Checksum error!" & @CR & _
								$sURL & @CR & "Please retry later.")
								If FileExists(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt ) Then _
									FileDelete(@ScriptDir & "\" & $szFName &"_" & $sNewVersion & $szExt) ; delete the file because it may be corrupt
								Return SetError(7, 0, 0)
							Else
								If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") : CRC Checksum successful " & Hex($iCRC32, 8) & @CRLF)
								; download successful, exit this program just after starting the temporary batch file that does following:
								; 1. wait for a few seconds to enable to current script to Exit (using ping trick)
								; 2. rename the current script (or .exe) version (scriptname_x.x.x.x.au3 or scriptname_x.x.x.x.exe) as a backup
								; 3. then rename the new downloaded version to the initially used script name (scriptname.au3 or scriptname.exe)
								; start the new script
								; finally auto-delete the temporary script
								; quit

								$UpdateScript = '@ECHO ON' & _
									@CRLF & _
									'ping 127.0.0.1 -n 5 -w 5000' & _
									@CRLF & _
									'rename "' & @ScriptDir & "\" & $szFName & $szExt & '" ' & $szFName & "_" & $sCurrentVersion & $szExt & _
									@CRLF & _
									'rename "' & @ScriptDir & "\" & $szFName & "_" & $sNewVersion & $szExt & '" ' & $szFName & $szExt & _
									@CRLF & _
									'start ' & $szFName & $szExt & _
									@CRLF & _
									'DEL /F "' & @ScriptDir & '\proc.cmd"' & _
									@CRLF

								If Not @Compiled Then ConsoleWrite($UpdateScript & @CR)

								$sTempFile = FileOpen(@ScriptDir & "\proc.cmd",2)
								FileWrite($sTempFile, $UpdateScript)
								FileClose($sTempFile)
								MsgBox(0, "Restart", "Wait a moment. Restarting the program", 2)
								Run('"' & @ScriptDir & '\proc.cmd"', "", @SW_HIDE)
								; done, now exit
								Exit
							EndIf
						EndIf
					EndIf
				Else
					If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") :" & " User refused Update" & @CRLF)
					Return SetError(2, 0, 0)
				EndIf
			Else
				If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") :" & " No newer version available" & @CRLF)
				Return SetError(1, 0, 0)
			EndIf
		Else
			If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") :" & " Version info not available in Update file" & @CRLF)
			Return SetError(4, 0, 0)
		EndIf
	Else
		If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") :" & " Update file not available" & @CRLF)
		Return SetError(3, 0, 0)
	EndIf

EndFunc	;==>CheckUpdate

; #FUNCTION# ================================================================================
; Name...........: IniMemoryRead
; Description ...: IniRead from an ini file loaded into memory
; Syntax.........: IniMemoryRead($sIniContent, $sSection, $sKey, $sDefault)
; Parameters ....: $sIniContent - The ini file loaded into memory
;				   $sSection    - The section name in the .ini file.
;				   $sKey 	    - The key name in the .ini file.
;				   $sDefault    - The default value to return if the requested key is not found.

; Return values .: Success - The requested key value as a string.
;                  Failure - The default string if requested key not found.
; Author ........: GreenCan
; Modified.......:
; Remarks .......:
; Related .......:
; Link ..........:
; Example .......:
; ===========================================================================================
Func IniMemoryRead($sIniContent, $sSection, $sKey, $sDefault)
	Local $aIniContent
    $aIniContent = StringSplit($sIniContent, @CRLF)
	; find section (case unsensitive)
    For $i = 1 To $aIniContent[0]
		$aIniContent[$i] = StringLower(StringStripWS($aIniContent[$i], $STR_STRIPLEADING + $STR_STRIPTRAILING)) ; remove blanks
        If $aIniContent[$i] == "[" & StringLower($sSection) & "]" Then ExitLoop
    Next
    If $i > $aIniContent[0] Then Return $sDefault

	; find key (case unsensitive)
    For $i = $i + 1 To $aIniContent[0]
        If $aIniContent[$i] = "" Then ContinueLoop ; skip empty line
		$aIniContent[$i] = StringStripWS($aIniContent[$i], $STR_STRIPLEADING + $STR_STRIPTRAILING) ; remove blanks
		If StringLeft($aIniContent[$i], 1) = ";" Then ContinueLoop ; skip remarks
        If StringLeft($aIniContent[$i],1) = "[" And StringRight($aIniContent[$i],1) = "]" Then Return $sDefault ; OK passed the complete section so return the Default result
        If  StringLower(StringLeft($aIniContent[$i], StringLen($sKey))) = StringLower($sKey) Then Return StringStripWS(StringTrimLeft($aIniContent[$i], StringLen($sKey) + 1), $STR_STRIPLEADING + $STR_STRIPTRAILING ) ; return Key
    Next
    Return $sDefault
EndFunc	;==>IniMemoryRead

; -----------------------------------------------------------------------------
; CRC Checksum Machine Code UDF
; Purpose: Provide The Machine Code Version of CRC16/CRC32 Algorithm In AutoIt
; Author: Ward
; http://www.autoitscript.com/forum/topic/121985-autoit-machine-code-algorithm-collection/
; -----------------------------------------------------------------------------
Func _CRC32_Exit()
	$_CRC32_CodeBuffer = 0
	_MemVirtualFree($_CRC32_CodeBufferMemory, 0, $MEM_RELEASE)
EndFunc	;==>_CRC32_Exit

Func _CRC32($Data, $Initial = -1, $Polynomial = 0xEDB88320)
	If Not IsDllStruct($_CRC32_CodeBuffer) Then
		If @AutoItX64 Then
			Local $Opcode = '0xC80004004989CA680001000059678D41FF516A0859D1E873034431C8E2F75989848DFCFBFFFFE2E589D14489C04D85D2741B67E318418A1230C2480FB6D2C1E80833849500FCFFFF49FFC2E2E8F7D0C9C3'
		Else
			Local $Opcode = '0xC8000400538B5514B9000100008D41FF516A0859D1E8730231D0E2F85989848DFCFBFFFFE2E78B5D088B4D0C8B451085DB7416E3148A1330C20FB6D2C1E80833849500FCFFFF43E2ECF7D05BC9C21000'
		EndIf
		$Opcode = Binary($Opcode)

		$_CRC32_CodeBufferMemory = _MemVirtualAlloc(0, BinaryLen($Opcode), $MEM_COMMIT, $PAGE_EXECUTE_READWRITE)
		$_CRC32_CodeBuffer = DllStructCreate("byte[" & BinaryLen($Opcode) & "]", $_CRC32_CodeBufferMemory)
		DllStructSetData($_CRC32_CodeBuffer, 1, $Opcode)
		OnAutoItExitRegister("_CRC32_Exit")
	EndIf

	$Data = Binary($Data)
	Local $InputLen = BinaryLen($Data)
	Local $Input = DllStructCreate("byte[" & $InputLen & "]")
	DllStructSetData($Input, 1, $Data)

	Local $Ret = DllCall("user32.dll", "uint", "CallWindowProc", "ptr", DllStructGetPtr($_CRC32_CodeBuffer), _
													"ptr", DllStructGetPtr($Input), _
													"uint", $InputLen, _
													"uint", $Initial, _
													"uint", $Polynomial)

	Return $Ret[0]
EndFunc	;==>_CRC32
#EndRegion update

; #FUNCTION# ================================================================================
; Name...........: InetgetProgress
; Description ...: Download a file with progress bar
; Syntax.........: InetgetProgress($sURL, $sFilename)
; Parameters ....: $sURL - URL of the file to be downloaded
;				   $sFilename - Local name of the destination file to be downloaded
; Return values .: Success - 0
;                  Failure - 1, sets @error
;                  |1 - Download failed
; Author ........: GreenCan
; Modified.......:
; Remarks .......:
; Related .......:
; Link ..........:
; Example .......:
; ===========================================================================================
Func InetgetProgress($sURL, $sFilename)
	Local $iSize, $iTotalSize, $hDownload, $iSec, $iCurrentBytes, $iReadBytes, $szDrive, $szDir, $szFName, $szExt
	_PathSplit($sURL, $szDrive, $szDir, $szFName, $szExt)
	$iSize = InetGetSize ($sURL)
	$iTotalSize = Round($iSize / 1024)
	If Not @Compiled Then ConsoleWrite("@@ Debug(" & @ScriptLineNumber & ") : $szFName & $szExt: " & $szFName & $szExt & " $sURL: " & $sURL & @CRLF)
	$hDownload = INetGet($sURL, $sFilename, 16, 1) ; from InetConstants.au3:  $INET_FORCEBYPASS (16) = By-pass forcing the connection online, $INET_DOWNLOADBACKGROUND (1) = Background download
	ProgressOn("Download " & $szFName & $szExt, "Download progress")
	Do
		$iSec = @SEC
		$iCurrentBytes = Round(InetGetInfo($hDownload,0))
		While @SEC = $iSec
			Sleep(1000)
		WEnd
		$iReadBytes = Round(InetGetInfo($hDownload,0))
		$iTotalSize = $iTotalSize - (($iReadBytes - $iCurrentBytes) /1024)
		ProgressSet(100 - Round($iTotalSize / $iSize * 100000), 100 - Round($iTotalSize / $iSize * 100000) & "%")
	Until InetGetInfo($hDownload,2)
	ProgressOff()
	If Not InetGetInfo($hDownload,3) Then Return SetError(1, 0, 0)
EndFunc	;==>InetgetProgress