!include "WinVer.nsh"
!include "drvsetup.nsh"
!include "x64.nsh"
 
;
; Written by Kuba Ober
; Copyright (c) 2004 Kuba Ober
;
; Permission is hereby granted, free of charge, to any person obtaining a 
; copy of this software and associated documentation files (the "Software"), 
; to deal in the Software without restriction, including without limitation 
; the rights to use, copy, modify, merge, publish, distribute, sublicense, 
; and/or sell copies of the Software, and to permit persons to whom the 
; Software is furnished to do so, subject to the following conditions:
;
; The above copyright notice and this permission notice shall be included in 
; all copies or substantial portions of the Software.
; 
; THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
; IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
; FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
; AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
; LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
; FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
; DEALINGS IN THE SOFTWARE.
 
;
; U S A G E
;
; Push "c:\program files\yoursoftware\driver" 
;  -- the directory of the .inf file
; Push "c:\program files\yoursoftware\driver\driver.inf"
;  -- the filepath of the .inf file (directory + filename)
; Push "USB\VID_1234&PID_5678"
;  -- the HID (Hardware ID) of your device
; Call InstallUpgradeDriver
;
; Your driver (minimally the .inf and .sys files) should already by installed
; by your NSIS script.
;
; Typically, you would put the driver either in $INSTDIR or $INSTDIR\Driver
; It's up to you, of course.
;
; The driver (i.e. .inf, .sys and related files) must be present for the
; lifetime of your application, you shouldn't remove them after calling
; this function!
;
; You DON'T want to put the driver in any of system directories. Windows
; will do it when the device is first plugged in.
 
Function InstallUpgradeDriver
 
 Pop $R0 ; HID
 Pop $R1 ; INFPATH
 Pop $R2 ; INFDIR
 
 ${If} ${AtLeastWin2000}
   ${AndIf} ${AtMostWin2003}
     Goto lbl_upgrade
 ${EndIf}

 ${If} ${AtLeastWin7}
   ${If} ${RunningX64}
     StrCpy $R3 "C:\Windows\winsxs\amd64_microsoft-windows-pnputil_31bf3856ad364e35_6.1.7600.16385_none_5958b438d6388d15\PnPutil.exe"
     goto lbl_pnputil
   ${Else}
     StrCpy $R3 "C:\Windows\winsxs\x86_microsoft-windows-pnputil_31bf3856ad364e35_6.1.7600.16385_none_fd3a18b51ddb1bdf\PnPutil.exe"
     goto lbl_pnputil
   ${EndIf}
${EndIf}

 ${If} ${AtLeastWinVista}
   StrCpy $R3 "pnputil.exe"
   Goto lbl_pnputil
 ${EndIf}

 DetailPrint "This Windows version doesn't support automatic driver updates."
 Goto lbl_noupgrade

lbl_upgrade:
 System::Get '${sysUpdateDriverForPlugAndPlayDevices}'
 Pop $0
 StrCmp $0 'error' lbl_noapi
 DetailPrint "Updating the driver..."
 ; 0, HID, INFPATH, 0, 0
 Push $INSTDIR ; Otherwise this function will swallow it, dunno why
 System::Call '${sysUpdateDriverForPlugAndPlayDevices}?e (0, R0, R1, 0, 0) .r0'
 Pop $1 ; last error
 Pop $INSTDIR
 IntCmp $0 1 lbl_done
 IntCmp $1 ${ERROR_NO_SUCH_DEVINST} lbl_notplugged
 
 DetailPrint "Driver update has failed. ($0,$1)"
 Goto lbl_noupgrade
lbl_notplugged:
 DetailPrint "The device is not plugged in, cannot update the driver."
 Goto lbl_noupgrade
lbl_noapi:
 DetailPrint "Your Windows version doesn't support driver updates."
 Goto lbl_noupgrade
 
lbl_noupgrade:
 ; Pre-install the driver
 System::Get '${sysSetupCopyOEMInf}'
 Pop $0
 StrCmp $0 'error' lbl_inoapi
 DetailPrint "Installing the driver..."
 ; INFPATH, INFDIR, SPOST_PATH, "", 0, 0, 0, 0
 System::Call '${sysSetupCopyOEMInf}?e (R1, R2, ${SPOST_PATH}, 0, 0, 0, 0, 0) .r0'
 Pop $1 ; last error
 IntCmp $0 1 lbl_nodriver
 DetailPrint 'Driver pre-installation has failed with error #$1'
 Goto lbl_done
lbl_inoapi:
 DetailPrint "Your Windows version doesn't support driver pre-installation."
 Goto lbl_nodriver
 
lbl_pnputil:
 DetailPrint 'pnputil.exe -i -a "$R1"'
 nsExec::ExecToLog '$R3 -i -a "$R1"'
 Pop $0
 StrCmp $0 "error" lbl_pnputil_not_found
 StrCmp $0 "timeout" lbl_pnputil_timeout
 IntCmp $0 "1" lbl_pnputil_already_exists
 IntCmp $0 0 lbl_done
 DetailPrint "pnputil.exe reported an error: $0"
 Goto lbl_nodriver
lbl_pnputil_not_found:
 DetailPrint "pnputil.exe not found"
 Goto lbl_nodriver
lbl_pnputil_timeout:
 DetailPrint "timeout waiting for pnputil.exe to run"
 Goto lbl_nodriver
lbl_pnputil_already_exists:
 DetailPrint "Driver does not need upgraded"
 Goto lbl_nodriver
 
lbl_nodriver:
lbl_done:
 
FunctionEnd