check.cpp 52 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300
  1. // Part of the Carbon Language project, under the Apache License v2.0 with LLVM
  2. // Exceptions. See /LICENSE for license information.
  3. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  4. #include "toolchain/check/check.h"
  5. #include <variant>
  6. #include "common/check.h"
  7. #include "common/error.h"
  8. #include "common/map.h"
  9. #include "common/variant_helpers.h"
  10. #include "common/vlog.h"
  11. #include "toolchain/base/kind_switch.h"
  12. #include "toolchain/base/pretty_stack_trace_function.h"
  13. #include "toolchain/check/context.h"
  14. #include "toolchain/check/diagnostic_helpers.h"
  15. #include "toolchain/check/function.h"
  16. #include "toolchain/check/handle.h"
  17. #include "toolchain/check/import.h"
  18. #include "toolchain/check/import_ref.h"
  19. #include "toolchain/check/sem_ir_diagnostic_converter.h"
  20. #include "toolchain/diagnostics/diagnostic.h"
  21. #include "toolchain/diagnostics/diagnostic_emitter.h"
  22. #include "toolchain/lex/token_kind.h"
  23. #include "toolchain/parse/node_ids.h"
  24. #include "toolchain/parse/tree.h"
  25. #include "toolchain/parse/tree_node_diagnostic_converter.h"
  26. #include "toolchain/sem_ir/file.h"
  27. #include "toolchain/sem_ir/ids.h"
  28. #include "toolchain/sem_ir/typed_insts.h"
  29. namespace Carbon::Check {
  30. namespace {
  31. struct UnitInfo {
  32. // A given import within the file, with its destination.
  33. struct Import {
  34. Parse::Tree::PackagingNames names;
  35. UnitInfo* unit_info;
  36. };
  37. // A file's imports corresponding to a single package, for the map.
  38. struct PackageImports {
  39. // Use the constructor so that the SmallVector is only constructed
  40. // as-needed.
  41. explicit PackageImports(IdentifierId package_id,
  42. Parse::ImportDeclId node_id)
  43. : package_id(package_id), node_id(node_id) {}
  44. // The identifier of the imported package.
  45. IdentifierId package_id;
  46. // The first `import` declaration in the file, which declared the package's
  47. // identifier (even if the import failed). Used for associating diagnostics
  48. // not specific to a single import.
  49. Parse::ImportDeclId node_id;
  50. // The associated `import` instruction. Only valid once a file is checked.
  51. SemIR::InstId import_decl_id = SemIR::InstId::Invalid;
  52. // Whether there's an import that failed to load.
  53. bool has_load_error = false;
  54. // The list of valid imports.
  55. llvm::SmallVector<Import> imports;
  56. };
  57. explicit UnitInfo(SemIR::CheckIRId check_ir_id, Unit& unit,
  58. Parse::NodeLocConverter& converter)
  59. : check_ir_id(check_ir_id),
  60. unit(&unit),
  61. err_tracker(*unit.consumer),
  62. emitter(converter, err_tracker) {}
  63. SemIR::CheckIRId check_ir_id;
  64. Unit* unit;
  65. // Emitter information.
  66. ErrorTrackingDiagnosticConsumer err_tracker;
  67. DiagnosticEmitter<Parse::NodeLoc> emitter;
  68. // List of the outgoing imports. If a package includes unavailable library
  69. // imports, it has an entry with has_load_error set. Invalid imports (for
  70. // example, `import Main;`) aren't added because they won't add identifiers to
  71. // name lookup.
  72. llvm::SmallVector<PackageImports> package_imports;
  73. // A map of the package names to the outgoing imports above.
  74. Map<IdentifierId, int32_t> package_imports_map;
  75. // The remaining number of imports which must be checked before this unit can
  76. // be processed.
  77. int32_t imports_remaining = 0;
  78. // A list of incoming imports. This will be empty for `impl` files, because
  79. // imports only touch `api` files.
  80. llvm::SmallVector<UnitInfo*> incoming_imports;
  81. // The corresponding `api` unit if this is an `impl` file. The entry should
  82. // also be in the corresponding `PackageImports`.
  83. UnitInfo* api_for_impl = nullptr;
  84. };
  85. } // namespace
  86. // Collects direct imports, for CollectTransitiveImports.
  87. static auto CollectDirectImports(llvm::SmallVector<SemIR::ImportIR>& results,
  88. llvm::MutableArrayRef<int> ir_to_result_index,
  89. SemIR::InstId import_decl_id,
  90. const UnitInfo::PackageImports& imports,
  91. bool is_local) -> void {
  92. for (const auto& import : imports.imports) {
  93. const auto& direct_ir = **import.unit_info->unit->sem_ir;
  94. auto& index = ir_to_result_index[direct_ir.check_ir_id().index];
  95. if (index != -1) {
  96. // This should only happen when doing API imports for an implementation
  97. // file. Don't change the entry; is_export doesn't matter.
  98. continue;
  99. }
  100. index = results.size();
  101. results.push_back({.decl_id = import_decl_id,
  102. // Only tag exports in API files, ignoring the value in
  103. // implementation files.
  104. .is_export = is_local && import.names.is_export,
  105. .sem_ir = &direct_ir});
  106. }
  107. }
  108. // Collects transitive imports, handling deduplication. These will be unified
  109. // between local_imports and api_imports.
  110. static auto CollectTransitiveImports(
  111. SemIR::InstId import_decl_id, const UnitInfo::PackageImports* local_imports,
  112. const UnitInfo::PackageImports* api_imports, int total_ir_count)
  113. -> llvm::SmallVector<SemIR::ImportIR> {
  114. llvm::SmallVector<SemIR::ImportIR> results;
  115. // Track whether an IR was imported in full, including `export import`. This
  116. // distinguishes from IRs that are indirectly added without all names being
  117. // exported to this IR.
  118. llvm::SmallVector<int> ir_to_result_index(total_ir_count, -1);
  119. // First add direct imports. This means that if an entity is imported both
  120. // directly and indirectly, the import path will reflect the direct import.
  121. if (local_imports) {
  122. CollectDirectImports(results, ir_to_result_index, import_decl_id,
  123. *local_imports,
  124. /*is_local=*/true);
  125. }
  126. if (api_imports) {
  127. CollectDirectImports(results, ir_to_result_index, import_decl_id,
  128. *api_imports,
  129. /*is_local=*/false);
  130. }
  131. // Loop through direct imports for any indirect exports. The underlying vector
  132. // is appended during iteration, so take the size first.
  133. const int direct_imports = results.size();
  134. for (int direct_index : llvm::seq(direct_imports)) {
  135. bool is_export = results[direct_index].is_export;
  136. for (const auto& indirect_ir :
  137. results[direct_index].sem_ir->import_irs().array_ref()) {
  138. if (!indirect_ir.is_export) {
  139. continue;
  140. }
  141. auto& indirect_index =
  142. ir_to_result_index[indirect_ir.sem_ir->check_ir_id().index];
  143. if (indirect_index == -1) {
  144. indirect_index = results.size();
  145. // TODO: In the case of a recursive `export import`, this only points at
  146. // the outermost import. May want something that better reflects the
  147. // recursion.
  148. results.push_back({.decl_id = results[direct_index].decl_id,
  149. .is_export = is_export,
  150. .sem_ir = indirect_ir.sem_ir});
  151. } else if (is_export) {
  152. results[indirect_index].is_export = true;
  153. }
  154. }
  155. }
  156. return results;
  157. }
  158. // Imports the current package.
  159. static auto ImportCurrentPackage(Context& context, UnitInfo& unit_info,
  160. int total_ir_count,
  161. SemIR::InstId package_inst_id,
  162. SemIR::TypeId namespace_type_id) -> void {
  163. // Add imports from the current package.
  164. auto import_map_lookup =
  165. unit_info.package_imports_map.Lookup(IdentifierId::Invalid);
  166. if (!import_map_lookup) {
  167. // Push the scope; there are no names to add.
  168. context.scope_stack().Push(package_inst_id, SemIR::NameScopeId::Package);
  169. return;
  170. }
  171. UnitInfo::PackageImports& self_import =
  172. unit_info.package_imports[import_map_lookup.value()];
  173. if (self_import.has_load_error) {
  174. context.name_scopes().Get(SemIR::NameScopeId::Package).has_error = true;
  175. }
  176. ImportLibrariesFromCurrentPackage(
  177. context, namespace_type_id,
  178. CollectTransitiveImports(self_import.import_decl_id, &self_import,
  179. /*api_imports=*/nullptr, total_ir_count));
  180. context.scope_stack().Push(
  181. package_inst_id, SemIR::NameScopeId::Package, SemIR::SpecificId::Invalid,
  182. context.name_scopes().Get(SemIR::NameScopeId::Package).has_error);
  183. }
  184. // Imports all other packages (excluding the current package).
  185. static auto ImportOtherPackages(Context& context, UnitInfo& unit_info,
  186. int total_ir_count,
  187. SemIR::TypeId namespace_type_id) -> void {
  188. // api_imports_list is initially the size of the current file's imports,
  189. // including for API files, for simplicity in iteration. It's only really used
  190. // when processing an implementation file, in order to combine the API file
  191. // imports.
  192. //
  193. // For packages imported by the API file, the IdentifierId is the package name
  194. // and the index is into the API's import list. Otherwise, the initial
  195. // {Invalid, -1} state remains.
  196. llvm::SmallVector<std::pair<IdentifierId, int32_t>> api_imports_list;
  197. api_imports_list.resize(unit_info.package_imports.size(),
  198. {IdentifierId::Invalid, -1});
  199. // When there's an API file, add the mapping to api_imports_list.
  200. if (unit_info.api_for_impl) {
  201. const auto& api_identifiers =
  202. unit_info.api_for_impl->unit->value_stores->identifiers();
  203. auto& impl_identifiers = unit_info.unit->value_stores->identifiers();
  204. for (auto [api_imports_index, api_imports] :
  205. llvm::enumerate(unit_info.api_for_impl->package_imports)) {
  206. // Skip the current package.
  207. if (!api_imports.package_id.is_valid()) {
  208. continue;
  209. }
  210. // Translate the package ID from the API file to the implementation file.
  211. auto impl_package_id =
  212. impl_identifiers.Add(api_identifiers.Get(api_imports.package_id));
  213. if (auto lookup = unit_info.package_imports_map.Lookup(impl_package_id)) {
  214. // On a hit, replace the entry to unify the API and implementation
  215. // imports.
  216. api_imports_list[lookup.value()] = {impl_package_id, api_imports_index};
  217. } else {
  218. // On a miss, add the package as API-only.
  219. api_imports_list.push_back({impl_package_id, api_imports_index});
  220. }
  221. }
  222. }
  223. for (auto [i, api_imports_entry] : llvm::enumerate(api_imports_list)) {
  224. // These variables are updated after figuring out which imports are present.
  225. auto import_decl_id = SemIR::InstId::Invalid;
  226. IdentifierId package_id = IdentifierId::Invalid;
  227. bool has_load_error = false;
  228. // Identify the local package imports if present.
  229. UnitInfo::PackageImports* local_imports = nullptr;
  230. if (i < unit_info.package_imports.size()) {
  231. local_imports = &unit_info.package_imports[i];
  232. if (!local_imports->package_id.is_valid()) {
  233. // Skip the current package.
  234. continue;
  235. }
  236. import_decl_id = local_imports->import_decl_id;
  237. package_id = local_imports->package_id;
  238. has_load_error |= local_imports->has_load_error;
  239. }
  240. // Identify the API package imports if present.
  241. UnitInfo::PackageImports* api_imports = nullptr;
  242. if (api_imports_entry.second != -1) {
  243. api_imports =
  244. &unit_info.api_for_impl->package_imports[api_imports_entry.second];
  245. if (local_imports) {
  246. CARBON_CHECK(package_id == api_imports_entry.first);
  247. } else {
  248. auto import_ir_inst_id = context.import_ir_insts().Add(
  249. {.ir_id = SemIR::ImportIRId::ApiForImpl,
  250. .inst_id = api_imports->import_decl_id});
  251. import_decl_id =
  252. context.AddInst(context.MakeImportedLocAndInst<SemIR::ImportDecl>(
  253. import_ir_inst_id, {.package_id = SemIR::NameId::ForIdentifier(
  254. api_imports_entry.first)}));
  255. package_id = api_imports_entry.first;
  256. }
  257. has_load_error |= api_imports->has_load_error;
  258. }
  259. // Do the actual import.
  260. ImportLibrariesFromOtherPackage(
  261. context, namespace_type_id, import_decl_id, package_id,
  262. CollectTransitiveImports(import_decl_id, local_imports, api_imports,
  263. total_ir_count),
  264. has_load_error);
  265. }
  266. }
  267. // Add imports to the root block.
  268. static auto InitPackageScopeAndImports(Context& context, UnitInfo& unit_info,
  269. int total_ir_count) -> void {
  270. // First create the constant values map for all imported IRs. We'll populate
  271. // these with mappings for namespaces as we go.
  272. size_t num_irs = 0;
  273. for (auto& package_imports : unit_info.package_imports) {
  274. num_irs += package_imports.imports.size();
  275. }
  276. if (!unit_info.api_for_impl) {
  277. // Leave an empty slot for ImportIRId::ApiForImpl.
  278. ++num_irs;
  279. }
  280. context.import_irs().Reserve(num_irs);
  281. context.import_ir_constant_values().reserve(num_irs);
  282. context.SetTotalIRCount(total_ir_count);
  283. // Importing makes many namespaces, so only canonicalize the type once.
  284. auto namespace_type_id =
  285. context.GetBuiltinType(SemIR::BuiltinInstKind::NamespaceType);
  286. // Define the package scope, with an instruction for `package` expressions to
  287. // reference.
  288. auto package_scope_id = context.name_scopes().Add(
  289. SemIR::InstId::PackageNamespace, SemIR::NameId::PackageNamespace,
  290. SemIR::NameScopeId::Invalid);
  291. CARBON_CHECK(package_scope_id == SemIR::NameScopeId::Package);
  292. auto package_inst_id = context.AddInst<SemIR::Namespace>(
  293. Parse::NodeId::Invalid, {.type_id = namespace_type_id,
  294. .name_scope_id = SemIR::NameScopeId::Package,
  295. .import_id = SemIR::InstId::Invalid});
  296. CARBON_CHECK(package_inst_id == SemIR::InstId::PackageNamespace);
  297. // If there is an implicit `api` import, set it first so that it uses the
  298. // ImportIRId::ApiForImpl when processed for imports.
  299. if (unit_info.api_for_impl) {
  300. const auto& names = context.parse_tree().packaging_decl()->names;
  301. auto import_decl_id = context.AddInst<SemIR::ImportDecl>(
  302. names.node_id,
  303. {.package_id = SemIR::NameId::ForIdentifier(names.package_id)});
  304. SetApiImportIR(context,
  305. {.decl_id = import_decl_id,
  306. .is_export = false,
  307. .sem_ir = &**unit_info.api_for_impl->unit->sem_ir});
  308. } else {
  309. SetApiImportIR(context,
  310. {.decl_id = SemIR::InstId::Invalid, .sem_ir = nullptr});
  311. }
  312. // Add import instructions for everything directly imported. Implicit imports
  313. // are handled separately.
  314. for (auto& package_imports : unit_info.package_imports) {
  315. CARBON_CHECK(!package_imports.import_decl_id.is_valid());
  316. package_imports.import_decl_id = context.AddInst<SemIR::ImportDecl>(
  317. package_imports.node_id, {.package_id = SemIR::NameId::ForIdentifier(
  318. package_imports.package_id)});
  319. }
  320. // Process the imports.
  321. if (unit_info.api_for_impl) {
  322. ImportApiFile(context, namespace_type_id,
  323. **unit_info.api_for_impl->unit->sem_ir);
  324. }
  325. ImportCurrentPackage(context, unit_info, total_ir_count, package_inst_id,
  326. namespace_type_id);
  327. CARBON_CHECK(context.scope_stack().PeekIndex() == ScopeIndex::Package);
  328. ImportOtherPackages(context, unit_info, total_ir_count, namespace_type_id);
  329. }
  330. namespace {
  331. // State used to track the next deferred function definition that we will
  332. // encounter and need to reorder.
  333. class NextDeferredDefinitionCache {
  334. public:
  335. explicit NextDeferredDefinitionCache(const Parse::Tree* tree) : tree_(tree) {
  336. SkipTo(Parse::DeferredDefinitionIndex(0));
  337. }
  338. // Set the specified deferred definition index as being the next one that will
  339. // be encountered.
  340. auto SkipTo(Parse::DeferredDefinitionIndex next_index) -> void {
  341. index_ = next_index;
  342. if (static_cast<std::size_t>(index_.index) ==
  343. tree_->deferred_definitions().size()) {
  344. start_id_ = Parse::NodeId::Invalid;
  345. } else {
  346. start_id_ = tree_->deferred_definitions().Get(index_).start_id;
  347. }
  348. }
  349. // Returns the index of the next deferred definition to be encountered.
  350. auto index() const -> Parse::DeferredDefinitionIndex { return index_; }
  351. // Returns the ID of the start node of the next deferred definition.
  352. auto start_id() const -> Parse::NodeId { return start_id_; }
  353. private:
  354. const Parse::Tree* tree_;
  355. Parse::DeferredDefinitionIndex index_ =
  356. Parse::DeferredDefinitionIndex::Invalid;
  357. Parse::NodeId start_id_ = Parse::NodeId::Invalid;
  358. };
  359. } // namespace
  360. // Determines whether this node kind is the start of a deferred definition
  361. // scope.
  362. static auto IsStartOfDeferredDefinitionScope(Parse::NodeKind kind) -> bool {
  363. switch (kind) {
  364. case Parse::NodeKind::ClassDefinitionStart:
  365. case Parse::NodeKind::ImplDefinitionStart:
  366. case Parse::NodeKind::InterfaceDefinitionStart:
  367. case Parse::NodeKind::NamedConstraintDefinitionStart:
  368. // TODO: Mixins.
  369. return true;
  370. default:
  371. return false;
  372. }
  373. }
  374. // Determines whether this node kind is the end of a deferred definition scope.
  375. static auto IsEndOfDeferredDefinitionScope(Parse::NodeKind kind) -> bool {
  376. switch (kind) {
  377. case Parse::NodeKind::ClassDefinition:
  378. case Parse::NodeKind::ImplDefinition:
  379. case Parse::NodeKind::InterfaceDefinition:
  380. case Parse::NodeKind::NamedConstraintDefinition:
  381. // TODO: Mixins.
  382. return true;
  383. default:
  384. return false;
  385. }
  386. }
  387. namespace {
  388. // A worklist of pending tasks to perform to check deferred function definitions
  389. // in the right order.
  390. class DeferredDefinitionWorklist {
  391. public:
  392. // A worklist task that indicates we should check a deferred function
  393. // definition that we previously skipped.
  394. struct CheckSkippedDefinition {
  395. // The definition that we skipped.
  396. Parse::DeferredDefinitionIndex definition_index;
  397. // The suspended function.
  398. SuspendedFunction suspended_fn;
  399. };
  400. // A worklist task that indicates we should enter a nested deferred definition
  401. // scope.
  402. struct EnterDeferredDefinitionScope {
  403. // The suspended scope. This is only set once we reach the end of the scope.
  404. std::optional<DeclNameStack::SuspendedName> suspended_name;
  405. // Whether this scope is itself within an outer deferred definition scope.
  406. // If so, we'll delay processing its contents until we reach the end of the
  407. // parent scope. For example:
  408. //
  409. // ```
  410. // class A {
  411. // class B {
  412. // fn F() -> A { return {}; }
  413. // }
  414. // } // A.B.F is type-checked here, with A complete.
  415. //
  416. // fn F() {
  417. // class C {
  418. // fn G() {}
  419. // } // C.G is type-checked here.
  420. // }
  421. // ```
  422. bool in_deferred_definition_scope;
  423. };
  424. // A worklist task that indicates we should leave a deferred definition scope.
  425. struct LeaveDeferredDefinitionScope {
  426. // Whether this scope is within another deferred definition scope.
  427. bool in_deferred_definition_scope;
  428. };
  429. // A pending type-checking task.
  430. using Task =
  431. std::variant<CheckSkippedDefinition, EnterDeferredDefinitionScope,
  432. LeaveDeferredDefinitionScope>;
  433. explicit DeferredDefinitionWorklist(llvm::raw_ostream* vlog_stream)
  434. : vlog_stream_(vlog_stream) {
  435. // See declaration of `worklist_`.
  436. worklist_.reserve(64);
  437. }
  438. static constexpr llvm::StringLiteral VlogPrefix =
  439. "DeferredDefinitionWorklist ";
  440. // Suspend the current function definition and push a task onto the worklist
  441. // to finish it later.
  442. auto SuspendFunctionAndPush(Context& context,
  443. Parse::DeferredDefinitionIndex index,
  444. Parse::FunctionDefinitionStartId node_id)
  445. -> void {
  446. worklist_.push_back(CheckSkippedDefinition{
  447. index, HandleFunctionDefinitionSuspend(context, node_id)});
  448. CARBON_VLOG("{0}Push CheckSkippedDefinition {1}\n", VlogPrefix,
  449. index.index);
  450. }
  451. // Push a task to re-enter a function scope, so that functions defined within
  452. // it are type-checked in the right context.
  453. auto PushEnterDeferredDefinitionScope(Context& context) -> void {
  454. bool nested = !entered_scopes_.empty() &&
  455. entered_scopes_.back().scope_index ==
  456. context.decl_name_stack().PeekInitialScopeIndex();
  457. entered_scopes_.push_back(
  458. {.worklist_start_index = worklist_.size(),
  459. .scope_index = context.scope_stack().PeekIndex()});
  460. worklist_.push_back(
  461. EnterDeferredDefinitionScope{.suspended_name = std::nullopt,
  462. .in_deferred_definition_scope = nested});
  463. CARBON_VLOG("{0}Push EnterDeferredDefinitionScope {1}\n", VlogPrefix,
  464. nested ? "(nested)" : "(non-nested)");
  465. }
  466. // Suspend the current deferred definition scope, which is finished but still
  467. // on the decl_name_stack, and push a task to leave the scope when we're
  468. // type-checking deferred definitions. Returns `true` if the current list of
  469. // deferred definitions should be type-checked immediately.
  470. auto SuspendFinishedScopeAndPush(Context& context) -> bool;
  471. // Pop the next task off the worklist.
  472. auto Pop() -> Task {
  473. if (vlog_stream_) {
  474. VariantMatch(
  475. worklist_.back(),
  476. [&](CheckSkippedDefinition& definition) {
  477. CARBON_VLOG("{0}Handle CheckSkippedDefinition {1}\n", VlogPrefix,
  478. definition.definition_index.index);
  479. },
  480. [&](EnterDeferredDefinitionScope& enter) {
  481. CARBON_CHECK(enter.in_deferred_definition_scope);
  482. CARBON_VLOG("{0}Handle EnterDeferredDefinitionScope (nested)\n",
  483. VlogPrefix);
  484. },
  485. [&](LeaveDeferredDefinitionScope& leave) {
  486. bool nested = leave.in_deferred_definition_scope;
  487. CARBON_VLOG("{0}Handle LeaveDeferredDefinitionScope {1}\n",
  488. VlogPrefix, nested ? "(nested)" : "(non-nested)");
  489. });
  490. }
  491. return worklist_.pop_back_val();
  492. }
  493. // CHECK that the work list has no further work.
  494. auto VerifyEmpty() {
  495. CARBON_CHECK(worklist_.empty() && entered_scopes_.empty())
  496. << "Tasks left behind on worklist.";
  497. }
  498. private:
  499. llvm::raw_ostream* vlog_stream_;
  500. // A worklist of type-checking tasks we'll need to do later.
  501. //
  502. // Don't allocate any inline storage here. A Task is fairly large, so we never
  503. // want this to live on the stack. Instead, we reserve space in the
  504. // constructor for a fairly large number of deferred definitions.
  505. llvm::SmallVector<Task, 0> worklist_;
  506. // A deferred definition scope that is currently still open.
  507. struct EnteredScope {
  508. // The index in worklist_ of the EnterDeferredDefinitionScope task.
  509. size_t worklist_start_index;
  510. // The corresponding lexical scope index.
  511. ScopeIndex scope_index;
  512. };
  513. // The deferred definition scopes for the current checking actions.
  514. llvm::SmallVector<EnteredScope> entered_scopes_;
  515. };
  516. } // namespace
  517. auto DeferredDefinitionWorklist::SuspendFinishedScopeAndPush(Context& context)
  518. -> bool {
  519. auto start_index = entered_scopes_.pop_back_val().worklist_start_index;
  520. // If we've not found any deferred definitions in this scope, clean up the
  521. // stack.
  522. if (start_index == worklist_.size() - 1) {
  523. context.decl_name_stack().PopScope();
  524. worklist_.pop_back();
  525. CARBON_VLOG("{0}Pop EnterDeferredDefinitionScope (empty)\n", VlogPrefix);
  526. return false;
  527. }
  528. // If we're finishing a nested deferred definition scope, keep track of that
  529. // but don't type-check deferred definitions now.
  530. auto& enter_scope = get<EnterDeferredDefinitionScope>(worklist_[start_index]);
  531. if (enter_scope.in_deferred_definition_scope) {
  532. // This is a nested deferred definition scope. Suspend the inner scope so we
  533. // can restore it when we come to type-check the deferred definitions.
  534. enter_scope.suspended_name = context.decl_name_stack().Suspend();
  535. // Enqueue a task to leave the nested scope.
  536. worklist_.push_back(
  537. LeaveDeferredDefinitionScope{.in_deferred_definition_scope = true});
  538. CARBON_VLOG("{0}Push LeaveDeferredDefinitionScope (nested)\n", VlogPrefix);
  539. return false;
  540. }
  541. // We're at the end of a non-nested deferred definition scope. Prepare to
  542. // start checking deferred definitions. Enqueue a task to leave this outer
  543. // scope and end checking deferred definitions.
  544. worklist_.push_back(
  545. LeaveDeferredDefinitionScope{.in_deferred_definition_scope = false});
  546. CARBON_VLOG("{0}Push LeaveDeferredDefinitionScope (non-nested)\n",
  547. VlogPrefix);
  548. // We'll process the worklist in reverse index order, so reverse the part of
  549. // it we're about to execute so we run our tasks in the order in which they
  550. // were pushed.
  551. std::reverse(worklist_.begin() + start_index, worklist_.end());
  552. // Pop the `EnterDeferredDefinitionScope` that's now on the end of the
  553. // worklist. We stay in that scope rather than suspending then immediately
  554. // resuming it.
  555. CARBON_CHECK(
  556. holds_alternative<EnterDeferredDefinitionScope>(worklist_.back()))
  557. << "Unexpected task in worklist.";
  558. worklist_.pop_back();
  559. CARBON_VLOG("{0}Handle EnterDeferredDefinitionScope (non-nested)\n",
  560. VlogPrefix);
  561. return true;
  562. }
  563. namespace {
  564. // A traversal of the node IDs in the parse tree, in the order in which we need
  565. // to check them.
  566. class NodeIdTraversal {
  567. public:
  568. explicit NodeIdTraversal(Context& context, llvm::raw_ostream* vlog_stream)
  569. : context_(context),
  570. next_deferred_definition_(&context.parse_tree()),
  571. worklist_(vlog_stream) {
  572. auto range = context.parse_tree().postorder();
  573. chunks_.push_back(
  574. {.it = range.begin(),
  575. .end = range.end(),
  576. .next_definition = Parse::DeferredDefinitionIndex::Invalid});
  577. }
  578. // Finds the next `NodeId` to type-check. Returns nullopt if the traversal is
  579. // complete.
  580. auto Next() -> std::optional<Parse::NodeId>;
  581. // Performs any processing necessary after we type-check a node.
  582. auto Handle(Parse::NodeKind parse_kind) -> void {
  583. // When we reach the start of a deferred definition scope, add a task to the
  584. // worklist to check future skipped definitions in the new context.
  585. if (IsStartOfDeferredDefinitionScope(parse_kind)) {
  586. worklist_.PushEnterDeferredDefinitionScope(context_);
  587. }
  588. // When we reach the end of a deferred definition scope, add a task to the
  589. // worklist to leave the scope. If this is not a nested scope, start
  590. // checking the deferred definitions now.
  591. if (IsEndOfDeferredDefinitionScope(parse_kind)) {
  592. chunks_.back().checking_deferred_definitions =
  593. worklist_.SuspendFinishedScopeAndPush(context_);
  594. }
  595. }
  596. private:
  597. // A chunk of the parse tree that we need to type-check.
  598. struct Chunk {
  599. Parse::Tree::PostorderIterator it;
  600. Parse::Tree::PostorderIterator end;
  601. // The next definition that will be encountered after this chunk completes.
  602. Parse::DeferredDefinitionIndex next_definition;
  603. // Whether we are currently checking deferred definitions, rather than the
  604. // tokens of this chunk. If so, we'll pull tasks off `worklist` and execute
  605. // them until we're done with this batch of deferred definitions. Otherwise,
  606. // we'll pull node IDs from `*it` until it reaches `end`.
  607. bool checking_deferred_definitions = false;
  608. };
  609. // Re-enter a nested deferred definition scope.
  610. auto PerformTask(
  611. DeferredDefinitionWorklist::EnterDeferredDefinitionScope&& enter)
  612. -> void {
  613. CARBON_CHECK(enter.suspended_name)
  614. << "Entering a scope with no suspension information.";
  615. context_.decl_name_stack().Restore(std::move(*enter.suspended_name));
  616. }
  617. // Leave a nested or top-level deferred definition scope.
  618. auto PerformTask(
  619. DeferredDefinitionWorklist::LeaveDeferredDefinitionScope&& leave)
  620. -> void {
  621. if (!leave.in_deferred_definition_scope) {
  622. // We're done with checking deferred definitions.
  623. chunks_.back().checking_deferred_definitions = false;
  624. }
  625. context_.decl_name_stack().PopScope();
  626. }
  627. // Resume checking a deferred definition.
  628. auto PerformTask(
  629. DeferredDefinitionWorklist::CheckSkippedDefinition&& parse_definition)
  630. -> void {
  631. auto& [definition_index, suspended_fn] = parse_definition;
  632. const auto& definition_info =
  633. context_.parse_tree().deferred_definitions().Get(definition_index);
  634. HandleFunctionDefinitionResume(context_, definition_info.start_id,
  635. std::move(suspended_fn));
  636. auto range = Parse::Tree::PostorderIterator::MakeRange(
  637. definition_info.start_id, definition_info.definition_id);
  638. chunks_.push_back({.it = range.begin() + 1,
  639. .end = range.end(),
  640. .next_definition = next_deferred_definition_.index()});
  641. ++definition_index.index;
  642. next_deferred_definition_.SkipTo(definition_index);
  643. }
  644. Context& context_;
  645. NextDeferredDefinitionCache next_deferred_definition_;
  646. DeferredDefinitionWorklist worklist_;
  647. llvm::SmallVector<Chunk> chunks_;
  648. };
  649. } // namespace
  650. auto NodeIdTraversal::Next() -> std::optional<Parse::NodeId> {
  651. while (true) {
  652. // If we're checking deferred definitions, find the next definition we
  653. // should check, restore its suspended state, and add a corresponding
  654. // `Chunk` to the top of the chunk list.
  655. if (chunks_.back().checking_deferred_definitions) {
  656. std::visit(
  657. [&](auto&& task) { PerformTask(std::forward<decltype(task)>(task)); },
  658. worklist_.Pop());
  659. continue;
  660. }
  661. // If we're not checking deferred definitions, produce the next parse node
  662. // for this chunk. If we've run out of parse nodes, we're done with this
  663. // chunk of the parse tree.
  664. if (chunks_.back().it == chunks_.back().end) {
  665. auto old_chunk = chunks_.pop_back_val();
  666. // If we're out of chunks, then we're done entirely.
  667. if (chunks_.empty()) {
  668. worklist_.VerifyEmpty();
  669. return std::nullopt;
  670. }
  671. next_deferred_definition_.SkipTo(old_chunk.next_definition);
  672. continue;
  673. }
  674. auto node_id = *chunks_.back().it;
  675. // If we've reached the start of a deferred definition, skip to the end of
  676. // it, and track that we need to check it later.
  677. if (node_id == next_deferred_definition_.start_id()) {
  678. const auto& definition_info =
  679. context_.parse_tree().deferred_definitions().Get(
  680. next_deferred_definition_.index());
  681. worklist_.SuspendFunctionAndPush(context_,
  682. next_deferred_definition_.index(),
  683. definition_info.start_id);
  684. // Continue type-checking the parse tree after the end of the definition.
  685. chunks_.back().it =
  686. Parse::Tree::PostorderIterator(definition_info.definition_id) + 1;
  687. next_deferred_definition_.SkipTo(definition_info.next_definition_index);
  688. continue;
  689. }
  690. ++chunks_.back().it;
  691. return node_id;
  692. }
  693. }
  694. // Emits a diagnostic for each declaration in context.definitions_required()
  695. // that doesn't have a definition.
  696. static auto DiagnoseMissingDefinitions(Context& context,
  697. Context::DiagnosticEmitter& emitter)
  698. -> void {
  699. CARBON_DIAGNOSTIC(MissingDefinitionInImpl, Error,
  700. "No definition found for declaration in impl file");
  701. for (SemIR::InstId decl_inst_id : context.definitions_required()) {
  702. SemIR::Inst decl_inst = context.insts().Get(decl_inst_id);
  703. CARBON_KIND_SWITCH(context.insts().Get(decl_inst_id)) {
  704. case CARBON_KIND(SemIR::ClassDecl class_decl): {
  705. if (!context.classes().Get(class_decl.class_id).is_defined()) {
  706. emitter.Emit(decl_inst_id, MissingDefinitionInImpl);
  707. }
  708. break;
  709. }
  710. case CARBON_KIND(SemIR::FunctionDecl function_decl): {
  711. if (context.functions().Get(function_decl.function_id).definition_id ==
  712. SemIR::InstId::Invalid) {
  713. emitter.Emit(decl_inst_id, MissingDefinitionInImpl);
  714. }
  715. break;
  716. }
  717. case CARBON_KIND(SemIR::ImplDecl impl_decl): {
  718. if (!context.impls().Get(impl_decl.impl_id).is_defined()) {
  719. emitter.Emit(decl_inst_id, MissingDefinitionInImpl);
  720. }
  721. break;
  722. }
  723. case SemIR::InterfaceDecl::Kind: {
  724. // TODO: handle `interface` as well, once we can test it without
  725. // triggering https://github.com/carbon-language/carbon-lang/issues/4071
  726. CARBON_FATAL()
  727. << "TODO: Support interfaces in DiagnoseMissingDefinitions";
  728. }
  729. default: {
  730. CARBON_FATAL() << "Unexpected inst in definitions_required: "
  731. << decl_inst;
  732. }
  733. }
  734. }
  735. }
  736. // Loops over all nodes in the tree. On some errors, this may return early,
  737. // for example if an unrecoverable state is encountered.
  738. // NOLINTNEXTLINE(readability-function-size)
  739. static auto ProcessNodeIds(Context& context, llvm::raw_ostream* vlog_stream,
  740. ErrorTrackingDiagnosticConsumer& err_tracker,
  741. Parse::NodeLocConverter& converter) -> bool {
  742. NodeIdTraversal traversal(context, vlog_stream);
  743. Parse::NodeId node_id = Parse::NodeId::Invalid;
  744. // On crash, report which token we were handling.
  745. PrettyStackTraceFunction node_dumper([&](llvm::raw_ostream& output) {
  746. auto loc = converter.ConvertLoc(
  747. node_id, [](DiagnosticLoc, const Internal::DiagnosticBase<>&) {});
  748. loc.FormatLocation(output);
  749. output << ": checking " << context.parse_tree().node_kind(node_id) << "\n";
  750. // Crash output has a tab indent; try to indent slightly past that.
  751. loc.FormatSnippet(output, /*indent=*/10);
  752. });
  753. while (auto maybe_node_id = traversal.Next()) {
  754. node_id = *maybe_node_id;
  755. auto parse_kind = context.parse_tree().node_kind(node_id);
  756. switch (parse_kind) {
  757. #define CARBON_PARSE_NODE_KIND(Name) \
  758. case Parse::NodeKind::Name: { \
  759. if (!HandleParseNode(context, Parse::Name##Id(node_id))) { \
  760. CARBON_CHECK(err_tracker.seen_error()) \
  761. << "Handle" #Name " returned false without printing a diagnostic"; \
  762. return false; \
  763. } \
  764. break; \
  765. }
  766. #include "toolchain/parse/node_kind.def"
  767. }
  768. traversal.Handle(parse_kind);
  769. }
  770. return true;
  771. }
  772. // Produces and checks the IR for the provided Parse::Tree.
  773. static auto CheckParseTree(
  774. llvm::MutableArrayRef<Parse::NodeLocConverter> node_converters,
  775. UnitInfo& unit_info, int total_ir_count, llvm::raw_ostream* vlog_stream)
  776. -> void {
  777. auto package_id = IdentifierId::Invalid;
  778. auto library_id = StringLiteralValueId::Invalid;
  779. if (const auto& packaging = unit_info.unit->parse_tree->packaging_decl()) {
  780. package_id = packaging->names.package_id;
  781. library_id = packaging->names.library_id;
  782. }
  783. unit_info.unit->sem_ir->emplace(
  784. unit_info.check_ir_id, package_id,
  785. SemIR::LibraryNameId::ForStringLiteralValueId(library_id),
  786. *unit_info.unit->value_stores,
  787. unit_info.unit->tokens->source().filename().str());
  788. SemIR::File& sem_ir = **unit_info.unit->sem_ir;
  789. SemIRDiagnosticConverter converter(node_converters, &sem_ir);
  790. Context::DiagnosticEmitter emitter(converter, unit_info.err_tracker);
  791. Context context(*unit_info.unit->tokens, emitter, *unit_info.unit->parse_tree,
  792. unit_info.unit->get_parse_tree_and_subtrees, sem_ir,
  793. vlog_stream);
  794. PrettyStackTraceFunction context_dumper(
  795. [&](llvm::raw_ostream& output) { context.PrintForStackDump(output); });
  796. // Add a block for the file.
  797. context.inst_block_stack().Push();
  798. InitPackageScopeAndImports(context, unit_info, total_ir_count);
  799. // Import all impls declared in imports.
  800. // TODO: Do this selectively when we see an impl query.
  801. ImportImpls(context);
  802. if (!ProcessNodeIds(context, vlog_stream, unit_info.err_tracker,
  803. node_converters[unit_info.check_ir_id.index])) {
  804. context.sem_ir().set_has_errors(true);
  805. return;
  806. }
  807. context.Finalize();
  808. DiagnoseMissingDefinitions(context, emitter);
  809. context.VerifyOnFinish();
  810. sem_ir.set_has_errors(unit_info.err_tracker.seen_error());
  811. #ifndef NDEBUG
  812. if (auto verify = sem_ir.Verify(); !verify.ok()) {
  813. CARBON_FATAL() << sem_ir << "Built invalid semantics IR: " << verify.error()
  814. << "\n";
  815. }
  816. #endif
  817. }
  818. // The package and library names, used as map keys.
  819. using ImportKey = std::pair<llvm::StringRef, llvm::StringRef>;
  820. // Returns a key form of the package object. file_package_id is only used for
  821. // imports, not the main package declaration; as a consequence, it will be
  822. // invalid for the main package declaration.
  823. static auto GetImportKey(UnitInfo& unit_info, IdentifierId file_package_id,
  824. Parse::Tree::PackagingNames names) -> ImportKey {
  825. auto* stores = unit_info.unit->value_stores;
  826. llvm::StringRef package_name =
  827. names.package_id.is_valid() ? stores->identifiers().Get(names.package_id)
  828. : file_package_id.is_valid() ? stores->identifiers().Get(file_package_id)
  829. : "";
  830. llvm::StringRef library_name =
  831. names.library_id.is_valid()
  832. ? stores->string_literal_values().Get(names.library_id)
  833. : "";
  834. return {package_name, library_name};
  835. }
  836. static constexpr llvm::StringLiteral ExplicitMainName = "Main";
  837. static auto RenderImportKey(ImportKey import_key) -> std::string {
  838. if (import_key.first.empty()) {
  839. import_key.first = ExplicitMainName;
  840. }
  841. if (import_key.second.empty()) {
  842. return import_key.first.str();
  843. }
  844. return llvm::formatv("{0}//{1}", import_key.first, import_key.second).str();
  845. }
  846. // Marks an import as required on both the source and target file.
  847. //
  848. // The ID comparisons between the import and unit are okay because they both
  849. // come from the same file.
  850. static auto TrackImport(Map<ImportKey, UnitInfo*>& api_map,
  851. Map<ImportKey, Parse::NodeId>* explicit_import_map,
  852. UnitInfo& unit_info, Parse::Tree::PackagingNames import)
  853. -> void {
  854. const auto& packaging = unit_info.unit->parse_tree->packaging_decl();
  855. IdentifierId file_package_id =
  856. packaging ? packaging->names.package_id : IdentifierId::Invalid;
  857. auto import_key = GetImportKey(unit_info, file_package_id, import);
  858. // True if the import has `Main` as the package name, even if it comes from
  859. // the file's packaging (diagnostics may differentiate).
  860. bool is_explicit_main = import_key.first == ExplicitMainName;
  861. // Explicit imports need more validation than implicit ones. We try to do
  862. // these in an order of imports that should be removed, followed by imports
  863. // that might be valid with syntax fixes.
  864. if (explicit_import_map) {
  865. // Diagnose redundant imports.
  866. if (auto insert_result =
  867. explicit_import_map->Insert(import_key, import.node_id);
  868. !insert_result.is_inserted()) {
  869. CARBON_DIAGNOSTIC(RepeatedImport, Error,
  870. "Library imported more than once.");
  871. CARBON_DIAGNOSTIC(FirstImported, Note, "First import here.");
  872. unit_info.emitter.Build(import.node_id, RepeatedImport)
  873. .Note(insert_result.value(), FirstImported)
  874. .Emit();
  875. return;
  876. }
  877. // True if the file's package is implicitly `Main` (by omitting an explicit
  878. // package name).
  879. bool is_file_implicit_main =
  880. !packaging || !packaging->names.package_id.is_valid();
  881. // True if the import is using implicit "current package" syntax (by
  882. // omitting an explicit package name).
  883. bool is_import_implicit_current_package = !import.package_id.is_valid();
  884. // True if the import is using `default` library syntax.
  885. bool is_import_default_library = !import.library_id.is_valid();
  886. // True if the import and file point at the same package, even by
  887. // incorrectly specifying the current package name to `import`.
  888. bool is_same_package = is_import_implicit_current_package ||
  889. import.package_id == file_package_id;
  890. // True if the import points at the same library as the file's library.
  891. bool is_same_library =
  892. is_same_package &&
  893. (packaging ? import.library_id == packaging->names.library_id
  894. : is_import_default_library);
  895. // Diagnose explicit imports of the same library, whether from `api` or
  896. // `impl`.
  897. if (is_same_library) {
  898. CARBON_DIAGNOSTIC(ExplicitImportApi, Error,
  899. "Explicit import of `api` from `impl` file is "
  900. "redundant with implicit import.");
  901. CARBON_DIAGNOSTIC(ImportSelf, Error, "File cannot import itself.");
  902. bool is_impl = !packaging || packaging->is_impl;
  903. unit_info.emitter.Emit(import.node_id,
  904. is_impl ? ExplicitImportApi : ImportSelf);
  905. return;
  906. }
  907. // Diagnose explicit imports of `Main//default`. There is no `api` for it.
  908. // This lets other diagnostics handle explicit `Main` package naming.
  909. if (is_file_implicit_main && is_import_implicit_current_package &&
  910. is_import_default_library) {
  911. CARBON_DIAGNOSTIC(ImportMainDefaultLibrary, Error,
  912. "Cannot import `Main//default`.");
  913. unit_info.emitter.Emit(import.node_id, ImportMainDefaultLibrary);
  914. return;
  915. }
  916. if (!is_import_implicit_current_package) {
  917. // Diagnose explicit imports of the same package that use the package
  918. // name.
  919. if (is_same_package || (is_file_implicit_main && is_explicit_main)) {
  920. CARBON_DIAGNOSTIC(
  921. ImportCurrentPackageByName, Error,
  922. "Imports from the current package must omit the package name.");
  923. unit_info.emitter.Emit(import.node_id, ImportCurrentPackageByName);
  924. return;
  925. }
  926. // Diagnose explicit imports from `Main`.
  927. if (is_explicit_main) {
  928. CARBON_DIAGNOSTIC(ImportMainPackage, Error,
  929. "Cannot import `Main` from other packages.");
  930. unit_info.emitter.Emit(import.node_id, ImportMainPackage);
  931. return;
  932. }
  933. }
  934. } else if (is_explicit_main) {
  935. // An implicit import with an explicit `Main` occurs when a `package` rule
  936. // has bad syntax, which will have been diagnosed when building the API map.
  937. // As a consequence, we return silently.
  938. return;
  939. }
  940. // Get the package imports, or create them if this is the first.
  941. auto create_imports = [&]() -> int32_t {
  942. int32_t index = unit_info.package_imports.size();
  943. unit_info.package_imports.push_back(
  944. UnitInfo::PackageImports(import.package_id, import.node_id));
  945. return index;
  946. };
  947. auto insert_result =
  948. unit_info.package_imports_map.Insert(import.package_id, create_imports);
  949. UnitInfo::PackageImports& package_imports =
  950. unit_info.package_imports[insert_result.value()];
  951. if (auto api_lookup = api_map.Lookup(import_key)) {
  952. // Add references between the file and imported api.
  953. UnitInfo* api = api_lookup.value();
  954. package_imports.imports.push_back({import, api});
  955. ++unit_info.imports_remaining;
  956. api->incoming_imports.push_back(&unit_info);
  957. // If this is the implicit import, note we have it.
  958. if (!explicit_import_map) {
  959. CARBON_CHECK(!unit_info.api_for_impl);
  960. unit_info.api_for_impl = api;
  961. }
  962. } else {
  963. // The imported api is missing.
  964. package_imports.has_load_error = true;
  965. CARBON_DIAGNOSTIC(LibraryApiNotFound, Error,
  966. "Corresponding API for '{0}' not found.", std::string);
  967. CARBON_DIAGNOSTIC(ImportNotFound, Error, "Imported API '{0}' not found.",
  968. std::string);
  969. unit_info.emitter.Emit(
  970. import.node_id,
  971. explicit_import_map ? ImportNotFound : LibraryApiNotFound,
  972. RenderImportKey(import_key));
  973. }
  974. }
  975. // Builds a map of `api` files which might be imported. Also diagnoses issues
  976. // related to the packaging because the strings are loaded as part of getting
  977. // the ImportKey (which we then do for `impl` files too).
  978. static auto BuildApiMapAndDiagnosePackaging(
  979. llvm::MutableArrayRef<UnitInfo> unit_infos) -> Map<ImportKey, UnitInfo*> {
  980. Map<ImportKey, UnitInfo*> api_map;
  981. for (auto& unit_info : unit_infos) {
  982. const auto& packaging = unit_info.unit->parse_tree->packaging_decl();
  983. // An import key formed from the `package` or `library` declaration. Or, for
  984. // Main//default, a placeholder key.
  985. auto import_key = packaging ? GetImportKey(unit_info, IdentifierId::Invalid,
  986. packaging->names)
  987. // Construct a boring key for Main//default.
  988. : ImportKey{"", ""};
  989. // Diagnose explicit `Main` uses before they become marked as possible
  990. // APIs.
  991. if (import_key.first == ExplicitMainName) {
  992. CARBON_DIAGNOSTIC(ExplicitMainPackage, Error,
  993. "`Main//default` must omit `package` declaration.");
  994. CARBON_DIAGNOSTIC(
  995. ExplicitMainLibrary, Error,
  996. "Use `library` declaration in `Main` package libraries.");
  997. unit_info.emitter.Emit(packaging->names.node_id,
  998. import_key.second.empty() ? ExplicitMainPackage
  999. : ExplicitMainLibrary);
  1000. continue;
  1001. }
  1002. bool is_impl = packaging && packaging->is_impl;
  1003. // Add to the `api` map and diagnose duplicates. This occurs before the
  1004. // file extension check because we might emit both diagnostics in situations
  1005. // where the user forgets (or has syntax errors with) a package line
  1006. // multiple times.
  1007. if (!is_impl) {
  1008. auto insert_result = api_map.Insert(import_key, &unit_info);
  1009. if (!insert_result.is_inserted()) {
  1010. llvm::StringRef prev_filename =
  1011. insert_result.value()->unit->tokens->source().filename();
  1012. if (packaging) {
  1013. CARBON_DIAGNOSTIC(DuplicateLibraryApi, Error,
  1014. "Library's API previously provided by `{0}`.",
  1015. std::string);
  1016. unit_info.emitter.Emit(packaging->names.node_id, DuplicateLibraryApi,
  1017. prev_filename.str());
  1018. } else {
  1019. CARBON_DIAGNOSTIC(DuplicateMainApi, Error,
  1020. "Main//default previously provided by `{0}`.",
  1021. std::string);
  1022. // Use the invalid node because there's no node to associate with.
  1023. unit_info.emitter.Emit(Parse::NodeId::Invalid, DuplicateMainApi,
  1024. prev_filename.str());
  1025. }
  1026. }
  1027. }
  1028. // Validate file extensions. Note imports rely the packaging declaration,
  1029. // not the extension. If the input is not a regular file, for example
  1030. // because it is stdin, no filename checking is performed.
  1031. if (unit_info.unit->tokens->source().is_regular_file()) {
  1032. auto filename = unit_info.unit->tokens->source().filename();
  1033. static constexpr llvm::StringLiteral ApiExt = ".carbon";
  1034. static constexpr llvm::StringLiteral ImplExt = ".impl.carbon";
  1035. bool is_api_with_impl_ext = !is_impl && filename.ends_with(ImplExt);
  1036. auto want_ext = is_impl ? ImplExt : ApiExt;
  1037. if (is_api_with_impl_ext || !filename.ends_with(want_ext)) {
  1038. CARBON_DIAGNOSTIC(IncorrectExtension, Error,
  1039. "File extension of `{0}` required for `{1}`.",
  1040. llvm::StringLiteral, Lex::TokenKind);
  1041. auto diag = unit_info.emitter.Build(
  1042. packaging ? packaging->names.node_id : Parse::NodeId::Invalid,
  1043. IncorrectExtension, want_ext,
  1044. is_impl ? Lex::TokenKind::Impl : Lex::TokenKind::Api);
  1045. if (is_api_with_impl_ext) {
  1046. CARBON_DIAGNOSTIC(IncorrectExtensionImplNote, Note,
  1047. "File extension of `{0}` only allowed for `{1}`.",
  1048. llvm::StringLiteral, Lex::TokenKind);
  1049. diag.Note(Parse::NodeId::Invalid, IncorrectExtensionImplNote, ImplExt,
  1050. Lex::TokenKind::Impl);
  1051. }
  1052. diag.Emit();
  1053. }
  1054. }
  1055. }
  1056. return api_map;
  1057. }
  1058. auto CheckParseTrees(
  1059. llvm::MutableArrayRef<Unit> units,
  1060. llvm::MutableArrayRef<Parse::NodeLocConverter> node_converters,
  1061. bool prelude_import, llvm::raw_ostream* vlog_stream) -> void {
  1062. // UnitInfo is big due to its SmallVectors, so we default to 0 on the stack.
  1063. llvm::SmallVector<UnitInfo, 0> unit_infos;
  1064. unit_infos.reserve(units.size());
  1065. for (auto [i, unit] : llvm::enumerate(units)) {
  1066. unit_infos.emplace_back(SemIR::CheckIRId(i), unit, node_converters[i]);
  1067. }
  1068. Map<ImportKey, UnitInfo*> api_map =
  1069. BuildApiMapAndDiagnosePackaging(unit_infos);
  1070. // Mark down imports for all files.
  1071. llvm::SmallVector<UnitInfo*> ready_to_check;
  1072. ready_to_check.reserve(units.size());
  1073. for (auto& unit_info : unit_infos) {
  1074. const auto& packaging = unit_info.unit->parse_tree->packaging_decl();
  1075. if (packaging && packaging->is_impl) {
  1076. // An `impl` has an implicit import of its `api`.
  1077. auto implicit_names = packaging->names;
  1078. implicit_names.package_id = IdentifierId::Invalid;
  1079. TrackImport(api_map, nullptr, unit_info, implicit_names);
  1080. }
  1081. Map<ImportKey, Parse::NodeId> explicit_import_map;
  1082. // Add the prelude import. It's added to explicit_import_map so that it can
  1083. // conflict with an explicit import of the prelude.
  1084. IdentifierId core_ident_id =
  1085. unit_info.unit->value_stores->identifiers().Add("Core");
  1086. if (prelude_import &&
  1087. !(packaging && packaging->names.package_id == core_ident_id)) {
  1088. auto prelude_id =
  1089. unit_info.unit->value_stores->string_literal_values().Add("prelude");
  1090. TrackImport(api_map, &explicit_import_map, unit_info,
  1091. {.node_id = Parse::InvalidNodeId(),
  1092. .package_id = core_ident_id,
  1093. .library_id = prelude_id});
  1094. }
  1095. for (const auto& import : unit_info.unit->parse_tree->imports()) {
  1096. TrackImport(api_map, &explicit_import_map, unit_info, import);
  1097. }
  1098. // If there were no imports, mark the file as ready to check for below.
  1099. if (unit_info.imports_remaining == 0) {
  1100. ready_to_check.push_back(&unit_info);
  1101. }
  1102. }
  1103. // Check everything with no dependencies. Earlier entries with dependencies
  1104. // will be checked as soon as all their dependencies have been checked.
  1105. for (int check_index = 0;
  1106. check_index < static_cast<int>(ready_to_check.size()); ++check_index) {
  1107. auto* unit_info = ready_to_check[check_index];
  1108. CheckParseTree(node_converters, *unit_info, units.size(), vlog_stream);
  1109. for (auto* incoming_import : unit_info->incoming_imports) {
  1110. --incoming_import->imports_remaining;
  1111. if (incoming_import->imports_remaining == 0) {
  1112. ready_to_check.push_back(incoming_import);
  1113. }
  1114. }
  1115. }
  1116. // If there are still units with remaining imports, it means there's a
  1117. // dependency loop.
  1118. if (ready_to_check.size() < unit_infos.size()) {
  1119. // Go through units and mask out unevaluated imports. This breaks everything
  1120. // associated with a loop equivalently, whether it's part of it or depending
  1121. // on a part of it.
  1122. // TODO: Better identify cycles, maybe try to untangle them.
  1123. for (auto& unit_info : unit_infos) {
  1124. if (unit_info.imports_remaining > 0) {
  1125. for (auto& package_imports : unit_info.package_imports) {
  1126. for (auto* import_it = package_imports.imports.begin();
  1127. import_it != package_imports.imports.end();) {
  1128. if (*import_it->unit_info->unit->sem_ir) {
  1129. // The import is checked, so continue.
  1130. ++import_it;
  1131. } else {
  1132. // The import hasn't been checked, indicating a cycle.
  1133. CARBON_DIAGNOSTIC(ImportCycleDetected, Error,
  1134. "Import cannot be used due to a cycle. Cycle "
  1135. "must be fixed to import.");
  1136. unit_info.emitter.Emit(import_it->names.node_id,
  1137. ImportCycleDetected);
  1138. // Make this look the same as an import which wasn't found.
  1139. package_imports.has_load_error = true;
  1140. if (unit_info.api_for_impl == import_it->unit_info) {
  1141. unit_info.api_for_impl = nullptr;
  1142. }
  1143. import_it = package_imports.imports.erase(import_it);
  1144. }
  1145. }
  1146. }
  1147. }
  1148. }
  1149. // Check the remaining file contents, which are probably broken due to
  1150. // incomplete imports.
  1151. for (auto& unit_info : unit_infos) {
  1152. if (unit_info.imports_remaining > 0) {
  1153. CheckParseTree(node_converters, unit_info, units.size(), vlog_stream);
  1154. }
  1155. }
  1156. }
  1157. }
  1158. } // namespace Carbon::Check