// SPDX-License-Identifier: BSD-2-Clause /* * Copyright (c) 2015, Linaro Limited */ #include #include #include #include #include #include #include #include #include #define TA_NAME "stats.ta" #define STATS_UUID \ { 0xd96a5b40, 0xe2c7, 0xb1af, \ { 0x87, 0x94, 0x10, 0x02, 0xa5, 0xd5, 0xc6, 0x1b } } #define STATS_CMD_PAGER_STATS 0 #define STATS_CMD_ALLOC_STATS 1 #define STATS_CMD_MEMLEAK_STATS 2 /* * UTEE_ENTRY_FUNC_DUMP_MEMSTATS * [out] memref[0] Array of context information of loaded TAs * * Each cell of the TA information array contains: * TEE_UUID TA UUID * uint32_t Non zero if TA panicked, 0 otherwise * uint32_t Number of sessions opened by the TA * uint32_t Byte size currently allocated in TA heap * uint32_t Max bytes allocated since last stats reset * uint32_t TA heap pool byte size * uint32_t Number of failed allocation requests * uint32_t Biggest byte size which allocation failed * uint32_t Biggest byte size which allocation succeeded */ #define STATS_CMD_TA_STATS 3 #define STATS_NB_POOLS 4 static TEE_Result get_alloc_stats(uint32_t type, TEE_Param p[TEE_NUM_PARAMS]) { struct malloc_stats *stats; uint32_t size_to_retrieve; uint32_t pool_id; uint32_t i; /* * p[0].value.a = pool id (from 0 to n) * - 0 means all the pools to be retrieved * - 1..n means pool id * p[0].value.b = 0 if no reset of the stats * p[1].memref.buffer = output buffer to struct malloc_stats */ if (TEE_PARAM_TYPES(TEE_PARAM_TYPE_VALUE_INPUT, TEE_PARAM_TYPE_MEMREF_OUTPUT, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE) != type) { return TEE_ERROR_BAD_PARAMETERS; } pool_id = p[0].value.a; if (pool_id > STATS_NB_POOLS) return TEE_ERROR_BAD_PARAMETERS; size_to_retrieve = sizeof(struct malloc_stats); if (!pool_id) size_to_retrieve *= STATS_NB_POOLS; if (p[1].memref.size < size_to_retrieve) { p[1].memref.size = size_to_retrieve; return TEE_ERROR_SHORT_BUFFER; } p[1].memref.size = size_to_retrieve; stats = p[1].memref.buffer; for (i = 1; i <= STATS_NB_POOLS; i++) { if ((pool_id) && (i != pool_id)) continue; switch (i) { case 1: malloc_get_stats(stats); strlcpy(stats->desc, "Heap", sizeof(stats->desc)); if (p[0].value.b) malloc_reset_stats(); break; case 2: EMSG("public DDR not managed by secure side anymore"); break; case 3: tee_mm_get_pool_stats(&tee_mm_sec_ddr, stats, !!p[0].value.b); strlcpy(stats->desc, "Secure DDR", sizeof(stats->desc)); break; #ifdef CFG_VIRTUALIZATION case 4: nex_malloc_get_stats(stats); strlcpy(stats->desc, "KHeap", sizeof(stats->desc)); if (p[0].value.b) nex_malloc_reset_stats(); break; #endif default: EMSG("Wrong pool id"); break; } stats++; } return TEE_SUCCESS; } static TEE_Result get_pager_stats(uint32_t type, TEE_Param p[TEE_NUM_PARAMS]) { struct tee_pager_stats stats; if (TEE_PARAM_TYPES(TEE_PARAM_TYPE_VALUE_OUTPUT, TEE_PARAM_TYPE_VALUE_OUTPUT, TEE_PARAM_TYPE_VALUE_OUTPUT, TEE_PARAM_TYPE_NONE) != type) { EMSG("expect 3 output values as argument"); return TEE_ERROR_BAD_PARAMETERS; } tee_pager_get_stats(&stats); p[0].value.a = stats.npages; p[0].value.b = stats.npages_all; p[1].value.a = stats.ro_hits; p[1].value.b = stats.rw_hits; p[2].value.a = stats.hidden_hits; p[2].value.b = stats.zi_released; return TEE_SUCCESS; } static TEE_Result get_memleak_stats(uint32_t type, TEE_Param p[TEE_NUM_PARAMS] __maybe_unused) { if (TEE_PARAM_TYPES(TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE) != type) return TEE_ERROR_BAD_PARAMETERS; mdbg_check(1); return TEE_SUCCESS; } static TEE_Result get_user_ta_stats(uint32_t type, TEE_Param p[TEE_NUM_PARAMS] __maybe_unused) { uint32_t res = TEE_SUCCESS; if (TEE_PARAM_TYPES(TEE_PARAM_TYPE_MEMREF_OUTPUT, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE, TEE_PARAM_TYPE_NONE) != type) return TEE_ERROR_BAD_PARAMETERS; #if defined(CFG_TA_STATS) res = tee_ta_instance_stats(p[0].memref.buffer, &p[0].memref.size); if (res != TEE_SUCCESS) DMSG("tee_ta_dump_stats return: 0x%"PRIx32, res); #else res = TEE_ERROR_NOT_SUPPORTED; #endif return res; } /* * Trusted Application Entry Points */ static TEE_Result invoke_command(void *psess __unused, uint32_t cmd, uint32_t ptypes, TEE_Param params[TEE_NUM_PARAMS]) { switch (cmd) { case STATS_CMD_PAGER_STATS: return get_pager_stats(ptypes, params); case STATS_CMD_ALLOC_STATS: return get_alloc_stats(ptypes, params); case STATS_CMD_MEMLEAK_STATS: return get_memleak_stats(ptypes, params); case STATS_CMD_TA_STATS: return get_user_ta_stats(ptypes, params); default: break; } return TEE_ERROR_BAD_PARAMETERS; } pseudo_ta_register(.uuid = STATS_UUID, .name = TA_NAME, .flags = PTA_DEFAULT_FLAGS, .invoke_command_entry_point = invoke_command);