UtilityFunction.cpp revision 309124 
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
33using namespace lldb_private;
34using 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//------------------------------------------------------------------
45UtilityFunction::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
58UtilityFunction::~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
72FunctionCaller *
73UtilityFunction::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