| 1 | //===--- FormatStringConverter.h - clang-tidy--------------------*- C++ -*-===// |
| 2 | // |
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | /// |
| 9 | /// \file |
| 10 | /// Declaration of the FormatStringConverter class which is used to convert |
| 11 | /// printf format strings to C++ std::formatter format strings. |
| 12 | /// |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #ifndef LLVM_CLANG_TOOLS_EXTRA_CLANG_TIDY_UTILS_FORMATSTRINGCONVERTER_H |
| 16 | #define |
| 17 | |
| 18 | #include "clang/AST/ASTContext.h" |
| 19 | #include "clang/AST/FormatString.h" |
| 20 | #include "clang/ASTMatchers/ASTMatchers.h" |
| 21 | #include <string> |
| 22 | |
| 23 | namespace clang::tidy::utils { |
| 24 | |
| 25 | /// Convert a printf-style format string to a std::formatter-style one, and |
| 26 | /// prepare any casts that are required to wrap the arguments to retain printf |
| 27 | /// compatibility. This class is expecting to work on the already-cooked format |
| 28 | /// string (i.e. all the escapes have been converted) so we have to convert them |
| 29 | /// back. This means that we might not convert them back using the same form. |
| 30 | class FormatStringConverter |
| 31 | : public clang::analyze_format_string::FormatStringHandler { |
| 32 | public: |
| 33 | using ConversionSpecifier = clang::analyze_format_string::ConversionSpecifier; |
| 34 | using PrintfSpecifier = analyze_printf::PrintfSpecifier; |
| 35 | |
| 36 | struct Configuration { |
| 37 | bool StrictMode = false; |
| 38 | bool AllowTrailingNewlineRemoval = false; |
| 39 | }; |
| 40 | |
| 41 | FormatStringConverter(ASTContext *Context, const CallExpr *Call, |
| 42 | unsigned FormatArgOffset, Configuration Config, |
| 43 | const LangOptions &LO, SourceManager &SM, |
| 44 | Preprocessor &PP); |
| 45 | |
| 46 | bool canApply() const { return ConversionNotPossibleReason.empty(); } |
| 47 | const std::string &conversionNotPossibleReason() const { |
| 48 | return ConversionNotPossibleReason; |
| 49 | } |
| 50 | void applyFixes(DiagnosticBuilder &Diag, SourceManager &SM); |
| 51 | bool usePrintNewlineFunction() const { return UsePrintNewlineFunction; } |
| 52 | |
| 53 | private: |
| 54 | ASTContext *Context; |
| 55 | const Configuration Config; |
| 56 | const bool CastMismatchedIntegerTypes; |
| 57 | const Expr *const *Args; |
| 58 | const unsigned NumArgs; |
| 59 | unsigned ArgsOffset; |
| 60 | const LangOptions &LangOpts; |
| 61 | std::string ConversionNotPossibleReason; |
| 62 | bool FormatStringNeededRewriting = false; |
| 63 | bool UsePrintNewlineFunction = false; |
| 64 | size_t PrintfFormatStringPos = 0U; |
| 65 | StringRef PrintfFormatString; |
| 66 | |
| 67 | /// Lazily-created c_str() call matcher |
| 68 | std::optional<clang::ast_matchers::StatementMatcher> |
| 69 | StringCStrCallExprMatcher; |
| 70 | |
| 71 | const StringLiteral *FormatExpr; |
| 72 | std::string StandardFormatString; |
| 73 | |
| 74 | /// Casts to be used to wrap arguments to retain printf compatibility. |
| 75 | struct ArgumentFix { |
| 76 | unsigned ArgIndex; |
| 77 | std::string Fix; |
| 78 | |
| 79 | // We currently need this for emplace_back. Roll on C++20. |
| 80 | explicit ArgumentFix(unsigned ArgIndex, std::string Fix) |
| 81 | : ArgIndex(ArgIndex), Fix(std::move(Fix)) {} |
| 82 | }; |
| 83 | |
| 84 | std::vector<ArgumentFix> ArgFixes; |
| 85 | std::vector<clang::ast_matchers::BoundNodes> ArgCStrRemovals; |
| 86 | |
| 87 | // Argument rotations to cope with the fact that std::print puts the value to |
| 88 | // be formatted first and the width and precision afterwards whereas printf |
| 89 | // puts the width and preicision first. |
| 90 | std::vector<std::tuple<unsigned, unsigned>> ArgRotates; |
| 91 | |
| 92 | void emitAlignment(const PrintfSpecifier &FS, std::string &FormatSpec); |
| 93 | void emitSign(const PrintfSpecifier &FS, std::string &FormatSpec); |
| 94 | void emitAlternativeForm(const PrintfSpecifier &FS, std::string &FormatSpec); |
| 95 | void emitFieldWidth(const PrintfSpecifier &FS, std::string &FormatSpec); |
| 96 | void emitPrecision(const PrintfSpecifier &FS, std::string &FormatSpec); |
| 97 | void emitStringArgument(unsigned ArgIndex, const Expr *Arg); |
| 98 | bool emitIntegerArgument(ConversionSpecifier::Kind ArgKind, const Expr *Arg, |
| 99 | unsigned ArgIndex, std::string &FormatSpec); |
| 100 | |
| 101 | bool emitType(const PrintfSpecifier &FS, const Expr *Arg, |
| 102 | std::string &FormatSpec); |
| 103 | bool convertArgument(const PrintfSpecifier &FS, const Expr *Arg, |
| 104 | std::string &StandardFormatString); |
| 105 | |
| 106 | void maybeRotateArguments(const PrintfSpecifier &FS); |
| 107 | |
| 108 | bool HandlePrintfSpecifier(const PrintfSpecifier &FS, |
| 109 | const char *StartSpecifier, unsigned SpecifierLen, |
| 110 | const TargetInfo &Target) override; |
| 111 | |
| 112 | void appendFormatText(StringRef Text); |
| 113 | void finalizeFormatText(); |
| 114 | static std::optional<StringRef> |
| 115 | formatStringContainsUnreplaceableMacro(const CallExpr *CallExpr, |
| 116 | const StringLiteral *FormatExpr, |
| 117 | SourceManager &SM, Preprocessor &PP); |
| 118 | bool conversionNotPossible(std::string Reason) { |
| 119 | ConversionNotPossibleReason = std::move(Reason); |
| 120 | return false; |
| 121 | } |
| 122 | }; |
| 123 | |
| 124 | } // namespace clang::tidy::utils |
| 125 | |
| 126 | #endif // LLVM_CLANG_TOOLS_EXTRA_CLANG_TIDY_UTILS_FORMATSTRINGCONVERTER_H |
| 127 | |