wireguard-go/ipc/uapi_windows.go

91 lines
1.7 KiB
Go
Raw Normal View History

2019-02-04 16:29:52 +00:00
/* SPDX-License-Identifier: MIT
*
* Copyright (C) 2017-2019 WireGuard LLC. All Rights Reserved.
*/
2019-03-03 03:04:41 +00:00
package ipc
2019-02-04 16:29:52 +00:00
import (
"net"
2019-05-14 07:09:52 +00:00
"golang.org/x/sys/windows"
2019-05-23 12:53:44 +00:00
"golang.zx2c4.com/wireguard/ipc/winpipe"
2019-02-04 16:29:52 +00:00
)
2019-06-06 21:00:15 +00:00
// TODO: replace these with actual standard windows error numbers from the win package
2019-02-04 16:29:52 +00:00
const (
2019-03-03 03:04:41 +00:00
IpcErrorIO = -int64(5)
IpcErrorProtocol = -int64(71)
IpcErrorInvalid = -int64(22)
IpcErrorPortInUse = -int64(98)
2019-02-04 16:29:52 +00:00
)
type UAPIListener struct {
listener net.Listener // unix socket listener
connNew chan net.Conn
connErr chan error
kqueueFd int
keventFd int
}
func (l *UAPIListener) Accept() (net.Conn, error) {
for {
select {
case conn := <-l.connNew:
return conn, nil
case err := <-l.connErr:
return nil, err
}
}
}
func (l *UAPIListener) Close() error {
return l.listener.Close()
}
func (l *UAPIListener) Addr() net.Addr {
return l.listener.Addr()
}
var UAPISecurityDescriptor *windows.SECURITY_DESCRIPTOR
func init() {
var err error
/* SDDL_DEVOBJ_SYS_ALL from the WDK */
UAPISecurityDescriptor, err = windows.SecurityDescriptorFromString("O:SYD:P(A;;GA;;;SY)")
if err != nil {
panic(err)
}
}
2019-02-04 16:29:52 +00:00
func UAPIListen(name string) (net.Listener, error) {
2019-05-23 12:53:44 +00:00
config := winpipe.PipeConfig{
SecurityDescriptor: UAPISecurityDescriptor,
2019-02-04 16:29:52 +00:00
}
2019-08-31 13:48:42 +00:00
listener, err := winpipe.ListenPipe(`\\.\pipe\ProtectedPrefix\Administrators\WireGuard\`+name, &config)
2019-02-04 16:29:52 +00:00
if err != nil {
return nil, err
}
uapi := &UAPIListener{
listener: listener,
connNew: make(chan net.Conn, 1),
connErr: make(chan error, 1),
}
go func(l *UAPIListener) {
for {
conn, err := l.listener.Accept()
if err != nil {
l.connErr <- err
break
}
l.connNew <- conn
}
}(uapi)
return uapi, nil
}