This commit is contained in:
jackdaw@loafle.com 2017-04-11 12:20:14 +09:00
parent 74b626cd9d
commit cf81c9a442

View File

@ -56,7 +56,7 @@ func (s *SocketHeahthCrawler) CheckHeahth(params map[string]interface{}) (bool,
return false, err
}
defer conn.Close()
if s.m.IsPrePacket() == true {
bytes := make([]byte, 1024)
n, _ := conn.Read(bytes)
@ -64,42 +64,42 @@ func (s *SocketHeahthCrawler) CheckHeahth(params map[string]interface{}) (bool,
if s.m.Match(0, p, nil) == false {
return false, nil
} else {
for i := 1 ; i < s.m.PacketCount(); i++ {
for i := 1; i < s.m.PacketCount(); i++ {
pack := s.m.Packet(i)
conn.Write(pack.Buffer)
bytes := make([]byte, 1024)
n, _ := conn.Read(bytes)
if s.m.IsNoResponse(i) == true { // empty last response
break
}
p := packet.NewPacket(bytes, n)
if s.m.Match(i, p, nil) == false {
return false, nil
}
}
}
} else {
for i := 0 ; i < s.m.PacketCount(); i++ {
for i := 0; i < s.m.PacketCount(); i++ {
pack := s.m.Packet(i)
conn.Write(pack.Buffer)
bytes := make([]byte, 1024)
n, _ := conn.Read(bytes)
if s.m.IsNoResponse(i) == true { // empty last response
break
}
p := packet.NewPacket(bytes, n)
if s.m.Match(i, p, nil) == false {
return false, nil
}
}
}
return true, nil
}