Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 1 | #include "AST.h" |
| 2 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 3 | #include "Coordinator.h" |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 4 | #include "Formatter.h" |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 5 | #include "FQName.h" |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 6 | #include "HandleType.h" |
Andreas Huber | bfd7621 | 2016-08-09 11:12:16 -0700 | [diff] [blame] | 7 | #include "Interface.h" |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 8 | #include "Scope.h" |
Andreas Huber | 8d3ac0c | 2016-08-04 14:49:23 -0700 | [diff] [blame] | 9 | #include "TypeDef.h" |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 10 | |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 11 | #include <android-base/logging.h> |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 12 | #include <iostream> |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 13 | #include <stdlib.h> |
| 14 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 15 | namespace android { |
| 16 | |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame] | 17 | AST::AST(Coordinator *coordinator, const std::string &path) |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 18 | : mCoordinator(coordinator), |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame] | 19 | mPath(path), |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 20 | mScanner(NULL), |
Andreas Huber | 9ed827c | 2016-08-22 12:31:13 -0700 | [diff] [blame] | 21 | mRootScope(new Scope("" /* localName */)) { |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 22 | enterScope(mRootScope); |
| 23 | } |
| 24 | |
| 25 | AST::~AST() { |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 26 | delete mRootScope; |
| 27 | mRootScope = NULL; |
| 28 | |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 29 | CHECK(mScanner == NULL); |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 30 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 31 | // Ownership of "coordinator" was NOT transferred. |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 32 | } |
| 33 | |
| 34 | void *AST::scanner() { |
| 35 | return mScanner; |
| 36 | } |
| 37 | |
| 38 | void AST::setScanner(void *scanner) { |
| 39 | mScanner = scanner; |
| 40 | } |
| 41 | |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame] | 42 | const std::string &AST::getFilename() const { |
| 43 | return mPath; |
| 44 | } |
| 45 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 46 | bool AST::setPackage(const char *package) { |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 47 | mPackage.setTo(package); |
| 48 | CHECK(mPackage.isValid()); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 49 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 50 | if (mPackage.package().empty() |
| 51 | || mPackage.version().empty() |
| 52 | || !mPackage.name().empty()) { |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 53 | return false; |
| 54 | } |
| 55 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 56 | return true; |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 57 | } |
| 58 | |
Andreas Huber | a2723d2 | 2016-07-29 15:36:07 -0700 | [diff] [blame] | 59 | FQName AST::package() const { |
| 60 | return mPackage; |
| 61 | } |
| 62 | |
| 63 | bool AST::isInterface(std::string *ifaceName) const { |
| 64 | return mRootScope->containsSingleInterface(ifaceName); |
| 65 | } |
| 66 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 67 | bool AST::addImport(const char *import) { |
| 68 | FQName fqName(import); |
| 69 | CHECK(fqName.isValid()); |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 70 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 71 | fqName.applyDefaults(mPackage.package(), mPackage.version()); |
| 72 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 73 | // LOG(INFO) << "importing " << fqName.string(); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 74 | |
| 75 | if (fqName.name().empty()) { |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 76 | std::vector<FQName> packageInterfaces; |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 77 | |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 78 | status_t err = |
Iliyan Malchev | 5bb1402 | 2016-08-09 15:04:39 -0700 | [diff] [blame] | 79 | mCoordinator->appendPackageInterfacesToSet(fqName, |
| 80 | &packageInterfaces); |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 81 | |
| 82 | if (err != OK) { |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 83 | return false; |
| 84 | } |
| 85 | |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 86 | for (const auto &subFQName : packageInterfaces) { |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 87 | AST *ast = mCoordinator->parse(subFQName, &mImportedASTs); |
| 88 | if (ast == NULL) { |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 89 | return false; |
| 90 | } |
| 91 | } |
| 92 | |
| 93 | return true; |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 94 | } |
| 95 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 96 | AST *importAST = mCoordinator->parse(fqName, &mImportedASTs); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 97 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 98 | if (importAST == NULL) { |
| 99 | return false; |
| 100 | } |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 101 | |
| 102 | return true; |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 103 | } |
| 104 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 105 | void AST::addImportedAST(AST *ast) { |
| 106 | mImportedASTs.insert(ast); |
| 107 | } |
| 108 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 109 | void AST::enterScope(Scope *container) { |
| 110 | mScopePath.push_back(container); |
| 111 | } |
| 112 | |
| 113 | void AST::leaveScope() { |
| 114 | mScopePath.pop(); |
| 115 | } |
| 116 | |
| 117 | Scope *AST::scope() { |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 118 | CHECK(!mScopePath.empty()); |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 119 | return mScopePath.top(); |
| 120 | } |
| 121 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 122 | bool AST::addTypeDef( |
| 123 | const char *localName, Type *type, std::string *errorMsg) { |
| 124 | // The reason we wrap the given type in a TypeDef is simply to suppress |
| 125 | // emitting any type definitions later on, since this is just an alias |
| 126 | // to a type defined elsewhere. |
| 127 | return addScopedTypeInternal( |
| 128 | localName, new TypeDef(type), errorMsg, true /* isTypeDef */); |
| 129 | } |
| 130 | |
Andreas Huber | 9ed827c | 2016-08-22 12:31:13 -0700 | [diff] [blame] | 131 | bool AST::addScopedType(NamedType *type, std::string *errorMsg) { |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 132 | return addScopedTypeInternal( |
Andreas Huber | 9ed827c | 2016-08-22 12:31:13 -0700 | [diff] [blame] | 133 | type->localName().c_str(), type, errorMsg, false /* isTypeDef */); |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 134 | } |
| 135 | |
| 136 | bool AST::addScopedTypeInternal( |
| 137 | const char *localName, |
| 138 | Type *type, |
| 139 | std::string *errorMsg, |
| 140 | bool isTypeDef) { |
| 141 | if (!isTypeDef) { |
| 142 | // Resolve typeDefs to the target type. |
| 143 | while (type->isTypeDef()) { |
| 144 | type = static_cast<TypeDef *>(type)->referencedType(); |
| 145 | } |
| 146 | } |
| 147 | |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 148 | // LOG(INFO) << "adding scoped type '" << localName << "'"; |
| 149 | |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame] | 150 | bool success = scope()->addType(localName, type, errorMsg); |
Andreas Huber | 5a54544 | 2016-08-03 10:44:56 -0700 | [diff] [blame] | 151 | if (!success) { |
| 152 | return false; |
| 153 | } |
| 154 | |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 155 | std::string path; |
| 156 | for (size_t i = 1; i < mScopePath.size(); ++i) { |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 157 | path.append(mScopePath[i]->localName()); |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 158 | path.append("."); |
| 159 | } |
| 160 | path.append(localName); |
| 161 | |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 162 | FQName fqName(mPackage.package(), mPackage.version(), path); |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 163 | |
| 164 | if (!isTypeDef) { |
| 165 | CHECK(type->isNamedType()); |
| 166 | |
| 167 | NamedType *namedType = static_cast<NamedType *>(type); |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 168 | namedType->setFullName(fqName); |
| 169 | } |
| 170 | |
| 171 | mDefinedTypesByFullName.add(fqName, type); |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 172 | |
Andreas Huber | 5a54544 | 2016-08-03 10:44:56 -0700 | [diff] [blame] | 173 | return true; |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 174 | } |
| 175 | |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame] | 176 | Type *AST::lookupType(const char *name) { |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 177 | FQName fqName(name); |
| 178 | CHECK(fqName.isValid()); |
| 179 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 180 | if (fqName.name().empty()) { |
| 181 | // Given a package and version??? |
| 182 | return NULL; |
| 183 | } |
| 184 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 185 | if (fqName.package().empty() && fqName.version().empty()) { |
| 186 | // This is just a plain identifier, resolve locally first if possible. |
| 187 | |
| 188 | for (size_t i = mScopePath.size(); i-- > 0;) { |
| 189 | Type *type = mScopePath[i]->lookupType(name); |
| 190 | |
| 191 | if (type != NULL) { |
Andreas Huber | 8d3ac0c | 2016-08-04 14:49:23 -0700 | [diff] [blame] | 192 | // Resolve typeDefs to the target type. |
| 193 | while (type->isTypeDef()) { |
| 194 | type = static_cast<TypeDef *>(type)->referencedType(); |
| 195 | } |
| 196 | |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame] | 197 | return type->ref(); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 198 | } |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 199 | } |
| 200 | } |
| 201 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 202 | Type *resolvedType = nullptr; |
| 203 | FQName resolvedName; |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 204 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 205 | for (const auto &importedAST : mImportedASTs) { |
| 206 | FQName matchingName; |
| 207 | Type *match = importedAST->findDefinedType(fqName, &matchingName); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 208 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 209 | if (match != nullptr) { |
| 210 | if (resolvedType != nullptr) { |
| 211 | std::cerr << "ERROR: Unable to resolve type name '" |
| 212 | << fqName.string() |
| 213 | << "', multiple matches found:\n"; |
Andreas Huber | 737080b | 2016-08-02 15:38:04 -0700 | [diff] [blame] | 214 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 215 | std::cerr << " " << resolvedName.string() << "\n"; |
| 216 | std::cerr << " " << matchingName.string() << "\n"; |
| 217 | |
| 218 | return NULL; |
| 219 | } |
| 220 | |
| 221 | resolvedType = match; |
| 222 | resolvedName = matchingName; |
| 223 | |
| 224 | // Keep going even after finding a match. |
| 225 | } |
| 226 | } |
| 227 | |
| 228 | if (resolvedType) { |
| 229 | #if 0 |
| 230 | LOG(INFO) << "found '" |
| 231 | << resolvedName.string() |
| 232 | << "' after looking for '" |
| 233 | << fqName.string() |
| 234 | << "'."; |
| 235 | #endif |
| 236 | |
| 237 | // Resolve typeDefs to the target type. |
| 238 | while (resolvedType->isTypeDef()) { |
| 239 | resolvedType = |
| 240 | static_cast<TypeDef *>(resolvedType)->referencedType(); |
| 241 | } |
| 242 | |
| 243 | if (!resolvedType->isInterface()) { |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 244 | // Non-interface types are declared in the associated types header. |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 245 | FQName typesName( |
| 246 | resolvedName.package(), resolvedName.version(), "types"); |
| 247 | |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 248 | mImportedNames.insert(typesName); |
| 249 | } else { |
Andreas Huber | bfd7621 | 2016-08-09 11:12:16 -0700 | [diff] [blame] | 250 | // Do _not_ use fqName, i.e. the name we used to look up the type, |
| 251 | // but instead use the name of the interface we found. |
| 252 | // This is necessary because if fqName pointed to a typedef which |
| 253 | // in turn referenced the found interface we'd mistakenly use the |
| 254 | // name of the typedef instead of the proper name of the interface. |
| 255 | |
| 256 | mImportedNames.insert( |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 257 | static_cast<Interface *>(resolvedType)->fqName()); |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 258 | } |
Andreas Huber | 737080b | 2016-08-02 15:38:04 -0700 | [diff] [blame] | 259 | } |
| 260 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 261 | return resolvedType->ref(); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 262 | } |
| 263 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 264 | Type *AST::findDefinedType(const FQName &fqName, FQName *matchingName) const { |
| 265 | for (size_t i = 0; i < mDefinedTypesByFullName.size(); ++i) { |
| 266 | const FQName &key = mDefinedTypesByFullName.keyAt(i); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 267 | |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 268 | if (key.endsWith(fqName)) { |
| 269 | *matchingName = key; |
| 270 | return mDefinedTypesByFullName.valueAt(i); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 271 | } |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 272 | } |
Andreas Huber | 39fa718 | 2016-08-19 14:27:33 -0700 | [diff] [blame] | 273 | |
| 274 | return nullptr; |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 275 | } |
| 276 | |
Iliyan Malchev | 5bb1402 | 2016-08-09 15:04:39 -0700 | [diff] [blame] | 277 | void AST::getImportedPackages(std::set<FQName> *importSet) const { |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 278 | for (const auto &fqName : mImportedNames) { |
| 279 | FQName packageName(fqName.package(), fqName.version(), ""); |
| 280 | |
| 281 | if (packageName == mPackage) { |
| 282 | // We only care about external imports, not our own package. |
| 283 | continue; |
| 284 | } |
| 285 | |
| 286 | importSet->insert(packageName); |
| 287 | } |
| 288 | } |
| 289 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 290 | } // namespace android; |