api. Validate OS in Subscription

NethServer · Jan 23, 2024 · 462c077 · 462c077
1 parent 457d729
commit 462c077
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 2 deletions.
diff --git a/core/imageroot/var/lib/nethserver/cluster/actions/get-subscription/10get_subscription b/core/imageroot/var/lib/nethserver/cluster/actions/get-subscription/10get_subscription
@@ -66,6 +66,25 @@ def fetch_subscription_info_nscom(rdb, attributes):
     }
     return info
 
+def _exit_validation_error(field, value, error, exit_code=2):
+    agent.set_status('validation-failed')
+    json.dump([{
+            'field': field,
+            'parameter': field,
+            'value': value,
+            'error': error
+        }],
+        fp=sys.stdout)
+    sys.exit(exit_code)
+
+try:
+    os_release = agent.read_envfile("/etc/os-release")
+    if os_release["ID"] != '"rocky"' or \
+        os_release["PLATFORM_ID"] != '"platform:el9"':
+        raise Exception("Invalid OS")
+except Exception as ex:
+    json.dump({"subscription": None, "error": "os_not_supported"}, fp=sys.stdout)
+    sys.exit(0)
 
 rdb = agent.redis_connect(privileged=True)
 hsubscription = rdb.hgetall("cluster/subscription") or {}

diff --git a/core/imageroot/var/lib/nethserver/cluster/actions/get-subscription/validate-output.json b/core/imageroot/var/lib/nethserver/cluster/actions/get-subscription/validate-output.json
@@ -3,5 +3,19 @@
     "title": "get-subscription",
     "description": "Get support subscription attributes",
     "$id": "http://schema.nethserver.org/cluster/get-subscription.json",
-    "examples": []
-}
+    "examples": [],
+    "type": [
+        "object"
+    ],
+    "required": [
+        "subscription"
+    ],
+    "properties": {
+        "subscription": {
+            "type": [
+                "object",
+                "null"
+            ]
+        }
+    }
+}
diff --git a/core/imageroot/var/lib/nethserver/cluster/actions/set-subscription/10set_subscription b/core/imageroot/var/lib/nethserver/cluster/actions/set-subscription/10set_subscription
@@ -140,6 +140,15 @@ else:
     if rdb.exists("cluster/subscription"):
         _exit_validation_error("", "", "system_already_subscribed")
 
+    try:
+        os_release = agent.read_envfile("/etc/os-release")
+        if os_release["ID"] != '"rocky"' or \
+            os_release["PLATFORM_ID"] != '"platform:el9"':
+            raise Exception("Invalid OS")
+    except Exception as ex:
+        print(agent.SD_WARNING, ex, file=sys.stderr)
+        _exit_validation_error("", "", "os_not_supported")
+
     prev_exceptions = []
     for subscribe_function in [subscribe_nsent, subscribe_nscom]:
         try: