mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-10-19 10:13:37 -04:00
130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
//go:build !windows && !wasm
|
|
// +build !windows,!wasm
|
|
|
|
package domainsocket
|
|
|
|
import (
|
|
"context"
|
|
gotls "crypto/tls"
|
|
"os"
|
|
"strings"
|
|
|
|
"golang.org/x/sys/unix"
|
|
|
|
"github.com/v2fly/v2ray-core/v5/common"
|
|
"github.com/v2fly/v2ray-core/v5/common/net"
|
|
"github.com/v2fly/v2ray-core/v5/transport/internet"
|
|
"github.com/v2fly/v2ray-core/v5/transport/internet/tls"
|
|
)
|
|
|
|
type Listener struct {
|
|
addr *net.UnixAddr
|
|
ln net.Listener
|
|
tlsConfig *gotls.Config
|
|
config *Config
|
|
addConn internet.ConnHandler
|
|
locker *fileLocker
|
|
}
|
|
|
|
func Listen(ctx context.Context, address net.Address, port net.Port, streamSettings *internet.MemoryStreamConfig, handler internet.ConnHandler) (internet.Listener, error) {
|
|
settings := streamSettings.ProtocolSettings.(*Config)
|
|
addr, err := settings.GetUnixAddr()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
unixListener, err := net.ListenUnix("unix", addr)
|
|
if err != nil {
|
|
return nil, newError("failed to listen domain socket").Base(err).AtWarning()
|
|
}
|
|
|
|
ln := &Listener{
|
|
addr: addr,
|
|
ln: unixListener,
|
|
config: settings,
|
|
addConn: handler,
|
|
}
|
|
|
|
if !settings.Abstract {
|
|
ln.locker = &fileLocker{
|
|
path: settings.Path + ".lock",
|
|
}
|
|
if err := ln.locker.Acquire(); err != nil {
|
|
unixListener.Close()
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
if config := tls.ConfigFromStreamSettings(streamSettings); config != nil {
|
|
ln.tlsConfig = config.GetTLSConfig()
|
|
}
|
|
|
|
go ln.run()
|
|
|
|
return ln, nil
|
|
}
|
|
|
|
func (ln *Listener) Addr() net.Addr {
|
|
return ln.addr
|
|
}
|
|
|
|
func (ln *Listener) Close() error {
|
|
if ln.locker != nil {
|
|
ln.locker.Release()
|
|
}
|
|
return ln.ln.Close()
|
|
}
|
|
|
|
func (ln *Listener) run() {
|
|
for {
|
|
conn, err := ln.ln.Accept()
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "closed") {
|
|
break
|
|
}
|
|
newError("failed to accepted raw connections").Base(err).AtWarning().WriteToLog()
|
|
continue
|
|
}
|
|
|
|
if ln.tlsConfig != nil {
|
|
conn = tls.Server(conn, ln.tlsConfig)
|
|
}
|
|
|
|
ln.addConn(internet.Connection(conn))
|
|
}
|
|
}
|
|
|
|
type fileLocker struct {
|
|
path string
|
|
file *os.File
|
|
}
|
|
|
|
func (fl *fileLocker) Acquire() error {
|
|
f, err := os.Create(fl.path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := unix.Flock(int(f.Fd()), unix.LOCK_EX); err != nil {
|
|
f.Close()
|
|
return newError("failed to lock file: ", fl.path).Base(err)
|
|
}
|
|
fl.file = f
|
|
return nil
|
|
}
|
|
|
|
func (fl *fileLocker) Release() {
|
|
if err := unix.Flock(int(fl.file.Fd()), unix.LOCK_UN); err != nil {
|
|
newError("failed to unlock file: ", fl.path).Base(err).WriteToLog()
|
|
}
|
|
if err := fl.file.Close(); err != nil {
|
|
newError("failed to close file: ", fl.path).Base(err).WriteToLog()
|
|
}
|
|
if err := os.Remove(fl.path); err != nil {
|
|
newError("failed to remove file: ", fl.path).Base(err).WriteToLog()
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
common.Must(internet.RegisterTransportListener(protocolName, Listen))
|
|
}
|