mirror of https://github.com/grpc/grpc.git
Esun Kim
4 years ago
committed by
GitHub
commit
e0cbc7e5dd
28 changed files with 564 additions and 33 deletions
@ -0,0 +1,43 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2020 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include "src/core/lib/gprpp/examine_stack.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
gpr_current_stack_trace_func g_current_stack_trace_provider = nullptr; |
||||
|
||||
gpr_current_stack_trace_func GetCurrentStackTraceProvider() { |
||||
return g_current_stack_trace_provider; |
||||
} |
||||
|
||||
void SetCurrentStackTraceProvider( |
||||
gpr_current_stack_trace_func current_stack_trace_provider) { |
||||
g_current_stack_trace_provider = current_stack_trace_provider; |
||||
} |
||||
|
||||
absl::optional<std::string> GetCurrentStackTrace() { |
||||
if (g_current_stack_trace_provider != nullptr) { |
||||
return g_current_stack_trace_provider(); |
||||
} |
||||
return absl::nullopt; |
||||
} |
||||
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,46 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2020 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_CORE_LIB_GPRPP_EXAMINE_STACK_H |
||||
#define GRPC_CORE_LIB_GPRPP_EXAMINE_STACK_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include <functional> |
||||
#include <string> |
||||
|
||||
#include "absl/types/optional.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
typedef std::string (*gpr_current_stack_trace_func)(); |
||||
|
||||
// Returns a current_stack_trace_provider.
|
||||
gpr_current_stack_trace_func GetCurrentStackTraceProvider(); |
||||
|
||||
// Sets current_stack_trace_provider which provides a current-stack trace.
|
||||
void SetCurrentStackTraceProvider( |
||||
gpr_current_stack_trace_func current_stack_trace_provider); |
||||
|
||||
// Returns the current stack trace as a string via current_stack_trace_provider
|
||||
// If current_stack_trace_provider is not set, it returns absl::nullopt.
|
||||
absl::optional<std::string> GetCurrentStackTrace(); |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif /* GRPC_CORE_LIB_GPRPP_EXAMINE_STACK_H */ |
||||
@ -0,0 +1,82 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2020 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include "src/core/lib/gprpp/examine_stack.h" |
||||
|
||||
#include <stdio.h> |
||||
#include <string.h> |
||||
|
||||
#include <gtest/gtest.h> |
||||
|
||||
#include "absl/debugging/stacktrace.h" |
||||
#include "absl/debugging/symbolize.h" |
||||
|
||||
#include <grpc/support/log.h> |
||||
|
||||
namespace { |
||||
|
||||
std::string SimpleCurrentStackTraceProvider() { return "stacktrace"; } |
||||
|
||||
std::string AbseilCurrentStackTraceProvider() { |
||||
std::string result = "Stack trace:\n"; |
||||
constexpr int kNumStackFrames = 10; |
||||
void* stack[kNumStackFrames]; |
||||
int frame_sizes[kNumStackFrames]; |
||||
int depth = absl::GetStackFrames(stack, frame_sizes, kNumStackFrames, 1); |
||||
for (int i = 0; i < depth; i++) { |
||||
char tmp[1024]; |
||||
const char* symbol = "(unknown)"; |
||||
if (absl::Symbolize(stack[i], tmp, sizeof(tmp))) { |
||||
symbol = tmp; |
||||
} |
||||
result += symbol; |
||||
result += +"\n"; |
||||
} |
||||
return result; |
||||
} |
||||
|
||||
} // namespace
|
||||
|
||||
TEST(ExamineStackTest, NullStackProvider) { |
||||
grpc_core::SetCurrentStackTraceProvider(nullptr); |
||||
EXPECT_EQ(grpc_core::GetCurrentStackTraceProvider(), nullptr); |
||||
EXPECT_EQ(grpc_core::GetCurrentStackTrace(), absl::nullopt); |
||||
} |
||||
|
||||
TEST(ExamineStackTest, SimpleStackProvider) { |
||||
grpc_core::SetCurrentStackTraceProvider(&SimpleCurrentStackTraceProvider); |
||||
EXPECT_NE(grpc_core::GetCurrentStackTraceProvider(), nullptr); |
||||
EXPECT_EQ(grpc_core::GetCurrentStackTrace(), "stacktrace"); |
||||
} |
||||
|
||||
TEST(ExamineStackTest, AbseilStackProvider) { |
||||
grpc_core::SetCurrentStackTraceProvider(&AbseilCurrentStackTraceProvider); |
||||
EXPECT_NE(grpc_core::GetCurrentStackTraceProvider(), nullptr); |
||||
const absl::optional<std::string> stack_trace = |
||||
grpc_core::GetCurrentStackTrace(); |
||||
EXPECT_NE(stack_trace, absl::nullopt); |
||||
gpr_log(GPR_INFO, "stack_trace=%s", stack_trace->c_str()); |
||||
EXPECT_TRUE(stack_trace->find("GetCurrentStackTrace") != -1); |
||||
} |
||||
|
||||
int main(int argc, char** argv) { |
||||
absl::InitializeSymbolizer(argv[0]); |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
int ret = RUN_ALL_TESTS(); |
||||
return ret; |
||||
} |
||||
@ -0,0 +1,41 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2020 the gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include "test/core/util/stack_tracer.h" |
||||
|
||||
#include <gtest/gtest.h> |
||||
#include <string> |
||||
|
||||
#include "absl/debugging/symbolize.h" |
||||
|
||||
#include <grpc/support/log.h> |
||||
|
||||
#include "test/core/util/test_config.h" |
||||
|
||||
TEST(StackTracerTest, Basic) { |
||||
std::string stack_trace = grpc_core::testing::GetCurrentStackTrace(); |
||||
gpr_log(GPR_INFO, "stack_trace=%s", stack_trace.c_str()); |
||||
EXPECT_TRUE(stack_trace.find("Basic") != -1); |
||||
} |
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc::testing::TestEnvironment env(argc, argv); |
||||
testing::InitGoogleTest(&argc, argv); |
||||
int ret = RUN_ALL_TESTS(); |
||||
return ret; |
||||
} |
||||
Loading…
Reference in new issue