1 //===-- Options.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/Interpreter/Options.h"
11 
12 // C Includes
13 // C++ Includes
14 #include <algorithm>
15 #include <bitset>
16 #include <set>
17 
18 // Other libraries and framework includes
19 // Project includes
20 #include "lldb/Interpreter/CommandObject.h"
21 #include "lldb/Interpreter/CommandReturnObject.h"
22 #include "lldb/Interpreter/CommandCompletions.h"
23 #include "lldb/Interpreter/CommandInterpreter.h"
24 #include "lldb/Core/StreamString.h"
25 #include "lldb/Target/Target.h"
26 
27 using namespace lldb;
28 using namespace lldb_private;
29 
30 //-------------------------------------------------------------------------
31 // Options
32 //-------------------------------------------------------------------------
33 Options::Options (CommandInterpreter &interpreter) :
34     m_interpreter (interpreter),
35     m_getopt_table ()
36 {
37     BuildValidOptionSets();
38 }
39 
40 Options::~Options ()
41 {
42 }
43 
44 void
45 Options::NotifyOptionParsingStarting ()
46 {
47     m_seen_options.clear();
48     // Let the subclass reset its option values
49     OptionParsingStarting ();
50 }
51 
52 Error
53 Options::NotifyOptionParsingFinished ()
54 {
55     return OptionParsingFinished ();
56 }
57 
58 void
59 Options::OptionSeen (int option_idx)
60 {
61     m_seen_options.insert ((char) option_idx);
62 }
63 
64 // Returns true is set_a is a subset of set_b;  Otherwise returns false.
65 
66 bool
67 Options::IsASubset (const OptionSet& set_a, const OptionSet& set_b)
68 {
69     bool is_a_subset = true;
70     OptionSet::const_iterator pos_a;
71     OptionSet::const_iterator pos_b;
72 
73     // set_a is a subset of set_b if every member of set_a is also a member of set_b
74 
75     for (pos_a = set_a.begin(); pos_a != set_a.end() && is_a_subset; ++pos_a)
76     {
77         pos_b = set_b.find(*pos_a);
78         if (pos_b == set_b.end())
79             is_a_subset = false;
80     }
81 
82     return is_a_subset;
83 }
84 
85 // Returns the set difference set_a - set_b, i.e. { x | ElementOf (x, set_a) && !ElementOf (x, set_b) }
86 
87 size_t
88 Options::OptionsSetDiff (const OptionSet& set_a, const OptionSet& set_b, OptionSet& diffs)
89 {
90     size_t num_diffs = 0;
91     OptionSet::const_iterator pos_a;
92     OptionSet::const_iterator pos_b;
93 
94     for (pos_a = set_a.begin(); pos_a != set_a.end(); ++pos_a)
95     {
96         pos_b = set_b.find(*pos_a);
97         if (pos_b == set_b.end())
98         {
99             ++num_diffs;
100             diffs.insert(*pos_a);
101         }
102     }
103 
104     return num_diffs;
105 }
106 
107 // Returns the union of set_a and set_b.  Does not put duplicate members into the union.
108 
109 void
110 Options::OptionsSetUnion (const OptionSet &set_a, const OptionSet &set_b, OptionSet &union_set)
111 {
112     OptionSet::const_iterator pos;
113     OptionSet::iterator pos_union;
114 
115     // Put all the elements of set_a into the union.
116 
117     for (pos = set_a.begin(); pos != set_a.end(); ++pos)
118         union_set.insert(*pos);
119 
120     // Put all the elements of set_b that are not already there into the union.
121     for (pos = set_b.begin(); pos != set_b.end(); ++pos)
122     {
123         pos_union = union_set.find(*pos);
124         if (pos_union == union_set.end())
125             union_set.insert(*pos);
126     }
127 }
128 
129 bool
130 Options::VerifyOptions (CommandReturnObject &result)
131 {
132     bool options_are_valid = false;
133 
134     int num_levels = GetRequiredOptions().size();
135     if (num_levels)
136     {
137         for (int i = 0; i < num_levels && !options_are_valid; ++i)
138         {
139             // This is the correct set of options if:  1). m_seen_options contains all of m_required_options[i]
140             // (i.e. all the required options at this level are a subset of m_seen_options); AND
141             // 2). { m_seen_options - m_required_options[i] is a subset of m_options_options[i] (i.e. all the rest of
142             // m_seen_options are in the set of optional options at this level.
143 
144             // Check to see if all of m_required_options[i] are a subset of m_seen_options
145             if (IsASubset (GetRequiredOptions()[i], m_seen_options))
146             {
147                 // Construct the set difference: remaining_options = {m_seen_options} - {m_required_options[i]}
148                 OptionSet remaining_options;
149                 OptionsSetDiff (m_seen_options, GetRequiredOptions()[i], remaining_options);
150                 // Check to see if remaining_options is a subset of m_optional_options[i]
151                 if (IsASubset (remaining_options, GetOptionalOptions()[i]))
152                     options_are_valid = true;
153             }
154         }
155     }
156     else
157     {
158         options_are_valid = true;
159     }
160 
161     if (options_are_valid)
162     {
163         result.SetStatus (eReturnStatusSuccessFinishNoResult);
164     }
165     else
166     {
167         result.AppendError ("invalid combination of options for the given command");
168         result.SetStatus (eReturnStatusFailed);
169     }
170 
171     return options_are_valid;
172 }
173 
174 // This is called in the Options constructor, though we could call it lazily if that ends up being
175 // a performance problem.
176 
177 void
178 Options::BuildValidOptionSets ()
179 {
180     // Check to see if we already did this.
181     if (m_required_options.size() != 0)
182         return;
183 
184     // Check to see if there are any options.
185     int num_options = NumCommandOptions ();
186     if (num_options == 0)
187         return;
188 
189     const OptionDefinition *opt_defs = GetDefinitions();
190     m_required_options.resize(1);
191     m_optional_options.resize(1);
192 
193     // First count the number of option sets we've got.  Ignore LLDB_ALL_OPTION_SETS...
194 
195     uint32_t num_option_sets = 0;
196 
197     for (int i = 0; i < num_options; i++)
198     {
199         uint32_t this_usage_mask = opt_defs[i].usage_mask;
200         if (this_usage_mask == LLDB_OPT_SET_ALL)
201         {
202             if (num_option_sets == 0)
203                 num_option_sets = 1;
204         }
205         else
206         {
207             for (int j = 0; j < LLDB_MAX_NUM_OPTION_SETS; j++)
208             {
209                 if (this_usage_mask & (1 << j))
210                 {
211                     if (num_option_sets <= j)
212                         num_option_sets = j + 1;
213                 }
214             }
215         }
216     }
217 
218     if (num_option_sets > 0)
219     {
220         m_required_options.resize(num_option_sets);
221         m_optional_options.resize(num_option_sets);
222 
223         for (int i = 0; i < num_options; ++i)
224         {
225             for (int j = 0; j < num_option_sets; j++)
226             {
227                 if (opt_defs[i].usage_mask & 1 << j)
228                 {
229                     if (opt_defs[i].required)
230                         m_required_options[j].insert(opt_defs[i].short_option);
231                     else
232                         m_optional_options[j].insert(opt_defs[i].short_option);
233                 }
234             }
235         }
236     }
237 }
238 
239 uint32_t
240 Options::NumCommandOptions ()
241 {
242     const OptionDefinition *opt_defs = GetDefinitions ();
243     if (opt_defs == NULL)
244         return 0;
245 
246     int i = 0;
247 
248     if (opt_defs != NULL)
249     {
250         while (opt_defs[i].long_option != NULL)
251             ++i;
252     }
253 
254     return i;
255 }
256 
257 struct option *
258 Options::GetLongOptions ()
259 {
260     // Check to see if this has already been done.
261     if (m_getopt_table.empty())
262     {
263         // Check to see if there are any options.
264         const uint32_t num_options = NumCommandOptions();
265         if (num_options == 0)
266             return NULL;
267 
268         uint32_t i;
269         uint32_t j;
270         const OptionDefinition *opt_defs = GetDefinitions();
271 
272         std::bitset<256> option_seen;
273 
274         m_getopt_table.resize(num_options + 1);
275         for (i = 0, j = 0; i < num_options; ++i)
276         {
277             char short_opt = opt_defs[i].short_option;
278 
279             if (option_seen.test(short_opt) == false)
280             {
281                 m_getopt_table[j].name    = opt_defs[i].long_option;
282                 m_getopt_table[j].has_arg = opt_defs[i].option_has_arg;
283                 m_getopt_table[j].flag    = NULL;
284                 m_getopt_table[j].val     = opt_defs[i].short_option;
285                 option_seen.set(short_opt);
286                 ++j;
287             }
288         }
289 
290         //getopt_long requires a NULL final entry in the table:
291 
292         m_getopt_table[j].name    = NULL;
293         m_getopt_table[j].has_arg = 0;
294         m_getopt_table[j].flag    = NULL;
295         m_getopt_table[j].val     = 0;
296     }
297 
298     if (m_getopt_table.empty())
299         return NULL;
300 
301     return &m_getopt_table.front();
302 }
303 
304 
305 // This function takes INDENT, which tells how many spaces to output at the front of each line; SPACES, which is
306 // a string containing 80 spaces; and TEXT, which is the text that is to be output.   It outputs the text, on
307 // multiple lines if necessary, to RESULT, with INDENT spaces at the front of each line.  It breaks lines on spaces,
308 // tabs or newlines, shortening the line if necessary to not break in the middle of a word.  It assumes that each
309 // output line should contain a maximum of OUTPUT_MAX_COLUMNS characters.
310 
311 
312 void
313 Options::OutputFormattedUsageText
314 (
315     Stream &strm,
316     const char *text,
317     uint32_t output_max_columns
318 )
319 {
320     int len = strlen (text);
321 
322     // Will it all fit on one line?
323 
324     if ((len + strm.GetIndentLevel()) < output_max_columns)
325     {
326         // Output it as a single line.
327         strm.Indent (text);
328         strm.EOL();
329     }
330     else
331     {
332         // We need to break it up into multiple lines.
333 
334         int text_width = output_max_columns - strm.GetIndentLevel() - 1;
335         int start = 0;
336         int end = start;
337         int final_end = strlen (text);
338         int sub_len;
339 
340         while (end < final_end)
341         {
342             // Don't start the 'text' on a space, since we're already outputting the indentation.
343             while ((start < final_end) && (text[start] == ' '))
344                 start++;
345 
346             end = start + text_width;
347             if (end > final_end)
348                 end = final_end;
349             else
350             {
351                 // If we're not at the end of the text, make sure we break the line on white space.
352                 while (end > start
353                        && text[end] != ' ' && text[end] != '\t' && text[end] != '\n')
354                     end--;
355             }
356 
357             sub_len = end - start;
358             if (start != 0)
359                 strm.EOL();
360             strm.Indent();
361             assert (start < final_end);
362             assert (start + sub_len <= final_end);
363             strm.Write(text + start, sub_len);
364             start = end + 1;
365         }
366         strm.EOL();
367     }
368 }
369 
370 bool
371 Options::SupportsLongOption (const char *long_option)
372 {
373     if (long_option && long_option[0])
374     {
375         const OptionDefinition *opt_defs = GetDefinitions ();
376         if (opt_defs)
377         {
378             const char *long_option_name = long_option;
379             if (long_option[0] == '-' && long_option[1] == '-')
380                 long_option_name += 2;
381 
382             for (uint32_t i = 0; opt_defs[i].long_option; ++i)
383             {
384                 if (strcmp(opt_defs[i].long_option, long_option_name) == 0)
385                     return true;
386             }
387         }
388     }
389     return false;
390 }
391 
392 void
393 Options::GenerateOptionUsage
394 (
395     Stream &strm,
396     CommandObject *cmd
397 )
398 {
399     const uint32_t screen_width = m_interpreter.GetDebugger().GetTerminalWidth();
400 
401     const OptionDefinition *opt_defs = GetDefinitions();
402     const uint32_t save_indent_level = strm.GetIndentLevel();
403     const char *name;
404 
405     StreamString arguments_str;
406 
407     if (cmd)
408     {
409         name = cmd->GetCommandName();
410         cmd->GetFormattedCommandArguments (arguments_str);
411     }
412     else
413         name = "";
414 
415     strm.PutCString ("\nCommand Options Usage:\n");
416 
417     strm.IndentMore(2);
418 
419     // First, show each usage level set of options, e.g. <cmd> [options-for-level-0]
420     //                                                   <cmd> [options-for-level-1]
421     //                                                   etc.
422 
423     const uint32_t num_options = NumCommandOptions();
424     if (num_options == 0)
425         return;
426 
427     int num_option_sets = GetRequiredOptions().size();
428 
429     uint32_t i;
430 
431     for (uint32_t opt_set = 0; opt_set < num_option_sets; ++opt_set)
432     {
433         uint32_t opt_set_mask;
434 
435         opt_set_mask = 1 << opt_set;
436         if (opt_set > 0)
437             strm.Printf ("\n");
438         strm.Indent (name);
439 
440         // Different option sets may require different args.
441         StreamString args_str;
442         if (cmd)
443             cmd->GetFormattedCommandArguments(args_str, opt_set_mask);
444 
445         // First go through and print all options that take no arguments as
446         // a single string. If a command has "-a" "-b" and "-c", this will show
447         // up as [-abc]
448 
449         std::set<char> options;
450         std::set<char>::const_iterator options_pos, options_end;
451         bool first;
452         for (i = 0, first = true; i < num_options; ++i)
453         {
454             if (opt_defs[i].usage_mask & opt_set_mask)
455             {
456                 // Add current option to the end of out_stream.
457 
458                 if (opt_defs[i].required == true &&
459                     opt_defs[i].option_has_arg == no_argument)
460                 {
461                     options.insert (opt_defs[i].short_option);
462                 }
463             }
464         }
465 
466         if (options.empty() == false)
467         {
468             // We have some required options with no arguments
469             strm.PutCString(" -");
470             for (i=0; i<2; ++i)
471                 for (options_pos = options.begin(), options_end = options.end();
472                      options_pos != options_end;
473                      ++options_pos)
474                 {
475                     if (i==0 && ::isupper (*options_pos))
476                         continue;
477                     if (i==1 && ::islower (*options_pos))
478                         continue;
479                     strm << *options_pos;
480                 }
481         }
482 
483         for (i = 0, options.clear(); i < num_options; ++i)
484         {
485             if (opt_defs[i].usage_mask & opt_set_mask)
486             {
487                 // Add current option to the end of out_stream.
488 
489                 if (opt_defs[i].required == false &&
490                     opt_defs[i].option_has_arg == no_argument)
491                 {
492                     options.insert (opt_defs[i].short_option);
493                 }
494             }
495         }
496 
497         if (options.empty() == false)
498         {
499             // We have some required options with no arguments
500             strm.PutCString(" [-");
501             for (i=0; i<2; ++i)
502                 for (options_pos = options.begin(), options_end = options.end();
503                      options_pos != options_end;
504                      ++options_pos)
505                 {
506                     if (i==0 && ::isupper (*options_pos))
507                         continue;
508                     if (i==1 && ::islower (*options_pos))
509                         continue;
510                     strm << *options_pos;
511                 }
512             strm.PutChar(']');
513         }
514 
515         // First go through and print the required options (list them up front).
516 
517         for (i = 0; i < num_options; ++i)
518         {
519             if (opt_defs[i].usage_mask & opt_set_mask)
520             {
521                 // Add current option to the end of out_stream.
522                 CommandArgumentType arg_type = opt_defs[i].argument_type;
523 
524                 if (opt_defs[i].required)
525                 {
526                     if (opt_defs[i].option_has_arg == required_argument)
527                     {
528                         strm.Printf (" -%c <%s>",
529                                      opt_defs[i].short_option,
530                                      CommandObject::GetArgumentName (arg_type));
531                     }
532                     else if (opt_defs[i].option_has_arg == optional_argument)
533                     {
534                         strm.Printf (" -%c [<%s>]",
535                                      opt_defs[i].short_option,
536                                      CommandObject::GetArgumentName (arg_type));
537                     }
538                 }
539             }
540         }
541 
542         // Now go through again, and this time only print the optional options.
543 
544         for (i = 0; i < num_options; ++i)
545         {
546             if (opt_defs[i].usage_mask & opt_set_mask)
547             {
548                 // Add current option to the end of out_stream.
549 
550                 CommandArgumentType arg_type = opt_defs[i].argument_type;
551 
552                 if (! opt_defs[i].required)
553                 {
554                     if (opt_defs[i].option_has_arg == required_argument)
555                         strm.Printf (" [-%c <%s>]", opt_defs[i].short_option,
556                                      CommandObject::GetArgumentName (arg_type));
557                     else if (opt_defs[i].option_has_arg == optional_argument)
558                         strm.Printf (" [-%c [<%s>]]", opt_defs[i].short_option,
559                                      CommandObject::GetArgumentName (arg_type));
560                 }
561             }
562         }
563 
564         if (args_str.GetSize() > 0)
565         {
566             if (cmd->WantsRawCommandString())
567                 strm.Printf(" --");
568 
569             strm.Printf (" %s", args_str.GetData());
570         }
571     }
572 
573     if (cmd &&
574         cmd->WantsRawCommandString() &&
575         arguments_str.GetSize() > 0)
576     {
577         strm.PutChar('\n');
578         strm.Indent(name);
579         strm.Printf(" %s", arguments_str.GetData());
580     }
581 
582     strm.Printf ("\n\n");
583 
584     // Now print out all the detailed information about the various options:  long form, short form and help text:
585     //   --long_name <argument>  ( -short <argument> )
586     //   help text
587 
588     // This variable is used to keep track of which options' info we've printed out, because some options can be in
589     // more than one usage level, but we only want to print the long form of its information once.
590 
591     OptionSet options_seen;
592     OptionSet::iterator pos;
593     strm.IndentMore (5);
594 
595     std::vector<char> sorted_options;
596 
597 
598     // Put the unique command options in a vector & sort it, so we can output them alphabetically (by short_option)
599     // when writing out detailed help for each option.
600 
601     for (i = 0; i < num_options; ++i)
602     {
603         pos = options_seen.find (opt_defs[i].short_option);
604         if (pos == options_seen.end())
605         {
606             options_seen.insert (opt_defs[i].short_option);
607             sorted_options.push_back (opt_defs[i].short_option);
608         }
609     }
610 
611     std::sort (sorted_options.begin(), sorted_options.end());
612 
613     // Go through the unique'd and alphabetically sorted vector of options, find the table entry for each option
614     // and write out the detailed help information for that option.
615 
616     int first_option_printed = 1;
617     size_t end = sorted_options.size();
618     for (size_t j = 0; j < end; ++j)
619     {
620         char option = sorted_options[j];
621         bool found = false;
622         for (i = 0; i < num_options && !found; ++i)
623         {
624             if (opt_defs[i].short_option == option)
625             {
626                 found = true;
627                 //Print out the help information for this option.
628 
629                 // Put a newline separation between arguments
630                 if (first_option_printed)
631                     first_option_printed = 0;
632                 else
633                     strm.EOL();
634 
635                 CommandArgumentType arg_type = opt_defs[i].argument_type;
636 
637                 StreamString arg_name_str;
638                 arg_name_str.Printf ("<%s>", CommandObject::GetArgumentName (arg_type));
639 
640                 strm.Indent ();
641                 strm.Printf ("-%c", opt_defs[i].short_option);
642                 if (arg_type != eArgTypeNone)
643                     strm.Printf (" <%s>",  CommandObject::GetArgumentName (arg_type));
644                 strm.Printf ("  ( --%s", opt_defs[i].long_option);
645                 if (arg_type != eArgTypeNone)
646                     strm.Printf (" <%s>", CommandObject::GetArgumentName (arg_type));
647                 strm.PutCString(" )\n");
648 
649                 strm.IndentMore (5);
650 
651                 if (opt_defs[i].usage_text)
652                     OutputFormattedUsageText (strm,
653                                               opt_defs[i].usage_text,
654                                               screen_width);
655                 if (opt_defs[i].enum_values != NULL)
656                 {
657                     strm.Indent ();
658                     strm.Printf("Values: ");
659                     for (int k = 0; opt_defs[i].enum_values[k].string_value != NULL; k++)
660                     {
661                         if (k == 0)
662                             strm.Printf("%s", opt_defs[i].enum_values[k].string_value);
663                         else
664                             strm.Printf(" | %s", opt_defs[i].enum_values[k].string_value);
665                     }
666                     strm.EOL();
667                 }
668                 strm.IndentLess (5);
669             }
670         }
671     }
672 
673     // Restore the indent level
674     strm.SetIndentLevel (save_indent_level);
675 }
676 
677 // This function is called when we have been given a potentially incomplete set of
678 // options, such as when an alias has been defined (more options might be added at
679 // at the time the alias is invoked).  We need to verify that the options in the set
680 // m_seen_options are all part of a set that may be used together, but m_seen_options
681 // may be missing some of the "required" options.
682 
683 bool
684 Options::VerifyPartialOptions (CommandReturnObject &result)
685 {
686     bool options_are_valid = false;
687 
688     int num_levels = GetRequiredOptions().size();
689     if (num_levels)
690       {
691         for (int i = 0; i < num_levels && !options_are_valid; ++i)
692           {
693             // In this case we are treating all options as optional rather than required.
694             // Therefore a set of options is correct if m_seen_options is a subset of the
695             // union of m_required_options and m_optional_options.
696             OptionSet union_set;
697             OptionsSetUnion (GetRequiredOptions()[i], GetOptionalOptions()[i], union_set);
698             if (IsASubset (m_seen_options, union_set))
699                 options_are_valid = true;
700           }
701       }
702 
703     return options_are_valid;
704 }
705 
706 bool
707 Options::HandleOptionCompletion
708 (
709     Args &input,
710     OptionElementVector &opt_element_vector,
711     int cursor_index,
712     int char_pos,
713     int match_start_point,
714     int max_return_elements,
715     bool &word_complete,
716     lldb_private::StringList &matches
717 )
718 {
719     word_complete = true;
720 
721     // For now we just scan the completions to see if the cursor position is in
722     // an option or its argument.  Otherwise we'll call HandleArgumentCompletion.
723     // In the future we can use completion to validate options as well if we want.
724 
725     const OptionDefinition *opt_defs = GetDefinitions();
726 
727     std::string cur_opt_std_str (input.GetArgumentAtIndex(cursor_index));
728     cur_opt_std_str.erase(char_pos);
729     const char *cur_opt_str = cur_opt_std_str.c_str();
730 
731     for (int i = 0; i < opt_element_vector.size(); i++)
732     {
733         int opt_pos = opt_element_vector[i].opt_pos;
734         int opt_arg_pos = opt_element_vector[i].opt_arg_pos;
735         int opt_defs_index = opt_element_vector[i].opt_defs_index;
736         if (opt_pos == cursor_index)
737         {
738             // We're completing the option itself.
739 
740             if (opt_defs_index == OptionArgElement::eBareDash)
741             {
742                 // We're completing a bare dash.  That means all options are open.
743                 // FIXME: We should scan the other options provided and only complete options
744                 // within the option group they belong to.
745                 char opt_str[3] = {'-', 'a', '\0'};
746 
747                 for (int j = 0 ; opt_defs[j].short_option != 0 ; j++)
748                 {
749                     opt_str[1] = opt_defs[j].short_option;
750                     matches.AppendString (opt_str);
751                 }
752                 return true;
753             }
754             else if (opt_defs_index == OptionArgElement::eBareDoubleDash)
755             {
756                 std::string full_name ("--");
757                 for (int j = 0 ; opt_defs[j].short_option != 0 ; j++)
758                 {
759                     full_name.erase(full_name.begin() + 2, full_name.end());
760                     full_name.append (opt_defs[j].long_option);
761                     matches.AppendString (full_name.c_str());
762                 }
763                 return true;
764             }
765             else if (opt_defs_index != OptionArgElement::eUnrecognizedArg)
766             {
767                 // We recognized it, if it an incomplete long option, complete it anyway (getopt_long is
768                 // happy with shortest unique string, but it's still a nice thing to do.)  Otherwise return
769                 // The string so the upper level code will know this is a full match and add the " ".
770                 if (cur_opt_str && strlen (cur_opt_str) > 2
771                     && cur_opt_str[0] == '-' && cur_opt_str[1] == '-'
772                     && strcmp (opt_defs[opt_defs_index].long_option, cur_opt_str) != 0)
773                 {
774                         std::string full_name ("--");
775                         full_name.append (opt_defs[opt_defs_index].long_option);
776                         matches.AppendString(full_name.c_str());
777                         return true;
778                 }
779                 else
780                 {
781                     matches.AppendString(input.GetArgumentAtIndex(cursor_index));
782                     return true;
783                 }
784             }
785             else
786             {
787                 // FIXME - not handling wrong options yet:
788                 // Check to see if they are writing a long option & complete it.
789                 // I think we will only get in here if the long option table has two elements
790                 // that are not unique up to this point.  getopt_long does shortest unique match
791                 // for long options already.
792 
793                 if (cur_opt_str && strlen (cur_opt_str) > 2
794                     && cur_opt_str[0] == '-' && cur_opt_str[1] == '-')
795                 {
796                     for (int j = 0 ; opt_defs[j].short_option != 0 ; j++)
797                     {
798                         if (strstr(opt_defs[j].long_option, cur_opt_str + 2) == opt_defs[j].long_option)
799                         {
800                             std::string full_name ("--");
801                             full_name.append (opt_defs[j].long_option);
802                             // The options definitions table has duplicates because of the
803                             // way the grouping information is stored, so only add once.
804                             bool duplicate = false;
805                             for (int k = 0; k < matches.GetSize(); k++)
806                             {
807                                 if (matches.GetStringAtIndex(k) == full_name)
808                                 {
809                                     duplicate = true;
810                                     break;
811                                 }
812                             }
813                             if (!duplicate)
814                                 matches.AppendString(full_name.c_str());
815                         }
816                     }
817                 }
818                 return true;
819             }
820 
821 
822         }
823         else if (opt_arg_pos == cursor_index)
824         {
825             // Okay the cursor is on the completion of an argument.
826             // See if it has a completion, otherwise return no matches.
827 
828             if (opt_defs_index != -1)
829             {
830                 HandleOptionArgumentCompletion (input,
831                                                 cursor_index,
832                                                 strlen (input.GetArgumentAtIndex(cursor_index)),
833                                                 opt_element_vector,
834                                                 i,
835                                                 match_start_point,
836                                                 max_return_elements,
837                                                 word_complete,
838                                                 matches);
839                 return true;
840             }
841             else
842             {
843                 // No completion callback means no completions...
844                 return true;
845             }
846 
847         }
848         else
849         {
850             // Not the last element, keep going.
851             continue;
852         }
853     }
854     return false;
855 }
856 
857 bool
858 Options::HandleOptionArgumentCompletion
859 (
860     Args &input,
861     int cursor_index,
862     int char_pos,
863     OptionElementVector &opt_element_vector,
864     int opt_element_index,
865     int match_start_point,
866     int max_return_elements,
867     bool &word_complete,
868     lldb_private::StringList &matches
869 )
870 {
871     const OptionDefinition *opt_defs = GetDefinitions();
872     std::auto_ptr<SearchFilter> filter_ap;
873 
874     int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
875     int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
876 
877     // See if this is an enumeration type option, and if so complete it here:
878 
879     OptionEnumValueElement *enum_values = opt_defs[opt_defs_index].enum_values;
880     if (enum_values != NULL)
881     {
882         bool return_value = false;
883         std::string match_string(input.GetArgumentAtIndex (opt_arg_pos), input.GetArgumentAtIndex (opt_arg_pos) + char_pos);
884         for (int i = 0; enum_values[i].string_value != NULL; i++)
885         {
886             if (strstr(enum_values[i].string_value, match_string.c_str()) == enum_values[i].string_value)
887             {
888                 matches.AppendString (enum_values[i].string_value);
889                 return_value = true;
890             }
891         }
892         return return_value;
893     }
894 
895     // If this is a source file or symbol type completion, and  there is a
896     // -shlib option somewhere in the supplied arguments, then make a search filter
897     // for that shared library.
898     // FIXME: Do we want to also have an "OptionType" so we don't have to match string names?
899 
900     uint32_t completion_mask = opt_defs[opt_defs_index].completion_type;
901 
902     if (completion_mask == 0)
903     {
904         lldb::CommandArgumentType option_arg_type = opt_defs[opt_defs_index].argument_type;
905         if (option_arg_type != eArgTypeNone)
906         {
907             CommandObject::ArgumentTableEntry *arg_entry = CommandObject::FindArgumentDataByType (opt_defs[opt_defs_index].argument_type);
908             if (arg_entry)
909                 completion_mask = arg_entry->completion_type;
910         }
911     }
912 
913     if (completion_mask & CommandCompletions::eSourceFileCompletion
914         || completion_mask & CommandCompletions::eSymbolCompletion)
915     {
916         for (int i = 0; i < opt_element_vector.size(); i++)
917         {
918             int cur_defs_index = opt_element_vector[i].opt_defs_index;
919             int cur_arg_pos    = opt_element_vector[i].opt_arg_pos;
920             const char *cur_opt_name = opt_defs[cur_defs_index].long_option;
921 
922             // If this is the "shlib" option and there was an argument provided,
923             // restrict it to that shared library.
924             if (strcmp(cur_opt_name, "shlib") == 0 && cur_arg_pos != -1)
925             {
926                 const char *module_name = input.GetArgumentAtIndex(cur_arg_pos);
927                 if (module_name)
928                 {
929                     FileSpec module_spec(module_name, false);
930                     lldb::TargetSP target_sp = m_interpreter.GetDebugger().GetSelectedTarget();
931                     // Search filters require a target...
932                     if (target_sp)
933                         filter_ap.reset (new SearchFilterByModule (target_sp, module_spec));
934                 }
935                 break;
936             }
937         }
938     }
939 
940     return CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter,
941                                                                 completion_mask,
942                                                                 input.GetArgumentAtIndex (opt_arg_pos),
943                                                                 match_start_point,
944                                                                 max_return_elements,
945                                                                 filter_ap.get(),
946                                                                 word_complete,
947                                                                 matches);
948 
949 }
950 
951 
952 void
953 OptionGroupOptions::Append (OptionGroup* group)
954 {
955     const OptionDefinition* group_option_defs = group->GetDefinitions ();
956     const uint32_t group_option_count = group->GetNumDefinitions();
957     for (uint32_t i=0; i<group_option_count; ++i)
958     {
959         m_option_infos.push_back (OptionInfo (group, i));
960         m_option_defs.push_back (group_option_defs[i]);
961     }
962 }
963 
964 void
965 OptionGroupOptions::Append (OptionGroup* group,
966                             uint32_t src_mask,
967                             uint32_t dst_mask)
968 {
969     const OptionDefinition* group_option_defs = group->GetDefinitions ();
970     const uint32_t group_option_count = group->GetNumDefinitions();
971     for (uint32_t i=0; i<group_option_count; ++i)
972     {
973         if (group_option_defs[i].usage_mask & src_mask)
974         {
975             m_option_infos.push_back (OptionInfo (group, i));
976             m_option_defs.push_back (group_option_defs[i]);
977             m_option_defs.back().usage_mask = dst_mask;
978         }
979     }
980 }
981 
982 void
983 OptionGroupOptions::Finalize ()
984 {
985     m_did_finalize = true;
986     OptionDefinition empty_option_def = { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL };
987     m_option_defs.push_back (empty_option_def);
988 }
989 
990 Error
991 OptionGroupOptions::SetOptionValue (uint32_t option_idx,
992                                     const char *option_value)
993 {
994     // After calling OptionGroupOptions::Append(...), you must finalize the groups
995     // by calling OptionGroupOptions::Finlize()
996     assert (m_did_finalize);
997     assert (m_option_infos.size() + 1 == m_option_defs.size());
998     Error error;
999     if (option_idx < m_option_infos.size())
1000     {
1001         error = m_option_infos[option_idx].option_group->SetOptionValue (m_interpreter,
1002                                                                          m_option_infos[option_idx].option_index,
1003                                                                          option_value);
1004 
1005     }
1006     else
1007     {
1008         error.SetErrorString ("invalid option index"); // Shouldn't happen...
1009     }
1010     return error;
1011 }
1012 
1013 void
1014 OptionGroupOptions::OptionParsingStarting ()
1015 {
1016     std::set<OptionGroup*> group_set;
1017     OptionInfos::iterator pos, end = m_option_infos.end();
1018     for (pos = m_option_infos.begin(); pos != end; ++pos)
1019     {
1020         OptionGroup* group = pos->option_group;
1021         if (group_set.find(group) == group_set.end())
1022         {
1023             group->OptionParsingStarting (m_interpreter);
1024             group_set.insert(group);
1025         }
1026     }
1027 }
1028 Error
1029 OptionGroupOptions::OptionParsingFinished ()
1030 {
1031     std::set<OptionGroup*> group_set;
1032     Error error;
1033     OptionInfos::iterator pos, end = m_option_infos.end();
1034     for (pos = m_option_infos.begin(); pos != end; ++pos)
1035     {
1036         OptionGroup* group = pos->option_group;
1037         if (group_set.find(group) == group_set.end())
1038         {
1039             error = group->OptionParsingFinished (m_interpreter);
1040             group_set.insert(group);
1041             if (error.Fail())
1042                 return error;
1043         }
1044     }
1045     return error;
1046 }
1047