Fix: fakeip missing host

common/cache/lrucache.go

@@ -86,6 +86,15 @@ func (c *LruCache) Get(key interface{}) (interface{}, bool) {
 	return value, true
 }
 
+// Exist returns if key exist in cache but not put item to the head of linked list
+func (c *LruCache) Exist(key interface{}) bool {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+
+	_, ok := c.cache[key]
+	return ok
+}
+
 // Set stores the interface{} representation of a response for a given key.
 func (c *LruCache) Set(key interface{}, value interface{}) {
 	c.mu.Lock()

component/fakeip/pool.go

@@ -22,8 +22,14 @@ type Pool struct {
 func (p *Pool) Lookup(host string) net.IP {
 	p.mux.Lock()
 	defer p.mux.Unlock()
-	if ip, exist := p.cache.Get(host); exist {
+	if elm, exist := p.cache.Get(host); exist {
-		return ip.(net.IP)
+		ip := elm.(net.IP)
+
+		// ensure ip --> host on head of linked list
+		n := ipToUint(ip.To4())
+		offset := n - p.min + 1
+		p.cache.Get(offset)
+		return ip
 	}
 
 	ip := p.get(host)
@@ -43,8 +49,12 @@ func (p *Pool) LookBack(ip net.IP) (string, bool) {
 	n := ipToUint(ip.To4())
 	offset := n - p.min + 1
 
-	if host, exist := p.cache.Get(offset); exist {
+	if elm, exist := p.cache.Get(offset); exist {
-		return host.(string), true
+		host := elm.(string)
+
+		// ensure host --> ip on head of linked list
+		p.cache.Get(host)
+		return host, true
 	}
 
 	return "", false
@@ -64,7 +74,7 @@ func (p *Pool) get(host string) net.IP {
 			break
 		}
 
-		if _, exist := p.cache.Get(p.offset); !exist {
+		if !p.cache.Exist(p.offset) {
 			break
 		}
 	}

component/fakeip/pool_test.go

@@ -43,6 +43,33 @@ func TestPool_MaxCacheSize(t *testing.T) {
 	assert.False(t, first.Equal(next))
 }
 
+func TestPool_DoubleMapping(t *testing.T) {
+	_, ipnet, _ := net.ParseCIDR("192.168.0.1/24")
+	pool, _ := New(ipnet, 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_Error(t *testing.T) {
 	_, ipnet, _ := net.ParseCIDR("192.168.0.1/31")
 	_, err := New(ipnet, 10)