[PP-EXT] Support usage of spec_index_cmp in conditions

Author: kpdev

clang/lib/Sema/SemaDecl.cpp

@@ -935,15 +935,22 @@ Sema::NameClassification Sema::ClassifyName(Scope *S, CXXScopeSpec &SS,
     }
     else if (IsSpecIdxCmp) {
       auto tfi = Context.CreateTypeSourceInfo(Context.VoidPtrTy);
-      ParmVarDecl* PVDecl = ParmVarDecl::Create(Context,
+      ParmVarDecl* PVDecl1 = ParmVarDecl::Create(Context,
         Context.getTranslationUnitDecl(),
         NameLoc, NameLoc, nullptr,
         Context.VoidPtrTy,
         tfi,
         clang::StorageClass::SC_None,
         nullptr);
-      Params.push_back(PVDecl);
-      Params.push_back(PVDecl);
+      ParmVarDecl* PVDecl2 = ParmVarDecl::Create(Context,
+        Context.getTranslationUnitDecl(),
+        NameLoc, NameLoc, nullptr,
+        Context.VoidPtrTy,
+        tfi,
+        clang::StorageClass::SC_None,
+        nullptr);
+      Params.push_back(PVDecl1);
+      Params.push_back(PVDecl2);
     }
     else if (IsGetSpecPtr) {
       auto tfi = Context.CreateTypeSourceInfo(Context.IntTy);

clang/lib/Sema/SemaExpr.cpp

@@ -2594,15 +2594,22 @@ Sema::ActOnIdExpression(Scope *S, CXXScopeSpec &SS,
     }
     else if (IsSpecIdxCmp) {
       auto tfi = Context.CreateTypeSourceInfo(Context.VoidPtrTy);
-      ParmVarDecl* PVDecl = ParmVarDecl::Create(Context,
+      ParmVarDecl* PVDecl1 = ParmVarDecl::Create(Context,
         Context.getTranslationUnitDecl(),
         NameLoc, NameLoc, nullptr,
         Context.VoidPtrTy,
         tfi,
         clang::StorageClass::SC_None,
         nullptr);
-      Params.push_back(PVDecl);
-      Params.push_back(PVDecl);
+      ParmVarDecl* PVDecl2 = ParmVarDecl::Create(Context,
+        Context.getTranslationUnitDecl(),
+        NameLoc, NameLoc, nullptr,
+        Context.VoidPtrTy,
+        tfi,
+        clang::StorageClass::SC_None,
+        nullptr);
+      Params.push_back(PVDecl1);
+      Params.push_back(PVDecl2);
     }
     NewD->setParams(Params);
     // TODO: Remove it

clang/test/CodeGen/Figure.c

@@ -262,6 +262,15 @@ int main() {
     // CHECK-RT-NEXT: Circ+Rect cmp: -1
     int res_cmp4 = spec_index_cmp(&fc, &fr);
     printf("Circ+Rect cmp: %d\n", res_cmp4);
+
+    // CHECK-RT-NEXT: Rect+Circ cmp: -1
+    res_cmp4 = spec_index_cmp(&fr, &fc);
+    printf("Rect+Circ cmp: %d\n", res_cmp4);
+
+    // CHECK-RT-NEXT: Checked usage spec_index_cmp in condition
+    if (spec_index_cmp(&fc, &fc) >= 0) {
+        printf("Checked usage spec_index_cmp in condition\n");
+    }
 }
 
 // This code just checking comilation