qwqdanchun
/
fscan
mirror of https://github.com/qwqdanchun/fscan.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #79 from lanyi1998/main 

修复两个线程阻塞的问题
This commit is contained in:
影舞者 2021-07-20 11:31:46 +08:00 committed by GitHub
parent 1499c7253a 6a4bbe3781
commit dc949e25b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Plugins/mongodb.go
View File

@ -28,6 +28,10 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) {
return flag, err
}
defer conn.Close()
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
if err != nil {
return flag, err
}
_, err = conn.Write(senddata)
if err != nil {
return flag, err

8
Plugins/redis.go
View File

@ -44,6 +44,10 @@ func RedisConn(info *common.HostInfo, pass string) (flag bool, err error) {
return flag, err
}
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)))
if err != nil {
return flag, err
@ -69,6 +73,10 @@ func RedisUnauth(info *common.HostInfo) (flag bool, err error) {
return flag, err
}
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"))
if err != nil {
return flag, err