mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-12-20 10:28:45 +00:00
35aae69f23
* Support armv7 and first abstraction * Change layout * Add more type hints * Fix imports * Update * move forward * add tests * fix type * fix lint & tests * fix tests * Fix unittests * Fix create folder * cleanup * Fix import order * cleanup loop parameter * cleanup init function * Allow changeable image name * fix setup * Fix load of arch * Fix lint * Add typing * fix init * fix hassos cli problem & stick on supervisor arch * address comments * cleanup * Fix image selfheal * Add comment * update uvloop * remove uvloop * fix tagging * Fix install name * Fix validate build config * Abstract image_name from system cache
120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
"""Util add-ons functions."""
|
|
from __future__ import annotations
|
|
import asyncio
|
|
import hashlib
|
|
import logging
|
|
from pathlib import Path
|
|
import re
|
|
from typing import TYPE_CHECKING
|
|
|
|
from ..const import (PRIVILEGED_DAC_READ_SEARCH, PRIVILEGED_NET_ADMIN,
|
|
PRIVILEGED_SYS_ADMIN, PRIVILEGED_SYS_MODULE,
|
|
PRIVILEGED_SYS_PTRACE, PRIVILEGED_SYS_RAWIO, ROLE_ADMIN,
|
|
ROLE_MANAGER, SECURITY_DISABLE, SECURITY_PROFILE)
|
|
|
|
if TYPE_CHECKING:
|
|
from .addon import Addon
|
|
|
|
RE_SHA1 = re.compile(r"[a-f0-9]{8}")
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def rating_security(addon: Addon) -> int:
|
|
"""Return 1-6 for security rating.
|
|
|
|
1 = not secure
|
|
6 = high secure
|
|
"""
|
|
rating = 5
|
|
|
|
# AppArmor
|
|
if addon.apparmor == SECURITY_DISABLE:
|
|
rating += -1
|
|
elif addon.apparmor == SECURITY_PROFILE:
|
|
rating += 1
|
|
|
|
# Home Assistant Login
|
|
if addon.access_auth_api:
|
|
rating += 1
|
|
|
|
# Privileged options
|
|
# pylint: disable=bad-continuation
|
|
if any(
|
|
privilege in addon.privileged for privilege in (
|
|
PRIVILEGED_NET_ADMIN,
|
|
PRIVILEGED_SYS_ADMIN,
|
|
PRIVILEGED_SYS_RAWIO,
|
|
PRIVILEGED_SYS_PTRACE,
|
|
PRIVILEGED_SYS_MODULE,
|
|
PRIVILEGED_DAC_READ_SEARCH,
|
|
)):
|
|
rating += -1
|
|
|
|
# API Hass.io role
|
|
if addon.hassio_role == ROLE_MANAGER:
|
|
rating += -1
|
|
elif addon.hassio_role == ROLE_ADMIN:
|
|
rating += -2
|
|
|
|
# Not secure Networking
|
|
if addon.host_network:
|
|
rating += -1
|
|
|
|
# Insecure PID namespace
|
|
if addon.host_pid:
|
|
rating += -2
|
|
|
|
# Full Access
|
|
if addon.with_full_access:
|
|
rating += -2
|
|
|
|
# Docker Access
|
|
if addon.access_docker_api:
|
|
rating = 1
|
|
|
|
return max(min(6, rating), 1)
|
|
|
|
|
|
def get_hash_from_repository(name: str) -> str:
|
|
"""Generate a hash from repository."""
|
|
key = name.lower().encode()
|
|
return hashlib.sha1(key).hexdigest()[:8]
|
|
|
|
|
|
def extract_hash_from_path(path: Path) -> str:
|
|
"""Extract repo id from path."""
|
|
repository_dir = path.parts[-1]
|
|
|
|
if not RE_SHA1.match(repository_dir):
|
|
return get_hash_from_repository(repository_dir)
|
|
return repository_dir
|
|
|
|
|
|
def check_installed(method):
|
|
"""Wrap function with check if add-on is installed."""
|
|
|
|
async def wrap_check(addon, *args, **kwargs):
|
|
"""Return False if not installed or the function."""
|
|
if not addon.is_installed:
|
|
_LOGGER.error("Addon %s is not installed", addon.slug)
|
|
return False
|
|
return await method(addon, *args, **kwargs)
|
|
|
|
return wrap_check
|
|
|
|
|
|
async def remove_data(folder: Path) -> None:
|
|
"""Remove folder and reset privileged."""
|
|
try:
|
|
proc = await asyncio.create_subprocess_exec(
|
|
"rm", "-rf", str(folder), stdout=asyncio.subprocess.DEVNULL)
|
|
|
|
_, error_msg = await proc.communicate()
|
|
except OSError as err:
|
|
error_msg = str(err)
|
|
else:
|
|
if proc.returncode == 0:
|
|
return
|
|
|
|
_LOGGER.error("Can't remove Add-on Data: %s", error_msg)
|