Line data Source code
1 : // Protocol Buffers - Google's data interchange format
2 : // Copyright 2008 Google Inc. All rights reserved.
3 : // https://developers.google.com/protocol-buffers/
4 : //
5 : // Redistribution and use in source and binary forms, with or without
6 : // modification, are permitted provided that the following conditions are
7 : // met:
8 : //
9 : // * Redistributions of source code must retain the above copyright
10 : // notice, this list of conditions and the following disclaimer.
11 : // * Redistributions in binary form must reproduce the above
12 : // copyright notice, this list of conditions and the following disclaimer
13 : // in the documentation and/or other materials provided with the
14 : // distribution.
15 : // * Neither the name of Google Inc. nor the names of its
16 : // contributors may be used to endorse or promote products derived from
17 : // this software without specific prior written permission.
18 : //
19 : // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 : // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 : // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 : // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 : // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 : // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 : // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 : // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 : // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 : // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 : // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 : #include <google/protobuf/stubs/stringpiece.h>
31 :
32 : #include <string.h>
33 : #include <algorithm>
34 : #include <climits>
35 : #include <string>
36 : #include <ostream>
37 :
38 : namespace google {
39 : namespace protobuf {
40 0 : std::ostream& operator<<(std::ostream& o, StringPiece piece) {
41 0 : o.write(piece.data(), piece.size());
42 0 : return o;
43 : }
44 :
45 : // Out-of-line error path.
46 0 : void StringPiece::LogFatalSizeTooBig(size_t size, const char* details) {
47 0 : GOOGLE_LOG(FATAL) << "size too big: " << size << " details: " << details;
48 0 : }
49 :
50 0 : StringPiece::StringPiece(StringPiece x, stringpiece_ssize_type pos)
51 0 : : ptr_(x.ptr_ + pos), length_(x.length_ - pos) {
52 : GOOGLE_DCHECK_LE(0, pos);
53 : GOOGLE_DCHECK_LE(pos, x.length_);
54 0 : }
55 :
56 0 : StringPiece::StringPiece(StringPiece x,
57 : stringpiece_ssize_type pos,
58 : stringpiece_ssize_type len)
59 0 : : ptr_(x.ptr_ + pos), length_(std::min(len, x.length_ - pos)) {
60 : GOOGLE_DCHECK_LE(0, pos);
61 : GOOGLE_DCHECK_LE(pos, x.length_);
62 : GOOGLE_DCHECK_GE(len, 0);
63 0 : }
64 :
65 0 : void StringPiece::CopyToString(string* target) const {
66 0 : target->assign(ptr_, length_);
67 0 : }
68 :
69 0 : void StringPiece::AppendToString(string* target) const {
70 0 : target->append(ptr_, length_);
71 0 : }
72 :
73 0 : bool StringPiece::Consume(StringPiece x) {
74 0 : if (starts_with(x)) {
75 0 : ptr_ += x.length_;
76 0 : length_ -= x.length_;
77 0 : return true;
78 : }
79 : return false;
80 : }
81 :
82 0 : bool StringPiece::ConsumeFromEnd(StringPiece x) {
83 0 : if (ends_with(x)) {
84 0 : length_ -= x.length_;
85 0 : return true;
86 : }
87 : return false;
88 : }
89 :
90 0 : stringpiece_ssize_type StringPiece::copy(char* buf,
91 : size_type n,
92 : size_type pos) const {
93 0 : stringpiece_ssize_type ret = std::min(length_ - pos, n);
94 0 : memcpy(buf, ptr_ + pos, ret);
95 0 : return ret;
96 : }
97 :
98 0 : bool StringPiece::contains(StringPiece s) const {
99 0 : return find(s, 0) != npos;
100 : }
101 :
102 0 : stringpiece_ssize_type StringPiece::find(StringPiece s, size_type pos) const {
103 0 : if (length_ <= 0 || pos > static_cast<size_type>(length_)) {
104 0 : if (length_ == 0 && pos == 0 && s.length_ == 0) return 0;
105 0 : return npos;
106 : }
107 : const char *result = std::search(ptr_ + pos, ptr_ + length_,
108 0 : s.ptr_, s.ptr_ + s.length_);
109 0 : return result == ptr_ + length_ ? npos : result - ptr_;
110 : }
111 :
112 0 : stringpiece_ssize_type StringPiece::find(char c, size_type pos) const {
113 0 : if (length_ <= 0 || pos >= static_cast<size_type>(length_)) {
114 : return npos;
115 : }
116 : const char* result = static_cast<const char*>(
117 0 : memchr(ptr_ + pos, c, length_ - pos));
118 0 : return result != NULL ? result - ptr_ : npos;
119 : }
120 :
121 0 : stringpiece_ssize_type StringPiece::rfind(StringPiece s, size_type pos) const {
122 0 : if (length_ < s.length_) return npos;
123 0 : const size_t ulen = length_;
124 0 : if (s.length_ == 0) return std::min(ulen, pos);
125 :
126 0 : const char* last = ptr_ + std::min(ulen - s.length_, pos) + s.length_;
127 0 : const char* result = std::find_end(ptr_, last, s.ptr_, s.ptr_ + s.length_);
128 0 : return result != last ? result - ptr_ : npos;
129 : }
130 :
131 : // Search range is [0..pos] inclusive. If pos == npos, search everything.
132 0 : stringpiece_ssize_type StringPiece::rfind(char c, size_type pos) const {
133 : // Note: memrchr() is not available on Windows.
134 0 : if (length_ <= 0) return npos;
135 0 : for (stringpiece_ssize_type i =
136 0 : std::min(pos, static_cast<size_type>(length_ - 1));
137 : i >= 0; --i) {
138 0 : if (ptr_[i] == c) {
139 : return i;
140 : }
141 : }
142 : return npos;
143 : }
144 :
145 : // For each character in characters_wanted, sets the index corresponding
146 : // to the ASCII code of that character to 1 in table. This is used by
147 : // the find_.*_of methods below to tell whether or not a character is in
148 : // the lookup table in constant time.
149 : // The argument `table' must be an array that is large enough to hold all
150 : // the possible values of an unsigned char. Thus it should be be declared
151 : // as follows:
152 : // bool table[UCHAR_MAX + 1]
153 : static inline void BuildLookupTable(StringPiece characters_wanted,
154 : bool* table) {
155 0 : const stringpiece_ssize_type length = characters_wanted.length();
156 0 : const char* const data = characters_wanted.data();
157 0 : for (stringpiece_ssize_type i = 0; i < length; ++i) {
158 0 : table[static_cast<unsigned char>(data[i])] = true;
159 : }
160 : }
161 :
162 0 : stringpiece_ssize_type StringPiece::find_first_of(StringPiece s,
163 : size_type pos) const {
164 0 : if (length_ <= 0 || s.length_ <= 0) {
165 : return npos;
166 : }
167 : // Avoid the cost of BuildLookupTable() for a single-character search.
168 0 : if (s.length_ == 1) return find_first_of(s.ptr_[0], pos);
169 :
170 0 : bool lookup[UCHAR_MAX + 1] = { false };
171 : BuildLookupTable(s, lookup);
172 0 : for (stringpiece_ssize_type i = pos; i < length_; ++i) {
173 0 : if (lookup[static_cast<unsigned char>(ptr_[i])]) {
174 : return i;
175 : }
176 : }
177 : return npos;
178 : }
179 :
180 0 : stringpiece_ssize_type StringPiece::find_first_not_of(StringPiece s,
181 : size_type pos) const {
182 0 : if (length_ <= 0) return npos;
183 0 : if (s.length_ <= 0) return 0;
184 : // Avoid the cost of BuildLookupTable() for a single-character search.
185 0 : if (s.length_ == 1) return find_first_not_of(s.ptr_[0], pos);
186 :
187 0 : bool lookup[UCHAR_MAX + 1] = { false };
188 : BuildLookupTable(s, lookup);
189 0 : for (stringpiece_ssize_type i = pos; i < length_; ++i) {
190 0 : if (!lookup[static_cast<unsigned char>(ptr_[i])]) {
191 : return i;
192 : }
193 : }
194 : return npos;
195 : }
196 :
197 0 : stringpiece_ssize_type StringPiece::find_first_not_of(char c,
198 : size_type pos) const {
199 0 : if (length_ <= 0) return npos;
200 :
201 0 : for (; pos < static_cast<size_type>(length_); ++pos) {
202 0 : if (ptr_[pos] != c) {
203 0 : return pos;
204 : }
205 : }
206 : return npos;
207 : }
208 :
209 0 : stringpiece_ssize_type StringPiece::find_last_of(StringPiece s,
210 : size_type pos) const {
211 0 : if (length_ <= 0 || s.length_ <= 0) return npos;
212 : // Avoid the cost of BuildLookupTable() for a single-character search.
213 0 : if (s.length_ == 1) return find_last_of(s.ptr_[0], pos);
214 :
215 0 : bool lookup[UCHAR_MAX + 1] = { false };
216 : BuildLookupTable(s, lookup);
217 0 : for (stringpiece_ssize_type i =
218 0 : std::min(pos, static_cast<size_type>(length_ - 1)); i >= 0; --i) {
219 0 : if (lookup[static_cast<unsigned char>(ptr_[i])]) {
220 : return i;
221 : }
222 : }
223 : return npos;
224 : }
225 :
226 0 : stringpiece_ssize_type StringPiece::find_last_not_of(StringPiece s,
227 : size_type pos) const {
228 0 : if (length_ <= 0) return npos;
229 :
230 0 : stringpiece_ssize_type i = std::min(pos, static_cast<size_type>(length_ - 1));
231 0 : if (s.length_ <= 0) return i;
232 :
233 : // Avoid the cost of BuildLookupTable() for a single-character search.
234 0 : if (s.length_ == 1) return find_last_not_of(s.ptr_[0], pos);
235 :
236 0 : bool lookup[UCHAR_MAX + 1] = { false };
237 : BuildLookupTable(s, lookup);
238 0 : for (; i >= 0; --i) {
239 0 : if (!lookup[static_cast<unsigned char>(ptr_[i])]) {
240 : return i;
241 : }
242 : }
243 : return npos;
244 : }
245 :
246 0 : stringpiece_ssize_type StringPiece::find_last_not_of(char c,
247 : size_type pos) const {
248 0 : if (length_ <= 0) return npos;
249 :
250 0 : for (stringpiece_ssize_type i =
251 0 : std::min(pos, static_cast<size_type>(length_ - 1)); i >= 0; --i) {
252 0 : if (ptr_[i] != c) {
253 : return i;
254 : }
255 : }
256 : return npos;
257 : }
258 :
259 0 : StringPiece StringPiece::substr(size_type pos, size_type n) const {
260 0 : if (pos > length_) pos = length_;
261 0 : if (n > length_ - pos) n = length_ - pos;
262 0 : return StringPiece(ptr_ + pos, n);
263 : }
264 :
265 : const StringPiece::size_type StringPiece::npos = size_type(-1);
266 :
267 : } // namespace protobuf
268 : } // namespace google
|