yuandian/slimming
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

测试

Browse Source
This commit is contained in:
黄思敏 2022-08-26 09:57:09 +08:00
parent 676acb3198
commit 4d0f25400a
2 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
config.go
View File

@ -23,7 +23,7 @@ type Config struct {
} `yaml:"network"` } `yaml:"network"`
IPv4Nodes map[uint32]string IPv4Nodes map[uint32]string
Lock sync.Mutex lock sync.Mutex
} }
var config = NewConfig() var config = NewConfig()
@ -56,3 +56,9 @@ func NewConfig() *Config {
return cnf return cnf
} }
func (cnf *Config) Lock(do func()) {
cnf.lock.Lock()
defer cnf.lock.Unlock()
do()
}

11
tap.go
View File

@ -167,12 +167,15 @@ func (nc *NetCard) runRead() {
continue continue
} }
var realAddr string var realAddr string = ""
config.Lock.Lock() config.Lock(func() {
if _, ok = config.IPv4Nodes[binary.LittleEndian.Uint32([]byte(waterutil.IPv4Destination(rframe)))]; !ok { if realAddr, ok = config.IPv4Nodes[binary.LittleEndian.Uint32([]byte(waterutil.IPv4Destination(rframe)))]; !ok {
return
}
})
if realAddr == "" {
continue continue
} }
config.Lock.Unlock()
// log.Printf("Payload: % x\n", rframe.Payload()) // log.Printf("Payload: % x\n", rframe.Payload())