Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 1 | // Copyright 2015 the V8 project authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "src/profiler/strings-storage.h" |
| 6 | |
| 7 | #include "src/base/smart-pointers.h" |
| 8 | #include "src/objects-inl.h" |
| 9 | |
| 10 | namespace v8 { |
| 11 | namespace internal { |
| 12 | |
| 13 | |
| 14 | bool StringsStorage::StringsMatch(void* key1, void* key2) { |
| 15 | return strcmp(reinterpret_cast<char*>(key1), reinterpret_cast<char*>(key2)) == |
| 16 | 0; |
| 17 | } |
| 18 | |
| 19 | |
| 20 | StringsStorage::StringsStorage(Heap* heap) |
| 21 | : hash_seed_(heap->HashSeed()), names_(StringsMatch) {} |
| 22 | |
| 23 | |
| 24 | StringsStorage::~StringsStorage() { |
Ben Murdoch | 61f157c | 2016-09-16 13:49:30 +0100 | [diff] [blame] | 25 | for (base::HashMap::Entry* p = names_.Start(); p != NULL; |
| 26 | p = names_.Next(p)) { |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 27 | DeleteArray(reinterpret_cast<const char*>(p->value)); |
| 28 | } |
| 29 | } |
| 30 | |
| 31 | |
| 32 | const char* StringsStorage::GetCopy(const char* src) { |
| 33 | int len = static_cast<int>(strlen(src)); |
Ben Murdoch | 61f157c | 2016-09-16 13:49:30 +0100 | [diff] [blame] | 34 | base::HashMap::Entry* entry = GetEntry(src, len); |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 35 | if (entry->value == NULL) { |
| 36 | Vector<char> dst = Vector<char>::New(len + 1); |
| 37 | StrNCpy(dst, src, len); |
| 38 | dst[len] = '\0'; |
| 39 | entry->key = dst.start(); |
| 40 | entry->value = entry->key; |
| 41 | } |
| 42 | return reinterpret_cast<const char*>(entry->value); |
| 43 | } |
| 44 | |
| 45 | |
| 46 | const char* StringsStorage::GetFormatted(const char* format, ...) { |
| 47 | va_list args; |
| 48 | va_start(args, format); |
| 49 | const char* result = GetVFormatted(format, args); |
| 50 | va_end(args); |
| 51 | return result; |
| 52 | } |
| 53 | |
| 54 | |
| 55 | const char* StringsStorage::AddOrDisposeString(char* str, int len) { |
Ben Murdoch | 61f157c | 2016-09-16 13:49:30 +0100 | [diff] [blame] | 56 | base::HashMap::Entry* entry = GetEntry(str, len); |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 57 | if (entry->value == NULL) { |
| 58 | // New entry added. |
| 59 | entry->key = str; |
| 60 | entry->value = str; |
| 61 | } else { |
| 62 | DeleteArray(str); |
| 63 | } |
| 64 | return reinterpret_cast<const char*>(entry->value); |
| 65 | } |
| 66 | |
| 67 | |
| 68 | const char* StringsStorage::GetVFormatted(const char* format, va_list args) { |
| 69 | Vector<char> str = Vector<char>::New(1024); |
| 70 | int len = VSNPrintF(str, format, args); |
| 71 | if (len == -1) { |
| 72 | DeleteArray(str.start()); |
| 73 | return GetCopy(format); |
| 74 | } |
| 75 | return AddOrDisposeString(str.start(), len); |
| 76 | } |
| 77 | |
| 78 | |
| 79 | const char* StringsStorage::GetName(Name* name) { |
| 80 | if (name->IsString()) { |
| 81 | String* str = String::cast(name); |
| 82 | int length = Min(kMaxNameSize, str->length()); |
| 83 | int actual_length = 0; |
| 84 | base::SmartArrayPointer<char> data = str->ToCString( |
| 85 | DISALLOW_NULLS, ROBUST_STRING_TRAVERSAL, 0, length, &actual_length); |
| 86 | return AddOrDisposeString(data.Detach(), actual_length); |
| 87 | } else if (name->IsSymbol()) { |
| 88 | return "<symbol>"; |
| 89 | } |
| 90 | return ""; |
| 91 | } |
| 92 | |
| 93 | |
| 94 | const char* StringsStorage::GetName(int index) { |
| 95 | return GetFormatted("%d", index); |
| 96 | } |
| 97 | |
| 98 | |
| 99 | const char* StringsStorage::GetFunctionName(Name* name) { |
| 100 | return GetName(name); |
| 101 | } |
| 102 | |
| 103 | |
| 104 | const char* StringsStorage::GetFunctionName(const char* name) { |
| 105 | return GetCopy(name); |
| 106 | } |
| 107 | |
| 108 | |
| 109 | size_t StringsStorage::GetUsedMemorySize() const { |
| 110 | size_t size = sizeof(*this); |
Ben Murdoch | 61f157c | 2016-09-16 13:49:30 +0100 | [diff] [blame] | 111 | size += sizeof(base::HashMap::Entry) * names_.capacity(); |
| 112 | for (base::HashMap::Entry* p = names_.Start(); p != NULL; |
| 113 | p = names_.Next(p)) { |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 114 | size += strlen(reinterpret_cast<const char*>(p->value)) + 1; |
| 115 | } |
| 116 | return size; |
| 117 | } |
| 118 | |
Ben Murdoch | 61f157c | 2016-09-16 13:49:30 +0100 | [diff] [blame] | 119 | base::HashMap::Entry* StringsStorage::GetEntry(const char* str, int len) { |
Ben Murdoch | 4a90d5f | 2016-03-22 12:00:34 +0000 | [diff] [blame] | 120 | uint32_t hash = StringHasher::HashSequentialString(str, len, hash_seed_); |
| 121 | return names_.LookupOrInsert(const_cast<char*>(str), hash); |
| 122 | } |
| 123 | } // namespace internal |
| 124 | } // namespace v8 |