xref: /llvm-project/flang/lib/Semantics/resolve-labels.cpp (revision a4bc44ad675a36c43eec9e08d5360f97ca97c389) 
1 //===-- lib/Semantics/resolve-labels.cpp ----------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "resolve-labels.h"
10 #include "flang/Common/enum-set.h"
11 #include "flang/Common/template.h"
12 #include "flang/Parser/parse-tree-visitor.h"
13 #include "flang/Semantics/semantics.h"
14 #include <cstdarg>
15 #include <type_traits>
16 
17 namespace Fortran::semantics {
18 
19 using namespace parser::literals;
20 
21 ENUM_CLASS(
22     TargetStatementEnum, Do, Branch, Format, CompatibleDo, CompatibleBranch)
23 using LabeledStmtClassificationSet =
24     common::EnumSet<TargetStatementEnum, TargetStatementEnum_enumSize>;
25 
26 using IndexList = std::vector<std::pair<parser::CharBlock, parser::CharBlock>>;
27 // A ProxyForScope is an integral proxy for a Fortran scope. This is required
28 // because the parse tree does not actually have the scopes required.
29 using ProxyForScope = unsigned;
30 // Minimal scope information
31 struct ScopeInfo {
32   ProxyForScope parent{};
33   bool isExteriorGotoFatal{false};
34   int depth{0};
35 };
36 struct LabeledStatementInfoTuplePOD {
37   ProxyForScope proxyForScope;
38   parser::CharBlock parserCharBlock;
39   LabeledStmtClassificationSet labeledStmtClassificationSet;
40   bool isExecutableConstructEndStmt;
41 };
42 using TargetStmtMap = std::map<parser::Label, LabeledStatementInfoTuplePOD>;
43 struct SourceStatementInfoTuplePOD {
44   SourceStatementInfoTuplePOD(const parser::Label &parserLabel,
45       const ProxyForScope &proxyForScope,
46       const parser::CharBlock &parserCharBlock)
47       : parserLabel{parserLabel}, proxyForScope{proxyForScope},
48         parserCharBlock{parserCharBlock} {}
49   parser::Label parserLabel;
50   ProxyForScope proxyForScope;
51   parser::CharBlock parserCharBlock;
52 };
53 using SourceStmtList = std::vector<SourceStatementInfoTuplePOD>;
54 enum class Legality { never, always, formerly };
55 
56 bool HasScope(ProxyForScope scope) { return scope != ProxyForScope{0u}; }
57 
58 // F18:R1131
59 template <typename A>
60 constexpr Legality IsLegalDoTerm(const parser::Statement<A> &) {
61   if (std::is_same_v<A, common::Indirection<parser::EndDoStmt>> ||
62       std::is_same_v<A, parser::EndDoStmt>) {
63     return Legality::always;
64   } else if (std::is_same_v<A, parser::EndForallStmt> ||
65       std::is_same_v<A, parser::EndWhereStmt>) {
66     // Executable construct end statements are also supported as
67     // an extension but they need special care because the associated
68     // construct create their own scope.
69     return Legality::formerly;
70   } else {
71     return Legality::never;
72   }
73 }
74 
75 constexpr Legality IsLegalDoTerm(
76     const parser::Statement<parser::ActionStmt> &actionStmt) {
77   if (std::holds_alternative<parser::ContinueStmt>(actionStmt.statement.u)) {
78     // See F08:C816
79     return Legality::always;
80   } else if (!(std::holds_alternative<
81                    common::Indirection<parser::ArithmeticIfStmt>>(
82                    actionStmt.statement.u) ||
83                  std::holds_alternative<common::Indirection<parser::CycleStmt>>(
84                      actionStmt.statement.u) ||
85                  std::holds_alternative<common::Indirection<parser::ExitStmt>>(
86                      actionStmt.statement.u) ||
87                  std::holds_alternative<common::Indirection<parser::StopStmt>>(
88                      actionStmt.statement.u) ||
89                  std::holds_alternative<common::Indirection<parser::GotoStmt>>(
90                      actionStmt.statement.u) ||
91                  std::holds_alternative<
92                      common::Indirection<parser::ReturnStmt>>(
93                      actionStmt.statement.u))) {
94     return Legality::formerly;
95   } else {
96     return Legality::never;
97   }
98 }
99 
100 template <typename A> constexpr bool IsFormat(const parser::Statement<A> &) {
101   return std::is_same_v<A, common::Indirection<parser::FormatStmt>>;
102 }
103 
104 template <typename A>
105 constexpr Legality IsLegalBranchTarget(const parser::Statement<A> &) {
106   if (std::is_same_v<A, parser::ActionStmt> ||
107       std::is_same_v<A, parser::AssociateStmt> ||
108       std::is_same_v<A, parser::EndAssociateStmt> ||
109       std::is_same_v<A, parser::IfThenStmt> ||
110       std::is_same_v<A, parser::EndIfStmt> ||
111       std::is_same_v<A, parser::SelectCaseStmt> ||
112       std::is_same_v<A, parser::EndSelectStmt> ||
113       std::is_same_v<A, parser::SelectRankStmt> ||
114       std::is_same_v<A, parser::SelectTypeStmt> ||
115       std::is_same_v<A, common::Indirection<parser::LabelDoStmt>> ||
116       std::is_same_v<A, parser::NonLabelDoStmt> ||
117       std::is_same_v<A, parser::EndDoStmt> ||
118       std::is_same_v<A, common::Indirection<parser::EndDoStmt>> ||
119       std::is_same_v<A, parser::BlockStmt> ||
120       std::is_same_v<A, parser::EndBlockStmt> ||
121       std::is_same_v<A, parser::CriticalStmt> ||
122       std::is_same_v<A, parser::EndCriticalStmt> ||
123       std::is_same_v<A, parser::ForallConstructStmt> ||
124       std::is_same_v<A, parser::WhereConstructStmt> ||
125       std::is_same_v<A, parser::EndFunctionStmt> ||
126       std::is_same_v<A, parser::EndMpSubprogramStmt> ||
127       std::is_same_v<A, parser::EndProgramStmt> ||
128       std::is_same_v<A, parser::EndSubroutineStmt>) {
129     return Legality::always;
130   } else {
131     return Legality::never;
132   }
133 }
134 
135 template <typename A>
136 constexpr LabeledStmtClassificationSet ConstructBranchTargetFlags(
137     const parser::Statement<A> &statement) {
138   LabeledStmtClassificationSet labeledStmtClassificationSet{};
139   if (IsLegalDoTerm(statement) == Legality::always) {
140     labeledStmtClassificationSet.set(TargetStatementEnum::Do);
141   } else if (IsLegalDoTerm(statement) == Legality::formerly) {
142     labeledStmtClassificationSet.set(TargetStatementEnum::CompatibleDo);
143   }
144   if (IsLegalBranchTarget(statement) == Legality::always) {
145     labeledStmtClassificationSet.set(TargetStatementEnum::Branch);
146   } else if (IsLegalBranchTarget(statement) == Legality::formerly) {
147     labeledStmtClassificationSet.set(TargetStatementEnum::CompatibleBranch);
148   }
149   if (IsFormat(statement)) {
150     labeledStmtClassificationSet.set(TargetStatementEnum::Format);
151   }
152   return labeledStmtClassificationSet;
153 }
154 
155 static unsigned SayLabel(parser::Label label) {
156   return static_cast<unsigned>(label);
157 }
158 
159 struct UnitAnalysis {
160   UnitAnalysis() { scopeModel.emplace_back(); }
161 
162   SourceStmtList doStmtSources;
163   SourceStmtList formatStmtSources;
164   SourceStmtList otherStmtSources;
165   SourceStmtList assignStmtSources;
166   TargetStmtMap targetStmts;
167   std::vector<ScopeInfo> scopeModel;
168 };
169 
170 // Some parse tree record for statements simply wrap construct names;
171 // others include them as tuple components.  Given a statement,
172 // return a pointer to its name if it has one.
173 template <typename A>
174 const parser::CharBlock *GetStmtName(const parser::Statement<A> &stmt) {
175   const std::optional<parser::Name> *name{nullptr};
176   if constexpr (WrapperTrait<A>) {
177     if constexpr (std::is_same_v<decltype(A::v), parser::Name>) {
178       return &stmt.statement.v.source;
179     } else {
180       name = &stmt.statement.v;
181     }
182   } else if constexpr (std::is_same_v<A, parser::SelectRankStmt> ||
183       std::is_same_v<A, parser::SelectTypeStmt>) {
184     name = &std::get<0>(stmt.statement.t);
185   } else if constexpr (common::HasMember<parser::Name,
186                            decltype(stmt.statement.t)>) {
187     return &std::get<parser::Name>(stmt.statement.t).source;
188   } else {
189     name = &std::get<std::optional<parser::Name>>(stmt.statement.t);
190   }
191   if (name && *name) {
192     return &(*name)->source;
193   }
194   return nullptr;
195 }
196 
197 class ParseTreeAnalyzer {
198 public:
199   ParseTreeAnalyzer(ParseTreeAnalyzer &&that) = default;
200   ParseTreeAnalyzer(SemanticsContext &context) : context_{context} {}
201 
202   template <typename A> constexpr bool Pre(const A &x) {
203     using LabeledProgramUnitStmts =
204         std::tuple<parser::MainProgram, parser::FunctionSubprogram,
205             parser::SubroutineSubprogram, parser::SeparateModuleSubprogram>;
206     if constexpr (common::HasMember<A, LabeledProgramUnitStmts>) {
207       const auto &endStmt{std::get<std::tuple_size_v<decltype(x.t)> - 1>(x.t)};
208       if (endStmt.label) {
209         // The END statement for a subprogram appears after any internal
210         // subprograms.  Visit that statement in advance so that results
211         // are placed in the correct programUnits_ slot.
212         auto targetFlags{ConstructBranchTargetFlags(endStmt)};
213         AddTargetLabelDefinition(
214             endStmt.label.value(), targetFlags, currentScope_);
215       }
216     }
217     return true;
218   }
219   template <typename A> constexpr void Post(const A &) {}
220 
221   template <typename A> bool Pre(const parser::Statement<A> &statement) {
222     currentPosition_ = statement.source;
223     const auto &label = statement.label;
224     if (!label) {
225       return true;
226     }
227     using LabeledConstructStmts = std::tuple<parser::AssociateStmt,
228         parser::BlockStmt, parser::ChangeTeamStmt, parser::CriticalStmt,
229         parser::IfThenStmt, parser::NonLabelDoStmt, parser::SelectCaseStmt,
230         parser::SelectRankStmt, parser::SelectTypeStmt,
231         parser::ForallConstructStmt, parser::WhereConstructStmt>;
232     using LabeledConstructEndStmts = std::tuple<parser::EndAssociateStmt,
233         parser::EndBlockStmt, parser::EndChangeTeamStmt,
234         parser::EndCriticalStmt, parser::EndDoStmt, parser::EndForallStmt,
235         parser::EndIfStmt, parser::EndWhereStmt>;
236     using LabeledProgramUnitEndStmts =
237         std::tuple<parser::EndFunctionStmt, parser::EndMpSubprogramStmt,
238             parser::EndProgramStmt, parser::EndSubroutineStmt>;
239     auto targetFlags{ConstructBranchTargetFlags(statement)};
240     if constexpr (common::HasMember<A, LabeledConstructStmts>) {
241       AddTargetLabelDefinition(label.value(), targetFlags, ParentScope());
242     } else if constexpr (std::is_same_v<A, parser::EndIfStmt> ||
243         std::is_same_v<A, parser::EndSelectStmt>) {
244       // the label on an END IF/SELECT is not in the last part/case
245       AddTargetLabelDefinition(label.value(), targetFlags, ParentScope(), true);
246     } else if constexpr (common::HasMember<A, LabeledConstructEndStmts>) {
247       constexpr bool isExecutableConstructEndStmt{true};
248       AddTargetLabelDefinition(label.value(), targetFlags, currentScope_,
249           isExecutableConstructEndStmt);
250     } else if constexpr (!common::HasMember<A, LabeledProgramUnitEndStmts>) {
251       // Program unit END statements have already been processed.
252       AddTargetLabelDefinition(label.value(), targetFlags, currentScope_);
253     }
254     return true;
255   }
256 
257   // see 11.1.1
258   bool Pre(const parser::ProgramUnit &) { return InitializeNewScopeContext(); }
259   bool Pre(const parser::InternalSubprogram &) {
260     return InitializeNewScopeContext();
261   }
262   bool Pre(const parser::ModuleSubprogram &) {
263     return InitializeNewScopeContext();
264   }
265   bool Pre(const parser::AssociateConstruct &associateConstruct) {
266     return PushConstructName(associateConstruct);
267   }
268   bool Pre(const parser::BlockConstruct &blockConstruct) {
269     return PushConstructName(blockConstruct);
270   }
271   bool Pre(const parser::ChangeTeamConstruct &changeTeamConstruct) {
272     return PushConstructName(changeTeamConstruct);
273   }
274   bool Pre(const parser::CriticalConstruct &criticalConstruct) {
275     return PushConstructName(criticalConstruct);
276   }
277   bool Pre(const parser::DoConstruct &doConstruct) {
278     const auto &optionalName{std::get<std::optional<parser::Name>>(
279         std::get<parser::Statement<parser::NonLabelDoStmt>>(doConstruct.t)
280             .statement.t)};
281     if (optionalName) {
282       constructNames_.emplace_back(optionalName->ToString());
283     }
284     // Allow FORTRAN '66 extended DO ranges
285     PushScope(false);
286     // Process labels of the DO and END DO statements, but not the
287     // statements themselves, so that a non-construct END DO
288     // can be distinguished (below).
289     Pre(std::get<parser::Statement<parser::NonLabelDoStmt>>(doConstruct.t));
290     Walk(std::get<parser::Block>(doConstruct.t), *this);
291     Pre(std::get<parser::Statement<parser::EndDoStmt>>(doConstruct.t));
292     PopConstructName(doConstruct);
293     return false;
294   }
295   void Post(const parser::EndDoStmt &endDoStmt) {
296     // Visited only for non-construct labeled DO termination
297     if (const auto &name{endDoStmt.v}) {
298       context_.Say(name->source, "Unexpected DO construct name '%s'"_err_en_US,
299           name->source);
300     }
301   }
302   bool Pre(const parser::IfConstruct &ifConstruct) {
303     return PushConstructName(ifConstruct);
304   }
305   void Post(const parser::IfThenStmt &) { PushScope(false); }
306   bool Pre(const parser::IfConstruct::ElseIfBlock &) {
307     return SwitchToNewScope();
308   }
309   bool Pre(const parser::IfConstruct::ElseBlock &) {
310     return SwitchToNewScope();
311   }
312   bool Pre(const parser::EndIfStmt &) {
313     PopScope();
314     return true;
315   }
316   bool Pre(const parser::CaseConstruct &caseConstruct) {
317     return PushConstructName(caseConstruct);
318   }
319   void Post(const parser::SelectCaseStmt &) { PushScope(false); }
320   bool Pre(const parser::CaseConstruct::Case &) { return SwitchToNewScope(); }
321   bool Pre(const parser::SelectRankConstruct &selectRankConstruct) {
322     return PushConstructName(selectRankConstruct);
323   }
324   void Post(const parser::SelectRankStmt &) { PushScope(true); }
325   bool Pre(const parser::SelectRankConstruct::RankCase &) {
326     return SwitchToNewScope();
327   }
328   bool Pre(const parser::SelectTypeConstruct &selectTypeConstruct) {
329     return PushConstructName(selectTypeConstruct);
330   }
331   void Post(const parser::SelectTypeStmt &) { PushScope(true); }
332   bool Pre(const parser::SelectTypeConstruct::TypeCase &) {
333     return SwitchToNewScope();
334   }
335   void Post(const parser::EndSelectStmt &) { PopScope(); }
336   bool Pre(const parser::WhereConstruct &whereConstruct) {
337     return PushConstructName(whereConstruct);
338   }
339   bool Pre(const parser::ForallConstruct &forallConstruct) {
340     return PushConstructName(forallConstruct);
341   }
342 
343   void Post(const parser::AssociateConstruct &associateConstruct) {
344     PopConstructName(associateConstruct);
345   }
346   void Post(const parser::BlockConstruct &blockConstruct) {
347     PopConstructName(blockConstruct);
348   }
349   void Post(const parser::ChangeTeamConstruct &changeTeamConstruct) {
350     PopConstructName(changeTeamConstruct);
351   }
352   void Post(const parser::CriticalConstruct &criticalConstruct) {
353     PopConstructName(criticalConstruct);
354   }
355   void Post(const parser::IfConstruct &ifConstruct) {
356     PopConstructName(ifConstruct);
357   }
358   void Post(const parser::CaseConstruct &caseConstruct) {
359     PopConstructName(caseConstruct);
360   }
361   void Post(const parser::SelectRankConstruct &selectRankConstruct) {
362     PopConstructName(selectRankConstruct);
363   }
364   void Post(const parser::SelectTypeConstruct &selectTypeConstruct) {
365     PopConstructName(selectTypeConstruct);
366   }
367   void Post(const parser::WhereConstruct &whereConstruct) {
368     PopConstructName(whereConstruct);
369   }
370   void Post(const parser::ForallConstruct &forallConstruct) {
371     PopConstructName(forallConstruct);
372   }
373 
374   // Checks for missing or mismatching names on various constructs (e.g., IF)
375   // and their intermediate or terminal statements that allow optional
376   // construct names(e.g., ELSE).  When an optional construct name is present,
377   // the construct as a whole must have a name that matches.
378   template <typename FIRST, typename CONSTRUCT, typename STMT>
379   void CheckOptionalName(const char *constructTag, const CONSTRUCT &a,
380       const parser::Statement<STMT> &stmt) {
381     if (const parser::CharBlock * name{GetStmtName(stmt)}) {
382       const auto &firstStmt{std::get<parser::Statement<FIRST>>(a.t)};
383       if (const parser::CharBlock * firstName{GetStmtName(firstStmt)}) {
384         if (*firstName != *name) {
385           context_.Say(*name, "%s name mismatch"_err_en_US, constructTag)
386               .Attach(*firstName, "should be"_en_US);
387         }
388       } else {
389         context_.Say(*name, "%s name not allowed"_err_en_US, constructTag)
390             .Attach(firstStmt.source, "in unnamed %s"_en_US, constructTag);
391       }
392     }
393   }
394 
395   // C1414
396   void Post(const parser::BlockData &blockData) {
397     CheckOptionalName<parser::BlockDataStmt>("BLOCK DATA subprogram", blockData,
398         std::get<parser::Statement<parser::EndBlockDataStmt>>(blockData.t));
399   }
400 
401   bool Pre(const parser::InterfaceBody &) {
402     PushDisposableMap();
403     return true;
404   }
405   void Post(const parser::InterfaceBody &) { PopDisposableMap(); }
406 
407   // C1564
408   void Post(const parser::InterfaceBody::Function &func) {
409     CheckOptionalName<parser::FunctionStmt>("FUNCTION", func,
410         std::get<parser::Statement<parser::EndFunctionStmt>>(func.t));
411   }
412 
413   // C1564
414   void Post(const parser::FunctionSubprogram &functionSubprogram) {
415     CheckOptionalName<parser::FunctionStmt>("FUNCTION", functionSubprogram,
416         std::get<parser::Statement<parser::EndFunctionStmt>>(
417             functionSubprogram.t));
418   }
419 
420   // C1502
421   void Post(const parser::InterfaceBlock &interfaceBlock) {
422     if (const auto &endGenericSpec{
423             std::get<parser::Statement<parser::EndInterfaceStmt>>(
424                 interfaceBlock.t)
425                 .statement.v}) {
426       const auto &interfaceStmt{
427           std::get<parser::Statement<parser::InterfaceStmt>>(interfaceBlock.t)};
428       if (std::holds_alternative<parser::Abstract>(interfaceStmt.statement.u)) {
429         context_
430             .Say(endGenericSpec->source,
431                 "END INTERFACE generic name (%s) may not appear for ABSTRACT INTERFACE"_err_en_US,
432                 endGenericSpec->source)
433             .Attach(
434                 interfaceStmt.source, "corresponding ABSTRACT INTERFACE"_en_US);
435       } else if (const auto &genericSpec{
436                      std::get<std::optional<parser::GenericSpec>>(
437                          interfaceStmt.statement.u)}) {
438         bool ok{genericSpec->source == endGenericSpec->source};
439         if (!ok) {
440           // Accept variant spellings of .LT. &c.
441           const auto *endOp{
442               std::get_if<parser::DefinedOperator>(&endGenericSpec->u)};
443           const auto *op{std::get_if<parser::DefinedOperator>(&genericSpec->u)};
444           if (endOp && op) {
445             const auto *endIntrin{
446                 std::get_if<parser::DefinedOperator::IntrinsicOperator>(
447                     &endOp->u)};
448             const auto *intrin{
449                 std::get_if<parser::DefinedOperator::IntrinsicOperator>(
450                     &op->u)};
451             ok = endIntrin && intrin && *endIntrin == *intrin;
452           }
453         }
454         if (!ok) {
455           context_
456               .Say(endGenericSpec->source,
457                   "END INTERFACE generic name (%s) does not match generic INTERFACE (%s)"_err_en_US,
458                   endGenericSpec->source, genericSpec->source)
459               .Attach(genericSpec->source, "corresponding INTERFACE"_en_US);
460         }
461       } else {
462         context_
463             .Say(endGenericSpec->source,
464                 "END INTERFACE generic name (%s) may not appear for non-generic INTERFACE"_err_en_US,
465                 endGenericSpec->source)
466             .Attach(interfaceStmt.source, "corresponding INTERFACE"_en_US);
467       }
468     }
469   }
470 
471   // C1402
472   void Post(const parser::Module &module) {
473     CheckOptionalName<parser::ModuleStmt>("MODULE", module,
474         std::get<parser::Statement<parser::EndModuleStmt>>(module.t));
475   }
476 
477   // C1569
478   void Post(const parser::SeparateModuleSubprogram &separateModuleSubprogram) {
479     CheckOptionalName<parser::MpSubprogramStmt>("MODULE PROCEDURE",
480         separateModuleSubprogram,
481         std::get<parser::Statement<parser::EndMpSubprogramStmt>>(
482             separateModuleSubprogram.t));
483   }
484 
485   // C1401
486   void Post(const parser::MainProgram &mainProgram) {
487     if (const parser::CharBlock *
488         endName{GetStmtName(std::get<parser::Statement<parser::EndProgramStmt>>(
489             mainProgram.t))}) {
490       if (const auto &program{
491               std::get<std::optional<parser::Statement<parser::ProgramStmt>>>(
492                   mainProgram.t)}) {
493         if (*endName != program->statement.v.source) {
494           context_.Say(*endName, "END PROGRAM name mismatch"_err_en_US)
495               .Attach(program->statement.v.source, "should be"_en_US);
496         }
497       } else {
498         context_.Say(*endName,
499             "END PROGRAM has name without PROGRAM statement"_err_en_US);
500       }
501     }
502   }
503 
504   // C1413
505   void Post(const parser::Submodule &submodule) {
506     CheckOptionalName<parser::SubmoduleStmt>("SUBMODULE", submodule,
507         std::get<parser::Statement<parser::EndSubmoduleStmt>>(submodule.t));
508   }
509 
510   // C1567
511   void Post(const parser::InterfaceBody::Subroutine &sub) {
512     CheckOptionalName<parser::SubroutineStmt>("SUBROUTINE", sub,
513         std::get<parser::Statement<parser::EndSubroutineStmt>>(sub.t));
514   }
515 
516   // C1567
517   void Post(const parser::SubroutineSubprogram &subroutineSubprogram) {
518     CheckOptionalName<parser::SubroutineStmt>("SUBROUTINE",
519         subroutineSubprogram,
520         std::get<parser::Statement<parser::EndSubroutineStmt>>(
521             subroutineSubprogram.t));
522   }
523 
524   // C739
525   bool Pre(const parser::DerivedTypeDef &) {
526     PushDisposableMap();
527     return true;
528   }
529   void Post(const parser::DerivedTypeDef &derivedTypeDef) {
530     CheckOptionalName<parser::DerivedTypeStmt>("derived type definition",
531         derivedTypeDef,
532         std::get<parser::Statement<parser::EndTypeStmt>>(derivedTypeDef.t));
533     PopDisposableMap();
534   }
535 
536   void Post(const parser::LabelDoStmt &labelDoStmt) {
537     AddLabelReferenceFromDoStmt(std::get<parser::Label>(labelDoStmt.t));
538   }
539   void Post(const parser::GotoStmt &gotoStmt) { AddLabelReference(gotoStmt.v); }
540   void Post(const parser::ComputedGotoStmt &computedGotoStmt) {
541     AddLabelReference(std::get<std::list<parser::Label>>(computedGotoStmt.t));
542   }
543   void Post(const parser::ArithmeticIfStmt &arithmeticIfStmt) {
544     AddLabelReference(std::get<1>(arithmeticIfStmt.t));
545     AddLabelReference(std::get<2>(arithmeticIfStmt.t));
546     AddLabelReference(std::get<3>(arithmeticIfStmt.t));
547   }
548   void Post(const parser::AssignStmt &assignStmt) {
549     AddLabelReferenceFromAssignStmt(std::get<parser::Label>(assignStmt.t));
550   }
551   void Post(const parser::AssignedGotoStmt &assignedGotoStmt) {
552     AddLabelReference(std::get<std::list<parser::Label>>(assignedGotoStmt.t));
553   }
554   void Post(const parser::AltReturnSpec &altReturnSpec) {
555     AddLabelReference(altReturnSpec.v);
556   }
557 
558   void Post(const parser::ErrLabel &errLabel) { AddLabelReference(errLabel.v); }
559   void Post(const parser::EndLabel &endLabel) { AddLabelReference(endLabel.v); }
560   void Post(const parser::EorLabel &eorLabel) { AddLabelReference(eorLabel.v); }
561   void Post(const parser::Format &format) {
562     if (const auto *labelPointer{std::get_if<parser::Label>(&format.u)}) {
563       AddLabelReferenceToFormatStmt(*labelPointer);
564     }
565   }
566   void Post(const parser::CycleStmt &cycleStmt) {
567     if (cycleStmt.v) {
568       CheckLabelContext("CYCLE", cycleStmt.v->source);
569     }
570   }
571   void Post(const parser::ExitStmt &exitStmt) {
572     if (exitStmt.v) {
573       CheckLabelContext("EXIT", exitStmt.v->source);
574     }
575   }
576 
577   const std::vector<UnitAnalysis> &ProgramUnits() const {
578     return programUnits_;
579   }
580   SemanticsContext &ErrorHandler() { return context_; }
581 
582 private:
583   ScopeInfo &PushScope(bool isExteriorGotoFatal) {
584     auto &model{programUnits_.back().scopeModel};
585     int newDepth{model.empty() ? 1 : model[currentScope_].depth + 1};
586     ScopeInfo &result{model.emplace_back()};
587     result.parent = currentScope_;
588     result.depth = newDepth;
589     result.isExteriorGotoFatal = isExteriorGotoFatal;
590     currentScope_ = model.size() - 1;
591     return result;
592   }
593   bool InitializeNewScopeContext() {
594     programUnits_.emplace_back(UnitAnalysis{});
595     currentScope_ = 0u;
596     PushScope(false);
597     return true;
598   }
599   ScopeInfo &PopScope() {
600     ScopeInfo &result{programUnits_.back().scopeModel[currentScope_]};
601     currentScope_ = result.parent;
602     return result;
603   }
604   ProxyForScope ParentScope() {
605     return programUnits_.back().scopeModel[currentScope_].parent;
606   }
607   bool SwitchToNewScope() {
608     PushScope(PopScope().isExteriorGotoFatal);
609     return true;
610   }
611 
612   template <typename A> bool PushConstructName(const A &a) {
613     const auto &optionalName{std::get<0>(std::get<0>(a.t).statement.t)};
614     if (optionalName) {
615       constructNames_.emplace_back(optionalName->ToString());
616     }
617     // Gotos into this construct from outside it are diagnosed, and
618     // are fatal unless the construct is a DO, IF, or SELECT CASE.
619     PushScope(!(std::is_same_v<A, parser::DoConstruct> ||
620         std::is_same_v<A, parser::IfConstruct> ||
621         std::is_same_v<A, parser::CaseConstruct>));
622     return true;
623   }
624   bool PushConstructName(const parser::BlockConstruct &blockConstruct) {
625     const auto &optionalName{
626         std::get<parser::Statement<parser::BlockStmt>>(blockConstruct.t)
627             .statement.v};
628     if (optionalName) {
629       constructNames_.emplace_back(optionalName->ToString());
630     }
631     PushScope(true);
632     return true;
633   }
634   template <typename A> void PopConstructNameIfPresent(const A &a) {
635     const auto &optionalName{std::get<0>(std::get<0>(a.t).statement.t)};
636     if (optionalName) {
637       constructNames_.pop_back();
638     }
639   }
640   void PopConstructNameIfPresent(const parser::BlockConstruct &blockConstruct) {
641     const auto &optionalName{
642         std::get<parser::Statement<parser::BlockStmt>>(blockConstruct.t)
643             .statement.v};
644     if (optionalName) {
645       constructNames_.pop_back();
646     }
647   }
648 
649   template <typename A> void PopConstructName(const A &a) {
650     CheckName(a);
651     PopScope();
652     PopConstructNameIfPresent(a);
653   }
654 
655   template <typename FIRST, typename CASEBLOCK, typename CASE,
656       typename CONSTRUCT>
657   void CheckSelectNames(const char *tag, const CONSTRUCT &construct) {
658     CheckEndName<FIRST, parser::EndSelectStmt>(tag, construct);
659     for (const auto &inner : std::get<std::list<CASEBLOCK>>(construct.t)) {
660       CheckOptionalName<FIRST>(
661           tag, construct, std::get<parser::Statement<CASE>>(inner.t));
662     }
663   }
664 
665   // C1144
666   void PopConstructName(const parser::CaseConstruct &caseConstruct) {
667     CheckSelectNames<parser::SelectCaseStmt, parser::CaseConstruct::Case,
668         parser::CaseStmt>("SELECT CASE", caseConstruct);
669     PopScope();
670     PopConstructNameIfPresent(caseConstruct);
671   }
672 
673   // C1154, C1156
674   void PopConstructName(
675       const parser::SelectRankConstruct &selectRankConstruct) {
676     CheckSelectNames<parser::SelectRankStmt,
677         parser::SelectRankConstruct::RankCase, parser::SelectRankCaseStmt>(
678         "SELECT RANK", selectRankConstruct);
679     PopScope();
680     PopConstructNameIfPresent(selectRankConstruct);
681   }
682 
683   // C1165
684   void PopConstructName(
685       const parser::SelectTypeConstruct &selectTypeConstruct) {
686     CheckSelectNames<parser::SelectTypeStmt,
687         parser::SelectTypeConstruct::TypeCase, parser::TypeGuardStmt>(
688         "SELECT TYPE", selectTypeConstruct);
689     PopScope();
690     PopConstructNameIfPresent(selectTypeConstruct);
691   }
692 
693   // Checks for missing or mismatching names on various constructs (e.g., BLOCK)
694   // and their END statements.  Both names must be present if either one is.
695   template <typename FIRST, typename END, typename CONSTRUCT>
696   void CheckEndName(const char *constructTag, const CONSTRUCT &a) {
697     const auto &constructStmt{std::get<parser::Statement<FIRST>>(a.t)};
698     const auto &endStmt{std::get<parser::Statement<END>>(a.t)};
699     const parser::CharBlock *endName{GetStmtName(endStmt)};
700     if (const parser::CharBlock * constructName{GetStmtName(constructStmt)}) {
701       if (endName) {
702         if (*constructName != *endName) {
703           context_
704               .Say(*endName, "%s construct name mismatch"_err_en_US,
705                   constructTag)
706               .Attach(*constructName, "should be"_en_US);
707         }
708       } else {
709         context_
710             .Say(endStmt.source,
711                 "%s construct name required but missing"_err_en_US,
712                 constructTag)
713             .Attach(*constructName, "should be"_en_US);
714       }
715     } else if (endName) {
716       context_
717           .Say(*endName, "%s construct name unexpected"_err_en_US, constructTag)
718           .Attach(
719               constructStmt.source, "unnamed %s statement"_en_US, constructTag);
720     }
721   }
722 
723   // C1106
724   void CheckName(const parser::AssociateConstruct &associateConstruct) {
725     CheckEndName<parser::AssociateStmt, parser::EndAssociateStmt>(
726         "ASSOCIATE", associateConstruct);
727   }
728   // C1117
729   void CheckName(const parser::CriticalConstruct &criticalConstruct) {
730     CheckEndName<parser::CriticalStmt, parser::EndCriticalStmt>(
731         "CRITICAL", criticalConstruct);
732   }
733   // C1131
734   void CheckName(const parser::DoConstruct &doConstruct) {
735     CheckEndName<parser::NonLabelDoStmt, parser::EndDoStmt>("DO", doConstruct);
736     if (auto label{std::get<std::optional<parser::Label>>(
737             std::get<parser::Statement<parser::NonLabelDoStmt>>(doConstruct.t)
738                 .statement.t)}) {
739       const auto &endDoStmt{
740           std::get<parser::Statement<parser::EndDoStmt>>(doConstruct.t)};
741       if (!endDoStmt.label || *endDoStmt.label != *label) {
742         context_
743             .Say(endDoStmt.source,
744                 "END DO statement must have the label '%d' matching its DO statement"_err_en_US,
745                 *label)
746             .Attach(std::get<parser::Statement<parser::NonLabelDoStmt>>(
747                         doConstruct.t)
748                         .source,
749                 "corresponding DO statement"_en_US);
750       }
751     }
752   }
753   // C1035
754   void CheckName(const parser::ForallConstruct &forallConstruct) {
755     CheckEndName<parser::ForallConstructStmt, parser::EndForallStmt>(
756         "FORALL", forallConstruct);
757   }
758 
759   // C1109
760   void CheckName(const parser::BlockConstruct &blockConstruct) {
761     CheckEndName<parser::BlockStmt, parser::EndBlockStmt>(
762         "BLOCK", blockConstruct);
763   }
764   // C1112
765   void CheckName(const parser::ChangeTeamConstruct &changeTeamConstruct) {
766     CheckEndName<parser::ChangeTeamStmt, parser::EndChangeTeamStmt>(
767         "CHANGE TEAM", changeTeamConstruct);
768   }
769 
770   // C1142
771   void CheckName(const parser::IfConstruct &ifConstruct) {
772     CheckEndName<parser::IfThenStmt, parser::EndIfStmt>("IF", ifConstruct);
773     for (const auto &elseIfBlock :
774         std::get<std::list<parser::IfConstruct::ElseIfBlock>>(ifConstruct.t)) {
775       CheckOptionalName<parser::IfThenStmt>("IF construct", ifConstruct,
776           std::get<parser::Statement<parser::ElseIfStmt>>(elseIfBlock.t));
777     }
778     if (const auto &elseBlock{
779             std::get<std::optional<parser::IfConstruct::ElseBlock>>(
780                 ifConstruct.t)}) {
781       CheckOptionalName<parser::IfThenStmt>("IF construct", ifConstruct,
782           std::get<parser::Statement<parser::ElseStmt>>(elseBlock->t));
783     }
784   }
785 
786   // C1033
787   void CheckName(const parser::WhereConstruct &whereConstruct) {
788     CheckEndName<parser::WhereConstructStmt, parser::EndWhereStmt>(
789         "WHERE", whereConstruct);
790     for (const auto &maskedElsewhere :
791         std::get<std::list<parser::WhereConstruct::MaskedElsewhere>>(
792             whereConstruct.t)) {
793       CheckOptionalName<parser::WhereConstructStmt>("WHERE construct",
794           whereConstruct,
795           std::get<parser::Statement<parser::MaskedElsewhereStmt>>(
796               maskedElsewhere.t));
797     }
798     if (const auto &elsewhere{
799             std::get<std::optional<parser::WhereConstruct::Elsewhere>>(
800                 whereConstruct.t)}) {
801       CheckOptionalName<parser::WhereConstructStmt>("WHERE construct",
802           whereConstruct,
803           std::get<parser::Statement<parser::ElsewhereStmt>>(elsewhere->t));
804     }
805   }
806 
807   // C1134, C1166
808   void CheckLabelContext(
809       const char *const stmtString, const parser::CharBlock &constructName) {
810     const auto iter{std::find(constructNames_.crbegin(),
811         constructNames_.crend(), constructName.ToString())};
812     if (iter == constructNames_.crend()) {
813       context_.Say(constructName, "%s construct-name is not in scope"_err_en_US,
814           stmtString);
815     }
816   }
817 
818   // 6.2.5, paragraph 2
819   void CheckLabelInRange(parser::Label label) {
820     if (label < 1 || label > 99999) {
821       context_.Say(currentPosition_, "Label '%u' is out of range"_err_en_US,
822           SayLabel(label));
823     }
824   }
825 
826   // 6.2.5., paragraph 2
827   void AddTargetLabelDefinition(parser::Label label,
828       LabeledStmtClassificationSet labeledStmtClassificationSet,
829       ProxyForScope scope, bool isExecutableConstructEndStmt = false) {
830     CheckLabelInRange(label);
831     TargetStmtMap &targetStmtMap{disposableMaps_.empty()
832             ? programUnits_.back().targetStmts
833             : disposableMaps_.back()};
834     const auto pair{targetStmtMap.emplace(label,
835         LabeledStatementInfoTuplePOD{scope, currentPosition_,
836             labeledStmtClassificationSet, isExecutableConstructEndStmt})};
837     if (!pair.second) {
838       context_.Say(currentPosition_, "Label '%u' is not distinct"_err_en_US,
839           SayLabel(label));
840     }
841   }
842 
843   void AddLabelReferenceFromDoStmt(parser::Label label) {
844     CheckLabelInRange(label);
845     programUnits_.back().doStmtSources.emplace_back(
846         label, currentScope_, currentPosition_);
847   }
848 
849   void AddLabelReferenceToFormatStmt(parser::Label label) {
850     CheckLabelInRange(label);
851     programUnits_.back().formatStmtSources.emplace_back(
852         label, currentScope_, currentPosition_);
853   }
854 
855   void AddLabelReferenceFromAssignStmt(parser::Label label) {
856     CheckLabelInRange(label);
857     programUnits_.back().assignStmtSources.emplace_back(
858         label, currentScope_, currentPosition_);
859   }
860 
861   void AddLabelReference(parser::Label label) {
862     CheckLabelInRange(label);
863     programUnits_.back().otherStmtSources.emplace_back(
864         label, currentScope_, currentPosition_);
865   }
866 
867   void AddLabelReference(const std::list<parser::Label> &labels) {
868     for (const parser::Label &label : labels) {
869       AddLabelReference(label);
870     }
871   }
872 
873   void PushDisposableMap() { disposableMaps_.emplace_back(); }
874   void PopDisposableMap() { disposableMaps_.pop_back(); }
875 
876   std::vector<UnitAnalysis> programUnits_;
877   SemanticsContext &context_;
878   parser::CharBlock currentPosition_;
879   ProxyForScope currentScope_;
880   std::vector<std::string> constructNames_;
881   // For labels in derived type definitions and procedure
882   // interfaces, which are their own inclusive scopes.  None
883   // of these labels can be used as a branch target, but they
884   // should be pairwise distinct.
885   std::vector<TargetStmtMap> disposableMaps_;
886 };
887 
888 bool InInclusiveScope(const std::vector<ScopeInfo> &scopes, ProxyForScope tail,
889     ProxyForScope head) {
890   for (; tail != head; tail = scopes[tail].parent) {
891     if (!HasScope(tail)) {
892       return false;
893     }
894   }
895   return true;
896 }
897 
898 ParseTreeAnalyzer LabelAnalysis(
899     SemanticsContext &context, const parser::Program &program) {
900   ParseTreeAnalyzer analysis{context};
901   Walk(program, analysis);
902   return analysis;
903 }
904 
905 bool InBody(const parser::CharBlock &position,
906     const std::pair<parser::CharBlock, parser::CharBlock> &pair) {
907   if (position.begin() >= pair.first.begin()) {
908     if (position.begin() < pair.second.end()) {
909       return true;
910     }
911   }
912   return false;
913 }
914 
915 LabeledStatementInfoTuplePOD GetLabel(
916     const TargetStmtMap &labels, const parser::Label &label) {
917   const auto iter{labels.find(label)};
918   if (iter == labels.cend()) {
919     return {0u, nullptr, LabeledStmtClassificationSet{}, false};
920   } else {
921     return iter->second;
922   }
923 }
924 
925 // 11.1.7.3
926 void CheckBranchesIntoDoBody(const SourceStmtList &branches,
927     const TargetStmtMap &labels, const IndexList &loopBodies,
928     SemanticsContext &context) {
929   for (const auto &branch : branches) {
930     const auto &label{branch.parserLabel};
931     auto branchTarget{GetLabel(labels, label)};
932     if (HasScope(branchTarget.proxyForScope)) {
933       const auto &fromPosition{branch.parserCharBlock};
934       const auto &toPosition{branchTarget.parserCharBlock};
935       for (const auto &body : loopBodies) {
936         if (!InBody(fromPosition, body) && InBody(toPosition, body) &&
937             context.ShouldWarn(common::LanguageFeature::BranchIntoConstruct)) {
938           context
939               .Say(
940                   fromPosition, "branch into loop body from outside"_warn_en_US)
941               .Attach(body.first, "the loop branched into"_en_US);
942         }
943       }
944     }
945   }
946 }
947 
948 void CheckDoNesting(const IndexList &loopBodies, SemanticsContext &context) {
949   for (auto i1{loopBodies.cbegin()}; i1 != loopBodies.cend(); ++i1) {
950     const auto &v1{*i1};
951     for (auto i2{i1 + 1}; i2 != loopBodies.cend(); ++i2) {
952       const auto &v2{*i2};
953       if (v2.first.begin() < v1.second.end() &&
954           v1.second.begin() < v2.second.begin()) {
955         context.Say(v1.first, "DO loop doesn't properly nest"_err_en_US)
956             .Attach(v2.first, "DO loop conflicts"_en_US);
957       }
958     }
959   }
960 }
961 
962 parser::CharBlock SkipLabel(const parser::CharBlock &position) {
963   const std::size_t maxPosition{position.size()};
964   if (maxPosition && parser::IsDecimalDigit(position[0])) {
965     std::size_t i{1l};
966     for (; (i < maxPosition) && parser::IsDecimalDigit(position[i]); ++i) {
967     }
968     for (; (i < maxPosition) && parser::IsWhiteSpace(position[i]); ++i) {
969     }
970     return parser::CharBlock{position.begin() + i, position.end()};
971   }
972   return position;
973 }
974 
975 ProxyForScope ParentScope(
976     const std::vector<ScopeInfo> &scopes, ProxyForScope scope) {
977   return scopes[scope].parent;
978 }
979 
980 void CheckLabelDoConstraints(const SourceStmtList &dos,
981     const SourceStmtList &branches, const TargetStmtMap &labels,
982     const std::vector<ScopeInfo> &scopes, SemanticsContext &context) {
983   IndexList loopBodies;
984   for (const auto &stmt : dos) {
985     const auto &label{stmt.parserLabel};
986     const auto &scope{stmt.proxyForScope};
987     const auto &position{stmt.parserCharBlock};
988     auto doTarget{GetLabel(labels, label)};
989     if (!HasScope(doTarget.proxyForScope)) {
990       // C1133
991       context.Say(
992           position, "Label '%u' cannot be found"_err_en_US, SayLabel(label));
993     } else if (doTarget.parserCharBlock.begin() < position.begin()) {
994       // R1119
995       context.Say(position,
996           "Label '%u' doesn't lexically follow DO stmt"_err_en_US,
997           SayLabel(label));
998 
999     } else if ((InInclusiveScope(scopes, scope, doTarget.proxyForScope) &&
1000                    doTarget.labeledStmtClassificationSet.test(
1001                        TargetStatementEnum::CompatibleDo)) ||
1002         (doTarget.isExecutableConstructEndStmt &&
1003             ParentScope(scopes, doTarget.proxyForScope) == scope)) {
1004       if (context.ShouldWarn(
1005               common::LanguageFeature::OldLabelDoEndStatements)) {
1006         context
1007             .Say(position,
1008                 "A DO loop should terminate with an END DO or CONTINUE"_port_en_US)
1009             .Attach(doTarget.parserCharBlock,
1010                 "DO loop currently ends at statement:"_en_US);
1011       }
1012     } else if (!InInclusiveScope(scopes, scope, doTarget.proxyForScope)) {
1013       context.Say(position, "Label '%u' is not in DO loop scope"_err_en_US,
1014           SayLabel(label));
1015     } else if (!doTarget.labeledStmtClassificationSet.test(
1016                    TargetStatementEnum::Do)) {
1017       context.Say(doTarget.parserCharBlock,
1018           "A DO loop should terminate with an END DO or CONTINUE"_err_en_US);
1019     } else {
1020       loopBodies.emplace_back(SkipLabel(position), doTarget.parserCharBlock);
1021     }
1022   }
1023 
1024   CheckBranchesIntoDoBody(branches, labels, loopBodies, context);
1025   CheckDoNesting(loopBodies, context);
1026 }
1027 
1028 // 6.2.5
1029 void CheckScopeConstraints(const SourceStmtList &stmts,
1030     const TargetStmtMap &labels, const std::vector<ScopeInfo> &scopes,
1031     SemanticsContext &context) {
1032   for (const auto &stmt : stmts) {
1033     const auto &label{stmt.parserLabel};
1034     const auto &scope{stmt.proxyForScope};
1035     const auto &position{stmt.parserCharBlock};
1036     auto target{GetLabel(labels, label)};
1037     if (!HasScope(target.proxyForScope)) {
1038       context.Say(
1039           position, "Label '%u' was not found"_err_en_US, SayLabel(label));
1040     } else if (!InInclusiveScope(scopes, scope, target.proxyForScope)) {
1041       // Clause 11.1.2.1 prohibits transfer of control to the interior of a
1042       // block from outside the block, but this does not apply to formats.
1043       // C1038 and C1034 forbid statements in FORALL and WHERE constructs
1044       // (resp.) from being branch targets.
1045       if (target.labeledStmtClassificationSet.test(
1046               TargetStatementEnum::Format)) {
1047         continue;
1048       }
1049       bool isFatal{false};
1050       ProxyForScope fromScope{scope};
1051       for (ProxyForScope toScope{target.proxyForScope}; HasScope(toScope);
1052            toScope = scopes[toScope].parent) {
1053         while (scopes[fromScope].depth > scopes[toScope].depth) {
1054           fromScope = scopes[fromScope].parent;
1055         }
1056         if (toScope == fromScope) {
1057           break;
1058         }
1059         if (scopes[toScope].isExteriorGotoFatal) {
1060           isFatal = true;
1061           break;
1062         }
1063       }
1064       if (isFatal) {
1065         context.Say(position,
1066             "Label '%u' is in a construct that prevents its use as a branch target here"_err_en_US,
1067             SayLabel(label));
1068       } else if (context.ShouldWarn(
1069                      common::LanguageFeature::BranchIntoConstruct)) {
1070         context.Say(position,
1071             "Label '%u' is in a construct that should not be used as a branch target here"_warn_en_US,
1072             SayLabel(label));
1073       }
1074     }
1075   }
1076 }
1077 
1078 void CheckBranchTargetConstraints(const SourceStmtList &stmts,
1079     const TargetStmtMap &labels, SemanticsContext &context) {
1080   for (const auto &stmt : stmts) {
1081     const auto &label{stmt.parserLabel};
1082     auto branchTarget{GetLabel(labels, label)};
1083     if (HasScope(branchTarget.proxyForScope)) {
1084       if (!branchTarget.labeledStmtClassificationSet.test(
1085               TargetStatementEnum::Branch) &&
1086           !branchTarget.labeledStmtClassificationSet.test(
1087               TargetStatementEnum::CompatibleBranch)) { // error
1088         context
1089             .Say(branchTarget.parserCharBlock,
1090                 "Label '%u' is not a branch target"_err_en_US, SayLabel(label))
1091             .Attach(stmt.parserCharBlock, "Control flow use of '%u'"_en_US,
1092                 SayLabel(label));
1093       } else if (!branchTarget.labeledStmtClassificationSet.test(
1094                      TargetStatementEnum::Branch) &&
1095           context.ShouldWarn(common::LanguageFeature::BadBranchTarget)) {
1096         context
1097             .Say(branchTarget.parserCharBlock,
1098                 "Label '%u' is not a branch target"_warn_en_US, SayLabel(label))
1099             .Attach(stmt.parserCharBlock, "Control flow use of '%u'"_en_US,
1100                 SayLabel(label));
1101       }
1102     }
1103   }
1104 }
1105 
1106 void CheckBranchConstraints(const SourceStmtList &branches,
1107     const TargetStmtMap &labels, const std::vector<ScopeInfo> &scopes,
1108     SemanticsContext &context) {
1109   CheckScopeConstraints(branches, labels, scopes, context);
1110   CheckBranchTargetConstraints(branches, labels, context);
1111 }
1112 
1113 void CheckDataXferTargetConstraints(const SourceStmtList &stmts,
1114     const TargetStmtMap &labels, SemanticsContext &context) {
1115   for (const auto &stmt : stmts) {
1116     const auto &label{stmt.parserLabel};
1117     auto ioTarget{GetLabel(labels, label)};
1118     if (HasScope(ioTarget.proxyForScope)) {
1119       if (!ioTarget.labeledStmtClassificationSet.test(
1120               TargetStatementEnum::Format)) {
1121         context
1122             .Say(ioTarget.parserCharBlock, "'%u' not a FORMAT"_err_en_US,
1123                 SayLabel(label))
1124             .Attach(stmt.parserCharBlock, "data transfer use of '%u'"_en_US,
1125                 SayLabel(label));
1126       }
1127     }
1128   }
1129 }
1130 
1131 void CheckDataTransferConstraints(const SourceStmtList &dataTransfers,
1132     const TargetStmtMap &labels, const std::vector<ScopeInfo> &scopes,
1133     SemanticsContext &context) {
1134   CheckScopeConstraints(dataTransfers, labels, scopes, context);
1135   CheckDataXferTargetConstraints(dataTransfers, labels, context);
1136 }
1137 
1138 void CheckAssignTargetConstraints(const SourceStmtList &stmts,
1139     const TargetStmtMap &labels, SemanticsContext &context) {
1140   for (const auto &stmt : stmts) {
1141     const auto &label{stmt.parserLabel};
1142     auto target{GetLabel(labels, label)};
1143     if (HasScope(target.proxyForScope) &&
1144         !target.labeledStmtClassificationSet.test(
1145             TargetStatementEnum::Branch) &&
1146         !target.labeledStmtClassificationSet.test(
1147             TargetStatementEnum::Format)) {
1148       parser::Message *msg{nullptr};
1149       if (!target.labeledStmtClassificationSet.test(
1150               TargetStatementEnum::CompatibleBranch)) {
1151         msg = &context.Say(target.parserCharBlock,
1152             "Label '%u' is not a branch target or FORMAT"_err_en_US,
1153             SayLabel(label));
1154       } else if (context.ShouldWarn(common::LanguageFeature::BadBranchTarget)) {
1155         msg = &context.Say(target.parserCharBlock,
1156             "Label '%u' is not a branch target or FORMAT"_warn_en_US,
1157             SayLabel(label));
1158       }
1159       if (msg) {
1160         msg->Attach(stmt.parserCharBlock, "ASSIGN statement use of '%u'"_en_US,
1161             SayLabel(label));
1162       }
1163     }
1164   }
1165 }
1166 
1167 void CheckAssignConstraints(const SourceStmtList &assigns,
1168     const TargetStmtMap &labels, const std::vector<ScopeInfo> &scopes,
1169     SemanticsContext &context) {
1170   CheckScopeConstraints(assigns, labels, scopes, context);
1171   CheckAssignTargetConstraints(assigns, labels, context);
1172 }
1173 
1174 bool CheckConstraints(ParseTreeAnalyzer &&parseTreeAnalysis) {
1175   auto &context{parseTreeAnalysis.ErrorHandler()};
1176   for (const auto &programUnit : parseTreeAnalysis.ProgramUnits()) {
1177     const auto &dos{programUnit.doStmtSources};
1178     const auto &branches{programUnit.otherStmtSources};
1179     const auto &labels{programUnit.targetStmts};
1180     const auto &scopes{programUnit.scopeModel};
1181     CheckLabelDoConstraints(dos, branches, labels, scopes, context);
1182     CheckBranchConstraints(branches, labels, scopes, context);
1183     const auto &dataTransfers{programUnit.formatStmtSources};
1184     CheckDataTransferConstraints(dataTransfers, labels, scopes, context);
1185     const auto &assigns{programUnit.assignStmtSources};
1186     CheckAssignConstraints(assigns, labels, scopes, context);
1187   }
1188   return !context.AnyFatalError();
1189 }
1190 
1191 bool ValidateLabels(SemanticsContext &context, const parser::Program &program) {
1192   return CheckConstraints(LabelAnalysis(context, program));
1193 }
1194 } // namespace Fortran::semantics
1195