contrib/llvm/tools/lldb/source/Host/common/OptionParser.cpp

   1 //===-- source/Host/common/OptionParser.cpp ---------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "lldb/Host/OptionParser.h"
  11 #include "lldb/Host/HostGetOpt.h"
  12 #include "lldb/lldb-private-types.h"
  13
  14 #include <vector>
  15
  16 using namespace lldb_private;
  17
  18 void OptionParser::Prepare(std::unique_lock<std::mutex> &lock) {
  19   static std::mutex g_mutex;
  20   lock = std::unique_lock<std::mutex>(g_mutex);
  21 #ifdef __GLIBC__
  22   optind = 0;
  23 #else
  24   optreset = 1;
  25   optind = 1;
  26 #endif
  27 }
  28
  29 void OptionParser::EnableError(bool error) { opterr = error ? 1 : 0; }
  30
  31 int OptionParser::Parse(int argc, char *const argv[], llvm::StringRef optstring,
  32                         const Option *longopts, int *longindex) {
  33   std::vector<option> opts;
  34   while (longopts->definition != nullptr) {
  35     option opt;
  36     opt.flag = longopts->flag;
  37     opt.val = longopts->val;
  38     opt.name = longopts->definition->long_option;
  39     opt.has_arg = longopts->definition->option_has_arg;
  40     opts.push_back(opt);
  41     ++longopts;
  42   }
  43   opts.push_back(option());
  44   std::string opt_cstr = optstring;
  45   return getopt_long_only(argc, argv, opt_cstr.c_str(), &opts[0], longindex);
  46 }
  47
  48 char *OptionParser::GetOptionArgument() { return optarg; }
  49
  50 int OptionParser::GetOptionIndex() { return optind; }
  51
  52 int OptionParser::GetOptionErrorCause() { return optopt; }
  53
  54 std::string OptionParser::GetShortOptionString(struct option *long_options) {
  55   std::string s;
  56   int i = 0;
  57   bool done = false;
  58   while (!done) {
  59     if (long_options[i].name == 0 && long_options[i].has_arg == 0 &&
  60         long_options[i].flag == 0 && long_options[i].val == 0) {
  61       done = true;
  62     } else {
  63       if (long_options[i].flag == NULL && isalpha(long_options[i].val)) {
  64         s.append(1, (char)long_options[i].val);
  65         switch (long_options[i].has_arg) {
  66         default:
  67         case no_argument:
  68           break;
  69
  70         case optional_argument:
  71           s.append(2, ':');
  72           break;
  73         case required_argument:
  74           s.append(1, ':');
  75           break;
  76         }
  77       }
  78       ++i;
  79     }
  80   }
  81   return s;
  82 }