diff --git a/clang/lib/Analysis/FlowSensitive/DataflowEnvironment.cpp b/clang/lib/Analysis/FlowSensitive/DataflowEnvironment.cpp index 07dc3a9f76ac2..196a1360a7750 100644 --- a/clang/lib/Analysis/FlowSensitive/DataflowEnvironment.cpp +++ b/clang/lib/Analysis/FlowSensitive/DataflowEnvironment.cpp @@ -782,8 +782,13 @@ Environment::getResultObjectLocation(const Expr &RecordPRValue) const { return Val->getLoc(); } - // Expression nodes that propagate a record prvalue should have exactly one - // child. + if (auto *Op = dyn_cast(&RecordPRValue); + Op && Op->isCommaOp()) { + return getResultObjectLocation(*Op->getRHS()); + } + + // All other expression nodes that propagate a record prvalue should have + // exactly one child. llvm::SmallVector children(RecordPRValue.child_begin(), RecordPRValue.child_end()); assert(children.size() == 1); diff --git a/clang/unittests/Analysis/FlowSensitive/TransferTest.cpp b/clang/unittests/Analysis/FlowSensitive/TransferTest.cpp index d0a0e6d3f5836..85ae24f0b6f16 100644 --- a/clang/unittests/Analysis/FlowSensitive/TransferTest.cpp +++ b/clang/unittests/Analysis/FlowSensitive/TransferTest.cpp @@ -2642,14 +2642,17 @@ TEST(TransferTest, ResultObjectLocation) { }; void target() { - A(); + 0, A(); (void)0; // [[p]] } )"; + using ast_matchers::binaryOperator; using ast_matchers::cxxBindTemporaryExpr; using ast_matchers::cxxTemporaryObjectExpr; using ast_matchers::exprWithCleanups; using ast_matchers::has; + using ast_matchers::hasOperatorName; + using ast_matchers::hasRHS; using ast_matchers::match; using ast_matchers::selectFirst; using ast_matchers::traverse; @@ -2659,26 +2662,33 @@ TEST(TransferTest, ResultObjectLocation) { ASTContext &ASTCtx) { const Environment &Env = getEnvironmentAtAnnotation(Results, "p"); - // The expresssion `A()` in the code above produces the following - // structure, consisting of three prvalues of record type. + // The expression `0, A()` in the code above produces the following + // structure, consisting of four prvalues of record type. // `Env.getResultObjectLocation()` should return the same location for // all of these. auto MatchResult = match( traverse(TK_AsIs, exprWithCleanups( - has(cxxBindTemporaryExpr( - has(cxxTemporaryObjectExpr().bind("toe"))) - .bind("bte"))) + has(binaryOperator( + hasOperatorName(","), + hasRHS(cxxBindTemporaryExpr( + has(cxxTemporaryObjectExpr().bind( + "toe"))) + .bind("bte"))) + .bind("comma"))) .bind("ewc")), ASTCtx); auto *TOE = selectFirst("toe", MatchResult); ASSERT_NE(TOE, nullptr); + auto *Comma = selectFirst("comma", MatchResult); + ASSERT_NE(Comma, nullptr); auto *EWC = selectFirst("ewc", MatchResult); ASSERT_NE(EWC, nullptr); auto *BTE = selectFirst("bte", MatchResult); ASSERT_NE(BTE, nullptr); RecordStorageLocation &Loc = Env.getResultObjectLocation(*TOE); + EXPECT_EQ(&Loc, &Env.getResultObjectLocation(*Comma)); EXPECT_EQ(&Loc, &Env.getResultObjectLocation(*EWC)); EXPECT_EQ(&Loc, &Env.getResultObjectLocation(*BTE)); });