mirror of
https://github.com/xjasonlyu/tun2socks.git
synced 2025-10-31 12:06:37 +08:00
36 lines
730 B
Go
36 lines
730 B
Go
package filter
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/xjasonlyu/tun2socks/common/log"
|
|
"github.com/xjasonlyu/tun2socks/common/packet"
|
|
)
|
|
|
|
type icmpFilter struct {
|
|
writer io.Writer
|
|
delay int
|
|
}
|
|
|
|
func NewICMPFilter(w io.Writer, delay int) Filter {
|
|
return &icmpFilter{writer: w, delay: delay}
|
|
}
|
|
|
|
func (w *icmpFilter) Write(buf []byte) (int, error) {
|
|
if uint8(buf[9]) == packet.PROTOCOL_ICMP {
|
|
payload := make([]byte, len(buf))
|
|
copy(payload, buf)
|
|
go func(data []byte) {
|
|
time.Sleep(time.Duration(w.delay) * time.Millisecond)
|
|
_, err := w.writer.Write(data)
|
|
if err != nil {
|
|
log.Fatalf("failed to input data to the stack: %v", err)
|
|
}
|
|
}(payload)
|
|
return len(buf), nil
|
|
} else {
|
|
return w.writer.Write(buf)
|
|
}
|
|
}
|