Mercurial > dropbear
view test/parent_dropbear_map.py @ 1870:0dcc5b0d93fa
Make re-exec work with "dropbearmulti dropbear"
The re-exec needs to know to use the dropbearmulti binary instead.
Add a test for this case.
author | Matt Johnston <matt@ucc.asn.au> |
---|---|
date | Tue, 01 Feb 2022 22:18:23 +0800 |
parents | 2b3a8026a6ce |
children | 1c9215154d4a |
line wrap: on
line source
#!/usr/bin/env python3 import os import sys import time import psutil from pathlib import Path want_name = "dropbear" # Walks up the parent process tree, prints the first line of /proc/pid/maps when # it finds the wanted name def main(): try: for p in psutil.Process().parents(): print(p.pid, file=sys.stderr) print(p.name(), file=sys.stderr) print(p.cmdline(), file=sys.stderr) if want_name in p.name(): with (Path('/proc') / str(p.pid) / "maps").open() as f: map0 = f.readline().rstrip() print(map0) return raise RuntimeError(f"Couldn't find parent {want_name} process") except Exception as e: print(psutil.Process().parents()) for p in psutil.Process().parents(): print(p.name()) print(e) # time.sleep(100) raise if __name__ == "__main__": main()