ginuerzh
ef58909fb7
Fix setMark early-return conditions that were inverted relative to Linux: on FreeBSD (SO_USER_COOKIE) and OpenBSD (SO_RTABLE), a non-zero mark was short-circuited instead of applied, disabling socket marking entirely. Also fix GetClientIP to trim leading whitespace from X-Forwarded-For entries (per RFC 7239), and fix Body.Read to subtract len(b) instead of n from recordSize when a single read exceeds the remaining quota. Add unit tests for internal/net/ (addr, dialer, http, ip, net, pipe, transport, resolve, proxyproto, udp).
141 lines
3.3 KiB
Go
141 lines
3.3 KiB
Go
package net
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/go-gost/core/hosts"
|
|
"github.com/go-gost/core/resolver"
|
|
)
|
|
|
|
type mockResolver struct {
|
|
ips []net.IP
|
|
err error
|
|
}
|
|
|
|
func (m *mockResolver) Resolve(ctx context.Context, network, host string, opts ...resolver.Option) ([]net.IP, error) {
|
|
if m.err != nil {
|
|
return nil, m.err
|
|
}
|
|
return m.ips, nil
|
|
}
|
|
|
|
type mockHostMapper struct {
|
|
ips []net.IP
|
|
}
|
|
|
|
func (m *mockHostMapper) Lookup(ctx context.Context, network, host string, opts ...hosts.Option) ([]net.IP, bool) {
|
|
if len(m.ips) == 0 {
|
|
return nil, false
|
|
}
|
|
return m.ips, true
|
|
}
|
|
|
|
func TestResolve_EmptyAddr(t *testing.T) {
|
|
ctx := context.Background()
|
|
addr, err := Resolve(ctx, "tcp", "", nil, nil, nil)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if addr != "" {
|
|
t.Errorf("expected empty, got %s", addr)
|
|
}
|
|
}
|
|
|
|
func TestResolve_NoHost(t *testing.T) {
|
|
ctx := context.Background()
|
|
addr, err := Resolve(ctx, "tcp", ":8080", nil, nil, nil)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if addr != ":8080" {
|
|
t.Errorf("expected :8080, got %s", addr)
|
|
}
|
|
}
|
|
|
|
func TestResolve_HostMapperMatch(t *testing.T) {
|
|
ctx := context.Background()
|
|
mapper := &mockHostMapper{ips: []net.IP{net.ParseIP("10.0.0.1")}}
|
|
r := &mockResolver{ips: []net.IP{net.ParseIP("192.168.1.1")}}
|
|
|
|
addr, err := Resolve(ctx, "tcp", "example.com:8080", r, mapper, nil)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if addr != "10.0.0.1:8080" {
|
|
t.Errorf("expected 10.0.0.1:8080, got %s", addr)
|
|
}
|
|
}
|
|
|
|
func TestResolve_Resolver(t *testing.T) {
|
|
ctx := context.Background()
|
|
r := &mockResolver{ips: []net.IP{net.ParseIP("192.168.1.1")}}
|
|
|
|
addr, err := Resolve(ctx, "tcp", "example.com:8080", r, nil, nil)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if addr != "192.168.1.1:8080" {
|
|
t.Errorf("expected 192.168.1.1:8080, got %s", addr)
|
|
}
|
|
}
|
|
|
|
func TestResolve_ResolverInvalid(t *testing.T) {
|
|
ctx := context.Background()
|
|
r := &mockResolver{err: resolver.ErrInvalid}
|
|
|
|
addr, err := Resolve(ctx, "tcp", "example.com:8080", r, nil, nil)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if addr != "example.com:8080" {
|
|
t.Errorf("expected example.com:8080, got %s", addr)
|
|
}
|
|
}
|
|
|
|
func TestResolve_ResolverOtherError(t *testing.T) {
|
|
ctx := context.Background()
|
|
r := &mockResolver{err: net.UnknownNetworkError("test-error")}
|
|
|
|
_, err := Resolve(ctx, "tcp", "example.com:8080", r, nil, nil)
|
|
if err == nil {
|
|
t.Error("expected error, got nil")
|
|
}
|
|
}
|
|
|
|
func TestResolve_ResolverEmptyResult(t *testing.T) {
|
|
ctx := context.Background()
|
|
r := &mockResolver{ips: nil}
|
|
|
|
_, err := Resolve(ctx, "tcp", "example.com:8080", r, nil, nil)
|
|
if err == nil {
|
|
t.Error("expected error, got nil")
|
|
}
|
|
}
|
|
|
|
func TestResolve_NoResolverNoHosts(t *testing.T) {
|
|
ctx := context.Background()
|
|
addr, err := Resolve(ctx, "tcp", "example.com:8080", nil, nil, nil)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if addr != "example.com:8080" {
|
|
t.Errorf("expected example.com:8080, got %s", addr)
|
|
}
|
|
}
|
|
|
|
func TestResolve_HostsNoMatchUsesResolver(t *testing.T) {
|
|
ctx := context.Background()
|
|
mapper := &mockHostMapper{}
|
|
r := &mockResolver{ips: []net.IP{net.ParseIP("10.0.0.1")}}
|
|
|
|
addr, err := Resolve(ctx, "tcp", "example.com:8080", r, mapper, nil)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if addr != "10.0.0.1:8080" {
|
|
t.Errorf("expected 10.0.0.1:8080, got %s", addr)
|
|
}
|
|
}
|