65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import json
|
|
import subprocess
|
|
import os
|
|
|
|
def set_annotation(config, arch, flavour, value, annotations_file):
|
|
"""Executes the Debian script to enforce a specific kernel configuration."""
|
|
|
|
cmd = [
|
|
"python3",
|
|
"debian/scripts/misc/annotations",
|
|
"--file", annotations_file,
|
|
"--arch", arch,
|
|
"--flavour", flavour,
|
|
"--config", config,
|
|
"--write", value
|
|
]
|
|
|
|
print(f"Setting {config} for {arch} ({flavour}) to '{value}'")
|
|
|
|
try:
|
|
subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL)
|
|
except subprocess.CalledProcessError:
|
|
print(f" [ERROR] Failed to apply {config} on {arch} ({flavour})", file=sys.stderr)
|
|
|
|
def main():
|
|
# Fetch the environment variable exported by helper
|
|
target_file = os.environ.get("KERNEL_CONFIG_ANNOTATIONS")
|
|
|
|
# Fail fast if the annotations file is not defined
|
|
if not target_file:
|
|
print("[FATAL ERROR] KERNEL_CONFIG_ANNOTATIONS environment variable is not set.", file=sys.stderr)
|
|
print("Aborting to prevent modifying the wrong configuration file.", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
print(f"[INFO] Target annotations file: {target_file}")
|
|
|
|
# Read and parse the JSON policy from stdin
|
|
try:
|
|
policy_dict = json.load(sys.stdin)
|
|
except json.JSONDecodeError as e:
|
|
print(f"[FATAL ERROR] Invalid JSON policy format: {e}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# Process the nested dictionary
|
|
for config, archs in policy_dict.items():
|
|
for key, value in archs.items():
|
|
|
|
# Check if the key contains a flavour delimiter (e.g., "arm64/generic-64k")
|
|
if "/" in key:
|
|
arch, flavour = key.split("/", 1)
|
|
set_annotation(config, arch, flavour, value, target_file)
|
|
|
|
# Backward compatibility for nested dicts (just in case)
|
|
elif isinstance(value, dict):
|
|
for flavour_nested, val_nested in value.items():
|
|
set_annotation(config, key, flavour_nested, val_nested, target_file)
|
|
|
|
# If it's just a plain string without a slash, default to 'generic'
|
|
else:
|
|
set_annotation(config, key, "generic", value, target_file)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|