diff --git a/Plugins/mongodb.go b/Plugins/mongodb.go index 8ca84e2..a691457 100644 --- a/Plugins/mongodb.go +++ b/Plugins/mongodb.go @@ -33,6 +33,7 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) { return flag, err } buf := make([]byte, 1024) + err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second)) count, err := conn.Read(buf) if err != nil { return flag, err @@ -43,6 +44,7 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) { if err != nil { return flag, err } + err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second)) count, err := conn.Read(buf) if err != nil { return flag, err diff --git a/Plugins/redis.go b/Plugins/redis.go index 6bbeecd..47f2bf2 100644 --- a/Plugins/redis.go +++ b/Plugins/redis.go @@ -262,6 +262,7 @@ func Readfile(filename string) (string, error) { func readreply(conn net.Conn) (result string, err error) { buf := make([]byte, 4096) for { + conn.SetReadDeadline(time.Now().Add(10 * time.Second)) count, err := conn.Read(buf) if err != nil { break