Austin/mihomo
1
0
Fork 0
mirror of https://github.com/MetaCubeX/mihomo.git synced 2024-11-16 11:42:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix: add mutex for fake ip pool

Browse Source
This commit is contained in:
Dreamacro 2019-05-23 23:27:29 +08:00
parent 89168e6c96
commit ad13ad8dba
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
} }