160 lines
5.2 KiB
C++
160 lines
5.2 KiB
C++
/*
|
|
* Copyright (C) 2010 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 "init_parser.h"
|
|
|
|
#include <dirent.h>
|
|
|
|
#include <android-base/logging.h>
|
|
#include <android-base/stringprintf.h>
|
|
#include <android-base/strings.h>
|
|
|
|
#include "parser.h"
|
|
#include "util.h"
|
|
|
|
Parser::Parser() {
|
|
}
|
|
|
|
Parser& Parser::GetInstance() {
|
|
static Parser instance;
|
|
return instance;
|
|
}
|
|
|
|
void Parser::AddSectionParser(const std::string& name,
|
|
std::unique_ptr<SectionParser> parser) {
|
|
section_parsers_[name] = std::move(parser);
|
|
}
|
|
|
|
void Parser::AddSingleLineParser(const std::string& prefix, LineCallback callback) {
|
|
line_callbacks_.emplace_back(prefix, callback);
|
|
}
|
|
|
|
void Parser::ParseData(const std::string& filename, const std::string& data) {
|
|
//TODO: Use a parser with const input and remove this copy
|
|
std::vector<char> data_copy(data.begin(), data.end());
|
|
data_copy.push_back('\0');
|
|
|
|
parse_state state;
|
|
state.line = 0;
|
|
state.ptr = &data_copy[0];
|
|
state.nexttoken = 0;
|
|
|
|
SectionParser* section_parser = nullptr;
|
|
std::vector<std::string> args;
|
|
|
|
for (;;) {
|
|
switch (next_token(&state)) {
|
|
case T_EOF:
|
|
if (section_parser) {
|
|
section_parser->EndSection();
|
|
}
|
|
return;
|
|
case T_NEWLINE:
|
|
state.line++;
|
|
if (args.empty()) {
|
|
break;
|
|
}
|
|
// If we have a line matching a prefix we recognize, call its callback and unset any
|
|
// current section parsers. This is meant for /sys/ and /dev/ line entries for uevent.
|
|
for (const auto& [prefix, callback] : line_callbacks_) {
|
|
if (android::base::StartsWith(args[0], prefix.c_str())) {
|
|
if (section_parser) section_parser->EndSection();
|
|
|
|
std::string ret_err;
|
|
if (!callback(std::move(args), &ret_err)) {
|
|
LOG(ERROR) << filename << ": " << state.line << ": " << ret_err;
|
|
}
|
|
section_parser = nullptr;
|
|
break;
|
|
}
|
|
}
|
|
if (section_parsers_.count(args[0])) {
|
|
if (section_parser) {
|
|
section_parser->EndSection();
|
|
}
|
|
section_parser = section_parsers_[args[0]].get();
|
|
std::string ret_err;
|
|
if (!section_parser->ParseSection(std::move(args), filename, state.line, &ret_err)) {
|
|
LOG(ERROR) << filename << ": " << state.line << ": " << ret_err;
|
|
section_parser = nullptr;
|
|
}
|
|
} else if (section_parser) {
|
|
std::string ret_err;
|
|
if (!section_parser->ParseLineSection(std::move(args), state.line, &ret_err)) {
|
|
LOG(ERROR) << filename << ": " << state.line << ": " << ret_err;
|
|
}
|
|
}
|
|
args.clear();
|
|
break;
|
|
case T_TEXT:
|
|
args.emplace_back(state.text);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool Parser::ParseConfigFile(const std::string& path) {
|
|
LOG(INFO) << "Parsing file " << path << "...";
|
|
Timer t;
|
|
std::string data;
|
|
if (!read_file(path, &data)) {
|
|
return false;
|
|
}
|
|
|
|
data.push_back('\n'); // TODO: fix parse_config.
|
|
ParseData(path, data);
|
|
for (const auto& [section_name, section_parser] : section_parsers_) {
|
|
section_parser->EndFile();
|
|
}
|
|
|
|
LOG(VERBOSE) << "(Parsing " << path << " took " << t << ".)";
|
|
return true;
|
|
}
|
|
|
|
bool Parser::ParseConfigDir(const std::string& path) {
|
|
LOG(INFO) << "Parsing directory " << path << "...";
|
|
std::unique_ptr<DIR, int(*)(DIR*)> config_dir(opendir(path.c_str()), closedir);
|
|
if (!config_dir) {
|
|
PLOG(ERROR) << "Could not import directory '" << path << "'";
|
|
return false;
|
|
}
|
|
dirent* current_file;
|
|
std::vector<std::string> files;
|
|
while ((current_file = readdir(config_dir.get()))) {
|
|
// Ignore directories and only process regular files.
|
|
if (current_file->d_type == DT_REG) {
|
|
std::string current_path =
|
|
android::base::StringPrintf("%s/%s", path.c_str(), current_file->d_name);
|
|
files.emplace_back(current_path);
|
|
}
|
|
}
|
|
// Sort first so we load files in a consistent order (bug 31996208)
|
|
std::sort(files.begin(), files.end());
|
|
for (const auto& file : files) {
|
|
if (!ParseConfigFile(file)) {
|
|
LOG(ERROR) << "could not import file '" << file << "'";
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool Parser::ParseConfig(const std::string& path) {
|
|
if (is_dir(path.c_str())) {
|
|
return ParseConfigDir(path);
|
|
}
|
|
return ParseConfigFile(path);
|
|
}
|