RepoMirrors/gperftools
1
0
Fork 0
mirror of https://github.com/gperftools/gperftools synced 2025-03-10 23:07:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

update stacktrace_unittest to test overflow issue

Browse Source 
Signed-off-by: Aliaksey Kandratsenka <alkondratenko@gmail.com>
[alkondratenko@gmail.com: squashed log update commit here]
This commit is contained in:
Yikai Zhao 2023-10-13 13:04:35 +08:00 committed by Aliaksey Kandratsenka
parent dad9e8ceb9
commit 5ba86d37a3
1 changed files with 24 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/tests/stacktrace_unittest.cc
View File

@ -52,6 +52,8 @@
# endif # endif
#endif #endif
#include <vector>
#include "base/commandlineflags.h" #include "base/commandlineflags.h"
#include "base/logging.h" #include "base/logging.h"
#include <gperftools/stacktrace.h> #include <gperftools/stacktrace.h>
@ -179,6 +181,8 @@ int ATTRIBUTE_NOINLINE CaptureLeafUContext(void **stack, int stack_len) {
gst_args.size_ptr = &size; gst_args.size_ptr = &size;
gst_args.result = stack; gst_args.result = stack;
gst_args.max_depth = stack_len; gst_args.max_depth = stack_len;
gst_args.captured = false;
gst_args.ready = false;
struct itimerval it; struct itimerval it;
it.it_interval.tv_sec = 0; it.it_interval.tv_sec = 0;
@ -186,8 +190,6 @@ int ATTRIBUTE_NOINLINE CaptureLeafUContext(void **stack, int stack_len) {
it.it_value.tv_sec = 0; it.it_value.tv_sec = 0;
it.it_value.tv_usec = 1; it.it_value.tv_usec = 1;
CHECK(!gst_args.captured);
rv = setitimer(ITIMER_PROF, &it, nullptr); rv = setitimer(ITIMER_PROF, &it, nullptr);
CHECK(rv == 0); CHECK(rv == 0);
@ -266,19 +268,21 @@ int ATTRIBUTE_NOINLINE CaptureLeafWSkip(void **stack, int stack_len) {
void ATTRIBUTE_NOINLINE CheckStackTrace(int); void ATTRIBUTE_NOINLINE CheckStackTrace(int);
int (*leaf_capture_fn)(void**, int) = CaptureLeafPlain; int (*leaf_capture_fn)(void**, int) = CaptureLeafPlain;
int leaf_capture_len = 20;
void ATTRIBUTE_NOINLINE CheckStackTraceLeaf(int i) { void ATTRIBUTE_NOINLINE CheckStackTraceLeaf(int i) {
const int STACK_LEN = 20; std::vector<void*> stack(leaf_capture_len + 1);
void *stack[STACK_LEN]; stack[leaf_capture_len] = (void*)0x42; // we will check that this value is not overwritten
int size; int size = 0;
ADJUST_ADDRESS_RANGE_FROM_RA(&expected_range[1]); ADJUST_ADDRESS_RANGE_FROM_RA(&expected_range[1]);
size = leaf_capture_fn(stack, STACK_LEN); size = leaf_capture_fn(stack.data(), leaf_capture_len);
CHECK_EQ(stack[leaf_capture_len], (void*)0x42);
#ifdef HAVE_EXECINFO_H #ifdef HAVE_EXECINFO_H
{ {
char **strings = backtrace_symbols(stack, size); char **strings = backtrace_symbols(stack.data(), size);
for (int i = 0; i < size; i++) for (int i = 0; i < size; i++)
printf("%s %p\n", strings[i], stack[i]); printf("%s %p\n", strings[i], stack[i]);
printf("CheckStackTrace() addr: %p\n", &CheckStackTrace); printf("CheckStackTrace() addr: %p\n", &CheckStackTrace);
@ -286,11 +290,12 @@ void ATTRIBUTE_NOINLINE CheckStackTraceLeaf(int i) {
} }
#endif #endif
for (int i = 0, j = 0; i < BACKTRACE_STEPS; i++, j++) { for (int i = 0, j = 0; i < BACKTRACE_STEPS && j < size; i++, j++) {
if (i == 1 && j == 1) { if (i == 1 && j == 1) {
// this is expected to be our function for which we don't // this is expected to be our function for which we don't
// establish bounds. So skip. // establish bounds. So skip.
j++; i--;
continue;
} }
printf("Backtrace %d: expected: %p..%p actual: %p ... ", printf("Backtrace %d: expected: %p..%p actual: %p ... ",
i, expected_range[i].start, expected_range[i].end, stack[j]); i, expected_range[i].start, expected_range[i].end, stack[j]);
@ -348,7 +353,7 @@ void ATTRIBUTE_NOINLINE CheckStackTrace(int i) {
//-----------------------------------------------------------------------// //-----------------------------------------------------------------------//
int main(int argc, char ** argv) { void RunTest() {
CheckStackTrace(0); CheckStackTrace(0);
printf("PASS\n"); printf("PASS\n");
@ -367,6 +372,15 @@ int main(int argc, char ** argv) {
CheckStackTrace(0); CheckStackTrace(0);
printf("PASS\n"); printf("PASS\n");
#endif // TEST_UCONTEXT_BITS #endif // TEST_UCONTEXT_BITS
}
int main(int argc, char ** argv) {
leaf_capture_len = 20;
RunTest();
printf("\nSet max capture length to 3:\n");
leaf_capture_len = 3; // less than stack depth
RunTest();
return 0; return 0;
} }