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 | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 7 | #include "Scope.h" |
| 8 | |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 9 | #include <android-base/logging.h> |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 10 | #include <stdlib.h> |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 11 | #include <sys/dir.h> |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 12 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 13 | namespace android { |
| 14 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 15 | AST::AST(Coordinator *coordinator) |
| 16 | : mCoordinator(coordinator), |
| 17 | mScanner(NULL), |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 18 | mRootScope(new Scope) { |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 19 | enterScope(mRootScope); |
| 20 | } |
| 21 | |
| 22 | AST::~AST() { |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 23 | delete mRootScope; |
| 24 | mRootScope = NULL; |
| 25 | |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 26 | CHECK(mScanner == NULL); |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 27 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 28 | // Ownership of "coordinator" was NOT transferred. |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 29 | } |
| 30 | |
| 31 | void *AST::scanner() { |
| 32 | return mScanner; |
| 33 | } |
| 34 | |
| 35 | void AST::setScanner(void *scanner) { |
| 36 | mScanner = scanner; |
| 37 | } |
| 38 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 39 | bool AST::setPackage(const char *package) { |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 40 | mPackage.setTo(package); |
| 41 | CHECK(mPackage.isValid()); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 42 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 43 | if (mPackage.package().empty() |
| 44 | || mPackage.version().empty() |
| 45 | || !mPackage.name().empty()) { |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 46 | return false; |
| 47 | } |
| 48 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 49 | return true; |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 50 | } |
| 51 | |
Andreas Huber | a2723d2 | 2016-07-29 15:36:07 -0700 | [diff] [blame] | 52 | FQName AST::package() const { |
| 53 | return mPackage; |
| 54 | } |
| 55 | |
| 56 | bool AST::isInterface(std::string *ifaceName) const { |
| 57 | return mRootScope->containsSingleInterface(ifaceName); |
| 58 | } |
| 59 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 60 | bool AST::addImport(const char *import) { |
| 61 | FQName fqName(import); |
| 62 | CHECK(fqName.isValid()); |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 63 | |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 64 | fqName.applyDefaults(mPackage.package(), mPackage.version()); |
| 65 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 66 | // LOG(INFO) << "importing " << fqName.string(); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 67 | |
| 68 | if (fqName.name().empty()) { |
Andreas Huber | dc98133 | 2016-07-29 15:46:54 -0700 | [diff] [blame] | 69 | const std::string packagePath = mCoordinator->getPackagePath(fqName); |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 70 | DIR *dir = opendir(packagePath.c_str()); |
| 71 | |
| 72 | if (dir == NULL) { |
| 73 | return false; |
| 74 | } |
| 75 | |
| 76 | // Enumerate all regular files ending in ".hal" in the package directory |
| 77 | // and attempt to import all of them. |
| 78 | |
| 79 | Vector<std::string> fileNames; |
| 80 | |
| 81 | struct dirent *ent; |
| 82 | while ((ent = readdir(dir)) != NULL) { |
| 83 | if (ent->d_type != DT_REG) { |
| 84 | continue; |
| 85 | } |
| 86 | |
Andreas Huber | f5dc73e | 2016-08-03 12:54:08 -0700 | [diff] [blame] | 87 | const auto suffix = ".hal"; |
| 88 | const auto suffix_len = std::strlen(suffix); |
| 89 | const auto d_namelen = strlen(ent->d_name); |
| 90 | |
| 91 | if (d_namelen < suffix_len |
| 92 | || strcmp(ent->d_name + d_namelen - suffix_len, suffix)) { |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 93 | continue; |
| 94 | } |
| 95 | |
Andreas Huber | f5dc73e | 2016-08-03 12:54:08 -0700 | [diff] [blame] | 96 | fileNames.push_back(std::string(ent->d_name, d_namelen - suffix_len)); |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 97 | } |
| 98 | |
| 99 | closedir(dir); |
| 100 | dir = NULL; |
| 101 | |
| 102 | for (size_t i = 0; i < fileNames.size(); ++i) { |
| 103 | FQName subFQName( |
| 104 | fqName.package() + fqName.version() + "::" + fileNames[i]); |
| 105 | |
| 106 | if (!subFQName.isValid()) { |
| 107 | LOG(WARNING) |
| 108 | << "Whole-package import encountered invalid filename '" |
| 109 | << fileNames[i] |
| 110 | << "' in package " |
| 111 | << fqName.package() |
| 112 | << fqName.version(); |
| 113 | |
| 114 | continue; |
| 115 | } |
| 116 | |
| 117 | if (mCoordinator->parse(subFQName) == NULL) { |
| 118 | return false; |
| 119 | } |
| 120 | } |
| 121 | |
| 122 | return true; |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 123 | } |
| 124 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 125 | AST *importAST = mCoordinator->parse(fqName); |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 126 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 127 | if (importAST == NULL) { |
| 128 | return false; |
| 129 | } |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 130 | |
| 131 | return true; |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 132 | } |
| 133 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 134 | void AST::enterScope(Scope *container) { |
| 135 | mScopePath.push_back(container); |
| 136 | } |
| 137 | |
| 138 | void AST::leaveScope() { |
| 139 | mScopePath.pop(); |
| 140 | } |
| 141 | |
| 142 | Scope *AST::scope() { |
Andreas Huber | eb1081f | 2016-07-28 13:13:24 -0700 | [diff] [blame] | 143 | CHECK(!mScopePath.empty()); |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 144 | return mScopePath.top(); |
| 145 | } |
| 146 | |
Andreas Huber | 5a54544 | 2016-08-03 10:44:56 -0700 | [diff] [blame] | 147 | bool AST::addScopedType(const char *localName, NamedType *type) { |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 148 | // LOG(INFO) << "adding scoped type '" << localName << "'"; |
| 149 | |
Andreas Huber | 5a54544 | 2016-08-03 10:44:56 -0700 | [diff] [blame] | 150 | bool success = scope()->addType(localName, type); |
| 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 | |
| 162 | type->setLocalName(localName); |
| 163 | |
| 164 | FQName fqName(mPackage.package(), mPackage.version(), path); |
| 165 | type->setFullName(fqName); |
| 166 | |
Andreas Huber | 5a54544 | 2016-08-03 10:44:56 -0700 | [diff] [blame] | 167 | return true; |
Andreas Huber | 31629bc | 2016-08-03 09:06:40 -0700 | [diff] [blame] | 168 | } |
| 169 | |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame^] | 170 | Type *AST::lookupType(const char *name) { |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 171 | FQName fqName(name); |
| 172 | CHECK(fqName.isValid()); |
| 173 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 174 | if (fqName.name().empty()) { |
| 175 | // Given a package and version??? |
| 176 | return NULL; |
| 177 | } |
| 178 | |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 179 | if (fqName.package().empty() && fqName.version().empty()) { |
| 180 | // This is just a plain identifier, resolve locally first if possible. |
| 181 | |
| 182 | for (size_t i = mScopePath.size(); i-- > 0;) { |
| 183 | Type *type = mScopePath[i]->lookupType(name); |
| 184 | |
| 185 | if (type != NULL) { |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame^] | 186 | return type->ref(); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 187 | } |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 188 | } |
| 189 | } |
| 190 | |
Andreas Huber | da51b8e | 2016-07-28 16:00:57 -0700 | [diff] [blame] | 191 | fqName.applyDefaults(mPackage.package(), mPackage.version()); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 192 | |
Andreas Huber | 68f2459 | 2016-07-29 14:53:48 -0700 | [diff] [blame] | 193 | // LOG(INFO) << "lookupType now looking for " << fqName.string(); |
Andreas Huber | 84f89de | 2016-07-28 15:39:51 -0700 | [diff] [blame] | 194 | |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame^] | 195 | Type *resultType = mCoordinator->lookupType(fqName); |
Andreas Huber | 737080b | 2016-08-02 15:38:04 -0700 | [diff] [blame] | 196 | |
| 197 | if (resultType) { |
Andreas Huber | fd4afab | 2016-08-03 13:02:57 -0700 | [diff] [blame^] | 198 | if (!resultType->isInterface()) { |
Andreas Huber | 0e00de4 | 2016-08-03 09:56:02 -0700 | [diff] [blame] | 199 | // Non-interface types are declared in the associated types header. |
| 200 | FQName typesName(fqName.package(), fqName.version(), "types"); |
| 201 | mImportedNames.insert(typesName); |
| 202 | } else { |
| 203 | mImportedNames.insert(fqName); |
| 204 | } |
Andreas Huber | 737080b | 2016-08-02 15:38:04 -0700 | [diff] [blame] | 205 | } |
| 206 | |
| 207 | return resultType; |
Andreas Huber | 5345ec2 | 2016-07-29 13:33:27 -0700 | [diff] [blame] | 208 | } |
| 209 | |
| 210 | Type *AST::lookupTypeInternal(const std::string &namePath) const { |
| 211 | Scope *scope = mRootScope; |
| 212 | |
| 213 | size_t startPos = 0; |
| 214 | for (;;) { |
| 215 | size_t dotPos = namePath.find('.', startPos); |
| 216 | |
| 217 | std::string component; |
| 218 | if (dotPos == std::string::npos) { |
| 219 | component = namePath.substr(startPos); |
| 220 | } else { |
| 221 | component = namePath.substr(startPos, dotPos - startPos); |
| 222 | } |
| 223 | |
| 224 | Type *type = scope->lookupType(component.c_str()); |
| 225 | |
| 226 | if (type == NULL) { |
| 227 | return NULL; |
| 228 | } |
| 229 | |
| 230 | if (dotPos == std::string::npos) { |
| 231 | return type; |
| 232 | } |
| 233 | |
| 234 | if (!type->isScope()) { |
| 235 | return NULL; |
| 236 | } |
| 237 | |
| 238 | scope = static_cast<Scope *>(type); |
| 239 | startPos = dotPos + 1; |
| 240 | } |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 241 | } |
| 242 | |
Andreas Huber | c9410c7 | 2016-07-28 12:18:40 -0700 | [diff] [blame] | 243 | } // namespace android; |