Author: marius doerner
Date: 2025-11-20T04:50:32+01:00
New Revision: 7198279707cb77b7a81b54184a6293a49af16f7c

URL: 
https://github.com/llvm/llvm-project/commit/7198279707cb77b7a81b54184a6293a49af16f7c
DIFF: 
https://github.com/llvm/llvm-project/commit/7198279707cb77b7a81b54184a6293a49af16f7c.diff

LOG: [clang][bytecode] Implement case ranges (#168418)

Fixes #165969

Implement GNU case ranges for constexpr bytecode interpreter.

Added: 
    

Modified: 
    clang/lib/AST/ByteCode/Compiler.cpp
    clang/test/AST/ByteCode/switch.cpp

Removed: 
    


################################################################################
diff  --git a/clang/lib/AST/ByteCode/Compiler.cpp 
b/clang/lib/AST/ByteCode/Compiler.cpp
index 8779ffab13b86..8f8f3d6a05820 100644
--- a/clang/lib/AST/ByteCode/Compiler.cpp
+++ b/clang/lib/AST/ByteCode/Compiler.cpp
@@ -6016,11 +6016,38 @@ bool Compiler<Emitter>::visitSwitchStmt(const 
SwitchStmt *S) {
   for (const SwitchCase *SC = S->getSwitchCaseList(); SC;
        SC = SC->getNextSwitchCase()) {
     if (const auto *CS = dyn_cast<CaseStmt>(SC)) {
-      // FIXME: Implement ranges.
-      if (CS->caseStmtIsGNURange())
-        return false;
       CaseLabels[SC] = this->getLabel();
 
+      if (CS->caseStmtIsGNURange()) {
+        LabelTy EndOfRangeCheck = this->getLabel();
+        const Expr *Low = CS->getLHS();
+        const Expr *High = CS->getRHS();
+        if (Low->isValueDependent() || High->isValueDependent())
+          return false;
+
+        if (!this->emitGetLocal(CondT, CondVar, CS))
+          return false;
+        if (!this->visit(Low))
+          return false;
+        PrimType LT = this->classifyPrim(Low->getType());
+        if (!this->emitGE(LT, S))
+          return false;
+        if (!this->jumpFalse(EndOfRangeCheck))
+          return false;
+
+        if (!this->emitGetLocal(CondT, CondVar, CS))
+          return false;
+        if (!this->visit(High))
+          return false;
+        PrimType HT = this->classifyPrim(High->getType());
+        if (!this->emitLE(HT, S))
+          return false;
+        if (!this->jumpTrue(CaseLabels[CS]))
+          return false;
+        this->emitLabel(EndOfRangeCheck);
+        continue;
+      }
+
       const Expr *Value = CS->getLHS();
       if (Value->isValueDependent())
         return false;

diff  --git a/clang/test/AST/ByteCode/switch.cpp 
b/clang/test/AST/ByteCode/switch.cpp
index 8136f92f5dfd3..151b4a5852e88 100644
--- a/clang/test/AST/ByteCode/switch.cpp
+++ b/clang/test/AST/ByteCode/switch.cpp
@@ -86,3 +86,106 @@ constexpr int another_test(int val) { // both-note 
{{declared here}}
 }
 static_assert(another_test(1) == 100, ""); // both-error {{static assertion 
failed}} \
                                            // both-note {{evaluates to}}
+
+namespace gnurange {
+  constexpr int l(int n) {
+    return n + 1;
+  }
+  constexpr int h(int n) {
+    return 2 * n + 1;
+  }
+  constexpr int f(int x) {
+    const int n = 2;
+    constexpr struct {
+      char lo {'a'};
+      char hi {'z'};
+    } s;
+
+    switch (x) {
+      case l(n) ... h(n):
+        return 1;
+      case -1 ... 1:
+        return 2;
+      case 9 ... 14:
+        return 3;
+      case 15:
+        return 4;
+      case 16 ... 20:
+        return 5;
+      case s.lo ... s.hi:
+        return 6;
+      default:
+        return -1;
+    }
+  }
+  static_assert(f(0) == 2);
+  static_assert(f(2) == -1);
+  static_assert(f(3) == 1);
+  static_assert(f(4) == 1);
+  static_assert(f(5) == 1);
+  static_assert(f(6) == -1);
+  static_assert(f(14) == 3);
+  static_assert(f(15) == 4);
+  static_assert(f(16) == 5);
+  static_assert(f(20) == 5);
+  static_assert(f('d') == 6);
+
+  template <int Lo, int Hi>
+  constexpr bool g(int x) {
+    switch (x) {
+      case Lo ... Hi:
+        break;
+      default:
+        return false;
+    }
+    return true;
+  }
+  static_assert(g<100, 200>(132));
+
+  constexpr bool m(int x) {
+    switch (x) {
+      case 10 ... 1: // both-warning {{empty case range specified}}
+        return true;
+      default:
+        return false;
+    }
+  }
+  static_assert(m(3)); // both-error {{static assertion failed due to 
requirement 'm(3)'}}
+  static_assert(!m(3));
+
+  constexpr bool j(int x) { // both-note {{declared here}}
+    switch (x) {
+      case bad(x) ... 100: // both-error {{case value is not a constant 
expression}} \
+                           // both-note {{cannot be used in a constant 
expression}}
+        return true;
+      default:
+        break;
+    }
+    return false;
+  }
+  static_assert(j(1)); // both-error {{static assertion failed}}
+
+  constexpr bool d(int x) { // both-note {{declared here}}
+    switch (x) {
+      case -100 ... bad(x): // both-error {{case value is not a constant 
expression}} \
+                            // both-note {{cannot be used in a constant 
expression}}
+        return true;
+      default:
+        break;
+    }
+    return false;
+  }
+  static_assert(d(1)); // both-error {{static assertion failed}}
+  
+  constexpr bool s(int x) { // both-note {{declared here}}
+    switch (x) {
+      case bad(x) - 100 ... bad(x) + 100: // both-error {{case value is not a 
constant expression}} \
+                                          // both-note {{cannot be used in a 
constant expression}}
+        return true;
+      default:
+        break;
+    }
+    return false;
+  }
+  static_assert(s(1)); // both-error {{static assertion failed}}
+}


        
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits

Reply via email to