| //===-- Lua.cpp -----------------------------------------------------------===// |
| // |
| // 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 |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "Lua.h" |
| #include "lldb/Host/FileSystem.h" |
| #include "lldb/Utility/FileSpec.h" |
| #include "llvm/Support/FormatVariadic.h" |
| |
| using namespace lldb_private; |
| using namespace lldb; |
| |
| llvm::Error Lua::Run(llvm::StringRef buffer) { |
| int error = |
| luaL_loadbuffer(m_lua_state, buffer.data(), buffer.size(), "buffer") || |
| lua_pcall(m_lua_state, 0, 0, 0); |
| if (!error) |
| return llvm::Error::success(); |
| |
| llvm::Error e = llvm::make_error<llvm::StringError>( |
| llvm::formatv("{0}\n", lua_tostring(m_lua_state, -1)), |
| llvm::inconvertibleErrorCode()); |
| // Pop error message from the stack. |
| lua_pop(m_lua_state, 1); |
| return e; |
| } |
| |
| llvm::Error Lua::LoadModule(llvm::StringRef filename) { |
| FileSpec file(filename); |
| if (!FileSystem::Instance().Exists(file)) { |
| return llvm::make_error<llvm::StringError>("invalid path", |
| llvm::inconvertibleErrorCode()); |
| } |
| |
| ConstString module_extension = file.GetFileNameExtension(); |
| if (module_extension != ".lua") { |
| return llvm::make_error<llvm::StringError>("invalid extension", |
| llvm::inconvertibleErrorCode()); |
| } |
| |
| int error = luaL_loadfile(m_lua_state, filename.data()) || |
| lua_pcall(m_lua_state, 0, 1, 0); |
| if (error) { |
| llvm::Error e = llvm::make_error<llvm::StringError>( |
| llvm::formatv("{0}\n", lua_tostring(m_lua_state, -1)), |
| llvm::inconvertibleErrorCode()); |
| // Pop error message from the stack. |
| lua_pop(m_lua_state, 1); |
| return e; |
| } |
| |
| ConstString module_name = file.GetFileNameStrippingExtension(); |
| lua_setglobal(m_lua_state, module_name.GetCString()); |
| return llvm::Error::success(); |
| } |
| |
| llvm::Error Lua::ChangeIO(FILE *out, FILE *err) { |
| assert(out != nullptr); |
| assert(err != nullptr); |
| |
| lua_getglobal(m_lua_state, "io"); |
| |
| lua_getfield(m_lua_state, -1, "stdout"); |
| if (luaL_Stream *s = static_cast<luaL_Stream *>( |
| luaL_testudata(m_lua_state, -1, LUA_FILEHANDLE))) { |
| s->f = out; |
| lua_pop(m_lua_state, 1); |
| } else { |
| lua_pop(m_lua_state, 2); |
| return llvm::make_error<llvm::StringError>("could not get stdout", |
| llvm::inconvertibleErrorCode()); |
| } |
| |
| lua_getfield(m_lua_state, -1, "stderr"); |
| if (luaL_Stream *s = static_cast<luaL_Stream *>( |
| luaL_testudata(m_lua_state, -1, LUA_FILEHANDLE))) { |
| s->f = out; |
| lua_pop(m_lua_state, 1); |
| } else { |
| lua_pop(m_lua_state, 2); |
| return llvm::make_error<llvm::StringError>("could not get stderr", |
| llvm::inconvertibleErrorCode()); |
| } |
| |
| lua_pop(m_lua_state, 1); |
| return llvm::Error::success(); |
| } |