2016-10-12 10:11:13 -04:00
|
|
|
package router_test
|
2015-11-30 10:14:38 -05:00
|
|
|
|
|
|
|
import (
|
2019-02-28 03:28:55 -05:00
|
|
|
"context"
|
2015-11-30 10:14:38 -05:00
|
|
|
"testing"
|
|
|
|
|
2018-10-22 09:58:52 -04:00
|
|
|
"github.com/golang/mock/gomock"
|
2021-02-16 15:31:50 -05:00
|
|
|
|
|
|
|
. "github.com/v2fly/v2ray-core/v4/app/router"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/common"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/common/net"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/common/session"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/features/outbound"
|
|
|
|
routing_session "github.com/v2fly/v2ray-core/v4/features/routing/session"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/testing/mocks"
|
2015-11-30 10:14:38 -05:00
|
|
|
)
|
|
|
|
|
2018-11-07 15:25:43 -05:00
|
|
|
type mockOutboundManager struct {
|
|
|
|
outbound.Manager
|
|
|
|
outbound.HandlerSelector
|
|
|
|
}
|
|
|
|
|
2015-11-30 10:14:38 -05:00
|
|
|
func TestSimpleRouter(t *testing.T) {
|
2018-10-22 09:58:52 -04:00
|
|
|
config := &Config{
|
|
|
|
Rule: []*RoutingRule{
|
|
|
|
{
|
2018-11-07 15:08:20 -05:00
|
|
|
TargetTag: &RoutingRule_Tag{
|
|
|
|
Tag: "test",
|
|
|
|
},
|
2018-11-20 10:58:26 -05:00
|
|
|
Networks: []net.Network{net.Network_TCP},
|
2018-10-22 09:58:52 -04:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
mockCtl := gomock.NewController(t)
|
|
|
|
defer mockCtl.Finish()
|
|
|
|
|
2020-10-11 07:22:46 -04:00
|
|
|
mockDNS := mocks.NewDNSClient(mockCtl)
|
2018-11-07 15:25:43 -05:00
|
|
|
mockOhm := mocks.NewOutboundManager(mockCtl)
|
|
|
|
mockHs := mocks.NewOutboundHandlerSelector(mockCtl)
|
2018-10-22 09:58:52 -04:00
|
|
|
|
|
|
|
r := new(Router)
|
2021-04-09 11:02:45 -04:00
|
|
|
common.Must(r.Init(context.TODO(), config, mockDNS, &mockOutboundManager{
|
2018-11-07 15:25:43 -05:00
|
|
|
Manager: mockOhm,
|
|
|
|
HandlerSelector: mockHs,
|
|
|
|
}))
|
|
|
|
|
2021-02-16 15:31:50 -05:00
|
|
|
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
|
2020-09-18 05:30:59 -04:00
|
|
|
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
|
2018-11-07 15:25:43 -05:00
|
|
|
common.Must(err)
|
2020-09-18 05:30:59 -04:00
|
|
|
if tag := route.GetOutboundTag(); tag != "test" {
|
2018-11-07 15:25:43 -05:00
|
|
|
t.Error("expect tag 'test', bug actually ", tag)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSimpleBalancer(t *testing.T) {
|
|
|
|
config := &Config{
|
|
|
|
Rule: []*RoutingRule{
|
|
|
|
{
|
|
|
|
TargetTag: &RoutingRule_BalancingTag{
|
|
|
|
BalancingTag: "balance",
|
|
|
|
},
|
2018-11-20 10:58:26 -05:00
|
|
|
Networks: []net.Network{net.Network_TCP},
|
2018-11-07 15:25:43 -05:00
|
|
|
},
|
|
|
|
},
|
|
|
|
BalancingRule: []*BalancingRule{
|
|
|
|
{
|
|
|
|
Tag: "balance",
|
|
|
|
OutboundSelector: []string{"test-"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
mockCtl := gomock.NewController(t)
|
|
|
|
defer mockCtl.Finish()
|
|
|
|
|
2020-10-11 07:22:46 -04:00
|
|
|
mockDNS := mocks.NewDNSClient(mockCtl)
|
2018-11-07 15:25:43 -05:00
|
|
|
mockOhm := mocks.NewOutboundManager(mockCtl)
|
|
|
|
mockHs := mocks.NewOutboundHandlerSelector(mockCtl)
|
|
|
|
|
|
|
|
mockHs.EXPECT().Select(gomock.Eq([]string{"test-"})).Return([]string{"test"})
|
|
|
|
|
|
|
|
r := new(Router)
|
2021-04-09 11:02:45 -04:00
|
|
|
common.Must(r.Init(context.TODO(), config, mockDNS, &mockOutboundManager{
|
2018-11-07 15:25:43 -05:00
|
|
|
Manager: mockOhm,
|
|
|
|
HandlerSelector: mockHs,
|
|
|
|
}))
|
2018-10-22 09:58:52 -04:00
|
|
|
|
2021-02-16 15:31:50 -05:00
|
|
|
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
|
2020-09-18 05:30:59 -04:00
|
|
|
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
|
2018-10-22 09:58:52 -04:00
|
|
|
common.Must(err)
|
2020-09-18 05:30:59 -04:00
|
|
|
if tag := route.GetOutboundTag(); tag != "test" {
|
2018-10-22 09:58:52 -04:00
|
|
|
t.Error("expect tag 'test', bug actually ", tag)
|
|
|
|
}
|
|
|
|
}
|
2015-11-30 10:14:38 -05:00
|
|
|
|
2018-10-22 09:58:52 -04:00
|
|
|
func TestIPOnDemand(t *testing.T) {
|
|
|
|
config := &Config{
|
|
|
|
DomainStrategy: Config_IpOnDemand,
|
|
|
|
Rule: []*RoutingRule{
|
|
|
|
{
|
2018-11-07 15:08:20 -05:00
|
|
|
TargetTag: &RoutingRule_Tag{
|
|
|
|
Tag: "test",
|
|
|
|
},
|
2018-10-22 09:58:52 -04:00
|
|
|
Cidr: []*CIDR{
|
2018-01-10 06:22:37 -05:00
|
|
|
{
|
2018-10-22 09:58:52 -04:00
|
|
|
Ip: []byte{192, 168, 0, 0},
|
|
|
|
Prefix: 16,
|
2018-01-10 06:22:37 -05:00
|
|
|
},
|
2016-10-11 17:02:44 -04:00
|
|
|
},
|
2018-10-22 09:58:52 -04:00
|
|
|
},
|
2016-02-01 17:21:54 -05:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2018-10-22 09:58:52 -04:00
|
|
|
mockCtl := gomock.NewController(t)
|
|
|
|
defer mockCtl.Finish()
|
|
|
|
|
2020-10-11 07:22:46 -04:00
|
|
|
mockDNS := mocks.NewDNSClient(mockCtl)
|
2021-03-19 03:55:18 -04:00
|
|
|
mockDNS.EXPECT().LookupIP(gomock.Eq("v2fly.org")).Return([]net.IP{{192, 168, 0, 1}}, nil).AnyTimes()
|
2015-11-30 10:14:38 -05:00
|
|
|
|
2018-10-22 09:58:52 -04:00
|
|
|
r := new(Router)
|
2021-04-09 11:02:45 -04:00
|
|
|
common.Must(r.Init(context.TODO(), config, mockDNS, nil))
|
2017-01-06 09:32:36 -05:00
|
|
|
|
2021-02-16 15:31:50 -05:00
|
|
|
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
|
2020-09-18 05:30:59 -04:00
|
|
|
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
|
2018-10-22 09:58:52 -04:00
|
|
|
common.Must(err)
|
2020-09-18 05:30:59 -04:00
|
|
|
if tag := route.GetOutboundTag(); tag != "test" {
|
2018-10-22 09:58:52 -04:00
|
|
|
t.Error("expect tag 'test', bug actually ", tag)
|
|
|
|
}
|
2015-11-30 10:14:38 -05:00
|
|
|
}
|
2018-12-04 14:36:51 -05:00
|
|
|
|
|
|
|
func TestIPIfNonMatchDomain(t *testing.T) {
|
|
|
|
config := &Config{
|
|
|
|
DomainStrategy: Config_IpIfNonMatch,
|
|
|
|
Rule: []*RoutingRule{
|
|
|
|
{
|
|
|
|
TargetTag: &RoutingRule_Tag{
|
|
|
|
Tag: "test",
|
|
|
|
},
|
|
|
|
Cidr: []*CIDR{
|
|
|
|
{
|
|
|
|
Ip: []byte{192, 168, 0, 0},
|
|
|
|
Prefix: 16,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
mockCtl := gomock.NewController(t)
|
|
|
|
defer mockCtl.Finish()
|
|
|
|
|
2020-10-11 07:22:46 -04:00
|
|
|
mockDNS := mocks.NewDNSClient(mockCtl)
|
2021-03-19 03:55:18 -04:00
|
|
|
mockDNS.EXPECT().LookupIP(gomock.Eq("v2fly.org")).Return([]net.IP{{192, 168, 0, 1}}, nil).AnyTimes()
|
2018-12-04 14:36:51 -05:00
|
|
|
|
|
|
|
r := new(Router)
|
2021-04-09 11:02:45 -04:00
|
|
|
common.Must(r.Init(context.TODO(), config, mockDNS, nil))
|
2018-12-04 14:36:51 -05:00
|
|
|
|
2021-02-16 15:31:50 -05:00
|
|
|
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
|
2020-09-18 05:30:59 -04:00
|
|
|
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
|
2018-12-04 14:36:51 -05:00
|
|
|
common.Must(err)
|
2020-09-18 05:30:59 -04:00
|
|
|
if tag := route.GetOutboundTag(); tag != "test" {
|
2018-12-04 14:36:51 -05:00
|
|
|
t.Error("expect tag 'test', bug actually ", tag)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestIPIfNonMatchIP(t *testing.T) {
|
|
|
|
config := &Config{
|
|
|
|
DomainStrategy: Config_IpIfNonMatch,
|
|
|
|
Rule: []*RoutingRule{
|
|
|
|
{
|
|
|
|
TargetTag: &RoutingRule_Tag{
|
|
|
|
Tag: "test",
|
|
|
|
},
|
|
|
|
Cidr: []*CIDR{
|
|
|
|
{
|
|
|
|
Ip: []byte{127, 0, 0, 0},
|
|
|
|
Prefix: 8,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
mockCtl := gomock.NewController(t)
|
|
|
|
defer mockCtl.Finish()
|
|
|
|
|
2020-10-11 07:22:46 -04:00
|
|
|
mockDNS := mocks.NewDNSClient(mockCtl)
|
2018-12-04 14:36:51 -05:00
|
|
|
|
|
|
|
r := new(Router)
|
2021-04-09 11:02:45 -04:00
|
|
|
common.Must(r.Init(context.TODO(), config, mockDNS, nil))
|
2018-12-04 14:36:51 -05:00
|
|
|
|
2019-02-28 03:28:55 -05:00
|
|
|
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.LocalHostIP, 80)})
|
2020-09-18 05:30:59 -04:00
|
|
|
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
|
2018-12-04 14:36:51 -05:00
|
|
|
common.Must(err)
|
2020-09-18 05:30:59 -04:00
|
|
|
if tag := route.GetOutboundTag(); tag != "test" {
|
2018-12-04 14:36:51 -05:00
|
|
|
t.Error("expect tag 'test', bug actually ", tag)
|
|
|
|
}
|
|
|
|
}
|