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"
201a8d4093SSean 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     {
58*3f4c09c1SCaroline Tice       //case 'l':
59*3f4c09c1SCaroline Tice       //if (language.SetLanguageFromCString (option_arg) == false)
60*3f4c09c1SCaroline Tice       //{
61*3f4c09c1SCaroline Tice       //    error.SetErrorStringWithFormat("Invalid language option argument '%s'.\n", option_arg);
62*3f4c09c1SCaroline Tice       //}
63*3f4c09c1SCaroline Tice       //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();
85*3f4c09c1SCaroline Tice     //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",
101*3f4c09c1SCaroline Tice             "Evaluate an Objective-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 {
1936961e878SSean Callanan     if (!m_exe_ctx.process)
1946961e878SSean Callanan     {
19591b9383bSJim Ingham         error_stream.Printf ("Execution context doesn't contain a process\n");
1966961e878SSean Callanan         return false;
1976961e878SSean Callanan     }
1986961e878SSean Callanan 
1996961e878SSean Callanan     if (!m_exe_ctx.process->GetDynamicCheckers())
2006961e878SSean Callanan     {
2016961e878SSean Callanan         DynamicCheckerFunctions *dynamic_checkers = new DynamicCheckerFunctions();
2026961e878SSean Callanan 
2036961e878SSean Callanan         StreamString install_errors;
2046961e878SSean Callanan 
2056961e878SSean Callanan         if (!dynamic_checkers->Install(install_errors, m_exe_ctx))
2066961e878SSean Callanan         {
20791b9383bSJim Ingham             error_stream.Printf("Couldn't install dynamic checkers into the execution context: %s\n", install_errors.GetData());
2086961e878SSean Callanan             return false;
2096961e878SSean Callanan         }
2106961e878SSean Callanan 
2116961e878SSean Callanan         m_exe_ctx.process->SetDynamicCheckers(dynamic_checkers);
2126961e878SSean Callanan     }
2136961e878SSean Callanan 
2141a8d4093SSean Callanan     ClangUserExpression user_expression (expr);
21516ad5faeSSean Callanan 
2161a8d4093SSean Callanan     if (!user_expression.Parse (error_stream, m_exe_ctx))
21716ad5faeSSean Callanan     {
21891b9383bSJim Ingham         error_stream.Printf ("Couldn't parse the expresssion\n");
21916ad5faeSSean Callanan         return false;
22016ad5faeSSean Callanan     }
22116ad5faeSSean Callanan 
2221a8d4093SSean Callanan     ClangExpressionVariable *expr_result;
22316ad5faeSSean Callanan 
2241a8d4093SSean Callanan     if (!user_expression.Execute (error_stream, m_exe_ctx, expr_result))
22530fdc8d8SChris Lattner     {
22691b9383bSJim Ingham         error_stream.Printf ("Couldn't execute the expresssion\n");
22716ad5faeSSean Callanan         return false;
22816ad5faeSSean Callanan     }
22930fdc8d8SChris Lattner 
230d1e5b439SSean Callanan     if (expr_result)
23130fdc8d8SChris Lattner     {
232d1e5b439SSean Callanan         StreamString ss;
23316ad5faeSSean Callanan 
234fcd43b71SJohnny Chen         Error rc = expr_result->Print (ss,
235d1e5b439SSean Callanan                                        m_exe_ctx,
236d1e5b439SSean Callanan                                        m_options.format,
237d1e5b439SSean Callanan                                        m_options.show_types,
238d1e5b439SSean Callanan                                        m_options.show_summary,
239d1e5b439SSean Callanan                                        m_options.debug);
24016ad5faeSSean Callanan 
241fcd43b71SJohnny Chen         if (rc.Fail()) {
242fcd43b71SJohnny Chen             error_stream.Printf ("Couldn't print result : %s\n", rc.AsCString());
243fcd43b71SJohnny Chen             return false;
244fcd43b71SJohnny Chen         }
245fcd43b71SJohnny Chen 
246d1e5b439SSean Callanan         output_stream.PutCString(ss.GetString().c_str());
247fcd43b71SJohnny Chen         if (result)
248fcd43b71SJohnny Chen             result->SetStatus (eReturnStatusSuccessFinishResult);
24930fdc8d8SChris Lattner     }
25030fdc8d8SChris Lattner     else
25130fdc8d8SChris Lattner     {
252d1e5b439SSean Callanan         error_stream.Printf ("Expression produced no result\n");
253fcd43b71SJohnny Chen         if (result)
254fcd43b71SJohnny Chen             result->SetStatus (eReturnStatusSuccessFinishNoResult);
25530fdc8d8SChris Lattner     }
25630fdc8d8SChris Lattner 
25716ad5faeSSean Callanan     return true;
25830fdc8d8SChris Lattner }
25930fdc8d8SChris Lattner 
26030fdc8d8SChris Lattner bool
26130fdc8d8SChris Lattner CommandObjectExpression::ExecuteRawCommandString
26230fdc8d8SChris Lattner (
2636611103cSGreg Clayton     CommandInterpreter &interpreter,
26430fdc8d8SChris Lattner     const char *command,
26530fdc8d8SChris Lattner     CommandReturnObject &result
26630fdc8d8SChris Lattner )
26730fdc8d8SChris Lattner {
26816ad5faeSSean Callanan     m_exe_ctx = interpreter.GetDebugger().GetExecutionContext();
26930fdc8d8SChris Lattner 
27030fdc8d8SChris Lattner     m_options.ResetOptionValues();
27130fdc8d8SChris Lattner 
27230fdc8d8SChris Lattner     const char * expr = NULL;
27330fdc8d8SChris Lattner 
27430fdc8d8SChris Lattner     if (command[0] == '\0')
27530fdc8d8SChris Lattner     {
27630fdc8d8SChris Lattner         m_expr_lines.clear();
27730fdc8d8SChris Lattner         m_expr_line_count = 0;
27830fdc8d8SChris Lattner 
2796611103cSGreg Clayton         InputReaderSP reader_sp (new InputReader(interpreter.GetDebugger()));
28030fdc8d8SChris Lattner         if (reader_sp)
28130fdc8d8SChris Lattner         {
28230fdc8d8SChris Lattner             Error err (reader_sp->Initialize (CommandObjectExpression::MultiLineExpressionCallback,
28330fdc8d8SChris Lattner                                               this,                         // baton
28430fdc8d8SChris Lattner                                               eInputReaderGranularityLine,  // token size, to pass to callback function
28530fdc8d8SChris Lattner                                               NULL,                         // end token
28630fdc8d8SChris Lattner                                               NULL,                         // prompt
28730fdc8d8SChris Lattner                                               true));                       // echo input
28830fdc8d8SChris Lattner             if (err.Success())
28930fdc8d8SChris Lattner             {
2906611103cSGreg Clayton                 interpreter.GetDebugger().PushInputReader (reader_sp);
29130fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
29230fdc8d8SChris Lattner             }
29330fdc8d8SChris Lattner             else
29430fdc8d8SChris Lattner             {
29530fdc8d8SChris Lattner                 result.AppendError (err.AsCString());
29630fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
29730fdc8d8SChris Lattner             }
29830fdc8d8SChris Lattner         }
29930fdc8d8SChris Lattner         else
30030fdc8d8SChris Lattner         {
30130fdc8d8SChris Lattner             result.AppendError("out of memory");
30230fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
30330fdc8d8SChris Lattner         }
30430fdc8d8SChris Lattner         return result.Succeeded();
30530fdc8d8SChris Lattner     }
30630fdc8d8SChris Lattner 
30730fdc8d8SChris Lattner     if (command[0] == '-')
30830fdc8d8SChris Lattner     {
30930fdc8d8SChris Lattner         // We have some options and these options MUST end with --.
31030fdc8d8SChris Lattner         const char *end_options = NULL;
31130fdc8d8SChris Lattner         const char *s = command;
31230fdc8d8SChris Lattner         while (s && s[0])
31330fdc8d8SChris Lattner         {
31430fdc8d8SChris Lattner             end_options = ::strstr (s, "--");
31530fdc8d8SChris Lattner             if (end_options)
31630fdc8d8SChris Lattner             {
31730fdc8d8SChris Lattner                 end_options += 2; // Get past the "--"
31830fdc8d8SChris Lattner                 if (::isspace (end_options[0]))
31930fdc8d8SChris Lattner                 {
32030fdc8d8SChris Lattner                     expr = end_options;
32130fdc8d8SChris Lattner                     while (::isspace (*expr))
32230fdc8d8SChris Lattner                         ++expr;
32330fdc8d8SChris Lattner                     break;
32430fdc8d8SChris Lattner                 }
32530fdc8d8SChris Lattner             }
32630fdc8d8SChris Lattner             s = end_options;
32730fdc8d8SChris Lattner         }
32830fdc8d8SChris Lattner 
32930fdc8d8SChris Lattner         if (end_options)
33030fdc8d8SChris Lattner         {
33130fdc8d8SChris Lattner             Args args (command, end_options - command);
3326611103cSGreg Clayton             if (!ParseOptions (interpreter, args, result))
33330fdc8d8SChris Lattner                 return false;
33430fdc8d8SChris Lattner         }
33530fdc8d8SChris Lattner     }
33630fdc8d8SChris Lattner 
33730fdc8d8SChris Lattner     if (expr == NULL)
33830fdc8d8SChris Lattner         expr = command;
33930fdc8d8SChris Lattner 
340fcd43b71SJohnny Chen     if (EvaluateExpression (expr, false, result.GetOutputStream(), result.GetErrorStream(), &result))
341fcd43b71SJohnny Chen         return true;
342fcd43b71SJohnny Chen 
343fcd43b71SJohnny Chen     result.SetStatus (eReturnStatusFailed);
344fcd43b71SJohnny Chen     return false;
34530fdc8d8SChris Lattner }
34630fdc8d8SChris Lattner 
34730fdc8d8SChris Lattner lldb::OptionDefinition
34830fdc8d8SChris Lattner CommandObjectExpression::CommandOptions::g_option_table[] =
34930fdc8d8SChris Lattner {
350*3f4c09c1SCaroline Tice   //{ LLDB_OPT_SET_ALL, false, "language",   'l', required_argument, NULL, 0, "[c|c++|objc|objc++]",          "Sets the language to use when parsing the expression."},
3511d389c4bSSean 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."},
3521d389c4bSSean Callanan { LLDB_OPT_SET_ALL, false, "debug",      'g', no_argument,       NULL, 0, NULL,                           "Enable verbose debug logging of the expression parsing and evaluation."},
3531d389c4bSSean 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."},
35430fdc8d8SChris Lattner { 0, false, NULL, 0, 0, NULL, NULL, NULL, NULL }
35530fdc8d8SChris Lattner };
35630fdc8d8SChris Lattner 
357