| /* |
| * Copyright (C) 2015 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "util/Util.h" |
| |
| #include <utils/Unicode.h> |
| #include <algorithm> |
| #include <ostream> |
| #include <string> |
| #include <vector> |
| |
| #include "androidfw/StringPiece.h" |
| |
| #include "util/BigBuffer.h" |
| #include "util/Maybe.h" |
| |
| using android::StringPiece; |
| using android::StringPiece16; |
| |
| namespace aapt { |
| namespace util { |
| |
| static std::vector<std::string> SplitAndTransform( |
| const StringPiece& str, char sep, const std::function<char(char)>& f) { |
| std::vector<std::string> parts; |
| const StringPiece::const_iterator end = std::end(str); |
| StringPiece::const_iterator start = std::begin(str); |
| StringPiece::const_iterator current; |
| do { |
| current = std::find(start, end, sep); |
| parts.emplace_back(str.substr(start, current).to_string()); |
| if (f) { |
| std::string& part = parts.back(); |
| std::transform(part.begin(), part.end(), part.begin(), f); |
| } |
| start = current + 1; |
| } while (current != end); |
| return parts; |
| } |
| |
| std::vector<std::string> Split(const StringPiece& str, char sep) { |
| return SplitAndTransform(str, sep, nullptr); |
| } |
| |
| std::vector<std::string> SplitAndLowercase(const StringPiece& str, char sep) { |
| return SplitAndTransform(str, sep, ::tolower); |
| } |
| |
| bool StartsWith(const StringPiece& str, const StringPiece& prefix) { |
| if (str.size() < prefix.size()) { |
| return false; |
| } |
| return str.substr(0, prefix.size()) == prefix; |
| } |
| |
| bool EndsWith(const StringPiece& str, const StringPiece& suffix) { |
| if (str.size() < suffix.size()) { |
| return false; |
| } |
| return str.substr(str.size() - suffix.size(), suffix.size()) == suffix; |
| } |
| |
| StringPiece TrimWhitespace(const StringPiece& str) { |
| if (str.size() == 0 || str.data() == nullptr) { |
| return str; |
| } |
| |
| const char* start = str.data(); |
| const char* end = str.data() + str.length(); |
| |
| while (start != end && isspace(*start)) { |
| start++; |
| } |
| |
| while (end != start && isspace(*(end - 1))) { |
| end--; |
| } |
| |
| return StringPiece(start, end - start); |
| } |
| |
| StringPiece::const_iterator FindNonAlphaNumericAndNotInSet( |
| const StringPiece& str, const StringPiece& allowed_chars) { |
| const auto end_iter = str.end(); |
| for (auto iter = str.begin(); iter != end_iter; ++iter) { |
| char c = *iter; |
| if ((c >= u'a' && c <= u'z') || (c >= u'A' && c <= u'Z') || |
| (c >= u'0' && c <= u'9')) { |
| continue; |
| } |
| |
| bool match = false; |
| for (char i : allowed_chars) { |
| if (c == i) { |
| match = true; |
| break; |
| } |
| } |
| |
| if (!match) { |
| return iter; |
| } |
| } |
| return end_iter; |
| } |
| |
| bool IsJavaClassName(const StringPiece& str) { |
| size_t pieces = 0; |
| for (const StringPiece& piece : Tokenize(str, '.')) { |
| pieces++; |
| if (piece.empty()) { |
| return false; |
| } |
| |
| // Can't have starting or trailing $ character. |
| if (piece.data()[0] == '$' || piece.data()[piece.size() - 1] == '$') { |
| return false; |
| } |
| |
| if (FindNonAlphaNumericAndNotInSet(piece, "$_") != piece.end()) { |
| return false; |
| } |
| } |
| return pieces >= 2; |
| } |
| |
| bool IsJavaPackageName(const StringPiece& str) { |
| if (str.empty()) { |
| return false; |
| } |
| |
| size_t pieces = 0; |
| for (const StringPiece& piece : Tokenize(str, '.')) { |
| pieces++; |
| if (piece.empty()) { |
| return false; |
| } |
| |
| if (piece.data()[0] == '_' || piece.data()[piece.size() - 1] == '_') { |
| return false; |
| } |
| |
| if (FindNonAlphaNumericAndNotInSet(piece, "_") != piece.end()) { |
| return false; |
| } |
| } |
| return pieces >= 1; |
| } |
| |
| Maybe<std::string> GetFullyQualifiedClassName(const StringPiece& package, |
| const StringPiece& classname) { |
| if (classname.empty()) { |
| return {}; |
| } |
| |
| if (util::IsJavaClassName(classname)) { |
| return classname.to_string(); |
| } |
| |
| if (package.empty()) { |
| return {}; |
| } |
| |
| std::string result(package.data(), package.size()); |
| if (classname.data()[0] != '.') { |
| result += '.'; |
| } |
| |
| result.append(classname.data(), classname.size()); |
| if (!IsJavaClassName(result)) { |
| return {}; |
| } |
| return result; |
| } |
| |
| static size_t ConsumeDigits(const char* start, const char* end) { |
| const char* c = start; |
| for (; c != end && *c >= '0' && *c <= '9'; c++) { |
| } |
| return static_cast<size_t>(c - start); |
| } |
| |
| bool VerifyJavaStringFormat(const StringPiece& str) { |
| const char* c = str.begin(); |
| const char* const end = str.end(); |
| |
| size_t arg_count = 0; |
| bool nonpositional = false; |
| while (c != end) { |
| if (*c == '%' && c + 1 < end) { |
| c++; |
| |
| if (*c == '%') { |
| c++; |
| continue; |
| } |
| |
| arg_count++; |
| |
| size_t num_digits = ConsumeDigits(c, end); |
| if (num_digits > 0) { |
| c += num_digits; |
| if (c != end && *c != '$') { |
| // The digits were a size, but not a positional argument. |
| nonpositional = true; |
| } |
| } else if (*c == '<') { |
| // Reusing last argument, bad idea since positions can be moved around |
| // during translation. |
| nonpositional = true; |
| |
| c++; |
| |
| // Optionally we can have a $ after |
| if (c != end && *c == '$') { |
| c++; |
| } |
| } else { |
| nonpositional = true; |
| } |
| |
| // Ignore size, width, flags, etc. |
| while (c != end && (*c == '-' || *c == '#' || *c == '+' || *c == ' ' || |
| *c == ',' || *c == '(' || (*c >= '0' && *c <= '9'))) { |
| c++; |
| } |
| |
| /* |
| * This is a shortcut to detect strings that are going to Time.format() |
| * instead of String.format() |
| * |
| * Comparison of String.format() and Time.format() args: |
| * |
| * String: ABC E GH ST X abcdefgh nost x |
| * Time: DEFGHKMS W Za d hkm s w yz |
| * |
| * Therefore we know it's definitely Time if we have: |
| * DFKMWZkmwyz |
| */ |
| if (c != end) { |
| switch (*c) { |
| case 'D': |
| case 'F': |
| case 'K': |
| case 'M': |
| case 'W': |
| case 'Z': |
| case 'k': |
| case 'm': |
| case 'w': |
| case 'y': |
| case 'z': |
| return true; |
| } |
| } |
| } |
| |
| if (c != end) { |
| c++; |
| } |
| } |
| |
| if (arg_count > 1 && nonpositional) { |
| // Multiple arguments were specified, but some or all were non positional. |
| // Translated |
| // strings may rearrange the order of the arguments, which will break the |
| // string. |
| return false; |
| } |
| return true; |
| } |
| |
| static Maybe<std::string> ParseUnicodeCodepoint(const char** start, |
| const char* end) { |
| char32_t code = 0; |
| for (size_t i = 0; i < 4 && *start != end; i++, (*start)++) { |
| char c = **start; |
| char32_t a; |
| if (c >= '0' && c <= '9') { |
| a = c - '0'; |
| } else if (c >= 'a' && c <= 'f') { |
| a = c - 'a' + 10; |
| } else if (c >= 'A' && c <= 'F') { |
| a = c - 'A' + 10; |
| } else { |
| return {}; |
| } |
| code = (code << 4) | a; |
| } |
| |
| ssize_t len = utf32_to_utf8_length(&code, 1); |
| if (len < 0) { |
| return {}; |
| } |
| |
| std::string result_utf8; |
| result_utf8.resize(len); |
| utf32_to_utf8(&code, 1, &*result_utf8.begin(), len + 1); |
| return result_utf8; |
| } |
| |
| StringBuilder& StringBuilder::Append(const StringPiece& str) { |
| if (!error_.empty()) { |
| return *this; |
| } |
| |
| // Where the new data will be appended to. |
| size_t new_data_index = str_.size(); |
| |
| const char* const end = str.end(); |
| const char* start = str.begin(); |
| const char* current = start; |
| while (current != end) { |
| if (last_char_was_escape_) { |
| switch (*current) { |
| case 't': |
| str_ += '\t'; |
| break; |
| case 'n': |
| str_ += '\n'; |
| break; |
| case '#': |
| str_ += '#'; |
| break; |
| case '@': |
| str_ += '@'; |
| break; |
| case '?': |
| str_ += '?'; |
| break; |
| case '"': |
| str_ += '"'; |
| break; |
| case '\'': |
| str_ += '\''; |
| break; |
| case '\\': |
| str_ += '\\'; |
| break; |
| case 'u': { |
| current++; |
| Maybe<std::string> c = ParseUnicodeCodepoint(¤t, end); |
| if (!c) { |
| error_ = "invalid unicode escape sequence"; |
| return *this; |
| } |
| str_ += c.value(); |
| current -= 1; |
| break; |
| } |
| |
| default: |
| // Ignore. |
| break; |
| } |
| last_char_was_escape_ = false; |
| start = current + 1; |
| } else if (*current == '"') { |
| if (!quote_ && trailing_space_) { |
| // We found an opening quote, and we have |
| // trailing space, so we should append that |
| // space now. |
| if (trailing_space_) { |
| // We had trailing whitespace, so |
| // replace with a single space. |
| if (!str_.empty()) { |
| str_ += ' '; |
| } |
| trailing_space_ = false; |
| } |
| } |
| quote_ = !quote_; |
| str_.append(start, current - start); |
| start = current + 1; |
| } else if (*current == '\'' && !quote_) { |
| // This should be escaped. |
| error_ = "unescaped apostrophe"; |
| return *this; |
| } else if (*current == '\\') { |
| // This is an escape sequence, convert to the real value. |
| if (!quote_ && trailing_space_) { |
| // We had trailing whitespace, so |
| // replace with a single space. |
| if (!str_.empty()) { |
| str_ += ' '; |
| } |
| trailing_space_ = false; |
| } |
| str_.append(start, current - start); |
| start = current + 1; |
| last_char_was_escape_ = true; |
| } else if (!quote_) { |
| // This is not quoted text, so look for whitespace. |
| if (isspace(*current)) { |
| // We found whitespace, see if we have seen some |
| // before. |
| if (!trailing_space_) { |
| // We didn't see a previous adjacent space, |
| // so mark that we did. |
| trailing_space_ = true; |
| str_.append(start, current - start); |
| } |
| |
| // Keep skipping whitespace. |
| start = current + 1; |
| } else if (trailing_space_) { |
| // We saw trailing space before, so replace all |
| // that trailing space with one space. |
| if (!str_.empty()) { |
| str_ += ' '; |
| } |
| trailing_space_ = false; |
| } |
| } |
| current++; |
| } |
| str_.append(start, end - start); |
| |
| // Accumulate the added string's UTF-16 length. |
| ssize_t len = utf8_to_utf16_length( |
| reinterpret_cast<const uint8_t*>(str_.data()) + new_data_index, |
| str_.size() - new_data_index); |
| if (len < 0) { |
| error_ = "invalid unicode code point"; |
| return *this; |
| } |
| utf16_len_ += len; |
| return *this; |
| } |
| |
| std::u16string Utf8ToUtf16(const StringPiece& utf8) { |
| ssize_t utf16_length = utf8_to_utf16_length( |
| reinterpret_cast<const uint8_t*>(utf8.data()), utf8.length()); |
| if (utf16_length <= 0) { |
| return {}; |
| } |
| |
| std::u16string utf16; |
| utf16.resize(utf16_length); |
| utf8_to_utf16(reinterpret_cast<const uint8_t*>(utf8.data()), utf8.length(), |
| &*utf16.begin(), utf16_length + 1); |
| return utf16; |
| } |
| |
| std::string Utf16ToUtf8(const StringPiece16& utf16) { |
| ssize_t utf8_length = utf16_to_utf8_length(utf16.data(), utf16.length()); |
| if (utf8_length <= 0) { |
| return {}; |
| } |
| |
| std::string utf8; |
| utf8.resize(utf8_length); |
| utf16_to_utf8(utf16.data(), utf16.length(), &*utf8.begin(), utf8_length + 1); |
| return utf8; |
| } |
| |
| bool WriteAll(std::ostream& out, const BigBuffer& buffer) { |
| for (const auto& b : buffer) { |
| if (!out.write(reinterpret_cast<const char*>(b.buffer.get()), b.size)) { |
| return false; |
| } |
| } |
| return true; |
| } |
| |
| std::unique_ptr<uint8_t[]> Copy(const BigBuffer& buffer) { |
| std::unique_ptr<uint8_t[]> data = |
| std::unique_ptr<uint8_t[]>(new uint8_t[buffer.size()]); |
| uint8_t* p = data.get(); |
| for (const auto& block : buffer) { |
| memcpy(p, block.buffer.get(), block.size); |
| p += block.size; |
| } |
| return data; |
| } |
| |
| typename Tokenizer::iterator& Tokenizer::iterator::operator++() { |
| const char* start = token_.end(); |
| const char* end = str_.end(); |
| if (start == end) { |
| end_ = true; |
| token_.assign(token_.end(), 0); |
| return *this; |
| } |
| |
| start += 1; |
| const char* current = start; |
| while (current != end) { |
| if (*current == separator_) { |
| token_.assign(start, current - start); |
| return *this; |
| } |
| ++current; |
| } |
| token_.assign(start, end - start); |
| return *this; |
| } |
| |
| bool Tokenizer::iterator::operator==(const iterator& rhs) const { |
| // We check equality here a bit differently. |
| // We need to know that the addresses are the same. |
| return token_.begin() == rhs.token_.begin() && |
| token_.end() == rhs.token_.end() && end_ == rhs.end_; |
| } |
| |
| bool Tokenizer::iterator::operator!=(const iterator& rhs) const { |
| return !(*this == rhs); |
| } |
| |
| Tokenizer::iterator::iterator(StringPiece s, char sep, StringPiece tok, |
| bool end) |
| : str_(s), separator_(sep), token_(tok), end_(end) {} |
| |
| Tokenizer::Tokenizer(StringPiece str, char sep) |
| : begin_(++iterator(str, sep, StringPiece(str.begin() - 1, 0), false)), |
| end_(str, sep, StringPiece(str.end(), 0), true) {} |
| |
| bool ExtractResFilePathParts(const StringPiece& path, StringPiece* out_prefix, |
| StringPiece* out_entry, StringPiece* out_suffix) { |
| const StringPiece res_prefix("res/"); |
| if (!StartsWith(path, res_prefix)) { |
| return false; |
| } |
| |
| StringPiece::const_iterator last_occurence = path.end(); |
| for (auto iter = path.begin() + res_prefix.size(); iter != path.end(); |
| ++iter) { |
| if (*iter == '/') { |
| last_occurence = iter; |
| } |
| } |
| |
| if (last_occurence == path.end()) { |
| return false; |
| } |
| |
| auto iter = std::find(last_occurence, path.end(), '.'); |
| *out_suffix = StringPiece(iter, path.end() - iter); |
| *out_entry = StringPiece(last_occurence + 1, iter - last_occurence - 1); |
| *out_prefix = StringPiece(path.begin(), last_occurence - path.begin() + 1); |
| return true; |
| } |
| |
| StringPiece16 GetString16(const android::ResStringPool& pool, size_t idx) { |
| size_t len; |
| const char16_t* str = pool.stringAt(idx, &len); |
| if (str != nullptr) { |
| return StringPiece16(str, len); |
| } |
| return StringPiece16(); |
| } |
| |
| std::string GetString(const android::ResStringPool& pool, size_t idx) { |
| size_t len; |
| const char* str = pool.string8At(idx, &len); |
| if (str != nullptr) { |
| return std::string(str, len); |
| } |
| return Utf16ToUtf8(GetString16(pool, idx)); |
| } |
| |
| } // namespace util |
| } // namespace aapt |