Skip to content

Commit 39fb38e

Browse files
committed
proxy: update activity pre and post packet io
To capture waiting for packet and received packet.
1 parent 1201b27 commit 39fb38e

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

proxy/proxy_device.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,11 +302,13 @@ func (self *ProxyDevice) Run() {
302302
case <-self.ctx.Done():
303303
return
304304
case receive <- packet:
305+
self.UpdateActivity()
305306
return
306307
case <-notify:
307308
}
308309
} else {
309310
self.tun.Write(packet)
311+
self.UpdateActivity()
310312
return
311313
}
312314
}
@@ -322,6 +324,9 @@ func (self *ProxyDevice) Run() {
322324
if err != nil {
323325
return
324326
}
327+
if !self.UpdateActivity() {
328+
return
329+
}
325330
success := self.deviceLocal.SendPacketNoCopy(packet, int32(len(packet)))
326331
if !success {
327332
connect.MessagePoolReturn(packet)
@@ -330,6 +335,9 @@ func (self *ProxyDevice) Run() {
330335
}
331336

332337
func (self *ProxyDevice) Send(packet []byte) bool {
338+
if !self.UpdateActivity() {
339+
return false
340+
}
333341
return self.deviceLocal.SendPacketNoCopy(packet, int32(len(packet)))
334342
}
335343

0 commit comments

Comments
 (0)