1 /*
2 * Copyright (c) 2025 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include <array>
17 #include "hitraceid.h"
18 #include "hitrace_chain_ani_util.h"
19 #include "hitrace_chain_ani_parameter_name.h"
20
21 using namespace OHOS::HiviewDFX;
22 #undef LOG_DOMAIN
23 #define LOG_DOMAIN 0xD002D33
24
25 #undef LOG_TAG
26 #define LOG_TAG "HitraceUtilAni"
27
IsRefUndefined(ani_env * env,ani_ref ref)28 bool HiTraceChainAniUtil::IsRefUndefined(ani_env* env, ani_ref ref)
29 {
30 ani_boolean isUndefined = false;
31 env->Reference_IsUndefined(ref, &isUndefined);
32 return isUndefined;
33 }
34
ParseStringValue(ani_env * env,ani_ref aniStrRef,std::string & name)35 ani_status HiTraceChainAniUtil::ParseStringValue(ani_env* env, ani_ref aniStrRef, std::string& name)
36 {
37 ani_size strSize = 0;
38 if (env->String_GetUTF8Size(static_cast<ani_string>(aniStrRef), &strSize) != ANI_OK) {
39 HILOG_ERROR(LOG_CORE, "String_GetUTF8Size failed");
40 return ANI_ERROR;
41 }
42 std::vector<char> buffer(strSize + 1);
43 char* utf8Buffer = buffer.data();
44 ani_size bytesWritten = 0;
45 if (env->String_GetUTF8(static_cast<ani_string>(aniStrRef), utf8Buffer, strSize + 1, &bytesWritten) != ANI_OK) {
46 HILOG_ERROR(LOG_CORE, "String_GetUTF8 failed");
47 return ANI_ERROR;
48 }
49 utf8Buffer[bytesWritten] = '\0';
50 name = std::string(utf8Buffer);
51 return ANI_OK;
52 }
53
ParseBigIntValue(ani_env * env,ani_ref elementRef)54 uint64_t HiTraceChainAniUtil::ParseBigIntValue(ani_env* env, ani_ref elementRef)
55 {
56 ani_long longNum = static_cast<ani_long>(0);
57 ani_class cls {};
58 if (env->FindClass(CLASS_NAME_BIGINT, &cls) != ANI_OK) {
59 HILOG_ERROR(LOG_CORE, "find class %{public}s failed", CLASS_NAME_BIGINT);
60 return static_cast<uint64_t>(longNum);
61 }
62 ani_method getLongMethod {};
63 if (env->Class_FindMethod(cls, FUNC_NAME_GETLONG, ":J", &getLongMethod) != ANI_OK) {
64 HILOG_ERROR(LOG_CORE, "find method %{public}s failed", FUNC_NAME_GETLONG);
65 return static_cast<uint64_t>(longNum);
66 }
67 if (env->Object_CallMethod_Long(static_cast<ani_object>(elementRef), getLongMethod, &longNum) != ANI_OK) {
68 HILOG_ERROR(LOG_CORE, "call method %{public}s failed", FUNC_NAME_GETLONG);
69 return static_cast<uint64_t>(longNum);
70 }
71 return static_cast<uint64_t>(longNum);
72 }
73
ParseNumberValueInt64(ani_env * env,ani_ref elementRef)74 int64_t HiTraceChainAniUtil::ParseNumberValueInt64(ani_env* env, ani_ref elementRef)
75 {
76 ani_double doubleVal = 0;
77 ani_class cls {};
78 if (env->FindClass(CLASS_NAME_DOUBLE, &cls) != ANI_OK) {
79 HILOG_ERROR(LOG_CORE, "find class %{public}s failed", CLASS_NAME_DOUBLE);
80 return static_cast<int64_t>(doubleVal);
81 }
82 ani_method unboxedMethod {};
83 if (env->Class_FindMethod(cls, FUNC_NAME_UNBOXED, ":D", &unboxedMethod) != ANI_OK) {
84 HILOG_ERROR(LOG_CORE, "find method %{public}s failed", FUNC_NAME_UNBOXED);
85 return static_cast<int64_t>(doubleVal);
86 }
87 if (env->Object_CallMethod_Double(static_cast<ani_object>(elementRef), unboxedMethod, &doubleVal) != ANI_OK) {
88 HILOG_ERROR(LOG_CORE, "call method %{public}s failed", FUNC_NAME_UNBOXED);
89 return static_cast<int64_t>(doubleVal);
90 }
91 return static_cast<int64_t>(doubleVal);
92 }
93
ParseNumberValueInt32(ani_env * env,ani_ref elementRef)94 int32_t HiTraceChainAniUtil::ParseNumberValueInt32(ani_env* env, ani_ref elementRef)
95 {
96 ani_double doubleVal = 0;
97 ani_class cls {};
98 if (env->FindClass(CLASS_NAME_DOUBLE, &cls) != ANI_OK) {
99 HILOG_ERROR(LOG_CORE, "find class %{public}s failed", CLASS_NAME_DOUBLE);
100 return static_cast<int32_t>(doubleVal);
101 }
102 ani_method unboxedMethod {};
103 if (env->Class_FindMethod(cls, FUNC_NAME_UNBOXED, ":D", &unboxedMethod) != ANI_OK) {
104 HILOG_ERROR(LOG_CORE, "find method %{public}s failed", FUNC_NAME_UNBOXED);
105 return static_cast<int32_t>(doubleVal);
106 }
107 if (env->Object_CallMethod_Double(static_cast<ani_object>(elementRef), unboxedMethod, &doubleVal) != ANI_OK) {
108 HILOG_ERROR(LOG_CORE, "call method %{public}s failed", FUNC_NAME_UNBOXED);
109 return static_cast<int32_t>(doubleVal);
110 }
111 return static_cast<int32_t>(doubleVal);
112 }
113
EnumGetValueInt32(ani_env * env,ani_enum_item enumItem,int32_t & value)114 ani_status HiTraceChainAniUtil::EnumGetValueInt32(ani_env* env, ani_enum_item enumItem, int32_t& value)
115 {
116 ani_int aniInt {};
117 ani_status result = env->EnumItem_GetValue_Int(enumItem, &aniInt);
118 if (result != ANI_OK) {
119 HILOG_ERROR(LOG_CORE, "get Int32 %{public}d failed", result);
120 return result;
121 }
122 value = static_cast<int32_t>(aniInt);
123 return result;
124 }
125
CreateBigInt(ani_env * env,uint64_t traceId)126 ani_object HiTraceChainAniUtil::CreateBigInt(ani_env* env, uint64_t traceId)
127 {
128 ani_object bigintCtor = nullptr;
129 ani_class bigIntCls;
130 if (env->FindClass(CLASS_NAME_BIGINT, &bigIntCls) != ANI_OK) {
131 HILOG_ERROR(LOG_CORE, "find Class %{public}s failed", CLASS_NAME_BIGINT);
132 return bigintCtor;
133 }
134 ani_method createLongMethod;
135 if (env->Class_FindMethod(bigIntCls, "<ctor>", "J:V", &createLongMethod) != ANI_OK) {
136 HILOG_ERROR(LOG_CORE, "find method %{public}s constructor failed", CLASS_NAME_BIGINT);
137 return bigintCtor;
138 }
139 ani_long longNum = static_cast<ani_long>(traceId);
140 if (env->Object_New(bigIntCls, createLongMethod, &bigintCtor, longNum) != ANI_OK) {
141 HILOG_ERROR(LOG_CORE, "create object %{public}s failed", CLASS_NAME_BIGINT);
142 return bigintCtor;
143 }
144 return bigintCtor;
145 }
146
CreateDoubleUint64(ani_env * env,uint64_t number)147 ani_object HiTraceChainAniUtil::CreateDoubleUint64(ani_env* env, uint64_t number)
148 {
149 ani_object personInfoObj = nullptr;
150 ani_class personCls;
151 if (env->FindClass(CLASS_NAME_DOUBLE, &personCls) != ANI_OK) {
152 HILOG_ERROR(LOG_CORE, "find class %{public}s failed", CLASS_NAME_DOUBLE);
153 return personInfoObj;
154 }
155 ani_method personInfoCtor;
156 env->Class_FindMethod(personCls, "<ctor>", "D:V", &personInfoCtor);
157 env->Object_New(personCls, personInfoCtor, &personInfoObj, static_cast<ani_double>(number));
158 return personInfoObj;
159 }
160
CreateDoubleInt(ani_env * env,int number)161 ani_object HiTraceChainAniUtil::CreateDoubleInt(ani_env* env, int number)
162 {
163 ani_object personInfoObj = nullptr;
164 ani_class personCls;
165 if (env->FindClass(CLASS_NAME_DOUBLE, &personCls) != ANI_OK) {
166 HILOG_ERROR(LOG_CORE, "find class %{public}s failed", CLASS_NAME_DOUBLE);
167 return personInfoObj;
168 }
169 ani_method personInfoCtor;
170 env->Class_FindMethod(personCls, "<ctor>", "D:V", &personInfoCtor);
171 env->Object_New(personCls, personInfoCtor, &personInfoObj, static_cast<ani_double>(number));
172 return personInfoObj;
173 }
174