archived-mpp/osal/mpp_singleton.c
Yanjun Liao 52f005f2df fix[mpp_singleton]: Fix 64-bit mask overflow
Use 64-bit literals in bit operations to prevent overflow
when singleton module ID >= 32.

Change-Id: Ib08190bbcbb1a0b46559ea3c2ae57d801b5efadf
Signed-off-by: Yanjun Liao <yanjun.liao@rock-chips.com>
2025-11-05 09:08:10 +08:00

120 lines
3 KiB
C

/* SPDX-License-Identifier: Apache-2.0 OR MIT */
/*
* Copyright (c) 2024 Rockchip Electronics Co., Ltd.
*/
#define MODULE_TAG "mpp_singleton"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "mpp_env.h"
#include "mpp_time.h"
#include "mpp_singleton.h"
#define sgln_dbg(fmt, ...) \
do { \
if (sgln_debug) \
printf(MODULE_TAG ": " fmt, ##__VA_ARGS__); \
} while (0)
static MppSingletonInfo sgln_info[MPP_SGLN_MAX_CNT] = {0};
static rk_u32 sgln_max_name_len = 12;
static rk_u64 sgln_mask = 0;
static rk_u32 sgln_debug = 0;
rk_s32 mpp_singleton_add(MppSingletonInfo *info, const char *caller)
{
mpp_env_get_u32("mpp_sgln_debug", &sgln_debug, 0);
if (!info) {
sgln_dbg("can not add NULL info at %s\n", caller);
return rk_nok;
}
if (info->id >= MPP_SGLN_MAX_CNT) {
sgln_dbg("id %d larger than max %d at %s\n", info->id, MPP_SGLN_MAX_CNT, caller);
return rk_nok;
}
if (sgln_mask & ((rk_u64)1 << info->id)) {
sgln_dbg("info %d has been registered at %s\n", info->id, caller);
return rk_nok;
}
sgln_info[info->id] = *info;
sgln_mask |= ((rk_u64)1 << info->id);
{
rk_u32 name_len = strlen(info->name);
if (name_len > sgln_max_name_len)
sgln_max_name_len = name_len;
}
sgln_dbg("info %2d %-*s registered at %s\n", info->id,
sgln_max_name_len, info->name, caller);
return rk_ok;
}
static void mpp_singleton_deinit(void)
{
rk_s32 i;
sgln_dbg("deinit enter\n");
/* NOTE: revert deinit order */
for (i = MPP_SGLN_MAX_CNT - 1; i >= 0; i--) {
if (sgln_mask & ((rk_u64)1 << i)) {
MppSingletonInfo *info = &sgln_info[i];
if (info->deinit) {
sgln_dbg("info %2d %-*s deinit start\n", info->id,
sgln_max_name_len, info->name);
info->deinit();
sgln_dbg("info %2d %-*s deinit finish\n", info->id,
sgln_max_name_len, info->name);
}
}
}
sgln_dbg("deinit leave\n");
}
__attribute__((constructor(65535))) static void mpp_singleton_init(void)
{
rk_s64 sum = 0;
rk_s32 i;
sgln_dbg("init enter\n");
/* NOTE: call atexit first to avoid init crash but not deinit */
atexit(mpp_singleton_deinit);
for (i = 0; i < MPP_SGLN_MAX_CNT; i++) {
if (sgln_mask & ((rk_u64)1 << i)) {
MppSingletonInfo *info = &sgln_info[i];
if (info->init) {
rk_s64 time;
sgln_dbg("info %2d %-*s init start\n", info->id,
sgln_max_name_len, info->name);
time = mpp_time();
info->init();
time = mpp_time() - time;
sum += time;
sgln_dbg("info %2d %-*s init finish %lld us\n", info->id,
sgln_max_name_len, info->name, time);
}
}
}
sgln_dbg("init leave total %lld us\n", sum);
}