mirror of
https://github.com/kawasaki/pyscrlink.git
synced 2025-09-07 18:20:07 +02:00
e552bd21bd
It was informed that the required tools setcap and getcap are not placed in /usr/bin in some user environment. They can not be executed since they are out of PATH. To allow the script to execute the commands out of PATH, have the script find the tools. Keep the found paths of each command in the dictionary 'tools', and refer it to execute the commands. Signed-off-by: Shin'ichiro Kawasaki <kawasaki@juno.dti.ne.jp>
82 lines
2.2 KiB
Python
Executable File
82 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
import os
|
|
import shutil
|
|
import bluepy
|
|
import subprocess
|
|
|
|
import logging
|
|
logLevel = logging.INFO
|
|
|
|
# for logging
|
|
logger = logging.getLogger(__name__)
|
|
formatter = logging.Formatter(fmt='%(asctime)s %(message)s')
|
|
handler = logging.StreamHandler()
|
|
handler.setLevel(logLevel)
|
|
handler.setFormatter(formatter)
|
|
logger.setLevel(logLevel)
|
|
logger.addHandler(handler)
|
|
logger.propagate = False
|
|
|
|
# Check dependent tools
|
|
DEPENDENT_TOOLS = {
|
|
"setcap": "libcap2-bin (Ubuntu) or libcap (Arch)",
|
|
"getcap": "libcap2-bin (Ubuntu) or libcap (Arch)",
|
|
}
|
|
|
|
tools = {}
|
|
|
|
for cmd in DEPENDENT_TOOLS:
|
|
# find the tools in PATH
|
|
path = shutil.which(cmd)
|
|
if path:
|
|
tools[cmd] = path
|
|
logger.debug(f"{cmd} found: {path}")
|
|
continue
|
|
# find the tools out of PATH but in major directories
|
|
for d in ["/usr/bin", "/bin", "/usr/sbin", "/sbin"]:
|
|
path = d + '/' + cmd
|
|
if os.path.isfile(path) and os.access(path, os.X_OK):
|
|
tools[cmd] = path
|
|
logger.debug(f"{cmd} found: {path}")
|
|
break
|
|
if not cmd in tools:
|
|
print(f"'{cmd}' not found. Install package {DEPENDENT_TOOLS[cmd]}.")
|
|
sys.exit(1)
|
|
|
|
def helper_path():
|
|
path = os.path.abspath(bluepy.__file__)
|
|
if not path:
|
|
logger.error("Bluepy module not found")
|
|
sys.exit(1)
|
|
if path.find("__init__.py") < 0:
|
|
logger.error(f"Unexpected bluepy module path: {path}")
|
|
sys.exit(1)
|
|
path = path.replace("__init__.py", "bluepy-helper")
|
|
return path
|
|
|
|
def is_set():
|
|
path = helper_path()
|
|
p = subprocess.run([tools["getcap"], path], stdout=subprocess.PIPE)
|
|
if p.returncode != 0:
|
|
logger.error(f"Failed to get capability of {path}")
|
|
return False
|
|
out = str(p.stdout)
|
|
return out.find("cap_net_admin") >= 0 and out.find("cap_net_raw") >= 0
|
|
|
|
def setcap():
|
|
path = helper_path()
|
|
if is_set():
|
|
return True
|
|
p = subprocess.run(["sudo", tools["setcap"],
|
|
"cap_net_raw,cap_net_admin+eip", path])
|
|
if p.returncode !=0:
|
|
logger.error(f"Failed to set capability to {path}")
|
|
return False
|
|
print(f"Set capacbility 'cap_net_raw,cap_net_admin' to {path}")
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
setcap()
|