130fdc8d8SChris Lattner //===-- CommandObjectExpression.cpp -----------------------------*- C++ -*-===//
230fdc8d8SChris Lattner //
330fdc8d8SChris Lattner //                     The LLVM Compiler Infrastructure
430fdc8d8SChris Lattner //
530fdc8d8SChris Lattner // This file is distributed under the University of Illinois Open Source
630fdc8d8SChris Lattner // License. See LICENSE.TXT for details.
730fdc8d8SChris Lattner //
830fdc8d8SChris Lattner //===----------------------------------------------------------------------===//
930fdc8d8SChris Lattner 
1030fdc8d8SChris Lattner #include "CommandObjectExpression.h"
1130fdc8d8SChris Lattner 
1230fdc8d8SChris Lattner // C Includes
1330fdc8d8SChris Lattner // C++ Includes
1430fdc8d8SChris Lattner // Other libraries and framework includes
1530fdc8d8SChris Lattner // Project includes
16ebf7707eSSean Callanan #include "lldb/Interpreter/Args.h"
1730fdc8d8SChris Lattner #include "lldb/Core/Value.h"
1830fdc8d8SChris Lattner #include "lldb/Core/InputReader.h"
1930fdc8d8SChris Lattner #include "lldb/Expression/ClangExpressionVariable.h"
20*1a8d4093SSean Callanan #include "lldb/Expression/ClangUserExpression.h"
21ebb84b24SStephen Wilson #include "lldb/Expression/ClangFunction.h"
2230fdc8d8SChris Lattner #include "lldb/Expression/DWARFExpression.h"
2330fdc8d8SChris Lattner #include "lldb/Host/Host.h"
24ebf7707eSSean Callanan #include "lldb/Core/Debugger.h"
256611103cSGreg Clayton #include "lldb/Interpreter/CommandInterpreter.h"
2630fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
2730fdc8d8SChris Lattner #include "lldb/Symbol/ObjectFile.h"
2830fdc8d8SChris Lattner #include "lldb/Symbol/Variable.h"
2930fdc8d8SChris Lattner #include "lldb/Target/Process.h"
3030fdc8d8SChris Lattner #include "lldb/Target/StackFrame.h"
3130fdc8d8SChris Lattner #include "lldb/Target/Target.h"
32ebf7707eSSean Callanan #include "llvm/ADT/StringRef.h"
3330fdc8d8SChris Lattner 
3430fdc8d8SChris Lattner using namespace lldb;
3530fdc8d8SChris Lattner using namespace lldb_private;
3630fdc8d8SChris Lattner 
3730fdc8d8SChris Lattner CommandObjectExpression::CommandOptions::CommandOptions () :
3830fdc8d8SChris Lattner     Options()
3930fdc8d8SChris Lattner {
4030fdc8d8SChris Lattner     // Keep only one place to reset the values to their defaults
4130fdc8d8SChris Lattner     ResetOptionValues();
4230fdc8d8SChris Lattner }
4330fdc8d8SChris Lattner 
4430fdc8d8SChris Lattner 
4530fdc8d8SChris Lattner CommandObjectExpression::CommandOptions::~CommandOptions ()
4630fdc8d8SChris Lattner {
4730fdc8d8SChris Lattner }
4830fdc8d8SChris Lattner 
4930fdc8d8SChris Lattner Error
5030fdc8d8SChris Lattner CommandObjectExpression::CommandOptions::SetOptionValue (int option_idx, const char *option_arg)
5130fdc8d8SChris Lattner {
5230fdc8d8SChris Lattner     Error error;
5330fdc8d8SChris Lattner 
5430fdc8d8SChris Lattner     char short_option = (char) m_getopt_table[option_idx].val;
5530fdc8d8SChris Lattner 
5630fdc8d8SChris Lattner     switch (short_option)
5730fdc8d8SChris Lattner     {
5830fdc8d8SChris Lattner     case 'l':
5930fdc8d8SChris Lattner         if (language.SetLanguageFromCString (option_arg) == false)
6030fdc8d8SChris Lattner         {
6130fdc8d8SChris Lattner             error.SetErrorStringWithFormat("Invalid language option argument '%s'.\n", option_arg);
6230fdc8d8SChris Lattner         }
6330fdc8d8SChris Lattner         break;
6430fdc8d8SChris Lattner 
6530fdc8d8SChris Lattner     case 'g':
6630fdc8d8SChris Lattner         debug = true;
6730fdc8d8SChris Lattner         break;
6830fdc8d8SChris Lattner 
6930fdc8d8SChris Lattner     case 'f':
7030fdc8d8SChris Lattner         error = Args::StringToFormat(option_arg, format);
7130fdc8d8SChris Lattner         break;
7230fdc8d8SChris Lattner 
7330fdc8d8SChris Lattner     default:
7430fdc8d8SChris Lattner         error.SetErrorStringWithFormat("Invalid short option character '%c'.\n", short_option);
7530fdc8d8SChris Lattner         break;
7630fdc8d8SChris Lattner     }
7730fdc8d8SChris Lattner 
7830fdc8d8SChris Lattner     return error;
7930fdc8d8SChris Lattner }
8030fdc8d8SChris Lattner 
8130fdc8d8SChris Lattner void
8230fdc8d8SChris Lattner CommandObjectExpression::CommandOptions::ResetOptionValues ()
8330fdc8d8SChris Lattner {
8430fdc8d8SChris Lattner     Options::ResetOptionValues();
8530fdc8d8SChris Lattner     language.Clear();
8630fdc8d8SChris Lattner     debug = false;
8730fdc8d8SChris Lattner     format = eFormatDefault;
8830fdc8d8SChris Lattner     show_types = true;
8930fdc8d8SChris Lattner     show_summary = true;
9030fdc8d8SChris Lattner }
9130fdc8d8SChris Lattner 
9230fdc8d8SChris Lattner const lldb::OptionDefinition*
9330fdc8d8SChris Lattner CommandObjectExpression::CommandOptions::GetDefinitions ()
9430fdc8d8SChris Lattner {
9530fdc8d8SChris Lattner     return g_option_table;
9630fdc8d8SChris Lattner }
9730fdc8d8SChris Lattner 
9830fdc8d8SChris Lattner CommandObjectExpression::CommandObjectExpression () :
9930fdc8d8SChris Lattner     CommandObject (
10030fdc8d8SChris Lattner             "expression",
10130fdc8d8SChris Lattner             "Evaluate a C expression in the current program context, using variables currently in scope.",
10230fdc8d8SChris Lattner             "expression [<cmd-options>] <expr>"),
10330fdc8d8SChris Lattner     m_expr_line_count (0),
10430fdc8d8SChris Lattner     m_expr_lines ()
10530fdc8d8SChris Lattner {
10630fdc8d8SChris Lattner   SetHelpLong(
10730fdc8d8SChris Lattner "Examples: \n\
10830fdc8d8SChris Lattner \n\
10930fdc8d8SChris Lattner    expr my_struct->a = my_array[3] \n\
11030fdc8d8SChris Lattner    expr -f bin -- (index * 8) + 5 \n\
11130fdc8d8SChris Lattner    expr char c[] = \"foo\"; c[0]\n");
11230fdc8d8SChris Lattner }
11330fdc8d8SChris Lattner 
11430fdc8d8SChris Lattner CommandObjectExpression::~CommandObjectExpression ()
11530fdc8d8SChris Lattner {
11630fdc8d8SChris Lattner }
11730fdc8d8SChris Lattner 
11830fdc8d8SChris Lattner Options *
11930fdc8d8SChris Lattner CommandObjectExpression::GetOptions ()
12030fdc8d8SChris Lattner {
12130fdc8d8SChris Lattner     return &m_options;
12230fdc8d8SChris Lattner }
12330fdc8d8SChris Lattner 
12430fdc8d8SChris Lattner 
12530fdc8d8SChris Lattner bool
12630fdc8d8SChris Lattner CommandObjectExpression::Execute
12730fdc8d8SChris Lattner (
1286611103cSGreg Clayton     CommandInterpreter &interpreter,
12930fdc8d8SChris Lattner     Args& command,
13030fdc8d8SChris Lattner     CommandReturnObject &result
13130fdc8d8SChris Lattner )
13230fdc8d8SChris Lattner {
13330fdc8d8SChris Lattner     return false;
13430fdc8d8SChris Lattner }
13530fdc8d8SChris Lattner 
13630fdc8d8SChris Lattner 
13730fdc8d8SChris Lattner size_t
13830fdc8d8SChris Lattner CommandObjectExpression::MultiLineExpressionCallback
13930fdc8d8SChris Lattner (
14030fdc8d8SChris Lattner     void *baton,
1416611103cSGreg Clayton     InputReader &reader,
14230fdc8d8SChris Lattner     lldb::InputReaderAction notification,
14330fdc8d8SChris Lattner     const char *bytes,
14430fdc8d8SChris Lattner     size_t bytes_len
14530fdc8d8SChris Lattner )
14630fdc8d8SChris Lattner {
14730fdc8d8SChris Lattner     CommandObjectExpression *cmd_object_expr = (CommandObjectExpression *) baton;
14830fdc8d8SChris Lattner 
14930fdc8d8SChris Lattner     switch (notification)
15030fdc8d8SChris Lattner     {
15130fdc8d8SChris Lattner     case eInputReaderActivate:
1526611103cSGreg Clayton         reader.GetDebugger().GetOutputStream().Printf("%s\n", "Enter expressions, then terminate with an empty line to evaluate:");
15330fdc8d8SChris Lattner         // Fall through
15430fdc8d8SChris Lattner     case eInputReaderReactivate:
15530fdc8d8SChris Lattner         //if (out_fh)
1566611103cSGreg Clayton         //    reader.GetDebugger().GetOutputStream().Printf ("%3u: ", cmd_object_expr->m_expr_line_count);
15730fdc8d8SChris Lattner         break;
15830fdc8d8SChris Lattner 
15930fdc8d8SChris Lattner     case eInputReaderDeactivate:
16030fdc8d8SChris Lattner         break;
16130fdc8d8SChris Lattner 
16230fdc8d8SChris Lattner     case eInputReaderGotToken:
16330fdc8d8SChris Lattner         ++cmd_object_expr->m_expr_line_count;
16430fdc8d8SChris Lattner         if (bytes && bytes_len)
16530fdc8d8SChris Lattner         {
16630fdc8d8SChris Lattner             cmd_object_expr->m_expr_lines.append (bytes, bytes_len + 1);
16730fdc8d8SChris Lattner         }
16830fdc8d8SChris Lattner 
16930fdc8d8SChris Lattner         if (bytes_len == 0)
1706611103cSGreg Clayton             reader.SetIsDone(true);
17130fdc8d8SChris Lattner         //else if (out_fh && !reader->IsDone())
17230fdc8d8SChris Lattner         //    ::fprintf (out_fh, "%3u: ", cmd_object_expr->m_expr_line_count);
17330fdc8d8SChris Lattner         break;
17430fdc8d8SChris Lattner 
17530fdc8d8SChris Lattner     case eInputReaderDone:
17630fdc8d8SChris Lattner         {
17730fdc8d8SChris Lattner             bool bare = false;
17830fdc8d8SChris Lattner             cmd_object_expr->EvaluateExpression (cmd_object_expr->m_expr_lines.c_str(),
17930fdc8d8SChris Lattner                                                  bare,
1806611103cSGreg Clayton                                                  reader.GetDebugger().GetOutputStream(),
1816611103cSGreg Clayton                                                  reader.GetDebugger().GetErrorStream());
18230fdc8d8SChris Lattner         }
18330fdc8d8SChris Lattner         break;
18430fdc8d8SChris Lattner     }
18530fdc8d8SChris Lattner 
18630fdc8d8SChris Lattner     return bytes_len;
18730fdc8d8SChris Lattner }
18830fdc8d8SChris Lattner 
18930fdc8d8SChris Lattner bool
190fcd43b71SJohnny Chen CommandObjectExpression::EvaluateExpression (const char *expr, bool bare, Stream &output_stream, Stream &error_stream,
191fcd43b71SJohnny Chen                                              CommandReturnObject *result)
19230fdc8d8SChris Lattner {
193*1a8d4093SSean Callanan     ClangUserExpression user_expression (expr);
19416ad5faeSSean Callanan 
195*1a8d4093SSean Callanan     if (!user_expression.Parse (error_stream, m_exe_ctx))
19616ad5faeSSean Callanan     {
197*1a8d4093SSean Callanan         error_stream.Printf ("Couldn't parse the expresssion");
19816ad5faeSSean Callanan         return false;
19916ad5faeSSean Callanan     }
20016ad5faeSSean Callanan 
201*1a8d4093SSean Callanan     ClangExpressionVariable *expr_result;
20216ad5faeSSean Callanan 
203*1a8d4093SSean Callanan     if (!user_expression.Execute (error_stream, m_exe_ctx, expr_result))
20430fdc8d8SChris Lattner     {
205*1a8d4093SSean Callanan         error_stream.Printf ("Couldn't execute the expresssion");
20616ad5faeSSean Callanan         return false;
20716ad5faeSSean Callanan     }
20830fdc8d8SChris Lattner 
209d1e5b439SSean Callanan     if (expr_result)
21030fdc8d8SChris Lattner     {
211d1e5b439SSean Callanan         StreamString ss;
21216ad5faeSSean Callanan 
213fcd43b71SJohnny Chen         Error rc = expr_result->Print (ss,
214d1e5b439SSean Callanan                                        m_exe_ctx,
215d1e5b439SSean Callanan                                        m_options.format,
216d1e5b439SSean Callanan                                        m_options.show_types,
217d1e5b439SSean Callanan                                        m_options.show_summary,
218d1e5b439SSean Callanan                                        m_options.debug);
21916ad5faeSSean Callanan 
220fcd43b71SJohnny Chen         if (rc.Fail()) {
221fcd43b71SJohnny Chen             error_stream.Printf ("Couldn't print result : %s\n", rc.AsCString());
222fcd43b71SJohnny Chen             return false;
223fcd43b71SJohnny Chen         }
224fcd43b71SJohnny Chen 
225d1e5b439SSean Callanan         output_stream.PutCString(ss.GetString().c_str());
226fcd43b71SJohnny Chen         if (result)
227fcd43b71SJohnny Chen             result->SetStatus (eReturnStatusSuccessFinishResult);
22830fdc8d8SChris Lattner     }
22930fdc8d8SChris Lattner     else
23030fdc8d8SChris Lattner     {
231d1e5b439SSean Callanan         error_stream.Printf ("Expression produced no result\n");
232fcd43b71SJohnny Chen         if (result)
233fcd43b71SJohnny Chen             result->SetStatus (eReturnStatusSuccessFinishNoResult);
23430fdc8d8SChris Lattner     }
23530fdc8d8SChris Lattner 
23616ad5faeSSean Callanan     return true;
23730fdc8d8SChris Lattner }
23830fdc8d8SChris Lattner 
23930fdc8d8SChris Lattner bool
24030fdc8d8SChris Lattner CommandObjectExpression::ExecuteRawCommandString
24130fdc8d8SChris Lattner (
2426611103cSGreg Clayton     CommandInterpreter &interpreter,
24330fdc8d8SChris Lattner     const char *command,
24430fdc8d8SChris Lattner     CommandReturnObject &result
24530fdc8d8SChris Lattner )
24630fdc8d8SChris Lattner {
24716ad5faeSSean Callanan     m_exe_ctx = interpreter.GetDebugger().GetExecutionContext();
24830fdc8d8SChris Lattner 
24930fdc8d8SChris Lattner     m_options.ResetOptionValues();
25030fdc8d8SChris Lattner 
25130fdc8d8SChris Lattner     const char * expr = NULL;
25230fdc8d8SChris Lattner 
25330fdc8d8SChris Lattner     if (command[0] == '\0')
25430fdc8d8SChris Lattner     {
25530fdc8d8SChris Lattner         m_expr_lines.clear();
25630fdc8d8SChris Lattner         m_expr_line_count = 0;
25730fdc8d8SChris Lattner 
2586611103cSGreg Clayton         InputReaderSP reader_sp (new InputReader(interpreter.GetDebugger()));
25930fdc8d8SChris Lattner         if (reader_sp)
26030fdc8d8SChris Lattner         {
26130fdc8d8SChris Lattner             Error err (reader_sp->Initialize (CommandObjectExpression::MultiLineExpressionCallback,
26230fdc8d8SChris Lattner                                               this,                         // baton
26330fdc8d8SChris Lattner                                               eInputReaderGranularityLine,  // token size, to pass to callback function
26430fdc8d8SChris Lattner                                               NULL,                         // end token
26530fdc8d8SChris Lattner                                               NULL,                         // prompt
26630fdc8d8SChris Lattner                                               true));                       // echo input
26730fdc8d8SChris Lattner             if (err.Success())
26830fdc8d8SChris Lattner             {
2696611103cSGreg Clayton                 interpreter.GetDebugger().PushInputReader (reader_sp);
27030fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
27130fdc8d8SChris Lattner             }
27230fdc8d8SChris Lattner             else
27330fdc8d8SChris Lattner             {
27430fdc8d8SChris Lattner                 result.AppendError (err.AsCString());
27530fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
27630fdc8d8SChris Lattner             }
27730fdc8d8SChris Lattner         }
27830fdc8d8SChris Lattner         else
27930fdc8d8SChris Lattner         {
28030fdc8d8SChris Lattner             result.AppendError("out of memory");
28130fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
28230fdc8d8SChris Lattner         }
28330fdc8d8SChris Lattner         return result.Succeeded();
28430fdc8d8SChris Lattner     }
28530fdc8d8SChris Lattner 
28630fdc8d8SChris Lattner     if (command[0] == '-')
28730fdc8d8SChris Lattner     {
28830fdc8d8SChris Lattner         // We have some options and these options MUST end with --.
28930fdc8d8SChris Lattner         const char *end_options = NULL;
29030fdc8d8SChris Lattner         const char *s = command;
29130fdc8d8SChris Lattner         while (s && s[0])
29230fdc8d8SChris Lattner         {
29330fdc8d8SChris Lattner             end_options = ::strstr (s, "--");
29430fdc8d8SChris Lattner             if (end_options)
29530fdc8d8SChris Lattner             {
29630fdc8d8SChris Lattner                 end_options += 2; // Get past the "--"
29730fdc8d8SChris Lattner                 if (::isspace (end_options[0]))
29830fdc8d8SChris Lattner                 {
29930fdc8d8SChris Lattner                     expr = end_options;
30030fdc8d8SChris Lattner                     while (::isspace (*expr))
30130fdc8d8SChris Lattner                         ++expr;
30230fdc8d8SChris Lattner                     break;
30330fdc8d8SChris Lattner                 }
30430fdc8d8SChris Lattner             }
30530fdc8d8SChris Lattner             s = end_options;
30630fdc8d8SChris Lattner         }
30730fdc8d8SChris Lattner 
30830fdc8d8SChris Lattner         if (end_options)
30930fdc8d8SChris Lattner         {
31030fdc8d8SChris Lattner             Args args (command, end_options - command);
3116611103cSGreg Clayton             if (!ParseOptions (interpreter, args, result))
31230fdc8d8SChris Lattner                 return false;
31330fdc8d8SChris Lattner         }
31430fdc8d8SChris Lattner     }
31530fdc8d8SChris Lattner 
31630fdc8d8SChris Lattner     if (expr == NULL)
31730fdc8d8SChris Lattner         expr = command;
31830fdc8d8SChris Lattner 
319fcd43b71SJohnny Chen     if (EvaluateExpression (expr, false, result.GetOutputStream(), result.GetErrorStream(), &result))
320fcd43b71SJohnny Chen         return true;
321fcd43b71SJohnny Chen 
322fcd43b71SJohnny Chen     result.SetStatus (eReturnStatusFailed);
323fcd43b71SJohnny Chen     return false;
32430fdc8d8SChris Lattner }
32530fdc8d8SChris Lattner 
32630fdc8d8SChris Lattner lldb::OptionDefinition
32730fdc8d8SChris Lattner CommandObjectExpression::CommandOptions::g_option_table[] =
32830fdc8d8SChris Lattner {
3290c5cd90dSGreg Clayton { LLDB_OPT_SET_ALL, false, "language",   'l', required_argument, NULL, 0, "[c|c++|objc|objc++]",          "Sets the language to use when parsing the expression."},
3301d389c4bSSean Callanan { LLDB_OPT_SET_ALL, false, "format",     'f', required_argument, NULL, 0, "[ [bool|b] | [bin] | [char|c] | [oct|o] | [dec|i|d|u] | [hex|x] | [float|f] | [cstr|s] ]",  "Specify the format that the expression output should use."},
3311d389c4bSSean Callanan { LLDB_OPT_SET_ALL, false, "debug",      'g', no_argument,       NULL, 0, NULL,                           "Enable verbose debug logging of the expression parsing and evaluation."},
3321d389c4bSSean Callanan { LLDB_OPT_SET_ALL, false, "use-ir",     'i', no_argument,       NULL, 0, NULL,                           "[Temporary] Instructs the expression evaluator to use IR instead of ASTs."},
33330fdc8d8SChris Lattner { 0, false, NULL, 0, 0, NULL, NULL, NULL, NULL }
33430fdc8d8SChris Lattner };
33530fdc8d8SChris Lattner 
336