fix(sniffer): server push cases for games

component/sniffing/sniffer.go

@@ -6,8 +6,10 @@
 package sniffing
 
 import (
+	"errors"
 	"github.com/mzz2017/softwind/pool"
 	"io"
+	"net"
 	"sync"
 )
 
@@ -44,6 +46,10 @@ func (s *Sniffer) SniffTcp() (d string, err error) {
 	if s.stream {
 		n, err := s.r.Read(s.buf)
 		if err != nil {
+			var netError net.Error
+			if errors.As(err, &netError) && netError.Timeout() {
+				return "", NotApplicableError
+			}
 			return "", err
 		}
 		s.buf = s.buf[:n]

control/tcp.go

@@ -29,11 +29,13 @@ func (c *ControlPlane) handleConn(lConn net.Conn) (err error) {
 	defer lConn.Close()
 
 	// Sniff target domain.
+	lConn.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
 	sniffer := sniffing.NewConnSniffer(lConn, TcpSniffBufSize)
 	domain, err := sniffer.SniffTcp()
 	if err != nil && !sniffing.IsSniffingError(err) {
 		return err
 	}
+	lConn.SetReadDeadline(time.Time{})
 	// ConnSniffer should be used later, so we cannot close it now.
 	defer sniffer.Close()