mirror of https://github.com/qwqdanchun/fscan.git
fix bug
This commit is contained in:
parent
1499c7253a
commit
71024954e2
|
@ -33,6 +33,7 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) {
|
||||||
return flag, err
|
return flag, err
|
||||||
}
|
}
|
||||||
buf := make([]byte, 1024)
|
buf := make([]byte, 1024)
|
||||||
|
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
|
||||||
count, err := conn.Read(buf)
|
count, err := conn.Read(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return flag, err
|
return flag, err
|
||||||
|
@ -43,6 +44,7 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return flag, err
|
return flag, err
|
||||||
}
|
}
|
||||||
|
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
|
||||||
count, err := conn.Read(buf)
|
count, err := conn.Read(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return flag, err
|
return flag, err
|
||||||
|
|
|
@ -262,6 +262,7 @@ 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
|
||||||
|
|
Loading…
Reference in New Issue