wintun: put mutex into private namespace
This commit is contained in:
parent
df2bf34373
commit
950ca2ba8c
99
tun/wintun/namespace_windows.go
Normal file
99
tun/wintun/namespace_windows.go
Normal file
@ -0,0 +1,99 @@
|
|||||||
|
/* SPDX-License-Identifier: MIT
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package wintun
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/hex"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"sync"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"golang.org/x/crypto/blake2s"
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
"golang.org/x/text/unicode/norm"
|
||||||
|
|
||||||
|
"golang.zx2c4.com/wireguard/ipc/winpipe"
|
||||||
|
"golang.zx2c4.com/wireguard/tun/wintun/namespaceapi"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
wintunObjectSecurityAttributes *windows.SecurityAttributes
|
||||||
|
hasInitializedNamespace bool
|
||||||
|
initializingNamespace sync.Mutex
|
||||||
|
)
|
||||||
|
|
||||||
|
func initializeNamespace() error {
|
||||||
|
initializingNamespace.Lock()
|
||||||
|
defer initializingNamespace.Unlock()
|
||||||
|
if hasInitializedNamespace {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
sd, err := winpipe.SddlToSecurityDescriptor("O:SYD:P(A;;GA;;;SY)")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("SddlToSecurityDescriptor failed: %v", err)
|
||||||
|
}
|
||||||
|
wintunObjectSecurityAttributes = &windows.SecurityAttributes{
|
||||||
|
Length: uint32(len(sd)),
|
||||||
|
SecurityDescriptor: uintptr(unsafe.Pointer(&sd[0])),
|
||||||
|
}
|
||||||
|
sid, err := windows.CreateWellKnownSid(windows.WinLocalSystemSid)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("CreateWellKnownSid(LOCAL_SYSTEM) failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
boundary, err := namespaceapi.CreateBoundaryDescriptor("Wintun")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("CreateBoundaryDescriptor failed: %v", err)
|
||||||
|
}
|
||||||
|
err = boundary.AddSid(sid)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("AddSIDToBoundaryDescriptor failed: %v", err)
|
||||||
|
}
|
||||||
|
for {
|
||||||
|
_, err = namespaceapi.CreatePrivateNamespace(wintunObjectSecurityAttributes, boundary, "Wintun")
|
||||||
|
if err == windows.ERROR_ALREADY_EXISTS {
|
||||||
|
_, err = namespaceapi.OpenPrivateNamespace(boundary, "Wintun")
|
||||||
|
if err == windows.ERROR_PATH_NOT_FOUND {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("Create/OpenPrivateNamespace failed: %v", err)
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
hasInitializedNamespace = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pool Pool) takeNameMutex() (windows.Handle, error) {
|
||||||
|
err := initializeNamespace()
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const mutexLabel = "WireGuard Adapter Name Mutex Stable Suffix v1 jason@zx2c4.com"
|
||||||
|
b2, _ := blake2s.New256(nil)
|
||||||
|
b2.Write([]byte(mutexLabel))
|
||||||
|
b2.Write(norm.NFC.Bytes([]byte(string(pool))))
|
||||||
|
mutexName := `Wintun\Wintun-Name-Mutex-` + hex.EncodeToString(b2.Sum(nil))
|
||||||
|
mutex, err := windows.CreateMutex(wintunObjectSecurityAttributes, false, windows.StringToUTF16Ptr(mutexName))
|
||||||
|
if err != nil {
|
||||||
|
err = fmt.Errorf("Error creating name mutex: %v", err)
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
event, err := windows.WaitForSingleObject(mutex, windows.INFINITE)
|
||||||
|
if err != nil {
|
||||||
|
windows.CloseHandle(mutex)
|
||||||
|
return 0, fmt.Errorf("Error waiting on name mutex: %v", err)
|
||||||
|
}
|
||||||
|
if event != windows.WAIT_OBJECT_0 {
|
||||||
|
windows.CloseHandle(mutex)
|
||||||
|
return 0, errors.New("Error with event trigger of name mutex")
|
||||||
|
}
|
||||||
|
return mutex, nil
|
||||||
|
}
|
@ -6,17 +6,14 @@
|
|||||||
package wintun
|
package wintun
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/hex"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
"unsafe"
|
"unsafe"
|
||||||
|
|
||||||
"golang.org/x/crypto/blake2s"
|
|
||||||
"golang.org/x/sys/windows"
|
"golang.org/x/sys/windows"
|
||||||
"golang.org/x/sys/windows/registry"
|
"golang.org/x/sys/windows/registry"
|
||||||
"golang.org/x/text/unicode/norm"
|
|
||||||
|
|
||||||
"golang.zx2c4.com/wireguard/tun/wintun/iphlpapi"
|
"golang.zx2c4.com/wireguard/tun/wintun/iphlpapi"
|
||||||
"golang.zx2c4.com/wireguard/tun/wintun/nci"
|
"golang.zx2c4.com/wireguard/tun/wintun/nci"
|
||||||
@ -758,26 +755,3 @@ func (wintun *Interface) GUID() windows.GUID {
|
|||||||
func (wintun *Interface) LUID() uint64 {
|
func (wintun *Interface) LUID() uint64 {
|
||||||
return ((uint64(wintun.luidIndex) & ((1 << 24) - 1)) << 24) | ((uint64(wintun.ifType) & ((1 << 16) - 1)) << 48)
|
return ((uint64(wintun.luidIndex) & ((1 << 24) - 1)) << 24) | ((uint64(wintun.ifType) & ((1 << 16) - 1)) << 48)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pool Pool) takeNameMutex() (windows.Handle, error) {
|
|
||||||
const mutexLabel = "WireGuard Adapter Name Mutex Stable Suffix v1 jason@zx2c4.com"
|
|
||||||
b2, _ := blake2s.New256(nil)
|
|
||||||
b2.Write([]byte(mutexLabel))
|
|
||||||
b2.Write(norm.NFC.Bytes([]byte(string(pool))))
|
|
||||||
mutexName := `Global\Wintun-Name-Mutex-` + hex.EncodeToString(b2.Sum(nil))
|
|
||||||
mutex, err := windows.CreateMutex(nil, false, windows.StringToUTF16Ptr(mutexName))
|
|
||||||
if err != nil {
|
|
||||||
err = fmt.Errorf("Error creating name mutex: %v", err)
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
event, err := windows.WaitForSingleObject(mutex, windows.INFINITE)
|
|
||||||
if err != nil {
|
|
||||||
windows.CloseHandle(mutex)
|
|
||||||
return 0, fmt.Errorf("Error waiting on name mutex: %v", err)
|
|
||||||
}
|
|
||||||
if event != windows.WAIT_OBJECT_0 {
|
|
||||||
windows.CloseHandle(mutex)
|
|
||||||
return 0, errors.New("Error with event trigger of name mutex")
|
|
||||||
}
|
|
||||||
return mutex, nil
|
|
||||||
}
|
|
Loading…
Reference in New Issue
Block a user