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

fix: fixed a logic bug which made targets lookup by alias fail.

Browse source
This commit is contained in:
evilsocket 2019-02-13 10:53:24 +01:00
commit ca421cf8e3
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
1 changed files with 19 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

34
network/net.go
View file

@ -80,7 +80,7 @@ func ParseMACs(targets string) (macs []net.HardwareAddr, err error) {
mac = NormalizeMac(mac)
hw, err := net.ParseMAC(mac)
if err != nil {
return nil, fmt.Errorf("Error while parsing MAC '%s': %s", mac, err)
return nil, fmt.Errorf("error while parsing MAC '%s': %s", mac, err)
}
macs = append(macs, hw)
@ -103,7 +103,7 @@ func ParseTargets(targets string, aliasMap *data.UnsortedKV) (ips []net.IP, macs
mac = NormalizeMac(mac)
hw, err := net.ParseMAC(mac)
if err != nil {
return nil, nil, fmt.Errorf("Error while parsing MAC '%s': %s", mac, err)
return nil, nil, fmt.Errorf("error while parsing MAC '%s': %s", mac, err)
}
macs = append(macs, hw)
@ -112,18 +112,22 @@ func ParseTargets(targets string, aliasMap *data.UnsortedKV) (ips []net.IP, macs
targets = strings.Trim(targets, ", ")
// check and resolve aliases
for _, alias := range aliasParser.FindAllString(targets, -1) {
if mac, found := aliasMap.Get(alias); found {
mac = NormalizeMac(mac)
hw, err := net.ParseMAC(mac)
if err != nil {
return nil, nil, fmt.Errorf("Error while parsing MAC '%s': %s", mac, err)
for _, targetAlias := range aliasParser.FindAllString(targets, -1) {
found := false
aliasMap.Each(func(mac, alias string) bool {
if alias == targetAlias {
if hw, err := net.ParseMAC(mac); err == nil {
macs = append(macs, hw)
targets = strings.Replace(targets, targetAlias, "", -1)
found = true
return true
}
}
return false
})
macs = append(macs, hw)
targets = strings.Replace(targets, alias, "", -1)
} else {
return nil, nil, fmt.Errorf("Could not resolve alias %s", alias)
if !found {
return nil, nil, fmt.Errorf("could not resolve alias %s", targetAlias)
}
}
targets = strings.Trim(targets, ", ")
@ -132,7 +136,7 @@ func ParseTargets(targets string, aliasMap *data.UnsortedKV) (ips []net.IP, macs
if targets != "" {
list, err := iprange.ParseList(targets)
if err != nil {
return nil, nil, fmt.Errorf("Error while parsing address list '%s': %s.", targets, err)
return nil, nil, fmt.Errorf("error while parsing address list '%s': %s.", targets, err)
}
ips = list.Expand()
@ -225,7 +229,7 @@ func findInterfaceByName(name string, ifaces []net.Interface) (*Endpoint, error)
}
}
return nil, fmt.Errorf("No interface matching '%s' found.", name)
return nil, fmt.Errorf("no interface matching '%s' found.", name)
}
func FindInterface(name string) (*Endpoint, error) {
@ -244,7 +248,7 @@ func FindInterface(name string) (*Endpoint, error) {
for _, iface := range ifaces {
addrs, err := iface.Addrs()
if err != nil {
fmt.Printf("WTF of the day: %s", err)
fmt.Printf("wtf of the day: %s", err)
continue
}