120 lines
2 KiB
Go
120 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"golang.org/x/sys/unix"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
ipcErrorIO = int64(unix.EIO)
|
|
ipcErrorNoPeer = int64(unix.EPROTO)
|
|
ipcErrorNoKeyValue = int64(unix.EPROTO)
|
|
ipcErrorInvalidKey = int64(unix.EPROTO)
|
|
ipcErrorInvalidValue = int64(unix.EPROTO)
|
|
)
|
|
|
|
/* TODO:
|
|
* This code can be improved by using fsnotify once:
|
|
* https://github.com/fsnotify/fsnotify/pull/205
|
|
* Is merged
|
|
*/
|
|
|
|
type UAPIListener struct {
|
|
listener net.Listener // unix socket listener
|
|
connNew chan net.Conn
|
|
connErr chan error
|
|
}
|
|
|
|
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 nil
|
|
}
|
|
|
|
func connectUnixSocket(path string) (net.Listener, error) {
|
|
|
|
// attempt inital connection
|
|
|
|
listener, err := net.Listen("unix", path)
|
|
if err == nil {
|
|
return listener, nil
|
|
}
|
|
|
|
// check if active
|
|
|
|
_, err = net.Dial("unix", path)
|
|
if err == nil {
|
|
return nil, errors.New("Unix socket in use")
|
|
}
|
|
|
|
// attempt cleanup
|
|
|
|
err = os.Remove(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return net.Listen("unix", path)
|
|
}
|
|
|
|
func NewUAPIListener(name string) (net.Listener, error) {
|
|
|
|
// open UNIX socket
|
|
|
|
socketPath := fmt.Sprintf("/var/run/wireguard/%s.sock", name)
|
|
listener, err := connectUnixSocket(socketPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
uapi := &UAPIListener{
|
|
listener: listener,
|
|
connNew: make(chan net.Conn, 1),
|
|
connErr: make(chan error, 1),
|
|
}
|
|
|
|
// watch for deletion of socket
|
|
|
|
go func(l *UAPIListener) {
|
|
for ; ; time.Sleep(time.Second) {
|
|
if _, err := os.Stat(socketPath); os.IsNotExist(err) {
|
|
l.connErr <- err
|
|
return
|
|
}
|
|
}
|
|
}(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
|
|
}
|