Ugh, fix some copy and pasted code that somehow never got compiled or something
[oota-llvm.git] / tools / llvm-db / Commands.cpp
1 //===-- Commands.cpp - Implement various commands for the CLI -------------===//
2 // 
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
7 // 
8 //===----------------------------------------------------------------------===//
9 // 
10 // This file implements many builtin user commands.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "CLIDebugger.h"
15 #include "CLICommand.h"
16 #include "llvm/Debugger/ProgramInfo.h"
17 #include "llvm/Debugger/RuntimeInfo.h"
18 #include "llvm/Debugger/SourceLanguage.h"
19 #include "llvm/Debugger/SourceFile.h"
20 #include "llvm/Debugger/InferiorProcess.h"
21 #include "Support/FileUtilities.h"
22 #include "Support/StringExtras.h"
23 #include <iostream>
24 using namespace llvm;
25
26 /// getCurrentLanguage - Return the current source language that the user is
27 /// playing around with.  This is aquired from the current stack frame of a
28 /// running program if one exists, but this value can be explicitly set by the
29 /// user as well.
30 const SourceLanguage &CLIDebugger::getCurrentLanguage() const {
31   // If the user explicitly switched languages with 'set language', use what
32   // they asked for.
33   if (CurrentLanguage) {
34     return *CurrentLanguage;
35   } else if (Dbg.isProgramRunning()) {
36     // Otherwise, if the program is running, infer the current language from it.
37     const GlobalVariable *FuncDesc =
38       getRuntimeInfo().getCurrentFrame().getFunctionDesc();
39     return getProgramInfo().getFunction(FuncDesc).getSourceFile().getLanguage();
40   } else {
41     // Otherwise, default to C like GDB apparently does.
42     return SourceLanguage::getCFamilyInstance();
43   }
44 }
45
46 /// startProgramRunning - If the program has been updated, reload it, then
47 /// start executing the program.
48 void CLIDebugger::startProgramRunning() {
49   eliminateRunInfo();
50
51   // If the program has been modified, reload it!
52   std::string Program = Dbg.getProgramPath();
53   if (TheProgramInfo->getProgramTimeStamp() != getFileTimestamp(Program)) {
54     std::cout << "'" << Program << "' has changed; re-reading program.\n";
55
56     // Unload an existing program.  This kills the program if necessary.
57     Dbg.unloadProgram();
58     delete TheProgramInfo;
59     TheProgramInfo = 0;
60     CurrentFile = 0;
61
62     Dbg.loadProgram(Program);
63     TheProgramInfo = new ProgramInfo(Dbg.getProgram());
64   }
65
66   std::cout << "Starting program: " << Dbg.getProgramPath() << "\n";
67   Dbg.createProgram();
68
69   // There was no current frame.
70   LastCurrentFrame = 0;
71 }
72
73 /// printSourceLine - Print the specified line of the current source file.
74 /// If the specified line is invalid (the source file could not be loaded or
75 /// the line number is out of range), don't print anything, but return true.
76 bool CLIDebugger::printSourceLine(unsigned LineNo) {
77   assert(CurrentFile && "There is no current source file to print!");
78   const char *LineStart, *LineEnd;
79   CurrentFile->getSourceLine(LineNo-1, LineStart, LineEnd);
80   if (LineStart == 0) return true;
81   std::cout << LineNo;
82
83   // If this is the line the program is currently stopped at, print a marker.
84   if (Dbg.isProgramRunning()) {
85     unsigned CurLineNo, CurColNo;
86     const SourceFileInfo *CurSFI;
87     getRuntimeInfo().getCurrentFrame().getSourceLocation(CurLineNo, CurColNo,
88                                                          CurSFI);
89
90     if (CurLineNo == LineNo && CurrentFile == &CurSFI->getSourceText())
91       std::cout << " ->";
92   }
93
94   std::cout << "\t" << std::string(LineStart, LineEnd) << "\n"; 
95   return false;
96 }
97
98 /// printProgramLocation - Print a line of the place where the current stack
99 /// frame has stopped and the source line it is on.
100 ///
101 void CLIDebugger::printProgramLocation(bool PrintLocation) {
102   assert(Dbg.isProgramLoaded() && Dbg.isProgramRunning() &&
103          "Error program is not loaded and running!");
104
105   // Figure out where the program stopped...
106   StackFrame &SF = getRuntimeInfo().getCurrentFrame();
107   unsigned LineNo, ColNo;
108   const SourceFileInfo *FileDesc;
109   SF.getSourceLocation(LineNo, ColNo, FileDesc);
110
111   // If requested, print out some program information about WHERE we are.
112   if (PrintLocation) {
113     // FIXME: print the current function arguments
114     if (const GlobalVariable *FuncDesc = SF.getFunctionDesc())
115       std::cout << getProgramInfo().getFunction(FuncDesc).getSymbolicName();
116     else
117       std::cout << "<unknown function>";
118     
119     CurrentFile = &FileDesc->getSourceText();
120     
121     std::cout << " at " << CurrentFile->getFilename() << ":" << LineNo;
122     if (ColNo) std::cout << ":" << ColNo << "\n";
123   }
124
125   if (printSourceLine(LineNo))
126     std::cout << "<could not load source file>\n";
127   else {
128     LineListedStart = LineNo-ListSize/2+1;
129     if ((int)LineListedStart < 1) LineListedStart = 1;
130     LineListedEnd = LineListedStart+1;
131   }
132 }
133
134 /// eliminateRunInfo - We are about to run the program.  Forget any state
135 /// about how the program used to be stopped.
136 void CLIDebugger::eliminateRunInfo() {
137   delete TheRuntimeInfo;
138   TheRuntimeInfo = 0;
139 }
140
141 /// programStoppedSuccessfully - This method updates internal data
142 /// structures to reflect the fact that the program just executed a while,
143 /// and has successfully stopped.
144 void CLIDebugger::programStoppedSuccessfully() {
145   assert(TheRuntimeInfo==0 && "Someone forgot to release the old RuntimeInfo!");
146
147   TheRuntimeInfo = new RuntimeInfo(TheProgramInfo, Dbg.getRunningProcess());
148
149   // FIXME: if there are any breakpoints at the current location, print them as
150   // well.
151
152   // Since the program as successfully stopped, print its location.
153   void *CurrentFrame = getRuntimeInfo().getCurrentFrame().getFrameID();
154   printProgramLocation(CurrentFrame != LastCurrentFrame);
155   LastCurrentFrame = CurrentFrame;
156 }
157
158
159
160 /// getUnsignedIntegerOption - Get an unsigned integer number from the Val
161 /// string.  Check to make sure that the string contains an unsigned integer
162 /// token, and if not, throw an exception.  If isOnlyOption is set, also throw
163 /// an exception if there is extra junk at the end of the string.
164 static unsigned getUnsignedIntegerOption(const char *Msg, std::string &Val,
165                                          bool isOnlyOption = true) {
166   std::string Tok = getToken(Val);
167   if (Tok.empty() || (isOnlyOption && !getToken(Val).empty()))
168     throw std::string(Msg) + " expects an unsigned integer argument.";
169   
170   char *EndPtr;
171   unsigned Result = strtoul(Tok.c_str(), &EndPtr, 0);
172   if (EndPtr != Tok.c_str()+Tok.size())
173     throw std::string(Msg) + " expects an unsigned integer argument.";
174
175   return Result;
176 }
177
178 /// getOptionalUnsignedIntegerOption - This method is just like
179 /// getUnsignedIntegerOption, but if the argument value is not specified, a
180 /// default is returned instead of causing an error.
181 static unsigned 
182 getOptionalUnsignedIntegerOption(const char *Msg, unsigned Default,
183                                  std::string &Val, bool isOnlyOption = true) {
184   // Check to see if the value was specified...
185   std::string TokVal = getToken(Val);
186   if (TokVal.empty()) return Default;
187
188   // If it was specified, add it back to the value we are parsing...
189   Val = TokVal+Val;
190
191   // And parse normally.
192   return getUnsignedIntegerOption(Msg, Val, isOnlyOption);
193 }
194
195
196 /// parseProgramOptions - This method parses the Options string and loads it
197 /// as options to be passed to the program.  This is used by the run command
198 /// and by 'set args'.
199 void CLIDebugger::parseProgramOptions(std::string &Options) {
200   // FIXME: tokenizing by whitespace is clearly incorrect.  Instead we should
201   // honor quotes and other things that a shell would.  Also in the future we
202   // should support redirection of standard IO.
203  
204   std::vector<std::string> Arguments;
205   for (std::string A = getToken(Options); !A.empty(); A = getToken(Options))
206     Arguments.push_back(A);
207   Dbg.setProgramArguments(Arguments.begin(), Arguments.end());
208 }
209                                                 
210
211 //===----------------------------------------------------------------------===//
212 //                   Program startup and shutdown options
213 //===----------------------------------------------------------------------===//
214
215
216 /// file command - If the user specifies an option, search the PATH for the
217 /// specified program/bytecode file and load it.  If the user does not specify
218 /// an option, unload the current program.
219 void CLIDebugger::fileCommand(std::string &Options) {
220   std::string Prog = getToken(Options);
221   if (!getToken(Options).empty())
222     throw "file command takes at most one argument.";
223
224   // Check to make sure the user knows what they are doing
225   if (Dbg.isProgramRunning() &&
226       !askYesNo("A program is already loaded.  Kill it?"))
227     return;
228
229   // Unload an existing program.  This kills the program if necessary.
230   eliminateRunInfo();
231   delete TheProgramInfo;
232   TheProgramInfo = 0;
233   Dbg.unloadProgram();
234   CurrentFile = 0;
235
236   // If requested, start the new program.
237   if (Prog.empty()) {
238     std::cout << "Unloaded program.\n";
239   } else {
240     std::cout << "Loading program... " << std::flush;
241     Dbg.loadProgram(Prog);
242     assert(Dbg.isProgramLoaded() &&
243            "loadProgram succeeded, but not program loaded!");
244     TheProgramInfo = new ProgramInfo(Dbg.getProgram());
245     std::cout << "successfully loaded '" << Dbg.getProgramPath() << "'!\n";
246   }
247 }
248
249
250 void CLIDebugger::createCommand(std::string &Options) {
251   if (!getToken(Options).empty())
252     throw "create command does not take any arguments.";
253   if (!Dbg.isProgramLoaded()) throw "No program loaded.";
254   if (Dbg.isProgramRunning() &&
255       !askYesNo("The program is already running.  Restart from the beginning?"))
256     return;
257
258   // Start the program running.
259   startProgramRunning();
260
261   // The program stopped!
262   programStoppedSuccessfully();
263 }
264
265 void CLIDebugger::killCommand(std::string &Options) {
266   if (!getToken(Options).empty())
267     throw "kill command does not take any arguments.";
268   if (!Dbg.isProgramRunning())
269     throw "No program is currently being run.";
270
271   if (askYesNo("Kill the program being debugged?"))
272     Dbg.killProgram();
273   eliminateRunInfo();
274 }
275
276 void CLIDebugger::quitCommand(std::string &Options) {
277   if (!getToken(Options).empty())
278     throw "quit command does not take any arguments.";
279
280   if (Dbg.isProgramRunning() &&
281       !askYesNo("The program is running.  Exit anyway?"))
282     return;
283
284   // Throw exception to get out of the user-input loop.
285   throw 0;
286 }
287
288
289 //===----------------------------------------------------------------------===//
290 //                        Program execution commands
291 //===----------------------------------------------------------------------===//
292
293 void CLIDebugger::runCommand(std::string &Options) {
294   if (!Dbg.isProgramLoaded()) throw "No program loaded.";
295   if (Dbg.isProgramRunning() &&
296       !askYesNo("The program is already running.  Restart from the beginning?"))
297     return;
298
299   // Parse all of the options to the run command, which specify program
300   // arguments to run with.
301   parseProgramOptions(Options);
302
303   eliminateRunInfo();
304
305   // Start the program running.
306   startProgramRunning();
307
308   // Start the program running...
309   Options = "";
310   contCommand(Options);
311 }
312
313 void CLIDebugger::contCommand(std::string &Options) {
314   if (!getToken(Options).empty()) throw "cont argument not supported yet.";
315   if (!Dbg.isProgramRunning()) throw "Program is not running.";
316
317   eliminateRunInfo();
318
319   Dbg.contProgram();
320
321   // The program stopped!
322   programStoppedSuccessfully();
323 }
324
325 void CLIDebugger::stepCommand(std::string &Options) {
326   if (!Dbg.isProgramRunning()) throw "Program is not running.";
327
328   // Figure out how many times to step.
329   unsigned Amount =
330     getOptionalUnsignedIntegerOption("'step' command", 1, Options);
331
332   eliminateRunInfo();
333
334   // Step the specified number of times.
335   for (; Amount; --Amount)
336     Dbg.stepProgram();
337
338   // The program stopped!
339   programStoppedSuccessfully();
340 }
341
342 void CLIDebugger::nextCommand(std::string &Options) {
343   if (!Dbg.isProgramRunning()) throw "Program is not running.";
344   unsigned Amount =
345     getOptionalUnsignedIntegerOption("'next' command", 1, Options);
346
347   eliminateRunInfo();
348
349   for (; Amount; --Amount)
350     Dbg.nextProgram();
351
352   // The program stopped!
353   programStoppedSuccessfully();
354 }
355
356 void CLIDebugger::finishCommand(std::string &Options) {
357   if (!getToken(Options).empty())
358     throw "finish command does not take any arguments.";
359   if (!Dbg.isProgramRunning()) throw "Program is not running.";
360
361   // Figure out where we are exactly.  If the user requests that we return from
362   // a frame that is not the top frame, make sure we get it.
363   void *CurrentFrame = getRuntimeInfo().getCurrentFrame().getFrameID();
364
365   eliminateRunInfo();
366
367   Dbg.finishProgram(CurrentFrame);
368
369   // The program stopped!
370   programStoppedSuccessfully();
371 }
372
373 //===----------------------------------------------------------------------===//
374 //                           Stack frame commands
375 //===----------------------------------------------------------------------===//
376
377 void CLIDebugger::backtraceCommand(std::string &Options) {
378   // Accepts "full", n, -n
379   if (!getToken(Options).empty())
380     throw "FIXME: bt command argument not implemented yet!";
381
382   RuntimeInfo &RI = getRuntimeInfo();
383   ProgramInfo &PI = getProgramInfo();
384
385   try {
386     for (unsigned i = 0; ; ++i) {
387       StackFrame &SF = RI.getStackFrame(i);
388       std::cout << "#" << i;
389       if (i == RI.getCurrentFrameIdx())
390         std::cout << " ->";
391       std::cout << "\t" << SF.getFrameID() << " in ";
392       if (const GlobalVariable *G = SF.getFunctionDesc())
393         std::cout << PI.getFunction(G).getSymbolicName();
394
395       unsigned LineNo, ColNo;
396       const SourceFileInfo *SFI;
397       SF.getSourceLocation(LineNo, ColNo, SFI);
398       if (!SFI->getBaseName().empty()) {
399         std::cout << " at " << SFI->getBaseName();
400         if (LineNo) {
401           std::cout << ":" << LineNo;
402           if (ColNo)
403             std::cout << ":" << ColNo;
404         }
405       }
406
407       // FIXME: when we support shared libraries, we should print ' from foo.so'
408       // if the stack frame is from a different object than the current one.
409
410       std::cout << "\n";
411     }
412   } catch (...) {
413     // Stop automatically when we run off the bottom of the stack.
414   }
415 }
416
417 void CLIDebugger::upCommand(std::string &Options) {
418   unsigned Num =
419     getOptionalUnsignedIntegerOption("'up' command", 1, Options);
420
421   RuntimeInfo &RI = getRuntimeInfo();
422   unsigned CurFrame = RI.getCurrentFrameIdx();
423
424   // Check to see if we go can up the specified number of frames.
425   try {
426     RI.getStackFrame(CurFrame+Num);
427   } catch (...) {
428     if (Num == 1)
429       throw "Initial frame selected; you cannot go up.";
430     else
431       throw "Cannot go up " + utostr(Num) + " frames!";
432   }
433
434   RI.setCurrentFrameIdx(CurFrame+Num);
435   printProgramLocation();
436 }
437
438 void CLIDebugger::downCommand(std::string &Options) {
439   unsigned Num =
440     getOptionalUnsignedIntegerOption("'down' command", 1, Options);
441
442   RuntimeInfo &RI = getRuntimeInfo();
443   unsigned CurFrame = RI.getCurrentFrameIdx();
444
445   // Check to see if we can go up the specified number of frames.
446   if (CurFrame < Num)
447     if (Num == 1)
448       throw "Bottom (i.e., innermost) frame selected; you cannot go down.";
449     else
450       throw "Cannot go down " + utostr(Num) + " frames!";
451
452   RI.setCurrentFrameIdx(CurFrame-Num);
453   printProgramLocation();
454 }
455
456 void CLIDebugger::frameCommand(std::string &Options) {
457   RuntimeInfo &RI = getRuntimeInfo();
458   unsigned CurFrame = RI.getCurrentFrameIdx();
459
460   unsigned Num =
461     getOptionalUnsignedIntegerOption("'frame' command", CurFrame, Options);
462
463   // Check to see if we go to the specified frame.
464   RI.getStackFrame(Num);
465
466   RI.setCurrentFrameIdx(Num);
467   printProgramLocation();
468 }
469
470
471 //===----------------------------------------------------------------------===//
472 //                        Breakpoint related commands
473 //===----------------------------------------------------------------------===//
474
475 void CLIDebugger::breakCommand(std::string &Options) {
476   // Figure out where the user wants a breakpoint.
477   const SourceFile *File;
478   unsigned LineNo;
479   
480   // Check to see if the user specified a line specifier.
481   std::string Option = getToken(Options);  // strip whitespace
482   if (!Option.empty()) {
483     Options = Option + Options;  // reconstruct string
484
485     // Parse the line specifier.
486     parseLineSpec(Options, File, LineNo);
487   } else {
488     // Build a line specifier for the current stack frame.
489     throw "FIXME: breaking at the current location is not implemented yet!";
490   }
491   
492   
493   
494   throw "breakpoints not implemented yet!";
495 }
496
497 //===----------------------------------------------------------------------===//
498 //                          Miscellaneous commands
499 //===----------------------------------------------------------------------===//
500
501 void CLIDebugger::infoCommand(std::string &Options) {
502   std::string What = getToken(Options);
503
504   if (What.empty() || !getToken(Options).empty())
505     throw "info command expects exactly one argument.";
506
507   if (What == "frame") {
508   } else if (What == "functions") {
509     const std::map<const GlobalVariable*, SourceFunctionInfo*> &Functions
510       = getProgramInfo().getSourceFunctions();
511     std::cout << "All defined functions:\n";
512     // FIXME: GDB groups these by source file.  We could do that I guess.
513     for (std::map<const GlobalVariable*, SourceFunctionInfo*>::const_iterator
514            I = Functions.begin(), E = Functions.end(); I != E; ++I) {
515       std::cout << I->second->getSymbolicName() << "\n";
516     }
517
518   } else if (What == "source") {
519     if (CurrentFile == 0)
520       throw "No current source file.";
521
522     // Get the SourceFile information for the current file.
523     const SourceFileInfo &SF =
524       getProgramInfo().getSourceFile(CurrentFile->getDescriptor());
525
526     std::cout << "Current source file is: " << SF.getBaseName() << "\n"
527               << "Compilation directory is: " << SF.getDirectory() << "\n";
528     if (unsigned NL = CurrentFile->getNumLines())
529       std::cout << "Located in: " << CurrentFile->getFilename() << "\n"
530                 << "Contains " << NL << " lines\n";
531     else
532       std::cout << "Could not find source file.\n";
533     std::cout << "Source language is "
534               << SF.getLanguage().getSourceLanguageName() << "\n";
535
536   } else if (What == "sources") {
537     const std::map<const GlobalVariable*, SourceFileInfo*> &SourceFiles = 
538       getProgramInfo().getSourceFiles();
539     std::cout << "Source files for the program:\n";
540     for (std::map<const GlobalVariable*, SourceFileInfo*>::const_iterator I =
541            SourceFiles.begin(), E = SourceFiles.end(); I != E;) {
542       std::cout << I->second->getDirectory() << "/"
543                 << I->second->getBaseName();
544       ++I;
545       if (I != E) std::cout << ", ";
546     }
547     std::cout << "\n";
548   } else if (What == "target") {
549     std::cout << Dbg.getRunningProcess().getStatus();
550   } else {
551     // See if this is something handled by the current language.
552     if (getCurrentLanguage().printInfo(What))
553       return;
554
555     throw "Unknown info command '" + What + "'.  Try 'help info'.";
556   }
557 }
558
559 /// parseLineSpec - Parses a line specifier, for use by the 'list' command.
560 /// If SourceFile is returned as a void pointer, then it was not specified.
561 /// If the line specifier is invalid, an exception is thrown.
562 void CLIDebugger::parseLineSpec(std::string &LineSpec,
563                                 const SourceFile *&SourceFile,
564                                 unsigned &LineNo) {
565   SourceFile = 0;
566   LineNo = 0;
567
568   // First, check to see if we have a : separator.
569   std::string FirstPart = getToken(LineSpec, ":");
570   std::string SecondPart = getToken(LineSpec, ":");
571   if (!getToken(LineSpec).empty()) throw "Malformed line specification!";
572
573   // If there is no second part, we must have either "function", "number",
574   // "+offset", or "-offset".
575   if (SecondPart.empty()) {
576     if (FirstPart.empty()) throw "Malformed line specification!";
577     if (FirstPart[0] == '+') {
578       FirstPart.erase(FirstPart.begin(), FirstPart.begin()+1);
579       // For +n, return LineListedEnd+n
580       LineNo = LineListedEnd +
581                getUnsignedIntegerOption("Line specifier '+'", FirstPart);
582
583     } else if (FirstPart[0] == '-') {
584       FirstPart.erase(FirstPart.begin(), FirstPart.begin()+1);
585       // For -n, return LineListedEnd-n
586       LineNo = LineListedEnd -
587                getUnsignedIntegerOption("Line specifier '-'", FirstPart);
588       if ((int)LineNo < 1) LineNo = 1;
589     } else if (FirstPart[0] == '*') {
590       throw "Address expressions not supported as source locations!";
591     } else {
592       // Ok, check to see if this is just a line number.
593       std::string Saved = FirstPart;
594       try {
595         LineNo = getUnsignedIntegerOption("", Saved);
596       } catch (...) {
597         // Ok, it's not a valid line number.  It must be a source-language
598         // entity name.
599         std::string Name = getToken(FirstPart);
600         if (!getToken(FirstPart).empty())
601           throw "Extra junk in line specifier after '" + Name + "'.";
602         SourceFunctionInfo *SFI = 
603           getCurrentLanguage().lookupFunction(Name, getProgramInfo(),
604                                               TheRuntimeInfo);
605         if (SFI == 0)
606           throw "Unknown identifier '" + Name + "'.";
607
608         unsigned L, C;
609         SFI->getSourceLocation(L, C);
610         if (L == 0) throw "Could not locate '" + Name + "'!";
611         LineNo = L;
612         SourceFile = &SFI->getSourceFile().getSourceText();
613         return;
614       }
615     }
616
617   } else {
618     // Ok, this must be a filename qualified line number or function name.
619     // First, figure out the source filename.
620     std::string SourceFilename = getToken(FirstPart);
621     if (!getToken(FirstPart).empty())
622       throw "Invalid filename qualified source location!";
623
624     // Next, check to see if this is just a line number.
625     std::string Saved = SecondPart;
626     try {
627       LineNo = getUnsignedIntegerOption("", Saved);
628     } catch (...) {
629       // Ok, it's not a valid line number.  It must be a function name.
630       throw "FIXME: Filename qualified function names are not support "
631             "as line specifiers yet!";
632     }
633
634     // Ok, we got the line number.  Now check out the source file name to make
635     // sure it's all good.  If it is, return it.  If not, throw exception.
636     SourceFile =&getProgramInfo().getSourceFile(SourceFilename).getSourceText();
637   }
638 }
639
640 void CLIDebugger::listCommand(std::string &Options) {
641   if (!Dbg.isProgramLoaded())
642     throw "No program is loaded.  Use the 'file' command.";
643
644   // Handle "list foo," correctly, by returning " " as the second token
645   Options += " ";
646   
647   std::string FirstLineSpec = getToken(Options, ",");
648   std::string SecondLineSpec = getToken(Options, ",");
649   if (!getToken(Options, ",").empty())
650     throw "list command only expects two source location specifiers!";
651
652   // StartLine, EndLine - The starting and ending line numbers to print.
653   unsigned StartLine = 0, EndLine = 0;
654
655   if (SecondLineSpec.empty()) {    // No second line specifier provided?
656     // Handle special forms like "", "+", "-", etc.
657     std::string TmpSpec = FirstLineSpec;
658     std::string Tok = getToken(TmpSpec);
659     if (getToken(TmpSpec).empty() && (Tok == "" || Tok == "+" || Tok == "-")) {
660       if (Tok == "+" || Tok == "") {
661         StartLine = LineListedEnd;
662         EndLine = StartLine + ListSize;
663       } else {
664         assert(Tok == "-");
665         StartLine = LineListedStart-ListSize;
666         EndLine = LineListedStart;
667         if ((int)StartLine <= 0) StartLine = 1;
668       }
669     } else {
670       // Must be a normal line specifier.
671       const SourceFile *File;
672       unsigned LineNo;
673       parseLineSpec(FirstLineSpec, File, LineNo);
674
675       // If the user only specified one file specifier, we should display
676       // ListSize lines centered at the specified line.
677       if (File != 0) CurrentFile = File;
678       StartLine = LineNo - (ListSize+1)/2;
679       if ((int)StartLine <= 0) StartLine = 1;
680       EndLine = StartLine + ListSize;
681     }
682
683   } else {
684     // Parse two line specifiers... 
685     const SourceFile *StartFile, *EndFile;
686     unsigned StartLineNo, EndLineNo;
687     parseLineSpec(FirstLineSpec, StartFile, StartLineNo);
688     unsigned SavedLLE = LineListedEnd;
689     LineListedEnd = StartLineNo;
690     try {
691       parseLineSpec(SecondLineSpec, EndFile, EndLineNo);
692     } catch (...) {
693       LineListedEnd = SavedLLE;
694       throw;
695     }
696
697     // Inherit file specified by the first line spec if there was one.
698     if (EndFile == 0) EndFile = StartFile;
699
700     if (StartFile != EndFile)
701       throw "Start and end line specifiers are in different files!";
702     CurrentFile = StartFile;
703     StartLine = StartLineNo;
704     EndLine = EndLineNo+1;
705   }
706
707   assert((int)StartLine > 0 && (int)EndLine > 0 && StartLine <= EndLine &&
708          "Error reading line specifiers!");
709
710   // If there was no current file, and the user didn't specify one to list, we
711   // have an error.
712   if (CurrentFile == 0)
713     throw "There is no current file to list.";
714
715   // Remember for next time.
716   LineListedStart = StartLine;
717   LineListedEnd = StartLine;
718
719   for (unsigned LineNo = StartLine; LineNo != EndLine; ++LineNo) {
720     // Print the source line, unless it is invalid.
721     if (printSourceLine(LineNo))
722       break;
723     LineListedEnd = LineNo+1;
724   }
725
726   // If we didn't print any lines, find out why.
727   if (LineListedEnd == StartLine) {
728     // See if we can read line #0 from the file, if not, we couldn't load the
729     // file.
730     const char *LineStart, *LineEnd;
731     CurrentFile->getSourceLine(0, LineStart, LineEnd);
732     if (LineStart == 0)
733       throw "Could not load source file '" + CurrentFile->getFilename() + "'!";
734     else
735       std::cout << "<end of file>\n";
736   }
737 }
738
739 void CLIDebugger::setCommand(std::string &Options) {
740   std::string What = getToken(Options);
741
742   if (What.empty())
743     throw "set command expects at least two arguments.";
744   if (What == "args") {
745     parseProgramOptions(Options);
746   } else if (What == "language") {
747     std::string Lang = getToken(Options);
748     if (!getToken(Options).empty())
749       throw "set language expects one argument at most.";
750     if (Lang == "") {
751       std::cout << "The currently understood settings are:\n\n"
752                 << "local or auto  Automatic setting based on source file\n"
753                 << "c              Use the C language\n"
754                 << "c++            Use the C++ language\n"
755                 << "unknown        Use when source language is not supported\n";
756     } else if (Lang == "local" || Lang == "auto") {
757       CurrentLanguage = 0;
758     } else if (Lang == "c") {
759       CurrentLanguage = &SourceLanguage::getCFamilyInstance();
760     } else if (Lang == "c++") {
761       CurrentLanguage = &SourceLanguage::getCPlusPlusInstance();
762     } else if (Lang == "unknown") {
763       CurrentLanguage = &SourceLanguage::getUnknownLanguageInstance();
764     } else {
765       throw "Unknown language '" + Lang + "'.";
766     }
767
768   } else if (What == "listsize") {
769     ListSize = getUnsignedIntegerOption("'set prompt' command", Options);
770   } else if (What == "prompt") {
771     // Include any trailing whitespace or other tokens, but not leading
772     // whitespace.
773     Prompt = getToken(Options);  // Strip leading whitespace
774     Prompt += Options;           // Keep trailing whitespace or other stuff
775   } else {
776     // FIXME: Try to parse this as a source-language program expression.
777     throw "Don't know how to set '" + What + "'!";
778   }
779 }
780
781 void CLIDebugger::showCommand(std::string &Options) {
782   std::string What = getToken(Options);
783
784   if (What.empty() || !getToken(Options).empty())
785     throw "show command expects one argument.";
786
787   if (What == "args") {
788     std::cout << "Argument list to give program when started is \"";
789     // FIXME: This doesn't print stuff correctly if the arguments have spaces in
790     // them, but currently the only way to get that is to use the --args command
791     // line argument.  This should really handle escaping all hard characters as
792     // needed.
793     for (unsigned i = 0, e = Dbg.getNumProgramArguments(); i != e; ++i)
794       std::cout << (i ? " " : "") << Dbg.getProgramArgument(i);
795     std::cout << "\"\n";
796
797   } else if (What == "language") {
798     std::cout << "The current source language is '";
799     if (CurrentLanguage)
800       std::cout << CurrentLanguage->getSourceLanguageName();
801     else
802       std::cout << "auto; currently "
803                 << getCurrentLanguage().getSourceLanguageName();
804     std::cout << "'.\n";
805   } else if (What == "listsize") {
806     std::cout << "Number of source lines llvm-db will list by default is "
807               << ListSize << ".\n";
808   } else if (What == "prompt") {
809     std::cout << "llvm-db's prompt is \"" << Prompt << "\".\n";
810   } else {
811     throw "Unknown show command '" + What + "'.  Try 'help show'.";
812   }
813 }
814
815 void CLIDebugger::helpCommand(std::string &Options) {
816   // Print out all of the commands in the CommandTable
817   std::string Command = getToken(Options);
818   if (!getToken(Options).empty())
819     throw "help command takes at most one argument.";
820
821   // Getting detailed help on a particular command?
822   if (!Command.empty()) {
823     CLICommand *C = getCommand(Command);
824     std::cout << C->getShortHelp() << ".\n" << C->getLongHelp();
825
826     // If there are aliases for this option, print them out.
827     const std::vector<std::string> &Names = C->getOptionNames();
828     if (Names.size() > 1) {
829       std::cout << "The '" << Command << "' command is known as: '"
830                 << Names[0] << "'";
831       for (unsigned i = 1, e = Names.size(); i != e; ++i)
832         std::cout << ", '" << Names[i] << "'";
833       std::cout << "\n";
834     }
835
836   } else {
837     unsigned MaxSize = 0;
838     for (std::map<std::string, CLICommand*>::iterator I = CommandTable.begin(),
839            E = CommandTable.end(); I != E; ++I)
840       if (I->first.size() > MaxSize &&
841           I->first == I->second->getPrimaryOptionName())
842         MaxSize = I->first.size();
843
844     // Loop over all of the commands, printing the short help version
845     for (std::map<std::string, CLICommand*>::iterator I = CommandTable.begin(),
846            E = CommandTable.end(); I != E; ++I)
847       if (I->first == I->second->getPrimaryOptionName())
848         std::cout << I->first << std::string(MaxSize - I->first.size(), ' ')
849                   << " - " << I->second->getShortHelp() << "\n";
850   }
851 }