mirror of https://github.com/v2ray/v2ray-core
102 lines
1.6 KiB
Go
102 lines
1.6 KiB
Go
package commander
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"sync"
|
|
|
|
"v2ray.com/core/common/signal"
|
|
"v2ray.com/core/transport/ray"
|
|
)
|
|
|
|
type OutboundListener struct {
|
|
buffer chan net.Conn
|
|
done *signal.Done
|
|
}
|
|
|
|
func (l *OutboundListener) add(conn net.Conn) {
|
|
select {
|
|
case l.buffer <- conn:
|
|
case <-l.done.C():
|
|
conn.Close()
|
|
default:
|
|
conn.Close()
|
|
}
|
|
}
|
|
|
|
func (l *OutboundListener) Accept() (net.Conn, error) {
|
|
select {
|
|
case <-l.done.C():
|
|
return nil, newError("listern closed")
|
|
case c := <-l.buffer:
|
|
return c, nil
|
|
}
|
|
}
|
|
|
|
func (l *OutboundListener) Close() error {
|
|
l.done.Close()
|
|
L:
|
|
for {
|
|
select {
|
|
case c := <-l.buffer:
|
|
c.Close()
|
|
default:
|
|
break L
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (l *OutboundListener) Addr() net.Addr {
|
|
return &net.TCPAddr{
|
|
IP: net.IP{0, 0, 0, 0},
|
|
Port: 0,
|
|
}
|
|
}
|
|
|
|
type CommanderOutbound struct {
|
|
tag string
|
|
listener *OutboundListener
|
|
access sync.RWMutex
|
|
closed bool
|
|
}
|
|
|
|
func (co *CommanderOutbound) Dispatch(ctx context.Context, r ray.OutboundRay) {
|
|
co.access.RLock()
|
|
|
|
if co.closed {
|
|
r.OutboundInput().CloseError()
|
|
r.OutboundOutput().CloseError()
|
|
co.access.RUnlock()
|
|
return
|
|
}
|
|
|
|
closeSignal := signal.NewNotifier()
|
|
c := ray.NewConnection(r.OutboundInput(), r.OutboundOutput(), ray.ConnCloseSignal(closeSignal))
|
|
co.listener.add(c)
|
|
co.access.RUnlock()
|
|
<-closeSignal.Wait()
|
|
|
|
return
|
|
}
|
|
|
|
func (co *CommanderOutbound) Tag() string {
|
|
return co.tag
|
|
}
|
|
|
|
func (co *CommanderOutbound) Start() error {
|
|
co.access.Lock()
|
|
co.closed = false
|
|
co.access.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (co *CommanderOutbound) Close() error {
|
|
co.access.Lock()
|
|
co.closed = true
|
|
co.listener.Close()
|
|
co.access.Unlock()
|
|
|
|
return nil
|
|
}
|