Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2016 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #include "proto/ProtoHelpers.h" |
| 18 | |
| 19 | namespace aapt { |
| 20 | |
Adam Lesinski | 5b6ee11 | 2017-07-28 17:10:35 -0700 | [diff] [blame] | 21 | void SerializeStringPoolToPb(const StringPool& pool, pb::StringPool* out_pb_pool) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 22 | BigBuffer buffer(1024); |
| 23 | StringPool::FlattenUtf8(&buffer, pool); |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 24 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 25 | std::string* data = out_pb_pool->mutable_data(); |
| 26 | data->reserve(buffer.size()); |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 27 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 28 | size_t offset = 0; |
| 29 | for (const BigBuffer::Block& block : buffer) { |
Adam Lesinski | 5b6ee11 | 2017-07-28 17:10:35 -0700 | [diff] [blame] | 30 | data->insert(data->begin() + offset, block.buffer.get(), block.buffer.get() + block.size); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 31 | offset += block.size; |
| 32 | } |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 33 | } |
| 34 | |
Adam Lesinski | 5b6ee11 | 2017-07-28 17:10:35 -0700 | [diff] [blame] | 35 | void SerializeSourceToPb(const Source& source, StringPool* src_pool, pb::Source* out_pb_source) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 36 | StringPool::Ref ref = src_pool->MakeRef(source.path); |
| 37 | out_pb_source->set_path_idx(static_cast<uint32_t>(ref.index())); |
| 38 | if (source.line) { |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 39 | out_pb_source->mutable_position()->set_line_number(static_cast<uint32_t>(source.line.value())); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 40 | } |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 41 | } |
| 42 | |
Adam Lesinski | 5b6ee11 | 2017-07-28 17:10:35 -0700 | [diff] [blame] | 43 | void DeserializeSourceFromPb(const pb::Source& pb_source, const android::ResStringPool& src_pool, |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 44 | Source* out_source) { |
| 45 | if (pb_source.has_path_idx()) { |
| 46 | out_source->path = util::GetString(src_pool, pb_source.path_idx()); |
| 47 | } |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 48 | |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 49 | if (pb_source.has_position()) { |
| 50 | out_source->line = static_cast<size_t>(pb_source.position().line_number()); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 51 | } |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 52 | } |
| 53 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 54 | pb::SymbolStatus_Visibility SerializeVisibilityToPb(SymbolState state) { |
| 55 | switch (state) { |
| 56 | case SymbolState::kPrivate: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 57 | return pb::SymbolStatus_Visibility_PRIVATE; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 58 | case SymbolState::kPublic: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 59 | return pb::SymbolStatus_Visibility_PUBLIC; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 60 | default: |
| 61 | break; |
| 62 | } |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 63 | return pb::SymbolStatus_Visibility_UNKNOWN; |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 64 | } |
| 65 | |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 66 | SymbolState DeserializeVisibilityFromPb(pb::SymbolStatus_Visibility pb_visibility) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 67 | switch (pb_visibility) { |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 68 | case pb::SymbolStatus_Visibility_PRIVATE: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 69 | return SymbolState::kPrivate; |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 70 | case pb::SymbolStatus_Visibility_PUBLIC: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 71 | return SymbolState::kPublic; |
| 72 | default: |
| 73 | break; |
| 74 | } |
| 75 | return SymbolState::kUndefined; |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 76 | } |
| 77 | |
Adam Lesinski | 5b6ee11 | 2017-07-28 17:10:35 -0700 | [diff] [blame] | 78 | void SerializeConfig(const ConfigDescription& config, pb::ConfigDescription* out_pb_config) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 79 | android::ResTable_config flat_config = config; |
| 80 | flat_config.size = sizeof(flat_config); |
| 81 | flat_config.swapHtoD(); |
| 82 | out_pb_config->set_data(&flat_config, sizeof(flat_config)); |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 83 | } |
| 84 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 85 | bool DeserializeConfigDescriptionFromPb(const pb::ConfigDescription& pb_config, |
| 86 | ConfigDescription* out_config) { |
| 87 | if (!pb_config.has_data()) { |
| 88 | return false; |
| 89 | } |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 90 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 91 | const android::ResTable_config* config; |
| 92 | if (pb_config.data().size() > sizeof(*config)) { |
| 93 | return false; |
| 94 | } |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 95 | |
Adam Lesinski | 5b6ee11 | 2017-07-28 17:10:35 -0700 | [diff] [blame] | 96 | config = reinterpret_cast<const android::ResTable_config*>(pb_config.data().data()); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 97 | out_config->copyFromDtoH(*config); |
| 98 | return true; |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 99 | } |
| 100 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 101 | pb::Reference_Type SerializeReferenceTypeToPb(Reference::Type type) { |
| 102 | switch (type) { |
| 103 | case Reference::Type::kResource: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 104 | return pb::Reference_Type_REFERENCE; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 105 | case Reference::Type::kAttribute: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 106 | return pb::Reference_Type_ATTRIBUTE; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 107 | default: |
| 108 | break; |
| 109 | } |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 110 | return pb::Reference_Type_REFERENCE; |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 111 | } |
| 112 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 113 | Reference::Type DeserializeReferenceTypeFromPb(pb::Reference_Type pb_type) { |
| 114 | switch (pb_type) { |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 115 | case pb::Reference_Type_REFERENCE: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 116 | return Reference::Type::kResource; |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 117 | case pb::Reference_Type_ATTRIBUTE: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 118 | return Reference::Type::kAttribute; |
| 119 | default: |
| 120 | break; |
| 121 | } |
| 122 | return Reference::Type::kResource; |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 123 | } |
| 124 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 125 | pb::Plural_Arity SerializePluralEnumToPb(size_t plural_idx) { |
| 126 | switch (plural_idx) { |
| 127 | case Plural::Zero: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 128 | return pb::Plural_Arity_ZERO; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 129 | case Plural::One: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 130 | return pb::Plural_Arity_ONE; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 131 | case Plural::Two: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 132 | return pb::Plural_Arity_TWO; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 133 | case Plural::Few: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 134 | return pb::Plural_Arity_FEW; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 135 | case Plural::Many: |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 136 | return pb::Plural_Arity_MANY; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 137 | default: |
| 138 | break; |
| 139 | } |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 140 | return pb::Plural_Arity_OTHER; |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 141 | } |
| 142 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 143 | size_t DeserializePluralEnumFromPb(pb::Plural_Arity arity) { |
| 144 | switch (arity) { |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 145 | case pb::Plural_Arity_ZERO: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 146 | return Plural::Zero; |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 147 | case pb::Plural_Arity_ONE: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 148 | return Plural::One; |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 149 | case pb::Plural_Arity_TWO: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 150 | return Plural::Two; |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 151 | case pb::Plural_Arity_FEW: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 152 | return Plural::Few; |
Adam Lesinski | b791721 | 2017-08-10 15:37:28 -0700 | [diff] [blame] | 153 | case pb::Plural_Arity_MANY: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 154 | return Plural::Many; |
| 155 | default: |
| 156 | break; |
| 157 | } |
| 158 | return Plural::Other; |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 159 | } |
| 160 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 161 | } // namespace aapt |