Added MAC vendor lookup functionality and updated UI to display vendor information in scan results; modified IP configuration and ping targets.

This commit is contained in:
Miguel 2025-04-04 21:27:13 +02:00
parent a948581619
commit 3f8a7643ce
4 changed files with 87 additions and 10 deletions

View File

@ -1 +1 @@
{"last_interface": "Ethernet", "last_ip_prefix": "10.1.20", "previous_config": {"name": "Ethernet", "ip_address": "10.1.20.249", "subnet_mask": "255.255.255.0", "gateway": "10.1.20.1", "dhcp_enabled": false}}
{"last_interface": "Ethernet", "last_ip_prefix": "10.1.20", "previous_config": {"name": "Ethernet", "ip_address": "10.1.20.249", "subnet_mask": "255.240.0.0", "gateway": "10.1.20.1", "dhcp_enabled": false}}

View File

@ -46,6 +46,21 @@ except ImportError:
"WMI module not found. Using alternative method for network interface detection."
)
# Add MAC vendor lookup library
try:
from mac_vendor_lookup import MacLookup
MAC_LOOKUP_AVAILABLE = True
# Initialize the MAC lookup service
mac_lookup = MacLookup()
# Update the database if needed (this can be time-consuming)
# mac_lookup.update_vendors() # Uncomment to update database on startup
except ImportError:
MAC_LOOKUP_AVAILABLE = False
print(
"mac-vendor-lookup module not found. Install with 'pip install mac-vendor-lookup' for MAC vendor lookup functionality."
)
@dataclass
class NetworkInterface:
@ -194,6 +209,10 @@ class IPChangerApp:
self.master.grid_columnconfigure(0, weight=1)
self.master.grid_rowconfigure(1, weight=1) # El log expandible
# Initialize MAC lookup if available
if MAC_LOOKUP_AVAILABLE:
self.mac_lookup = MacLookup()
def setup_mask_traces(self):
"""Set up traces for subnet mask and CIDR fields safely"""
# First initialize variables
@ -538,7 +557,7 @@ class IPChangerApp:
# Replace Listbox with Treeview that has columns
self.scan_results_tree = ttk.Treeview(
self.results_frame,
columns=("ip", "hostname", "mac"),
columns=("ip", "hostname", "mac", "vendor"), # Add vendor column
show="headings",
height=10,
)
@ -547,11 +566,17 @@ class IPChangerApp:
self.scan_results_tree.heading("ip", text="IP Address")
self.scan_results_tree.heading("hostname", text="Hostname")
self.scan_results_tree.heading("mac", text="MAC Address")
self.scan_results_tree.heading(
"vendor", text="MAC Vendor"
) # Add vendor heading
# Set column widths
self.scan_results_tree.column("ip", width=120, anchor="w")
self.scan_results_tree.column("hostname", width=200, anchor="w")
self.scan_results_tree.column("mac", width=150, anchor="w")
self.scan_results_tree.column(
"vendor", width=180, anchor="w"
) # Add vendor column width
# Add scrollbar
self.scan_results_scrollbar = ttk.Scrollbar(
@ -931,6 +956,7 @@ class IPChangerApp:
return
# Save the ping target when used
self.save_current_ping_target()
# Check if continuous ping is enabled
@ -1114,11 +1140,18 @@ class IPChangerApp:
# Try to get hostname and MAC address
hostname = self.get_hostname(target)
mac_address = self.get_mac_address(target)
vendor = ""
# Look up vendor if we have a MAC
if mac_address and MAC_LOOKUP_AVAILABLE:
vendor = self.get_mac_vendor(mac_address)
if hostname:
self.log_message(f"Hostname: {hostname}")
if mac_address:
self.log_message(f"MAC Address: {mac_address}")
if vendor:
self.log_message(f"MAC Vendor: {vendor}")
except Exception as e:
self.log_message(f"Error executing ping: {str(e)}")
@ -1247,7 +1280,7 @@ class IPChangerApp:
# Add to results tree with placeholder values
self.master.after(
0, lambda ip=ip_str: self.add_scan_result(ip, "", "")
0, lambda ip=ip_str: self.add_scan_result(ip, "", "", "")
)
# Update progress
@ -1294,6 +1327,15 @@ class IPChangerApp:
# Get hostname and MAC
hostname = self.get_hostname(ip)
mac_address = self.get_mac_address(ip)
vendor = ""
# Look up vendor information if we have a MAC address
if mac_address and MAC_LOOKUP_AVAILABLE:
vendor = self.get_mac_vendor(mac_address)
if vendor:
self.log_message(f" MAC Vendor: {vendor}")
else:
self.log_message(f" MAC Vendor: Unknown")
# Log what we found
if hostname:
@ -1306,8 +1348,8 @@ class IPChangerApp:
else:
self.log_message(f" MAC Address: Not found")
# Update the tree
self.update_host_in_tree(ip, hostname, mac_address)
# Update the tree with vendor information
self.update_host_in_tree(ip, hostname, mac_address, vendor)
# Update progress
self.scan_progress_var.set(i + 1)
@ -1317,7 +1359,7 @@ class IPChangerApp:
except Exception as e:
self.log_message(f"Error gathering host information: {str(e)}")
def update_host_in_tree(self, ip, hostname, mac):
def update_host_in_tree(self, ip, hostname, mac, vendor=""):
"""Update an existing host in the treeview with obtained information"""
try:
# Find the item with this IP
@ -1327,21 +1369,24 @@ class IPChangerApp:
# Update the values
hostname_str = hostname if hostname else "Not resolved"
mac_str = mac if mac else "Not found"
vendor_str = vendor if vendor else "Unknown"
self.scan_results_tree.item(
item_id, values=(ip, hostname_str, mac_str)
item_id, values=(ip, hostname_str, mac_str, vendor_str)
)
break
except Exception as e:
self.log_message(f"Error updating host in treeview: {str(e)}")
def add_scan_result(self, ip, hostname, mac):
def add_scan_result(self, ip, hostname, mac, vendor=""):
"""Helper method to add scan result to the treeview"""
try:
# Insert the item with all values, even if some are empty
hostname = hostname if hostname else "Not resolved"
mac = mac if mac else "Not found"
vendor = vendor if vendor else "Unknown"
self.scan_results_tree.insert("", "end", values=(ip, hostname, mac))
self.scan_results_tree.insert("", "end", values=(ip, hostname, mac, vendor))
except Exception as e:
self.log_message(f"Error adding scan result to UI: {str(e)}")
@ -1784,6 +1829,30 @@ class IPChangerApp:
self.log_message(f"Error getting MAC address for {ip_address}: {str(e)}")
return ""
def get_mac_vendor(self, mac_address):
"""Look up the vendor for a MAC address"""
try:
if not MAC_LOOKUP_AVAILABLE:
return ""
# Clean up the MAC address format - the library expects a standardized format
# Remove spaces, dashes, colons and ensure uppercase
clean_mac = (
mac_address.replace(" ", "").replace("-", "").replace(":", "").upper()
)
# Add colons to form a standard MAC format for lookup
formatted_mac = ":".join(
[clean_mac[i : i + 2] for i in range(0, len(clean_mac), 2)]
)
# Look up the vendor
vendor = self.mac_lookup.lookup(formatted_mac)
return vendor
except Exception as e:
self.log_message(f"Error looking up MAC vendor: {str(e)}")
return ""
def main():
root = tk.Tk()

View File

@ -1 +1 @@
{"10.1.22": "10.1.22.11", "10.138.182": "10.138.182.94", "10.1.20": "10.1.20.11", "10.255.255": "10.255.255.1"}
{"10.1.22": "10.1.22.11", "10.138.182": "10.138.182.94", "10.1.20": "10.1.33.11", "10.255.255": "10.255.255.1", "192.168.88": "192.168.88.1"}

8
requirements.txt Normal file
View File

@ -0,0 +1,8 @@
# Network-related packages
pythonping>=1.1.4
pysnmp>=4.4.12
wmi>=1.5.1
mac-vendor-lookup>=0.1.11
# Other dependencies
ipaddress>=1.0.23