mirror of
https://github.com/p4gefau1t/trojan-go.git
synced 2024-04-21 12:21:34 +00:00
71 lines
2.1 KiB
Go
71 lines
2.1 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/p4gefau1t/trojan-go/config"
|
|
"github.com/p4gefau1t/trojan-go/proxy"
|
|
"github.com/p4gefau1t/trojan-go/proxy/client"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/freedom"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/mux"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/router"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/shadowsocks"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/simplesocks"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/tls"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/transport"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/trojan"
|
|
"github.com/p4gefau1t/trojan-go/tunnel/websocket"
|
|
)
|
|
|
|
const Name = "SERVER"
|
|
|
|
func init() {
|
|
proxy.RegisterProxyCreator(Name, func(ctx context.Context) (*proxy.Proxy, error) {
|
|
cfg := config.FromContext(ctx, Name).(*client.Config)
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
transportServer, err := transport.NewServer(ctx, nil)
|
|
if err != nil {
|
|
cancel()
|
|
return nil, err
|
|
}
|
|
clientStack := []string{freedom.Name}
|
|
if cfg.Router.Enabled {
|
|
clientStack = []string{freedom.Name, router.Name}
|
|
}
|
|
|
|
root := &proxy.Node{
|
|
Name: transport.Name,
|
|
Next: make(map[string]*proxy.Node),
|
|
IsEndpoint: false,
|
|
Context: ctx,
|
|
Server: transportServer,
|
|
}
|
|
|
|
if !cfg.TransportPlugin.Enabled {
|
|
root = root.BuildNext(tls.Name)
|
|
}
|
|
|
|
trojanSubTree := root
|
|
if cfg.Shadowsocks.Enabled {
|
|
trojanSubTree = trojanSubTree.BuildNext(shadowsocks.Name)
|
|
}
|
|
trojanSubTree.BuildNext(trojan.Name).BuildNext(mux.Name).BuildNext(simplesocks.Name).IsEndpoint = true
|
|
trojanSubTree.BuildNext(trojan.Name).IsEndpoint = true
|
|
|
|
wsSubTree := root.BuildNext(websocket.Name)
|
|
if cfg.Shadowsocks.Enabled {
|
|
wsSubTree = wsSubTree.BuildNext(shadowsocks.Name)
|
|
}
|
|
wsSubTree.BuildNext(trojan.Name).BuildNext(mux.Name).BuildNext(simplesocks.Name).IsEndpoint = true
|
|
wsSubTree.BuildNext(trojan.Name).IsEndpoint = true
|
|
|
|
serverList := proxy.FindAllEndpoints(root)
|
|
clientList, err := proxy.CreateClientStack(ctx, clientStack)
|
|
if err != nil {
|
|
cancel()
|
|
return nil, err
|
|
}
|
|
return proxy.NewProxy(ctx, cancel, serverList, clientList), nil
|
|
})
|
|
}
|