github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 10:46:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

misc: small fix or general refactoring i did not bother commenting

Browse source
This commit is contained in:
evilsocket 2018-09-09 13:21:42 +03:00
commit 549fdc45af
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
2 changed files with 17 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

20
modules/net_probe_upnp.go
View file

@ -1,6 +1,7 @@
package modules
import (
"fmt"
"net"
"github.com/bettercap/bettercap/log"
@ -8,13 +9,18 @@ import (
)
func (p *Prober) sendProbeUPNP(from net.IP, from_hw net.HardwareAddr) {
err, raw := packets.NewUPNPProbe(from, from_hw)
if err != nil {
log.Error("error while sending upnp probe: %v", err)
return
} else if err := p.Session.Queue.Send(raw); err != nil {
log.Error("error sending upnp packet: %s", err)
name := fmt.Sprintf("%s:%d", packets.UPNPDestIP, packets.UPNPPort)
if addr, err := net.ResolveUDPAddr("udp", name); err != nil {
log.Debug("could not resolve %s.", name)
} else if con, err := net.DialUDP("udp", nil, addr); err != nil {
log.Debug("could not dial %s.", name)
} else {
log.Debug("sent %d bytes of UPNP probe", len(raw))
defer con.Close()
if wrote, _ := con.Write(packets.UPNPDiscoveryPayload); wrote > 0 {
p.Session.Queue.TrackSent(uint64(wrote))
} else {
p.Session.Queue.TrackError()
}
}
}