1 //===-- lib/Semantics/semantics.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 "flang/Semantics/semantics.h"
10 #include "assignment.h"
11 #include "canonicalize-acc.h"
12 #include "canonicalize-do.h"
13 #include "canonicalize-omp.h"
14 #include "check-acc-structure.h"
15 #include "check-allocate.h"
16 #include "check-arithmeticif.h"
17 #include "check-case.h"
18 #include "check-coarray.h"
19 #include "check-data.h"
20 #include "check-deallocate.h"
21 #include "check-declarations.h"
22 #include "check-do-forall.h"
23 #include "check-if-stmt.h"
24 #include "check-io.h"
25 #include "check-namelist.h"
26 #include "check-nullify.h"
27 #include "check-omp-structure.h"
28 #include "check-purity.h"
29 #include "check-return.h"
30 #include "check-select-rank.h"
31 #include "check-select-type.h"
32 #include "check-stop.h"
33 #include "compute-offsets.h"
34 #include "mod-file.h"
35 #include "resolve-labels.h"
36 #include "resolve-names.h"
37 #include "rewrite-parse-tree.h"
38 #include "flang/Common/default-kinds.h"
39 #include "flang/Parser/parse-tree-visitor.h"
40 #include "flang/Parser/tools.h"
41 #include "flang/Semantics/expression.h"
42 #include "flang/Semantics/scope.h"
43 #include "flang/Semantics/symbol.h"
44 #include "llvm/Support/raw_ostream.h"
45 
46 namespace Fortran::semantics {
47 
48 using NameToSymbolMap = std::map<const char *, SymbolRef>;
49 static void DoDumpSymbols(llvm::raw_ostream &, const Scope &, int indent = 0);
50 static void PutIndent(llvm::raw_ostream &, int indent);
51 
52 static void GetSymbolNames(const Scope &scope, NameToSymbolMap &symbols) {
53   // Finds all symbol names in the scope without collecting duplicates.
54   for (const auto &pair : scope) {
55     symbols.emplace(pair.second->name().begin(), *pair.second);
56   }
57   for (const auto &pair : scope.commonBlocks()) {
58     symbols.emplace(pair.second->name().begin(), *pair.second);
59   }
60   for (const auto &child : scope.children()) {
61     GetSymbolNames(child, symbols);
62   }
63 }
64 
65 // A parse tree visitor that calls Enter/Leave functions from each checker
66 // class C supplied as template parameters. Enter is called before the node's
67 // children are visited, Leave is called after. No two checkers may have the
68 // same Enter or Leave function. Each checker must be constructible from
69 // SemanticsContext and have BaseChecker as a virtual base class.
70 template <typename... C> class SemanticsVisitor : public virtual C... {
71 public:
72   using C::Enter...;
73   using C::Leave...;
74   using BaseChecker::Enter;
75   using BaseChecker::Leave;
76   SemanticsVisitor(SemanticsContext &context)
77       : C{context}..., context_{context} {}
78 
79   template <typename N> bool Pre(const N &node) {
80     if constexpr (common::HasMember<const N *, ConstructNode>) {
81       context_.PushConstruct(node);
82     }
83     Enter(node);
84     return true;
85   }
86   template <typename N> void Post(const N &node) {
87     Leave(node);
88     if constexpr (common::HasMember<const N *, ConstructNode>) {
89       context_.PopConstruct();
90     }
91   }
92 
93   template <typename T> bool Pre(const parser::Statement<T> &node) {
94     context_.set_location(node.source);
95     Enter(node);
96     return true;
97   }
98   template <typename T> bool Pre(const parser::UnlabeledStatement<T> &node) {
99     context_.set_location(node.source);
100     Enter(node);
101     return true;
102   }
103   template <typename T> void Post(const parser::Statement<T> &node) {
104     Leave(node);
105     context_.set_location(std::nullopt);
106   }
107   template <typename T> void Post(const parser::UnlabeledStatement<T> &node) {
108     Leave(node);
109     context_.set_location(std::nullopt);
110   }
111 
112   bool Walk(const parser::Program &program) {
113     parser::Walk(program, *this);
114     return !context_.AnyFatalError();
115   }
116 
117 private:
118   SemanticsContext &context_;
119 };
120 
121 class MiscChecker : public virtual BaseChecker {
122 public:
123   explicit MiscChecker(SemanticsContext &context) : context_{context} {}
124   void Leave(const parser::EntryStmt &) {
125     if (!context_.constructStack().empty()) { // C1571
126       context_.Say("ENTRY may not appear in an executable construct"_err_en_US);
127     }
128   }
129   void Leave(const parser::AssignStmt &stmt) {
130     CheckAssignGotoName(std::get<parser::Name>(stmt.t));
131   }
132   void Leave(const parser::AssignedGotoStmt &stmt) {
133     CheckAssignGotoName(std::get<parser::Name>(stmt.t));
134   }
135 
136 private:
137   void CheckAssignGotoName(const parser::Name &name) {
138     if (context_.HasError(name.symbol)) {
139       return;
140     }
141     const Symbol &symbol{DEREF(name.symbol)};
142     auto type{evaluate::DynamicType::From(symbol)};
143     if (!IsVariableName(symbol) || symbol.Rank() != 0 || !type ||
144         type->category() != TypeCategory::Integer ||
145         type->kind() !=
146             context_.defaultKinds().GetDefaultKind(TypeCategory::Integer)) {
147       context_
148           .Say(name.source,
149               "'%s' must be a default integer scalar variable"_err_en_US,
150               name.source)
151           .Attach(symbol.name(), "Declaration of '%s'"_en_US, symbol.name());
152     }
153   }
154 
155   SemanticsContext &context_;
156 };
157 
158 using StatementSemanticsPass1 = ExprChecker;
159 using StatementSemanticsPass2 = SemanticsVisitor<AccStructureChecker,
160     AllocateChecker, ArithmeticIfStmtChecker, AssignmentChecker, CaseChecker,
161     CoarrayChecker, DataChecker, DeallocateChecker, DoForallChecker,
162     IfStmtChecker, IoChecker, MiscChecker, NamelistChecker, NullifyChecker,
163     OmpStructureChecker, PurityChecker, ReturnStmtChecker,
164     SelectRankConstructChecker, SelectTypeChecker, StopChecker>;
165 
166 static bool PerformStatementSemantics(
167     SemanticsContext &context, parser::Program &program) {
168   ResolveNames(context, program);
169   RewriteParseTree(context, program);
170   ComputeOffsets(context);
171   CheckDeclarations(context);
172   StatementSemanticsPass1{context}.Walk(program);
173   StatementSemanticsPass2 pass2{context};
174   pass2.Walk(program);
175   if (!context.AnyFatalError()) {
176     pass2.CompileDataInitializationsIntoInitializers();
177   }
178   return !context.AnyFatalError();
179 }
180 
181 SemanticsContext::SemanticsContext(
182     const common::IntrinsicTypeDefaultKinds &defaultKinds,
183     const common::LanguageFeatureControl &languageFeatures,
184     parser::AllSources &allSources)
185     : defaultKinds_{defaultKinds}, languageFeatures_{languageFeatures},
186       allSources_{allSources},
187       intrinsics_{evaluate::IntrinsicProcTable::Configure(defaultKinds_)},
188       foldingContext_{
189           parser::ContextualMessages{&messages_}, defaultKinds_, intrinsics_} {}
190 
191 SemanticsContext::~SemanticsContext() {}
192 
193 int SemanticsContext::GetDefaultKind(TypeCategory category) const {
194   return defaultKinds_.GetDefaultKind(category);
195 }
196 
197 bool SemanticsContext::IsEnabled(common::LanguageFeature feature) const {
198   return languageFeatures_.IsEnabled(feature);
199 }
200 
201 bool SemanticsContext::ShouldWarn(common::LanguageFeature feature) const {
202   return languageFeatures_.ShouldWarn(feature);
203 }
204 
205 const DeclTypeSpec &SemanticsContext::MakeNumericType(
206     TypeCategory category, int kind) {
207   if (kind == 0) {
208     kind = GetDefaultKind(category);
209   }
210   return globalScope_.MakeNumericType(category, KindExpr{kind});
211 }
212 const DeclTypeSpec &SemanticsContext::MakeLogicalType(int kind) {
213   if (kind == 0) {
214     kind = GetDefaultKind(TypeCategory::Logical);
215   }
216   return globalScope_.MakeLogicalType(KindExpr{kind});
217 }
218 
219 bool SemanticsContext::AnyFatalError() const {
220   return !messages_.empty() &&
221       (warningsAreErrors_ || messages_.AnyFatalError());
222 }
223 bool SemanticsContext::HasError(const Symbol &symbol) {
224   return CheckError(symbol.test(Symbol::Flag::Error));
225 }
226 bool SemanticsContext::HasError(const Symbol *symbol) {
227   return CheckError(!symbol || HasError(*symbol));
228 }
229 bool SemanticsContext::HasError(const parser::Name &name) {
230   return HasError(name.symbol);
231 }
232 void SemanticsContext::SetError(Symbol &symbol, bool value) {
233   if (value) {
234     CHECK(AnyFatalError());
235     symbol.set(Symbol::Flag::Error);
236   }
237 }
238 bool SemanticsContext::CheckError(bool error) {
239   CHECK(!error || AnyFatalError());
240   return error;
241 }
242 
243 const Scope &SemanticsContext::FindScope(parser::CharBlock source) const {
244   return const_cast<SemanticsContext *>(this)->FindScope(source);
245 }
246 
247 Scope &SemanticsContext::FindScope(parser::CharBlock source) {
248   if (auto *scope{globalScope_.FindScope(source)}) {
249     return *scope;
250   } else {
251     common::die("SemanticsContext::FindScope(): invalid source location");
252   }
253 }
254 
255 void SemanticsContext::PopConstruct() {
256   CHECK(!constructStack_.empty());
257   constructStack_.pop_back();
258 }
259 
260 void SemanticsContext::CheckIndexVarRedefine(const parser::CharBlock &location,
261     const Symbol &variable, parser::MessageFixedText &&message) {
262   if (const Symbol * root{GetAssociationRoot(variable)}) {
263     auto it{activeIndexVars_.find(*root)};
264     if (it != activeIndexVars_.end()) {
265       std::string kind{EnumToString(it->second.kind)};
266       Say(location, std::move(message), kind, root->name())
267           .Attach(it->second.location, "Enclosing %s construct"_en_US, kind);
268     }
269   }
270 }
271 
272 void SemanticsContext::WarnIndexVarRedefine(
273     const parser::CharBlock &location, const Symbol &variable) {
274   CheckIndexVarRedefine(
275       location, variable, "Possible redefinition of %s variable '%s'"_en_US);
276 }
277 
278 void SemanticsContext::CheckIndexVarRedefine(
279     const parser::CharBlock &location, const Symbol &variable) {
280   CheckIndexVarRedefine(
281       location, variable, "Cannot redefine %s variable '%s'"_err_en_US);
282 }
283 
284 void SemanticsContext::CheckIndexVarRedefine(const parser::Variable &variable) {
285   if (const Symbol * entity{GetLastName(variable).symbol}) {
286     CheckIndexVarRedefine(variable.GetSource(), *entity);
287   }
288 }
289 
290 void SemanticsContext::CheckIndexVarRedefine(const parser::Name &name) {
291   if (const Symbol * entity{name.symbol}) {
292     CheckIndexVarRedefine(name.source, *entity);
293   }
294 }
295 
296 void SemanticsContext::ActivateIndexVar(
297     const parser::Name &name, IndexVarKind kind) {
298   CheckIndexVarRedefine(name);
299   if (const Symbol * indexVar{name.symbol}) {
300     if (const Symbol * root{GetAssociationRoot(*indexVar)}) {
301       activeIndexVars_.emplace(*root, IndexVarInfo{name.source, kind});
302     }
303   }
304 }
305 
306 void SemanticsContext::DeactivateIndexVar(const parser::Name &name) {
307   if (Symbol * indexVar{name.symbol}) {
308     if (const Symbol * root{GetAssociationRoot(*indexVar)}) {
309       auto it{activeIndexVars_.find(*root)};
310       if (it != activeIndexVars_.end() && it->second.location == name.source) {
311         activeIndexVars_.erase(it);
312       }
313     }
314   }
315 }
316 
317 SymbolVector SemanticsContext::GetIndexVars(IndexVarKind kind) {
318   SymbolVector result;
319   for (const auto &[symbol, info] : activeIndexVars_) {
320     if (info.kind == kind) {
321       result.push_back(symbol);
322     }
323   }
324   return result;
325 }
326 
327 bool Semantics::Perform() {
328   return ValidateLabels(context_, program_) &&
329       parser::CanonicalizeDo(program_) && // force line break
330       CanonicalizeAcc(context_.messages(), program_) &&
331       CanonicalizeOmp(context_.messages(), program_) &&
332       PerformStatementSemantics(context_, program_) &&
333       ModFileWriter{context_}.WriteAll();
334 }
335 
336 void Semantics::EmitMessages(llvm::raw_ostream &os) const {
337   context_.messages().Emit(os, cooked_);
338 }
339 
340 void Semantics::DumpSymbols(llvm::raw_ostream &os) {
341   DoDumpSymbols(os, context_.globalScope());
342 }
343 
344 void Semantics::DumpSymbolsSources(llvm::raw_ostream &os) const {
345   NameToSymbolMap symbols;
346   GetSymbolNames(context_.globalScope(), symbols);
347   for (const auto &pair : symbols) {
348     const Symbol &symbol{pair.second};
349     if (auto sourceInfo{cooked_.GetSourcePositionRange(symbol.name())}) {
350       os << symbol.name().ToString() << ": " << sourceInfo->first.file.path()
351          << ", " << sourceInfo->first.line << ", " << sourceInfo->first.column
352          << "-" << sourceInfo->second.column << "\n";
353     } else if (symbol.has<semantics::UseDetails>()) {
354       os << symbol.name().ToString() << ": "
355          << symbol.GetUltimate().owner().symbol()->name().ToString() << "\n";
356     }
357   }
358 }
359 
360 void DoDumpSymbols(llvm::raw_ostream &os, const Scope &scope, int indent) {
361   PutIndent(os, indent);
362   os << Scope::EnumToString(scope.kind()) << " scope:";
363   if (const auto *symbol{scope.symbol()}) {
364     os << ' ' << symbol->name();
365   }
366   if (scope.size()) {
367     os << " size=" << scope.size() << " alignment=" << scope.alignment();
368   }
369   if (scope.derivedTypeSpec()) {
370     os << " instantiation of " << *scope.derivedTypeSpec();
371   }
372   os << '\n';
373   ++indent;
374   for (const auto &pair : scope) {
375     const auto &symbol{*pair.second};
376     PutIndent(os, indent);
377     os << symbol << '\n';
378     if (const auto *details{symbol.detailsIf<GenericDetails>()}) {
379       if (const auto &type{details->derivedType()}) {
380         PutIndent(os, indent);
381         os << *type << '\n';
382       }
383     }
384   }
385   if (!scope.equivalenceSets().empty()) {
386     PutIndent(os, indent);
387     os << "Equivalence Sets:";
388     for (const auto &set : scope.equivalenceSets()) {
389       os << ' ';
390       char sep = '(';
391       for (const auto &object : set) {
392         os << sep << object.AsFortran();
393         sep = ',';
394       }
395       os << ')';
396     }
397     os << '\n';
398   }
399   if (!scope.crayPointers().empty()) {
400     PutIndent(os, indent);
401     os << "Cray Pointers:";
402     for (const auto &[pointee, pointer] : scope.crayPointers()) {
403       os << " (" << pointer->name() << ',' << pointee << ')';
404     }
405   }
406   for (const auto &pair : scope.commonBlocks()) {
407     const auto &symbol{*pair.second};
408     PutIndent(os, indent);
409     os << symbol << '\n';
410   }
411   for (const auto &child : scope.children()) {
412     DoDumpSymbols(os, child, indent);
413   }
414   --indent;
415 }
416 
417 static void PutIndent(llvm::raw_ostream &os, int indent) {
418   for (int i = 0; i < indent; ++i) {
419     os << "  ";
420   }
421 }
422 } // namespace Fortran::semantics
423