Merge pull request #515 from vmg/vmg/static-lua

Lua tooling improvements

examples/lua/bashreadline.lua

+#!/usr/bin/env bcc-lua
 --[[
 Copyright 2016 GitHub, Inc
 

examples/lua/memleak.lua

+#!/usr/bin/env bcc-lua
 --[[
 Copyright 2016 GitHub, Inc
 

examples/lua/offcputime.lua

+#!/usr/bin/env bcc-lua
 --[[
 Copyright 2016 GitHub, Inc
 

examples/lua/strlen_count.lua

+#!/usr/bin/env bcc-lua
 --[[
 Copyright 2016 GitHub, Inc
 

examples/lua/task_switch.lua

+#!/usr/bin/env bcc-lua
 --[[
 Copyright 2016 GitHub, Inc
 

src/lua/CMakeLists.txt

@@ -2,7 +2,9 @@ find_package(LuaJIT)
 find_program(LUAJIT luajit)
 
 if (LUAJIT_LIBRARIES AND LUAJIT)
-	FILE(GLOB_RECURSE SRC_LUA ${CMAKE_CURRENT_SOURCE_DIR}/bcc/*/*.lua)
+	FILE(GLOB_RECURSE SRC_LUA
+		${CMAKE_CURRENT_SOURCE_DIR}/bcc/*.lua
+		${CMAKE_CURRENT_SOURCE_DIR}/bcc/vendor/*.lua)
 
 	ADD_CUSTOM_COMMAND(
 		OUTPUT bcc.lua
@@ -18,7 +20,9 @@ if (LUAJIT_LIBRARIES AND LUAJIT)
 
 	include_directories(${LUAJIT_INCLUDE_DIR})
 	add_executable(bcc-lua src/main.c bcc.o)
+	set_target_properties(bcc-lua PROPERTIES LINKER_LANGUAGE C)
 	target_link_libraries(bcc-lua ${LUAJIT_LIBRARIES})
+	target_link_libraries(bcc-lua -Wl,--whole-archive bcc-static -Wl,--no-whole-archive)
 
 	install(TARGETS bcc-lua RUNTIME DESTINATION bin)
 endif()

src/lua/bcc-probe

@@ -17,5 +17,4 @@ limitations under the License.
 local str = require("debug").getinfo(1, "S").source:sub(2)
 local script_path = str:match("(.*/)").."/?.lua;"
 package.path = script_path..package.path
-rawset(_G, "BCC_STANDALONE_NAME", "bcc-probe")
 require("bcc.run")()

src/lua/bcc/bpf.lua

@@ -41,7 +41,8 @@ function Bpf.static.cleanup_probes()
   local function detach_all(probe_type, all_probes)
     for key, probe in pairs(all_probes) do
       libbcc.perf_reader_free(probe)
-      if type(key) == "string" then
+      -- skip bcc-specific kprobes
+      if not key:starts("bcc:") then
         local desc = string.format("-:%s/%s", probe_type, key)
         log.info("detaching %s", desc)
 

src/lua/bcc/libbcc.lua

@@ -112,4 +112,11 @@ int bcc_symcache_resolve(void *symcache, uint64_t addr, struct bcc_symbol *sym);
 void bcc_symcache_refresh(void *resolver);
 ]]
 
-return ffi.load(os.getenv("LIBBCC_SO_PATH") or rawget(_G, "LIBBCC_SO_PATH") or "bcc")
+if rawget(_G, "BCC_STANDALONE") then
+  return ffi.C
+else
+  return ffi.load(
+    os.getenv("LIBBCC_SO_PATH") or
+    rawget(_G, "LIBBCC_SO_PATH") or
+    "bcc")
+end

src/lua/bcc/run.lua

@@ -16,11 +16,12 @@ limitations under the License.
 
 return function()
   require("bcc.vendor.helpers")
-  local progname = rawget(_G, "BCC_STANDALONE_NAME") or "bcc-lua"
+  local standalone = rawget(_G, "BCC_STANDALONE")
+  local progname = standalone or "bcc-probe"
 
   local function print_usage()
     io.stderr:write(string.format(
-      "usage: %s [[--so-path=PATH|--version|--verbose] --] path_to_script.lua [...]\n",
+      "usage: %s [[--version|--verbose] --] path_to_script.lua [...]\n",
       progname))
     os.exit(1)
   end
@@ -37,7 +38,7 @@ return function()
     local k = table.remove(arg, 1)
     if k == "--" then
       break
-    elseif string.starts(k, "--so-path=") then
+    elseif standalone == nil and string.starts(k, "--so-path=") then
       rawset(_G, "LIBBCC_SO_PATH", string.lstrip(k, "--so-path="))
     elseif k == "--llvm-debug" then
       rawset(_G, "LIBBCC_LLVM_DEBUG", 1)

src/lua/bcc/table.lua

@@ -236,7 +236,7 @@ function PerfEventArray:initialize(bpf, map_id, map_fd, key_type, leaf_type)
 end
 
 local function _perf_id(id, cpu)
-  return string.format("perf_event_array:%d:%d", tonumber(id), cpu or 0)
+  return string.format("bcc:perf_event_array:%d:%d", tonumber(id), cpu or 0)
 end
 
 function PerfEventArray:_open_perf_buffer(cpu, callback, ctype)

src/lua/src/main.c

@@ -29,46 +29,43 @@
 #include <string.h>
 #include <unistd.h>
 
-#include "lua.h"
 #include "lauxlib.h"
+#include "lua.h"
 #include "lualib.h"
 
 static lua_State *globalL = NULL;
 static const char *progname = NULL;
 
-static void lstop(lua_State *L, lua_Debug *ar)
-{
+static void lstop(lua_State *L, lua_Debug *ar) {
   (void)ar; /* unused arg. */
   lua_sethook(L, NULL, 0, 0);
   luaL_error(L, "interrupted!");
 }
 
-static void laction(int i)
-{
+static void laction(int i) {
   signal(i, SIG_DFL);
   lua_sethook(globalL, lstop, LUA_MASKCALL | LUA_MASKRET | LUA_MASKCOUNT, 1);
 }
 
-static void l_message(const char *pname, const char *msg)
-{
-	if (pname) fprintf(stderr, "%s: ", pname);
+static void l_message(const char *pname, const char *msg) {
+  if (pname)
+    fprintf(stderr, "%s: ", pname);
   fprintf(stderr, "%s\n", msg);
   fflush(stderr);
 }
 
-static int report(lua_State *L, int status)
-{
+static int report(lua_State *L, int status) {
   if (status && !lua_isnil(L, -1)) {
     const char *msg = lua_tostring(L, -1);
-		if (msg == NULL) msg = "(error object is not a string)";
+    if (msg == NULL)
+      msg = "(error object is not a string)";
     l_message(progname, msg);
     lua_pop(L, 1);
   }
   return status;
 }
 
-static int traceback(lua_State *L)
-{
+static int traceback(lua_State *L) {
   if (!lua_isstring(L, 1)) /* 'message' not a string? */
     return 1;              /* keep it intact */
   lua_getfield(L, LUA_GLOBALSINDEX, "debug");
@@ -87,8 +84,7 @@ static int traceback(lua_State *L)
   return 1;
 }
 
-static int docall(lua_State *L, int narg, int clear)
-{
+static int docall(lua_State *L, int narg, int clear) {
   int status;
   int base = lua_gettop(L) - narg; /* function index */
   lua_pushcfunction(L, traceback); /* push traceback function */
@@ -98,12 +94,12 @@ static int docall(lua_State *L, int narg, int clear)
   signal(SIGINT, SIG_DFL);
   lua_remove(L, base); /* remove traceback function */
   /* force a complete garbage collection in case of errors */
-	if (status != 0) lua_gc(L, LUA_GCCOLLECT, 0);
+  if (status != 0)
+    lua_gc(L, LUA_GCCOLLECT, 0);
   return status;
 }
 
-static int dolibrary(lua_State *L, const char *name, int clear)
-{
+static int dolibrary(lua_State *L, const char *name, int clear) {
   lua_getglobal(L, "require");
   lua_pushstring(L, name);
   return report(L, docall(L, 1, clear));
@@ -115,8 +111,7 @@ struct Smain {
   int status;
 };
 
-static void pushargv(lua_State *L, char **argv, int argc, int offset)
-{
+static void pushargv(lua_State *L, char **argv, int argc, int offset) {
   int i, j;
   lua_createtable(L, argc, 0);
   for (i = offset, j = 1; i < argc; i++, j++) {
@@ -125,43 +120,20 @@ static void pushargv(lua_State *L, char **argv, int argc, int offset)
   }
 }
 
-static void find_local_libbcc(lua_State *L)
-{
-	char buffer[4096];
-	char *dirname;
-
-	if (readlink("/proc/self/exe", buffer, sizeof(buffer)) < 0)
-		return;
-
-	dirname = strrchr(buffer, '/');
-	if (dirname == NULL)
-		return;
-
-	strcpy(dirname + 1, "libbcc.so");
-
-	if (access(buffer, F_OK|R_OK|X_OK) != 0)
-		return;
-
-	lua_pushstring(L, buffer);
-	lua_setglobal(L, "LIBBCC_SO_PATH");
-}
-
-static int pmain(lua_State *L)
-{
+static int pmain(lua_State *L) {
   struct Smain *s = (struct Smain *)lua_touserdata(L, 1);
   globalL = L;
 
   lua_gc(L, LUA_GCSTOP, 0);
   luaL_openlibs(L);
   lua_gc(L, LUA_GCRESTART, 0);
-	find_local_libbcc(L);
 
   s->status = dolibrary(L, "bcc", 0);
   if (s->status)
     return 0;
 
   lua_pushstring(L, progname);
-	lua_setglobal(L, "BCC_STANDALONE_NAME");
+  lua_setglobal(L, "BCC_STANDALONE");
 
   pushargv(L, s->argv, s->argc, 1);
   lua_setglobal(L, "arg");
@@ -170,8 +142,7 @@ static int pmain(lua_State *L)
   return 0;
 }
 
-int main(int argc, char **argv)
-{
+int main(int argc, char **argv) {
   int status;
   struct Smain s;
   lua_State *L = lua_open(); /* create state */
@@ -181,6 +152,11 @@ int main(int argc, char **argv)
     return EXIT_FAILURE;
   }
 
+  if (geteuid() != 0) {
+    l_message(argv[0], "bcc-lua must be ran as root");
+    return EXIT_FAILURE;
+  }
+
   progname = argv[0];
   s.argc = argc;
   s.argv = argv;

src/python/bcc/table.py

@@ -372,7 +372,7 @@ class PerfEventArray(ArrayBase):
         super(PerfEventArray, self).__init__(*args, **kwargs)
 
     def __delitem__(self, key):
-        super(PerfEventArray, self).__init__(key)
+        super(PerfEventArray, self).__delitem__(key)
         self.close_perf_buffer(key)
 
     def open_perf_buffer(self, callback):