From c37304d7059623c1d5cc4a01b2e6c2e9670dcbc5 Mon Sep 17 00:00:00 2001
From: Mikhail Osipov <mike.osipov@gmail.com>
Date: Mon, 2 Mar 2020 23:12:12 +0300
Subject: [hook/proxy] add timeout

---
 pkg/server/hook/proxy.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'pkg/server/hook')

diff --git a/pkg/server/hook/proxy.go b/pkg/server/hook/proxy.go
index 64db784..6c41742 100644
--- a/pkg/server/hook/proxy.go
+++ b/pkg/server/hook/proxy.go
@@ -9,6 +9,7 @@ import (
 	"io"
 	"regexp"
 	"strconv"
+	"time"
 	"tunnel/pkg/server/env"
 	"tunnel/pkg/server/opts"
 	"tunnel/pkg/server/queue"
@@ -87,7 +88,7 @@ func (p *proxy) Recv(rq, wq queue.Q) (err error) {
 		}
 	}()
 
-	s := bufio.NewScanner(rq.Reader())
+	s := bufio.NewScanner(rq.TimeoutReader(5 * time.Second))
 
 	var resp bool
 
-- 
cgit v1.2.3-70-g09d2