mirror of https://github.com/aria2/aria2
304 lines
8.5 KiB
C++
304 lines
8.5 KiB
C++
/* <!-- copyright */
|
|
/*
|
|
* aria2 - The high speed download utility
|
|
*
|
|
* Copyright (C) 2006 Tatsuhiro Tsujikawa
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*
|
|
* In addition, as a special exception, the copyright holders give
|
|
* permission to link the code of portions of this program with the
|
|
* OpenSSL library under certain conditions as described in each
|
|
* individual source file, and distribute linked combinations
|
|
* including the two.
|
|
* You must obey the GNU General Public License in all respects
|
|
* for all of the code used other than OpenSSL. If you modify
|
|
* file(s) with this exception, you may extend this exception to your
|
|
* version of the file(s), but you are not obligated to do so. If you
|
|
* do not wish to do so, delete this exception statement from your
|
|
* version. If you delete this exception statement from all source
|
|
* files in the program, then also delete it here.
|
|
*/
|
|
/* copyright --> */
|
|
#include "OptionParser.h"
|
|
|
|
#include <unistd.h>
|
|
#include <getopt.h>
|
|
|
|
#include <cstring>
|
|
#include <istream>
|
|
#include <utility>
|
|
|
|
#include "util.h"
|
|
#include "OptionHandlerImpl.h"
|
|
#include "Option.h"
|
|
#include "A2STR.h"
|
|
#include "a2functional.h"
|
|
#include "array_fun.h"
|
|
#include "OptionHandlerFactory.h"
|
|
#include "DlAbortEx.h"
|
|
#include "error_code.h"
|
|
#include "prefs.h"
|
|
|
|
namespace aria2 {
|
|
|
|
OptionParser::OptionParser()
|
|
: handlers_(option::countOption()),
|
|
shortOpts_(256)
|
|
{}
|
|
|
|
OptionParser::~OptionParser() {}
|
|
|
|
namespace {
|
|
template<typename InputIterator>
|
|
size_t countPublicOption(InputIterator first, InputIterator last)
|
|
{
|
|
size_t count = 0;
|
|
for(; first != last; ++first) {
|
|
if(*first && !(*first)->isHidden()) {
|
|
++count;
|
|
}
|
|
}
|
|
return count;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
template<typename InputIterator>
|
|
void putOptions(struct option* longOpts, int* plopt,
|
|
InputIterator first, InputIterator last)
|
|
{
|
|
for(; first != last; ++first) {
|
|
if(*first && !(*first)->isHidden()) {
|
|
#ifdef HAVE_OPTION_CONST_NAME
|
|
(*longOpts).name = (*first)->getName().c_str();
|
|
#else // !HAVE_OPTION_CONST_NAME
|
|
(*longOpts).name = strdup((*first)->getName().c_str());
|
|
#endif // !HAVE_OPTION_CONST_NAME
|
|
switch((*first)->getArgType()) {
|
|
case OptionHandler::REQ_ARG:
|
|
(*longOpts).has_arg = required_argument;
|
|
break;
|
|
case OptionHandler::OPT_ARG:
|
|
(*longOpts).has_arg = optional_argument;
|
|
break;
|
|
case OptionHandler::NO_ARG:
|
|
(*longOpts).has_arg = no_argument;
|
|
break;
|
|
default:
|
|
abort();
|
|
}
|
|
if((*first)->getShortName() == 0) {
|
|
(*longOpts).flag = plopt;
|
|
(*longOpts).val = (*first)->getPref()->i;
|
|
} else {
|
|
(*longOpts).flag = 0;
|
|
(*longOpts).val = (*first)->getShortName();
|
|
}
|
|
++longOpts;
|
|
}
|
|
}
|
|
(*longOpts).name = 0;
|
|
(*longOpts).has_arg = 0;
|
|
(*longOpts).flag = 0;
|
|
(*longOpts).val = 0;
|
|
}
|
|
} // namespace
|
|
|
|
namespace {
|
|
template<typename InputIterator>
|
|
std::string createOptstring(InputIterator first, InputIterator last)
|
|
{
|
|
std::string str = "";
|
|
for(; first != last; ++first) {
|
|
if(*first && !(*first)->isHidden()) {
|
|
if((*first)->getShortName() != 0) {
|
|
str += (*first)->getShortName();
|
|
if((*first)->getArgType() == OptionHandler::REQ_ARG) {
|
|
str += ":";
|
|
} else if((*first)->getArgType() == OptionHandler::OPT_ARG) {
|
|
str += "::";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return str;
|
|
}
|
|
} // namespace
|
|
|
|
void OptionParser::parseArg
|
|
(std::ostream& out, std::vector<std::string>& nonopts,
|
|
int argc, char* argv[]) const
|
|
{
|
|
size_t numPublicOption = countPublicOption(handlers_.begin(),
|
|
handlers_.end());
|
|
int lopt;
|
|
array_ptr<struct option> longOpts(new struct option[numPublicOption+1]);
|
|
putOptions(longOpts, &lopt,handlers_.begin(),handlers_.end());
|
|
std::string optstring = createOptstring(handlers_.begin(), handlers_.end());
|
|
while(1) {
|
|
int c = getopt_long(argc, argv, optstring.c_str(), longOpts, 0);
|
|
if(c == -1) {
|
|
break;
|
|
}
|
|
SharedHandle<OptionHandler> op;
|
|
if(c == 0) {
|
|
op = findById(lopt);
|
|
} else {
|
|
op = findByShortName(c);
|
|
}
|
|
if(!op) {
|
|
throw DL_ABORT_EX2("Failed to parse command-line options.",
|
|
error_code::OPTION_ERROR);
|
|
}
|
|
out << op->getName() << "=";
|
|
if(optarg) {
|
|
out << optarg;
|
|
if(op->getEraseAfterParse()) {
|
|
for(char* p = optarg; *p != '\0'; ++p) {
|
|
*p = '*';
|
|
}
|
|
}
|
|
}
|
|
out << "\n";
|
|
}
|
|
std::copy(argv+optind, argv+argc, std::back_inserter(nonopts));
|
|
}
|
|
|
|
void OptionParser::parse(Option& option, std::istream& is) const
|
|
{
|
|
std::string line;
|
|
while(getline(is, line)) {
|
|
if(util::startsWith(line, A2STR::SHARP_C)) {
|
|
continue;
|
|
}
|
|
std::pair<std::string, std::string> nv;
|
|
util::divide(nv, line, '=');
|
|
if(nv.first.empty()) {
|
|
continue;
|
|
}
|
|
const SharedHandle<OptionHandler>& handler = find(option::k2p(nv.first));
|
|
if(handler) {
|
|
handler->parse(option, nv.second);
|
|
}
|
|
}
|
|
}
|
|
|
|
void OptionParser::setOptionHandlers
|
|
(const std::vector<SharedHandle<OptionHandler> >& handlers)
|
|
{
|
|
for(std::vector<SharedHandle<OptionHandler> >::const_iterator i =
|
|
handlers.begin(), eoi = handlers.end(); i != eoi; ++i) {
|
|
addOptionHandler(*i);
|
|
}
|
|
}
|
|
|
|
void OptionParser::addOptionHandler(const SharedHandle<OptionHandler>& handler)
|
|
{
|
|
size_t optId = handler->getPref()->i;
|
|
assert(optId < handlers_.size());
|
|
handlers_[optId] = handler;
|
|
if(handler->getShortName()) {
|
|
shortOpts_[static_cast<unsigned char>(handler->getShortName())] = optId;
|
|
}
|
|
}
|
|
|
|
void OptionParser::parseDefaultValues(Option& option) const
|
|
{
|
|
for(std::vector<SharedHandle<OptionHandler> >::const_iterator i =
|
|
handlers_.begin(), eoi = handlers_.end(); i != eoi; ++i) {
|
|
if(*i && !(*i)->getDefaultValue().empty()) {
|
|
(*i)->parse(option, (*i)->getDefaultValue());
|
|
}
|
|
}
|
|
}
|
|
|
|
std::vector<SharedHandle<OptionHandler> >
|
|
OptionParser::findByTag(const std::string& tag) const
|
|
{
|
|
std::vector<SharedHandle<OptionHandler> > result;
|
|
for(std::vector<SharedHandle<OptionHandler> >::const_iterator i =
|
|
handlers_.begin(), eoi = handlers_.end(); i != eoi; ++i) {
|
|
if(*i && !(*i)->isHidden() && (*i)->hasTag(tag)) {
|
|
result.push_back(*i);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
std::vector<SharedHandle<OptionHandler> >
|
|
OptionParser::findByNameSubstring(const std::string& substring) const
|
|
{
|
|
std::vector<SharedHandle<OptionHandler> > result;
|
|
for(std::vector<SharedHandle<OptionHandler> >::const_iterator i =
|
|
handlers_.begin(), eoi = handlers_.end(); i != eoi; ++i) {
|
|
if(*i && !(*i)->isHidden() &&
|
|
(*i)->getName().find(substring) != std::string::npos) {
|
|
result.push_back(*i);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
std::vector<SharedHandle<OptionHandler> > OptionParser::findAll() const
|
|
{
|
|
std::vector<SharedHandle<OptionHandler> > result;
|
|
for(std::vector<SharedHandle<OptionHandler> >::const_iterator i =
|
|
handlers_.begin(), eoi = handlers_.end(); i != eoi; ++i) {
|
|
if(*i && !(*i)->isHidden()) {
|
|
result.push_back(*i);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
const SharedHandle<OptionHandler>& OptionParser::find(const Pref* pref) const
|
|
{
|
|
return findById(pref->i);
|
|
}
|
|
|
|
const SharedHandle<OptionHandler>& OptionParser::findById(size_t id) const
|
|
{
|
|
if(id >= handlers_.size()) {
|
|
return handlers_[0];
|
|
}
|
|
const SharedHandle<OptionHandler>& h = handlers_[id];
|
|
if(!h || h->isHidden()) {
|
|
return handlers_[0];
|
|
} else {
|
|
return h;
|
|
}
|
|
}
|
|
|
|
const SharedHandle<OptionHandler>& OptionParser::findByShortName
|
|
(char shortName) const
|
|
{
|
|
size_t idx = static_cast<unsigned char>(shortName);
|
|
return findById(shortOpts_[idx]);
|
|
}
|
|
|
|
SharedHandle<OptionParser> OptionParser::optionParser_;
|
|
|
|
const SharedHandle<OptionParser>& OptionParser::getInstance()
|
|
{
|
|
if(!optionParser_) {
|
|
optionParser_.reset(new OptionParser());
|
|
optionParser_->setOptionHandlers(OptionHandlerFactory::createOptionHandlers());
|
|
}
|
|
return optionParser_;
|
|
}
|
|
|
|
} // namespace aria2
|