1 /*
2 * Copyright (c) 2021 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 "base/log/ace_trace.h"
17
18 #ifndef WINDOWS_PLATFORM
19 #include "securec.h"
20 #endif
21
22 namespace OHOS::Ace {
23 namespace {
24
25 const size_t MAX_STRING_SIZE = 128;
26
27 }
28
AceTraceBeginWithArgv(const char * format,va_list args)29 bool AceTraceBeginWithArgv(const char* format, va_list args)
30 {
31 char name[MAX_STRING_SIZE] = { 0 };
32 if (vsnprintf_s(name, sizeof(name), sizeof(name) - 1, format, args) < 0) {
33 return false;
34 }
35 AceTraceBegin(name);
36 return true;
37 }
38
AceTraceBeginWithArgs(const char * format,...)39 bool AceTraceBeginWithArgs(const char* format, ...)
40 {
41 va_list args;
42 va_start(args, format);
43 bool retVal = AceTraceBeginWithArgv(format, args);
44 va_end(args);
45 return retVal;
46 }
47
AceScopedTrace(const char * format,...)48 AceScopedTrace::AceScopedTrace(const char* format, ...) : traceEnabled_(AceTraceEnabled())
49 {
50 if (traceEnabled_) {
51 va_list args;
52 va_start(args, format);
53 traceEnabled_ = AceTraceBeginWithArgv(format, args);
54 va_end(args);
55 }
56 }
57
~AceScopedTrace()58 AceScopedTrace::~AceScopedTrace()
59 {
60 if (traceEnabled_) {
61 AceTraceEnd();
62 }
63 }
64
AceScopedTraceFlag(bool flag,const char * format,...)65 AceScopedTraceFlag::AceScopedTraceFlag(bool flag, const char* format, ...)
66 {
67 if (flag && AceTraceEnabled()) {
68 va_list args;
69 va_start(args, format);
70 flagTraceEnabled_ = AceTraceBeginWithArgv(format, args);
71 va_end(args);
72 }
73 }
74
~AceScopedTraceFlag()75 AceScopedTraceFlag::~AceScopedTraceFlag()
76 {
77 if (flagTraceEnabled_) {
78 AceTraceEnd();
79 }
80 }
81
AceAsyncTraceBeginWithArgv(int32_t taskId,const char * format,va_list args)82 std::string ACE_EXPORT AceAsyncTraceBeginWithArgv(int32_t taskId, const char* format, va_list args)
83 {
84 char name[MAX_STRING_SIZE] = { 0 };
85 if (vsnprintf_s(name, sizeof(name), sizeof(name) - 1, format, args) < 0) {
86 return "";
87 }
88 AceAsyncTraceBegin(taskId, name);
89 return name;
90 }
91
AceAsyncTraceBeginWithArgs(int32_t taskId,char * format,...)92 std::string ACE_EXPORT AceAsyncTraceBeginWithArgs(int32_t taskId, char* format, ...)
93 {
94 va_list args;
95 va_start(args, format);
96 std::string name = AceAsyncTraceBeginWithArgv(taskId, format, args);
97 va_end(args);
98 return name;
99 }
100
101 std::atomic<std::int32_t> AceAsyncScopedTrace::id_ = 0;
102
AceAsyncScopedTrace(const char * format,...)103 AceAsyncScopedTrace::AceAsyncScopedTrace(const char* format, ...) : asyncTraceEnabled_(AceTraceEnabled())
104 {
105 taskId_ = id_++;
106 if (asyncTraceEnabled_) {
107 va_list args;
108 va_start(args, format);
109 name_ = AceAsyncTraceBeginWithArgv(taskId_, format, args);
110 va_end(args);
111 }
112 }
113
~AceAsyncScopedTrace()114 AceAsyncScopedTrace::~AceAsyncScopedTrace()
115 {
116 if (!name_.empty()) {
117 AceAsyncTraceEnd(taskId_, name_.c_str());
118 }
119 }
120 } // namespace OHOS::Ace
121