[Sumover-dev] [svn commit] r3841 - vic/trunk/win32
sumover-dev at cs.ucl.ac.uk
sumover-dev at cs.ucl.ac.uk
Mon Sep 11 14:32:48 BST 2006
Author: piers
Date: Mon Sep 11 14:32:33 2006
New Revision: 3841
Added:
vic/trunk/win32/utility-macros.nsh
Log:
Added utility functions - needed for NSIS scripts
Added: vic/trunk/win32/utility-macros.nsh
==============================================================================
--- (empty file)
+++ vic/trunk/win32/utility-macros.nsh Mon Sep 11 14:32:33 2006
@@ -0,0 +1,473 @@
+###########################################
+# Various Utility Functions #
+###########################################
+
+!ifndef _AddToPath_nsh
+!define _AddToPath_nsh
+
+; for path manipulation
+;http://nsis.sourceforge.net/wiki/Path_Manipulation
+
+!verbose 3
+!include "WinMessages.NSH"
+!verbose 4
+
+!ifndef WriteEnvStr_RegKey
+ !ifdef ALL_USERS
+ !define WriteEnvStr_RegKey \
+ 'HKLM "SYSTEM\CurrentControlSet\Control\Session Manager\Environment"'
+ !else
+ !define WriteEnvStr_RegKey 'HKCU "Environment"'
+ !endif
+!endif
+
+; AddToPath - Adds the given dir to the search path.
+; Input - head of the stack
+; Note - Win9x systems requires reboot
+
+Function AddToPath
+ Exch $0
+ Push $1
+ Push $2
+ Push $3
+
+ # don't add if the path doesn't exist
+ IfFileExists "$0\*.*" "" AddToPath_done
+
+ ReadEnvStr $1 PATH
+ Push "$1;"
+ Push "$0;"
+ Call StrStr
+ Pop $2
+ StrCmp $2 "" "" AddToPath_done
+ Push "$1;"
+ Push "$0\;"
+ Call StrStr
+ Pop $2
+ StrCmp $2 "" "" AddToPath_done
+ GetFullPathName /SHORT $3 $0
+ Push "$1;"
+ Push "$3;"
+ Call StrStr
+ Pop $2
+ StrCmp $2 "" "" AddToPath_done
+ Push "$1;"
+ Push "$3\;"
+ Call StrStr
+ Pop $2
+ StrCmp $2 "" "" AddToPath_done
+
+ Call IsNT
+ Pop $1
+ StrCmp $1 1 AddToPath_NT
+ ; Not on NT
+ StrCpy $1 $WINDIR 2
+ FileOpen $1 "$1\autoexec.bat" a
+ FileSeek $1 -1 END
+ FileReadByte $1 $2
+ IntCmp $2 26 0 +2 +2 # DOS EOF
+ FileSeek $1 -1 END # write over EOF
+ FileWrite $1 "$\r$\nSET PATH=%PATH%;$3$\r$\n"
+ FileClose $1
+ SetRebootFlag true
+ Goto AddToPath_done
+
+ AddToPath_NT:
+ ReadRegStr $1 ${WriteEnvStr_RegKey} "PATH"
+ StrCpy $2 $1 1 -1 # copy last char
+ StrCmp $2 ";" 0 +2 # if last char == ;
+ StrCpy $1 $1 -1 # remove last char
+ StrCmp $1 "" AddToPath_NTdoIt
+ StrCpy $0 "$1;$0"
+ AddToPath_NTdoIt:
+ WriteRegExpandStr ${WriteEnvStr_RegKey} "PATH" $0
+ SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
+
+ AddToPath_done:
+ Pop $3
+ Pop $2
+ Pop $1
+ Pop $0
+FunctionEnd
+
+; RemoveFromPath - Remove a given dir from the path
+; Input: head of the stack
+
+Function un.RemoveFromPath
+ Exch $0
+ Push $1
+ Push $2
+ Push $3
+ Push $4
+ Push $5
+ Push $6
+
+ IntFmt $6 "%c" 26 # DOS EOF
+
+ Call un.IsNT
+ Pop $1
+ StrCmp $1 1 unRemoveFromPath_NT
+ ; Not on NT
+ StrCpy $1 $WINDIR 2
+ FileOpen $1 "$1\autoexec.bat" r
+ GetTempFileName $4
+ FileOpen $2 $4 w
+ GetFullPathName /SHORT $0 $0
+ StrCpy $0 "SET PATH=%PATH%;$0"
+ Goto unRemoveFromPath_dosLoop
+
+ unRemoveFromPath_dosLoop:
+ FileRead $1 $3
+ StrCpy $5 $3 1 -1 # read last char
+ StrCmp $5 $6 0 +2 # if DOS EOF
+ StrCpy $3 $3 -1 # remove DOS EOF so we can compare
+ StrCmp $3 "$0$\r$\n" unRemoveFromPath_dosLoopRemoveLine
+ StrCmp $3 "$0$\n" unRemoveFromPath_dosLoopRemoveLine
+ StrCmp $3 "$0" unRemoveFromPath_dosLoopRemoveLine
+ StrCmp $3 "" unRemoveFromPath_dosLoopEnd
+ FileWrite $2 $3
+ Goto unRemoveFromPath_dosLoop
+ unRemoveFromPath_dosLoopRemoveLine:
+ SetRebootFlag true
+ Goto unRemoveFromPath_dosLoop
+
+ unRemoveFromPath_dosLoopEnd:
+ FileClose $2
+ FileClose $1
+ StrCpy $1 $WINDIR 2
+ Delete "$1\autoexec.bat"
+ CopyFiles /SILENT $4 "$1\autoexec.bat"
+ Delete $4
+ Goto unRemoveFromPath_done
+
+ unRemoveFromPath_NT:
+ ReadRegStr $1 ${WriteEnvStr_RegKey} "PATH"
+ StrCpy $5 $1 1 -1 # copy last char
+ StrCmp $5 ";" +2 # if last char != ;
+ StrCpy $1 "$1;" # append ;
+ Push $1
+ Push "$0;"
+ Call un.StrStr ; Find `$0;` in $1
+ Pop $2 ; pos of our dir
+ StrCmp $2 "" unRemoveFromPath_done
+ ; else, it is in path
+ # $0 - path to add
+ # $1 - path var
+ StrLen $3 "$0;"
+ StrLen $4 $2
+ StrCpy $5 $1 -$4 # $5 is now the part before the path to remove
+ StrCpy $6 $2 "" $3 # $6 is now the part after the path to remove
+ StrCpy $3 $5$6
+
+ StrCpy $5 $3 1 -1 # copy last char
+ StrCmp $5 ";" 0 +2 # if last char == ;
+ StrCpy $3 $3 -1 # remove last char
+
+ WriteRegExpandStr ${WriteEnvStr_RegKey} "PATH" $3
+ SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
+
+ unRemoveFromPath_done:
+ Pop $6
+ Pop $5
+ Pop $4
+ Pop $3
+ Pop $2
+ Pop $1
+ Pop $0
+FunctionEnd
+
+
+
+; AddToEnvVar - Adds the given value to the given environment var
+; Input - head of the stack $0 environement variable $1=value to add
+; Note - Win9x systems requires reboot
+
+Function AddToEnvVar
+
+ Exch $1 ; $1 has environment variable value
+ Exch
+ Exch $0 ; $0 has environment variable name
+
+ DetailPrint "Adding $1 to $0"
+ Push $2
+ Push $3
+ Push $4
+
+
+ ReadEnvStr $2 $0
+ Push "$2;"
+ Push "$1;"
+ Call StrStr
+ Pop $3
+ StrCmp $3 "" "" AddToEnvVar_done
+
+ Push "$2;"
+ Push "$1\;"
+ Call StrStr
+ Pop $3
+ StrCmp $3 "" "" AddToEnvVar_done
+
+
+ Call IsNT
+ Pop $2
+ StrCmp $2 1 AddToEnvVar_NT
+ ; Not on NT
+ StrCpy $2 $WINDIR 2
+ FileOpen $2 "$2\autoexec.bat" a
+ FileSeek $2 -1 END
+ FileReadByte $2 $3
+ IntCmp $3 26 0 +2 +2 # DOS EOF
+ FileSeek $2 -1 END # write over EOF
+ FileWrite $2 "$\r$\nSET $0=%$0%;$4$\r$\n"
+ FileClose $2
+ SetRebootFlag true
+ Goto AddToEnvVar_done
+
+ AddToEnvVar_NT:
+ ReadRegStr $2 ${WriteEnvStr_RegKey} $0
+ StrCpy $3 $2 1 -1 # copy last char
+ StrCmp $3 ";" 0 +2 # if last char == ;
+ StrCpy $2 $2 -1 # remove last char
+ StrCmp $2 "" AddToEnvVar_NTdoIt
+ StrCpy $1 "$2;$1"
+ AddToEnvVar_NTdoIt:
+ WriteRegExpandStr ${WriteEnvStr_RegKey} $0 $1
+ SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
+
+ AddToEnvVar_done:
+ Pop $4
+ Pop $3
+ Pop $2
+ Pop $1
+ Pop $0
+
+FunctionEnd
+
+; RemoveFromEnvVar - Remove a given value from a environment var
+; Input: head of the stack
+
+Function un.RemoveFromEnvVar
+
+ Exch $1 ; $1 has environment variable value
+ Exch
+ Exch $0 ; $0 has environment variable name
+
+ DetailPrint "Removing $1 from $0"
+ Push $2
+ Push $3
+ Push $4
+ Push $5
+ Push $6
+ Push $7
+
+ IntFmt $7 "%c" 26 # DOS EOF
+
+ Call un.IsNT
+ Pop $2
+ StrCmp $2 1 unRemoveFromEnvVar_NT
+ ; Not on NT
+ StrCpy $2 $WINDIR 2
+ FileOpen $2 "$2\autoexec.bat" r
+ GetTempFileName $5
+ FileOpen $3 $5 w
+ GetFullPathName /SHORT $1 $1
+ StrCpy $1 "SET $0=%$0%;$1"
+ Goto unRemoveFromEnvVar_dosLoop
+
+ unRemoveFromEnvVar_dosLoop:
+ FileRead $2 $4
+ StrCpy $6 $4 1 -1 # read last char
+ StrCmp $6 $7 0 +2 # if DOS EOF
+ StrCpy $4 $4 -1 # remove DOS EOF so we can compare
+ StrCmp $4 "$1$\r$\n" unRemoveFromEnvVar_dosLoopRemoveLine
+ StrCmp $4 "$1$\n" unRemoveFromEnvVar_dosLoopRemoveLine
+ StrCmp $4 "$1" unRemoveFromEnvVar_dosLoopRemoveLine
+ StrCmp $4 "" unRemoveFromEnvVar_dosLoopEnd
+ FileWrite $3 $4
+ Goto unRemoveFromEnvVar_dosLoop
+ unRemoveFromEnvVar_dosLoopRemoveLine:
+ SetRebootFlag true
+ Goto unRemoveFromEnvVar_dosLoop
+
+ unRemoveFromEnvVar_dosLoopEnd:
+ FileClose $3
+ FileClose $2
+ StrCpy $2 $WINDIR 2
+ Delete "$2\autoexec.bat"
+ CopyFiles /SILENT $5 "$2\autoexec.bat"
+ Delete $5
+ Goto unRemoveFromEnvVar_done
+
+ unRemoveFromEnvVar_NT:
+ ReadRegStr $2 ${WriteEnvStr_RegKey} $0
+ StrCpy $6 $2 1 -1 # copy last char
+ StrCmp $6 ";" +2 # if last char != ;
+ StrCpy $2 "$2;" # append ;
+ Push $2
+ Push "$1;"
+ Call un.StrStr ; Find `$1;` in $2
+ Pop $3 ; pos of our dir
+ StrCmp $3 "" unRemoveFromEnvVar_done
+ ; else, it is in path
+ # $1 - path to add
+ # $2 - path var
+ StrLen $4 "$1;"
+ StrLen $5 $3
+ StrCpy $6 $2 -$5 # $6 is now the part before the path to remove
+ StrCpy $7 $3 "" $4 # $7 is now the part after the path to remove
+ StrCpy $4 $6$7
+
+ StrCpy $6 $4 1 -1 # copy last char
+ StrCmp $6 ";" 0 +2 # if last char == ;
+ StrCpy $4 $4 -1 # remove last char
+
+ WriteRegExpandStr ${WriteEnvStr_RegKey} $0 $4
+ SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
+
+ unRemoveFromEnvVar_done:
+ Pop $7
+ Pop $6
+ Pop $5
+ Pop $4
+ Pop $3
+ Pop $2
+ Pop $1
+ Pop $0
+FunctionEnd
+
+
+
+
+!ifndef IsNT_KiCHiK
+!define IsNT_KiCHiK
+
+; IsNT
+; no input
+; output, top of the stack = 1 if NT or 0 if not
+;
+; Usage:
+; Call IsNT
+; Pop $R0
+; ($R0 at this point is 1 or 0)
+
+!macro IsNT un
+Function ${un}IsNT
+ Push $0
+ ReadRegStr $0 HKLM "SOFTWARE\Microsoft\Windows NT\CurrentVersion" CurrentVersion
+ StrCmp $0 "" 0 IsNT_yes
+ ; we are not NT.
+ Pop $0
+ Push 0
+ Return
+
+ IsNT_yes:
+ ; NT!!!
+ Pop $0
+ Push 1
+FunctionEnd
+!macroend
+!insertmacro IsNT ""
+!insertmacro IsNT "un."
+
+!endif ; IsNT_KiCHiK
+
+; StrStr
+; input, top of stack = string to search for
+; top of stack-1 = string to search in
+; output, top of stack (replaces with the portion of the string remaining)
+; modifies no other variables.
+;
+; Usage:
+; Push "this is a long ass string"
+; Push "ass"
+; Call StrStr
+; Pop $R0
+; ($R0 at this point is "ass string")
+
+!macro StrStr un
+Function ${un}StrStr
+Exch $R1 ; st=haystack,old$R1, $R1=needle
+ Exch ; st=old$R1,haystack
+ Exch $R2 ; st=old$R1,old$R2, $R2=haystack
+ Push $R3
+ Push $R4
+ Push $R5
+ StrLen $R3 $R1
+ StrCpy $R4 0
+ ; $R1=needle
+ ; $R2=haystack
+ ; $R3=len(needle)
+ ; $R4=cnt
+ ; $R5=tmp
+ loop:
+ StrCpy $R5 $R2 $R3 $R4
+ StrCmp $R5 $R1 done
+ StrCmp $R5 "" done
+ IntOp $R4 $R4 + 1
+ Goto loop
+done:
+ StrCpy $R1 $R2 "" $R4
+ Pop $R5
+ Pop $R4
+ Pop $R3
+ Pop $R2
+ Exch $R1
+FunctionEnd
+!macroend
+!insertmacro StrStr ""
+!insertmacro StrStr "un."
+
+!endif ; _AddToPath_nsh
+
+!ifndef _isEmptyDir_nsh
+!define _isEmptyDir_nsh
+
+; Macro
+;http://nsis.sourceforge.net/wiki/Delete_dir_only_if_empty
+
+Function un.DeleteDirIfEmpty
+ FindFirst $R0 $R1 "$0\*.*"
+ strcmp $R1 "." 0 NoDelete
+ FindNext $R0 $R1
+ strcmp $R1 ".." 0 NoDelete
+ ClearErrors
+ FindNext $R0 $R1
+ IfErrors 0 NoDelete
+ FindClose $R0
+ Sleep 1000
+ RMDir "$0"
+ NoDelete:
+ FindClose $R0
+FunctionEnd
+
+;http://nsis.sourceforge.net/Check_if_dir_is_empty
+
+!macro isEmptyDir un
+Function ${un}isEmptyDir
+ # Stack -> # Stack: <directory>
+ Exch $0 # Stack: $0
+ Push $1 # Stack: $1, $0
+ FindFirst $0 $1 "$0\*.*"
+ strcmp $1 "." 0 _notempty
+ FindNext $0 $1
+ strcmp $1 ".." 0 _notempty
+ ClearErrors
+ FindNext $0 $1
+ IfErrors 0 _notempty
+ FindClose $0
+ Pop $1 # Stack: $0
+ StrCpy $0 1
+ Exch $0 # Stack: 1 (true)
+ goto _end
+ _notempty:
+ FindClose $0
+ Pop $1 # Stack: $0
+ StrCpy $0 0
+ Exch $0 # Stack: 0 (false)
+ _end:
+FunctionEnd
+!macroend
+!insertmacro isEmptyDir ""
+!insertmacro isEmptyDir "un."
+
+!endif ; _isEmptyDir_nsh
\ No newline at end of file
More information about the Sumover-dev
mailing list