fix -dM with variadic macros, PR5699
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@90735 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Frontend/PrintPreprocessedOutput.cpp b/lib/Frontend/PrintPreprocessedOutput.cpp
index e0186ab..c23c6e3 100644
--- a/lib/Frontend/PrintPreprocessedOutput.cpp
+++ b/lib/Frontend/PrintPreprocessedOutput.cpp
@@ -28,6 +28,13 @@
#include <cstdio>
using namespace clang;
+static void PrintArgName(const IdentifierInfo *II, llvm::raw_ostream &OS) {
+ if (II->getName() == "__VA_ARGS__")
+ OS << "...";
+ else
+ OS << II->getName();
+}
+
/// PrintMacroDefinition - Print a macro definition in a form that will be
/// properly accepted back as a definition.
static void PrintMacroDefinition(const IdentifierInfo &II, const MacroInfo &MI,
@@ -39,19 +46,19 @@
if (MI.arg_empty())
;
else if (MI.getNumArgs() == 1)
- OS << (*MI.arg_begin())->getName();
+ PrintArgName(*MI.arg_begin(), OS);
else {
MacroInfo::arg_iterator AI = MI.arg_begin(), E = MI.arg_end();
OS << (*AI++)->getName();
- while (AI != E)
- OS << ',' << (*AI++)->getName();
+ while (AI != E) {
+ OS << ',';
+ PrintArgName(*AI++, OS);
+ }
}
- if (MI.isVariadic()) {
- if (!MI.arg_empty())
- OS << ',';
- OS << "...";
- }
+ if (MI.isGNUVarargs())
+ OS << "..."; // #define foo(x...)
+
OS << ')';
}
diff --git a/test/Preprocessor/dump_macros.c b/test/Preprocessor/dump_macros.c
index 29b1d29..b4d2906 100644
--- a/test/Preprocessor/dump_macros.c
+++ b/test/Preprocessor/dump_macros.c
@@ -27,3 +27,12 @@
#define G 1
#undef G
#define G 2
+
+// Variadic macros of various sorts. PR5699
+
+// CHECK: H(x,...) __VA_ARGS__
+#define H(x, ...) __VA_ARGS__
+// CHECK: I(...) __VA_ARGS__
+#define I(...) __VA_ARGS__
+// CHECK: J(x...) __VA_ARGS__
+#define J(x ...) __VA_ARGS__