mihomo/listener/mixed/mixed.go

71 lines
1.3 KiB
Go
Raw Normal View History

package mixed
import (
"net"
"time"
"github.com/Dreamacro/clash/common/cache"
2021-06-15 17:13:40 +08:00
N "github.com/Dreamacro/clash/common/net"
2021-06-13 17:23:10 +08:00
C "github.com/Dreamacro/clash/constant"
"github.com/Dreamacro/clash/listener/http"
"github.com/Dreamacro/clash/listener/socks"
"github.com/Dreamacro/clash/transport/socks4"
2021-05-13 22:18:49 +08:00
"github.com/Dreamacro/clash/transport/socks5"
)
2021-06-13 17:23:10 +08:00
type Listener struct {
listener net.Listener
address string
closed bool
cache *cache.Cache
}
2021-06-13 17:23:10 +08:00
func New(addr string, in chan<- C.ConnContext) (*Listener, error) {
l, err := net.Listen("tcp", addr)
if err != nil {
return nil, err
}
2021-06-13 17:23:10 +08:00
ml := &Listener{l, addr, false, cache.New(30 * time.Second)}
go func() {
for {
c, err := ml.listener.Accept()
if err != nil {
if ml.closed {
break
}
continue
}
2021-06-13 17:23:10 +08:00
go handleConn(c, in, ml.cache)
}
}()
return ml, nil
}
2021-06-13 17:23:10 +08:00
func (l *Listener) Close() {
l.closed = true
l.listener.Close()
}
2021-06-13 17:23:10 +08:00
func (l *Listener) Address() string {
return l.address
}
2021-06-13 17:23:10 +08:00
func handleConn(conn net.Conn, in chan<- C.ConnContext, cache *cache.Cache) {
2021-06-15 17:13:40 +08:00
bufConn := N.NewBufferedConn(conn)
head, err := bufConn.Peek(1)
if err != nil {
return
}
switch head[0] {
case socks4.Version:
socks.HandleSocks4(bufConn, in)
case socks5.Version:
socks.HandleSocks5(bufConn, in)
default:
http.HandleConn(bufConn, in, cache)
}
}