From 85b85e62e52198045ed853b81617db1f8c83c65f Mon Sep 17 00:00:00 2001 From: Simon Rozman Date: Tue, 4 Jun 2019 13:57:36 +0200 Subject: [PATCH] wintun: set DI_QUIETINSTALL flag for GUI-less device management Signed-off-by: Simon Rozman --- tun/wintun/wintun_windows.go | 29 ++++++++++++++++++++++++++--- 1 file changed, 26 insertions(+), 3 deletions(-) diff --git a/tun/wintun/wintun_windows.go b/tun/wintun/wintun_windows.go index 7b99a50..9dc4030 100644 --- a/tun/wintun/wintun_windows.go +++ b/tun/wintun/wintun_windows.go @@ -206,6 +206,13 @@ func CreateInterface(description string, hwndParent uintptr) (*Wintun, bool, err return nil, false, fmt.Errorf("SetupDiCreateDeviceInfo failed: %v", err) } + if hwndParent == 0 { + err = setQuietInstall(devInfoList, deviceData) + if err != nil { + return nil, false, fmt.Errorf("Setting quiet installation failed: %v", err) + } + } + // Set a device information element as the selected member of a device information set. err = devInfoList.SetSelectedDevice(deviceData) if err != nil { @@ -441,6 +448,13 @@ func (wintun *Wintun) DeleteInterface(hwndParent uintptr) (bool, bool, error) { } if wintun.cfgInstanceID == wintun2.cfgInstanceID { + if hwndParent == 0 { + err = setQuietInstall(devInfoList, deviceData) + if err != nil { + return false, false, fmt.Errorf("Setting quiet installation failed: %v", err) + } + } + // Remove the device. removeDeviceParams := setupapi.RemoveDeviceParams{ ClassInstallHeader: *setupapi.MakeClassInstallHeader(setupapi.DIF_REMOVE), @@ -480,11 +494,20 @@ func checkReboot(deviceInfoSet setupapi.DevInfo, deviceInfoData *setupapi.DevInf return false, err } - if (devInstallParams.Flags & (setupapi.DI_NEEDREBOOT | setupapi.DI_NEEDRESTART)) != 0 { - return true, nil + return (devInstallParams.Flags & (setupapi.DI_NEEDREBOOT | setupapi.DI_NEEDRESTART)) != 0, nil +} + +// +// setQuietInstall sets device install parameters for a quiet installation +// +func setQuietInstall(deviceInfoSet setupapi.DevInfo, deviceInfoData *setupapi.DevInfoData) error { + devInstallParams, err := deviceInfoSet.DeviceInstallParams(deviceInfoData) + if err != nil { + return err } - return false, nil + devInstallParams.Flags |= setupapi.DI_QUIETINSTALL + return deviceInfoSet.SetDeviceInstallParams(deviceInfoData, devInstallParams) } //