RDKB-60656 : Available memory check for firmware downloads

configure.ac

@@ -160,6 +160,7 @@ AC_CONFIG_FILES(
 	source/Psmcli/Makefile
 	source/LTime/Makefile
 	source/TimeConv/Makefile
+  source/FwDownloadChk/Makefile
 	source/parcon/Makefile
 	source/ServiceCtrl/Makefile
 	source/webcfg_decoder/Makefile

source/FwDownloadChk/Makefile.am

@@ -0,0 +1,10 @@
+# SPDX-License-Identifier: Apache-2.0
+
+AM_CFLAGS =
+
+lib_LTLIBRARIES = libfw_download_chk.la
+
+libfw_download_chk_la_SOURCES = fw_download_check.c
+
+# syscfg for syscfg_get, ccsp_common for logging headers
+libfw_download_chk_la_LDFLAGS = -lsyscfg -lccsp_common

source/FwDownloadChk/fw_download_check.c

@@ -0,0 +1,171 @@
+/*
+ * If not stated otherwise in this file or this component's LICENSE file the
+ * following copyright and licenses apply:
+ *
+ * Copyright 2023 RDK Management
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+*/
+#define _GNU_SOURCE 1
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdint.h>
+#include <inttypes.h>
+#include <ctype.h>
+#include <time.h>
+#include <stdarg.h>
+
+#include <syscfg/syscfg.h>
+#include <telemetry_busmessage_sender.h>
+#include "fw_download_check.h"
+
+#define XCONF_LOG_PATH "/rdklogs/logs/xconf.txt.0"
+#define XCONF_LOG_INFO(...)  xconf_log_write("INFO", __VA_ARGS__)
+#define XCONF_LOG_ERROR(...) xconf_log_write("ERROR", __VA_ARGS__)
+
+static void xconf_log_write(const char *level, const char *fmt, ...)
+{
+    FILE *fp = fopen(XCONF_LOG_PATH, "a");
+    if (!fp) {
+        fp = stderr;
+    }
+
+    time_t now = time(NULL);
+    struct tm tm_info;
+    localtime_r(&now, &tm_info);
+    char ts[32];
+    strftime(ts, sizeof(ts), "%Y-%m-%d %H:%M:%S", &tm_info);
+
+    fprintf(fp, "%s [%s] ", ts, level);
+
+    va_list ap;
+    va_start(ap, fmt);
+    vfprintf(fp, fmt, ap);
+    va_end(ap);
+
+    if (fp != stderr) {
+        fflush(fp);
+        fclose(fp);
+    } else {
+        fflush(stderr);
+    }
+}
+
+int can_proceed_fw_download(void)
+{
+    char url[1024] = {0};
+    char fname[256] = {0};
+    char buf[64] = {0};
+    char line[256] = {0};
+
+    uint64_t fw_kb = 0;
+    uint64_t avail_kb = 0;
+    uint64_t rsrv_mb = 0, rsrv_kb = 0;
+    int imgp_pct = 0;
+
+    if(syscfg_get(NULL,"xconf_url",url, sizeof(url)) != 0){
+        XCONF_LOG_ERROR("[FWCHK] Failed to get xconf_url\n");
+        return -1;
+    }
+    XCONF_LOG_INFO("[FWCHK] xconf_url: %s\n", url);
+
+    if(syscfg_get(NULL,"fw_to_upgrade",fname,sizeof(fname)) != 0){
+        XCONF_LOG_ERROR("[FWCHK] Failed to get fw_to_upgrade\n");
+        return -1;
+    }
+    XCONF_LOG_INFO("[FWCHK] fw_to_upgrade: %s\n", fname);
+
+    {
+        char cmd[1500];
+        snprintf(cmd, sizeof(cmd),
+                 "curl -sI '%s%s%s' | grep -i Content-Length | awk '{print $2}'",
+                 url, (url[strlen(url)-1] == '/') ? "" : "/", fname);
+
+        FILE *fp = popen(cmd, "r");
+        if (!fp) {
+            XCONF_LOG_ERROR("[FWCHK] popen() failed for curl\n");
+            return -1;
+        }
+
+        if (fgets(line, sizeof(line), fp) == NULL) {
+            XCONF_LOG_ERROR("[FWCHK] Content-Length not found\n");
+            pclose(fp);
+            return -1;
+        }
+        pclose(fp);
+
+        uint64_t bytes = strtoull(line, NULL, 10);
+        if (bytes == 0) {
+            XCONF_LOG_ERROR("[FWCHK] Invalid Content-Length\n");
+            return -1;
+        }
+
+        fw_kb = (bytes + 1023ULL) / 1024ULL;
+        XCONF_LOG_INFO("[FWCHK] Firmware size: %" PRIu64 " kB\n", fw_kb);
+    }
+
+    {
+        FILE *fp = fopen("/proc/meminfo", "r");
+        if (!fp) {
+            XCONF_LOG_ERROR("[FWCHK] Cannot read /proc/meminfo\n");
+            return -1;
+        }
+
+        while (fgets(line, sizeof(line), fp)) {
+            if (strncmp(line, "MemAvailable:", 13) == 0) {
+                char *p = line + 13;
+                while (*p && !isdigit((unsigned char)*p)) ++p;
+                avail_kb = strtoull(p, NULL, 10);
+                break;
+            }
+        }
+        fclose(fp);
+
+        if (avail_kb == 0) {
+            XCONF_LOG_ERROR("[FWCHK] MemAvailable not found\n");
+            return -1;
+        }
+        XCONF_LOG_INFO("[FWCHK] MemAvailable: %" PRIu64 " kB\n", avail_kb);
+    }
+
+    if(syscfg_get(NULL, "FwDwld_AvlMem_RsrvThreshold", buf, sizeof(buf)) != 0){
+        XCONF_LOG_ERROR("[FWCHK] Failed to get FwDwld_AvlMem_RsrvThreshold\n");
+        return -1;
+    }
+    rsrv_mb = (uint64_t)atoi(buf);
+    rsrv_kb = rsrv_mb * 1024ULL;
+    XCONF_LOG_INFO("[FWCHK] ReserveThreshold: %llu MB\n", (unsigned long long)rsrv_mb);
+
+    if(syscfg_get(NULL, "FwDwld_ImageProcMemPercent", buf, sizeof(buf)) != 0){
+        XCONF_LOG_ERROR("[FWCHK] Failed to get FwDwld_ImageProcMemPercent\n");
+        return -1;
+    }
+    imgp_pct = atoi(buf);
+    if (imgp_pct < 0) imgp_pct = 0;
+    XCONF_LOG_INFO("[FWCHK] ImageProcPercent: %d %%\n", imgp_pct);
+
+    uint64_t img_proc_kb = (fw_kb * (uint64_t)imgp_pct + 99ULL) / 100ULL;
+    uint64_t required_kb = fw_kb + rsrv_kb + img_proc_kb;
+
+    XCONF_LOG_INFO("[FWCHK] Required Memory: %" PRIu64 " kB\n", required_kb);
+
+    if (avail_kb >= required_kb) {
+        XCONF_LOG_INFO("[FWCHK] Verdict: PROCEED\n");
+        return 1;
+    }
+
+    XCONF_LOG_INFO("[FWCHK] Verdict: BLOCK\n");
+    t2_event_d("XCONF_Dwld_Ignored_Not_EnoughMem", 1);
+    return 0;
+}

source/FwDownloadChk/fw_download_check.h

@@ -0,0 +1,24 @@
+/*
+ * If not stated otherwise in this file or this component's LICENSE file the
+ * following copyright and licenses apply:
+ *
+ * Copyright 2023 RDK Management
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+*/
+#ifndef FW_DOWNLOAD_CHECK_H
+#define FW_DOWNLOAD_CHECK_H
+
+int can_proceed_fw_download(void);
+
+#endif /* FW_DOWNLOAD_CHECK_H */

source/Makefile.am

@@ -53,5 +53,5 @@ endif
 if UNIT_TEST_DOCKER_SUPPORT
 SUBDIRS = test
 else
-SUBDIRS = Psmcli parcon LTime ServiceCtrl TimeConv webcfg_decoder MemFrag_Calc $(SETLED) $(MULTIPART_UTIL) $(BRIDGE_UTIL) $(WBCFG_TEST) $(DHCP_CLIENT_UTILS_EXAMPLE) msgq_util EventSubscription FwBankInfo $(RDKSCHEDULER_TESTAPP) $(SOCKET_EXAMPLE)
+SUBDIRS = Psmcli parcon LTime ServiceCtrl TimeConv FwDownloadChk webcfg_decoder MemFrag_Calc $(SETLED) $(MULTIPART_UTIL) $(BRIDGE_UTIL) $(WBCFG_TEST) $(DHCP_CLIENT_UTILS_EXAMPLE) msgq_util EventSubscription FwBankInfo $(RDKSCHEDULER_TESTAPP) $(SOCKET_EXAMPLE)
 endif

source/bridge_utils/bridge_utils_bin/Makefile.am

@@ -31,7 +31,7 @@ bridgeUtils_CPPFLAGS = -I$(top_srcdir)/include \
 
 bridgeUtils_SOURCES = bridge_util.c bridge_creation.c main.c bridge_util_generic.c
 
-bridgeUtils_LDFLAGS =  -lrt -lccsp_common -lsyscfg -lsysevent -lprivilege -lbridge_utils -lsecure_wrapper
+bridgeUtils_LDFLAGS =  -lrt -lccsp_common -lsyscfg -lsysevent -lprivilege -lbridge_utils -lsecure_wrapper -lOvsAgentApi -lbreakpadwrapper -lpthread -lstdc++ -lm -lcjson 
 
 if FEATURE_VTS_BRIDGE_UTIL
 lib_LTLIBRARIES = libbridgeutils.la