Permalink
Switch branches/tags
Nothing to show
Find file
Fetching contributors…
Cannot retrieve contributors at this time
396 lines (336 sloc) 8.83 KB
!ifndef _AddToPath_nsh
!define _AddToPath_nsh
!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"
StrCmp $1 "" AddToPath_NTdoIt
Push $1
Call Trim
Pop $1
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 $0
Pop $1
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
; delete reg value if null
StrCmp $4 "" 0 +2 # if null delete reg
DeleteRegValue ${WriteEnvStr_RegKey} $0
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
###########################################
# Utility Functions #
###########################################
; 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
Function Trim ; Added by Pelaca
Exch $R1
Push $R2
Loop:
StrCpy $R2 "$R1" 1 -1
StrCmp "$R2" " " RTrim
StrCmp "$R2" "$\n" RTrim
StrCmp "$R2" "$\r" RTrim
StrCmp "$R2" ";" RTrim
GoTo Done
RTrim:
StrCpy $R1 "$R1" -1
Goto Loop
Done:
Pop $R2
Exch $R1
FunctionEnd
!endif ; _AddToPath_nsh