1 //===-- ScriptedPythonInterface.cpp ---------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "lldb/Host/Config.h"
10 #include "lldb/Utility/Log.h"
11 #include "lldb/Utility/Logging.h"
12 #include "lldb/lldb-enumerations.h"
13 
14 #if LLDB_ENABLE_PYTHON
15 
16 // LLDB Python header must be included first
17 #include "lldb-python.h"
18 
19 #include "SWIGPythonBridge.h"
20 #include "ScriptInterpreterPythonImpl.h"
21 #include "ScriptedPythonInterface.h"
22 
23 using namespace lldb;
24 using namespace lldb_private;
25 
26 ScriptedPythonInterface::ScriptedPythonInterface(
27     ScriptInterpreterPythonImpl &interpreter)
28     : ScriptedInterface(), m_interpreter(interpreter) {}
29 
30 Status
31 ScriptedPythonInterface::GetStatusFromMethod(llvm::StringRef method_name) {
32   Status error;
33   Dispatch<Status>(method_name, error);
34 
35   return error;
36 }
37 
38 template <>
39 StructuredData::DictionarySP
40 ScriptedPythonInterface::ExtractValueFromPythonObject<
41     StructuredData::DictionarySP>(python::PythonObject &p, Status &error) {
42   python::PythonDictionary result_dict(python::PyRefType::Borrowed, p.get());
43   return result_dict.CreateStructuredDictionary();
44 }
45 
46 template <>
47 Status ScriptedPythonInterface::ExtractValueFromPythonObject<Status>(
48     python::PythonObject &p, Status &error) {
49   if (lldb::SBError *sb_error = reinterpret_cast<lldb::SBError *>(
50           LLDBSWIGPython_CastPyObjectToSBError(p.get())))
51     error = m_interpreter.GetStatusFromSBError(*sb_error);
52   else
53     error.SetErrorString("Couldn't cast lldb::SBError to lldb::Status.");
54 
55   return error;
56 }
57 
58 template <>
59 lldb::DataExtractorSP
60 ScriptedPythonInterface::ExtractValueFromPythonObject<lldb::DataExtractorSP>(
61     python::PythonObject &p, Status &error) {
62   lldb::SBData *sb_data = reinterpret_cast<lldb::SBData *>(
63       LLDBSWIGPython_CastPyObjectToSBData(p.get()));
64 
65   if (!sb_data) {
66     error.SetErrorString(
67         "Couldn't cast lldb::SBData to lldb::DataExtractorSP.");
68     return nullptr;
69   }
70 
71   return m_interpreter.GetDataExtractorFromSBData(*sb_data);
72 }
73 
74 template <>
75 llvm::Optional<MemoryRegionInfo>
76 ScriptedPythonInterface::ExtractValueFromPythonObject<
77     llvm::Optional<MemoryRegionInfo>>(python::PythonObject &p, Status &error) {
78 
79   lldb::SBMemoryRegionInfo *sb_mem_reg_info =
80       reinterpret_cast<lldb::SBMemoryRegionInfo *>(
81           LLDBSWIGPython_CastPyObjectToSBMemoryRegionInfo(p.get()));
82 
83   if (!sb_mem_reg_info) {
84     error.SetErrorString(
85         "Couldn't cast lldb::SBMemoryRegionInfo to lldb::MemoryRegionInfoSP.");
86     return {};
87   }
88 
89   return m_interpreter.GetOpaqueTypeFromSBMemoryRegionInfo(*sb_mem_reg_info);
90 }
91 
92 #endif
93