Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[clang] Fix segmentation fault caused by stack overflow on deeply nested expressions #111701

Merged
merged 9 commits into from
Oct 14, 2024

Conversation

bricknerb
Copy link
Contributor

Done by calling clang::runWithSufficientStackSpace().
Added CodeGenModule::runWithSufficientStackSpace() method similar to the one in Sema to provide a single warning when this triggers
Fixes: #111699

…ted expressions.

This is done by calling clang::runWithSufficientStackSpace().
Added CodeGenModule::runWithSufficientStackSpace() method similar to the one in Sema to provide a single warning when this triggers.
Copy link

github-actions bot commented Oct 9, 2024

Thank you for submitting a Pull Request (PR) to the LLVM Project!

This PR will be automatically labeled and the relevant teams will be notified.

If you wish to, you can add reviewers by using the "Reviewers" section on this page.

If this is not working for you, it is probably because you do not have write permissions for the repository. In which case you can instead tag reviewers by name in a comment by using @ followed by their GitHub username.

If you have received no comments on your PR for a week, you can request a review by "ping"ing the PR by adding a comment “Ping”. The common courtesy "ping" rate is once a week. Please remember that you are asking for valuable time from other developers.

If you have further questions, they may be answered by the LLVM GitHub User Guide.

You can also ask questions in a comment on this PR, on the LLVM Discord or on the forums.

@llvmbot llvmbot added clang Clang issues not falling into any other category clang:codegen labels Oct 9, 2024
@llvmbot
Copy link
Member

llvmbot commented Oct 9, 2024

@llvm/pr-subscribers-clang-codegen

Author: Boaz Brickner (bricknerb)

Changes

Done by calling clang::runWithSufficientStackSpace().
Added CodeGenModule::runWithSufficientStackSpace() method similar to the one in Sema to provide a single warning when this triggers
Fixes: #111699


Patch is 82.20 KiB, truncated to 20.00 KiB below, full version: https://github.com/llvm/llvm-project/pull/111701.diff

4 Files Affected:

  • (modified) clang/lib/CodeGen/CGExpr.cpp (+4-1)
  • (modified) clang/lib/CodeGen/CodeGenModule.cpp (+14)
  • (modified) clang/lib/CodeGen/CodeGenModule.h (+11)
  • (added) clang/test/CodeGen/deeply-nested-expressions.cpp (+1013)
diff --git a/clang/lib/CodeGen/CGExpr.cpp b/clang/lib/CodeGen/CGExpr.cpp
index 451442765620f7..5ececf7d940520 100644
--- a/clang/lib/CodeGen/CGExpr.cpp
+++ b/clang/lib/CodeGen/CGExpr.cpp
@@ -5817,7 +5817,10 @@ LValue CodeGenFunction::EmitHLSLArrayAssignLValue(const BinaryOperator *E) {
 
 LValue CodeGenFunction::EmitCallExprLValue(const CallExpr *E,
                                            llvm::CallBase **CallOrInvoke) {
-  RValue RV = EmitCallExpr(E, ReturnValueSlot(), CallOrInvoke);
+  RValue RV;
+  CGM.runWithSufficientStackSpace(E->getExprLoc(), [&] {
+    RV = EmitCallExpr(E, ReturnValueSlot(), CallOrInvoke);
+  });
 
   if (!RV.isScalar())
     return MakeAddrLValue(RV.getAggregateAddress(), E->getType(),
diff --git a/clang/lib/CodeGen/CodeGenModule.cpp b/clang/lib/CodeGen/CodeGenModule.cpp
index 5ba098144a74e7..424455cbf4da39 100644
--- a/clang/lib/CodeGen/CodeGenModule.cpp
+++ b/clang/lib/CodeGen/CodeGenModule.cpp
@@ -44,6 +44,7 @@
 #include "clang/Basic/FileManager.h"
 #include "clang/Basic/Module.h"
 #include "clang/Basic/SourceManager.h"
+#include "clang/Basic/Stack.h"
 #include "clang/Basic/TargetInfo.h"
 #include "clang/Basic/Version.h"
 #include "clang/CodeGen/BackendUtil.h"
@@ -1596,6 +1597,19 @@ void CodeGenModule::ErrorUnsupported(const Decl *D, const char *Type) {
   getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID) << Msg;
 }
 
+void CodeGenModule::warnStackExhausted(SourceLocation Loc) {
+  // Only warn about this once.
+  if (!WarnedStackExhausted) {
+    getDiags().Report(Loc, diag::warn_stack_exhausted);
+    WarnedStackExhausted = true;
+  }
+}
+
+void CodeGenModule::runWithSufficientStackSpace(SourceLocation Loc,
+                                                llvm::function_ref<void()> Fn) {
+  clang::runWithSufficientStackSpace([&] { warnStackExhausted(Loc); }, Fn);
+}
+
 llvm::ConstantInt *CodeGenModule::getSize(CharUnits size) {
   return llvm::ConstantInt::get(SizeTy, size.getQuantity());
 }
diff --git a/clang/lib/CodeGen/CodeGenModule.h b/clang/lib/CodeGen/CodeGenModule.h
index c58bb88035ca8a..57e06cbfac13a9 100644
--- a/clang/lib/CodeGen/CodeGenModule.h
+++ b/clang/lib/CodeGen/CodeGenModule.h
@@ -336,6 +336,7 @@ class CodeGenModule : public CodeGenTypeCache {
   std::unique_ptr<llvm::IndexedInstrProfReader> PGOReader;
   InstrProfStats PGOStats;
   std::unique_ptr<llvm::SanitizerStatReport> SanStats;
+  bool WarnedStackExhausted = false;
 
   // A set of references that have only been seen via a weakref so far. This is
   // used to remove the weak of the reference if we ever see a direct reference
@@ -1297,6 +1298,16 @@ class CodeGenModule : public CodeGenTypeCache {
   /// Print out an error that codegen doesn't support the specified decl yet.
   void ErrorUnsupported(const Decl *D, const char *Type);
 
+  /// Warn that the stack is nearly exhausted.
+  void warnStackExhausted(SourceLocation Loc);
+
+  /// Run some code with "sufficient" stack space. (Currently, at least 256K is
+  /// guaranteed). Produces a warning if we're low on stack space and allocates
+  /// more in that case. Use this in code that may recurse deeply to avoid stack
+  /// overflow.
+  void runWithSufficientStackSpace(SourceLocation Loc,
+                                   llvm::function_ref<void()> Fn);
+
   /// Set the attributes on the LLVM function for the given decl and function
   /// info. This applies attributes necessary for handling the ABI as well as
   /// user specified attributes like section.
diff --git a/clang/test/CodeGen/deeply-nested-expressions.cpp b/clang/test/CodeGen/deeply-nested-expressions.cpp
new file mode 100644
index 00000000000000..3f7b55d35ed76d
--- /dev/null
+++ b/clang/test/CodeGen/deeply-nested-expressions.cpp
@@ -0,0 +1,1013 @@
+// RUN: %clang_cc1 %s -emit-llvm -o - -Wstack-exhausted -verify
+
+class AClass {
+public:
+  AClass() {}
+  AClass &foo() { return *this; }
+};
+
+void test_bar() {
+  AClass a;
+  // expected-warning@* {{stack nearly exhausted; compilation time may suffer, and crashes due to stack overflow are likely}}
+  a.foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().f...
[truncated]

@llvmbot
Copy link
Member

llvmbot commented Oct 9, 2024

@llvm/pr-subscribers-clang

Author: Boaz Brickner (bricknerb)

Changes

Done by calling clang::runWithSufficientStackSpace().
Added CodeGenModule::runWithSufficientStackSpace() method similar to the one in Sema to provide a single warning when this triggers
Fixes: #111699


Patch is 82.20 KiB, truncated to 20.00 KiB below, full version: https://github.com/llvm/llvm-project/pull/111701.diff

4 Files Affected:

  • (modified) clang/lib/CodeGen/CGExpr.cpp (+4-1)
  • (modified) clang/lib/CodeGen/CodeGenModule.cpp (+14)
  • (modified) clang/lib/CodeGen/CodeGenModule.h (+11)
  • (added) clang/test/CodeGen/deeply-nested-expressions.cpp (+1013)
diff --git a/clang/lib/CodeGen/CGExpr.cpp b/clang/lib/CodeGen/CGExpr.cpp
index 451442765620f7..5ececf7d940520 100644
--- a/clang/lib/CodeGen/CGExpr.cpp
+++ b/clang/lib/CodeGen/CGExpr.cpp
@@ -5817,7 +5817,10 @@ LValue CodeGenFunction::EmitHLSLArrayAssignLValue(const BinaryOperator *E) {
 
 LValue CodeGenFunction::EmitCallExprLValue(const CallExpr *E,
                                            llvm::CallBase **CallOrInvoke) {
-  RValue RV = EmitCallExpr(E, ReturnValueSlot(), CallOrInvoke);
+  RValue RV;
+  CGM.runWithSufficientStackSpace(E->getExprLoc(), [&] {
+    RV = EmitCallExpr(E, ReturnValueSlot(), CallOrInvoke);
+  });
 
   if (!RV.isScalar())
     return MakeAddrLValue(RV.getAggregateAddress(), E->getType(),
diff --git a/clang/lib/CodeGen/CodeGenModule.cpp b/clang/lib/CodeGen/CodeGenModule.cpp
index 5ba098144a74e7..424455cbf4da39 100644
--- a/clang/lib/CodeGen/CodeGenModule.cpp
+++ b/clang/lib/CodeGen/CodeGenModule.cpp
@@ -44,6 +44,7 @@
 #include "clang/Basic/FileManager.h"
 #include "clang/Basic/Module.h"
 #include "clang/Basic/SourceManager.h"
+#include "clang/Basic/Stack.h"
 #include "clang/Basic/TargetInfo.h"
 #include "clang/Basic/Version.h"
 #include "clang/CodeGen/BackendUtil.h"
@@ -1596,6 +1597,19 @@ void CodeGenModule::ErrorUnsupported(const Decl *D, const char *Type) {
   getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID) << Msg;
 }
 
+void CodeGenModule::warnStackExhausted(SourceLocation Loc) {
+  // Only warn about this once.
+  if (!WarnedStackExhausted) {
+    getDiags().Report(Loc, diag::warn_stack_exhausted);
+    WarnedStackExhausted = true;
+  }
+}
+
+void CodeGenModule::runWithSufficientStackSpace(SourceLocation Loc,
+                                                llvm::function_ref<void()> Fn) {
+  clang::runWithSufficientStackSpace([&] { warnStackExhausted(Loc); }, Fn);
+}
+
 llvm::ConstantInt *CodeGenModule::getSize(CharUnits size) {
   return llvm::ConstantInt::get(SizeTy, size.getQuantity());
 }
diff --git a/clang/lib/CodeGen/CodeGenModule.h b/clang/lib/CodeGen/CodeGenModule.h
index c58bb88035ca8a..57e06cbfac13a9 100644
--- a/clang/lib/CodeGen/CodeGenModule.h
+++ b/clang/lib/CodeGen/CodeGenModule.h
@@ -336,6 +336,7 @@ class CodeGenModule : public CodeGenTypeCache {
   std::unique_ptr<llvm::IndexedInstrProfReader> PGOReader;
   InstrProfStats PGOStats;
   std::unique_ptr<llvm::SanitizerStatReport> SanStats;
+  bool WarnedStackExhausted = false;
 
   // A set of references that have only been seen via a weakref so far. This is
   // used to remove the weak of the reference if we ever see a direct reference
@@ -1297,6 +1298,16 @@ class CodeGenModule : public CodeGenTypeCache {
   /// Print out an error that codegen doesn't support the specified decl yet.
   void ErrorUnsupported(const Decl *D, const char *Type);
 
+  /// Warn that the stack is nearly exhausted.
+  void warnStackExhausted(SourceLocation Loc);
+
+  /// Run some code with "sufficient" stack space. (Currently, at least 256K is
+  /// guaranteed). Produces a warning if we're low on stack space and allocates
+  /// more in that case. Use this in code that may recurse deeply to avoid stack
+  /// overflow.
+  void runWithSufficientStackSpace(SourceLocation Loc,
+                                   llvm::function_ref<void()> Fn);
+
   /// Set the attributes on the LLVM function for the given decl and function
   /// info. This applies attributes necessary for handling the ABI as well as
   /// user specified attributes like section.
diff --git a/clang/test/CodeGen/deeply-nested-expressions.cpp b/clang/test/CodeGen/deeply-nested-expressions.cpp
new file mode 100644
index 00000000000000..3f7b55d35ed76d
--- /dev/null
+++ b/clang/test/CodeGen/deeply-nested-expressions.cpp
@@ -0,0 +1,1013 @@
+// RUN: %clang_cc1 %s -emit-llvm -o - -Wstack-exhausted -verify
+
+class AClass {
+public:
+  AClass() {}
+  AClass &foo() { return *this; }
+};
+
+void test_bar() {
+  AClass a;
+  // expected-warning@* {{stack nearly exhausted; compilation time may suffer, and crashes due to stack overflow are likely}}
+  a.foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo().foo()
+      .foo().foo().f...
[truncated]

@bricknerb bricknerb changed the title [clang] Fix segmentation fault caused by stack overflow on deeply nes… [clang] Fix segmentation fault caused by stack overflow on deeply nested expressions Oct 9, 2024
Copy link
Contributor

@ilya-biryukov ilya-biryukov left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I have a few suggestions, but definitely support this.

I've also added a few folks suggested by GitHub as reviewers to make sure we have coverage of someone who looks at CodeGen.

clang/test/CodeGen/deeply-nested-expressions.cpp Outdated Show resolved Hide resolved
clang/lib/CodeGen/CGExpr.cpp Outdated Show resolved Hide resolved
@ilya-biryukov
Copy link
Contributor

The breakage from buildkite is realted: we don't seem to run out of stack space on Windows, unfortunately it's quite common for these things to be vary between platforms. We could either limit the tests to Linux, or increase the depth, or figure some other way to reconcile it 🤷

…deGenFunction::EmitCallExprLValue() to CodeGenFunction::EmitLValue().
…ch it where a bigger stack is available.

This warning might trigger twice now because the can also trigger this warning in Sema and not just in CodeGen.
getDiags().Report(Loc, diag::warn_stack_exhausted);
WarnedStackExhausted = true;
}
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there a way to avoid the duplication with Sema here?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I was thinking that as a follow up I will extract this small logic into a class which both Sema and CodeGen would own.
I also have #111799 to make them more similar.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

SGTM and also feels like something that we could do in a follow-up, even if the code here is duplicated for some short period of time.

I suspect there's some wiring up to figure out for sharing this in the same class, so it'd be good to land this while that's still happening.

Copy link
Contributor

@ilya-biryukov ilya-biryukov left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, I feel we could land this right away and figure out how to share with Sema in a follow-up.

Hopefully @cor3ntin did not mean to block this until that happens, but let's not hold off the follow-up for long to make sure we don't make this a permanent state.

clang/test/CodeGen/deeply-nested-expressions.cpp Outdated Show resolved Hide resolved
getDiags().Report(Loc, diag::warn_stack_exhausted);
WarnedStackExhausted = true;
}
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

SGTM and also feels like something that we could do in a follow-up, even if the code here is duplicated for some short period of time.

I suspect there's some wiring up to figure out for sharing this in the same class, so it'd be good to land this while that's still happening.

Copy link

github-actions bot commented Oct 14, 2024

✅ With the latest revision this PR passed the C/C++ code formatter.

@cor3ntin
Copy link
Contributor

I'm still concerned that the test is going to be super flaky - I would remove it and mention the issue this fixes by adding a comment where runWithSufficientStackSpace is called

Copy link
Contributor

@ilya-biryukov ilya-biryukov left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Marking with "request changes" to ask removing the test per @cor3ntin's comments

…o avoid having a potentially flaky test.

Add a comment to explain why we call runWithSufficientStackSpace() instead.
Copy link
Contributor

@ilya-biryukov ilya-biryukov left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@ilya-biryukov ilya-biryukov merged commit c978f0f into llvm:main Oct 14, 2024
8 checks passed
Copy link

@bricknerb Congratulations on having your first Pull Request (PR) merged into the LLVM Project!

Your changes will be combined with recent changes from other authors, then tested by our build bots. If there is a problem with a build, you may receive a report in an email or a comment on this PR.

Please check whether problems have been caused by your change specifically, as the builds can include changes from many authors. It is not uncommon for your change to be included in a build that fails due to someone else's changes, or infrastructure issues.

How to do this, and the rest of the post-merge process, is covered in detail here.

If your change does cause a problem, it may be reverted, or you can revert it yourself. This is a normal part of LLVM development. You can fix your changes and open a new PR to merge them again.

If you don't get any reports, no action is required from you. Your changes are working as expected, well done!

DanielCChen pushed a commit to DanielCChen/llvm-project that referenced this pull request Oct 16, 2024
…ted expressions (llvm#111701)

Done by calling clang::runWithSufficientStackSpace().
Added CodeGenModule::runWithSufficientStackSpace() method similar to the
one in Sema to provide a single warning when this triggers
Fixes: llvm#111699
bricknerb added a commit to bricknerb/llvm-project that referenced this pull request Oct 17, 2024
…ted expressions (llvm#111701)

Done by calling clang::runWithSufficientStackSpace().
Added CodeGenModule::runWithSufficientStackSpace() method similar to the
one in Sema to provide a single warning when this triggers
Fixes: llvm#111699
@bricknerb bricknerb deleted the stack3 branch October 18, 2024 08:42
EricWF pushed a commit to efcs/llvm-project that referenced this pull request Oct 22, 2024
…ted expressions (llvm#111701)

Done by calling clang::runWithSufficientStackSpace().
Added CodeGenModule::runWithSufficientStackSpace() method similar to the
one in Sema to provide a single warning when this triggers
Fixes: llvm#111699
bricknerb added a commit to bricknerb/llvm-project that referenced this pull request Jan 23, 2025
…flow on deeply nested expressions

This happens when using -O2.

Added a test that reproduces it based on a test that was reverted in llvm#111701:
https://github.com/bricknerb/llvm-project/blob/93e4a7386ec897e53d7330c6206d38759a858be2/clang/test/CodeGen/deeply-nested-expressions.cpp
However, this test is slow and likely to be hard to maintained as discussed in https://github.com/llvm/llvm-project/pull/111701/files/1a63281b6c240352653fd2e4299755c1f32a76f4#r1795518779 so unless there are objections I plan to remove this test before merging.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
clang:codegen clang Clang issues not falling into any other category
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Clang crashes segmentation fault due to stack overflow on deeply nested expressions
5 participants