fix: some safeConnClose forget using original

This commit is contained in:
wwqgtxx
2022-12-22 12:31:45 +08:00
parent 980454beb2
commit daf0b23805
3 changed files with 16 additions and 16 deletions

View File

@ -153,9 +153,9 @@ func (t *Trojan) ListenPacketContext(ctx context.Context, metadata *C.Metadata,
if err != nil {
return nil, fmt.Errorf("%s connect error: %w", t.addr, err)
}
defer func() {
defer func(c net.Conn) {
safeConnClose(c, err)
}()
}(c)
err = t.instance.WriteHeader(c, trojan.CommandUDP, serializesSocksAddr(metadata))
if err != nil {
return nil, err
@ -173,9 +173,9 @@ func (t *Trojan) ListenPacketWithDialer(ctx context.Context, dialer C.Dialer, me
if err != nil {
return nil, fmt.Errorf("%s connect error: %w", t.addr, err)
}
defer func() {
defer func(c net.Conn) {
safeConnClose(c, err)
}()
}(c)
tcpKeepAlive(c)
c, err = t.plainStream(c)
if err != nil {