1*7eba2f3bSAndroid Build Coastguard Worker /*
2*7eba2f3bSAndroid Build Coastguard Worker * Copyright 2017 The Android Open Source Project
3*7eba2f3bSAndroid Build Coastguard Worker *
4*7eba2f3bSAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*7eba2f3bSAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*7eba2f3bSAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*7eba2f3bSAndroid Build Coastguard Worker *
8*7eba2f3bSAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*7eba2f3bSAndroid Build Coastguard Worker *
10*7eba2f3bSAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*7eba2f3bSAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*7eba2f3bSAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*7eba2f3bSAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*7eba2f3bSAndroid Build Coastguard Worker * limitations under the License.
15*7eba2f3bSAndroid Build Coastguard Worker */
16*7eba2f3bSAndroid Build Coastguard Worker #include "config.h"
17*7eba2f3bSAndroid Build Coastguard Worker
18*7eba2f3bSAndroid Build Coastguard Worker #include <android-base/file.h>
19*7eba2f3bSAndroid Build Coastguard Worker #include <android-base/logging.h>
20*7eba2f3bSAndroid Build Coastguard Worker #include <android-base/parseint.h>
21*7eba2f3bSAndroid Build Coastguard Worker #include <android-base/strings.h>
22*7eba2f3bSAndroid Build Coastguard Worker
23*7eba2f3bSAndroid Build Coastguard Worker using namespace ::std;
24*7eba2f3bSAndroid Build Coastguard Worker using namespace ::android::base;
25*7eba2f3bSAndroid Build Coastguard Worker
26*7eba2f3bSAndroid Build Coastguard Worker namespace {
27*7eba2f3bSAndroid Build Coastguard Worker
parseBytesString(std::string in,std::vector<uint8_t> & out)28*7eba2f3bSAndroid Build Coastguard Worker bool parseBytesString(std::string in, std::vector<uint8_t>& out) {
29*7eba2f3bSAndroid Build Coastguard Worker vector<string> values = Split(in, ":");
30*7eba2f3bSAndroid Build Coastguard Worker if (values.size() == 0) return false;
31*7eba2f3bSAndroid Build Coastguard Worker for (const string& value : values) {
32*7eba2f3bSAndroid Build Coastguard Worker if (value.length() != 2) return false;
33*7eba2f3bSAndroid Build Coastguard Worker uint8_t tmp = 0;
34*7eba2f3bSAndroid Build Coastguard Worker string hexified = "0x";
35*7eba2f3bSAndroid Build Coastguard Worker hexified.append(value);
36*7eba2f3bSAndroid Build Coastguard Worker if (!ParseUint(hexified.c_str(), &tmp)) return false;
37*7eba2f3bSAndroid Build Coastguard Worker out.push_back(tmp);
38*7eba2f3bSAndroid Build Coastguard Worker }
39*7eba2f3bSAndroid Build Coastguard Worker return true;
40*7eba2f3bSAndroid Build Coastguard Worker }
41*7eba2f3bSAndroid Build Coastguard Worker
42*7eba2f3bSAndroid Build Coastguard Worker } // namespace
43*7eba2f3bSAndroid Build Coastguard Worker
ConfigValue()44*7eba2f3bSAndroid Build Coastguard Worker ConfigValue::ConfigValue() {
45*7eba2f3bSAndroid Build Coastguard Worker type_ = UNSIGNED;
46*7eba2f3bSAndroid Build Coastguard Worker value_unsigned_ = 0;
47*7eba2f3bSAndroid Build Coastguard Worker }
48*7eba2f3bSAndroid Build Coastguard Worker
ConfigValue(std::string value)49*7eba2f3bSAndroid Build Coastguard Worker ConfigValue::ConfigValue(std::string value) {
50*7eba2f3bSAndroid Build Coastguard Worker // Don't allow empty strings
51*7eba2f3bSAndroid Build Coastguard Worker CHECK(!(value.empty()));
52*7eba2f3bSAndroid Build Coastguard Worker type_ = STRING;
53*7eba2f3bSAndroid Build Coastguard Worker value_string_ = value;
54*7eba2f3bSAndroid Build Coastguard Worker value_unsigned_ = 0;
55*7eba2f3bSAndroid Build Coastguard Worker }
56*7eba2f3bSAndroid Build Coastguard Worker
ConfigValue(unsigned value)57*7eba2f3bSAndroid Build Coastguard Worker ConfigValue::ConfigValue(unsigned value) {
58*7eba2f3bSAndroid Build Coastguard Worker type_ = UNSIGNED;
59*7eba2f3bSAndroid Build Coastguard Worker value_unsigned_ = value;
60*7eba2f3bSAndroid Build Coastguard Worker }
61*7eba2f3bSAndroid Build Coastguard Worker
ConfigValue(std::vector<int8_t> value)62*7eba2f3bSAndroid Build Coastguard Worker ConfigValue::ConfigValue(std::vector<int8_t> value) {
63*7eba2f3bSAndroid Build Coastguard Worker CHECK(!(value.empty()));
64*7eba2f3bSAndroid Build Coastguard Worker type_ = BYTES;
65*7eba2f3bSAndroid Build Coastguard Worker value_bytes_ = std::vector<uint8_t>(value.begin(), value.end());
66*7eba2f3bSAndroid Build Coastguard Worker value_unsigned_ = 0;
67*7eba2f3bSAndroid Build Coastguard Worker }
68*7eba2f3bSAndroid Build Coastguard Worker
ConfigValue(std::vector<uint8_t> value)69*7eba2f3bSAndroid Build Coastguard Worker ConfigValue::ConfigValue(std::vector<uint8_t> value) {
70*7eba2f3bSAndroid Build Coastguard Worker CHECK(!(value.empty()));
71*7eba2f3bSAndroid Build Coastguard Worker type_ = BYTES;
72*7eba2f3bSAndroid Build Coastguard Worker value_bytes_ = value;
73*7eba2f3bSAndroid Build Coastguard Worker value_unsigned_ = 0;
74*7eba2f3bSAndroid Build Coastguard Worker }
75*7eba2f3bSAndroid Build Coastguard Worker
getType() const76*7eba2f3bSAndroid Build Coastguard Worker ConfigValue::Type ConfigValue::getType() const { return type_; }
77*7eba2f3bSAndroid Build Coastguard Worker
getString() const78*7eba2f3bSAndroid Build Coastguard Worker std::string ConfigValue::getString() const {
79*7eba2f3bSAndroid Build Coastguard Worker CHECK(type_ == STRING);
80*7eba2f3bSAndroid Build Coastguard Worker return value_string_;
81*7eba2f3bSAndroid Build Coastguard Worker };
82*7eba2f3bSAndroid Build Coastguard Worker
getUnsigned() const83*7eba2f3bSAndroid Build Coastguard Worker unsigned ConfigValue::getUnsigned() const {
84*7eba2f3bSAndroid Build Coastguard Worker CHECK(type_ == UNSIGNED);
85*7eba2f3bSAndroid Build Coastguard Worker return value_unsigned_;
86*7eba2f3bSAndroid Build Coastguard Worker };
87*7eba2f3bSAndroid Build Coastguard Worker
getBytes() const88*7eba2f3bSAndroid Build Coastguard Worker std::vector<uint8_t> ConfigValue::getBytes() const {
89*7eba2f3bSAndroid Build Coastguard Worker CHECK(type_ == BYTES);
90*7eba2f3bSAndroid Build Coastguard Worker return value_bytes_;
91*7eba2f3bSAndroid Build Coastguard Worker };
92*7eba2f3bSAndroid Build Coastguard Worker
parseFromString(std::string in)93*7eba2f3bSAndroid Build Coastguard Worker bool ConfigValue::parseFromString(std::string in) {
94*7eba2f3bSAndroid Build Coastguard Worker if (in.length() > 1 && in[0] == '"' && in[in.length() - 1] == '"') {
95*7eba2f3bSAndroid Build Coastguard Worker CHECK(in.length() > 2); // Don't allow empty strings
96*7eba2f3bSAndroid Build Coastguard Worker type_ = STRING;
97*7eba2f3bSAndroid Build Coastguard Worker value_string_ = in.substr(1, in.length() - 2);
98*7eba2f3bSAndroid Build Coastguard Worker return true;
99*7eba2f3bSAndroid Build Coastguard Worker }
100*7eba2f3bSAndroid Build Coastguard Worker
101*7eba2f3bSAndroid Build Coastguard Worker if (in.length() > 1 && in[0] == '{' && in[in.length() - 1] == '}') {
102*7eba2f3bSAndroid Build Coastguard Worker CHECK(in.length() >= 4); // Needs at least one byte
103*7eba2f3bSAndroid Build Coastguard Worker type_ = BYTES;
104*7eba2f3bSAndroid Build Coastguard Worker return parseBytesString(in.substr(1, in.length() - 2), value_bytes_);
105*7eba2f3bSAndroid Build Coastguard Worker }
106*7eba2f3bSAndroid Build Coastguard Worker
107*7eba2f3bSAndroid Build Coastguard Worker unsigned tmp = 0;
108*7eba2f3bSAndroid Build Coastguard Worker if (ParseUint(in.c_str(), &tmp)) {
109*7eba2f3bSAndroid Build Coastguard Worker type_ = UNSIGNED;
110*7eba2f3bSAndroid Build Coastguard Worker value_unsigned_ = tmp;
111*7eba2f3bSAndroid Build Coastguard Worker return true;
112*7eba2f3bSAndroid Build Coastguard Worker }
113*7eba2f3bSAndroid Build Coastguard Worker
114*7eba2f3bSAndroid Build Coastguard Worker return false;
115*7eba2f3bSAndroid Build Coastguard Worker }
116*7eba2f3bSAndroid Build Coastguard Worker
addConfig(const std::string & key,ConfigValue & value)117*7eba2f3bSAndroid Build Coastguard Worker void ConfigFile::addConfig(const std::string& key, ConfigValue& value) {
118*7eba2f3bSAndroid Build Coastguard Worker CHECK(!hasKey(key));
119*7eba2f3bSAndroid Build Coastguard Worker values_.emplace(key, value);
120*7eba2f3bSAndroid Build Coastguard Worker }
121*7eba2f3bSAndroid Build Coastguard Worker
parseFromFile(const std::string & file_name)122*7eba2f3bSAndroid Build Coastguard Worker void ConfigFile::parseFromFile(const std::string& file_name) {
123*7eba2f3bSAndroid Build Coastguard Worker string config;
124*7eba2f3bSAndroid Build Coastguard Worker bool config_read = ReadFileToString(file_name, &config);
125*7eba2f3bSAndroid Build Coastguard Worker CHECK(config_read);
126*7eba2f3bSAndroid Build Coastguard Worker LOG(INFO) << "ConfigFile - Parsing file '" << file_name << "'";
127*7eba2f3bSAndroid Build Coastguard Worker parseFromString(config);
128*7eba2f3bSAndroid Build Coastguard Worker }
129*7eba2f3bSAndroid Build Coastguard Worker
parseFromString(const std::string & config)130*7eba2f3bSAndroid Build Coastguard Worker void ConfigFile::parseFromString(const std::string& config) {
131*7eba2f3bSAndroid Build Coastguard Worker stringstream ss(config);
132*7eba2f3bSAndroid Build Coastguard Worker string line;
133*7eba2f3bSAndroid Build Coastguard Worker while (getline(ss, line)) {
134*7eba2f3bSAndroid Build Coastguard Worker line = Trim(line);
135*7eba2f3bSAndroid Build Coastguard Worker if (line.empty()) continue;
136*7eba2f3bSAndroid Build Coastguard Worker if (line.at(0) == '#') continue;
137*7eba2f3bSAndroid Build Coastguard Worker if (line.at(0) == 0) continue;
138*7eba2f3bSAndroid Build Coastguard Worker
139*7eba2f3bSAndroid Build Coastguard Worker auto search = line.find('=');
140*7eba2f3bSAndroid Build Coastguard Worker CHECK(search != string::npos);
141*7eba2f3bSAndroid Build Coastguard Worker
142*7eba2f3bSAndroid Build Coastguard Worker string key(Trim(line.substr(0, search)));
143*7eba2f3bSAndroid Build Coastguard Worker string value_string(Trim(line.substr(search + 1, string::npos)));
144*7eba2f3bSAndroid Build Coastguard Worker
145*7eba2f3bSAndroid Build Coastguard Worker ConfigValue value;
146*7eba2f3bSAndroid Build Coastguard Worker bool value_parsed = value.parseFromString(value_string);
147*7eba2f3bSAndroid Build Coastguard Worker CHECK(value_parsed);
148*7eba2f3bSAndroid Build Coastguard Worker addConfig(key, value);
149*7eba2f3bSAndroid Build Coastguard Worker
150*7eba2f3bSAndroid Build Coastguard Worker LOG(INFO) << "ConfigFile - [" << key << "] = " << value_string;
151*7eba2f3bSAndroid Build Coastguard Worker }
152*7eba2f3bSAndroid Build Coastguard Worker }
153*7eba2f3bSAndroid Build Coastguard Worker
hasKey(const std::string & key)154*7eba2f3bSAndroid Build Coastguard Worker bool ConfigFile::hasKey(const std::string& key) {
155*7eba2f3bSAndroid Build Coastguard Worker return values_.count(key) != 0;
156*7eba2f3bSAndroid Build Coastguard Worker }
157*7eba2f3bSAndroid Build Coastguard Worker
getValue(const std::string & key)158*7eba2f3bSAndroid Build Coastguard Worker ConfigValue& ConfigFile::getValue(const std::string& key) {
159*7eba2f3bSAndroid Build Coastguard Worker auto search = values_.find(key);
160*7eba2f3bSAndroid Build Coastguard Worker CHECK(search != values_.end());
161*7eba2f3bSAndroid Build Coastguard Worker return search->second;
162*7eba2f3bSAndroid Build Coastguard Worker }
163*7eba2f3bSAndroid Build Coastguard Worker
getString(const std::string & key)164*7eba2f3bSAndroid Build Coastguard Worker std::string ConfigFile::getString(const std::string& key) {
165*7eba2f3bSAndroid Build Coastguard Worker return getValue(key).getString();
166*7eba2f3bSAndroid Build Coastguard Worker }
167*7eba2f3bSAndroid Build Coastguard Worker
getUnsigned(const std::string & key)168*7eba2f3bSAndroid Build Coastguard Worker unsigned ConfigFile::getUnsigned(const std::string& key) {
169*7eba2f3bSAndroid Build Coastguard Worker return getValue(key).getUnsigned();
170*7eba2f3bSAndroid Build Coastguard Worker }
171*7eba2f3bSAndroid Build Coastguard Worker
getBytes(const std::string & key)172*7eba2f3bSAndroid Build Coastguard Worker std::vector<uint8_t> ConfigFile::getBytes(const std::string& key) {
173*7eba2f3bSAndroid Build Coastguard Worker return getValue(key).getBytes();
174*7eba2f3bSAndroid Build Coastguard Worker }
175*7eba2f3bSAndroid Build Coastguard Worker
isEmpty()176*7eba2f3bSAndroid Build Coastguard Worker bool ConfigFile::isEmpty() { return values_.empty(); }
clear()177*7eba2f3bSAndroid Build Coastguard Worker void ConfigFile::clear() { values_.clear(); }
178