Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 1 | //===- CXString.cpp - Routines for manipulating CXStrings -----------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file defines routines for manipulating CXStrings. It should be the |
| 11 | // only file that has internal knowledge of the encoding of the data in |
| 12 | // CXStrings. |
| 13 | // |
| 14 | //===----------------------------------------------------------------------===// |
| 15 | |
| 16 | #include "CXString.h" |
Ted Kremenek | 7df92ae | 2010-11-17 23:24:11 +0000 | [diff] [blame] | 17 | #include "CXTranslationUnit.h" |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 18 | #include "clang-c/Index.h" |
Chandler Carruth | cc0694c | 2012-12-04 09:25:21 +0000 | [diff] [blame] | 19 | #include "clang/Frontend/ASTUnit.h" |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 20 | #include "llvm/Support/ErrorHandling.h" |
| 21 | |
| 22 | using namespace clang; |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 23 | |
Dmitri Gribenko | 74f3e1e | 2013-02-01 16:36:31 +0000 | [diff] [blame] | 24 | /// Describes the kind of underlying data in CXString. |
| 25 | enum CXStringFlag { |
| 26 | /// CXString contains a 'const char *' that it doesn't own. |
| 27 | CXS_Unmanaged, |
| 28 | |
| 29 | /// CXString contains a 'const char *' that it allocated with malloc(). |
| 30 | CXS_Malloc, |
| 31 | |
| 32 | /// CXString contains a CXStringBuf that needs to be returned to the |
| 33 | /// CXStringPool. |
| 34 | CXS_StringBuf |
| 35 | }; |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 36 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 37 | namespace clang { |
| 38 | namespace cxstring { |
| 39 | |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 40 | //===----------------------------------------------------------------------===// |
| 41 | // Basic generation of CXStrings. |
| 42 | //===----------------------------------------------------------------------===// |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 43 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 44 | CXString createEmpty() { |
Dmitri Gribenko | 36a6dd0 | 2013-02-01 14:21:22 +0000 | [diff] [blame] | 45 | CXString Str; |
| 46 | Str.data = ""; |
| 47 | Str.private_flags = CXS_Unmanaged; |
| 48 | return Str; |
| 49 | } |
| 50 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 51 | CXString createNull() { |
Dmitri Gribenko | f98dfba | 2013-02-01 14:13:32 +0000 | [diff] [blame] | 52 | CXString Str; |
Craig Topper | 69186e7 | 2014-06-08 08:38:04 +0000 | [diff] [blame] | 53 | Str.data = nullptr; |
Dmitri Gribenko | f98dfba | 2013-02-01 14:13:32 +0000 | [diff] [blame] | 54 | Str.private_flags = CXS_Unmanaged; |
| 55 | return Str; |
| 56 | } |
| 57 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 58 | CXString createRef(const char *String) { |
Dmitri Gribenko | 3c66b0b | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 59 | if (String && String[0] == '\0') |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 60 | return createEmpty(); |
Dmitri Gribenko | 3c66b0b | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 61 | |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 62 | CXString Str; |
Dmitri Gribenko | 3c66b0b | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 63 | Str.data = String; |
| 64 | Str.private_flags = CXS_Unmanaged; |
| 65 | return Str; |
| 66 | } |
| 67 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 68 | CXString createDup(const char *String) { |
Dmitri Gribenko | 3c66b0b | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 69 | if (!String) |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 70 | return createNull(); |
Dmitri Gribenko | 3c66b0b | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 71 | |
| 72 | if (String[0] == '\0') |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 73 | return createEmpty(); |
Dmitri Gribenko | 3c66b0b | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 74 | |
| 75 | CXString Str; |
| 76 | Str.data = strdup(String); |
| 77 | Str.private_flags = CXS_Malloc; |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 78 | return Str; |
| 79 | } |
| 80 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 81 | CXString createRef(StringRef String) { |
Dmitri Gribenko | 2f23e9c | 2013-02-02 02:19:29 +0000 | [diff] [blame] | 82 | // If the string is not nul-terminated, we have to make a copy. |
Alp Toker | 541d507 | 2014-06-07 23:30:53 +0000 | [diff] [blame] | 83 | |
| 84 | // FIXME: This is doing a one past end read, and should be removed! For memory |
| 85 | // we don't manage, the API string can become unterminated at any time outside |
| 86 | // our control. |
| 87 | |
Dmitri Gribenko | 2f23e9c | 2013-02-02 02:19:29 +0000 | [diff] [blame] | 88 | if (!String.empty() && String.data()[String.size()] != 0) |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 89 | return createDup(String); |
Dmitri Gribenko | 2f23e9c | 2013-02-02 02:19:29 +0000 | [diff] [blame] | 90 | |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 91 | CXString Result; |
Dmitri Gribenko | 2f23e9c | 2013-02-02 02:19:29 +0000 | [diff] [blame] | 92 | Result.data = String.data(); |
| 93 | Result.private_flags = (unsigned) CXS_Unmanaged; |
| 94 | return Result; |
| 95 | } |
| 96 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 97 | CXString createDup(StringRef String) { |
Dmitri Gribenko | 2f23e9c | 2013-02-02 02:19:29 +0000 | [diff] [blame] | 98 | CXString Result; |
| 99 | char *Spelling = static_cast<char *>(malloc(String.size() + 1)); |
| 100 | memmove(Spelling, String.data(), String.size()); |
| 101 | Spelling[String.size()] = 0; |
| 102 | Result.data = Spelling; |
| 103 | Result.private_flags = (unsigned) CXS_Malloc; |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 104 | return Result; |
| 105 | } |
| 106 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 107 | CXString createCXString(CXStringBuf *buf) { |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 108 | CXString Str; |
| 109 | Str.data = buf; |
| 110 | Str.private_flags = (unsigned) CXS_StringBuf; |
| 111 | return Str; |
| 112 | } |
| 113 | |
Saleem Abdulrasool | 5d92eae | 2015-11-12 03:57:16 +0000 | [diff] [blame] | 114 | CXStringSet *createSet(const std::vector<std::string> &Strings) { |
| 115 | CXStringSet *Set = new CXStringSet; |
| 116 | Set->Count = Strings.size(); |
| 117 | Set->Strings = new CXString[Set->Count]; |
| 118 | for (unsigned SI = 0, SE = Set->Count; SI < SE; ++SI) |
| 119 | Set->Strings[SI] = createDup(Strings[SI]); |
| 120 | return Set; |
| 121 | } |
| 122 | |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 123 | |
| 124 | //===----------------------------------------------------------------------===// |
| 125 | // String pools. |
| 126 | //===----------------------------------------------------------------------===// |
| 127 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 128 | CXStringPool::~CXStringPool() { |
Dmitri Gribenko | b95b3f1 | 2013-01-26 22:44:19 +0000 | [diff] [blame] | 129 | for (std::vector<CXStringBuf *>::iterator I = Pool.begin(), E = Pool.end(); |
| 130 | I != E; ++I) { |
| 131 | delete *I; |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 132 | } |
| 133 | } |
| 134 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 135 | CXStringBuf *CXStringPool::getCXStringBuf(CXTranslationUnit TU) { |
Dmitri Gribenko | b95b3f1 | 2013-01-26 22:44:19 +0000 | [diff] [blame] | 136 | if (Pool.empty()) |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 137 | return new CXStringBuf(TU); |
Dmitri Gribenko | b95b3f1 | 2013-01-26 22:44:19 +0000 | [diff] [blame] | 138 | |
| 139 | CXStringBuf *Buf = Pool.back(); |
| 140 | Buf->Data.clear(); |
| 141 | Pool.pop_back(); |
| 142 | return Buf; |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 143 | } |
| 144 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 145 | CXStringBuf *getCXStringBuf(CXTranslationUnit TU) { |
Dmitri Gribenko | b95b3f1 | 2013-01-26 22:44:19 +0000 | [diff] [blame] | 146 | return TU->StringPool->getCXStringBuf(TU); |
| 147 | } |
| 148 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 149 | void CXStringBuf::dispose() { |
Dmitri Gribenko | b95b3f1 | 2013-01-26 22:44:19 +0000 | [diff] [blame] | 150 | TU->StringPool->Pool.push_back(this); |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 151 | } |
| 152 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 153 | bool isManagedByPool(CXString str) { |
Ted Kremenek | 5b8ad40 | 2011-08-17 22:19:53 +0000 | [diff] [blame] | 154 | return ((CXStringFlag) str.private_flags) == CXS_StringBuf; |
| 155 | } |
| 156 | |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 157 | } // end namespace cxstring |
| 158 | } // end namespace clang |
| 159 | |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 160 | //===----------------------------------------------------------------------===// |
| 161 | // libClang public APIs. |
| 162 | //===----------------------------------------------------------------------===// |
| 163 | |
| 164 | extern "C" { |
| 165 | const char *clang_getCString(CXString string) { |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 166 | if (string.private_flags == (unsigned) CXS_StringBuf) { |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 167 | return static_cast<const cxstring::CXStringBuf *>(string.data)->Data.data(); |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 168 | } |
Dmitri Gribenko | 037f9d1 | 2013-01-11 23:13:36 +0000 | [diff] [blame] | 169 | return static_cast<const char *>(string.data); |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 170 | } |
| 171 | |
| 172 | void clang_disposeString(CXString string) { |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 173 | switch ((CXStringFlag) string.private_flags) { |
| 174 | case CXS_Unmanaged: |
| 175 | break; |
| 176 | case CXS_Malloc: |
| 177 | if (string.data) |
Dmitri Gribenko | ec3a16f | 2013-01-11 23:08:18 +0000 | [diff] [blame] | 178 | free(const_cast<void *>(string.data)); |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 179 | break; |
| 180 | case CXS_StringBuf: |
Dmitri Gribenko | 18aa3bb | 2013-02-03 13:54:26 +0000 | [diff] [blame] | 181 | static_cast<cxstring::CXStringBuf *>( |
Dmitri Gribenko | b95b3f1 | 2013-01-26 22:44:19 +0000 | [diff] [blame] | 182 | const_cast<void *>(string.data))->dispose(); |
Ted Kremenek | 9155428 | 2010-11-16 08:15:36 +0000 | [diff] [blame] | 183 | break; |
| 184 | } |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 185 | } |
Saleem Abdulrasool | 5d92eae | 2015-11-12 03:57:16 +0000 | [diff] [blame] | 186 | |
| 187 | void clang_disposeStringSet(CXStringSet *set) { |
Reid Kleckner | 356be39 | 2015-11-18 01:06:39 +0000 | [diff] [blame] | 188 | for (unsigned SI = 0, SE = set->Count; SI < SE; ++SI) |
| 189 | clang_disposeString(set->Strings[SI]); |
Saleem Abdulrasool | 5d92eae | 2015-11-12 03:57:16 +0000 | [diff] [blame] | 190 | delete[] set->Strings; |
| 191 | delete set; |
| 192 | } |
| 193 | |
Ted Kremenek | 64dddfb | 2010-11-16 02:03:55 +0000 | [diff] [blame] | 194 | } // end: extern "C" |
| 195 | |