mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-22 03:10:42 +08:00
fix: UDP插件改用 conn.SetDeadline 替代类型断言
This commit is contained in:
@@ -34,9 +34,7 @@ func (p *BACnetPlugin) Scan(ctx context.Context, info *common.HostInfo, session
|
|||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
if dl, ok := conn.(interface{ SetReadDeadline(time.Time) error }); ok {
|
_ = conn.SetDeadline(time.Now().Add(timeout))
|
||||||
_ = dl.SetReadDeadline(time.Now().Add(timeout))
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := conn.Write(bacnetWhoIs); err != nil {
|
if _, err := conn.Write(bacnetWhoIs); err != nil {
|
||||||
return &ScanResult{Success: false, Service: "bacnet"}
|
return &ScanResult{Success: false, Service: "bacnet"}
|
||||||
|
|||||||
@@ -34,9 +34,7 @@ func (p *DNSPlugin) Scan(ctx context.Context, info *common.HostInfo, session *co
|
|||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
if dl, ok := conn.(interface{ SetReadDeadline(time.Time) error }); ok {
|
_ = conn.SetDeadline(time.Now().Add(timeout))
|
||||||
_ = dl.SetReadDeadline(time.Now().Add(timeout))
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := conn.Write(query); err != nil {
|
if _, err := conn.Write(query); err != nil {
|
||||||
return &ScanResult{Success: false, Service: "dns"}
|
return &ScanResult{Success: false, Service: "dns"}
|
||||||
|
|||||||
@@ -53,9 +53,7 @@ func (p *IPMIPlugin) rmcpPing(ctx context.Context, target string, timeout time.D
|
|||||||
0x00, // data length = 0
|
0x00, // data length = 0
|
||||||
}
|
}
|
||||||
|
|
||||||
if dl, ok := conn.(interface{ SetReadDeadline(time.Time) error }); ok {
|
_ = conn.SetDeadline(time.Now().Add(timeout))
|
||||||
_ = dl.SetReadDeadline(time.Now().Add(timeout))
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := conn.Write(ping); err != nil {
|
if _, err := conn.Write(ping); err != nil {
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
@@ -75,9 +75,7 @@ func (p *SNMPPlugin) probe(ctx context.Context, target, community string, timeou
|
|||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
if dl, ok := conn.(interface{ SetReadDeadline(time.Time) error }); ok {
|
_ = conn.SetDeadline(time.Now().Add(timeout))
|
||||||
_ = dl.SetReadDeadline(time.Now().Add(timeout))
|
|
||||||
}
|
|
||||||
|
|
||||||
pkt := buildSNMPGetRequest(community, []int{1, 3, 6, 1, 2, 1, 1, 1, 0})
|
pkt := buildSNMPGetRequest(community, []int{1, 3, 6, 1, 2, 1, 1, 1, 0})
|
||||||
if _, err := conn.Write(pkt); err != nil {
|
if _, err := conn.Write(pkt); err != nil {
|
||||||
|
|||||||
@@ -33,9 +33,7 @@ func (p *TFTPPlugin) Scan(ctx context.Context, info *common.HostInfo, session *c
|
|||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
if dl, ok := conn.(interface{ SetReadDeadline(time.Time) error }); ok {
|
_ = conn.SetDeadline(time.Now().Add(timeout))
|
||||||
_ = dl.SetReadDeadline(time.Now().Add(timeout))
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := conn.Write(buildTFTPReadRequest("probe")); err != nil {
|
if _, err := conn.Write(buildTFTPReadRequest("probe")); err != nil {
|
||||||
return &ScanResult{Success: false, Service: "tftp"}
|
return &ScanResult{Success: false, Service: "tftp"}
|
||||||
|
|||||||
Reference in New Issue
Block a user