black-internet-days-ahead/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2024-11-29 20:11:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

block: Darin packets in 5 seconds before close

Browse source
This commit is contained in:
世界 2024-07-23 14:54:54 +08:00
parent aff6768ff6
commit a7e0be485a
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 21 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
outbound/block.go
View file

@ -4,10 +4,13 @@ import (
"context"
"io"
"net"
"time"
"github.com/sagernet/sing-box/adapter"
C "github.com/sagernet/sing-box/constant"
"github.com/sagernet/sing-box/log"
"github.com/sagernet/sing/common/buf"
"github.com/sagernet/sing/common/bufio"
M "github.com/sagernet/sing/common/metadata"
N "github.com/sagernet/sing/common/network"
)
@ -46,7 +49,24 @@ func (h *Block) NewConnection(ctx context.Context, conn net.Conn, metadata adapt
}
func (h *Block) NewPacketConnection(ctx context.Context, conn N.PacketConn, metadata adapter.InboundContext) error {
conn.Close()
h.logger.InfoContext(ctx, "blocked packet connection to ", metadata.Destination)
writer := &discardPacketWriter{
timer: time.AfterFunc(C.UDPTimeout, func() {
_ = conn.Close()
}),
}
_, _ = bufio.CopyPacket(writer, conn)
return nil
}
type discardPacketWriter struct {
timer *time.Timer
}
func (w *discardPacketWriter) WritePacket(buffer *buf.Buffer, destination M.Socksaddr) error {
if w.timer.Stop() {
w.timer.Reset(C.UDPTimeout)
}
buffer.Release()
return nil
}