bbef51eb43
It is surprisingly difficult to write a simple python script that
can reliably `import lldb` without failing, or crashing. I'm
currently resorting to convolutions like this:
def find_lldb(may_reexec=False):
if prefix := os.environ.get('LLDB_PYTHON_PREFIX'):
if os.path.realpath(prefix) != os.path.realpath(sys.prefix):
raise Exception("cannot import lldb.\n"
f" sys.prefix should be: {prefix}\n"
f" but it is: {sys.prefix}")
else:
line1, line2 = subprocess.run(
['lldb', '-x', '-b', '-o', 'script print(sys.prefix)'],
encoding='utf8', stdout=subprocess.PIPE,
check=True).stdout.strip().splitlines()
assert line1.strip() == '(lldb) script print(sys.prefix)'
prefix = line2.strip()
os.environ['LLDB_PYTHON_PREFIX'] = prefix
if sys.prefix != prefix:
if not may_reexec:
raise Exception(
"cannot import lldb.\n" +
f" This python, at {sys.prefix}\n"
f" does not math LLDB's python at {prefix}")
os.environ['LLDB_PYTHON_PREFIX'] = prefix
python_exe = os.path.join(prefix, 'bin', 'python3')
os.execl(python_exe, python_exe, *sys.argv)
lldb_path = subprocess.run(['lldb', '-P'],
check=True, stdout=subprocess.PIPE,
encoding='utf8').stdout.strip()
sys.path = [lldb_path] + sys.path
This patch aims to replace all that with:
#!/usr/bin/env lldb-python
import lldb
...
... by adding the following features:
* new command line option: --print-script-interpreter-info. This
prints language-specific information about the script interpreter
in JSON format.
* new tool (unix only): lldb-python which finds python and exec's it.
Reviewed By: JDevlieghere
Differential Revision: https://reviews.llvm.org/D112973
104 lines
2.8 KiB
C++
104 lines
2.8 KiB
C++
//===-- Driver.h ------------------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLDB_TOOLS_DRIVER_DRIVER_H
|
|
#define LLDB_TOOLS_DRIVER_DRIVER_H
|
|
|
|
#include "Platform.h"
|
|
|
|
#include "lldb/API/SBBroadcaster.h"
|
|
#include "lldb/API/SBDebugger.h"
|
|
#include "lldb/API/SBDefines.h"
|
|
#include "lldb/API/SBError.h"
|
|
|
|
#include "llvm/Option/Arg.h"
|
|
#include "llvm/Option/ArgList.h"
|
|
#include "llvm/Option/Option.h"
|
|
|
|
#include <set>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
class Driver : public lldb::SBBroadcaster {
|
|
public:
|
|
enum CommandPlacement {
|
|
eCommandPlacementBeforeFile,
|
|
eCommandPlacementAfterFile,
|
|
eCommandPlacementAfterCrash,
|
|
};
|
|
|
|
Driver();
|
|
|
|
virtual ~Driver();
|
|
|
|
/// Runs the main loop.
|
|
///
|
|
/// \return The exit code that the process should return.
|
|
int MainLoop();
|
|
|
|
lldb::SBError ProcessArgs(const llvm::opt::InputArgList &args, bool &exiting);
|
|
|
|
void WriteCommandsForSourcing(CommandPlacement placement,
|
|
lldb::SBStream &strm);
|
|
|
|
struct OptionData {
|
|
void AddInitialCommand(std::string command, CommandPlacement placement,
|
|
bool is_file, lldb::SBError &error);
|
|
|
|
struct InitialCmdEntry {
|
|
InitialCmdEntry(std::string contents, bool in_is_file,
|
|
bool in_quiet = false)
|
|
: contents(std::move(contents)), is_file(in_is_file),
|
|
source_quietly(in_quiet) {}
|
|
|
|
std::string contents;
|
|
bool is_file;
|
|
bool source_quietly;
|
|
};
|
|
|
|
std::vector<std::string> m_args;
|
|
|
|
lldb::LanguageType m_repl_lang = lldb::eLanguageTypeUnknown;
|
|
lldb::pid_t m_process_pid = LLDB_INVALID_PROCESS_ID;
|
|
|
|
std::string m_core_file;
|
|
std::string m_crash_log;
|
|
std::string m_repl_options;
|
|
std::string m_process_name;
|
|
|
|
std::vector<InitialCmdEntry> m_initial_commands;
|
|
std::vector<InitialCmdEntry> m_after_file_commands;
|
|
std::vector<InitialCmdEntry> m_after_crash_commands;
|
|
|
|
bool m_debug_mode = false;
|
|
bool m_source_quietly = false;
|
|
bool m_print_version = false;
|
|
bool m_print_python_path = false;
|
|
bool m_print_script_interpreter_info = false;
|
|
bool m_wait_for = false;
|
|
bool m_repl = false;
|
|
bool m_batch = false;
|
|
|
|
// FIXME: When we have set/show variables we can remove this from here.
|
|
bool m_use_external_editor = false;
|
|
|
|
using OptionSet = std::set<char>;
|
|
OptionSet m_seen_options;
|
|
};
|
|
|
|
lldb::SBDebugger &GetDebugger() { return m_debugger; }
|
|
|
|
void ResizeWindow(unsigned short col);
|
|
|
|
private:
|
|
lldb::SBDebugger m_debugger;
|
|
OptionData m_option_data;
|
|
};
|
|
|
|
#endif // LLDB_TOOLS_DRIVER_DRIVER_H
|