Lua.cpp revision 357095 
1//===-- Lua.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 "Lua.h"
10#include "lldb/Host/FileSystem.h"
11#include "lldb/Utility/FileSpec.h"
12#include "llvm/Support/FormatVariadic.h"
13
14using namespace lldb_private;
15using namespace lldb;
16
17llvm::Error Lua::Run(llvm::StringRef buffer) {
18  int error =
19      luaL_loadbuffer(m_lua_state, buffer.data(), buffer.size(), "buffer") ||
20      lua_pcall(m_lua_state, 0, 0, 0);
21  if (!error)
22    return llvm::Error::success();
23
24  llvm::Error e = llvm::make_error<llvm::StringError>(
25      llvm::formatv("{0}\n", lua_tostring(m_lua_state, -1)),
26      llvm::inconvertibleErrorCode());
27  // Pop error message from the stack.
28  lua_pop(m_lua_state, 1);
29  return e;
30}
31
32llvm::Error Lua::LoadModule(llvm::StringRef filename) {
33  FileSpec file(filename);
34  if (!FileSystem::Instance().Exists(file)) {
35    return llvm::make_error<llvm::StringError>("invalid path",
36                                               llvm::inconvertibleErrorCode());
37  }
38
39  ConstString module_extension = file.GetFileNameExtension();
40  if (module_extension != ".lua") {
41    return llvm::make_error<llvm::StringError>("invalid extension",
42                                               llvm::inconvertibleErrorCode());
43  }
44
45  int error = luaL_loadfile(m_lua_state, filename.data()) ||
46              lua_pcall(m_lua_state, 0, 1, 0);
47  if (error) {
48    llvm::Error e = llvm::make_error<llvm::StringError>(
49        llvm::formatv("{0}\n", lua_tostring(m_lua_state, -1)),
50        llvm::inconvertibleErrorCode());
51    // Pop error message from the stack.
52    lua_pop(m_lua_state, 1);
53    return e;
54  }
55
56  ConstString module_name = file.GetFileNameStrippingExtension();
57  lua_setglobal(m_lua_state, module_name.GetCString());
58  return llvm::Error::success();
59}
60