LLVM: lib/Support/FormatVariadic.cpp Source File

LLVM 22.0.0git
FormatVariadic.cpp
Go to the documentation of this file.
1//===- FormatVariadic.cpp - Format string parsing and analysis ----*-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#include "llvm/Support/FormatVariadic.h"
9#include <cassert>
10#include <optional>
11
12using namespace llvm;
13
14 static std::optional<AlignStyle> translateLocChar(char C) {
15 switch (C) {
16 case '-':
17 return AlignStyle::Left;
18 case '=':
19 return AlignStyle::Center;
20 case '+':
21 return AlignStyle::Right;
22 default:
23 return std::nullopt;
24 }
25 LLVM_BUILTIN_UNREACHABLE;
26}
27
29 unsigned &Align, char &Pad) {
30 Where = AlignStyle::Right;
31 Align = 0;
32 Pad = ' ';
33 if (Spec.empty())
34 return true;
35
36 if (Spec.size() > 1) {
37 // A maximum of 2 characters at the beginning can be used for something
38 // other than the width.
39 // If Spec[1] is a loc char, then Spec[0] is a pad char and Spec[2:...]
40 // contains the width.
41 // Otherwise, if Spec[0] is a loc char, then Spec[1:...] contains the width.
42 // Otherwise, Spec[0:...] contains the width.
43 if (auto Loc = translateLocChar(Spec[1])) {
44 Pad = Spec[0];
45 Where = *Loc;
46 Spec = Spec.drop_front(2);
47 } else if (auto Loc = translateLocChar(Spec[0])) {
48 Where = *Loc;
49 Spec = Spec.drop_front(1);
50 }
51 }
52
53 bool Failed = Spec.consumeInteger(0, Align);
54 return !Failed;
55}
56
57 static std::optional<ReplacementItem> parseReplacementItem(StringRef Spec) {
58 StringRef RepString = Spec.trim("{}");
59
60 // If the replacement sequence does not start with a non-negative integer,
61 // this is an error.
62 char Pad = ' ';
63 unsigned Align = 0;
66 unsigned Index = ~0U;
67 RepString = RepString.ltrim();
68
69 // If index is not specified, keep it ~0U to indicate unresolved index.
70 RepString.consumeInteger(0, Index);
71
72 if (RepString.consume_front(",")) {
73 if (!consumeFieldLayout(RepString, Where, Align, Pad)) {
74 assert(false && "Invalid replacement field layout specification!");
75 return std::nullopt;
76 }
77 }
78 RepString = RepString.ltrim();
79 if (RepString.consume_front(":")) {
80 Options = RepString;
81 RepString = StringRef();
82 }
83 RepString = RepString.trim();
84 if (!RepString.empty()) {
85 assert(0 && "Unexpected characters found in replacement string!");
86 return std::nullopt;
87 }
88
89 return ReplacementItem(Spec, Index, Align, Where, Pad, Options);
90}
91
92static std::pair<std::optional<ReplacementItem>, StringRef>
94 assert(!Fmt.empty());
95 // Everything up until the first brace is a literal.
96 if (Fmt.front() != '{') {
97 size_t BO = Fmt.find_first_of('{');
98 return {ReplacementItem{Fmt.substr(0, BO)}, Fmt.substr(BO)};
99 }
100
101 StringRef Braces = Fmt.take_while([](char C) { return C == '{'; });
102 // If there is more than one brace, then some of them are escaped. Treat
103 // these as replacements.
104 if (Braces.size() > 1) {
105 size_t NumEscapedBraces = Braces.size() / 2;
106 StringRef Middle = Fmt.take_front(NumEscapedBraces);
107 StringRef Right = Fmt.drop_front(NumEscapedBraces * 2);
108 return {ReplacementItem(Middle), Right};
109 }
110 // An unterminated open brace is undefined. Assert to indicate that this is
111 // undefined and that we consider it an error. When asserts are disabled,
112 // build a replacement item with an error message.
113 size_t BC = Fmt.find_first_of('}');
114 if (BC == StringRef::npos) {
115 assert(false &&
116 "Unterminated brace sequence. Escape with {{ for a literal brace.");
117 return {ReplacementItem("Unterminated brace sequence. Escape with {{ for a "
118 "literal brace."),
119 StringRef()};
120 }
121
122 // Even if there is a closing brace, if there is another open brace before
123 // this closing brace, treat this portion as literal, and try again with the
124 // next one.
125 size_t BO2 = Fmt.find_first_of('{', 1);
126 if (BO2 < BC)
127 return {ReplacementItem(Fmt.substr(0, BO2)), Fmt.substr(BO2)};
128
129 StringRef Spec = Fmt.slice(1, BC);
130 StringRef Right = Fmt.substr(BC + 1);
131
133}
134
135#ifndef NDEBUG
136 #define ENABLE_VALIDATION 1
137#else
138#define ENABLE_VALIDATION 0 // Conveniently enable validation in release mode.
139#endif
140
141SmallVector<ReplacementItem, 2>
143 bool Validate) {
145 unsigned NextAutomaticIndex = 0;
146
147#if ENABLE_VALIDATION
148 const StringRef SavedFmtStr = Fmt;
149 unsigned NumExpectedArgs = 0;
150 bool HasExplicitIndex = false;
151#endif
152
153 while (!Fmt.empty()) {
154 std::optional<ReplacementItem> I;
155 std::tie(I, Fmt) = splitLiteralAndReplacement(Fmt);
156 if (!I)
157 continue;
158 if (I->Type == ReplacementType::Format) {
159 if (I->Index == ~0U)
160 I->Index = NextAutomaticIndex++;
161#if ENABLE_VALIDATION
162 else
163 HasExplicitIndex = true;
164 NumExpectedArgs = std::max(NumExpectedArgs, I->Index + 1);
165#endif
166 }
167
168 Replacements.emplace_back(*I);
169 }
170
171#if ENABLE_VALIDATION
172 if (!Validate)
173 return Replacements;
174
175 // Perform additional validation. Verify that the number of arguments matches
176 // the number of replacement indices and that there are no holes in the
177 // replacement indices.
178
179 // When validation fails, return an array of replacement items that
180 // will print an error message as the outout of this formatv() (used when
181 // validation is enabled in release mode).
182 auto getErrorReplacements = [SavedFmtStr](StringLiteral ErrorMsg) {
184 ReplacementItem("Invalid formatv() call: "), ReplacementItem(ErrorMsg),
185 ReplacementItem(" for format string: "), ReplacementItem(SavedFmtStr)};
186 };
187
188 if (NumExpectedArgs != NumArgs) {
189 errs() << formatv("Expected {} Args, but got {} for format string '{}'\n",
190 NumExpectedArgs, NumArgs, SavedFmtStr);
191 assert(0 && "Invalid formatv() call");
192 return getErrorReplacements("Unexpected number of arguments");
193 }
194
195 // Find the number of unique indices seen. All replacement indices
196 // are < NumExpectedArgs.
197 SmallVector<bool> Indices(NumExpectedArgs);
198 unsigned Count = 0;
199 for (const ReplacementItem &I : Replacements) {
200 if (I.Type != ReplacementType::Format || Indices[I.Index])
201 continue;
202 Indices[I.Index] = true;
203 ++Count;
204 }
205
206 if (Count != NumExpectedArgs) {
207 errs() << formatv(
208 "Replacement field indices cannot have holes for format string '{}'\n",
209 SavedFmtStr);
210 assert(0 && "Invalid format string");
211 return getErrorReplacements("Replacement indices have holes");
212 }
213
214 // Fail validation if we see both automatic index and explicit index.
215 if (NextAutomaticIndex != 0 && HasExplicitIndex) {
216 errs() << formatv(
217 "Cannot mix automatic and explicit indices for format string '{}'\n",
218 SavedFmtStr);
219 assert(0 && "Invalid format string");
220 return getErrorReplacements("Cannot mix automatic and explicit indices");
221 }
222#endif // ENABLE_VALIDATION
223 return Replacements;
224}
225
226void support::detail::format_adapter::anchor() {}
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
static bool consumeFieldLayout(StringRef &Spec, AlignStyle &Where, unsigned &Align, char &Pad)
static std::pair< std::optional< ReplacementItem >, StringRef > splitLiteralAndReplacement(StringRef Fmt)
static std::optional< ReplacementItem > parseReplacementItem(StringRef Spec)
static std::optional< AlignStyle > translateLocChar(char C)
static LVOptions Options
Definition LVOptions.cpp:25
I
#define I(x, y, z)
Definition MD5.cpp:58
reference emplace_back(ArgTypes &&... Args)
Definition SmallVector.h:944
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
Definition SmallVector.h:1203
A wrapper around a string literal that serves as a proxy for constructing global tables of StringRefs...
Definition StringRef.h:854
StringRef - Represent a constant reference to a string, i.e.
Definition StringRef.h:55
static constexpr size_t npos
Definition StringRef.h:57
bool consumeInteger(unsigned Radix, T &Result)
Parse the current string as an integer of the specified radix.
Definition StringRef.h:501
constexpr StringRef substr(size_t Start, size_t N=npos) const
Return a reference to the substring from [Start, Start + N).
Definition StringRef.h:573
constexpr bool empty() const
empty - Check if the string is empty.
Definition StringRef.h:143
StringRef take_while(function_ref< bool(char)> F) const
Return the longest prefix of 'this' such that every character in the prefix satisfies the given predi...
Definition StringRef.h:599
StringRef drop_front(size_t N=1) const
Return a StringRef equal to 'this' but with the first N elements dropped.
Definition StringRef.h:611
StringRef slice(size_t Start, size_t End) const
Return a reference to the substring from [Start, End).
Definition StringRef.h:686
constexpr size_t size() const
size - Get the string size.
Definition StringRef.h:146
char front() const
front - Get the first character in the string.
Definition StringRef.h:149
StringRef ltrim(char Char) const
Return string with consecutive Char characters starting from the the left removed.
Definition StringRef.h:792
bool consume_front(StringRef Prefix)
Returns true if this StringRef has the given prefix and removes that prefix.
Definition StringRef.h:637
size_t find_first_of(char C, size_t From=0) const
Find the first character in the string that is C, or npos if not found.
Definition StringRef.h:376
StringRef take_front(size_t N=1) const
Return a StringRef equal to 'this' but with only the first N elements remaining.
Definition StringRef.h:582
StringRef trim(char Char) const
Return string with consecutive Char characters starting from the left and right removed.
Definition StringRef.h:816
bool Validate
static LLVM_ABI SmallVector< ReplacementItem, 2 > parseFormatString(StringRef Fmt, size_t NumArgs, bool Validate)
StringRef Fmt
@ C
The default llvm calling convention, compatible with C.
Definition CallingConv.h:34
This is an optimization pass for GlobalISel generic memory operations.
Definition AddressRanges.h:18
AlignStyle
Definition FormatCommon.h:17
testing::Matcher< const detail::ErrorHolder & > Failed()
Definition Error.h:198
auto formatv(bool Validate, const char *Fmt, Ts &&...Vals)
FunctionAddr VTableAddr Count
Definition InstrProf.h:139
LLVM_ABI raw_fd_ostream & errs()
This returns a reference to a raw_ostream for standard error.
This struct is a compact representation of a valid (non-zero power of two) alignment.
Definition Alignment.h:39
Generated on for LLVM by doxygen 1.14.0

AltStyle によって変換されたページ (->オリジナル) /