2019-01-01 19:55:51 -05:00
|
|
|
/* SPDX-License-Identifier: MIT
|
2018-05-03 09:04:00 -04:00
|
|
|
*
|
2022-09-20 11:21:32 -04:00
|
|
|
* Copyright (C) 2017-2023 WireGuard LLC. All Rights Reserved.
|
2018-05-03 09:04:00 -04:00
|
|
|
*/
|
|
|
|
|
2019-03-02 22:04:41 -05:00
|
|
|
package ipc
|
2017-07-15 07:41:02 -04:00
|
|
|
|
|
|
|
import (
|
|
|
|
"net"
|
|
|
|
"os"
|
2019-05-14 03:09:52 -04:00
|
|
|
|
|
|
|
"golang.org/x/sys/unix"
|
2024-01-07 14:03:11 -05:00
|
|
|
"gitea.hbanafa.com/hesham/wireguard-go/rwcancel"
|
2017-07-15 07:41:02 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
type UAPIListener struct {
|
2018-05-13 18:28:30 -04:00
|
|
|
listener net.Listener // unix socket listener
|
|
|
|
connNew chan net.Conn
|
|
|
|
connErr chan error
|
|
|
|
inotifyFd int
|
|
|
|
inotifyRWCancel *rwcancel.RWCancel
|
2017-07-15 07:41:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
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 {
|
2017-08-11 10:18:20 -04:00
|
|
|
err1 := unix.Close(l.inotifyFd)
|
2018-05-14 06:46:06 -04:00
|
|
|
err2 := l.inotifyRWCancel.Cancel()
|
|
|
|
err3 := l.listener.Close()
|
2017-08-11 10:18:20 -04:00
|
|
|
if err1 != nil {
|
|
|
|
return err1
|
|
|
|
}
|
2018-05-13 18:28:30 -04:00
|
|
|
if err2 != nil {
|
|
|
|
return err2
|
|
|
|
}
|
|
|
|
return err3
|
2017-07-15 07:41:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func (l *UAPIListener) Addr() net.Addr {
|
2018-05-14 06:46:06 -04:00
|
|
|
return l.listener.Addr()
|
2017-07-15 07:41:02 -04:00
|
|
|
}
|
|
|
|
|
2017-11-17 08:36:08 -05:00
|
|
|
func UAPIListen(name string, file *os.File) (net.Listener, error) {
|
|
|
|
// wrap file in listener
|
2017-08-01 06:45:11 -04:00
|
|
|
|
2017-11-17 08:36:08 -05:00
|
|
|
listener, err := net.FileListener(file)
|
2017-07-15 07:41:02 -04:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-14 06:46:06 -04:00
|
|
|
if unixListener, ok := listener.(*net.UnixListener); ok {
|
|
|
|
unixListener.SetUnlinkOnClose(true)
|
|
|
|
}
|
|
|
|
|
2017-07-15 07:41:02 -04:00
|
|
|
uapi := &UAPIListener{
|
|
|
|
listener: listener,
|
|
|
|
connNew: make(chan net.Conn, 1),
|
|
|
|
connErr: make(chan error, 1),
|
|
|
|
}
|
|
|
|
|
|
|
|
// watch for deletion of socket
|
|
|
|
|
2020-05-02 02:28:33 -04:00
|
|
|
socketPath := sockPath(name)
|
2017-11-17 08:36:08 -05:00
|
|
|
|
2017-08-07 09:25:04 -04:00
|
|
|
uapi.inotifyFd, err = unix.InotifyInit()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = unix.InotifyAddWatch(
|
|
|
|
uapi.inotifyFd,
|
|
|
|
socketPath,
|
|
|
|
unix.IN_ATTRIB|
|
|
|
|
unix.IN_DELETE|
|
|
|
|
unix.IN_DELETE_SELF,
|
|
|
|
)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-13 18:28:30 -04:00
|
|
|
uapi.inotifyRWCancel, err = rwcancel.NewRWCancel(uapi.inotifyFd)
|
|
|
|
if err != nil {
|
|
|
|
unix.Close(uapi.inotifyFd)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2017-07-15 07:41:02 -04:00
|
|
|
go func(l *UAPIListener) {
|
2023-03-13 12:55:05 -04:00
|
|
|
var buf [0]byte
|
2017-08-07 09:25:04 -04:00
|
|
|
for {
|
2021-02-09 14:18:21 -05:00
|
|
|
defer uapi.inotifyRWCancel.Close()
|
2017-11-17 08:36:08 -05:00
|
|
|
// start with lstat to avoid race condition
|
2017-08-07 09:25:04 -04:00
|
|
|
if _, err := os.Lstat(socketPath); os.IsNotExist(err) {
|
2017-07-15 07:41:02 -04:00
|
|
|
l.connErr <- err
|
|
|
|
return
|
|
|
|
}
|
2023-03-13 12:55:05 -04:00
|
|
|
_, err := uapi.inotifyRWCancel.Read(buf[:])
|
2018-05-13 18:28:30 -04:00
|
|
|
if err != nil {
|
|
|
|
l.connErr <- err
|
|
|
|
return
|
|
|
|
}
|
2017-07-15 07:41:02 -04:00
|
|
|
}
|
|
|
|
}(uapi)
|
|
|
|
|
|
|
|
// watch for new connections
|
|
|
|
|
|
|
|
go func(l *UAPIListener) {
|
|
|
|
for {
|
|
|
|
conn, err := l.listener.Accept()
|
|
|
|
if err != nil {
|
|
|
|
l.connErr <- err
|
|
|
|
break
|
|
|
|
}
|
|
|
|
l.connNew <- conn
|
|
|
|
}
|
|
|
|
}(uapi)
|
|
|
|
|
|
|
|
return uapi, nil
|
|
|
|
}
|