1 //===-- ClangUserExpression.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 // C Includes
11 #include <stdio.h>
12 #if HAVE_SYS_TYPES_H
13 #  include <sys/types.h>
14 #endif
15 
16 // C++ Includes
17 
18 #include "lldb/Core/ConstString.h"
19 #include "lldb/Core/Log.h"
20 #include "lldb/Core/Module.h"
21 #include "lldb/Core/Stream.h"
22 #include "lldb/Core/StreamFile.h"
23 #include "lldb/Expression/DiagnosticManager.h"
24 #include "lldb/Expression/ExpressionSourceCode.h"
25 #include "lldb/Expression/FunctionCaller.h"
26 #include "lldb/Expression/IRExecutionUnit.h"
27 #include "lldb/Expression/UtilityFunction.h"
28 #include "lldb/Host/Host.h"
29 #include "lldb/Target/ExecutionContext.h"
30 #include "lldb/Target/Process.h"
31 #include "lldb/Target/Target.h"
32 
33 using namespace lldb_private;
34 using namespace lldb;
35 
36 //------------------------------------------------------------------
37 /// Constructor
38 ///
39 /// @param[in] text
40 ///     The text of the function.  Must be a full translation unit.
41 ///
42 /// @param[in] name
43 ///     The name of the function, as used in the text.
44 //------------------------------------------------------------------
45 UtilityFunction::UtilityFunction (ExecutionContextScope &exe_scope,
46                                   const char *text,
47                                   const char *name) :
48     Expression (exe_scope),
49     m_execution_unit_sp (),
50     m_jit_module_wp (),
51     m_function_text (ExpressionSourceCode::g_expression_prefix),
52     m_function_name (name)
53 {
54     if (text && text[0])
55         m_function_text.append (text);
56 }
57 
58 UtilityFunction::~UtilityFunction ()
59 {
60     lldb::ProcessSP process_sp (m_jit_process_wp.lock());
61     if (process_sp)
62     {
63         lldb::ModuleSP jit_module_sp (m_jit_module_wp.lock());
64         if (jit_module_sp)
65             process_sp->GetTarget().GetImages().Remove(jit_module_sp);
66     }
67 
68 }
69 
70 // FIXME: We should check that every time this is called it is called with the same return type & arguments...
71 
72 FunctionCaller *
73 UtilityFunction::MakeFunctionCaller (const CompilerType &return_type, const ValueList &arg_value_list, lldb::ThreadSP thread_to_use_sp, Error &error)
74 {
75     if (m_caller_up)
76         return m_caller_up.get();
77 
78     ProcessSP process_sp = m_jit_process_wp.lock();
79     if (!process_sp)
80     {
81         error.SetErrorString("Can't make a function caller without a process.");
82         return nullptr;
83     }
84 
85     Address impl_code_address;
86     impl_code_address.SetOffset(StartAddress());
87     std::string name(m_function_name);
88     name.append("-caller");
89 
90     m_caller_up.reset (process_sp->GetTarget().GetFunctionCallerForLanguage (Language(),
91                                                                              return_type,
92                                                                              impl_code_address,
93                                                                              arg_value_list,
94                                                                              name.c_str(),
95                                                                              error));
96     if (error.Fail())
97     {
98 
99         return nullptr;
100     }
101     if (m_caller_up)
102     {
103         DiagnosticManager diagnostics;
104 
105         unsigned num_errors = m_caller_up->CompileFunction(thread_to_use_sp, diagnostics);
106         if (num_errors)
107         {
108             error.SetErrorStringWithFormat("Error compiling %s caller function: \"%s\".", m_function_name.c_str(),
109                                            diagnostics.GetString().c_str());
110             m_caller_up.reset();
111             return nullptr;
112         }
113 
114         diagnostics.Clear();
115         ExecutionContext exe_ctx(process_sp);
116 
117         if (!m_caller_up->WriteFunctionWrapper(exe_ctx, diagnostics))
118         {
119             error.SetErrorStringWithFormat("Error inserting caller function for %s: \"%s\".", m_function_name.c_str(),
120                                            diagnostics.GetString().c_str());
121             m_caller_up.reset();
122             return nullptr;
123         }
124     }
125     return m_caller_up.get();
126 }
127