package fakeip import ( "fmt" "net" "os" "testing" "time" "github.com/Dreamacro/clash/component/profile/cachefile" "github.com/Dreamacro/clash/component/trie" "github.com/stretchr/testify/assert" "go.etcd.io/bbolt" ) func createPools(options Options) ([]*Pool, string, error) { pool, err := New(options) if err != nil { return nil, "", err } filePool, tempfile, err := createCachefileStore(options) if err != nil { return nil, "", err } return []*Pool{pool, filePool}, tempfile, nil } func createCachefileStore(options Options) (*Pool, string, error) { pool, err := New(options) if err != nil { return nil, "", err } f, err := os.CreateTemp("", "clash") if err != nil { return nil, "", err } db, err := bbolt.Open(f.Name(), 0o666, &bbolt.Options{Timeout: time.Second}) if err != nil { return nil, "", err } pool.store = &cachefileStore{ cache: &cachefile.CacheFile{DB: db}, } return pool, f.Name(), nil } func TestPool_Basic(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.0/28") pools, tempfile, err := createPools(Options{ IPNet: ipnet, Size: 10, }) assert.Nil(t, err) defer os.Remove(tempfile) for _, pool := range pools { first := pool.Lookup("foo.com") last := pool.Lookup("bar.com") bar, exist := pool.LookBack(last) assert.True(t, first.Equal(net.IP{192, 168, 0, 3})) assert.Equal(t, pool.Lookup("foo.com"), net.IP{192, 168, 0, 3}) assert.True(t, last.Equal(net.IP{192, 168, 0, 4})) assert.True(t, exist) assert.Equal(t, bar, "bar.com") assert.Equal(t, pool.Gateway(), net.IP{192, 168, 0, 1}) assert.Equal(t, pool.Broadcast(), net.IP{192, 168, 0, 15}) assert.Equal(t, pool.IPNet().String(), ipnet.String()) assert.True(t, pool.Exist(net.IP{192, 168, 0, 4})) assert.False(t, pool.Exist(net.IP{192, 168, 0, 5})) assert.False(t, pool.Exist(net.ParseIP("::1"))) } } func TestPool_CycleUsed(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.16/28") pools, tempfile, err := createPools(Options{ IPNet: ipnet, Size: 10, }) assert.Nil(t, err) defer os.Remove(tempfile) for _, pool := range pools { foo := pool.Lookup("foo.com") bar := pool.Lookup("bar.com") for i := 0; i < 9; i++ { pool.Lookup(fmt.Sprintf("%d.com", i)) } baz := pool.Lookup("baz.com") next := pool.Lookup("foo.com") assert.True(t, foo.Equal(baz)) assert.True(t, next.Equal(bar)) } } func TestPool_Skip(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.1/29") tree := trie.New[bool]() tree.Insert("example.com", true) pools, tempfile, err := createPools(Options{ IPNet: ipnet, Size: 10, Host: tree, }) assert.Nil(t, err) defer os.Remove(tempfile) for _, pool := range pools { assert.True(t, pool.ShouldSkipped("example.com")) assert.False(t, pool.ShouldSkipped("foo.com")) } } func TestPool_MaxCacheSize(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.1/24") pool, _ := New(Options{ IPNet: ipnet, Size: 2, }) first := pool.Lookup("foo.com") pool.Lookup("bar.com") pool.Lookup("baz.com") next := pool.Lookup("foo.com") assert.False(t, first.Equal(next)) } func TestPool_DoubleMapping(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.1/24") pool, _ := New(Options{ IPNet: ipnet, Size: 2, }) // fill cache fooIP := pool.Lookup("foo.com") bazIP := pool.Lookup("baz.com") // make foo.com hot pool.Lookup("foo.com") // should drop baz.com barIP := pool.Lookup("bar.com") _, fooExist := pool.LookBack(fooIP) _, bazExist := pool.LookBack(bazIP) _, barExist := pool.LookBack(barIP) newBazIP := pool.Lookup("baz.com") assert.True(t, fooExist) assert.False(t, bazExist) assert.True(t, barExist) assert.False(t, bazIP.Equal(newBazIP)) } func TestPool_Clone(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.1/24") pool, _ := New(Options{ IPNet: ipnet, Size: 2, }) first := pool.Lookup("foo.com") last := pool.Lookup("bar.com") assert.True(t, first.Equal(net.IP{192, 168, 0, 3})) assert.True(t, last.Equal(net.IP{192, 168, 0, 4})) newPool, _ := New(Options{ IPNet: ipnet, Size: 2, }) newPool.CloneFrom(pool) _, firstExist := newPool.LookBack(first) _, lastExist := newPool.LookBack(last) assert.True(t, firstExist) assert.True(t, lastExist) } func TestPool_Error(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.1/31") _, err := New(Options{ IPNet: ipnet, Size: 10, }) assert.Error(t, err) } func TestPool_FlushFileCache(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.1/28") pools, tempfile, err := createPools(Options{ IPNet: ipnet, Size: 10, }) assert.Nil(t, err) defer os.Remove(tempfile) for _, pool := range pools { foo := pool.Lookup("foo.com") bar := pool.Lookup("baz.com") bax := pool.Lookup("baz.com") fox := pool.Lookup("foo.com") err = pool.FlushFakeIP() assert.Nil(t, err) baz := pool.Lookup("foo.com") next := pool.Lookup("baz.com") nero := pool.Lookup("foo.com") assert.Equal(t, foo, fox) assert.NotEqual(t, foo, baz) assert.Equal(t, bar, bax) assert.NotEqual(t, bar, next) assert.Equal(t, baz, nero) } } func TestPool_FlushMemoryCache(t *testing.T) { _, ipnet, _ := net.ParseCIDR("192.168.0.1/28") pool, _ := New(Options{ IPNet: ipnet, Size: 10, }) foo := pool.Lookup("foo.com") bar := pool.Lookup("baz.com") bax := pool.Lookup("baz.com") fox := pool.Lookup("foo.com") err := pool.FlushFakeIP() assert.Nil(t, err) baz := pool.Lookup("foo.com") next := pool.Lookup("baz.com") nero := pool.Lookup("foo.com") assert.Equal(t, foo, fox) assert.NotEqual(t, foo, baz) assert.Equal(t, bar, bax) assert.NotEqual(t, bar, next) assert.Equal(t, baz, nero) }