add unveil to list of exported functions

Author: n0la

CMakeLists.txt

@@ -1,4 +1,5 @@
 CMAKE_MINIMUM_REQUIRED(VERSION 2.8)
+PROJECT(lua-openbsd)
 
 FIND_PACKAGE(PkgConfig)
 INCLUDE(CheckFunctionExists)
@@ -13,6 +14,7 @@ ADD_DEFINITIONS("-Wall -Werror")
 CHECK_FUNCTION_EXISTS("pledge" HAVE_PLEDGE)
 CHECK_FUNCTION_EXISTS("arc4random" HAVE_ARC4RANDOM)
 CHECK_FUNCTION_EXISTS("arc4random_uniform" HAVE_ARC4RANDOM_UNIFORM)
+CHECK_FUNCTION_EXISTS("unveil" HAVE_UNVEIL)
 CONFIGURE_FILE("${CMAKE_SOURCE_DIR}/config.h.in"
   "${CMAKE_BINARY_DIR}/config.h")
 

README

@@ -5,6 +5,7 @@ Implements:
  o pledge()
  o arc4random()
  o arc4random_uniform()
+ o unveil()
 
 Works and has been tested on Lua 5.1, 5.2 and 5.3.
 

config.h.in

@@ -13,4 +13,8 @@
  */
 #cmakedefine HAVE_ARC4RANDOM_UNIFORM
 
+/* Do we have unveil()?
+ */
+#cmakedefine HAVE_UNVEIL
+
 #endif

example/openbsd-example.lua

@@ -1,13 +1,13 @@
 o = require("openbsd")
 
+-- Test out unveil
+ret = o.unveil(".", "rwx")
+print("unveil: " .. ret)
+
 -- Same as pledge("rpath", NULL)
 ret, s = o.pledge("rpath stdio")
 print(ret, s)
 
--- Same as pledge("rpath", { NULL })
-ret, s = o.pledge("rpath stdio", {})
-print(ret, s)
-
 -- Same as pledge("rpath stdio wpath", "rpath stdio")
 ret, s = o.pledge("rpath stdio wpath", "rpath stdio")
 

src/lua-openbsd.c

@@ -78,6 +78,29 @@ int lua_arc4random_uniform(lua_State *L)
     return 1;
 }
 
+int lua_unveil(lua_State *L)
+{
+    char const *path = NULL, *mode = NULL;
+    int ret = 0;
+
+#ifndef HAVE_UNVEIL
+    lo_die(L, "unveil: not supported");
+#endif
+
+    luaL_argcheck(L, lua_isstring(L, 1), 1,
+                  "unveil: first argument must be string");
+    luaL_argcheck(L, lua_isstring(L, 2), 1,
+                  "unveil: second argument must be string");
+
+    path = lua_tostring(L, 1);
+    mode = lua_tostring(L, 2);
+
+    ret = unveil(path, mode);
+    lua_pushnumber(L, ret);
+
+    return 1;
+}
+
 #if !defined LUA_VERSION_NUM || LUA_VERSION_NUM==501
 /* Adapted from Lua 5.2.0
  */
@@ -100,6 +123,7 @@ static const struct luaL_Reg l_pledge[] = {
     { "pledge", lua_pledge },
     { "arc4random", lua_arc4random },
     { "arc4random_uniform", lua_arc4random_uniform },
+    { "unveil", lua_unveil },
     { NULL, NULL },
 };