xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix: add mutex for fake ip pool

Browse source
This commit is contained in:
gVisor bot 2019-05-23 23:27:29 +08:00
parent 0a6f9b3878
commit d883e7986e
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
component/fakeip/pool.go
View file

@ -3,6 +3,7 @@ package fakeip
import ( import (
"errors" "errors"
"net" "net"
"sync"
) )
// Pool is a implementation about fake ip generator without storage // Pool is a implementation about fake ip generator without storage
@ -10,10 +11,13 @@ type Pool struct {
max uint32 max uint32
min uint32 min uint32
offset uint32 offset uint32
mux *sync.Mutex
} }
// Get return a new fake ip // Get return a new fake ip
func (p *Pool) Get() net.IP { func (p *Pool) Get() net.IP {
p.mux.Lock()
defer p.mux.Unlock()
ip := uintToIP(p.min + p.offset) ip := uintToIP(p.min + p.offset)
p.offset = (p.offset + 1) % (p.max - p.min) p.offset = (p.offset + 1) % (p.max - p.min)
return ip return ip
@ -46,5 +50,6 @@ func New(ipnet *net.IPNet) (*Pool, error) {
return &Pool{ return &Pool{
min: min, min: min,
max: max, max: max,
mux: &sync.Mutex{},
}, nil }, nil
} }