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 | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 12 | #include <stdlib.h> |
| 13 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 14 | namespace android { |
| 15 | |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame^] | 16 | AST::AST(Coordinator *coordinator, const std::string &path) |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 17 | : mCoordinator(coordinator), |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame^] | 18 | mPath(path), |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 19 | mScanner(NULL), |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 20 | mRootScope(new Scope) { |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 21 | enterScope(mRootScope); |
| 22 | } |
| 23 | |
| 24 | AST::~AST() { |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 25 | delete mRootScope; |
| 26 | mRootScope = NULL; |
| 27 | |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 28 | CHECK(mScanner == NULL); |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 29 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 30 | // Ownership of "coordinator" was NOT transferred. |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 31 | } |
| 32 | |
| 33 | void *AST::scanner() { |
| 34 | return mScanner; |
| 35 | } |
| 36 | |
| 37 | void AST::setScanner(void *scanner) { |
| 38 | mScanner = scanner; |
| 39 | } |
| 40 | |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame^] | 41 | const std::string &AST::getFilename() const { |
| 42 | return mPath; |
| 43 | } |
| 44 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 45 | bool AST::setPackage(const char *package) { |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 46 | mPackage.setTo(package); |
| 47 | CHECK(mPackage.isValid()); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 48 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 49 | if (mPackage.package().empty() |
| 50 | || mPackage.version().empty() |
| 51 | || !mPackage.name().empty()) { |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 52 | return false; |
| 53 | } |
| 54 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 55 | return true; |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 56 | } |
| 57 | |
Andreas Huber | a2723d2 | 2016-07-29 15:36:07 -0700 | [diff] [blame] | 58 | FQName AST::package() const { |
| 59 | return mPackage; |
| 60 | } |
| 61 | |
| 62 | bool AST::isInterface(std::string *ifaceName) const { |
| 63 | return mRootScope->containsSingleInterface(ifaceName); |
| 64 | } |
| 65 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 66 | bool AST::addImport(const char *import) { |
| 67 | FQName fqName(import); |
| 68 | CHECK(fqName.isValid()); |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 69 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 70 | fqName.applyDefaults(mPackage.package(), mPackage.version()); |
| 71 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 72 | // LOG(INFO) << "importing " << fqName.string(); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 73 | |
| 74 | if (fqName.name().empty()) { |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 75 | std::vector<FQName> packageInterfaces; |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 76 | |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 77 | status_t err = |
Iliyan Malchev | 5bb1402 | 2016-08-09 15:04:39 -0700 | [diff] [blame] | 78 | mCoordinator->appendPackageInterfacesToSet(fqName, |
| 79 | &packageInterfaces); |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 80 | |
| 81 | if (err != OK) { |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 82 | return false; |
| 83 | } |
| 84 | |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 85 | for (const auto &subFQName : packageInterfaces) { |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 86 | if (mCoordinator->parse(subFQName) == NULL) { |
| 87 | return false; |
| 88 | } |
| 89 | } |
| 90 | |
| 91 | return true; |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 92 | } |
| 93 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 94 | AST *importAST = mCoordinator->parse(fqName); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 95 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 96 | if (importAST == NULL) { |
| 97 | return false; |
| 98 | } |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 99 | |
| 100 | return true; |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 101 | } |
| 102 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 103 | void AST::enterScope(Scope *container) { |
| 104 | mScopePath.push_back(container); |
| 105 | } |
| 106 | |
| 107 | void AST::leaveScope() { |
| 108 | mScopePath.pop(); |
| 109 | } |
| 110 | |
| 111 | Scope *AST::scope() { |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 112 | CHECK(!mScopePath.empty()); |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 113 | return mScopePath.top(); |
| 114 | } |
| 115 | |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame^] | 116 | bool AST::addScopedType( |
| 117 | const char *localName, NamedType *type, std::string *errorMsg) { |
Andreas Huber | c7dfef3 | 2016-08-16 10:57:14 -0700 | [diff] [blame] | 118 | std::string anonName; |
| 119 | |
| 120 | if (localName == nullptr) { |
| 121 | // Anonymous type declaration. |
| 122 | anonName = scope()->pickUniqueAnonymousName(); |
| 123 | localName = anonName.c_str(); |
| 124 | } |
| 125 | |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 126 | // LOG(INFO) << "adding scoped type '" << localName << "'"; |
| 127 | |
Andreas Huber | 0d0f9a2 | 2016-08-17 10:26:11 -0700 | [diff] [blame^] | 128 | bool success = scope()->addType(localName, type, errorMsg); |
Andreas Huber | 5a54544 | 2016-08-03 10:44:56 -0700 | [diff] [blame] | 129 | if (!success) { |
| 130 | return false; |
| 131 | } |
| 132 | |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 133 | std::string path; |
| 134 | for (size_t i = 1; i < mScopePath.size(); ++i) { |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 135 | path.append(mScopePath[i]->localName()); |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 136 | path.append("."); |
| 137 | } |
| 138 | path.append(localName); |
| 139 | |
| 140 | type->setLocalName(localName); |
| 141 | |
| 142 | FQName fqName(mPackage.package(), mPackage.version(), path); |
| 143 | type->setFullName(fqName); |
| 144 | |
Andreas Huber | 5a54544 | 2016-08-03 10:44:56 -0700 | [diff] [blame] | 145 | return true; |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 146 | } |
| 147 | |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame] | 148 | Type *AST::lookupType(const char *name) { |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 149 | FQName fqName(name); |
| 150 | CHECK(fqName.isValid()); |
| 151 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 152 | if (fqName.name().empty()) { |
| 153 | // Given a package and version??? |
| 154 | return NULL; |
| 155 | } |
| 156 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 157 | if (fqName.package().empty() && fqName.version().empty()) { |
| 158 | // This is just a plain identifier, resolve locally first if possible. |
| 159 | |
| 160 | for (size_t i = mScopePath.size(); i-- > 0;) { |
| 161 | Type *type = mScopePath[i]->lookupType(name); |
| 162 | |
| 163 | if (type != NULL) { |
Andreas Huber | 8d3ac0c | 2016-08-04 14:49:23 -0700 | [diff] [blame] | 164 | // Resolve typeDefs to the target type. |
| 165 | while (type->isTypeDef()) { |
| 166 | type = static_cast<TypeDef *>(type)->referencedType(); |
| 167 | } |
| 168 | |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame] | 169 | return type->ref(); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 170 | } |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 171 | } |
| 172 | } |
| 173 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 174 | fqName.applyDefaults(mPackage.package(), mPackage.version()); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 175 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 176 | // LOG(INFO) << "lookupType now looking for " << fqName.string(); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 177 | |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame] | 178 | Type *resultType = mCoordinator->lookupType(fqName); |
Andreas Huber | 737080b | 2016-08-02 15:38:04 -0700 | [diff] [blame] | 179 | |
| 180 | if (resultType) { |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame] | 181 | if (!resultType->isInterface()) { |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 182 | // Non-interface types are declared in the associated types header. |
| 183 | FQName typesName(fqName.package(), fqName.version(), "types"); |
| 184 | mImportedNames.insert(typesName); |
| 185 | } else { |
Andreas Huber | bfd7621 | 2016-08-09 11:12:16 -0700 | [diff] [blame] | 186 | // Do _not_ use fqName, i.e. the name we used to look up the type, |
| 187 | // but instead use the name of the interface we found. |
| 188 | // This is necessary because if fqName pointed to a typedef which |
| 189 | // in turn referenced the found interface we'd mistakenly use the |
| 190 | // name of the typedef instead of the proper name of the interface. |
| 191 | |
| 192 | mImportedNames.insert( |
| 193 | static_cast<Interface *>(resultType)->fqName()); |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 194 | } |
Andreas Huber | 737080b | 2016-08-02 15:38:04 -0700 | [diff] [blame] | 195 | } |
| 196 | |
| 197 | return resultType; |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 198 | } |
| 199 | |
| 200 | Type *AST::lookupTypeInternal(const std::string &namePath) const { |
| 201 | Scope *scope = mRootScope; |
| 202 | |
| 203 | size_t startPos = 0; |
| 204 | for (;;) { |
| 205 | size_t dotPos = namePath.find('.', startPos); |
| 206 | |
| 207 | std::string component; |
| 208 | if (dotPos == std::string::npos) { |
| 209 | component = namePath.substr(startPos); |
| 210 | } else { |
| 211 | component = namePath.substr(startPos, dotPos - startPos); |
| 212 | } |
| 213 | |
| 214 | Type *type = scope->lookupType(component.c_str()); |
| 215 | |
| 216 | if (type == NULL) { |
| 217 | return NULL; |
| 218 | } |
| 219 | |
| 220 | if (dotPos == std::string::npos) { |
Andreas Huber | 8d3ac0c | 2016-08-04 14:49:23 -0700 | [diff] [blame] | 221 | // Resolve typeDefs to the target type. |
| 222 | while (type->isTypeDef()) { |
| 223 | type = static_cast<TypeDef *>(type)->referencedType(); |
| 224 | } |
| 225 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 226 | return type; |
| 227 | } |
| 228 | |
| 229 | if (!type->isScope()) { |
| 230 | return NULL; |
| 231 | } |
| 232 | |
| 233 | scope = static_cast<Scope *>(type); |
| 234 | startPos = dotPos + 1; |
| 235 | } |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 236 | } |
| 237 | |
Iliyan Malchev | 5bb1402 | 2016-08-09 15:04:39 -0700 | [diff] [blame] | 238 | void AST::getImportedPackages(std::set<FQName> *importSet) const { |
Andreas Huber | d2943e1 | 2016-08-05 11:59:31 -0700 | [diff] [blame] | 239 | for (const auto &fqName : mImportedNames) { |
| 240 | FQName packageName(fqName.package(), fqName.version(), ""); |
| 241 | |
| 242 | if (packageName == mPackage) { |
| 243 | // We only care about external imports, not our own package. |
| 244 | continue; |
| 245 | } |
| 246 | |
| 247 | importSet->insert(packageName); |
| 248 | } |
| 249 | } |
| 250 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 251 | } // namespace android; |