mirror of https://github.com/v2fly/v2ray-core.git
97 lines
2.0 KiB
Go
97 lines
2.0 KiB
Go
package reverse
|
|
|
|
//go:generate go run github.com/v2fly/v2ray-core/v4/common/errors/errorgen
|
|
|
|
import (
|
|
"context"
|
|
|
|
core "github.com/v2fly/v2ray-core/v4"
|
|
"github.com/v2fly/v2ray-core/v4/common"
|
|
"github.com/v2fly/v2ray-core/v4/common/errors"
|
|
"github.com/v2fly/v2ray-core/v4/common/net"
|
|
"github.com/v2fly/v2ray-core/v4/features/outbound"
|
|
"github.com/v2fly/v2ray-core/v4/features/routing"
|
|
)
|
|
|
|
const (
|
|
internalDomain = "reverse.internal.v2fly.org"
|
|
)
|
|
|
|
func isDomain(dest net.Destination, domain string) bool {
|
|
return dest.Address.Family().IsDomain() && dest.Address.Domain() == domain
|
|
}
|
|
|
|
func isInternalDomain(dest net.Destination) bool {
|
|
return isDomain(dest, internalDomain)
|
|
}
|
|
|
|
func init() {
|
|
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
|
|
r := new(Reverse)
|
|
if err := core.RequireFeatures(ctx, func(d routing.Dispatcher, om outbound.Manager) error {
|
|
return r.Init(config.(*Config), d, om)
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
return r, nil
|
|
}))
|
|
}
|
|
|
|
type Reverse struct {
|
|
bridges []*Bridge
|
|
portals []*Portal
|
|
}
|
|
|
|
func (r *Reverse) Init(config *Config, d routing.Dispatcher, ohm outbound.Manager) error {
|
|
for _, bConfig := range config.BridgeConfig {
|
|
b, err := NewBridge(bConfig, d)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
r.bridges = append(r.bridges, b)
|
|
}
|
|
|
|
for _, pConfig := range config.PortalConfig {
|
|
p, err := NewPortal(pConfig, ohm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
r.portals = append(r.portals, p)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *Reverse) Type() interface{} {
|
|
return (*Reverse)(nil)
|
|
}
|
|
|
|
func (r *Reverse) Start() error {
|
|
for _, b := range r.bridges {
|
|
if err := b.Start(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, p := range r.portals {
|
|
if err := p.Start(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *Reverse) Close() error {
|
|
var errs []error
|
|
for _, b := range r.bridges {
|
|
errs = append(errs, b.Close())
|
|
}
|
|
|
|
for _, p := range r.portals {
|
|
errs = append(errs, p.Close())
|
|
}
|
|
|
|
return errors.Combine(errs...)
|
|
}
|