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
AceCountTraceWidthArgs(int32_t count,const char * format,...)48 void AceCountTraceWidthArgs(int32_t count, const char* format, ...)
49 {
50 char name[MAX_STRING_SIZE] = { 0 };
51 va_list args;
52 va_start(args, format);
53 if (vsnprintf_s(name, sizeof(name), sizeof(name) - 1, format, args) < 0) {
54 return;
55 }
56 va_end(args);
57 AceCountTrace(name, count);
58 }
59
AceScopedTrace(const char * format,...)60 AceScopedTrace::AceScopedTrace(const char* format, ...) : traceEnabled_(AceTraceEnabled())
61 {
62 if (traceEnabled_) {
63 va_list args;
64 va_start(args, format);
65 traceEnabled_ = AceTraceBeginWithArgv(format, args);
66 va_end(args);
67 }
68 }
69
~AceScopedTrace()70 AceScopedTrace::~AceScopedTrace()
71 {
72 if (traceEnabled_) {
73 AceTraceEnd();
74 }
75 }
76
AceScopedTraceFlag(bool flag,const char * format,...)77 AceScopedTraceFlag::AceScopedTraceFlag(bool flag, const char* format, ...)
78 {
79 if (flag && AceTraceEnabled()) {
80 va_list args;
81 va_start(args, format);
82 flagTraceEnabled_ = AceTraceBeginWithArgv(format, args);
83 va_end(args);
84 }
85 }
86
~AceScopedTraceFlag()87 AceScopedTraceFlag::~AceScopedTraceFlag()
88 {
89 if (flagTraceEnabled_) {
90 AceTraceEnd();
91 }
92 }
93
AceAsyncTraceBeginWithArgv(int32_t taskId,const char * format,va_list args)94 std::string ACE_EXPORT AceAsyncTraceBeginWithArgv(int32_t taskId, const char* format, va_list args)
95 {
96 char name[MAX_STRING_SIZE] = { 0 };
97 if (vsnprintf_s(name, sizeof(name), sizeof(name) - 1, format, args) < 0) {
98 return "";
99 }
100 AceAsyncTraceBegin(taskId, name);
101 return name;
102 }
103
AceAsyncTraceBeginWithArgs(int32_t taskId,char * format,...)104 std::string ACE_EXPORT AceAsyncTraceBeginWithArgs(int32_t taskId, char* format, ...)
105 {
106 va_list args;
107 va_start(args, format);
108 std::string name = AceAsyncTraceBeginWithArgv(taskId, format, args);
109 va_end(args);
110 return name;
111 }
112
113 std::atomic<std::int32_t> AceAsyncScopedTrace::id_ = 0;
114
AceAsyncScopedTrace(const char * format,...)115 AceAsyncScopedTrace::AceAsyncScopedTrace(const char* format, ...) : asyncTraceEnabled_(AceTraceEnabled())
116 {
117 taskId_ = id_++;
118 if (asyncTraceEnabled_) {
119 va_list args;
120 va_start(args, format);
121 name_ = AceAsyncTraceBeginWithArgv(taskId_, format, args);
122 va_end(args);
123 }
124 }
125
~AceAsyncScopedTrace()126 AceAsyncScopedTrace::~AceAsyncScopedTrace()
127 {
128 if (!name_.empty()) {
129 AceAsyncTraceEnd(taskId_, name_.c_str());
130 }
131 }
132 } // namespace OHOS::Ace
133