OLD | NEW |
1 // Copyright (c) 2008, Google Inc. | 1 // Copyright (c) 2008, Google Inc. |
2 // All rights reserved. | 2 // All rights reserved. |
3 // | 3 // |
4 // Redistribution and use in source and binary forms, with or without | 4 // Redistribution and use in source and binary forms, with or without |
5 // modification, are permitted provided that the following conditions are | 5 // modification, are permitted provided that the following conditions are |
6 // met: | 6 // met: |
7 // | 7 // |
8 // * Redistributions of source code must retain the above copyright | 8 // * Redistributions of source code must retain the above copyright |
9 // notice, this list of conditions and the following disclaimer. | 9 // notice, this list of conditions and the following disclaimer. |
10 // * Redistributions in binary form must reproduce the above | 10 // * Redistributions in binary form must reproduce the above |
(...skipping 18 matching lines...) Expand all Loading... |
29 | 29 |
30 // --- | 30 // --- |
31 // Author: Craig Silverstein | 31 // Author: Craig Silverstein |
32 // | 32 // |
33 // Simple test of malloc_extension. Includes test of C shims. | 33 // Simple test of malloc_extension. Includes test of C shims. |
34 | 34 |
35 #include "config_for_unittests.h" | 35 #include "config_for_unittests.h" |
36 #include <stdio.h> | 36 #include <stdio.h> |
37 #include <sys/types.h> | 37 #include <sys/types.h> |
38 #include "base/logging.h" | 38 #include "base/logging.h" |
39 #include <gperftools/malloc_extension.h> | 39 #include <google/malloc_extension.h> |
40 #include <gperftools/malloc_extension_c.h> | 40 #include <google/malloc_extension_c.h> |
41 | 41 |
42 using STL_NAMESPACE::vector; | 42 using STL_NAMESPACE::vector; |
43 | 43 |
44 int main(int argc, char** argv) { | 44 int main(int argc, char** argv) { |
45 void* a = malloc(1000); | 45 void* a = malloc(1000); |
46 | 46 |
47 size_t cxx_bytes_used, c_bytes_used; | 47 size_t cxx_bytes_used, c_bytes_used; |
48 ASSERT_TRUE(MallocExtension::instance()->GetNumericProperty( | 48 ASSERT_TRUE(MallocExtension::instance()->GetNumericProperty( |
49 "generic.current_allocated_bytes", &cxx_bytes_used)); | 49 "generic.current_allocated_bytes", &cxx_bytes_used)); |
50 ASSERT_TRUE(MallocExtension_GetNumericProperty( | 50 ASSERT_TRUE(MallocExtension_GetNumericProperty( |
51 "generic.current_allocated_bytes", &c_bytes_used)); | 51 "generic.current_allocated_bytes", &c_bytes_used)); |
52 ASSERT_GT(cxx_bytes_used, 1000); | 52 ASSERT_GT(cxx_bytes_used, 1000); |
53 ASSERT_EQ(cxx_bytes_used, c_bytes_used); | 53 ASSERT_EQ(cxx_bytes_used, c_bytes_used); |
54 | 54 |
55 ASSERT_TRUE(MallocExtension::instance()->VerifyAllMemory()); | 55 ASSERT_TRUE(MallocExtension::instance()->VerifyAllMemory()); |
56 ASSERT_TRUE(MallocExtension_VerifyAllMemory()); | 56 ASSERT_TRUE(MallocExtension_VerifyAllMemory()); |
57 | 57 |
58 ASSERT_EQ(MallocExtension::kOwned, | |
59 MallocExtension::instance()->GetOwnership(a)); | |
60 // TODO(csilvers): this relies on undocumented behavior that | |
61 // GetOwnership works on stack-allocated variables. Use a better test. | |
62 ASSERT_EQ(MallocExtension::kNotOwned, | |
63 MallocExtension::instance()->GetOwnership(&cxx_bytes_used)); | |
64 ASSERT_EQ(MallocExtension::kNotOwned, | |
65 MallocExtension::instance()->GetOwnership(NULL)); | |
66 ASSERT_GE(MallocExtension::instance()->GetAllocatedSize(a), 1000); | 58 ASSERT_GE(MallocExtension::instance()->GetAllocatedSize(a), 1000); |
67 // This is just a sanity check. If we allocated too much, tcmalloc is broken | 59 // This is just a sanity check. If we allocated too much, tcmalloc is broken |
68 ASSERT_LE(MallocExtension::instance()->GetAllocatedSize(a), 5000); | 60 ASSERT_LE(MallocExtension::instance()->GetAllocatedSize(a), 5000); |
69 ASSERT_GE(MallocExtension::instance()->GetEstimatedAllocatedSize(1000), 1000); | 61 ASSERT_GE(MallocExtension::instance()->GetEstimatedAllocatedSize(1000), 1000); |
70 | 62 |
71 for (int i = 0; i < 10; ++i) { | 63 for (int i = 0; i < 10; ++i) { |
72 void *p = malloc(i); | 64 void *p = malloc(i); |
73 ASSERT_GE(MallocExtension::instance()->GetAllocatedSize(p), | 65 ASSERT_GE(MallocExtension::instance()->GetAllocatedSize(p), |
74 MallocExtension::instance()->GetEstimatedAllocatedSize(i)); | 66 MallocExtension::instance()->GetEstimatedAllocatedSize(i)); |
75 free(p); | 67 free(p); |
76 } | 68 } |
77 | 69 |
78 // Check the c-shim version too. | 70 // Check the c-shim version too. |
79 ASSERT_EQ(MallocExtension_kOwned, MallocExtension_GetOwnership(a)); | |
80 ASSERT_EQ(MallocExtension_kNotOwned, | |
81 MallocExtension_GetOwnership(&cxx_bytes_used)); | |
82 ASSERT_EQ(MallocExtension_kNotOwned, MallocExtension_GetOwnership(NULL)); | |
83 ASSERT_GE(MallocExtension_GetAllocatedSize(a), 1000); | 71 ASSERT_GE(MallocExtension_GetAllocatedSize(a), 1000); |
84 ASSERT_LE(MallocExtension_GetAllocatedSize(a), 5000); | 72 ASSERT_LE(MallocExtension_GetAllocatedSize(a), 5000); |
85 ASSERT_GE(MallocExtension_GetEstimatedAllocatedSize(1000), 1000); | 73 ASSERT_GE(MallocExtension_GetEstimatedAllocatedSize(1000), 1000); |
86 | 74 |
87 free(a); | 75 free(a); |
88 | 76 |
89 // Verify that the .cc file and .h file have the same enum values. | |
90 ASSERT_EQ(static_cast<int>(MallocExtension::kUnknownOwnership), | |
91 static_cast<int>(MallocExtension_kUnknownOwnership)); | |
92 ASSERT_EQ(static_cast<int>(MallocExtension::kOwned), | |
93 static_cast<int>(MallocExtension_kOwned)); | |
94 ASSERT_EQ(static_cast<int>(MallocExtension::kNotOwned), | |
95 static_cast<int>(MallocExtension_kNotOwned)); | |
96 | |
97 printf("DONE\n"); | 77 printf("DONE\n"); |
98 return 0; | 78 return 0; |
99 } | 79 } |
OLD | NEW |