mirror of
https://github.com/nikdoof/globaltalk-scraper.git
synced 2025-12-17 12:59:23 +00:00
119 lines
3.2 KiB
Python
Executable File
119 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
GlobalTalk Scraper
|
|
|
|
A script that uses netatalk's `getzones` and `nbplkup` to discover devices on
|
|
the GlobalTalk network, and create some nice visualisations of the network.
|
|
|
|
Created for MARCHintosh 2025
|
|
"""
|
|
|
|
import logging
|
|
import subprocess
|
|
import re
|
|
import json
|
|
import sys
|
|
import os
|
|
import argparse
|
|
import shutil
|
|
|
|
NPBLKUP_RESULTS = re.compile(r"^(.*):(.*)\s(\d*\.\d*:\d*)$")
|
|
|
|
|
|
def getzones() -> list[str]:
|
|
"""Returns a list of AppleTalk zones using `getzones`"""
|
|
result = subprocess.run(["getzones"], capture_output=True, text=True)
|
|
return [x.strip() for x in result.stdout.split("\n") if x.strip() != ""]
|
|
|
|
|
|
def nbplkup(zone: str) -> list[(str, str)]:
|
|
"""Lookup members of a zone, and return a list of name and address tuples"""
|
|
if zone not in getzones():
|
|
logging.warning("%s not currently a known zone", zone)
|
|
return []
|
|
|
|
zone_results = []
|
|
|
|
# Define the charset for the console as 'mac-roman' to avoid any
|
|
# translation issues with some device names (AsanteTalk and such)
|
|
environ = os.environ.copy()
|
|
environ["ATALK_UNIX_CHARSET"] = "mac-roman"
|
|
|
|
cmd = subprocess.run(
|
|
["nbplkup", "@{0}".format(zone)],
|
|
capture_output=True,
|
|
text=True,
|
|
encoding="mac-roman",
|
|
env=environ,
|
|
)
|
|
|
|
# Use regex to split the row nicely
|
|
for node in [
|
|
NPBLKUP_RESULTS.match(x.strip()).groups()
|
|
for x in cmd.stdout.split("\n")
|
|
if x.strip() != ""
|
|
]:
|
|
object, endpoint_type, address = node
|
|
address, socket = address.split(":")
|
|
zone_results.append(
|
|
{
|
|
"object": object.strip(),
|
|
"type": endpoint_type.strip(),
|
|
"address": address,
|
|
"socket": socket,
|
|
"zone": zone,
|
|
}
|
|
)
|
|
return zone_results
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser("gtscraper")
|
|
parser.add_argument(
|
|
"--zone", required=False, default=None, help="Scan an individual zone"
|
|
)
|
|
parser.add_argument(
|
|
"--output",
|
|
type=argparse.FileType("w"),
|
|
default=sys.stdout,
|
|
help="Filename to write the resulting JSON to",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
logging.basicConfig(level=logging.INFO, stream=sys.stderr)
|
|
|
|
# Check we have the pre-reqs to actually run
|
|
for cmd in ["getzones", "nbplkup"]:
|
|
if not shutil.which(cmd):
|
|
logging.error(
|
|
"%s is missing from your PATH, is netatalk installed and working?", cmd
|
|
)
|
|
return
|
|
|
|
# Init the results dict
|
|
zone_results = {
|
|
"format": "v1",
|
|
"zones": getzones(),
|
|
"nodes": [],
|
|
}
|
|
|
|
# Iterate the zones and scan them
|
|
for zone in zone_results["zones"]:
|
|
if args.zone and zone != args.zone:
|
|
continue
|
|
logging.info("Scanning %s", zone)
|
|
node_data = nbplkup(zone)
|
|
zone_results["nodes"].extend(node_data)
|
|
|
|
nodes = len(zone_results["nodes"])
|
|
zones = len(zone_results["zones"])
|
|
|
|
print("{0} zones, {1} nodes".format(zones, nodes), file=sys.stderr)
|
|
|
|
# Dump out the resulting JSON to stdout
|
|
sys.stdout.write(json.dumps(zone_results))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|