This commit is contained in:
黄思敏 2022-08-29 10:10:45 +08:00
parent d1ed742281
commit 0a95455170

View File

@ -11,6 +11,7 @@ import (
"os/exec"
gen "slimming/proto/gen"
"strings"
"sync"
"sync/atomic"
"time"
@ -31,6 +32,14 @@ type NetCard struct {
server *RPCServer
clientMap map[string]*RPCClient
lock sync.Mutex
}
func (nc *NetCard) Lock(do func()) {
nc.lock.Lock()
defer nc.lock.Unlock()
do()
}
type RPCClient struct {
@ -176,11 +185,14 @@ func (nc *NetCard) runRead() {
var rframe ethernet.Frame
rframe.Resize(1500)
nc.Lock(func() {
n, err := ifce.Read([]byte(rframe))
if err != nil {
log.Fatal(err)
}
rframe = rframe[:n]
})
if !waterutil.IsIPv4(rframe) || waterutil.IPv4Source(rframe).Equal(net.IPv4(0, 0, 0, 0)) {
continue
}
@ -260,12 +272,15 @@ func (nc *NetCard) runWrite() {
log.Printf("get wframes decode len: %d", len(bufs[0]))
nc.Lock(func() {
for _, buf := range bufs {
_, err := ifce.Write(buf)
if err != nil {
log.Panic(err)
}
}
})
}
}