archive/fscan
1
0
Fork 0
mirror of https://github.com/shadow1ng/fscan.git synced 2026-09-26 05:01:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update redis.go

Browse Source
This commit is contained in:
影舞者
2021-07-20 11:29:43 +08:00
committed by GitHub
parent f64b185e6d
commit c322700c6d
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Plugins/redis.go
+8 -1
View File
@@ -44,6 +44,10 @@ func RedisConn(info *common.HostInfo, pass string) (flag bool, err error) {
return flag, err return flag, err
} }
defer conn.Close() defer conn.Close()
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
if err != nil {
return flag, err
}
_, err = conn.Write([]byte(fmt.Sprintf("auth %s\r\n", pass))) _, err = conn.Write([]byte(fmt.Sprintf("auth %s\r\n", pass)))
if err != nil { if err != nil {
return flag, err return flag, err
@@ -69,6 +73,10 @@ func RedisUnauth(info *common.HostInfo) (flag bool, err error) {
return flag, err return flag, err
} }
defer conn.Close() defer conn.Close()
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
if err != nil {
return flag, err
}
_, err = conn.Write([]byte("info\r\n")) _, err = conn.Write([]byte("info\r\n"))
if err != nil { if err != nil {
return flag, err return flag, err
@@ -262,7 +270,6 @@ func Readfile(filename string) (string, error) {
func readreply(conn net.Conn) (result string, err error) { func readreply(conn net.Conn) (result string, err error) {
buf := make([]byte, 4096) buf := make([]byte, 4096)
for { for {
conn.SetReadDeadline(time.Now().Add(10 * time.Second))
count, err := conn.Read(buf) count, err := conn.Read(buf)
if err != nil { if err != nil {
break break