| 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 |