================ @@ -5335,6 +5335,221 @@ static void handlePreferredTypeAttr(Sema &S, Decl *D, const ParsedAttr &AL) { D->addAttr(::new (S.Context) PreferredTypeAttr(S.Context, AL, ParmTSI)); } +// This function is called only if function call is not inside template body. +// TODO: Add call for function calls inside template body. +// Emit warnings if parent function misses format attributes. +void Sema::DiagnoseMissingFormatAttributes(Stmt *Body, + const FunctionDecl *FDecl) { + assert(FDecl); + + // If there are no function body, exit. + if (!Body) + return; + + // Get missing format attributes + std::vector<FormatAttr *> MissingFormatAttributes = + GetMissingFormatAttributes(Body, FDecl); + if (MissingFormatAttributes.empty()) + return; + + // Check if there are more than one format type found. In that case do not + // emit diagnostic. + const clang::IdentifierInfo *AttrType = MissingFormatAttributes[0]->getType(); + if (llvm::any_of(MissingFormatAttributes, [&](const FormatAttr *Attr) { + return AttrType != Attr->getType(); + })) + return; + + for (const FormatAttr *FA : MissingFormatAttributes) { + // If format index and first-to-check argument index are negative, it means + // that this attribute is only saved for multiple format types checking. + if (FA->getFormatIdx() < 0 || FA->getFirstArg() < 0) + continue; + + // Emit diagnostic + SourceLocation Loc = FDecl->getLocation(); + Diag(Loc, diag::warn_missing_format_attribute) + << FA->getType() << FDecl + << FixItHint::CreateInsertion(Loc, + (llvm::Twine("__attribute__((format(") + + FA->getType()->getName() + ", " + + llvm::Twine(FA->getFormatIdx()) + ", " + + llvm::Twine(FA->getFirstArg()) + ")))") + .str()); + } +} + +// Returns vector of format attributes. There are no two attributes with same +// arguments in returning vector. There can be attributes that effectivelly only +// store information about format type. +std::vector<FormatAttr *> +Sema::GetMissingFormatAttributes(Stmt *Body, const FunctionDecl *FDecl) { + unsigned int FunctionFormatArgumentIndexOffset = + checkIfMethodHasImplicitObjectParameter(FDecl) ? 2 : 1; + + std::vector<FormatAttr *> MissingAttributes; + + // Iterate over body statements. + for (auto *Child : Body->children()) { + // If child statement is compound statement, recursively get missing + // attributes. + if (dyn_cast_or_null<CompoundStmt>(Child)) { + std::vector<FormatAttr *> CompoundStmtMissingAttributes = + GetMissingFormatAttributes(Child, FDecl); + + // If there are already missing attributes with same arguments, do not add + // duplicates. + for (FormatAttr *FA : CompoundStmtMissingAttributes) { + if (!llvm::any_of(MissingAttributes, [&](const FormatAttr *Attr) { + return FA->getType() == Attr->getType() && + FA->getFormatIdx() == Attr->getFormatIdx() && + FA->getFirstArg() == Attr->getFirstArg(); + })) + MissingAttributes.push_back(FA); + } + + continue; + } + + ValueStmt *VS = dyn_cast_or_null<ValueStmt>(Child); + if (!VS) + continue; + Expr *TheExpr = VS->getExprStmt(); + if (!TheExpr) + continue; + CallExpr *TheCall = dyn_cast_or_null<CallExpr>(TheExpr); + if (!TheCall) + continue; + const FunctionDecl *ChildFunction = + dyn_cast_or_null<FunctionDecl>(TheCall->getCalleeDecl()); + if (!ChildFunction) + continue; + + Expr **Args = TheCall->getArgs(); + unsigned int NumArgs = TheCall->getNumArgs(); + + // If child expression is function, check if it is format function. + // If it is, check if parent function misses format attributes. + + unsigned int ChildFunctionFormatArgumentIndexOffset = + checkIfMethodHasImplicitObjectParameter(ChildFunction) ? 2 : 1; + + if (!ChildFunction->hasAttr<FormatAttr>()) + continue; ---------------- AaronBallman wrote:
```suggestion ``` https://github.com/llvm/llvm-project/pull/105479 _______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits