libexec/flua/modules/lposix.c

   1 /*-
   2  * Copyright (c) 2019 Kyle Evans <kevans@FreeBSD.org>
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  14  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  16  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  17  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  18  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  19  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  20  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  21  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  22  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  23  * SUCH DAMAGE.
  24  *
  25  */
  26
  27 #include <sys/cdefs.h>
  28 __FBSDID("$FreeBSD$");
  29
  30 #include <sys/stat.h>
  31
  32 #include <errno.h>
  33 #include <string.h>
  34 #include <unistd.h>
  35
  36 #include <lua.h>
  37 #include "lauxlib.h"
  38 #include "lposix.h"
  39
  40 /*
  41  * Minimal implementation of luaposix needed for internal FreeBSD bits.
  42  */
  43
  44 static int
  45 lua_chmod(lua_State *L)
  46 {
  47         int n;
  48         const char *path;
  49         mode_t mode;
  50
  51         n = lua_gettop(L);
  52         luaL_argcheck(L, n == 2, n > 2 ? 3 : n,
  53             "chmod takes exactly two arguments");
  54         path = luaL_checkstring(L, 1);
  55         mode = (mode_t)luaL_checkinteger(L, 2);
  56         if (chmod(path, mode) == -1) {
  57                 lua_pushnil(L);
  58                 lua_pushstring(L, strerror(errno));
  59                 lua_pushinteger(L, errno);
  60                 return 3;
  61         }
  62         lua_pushinteger(L, 0);
  63         return 1;
  64 }
  65
  66 static int
  67 lua_getpid(lua_State *L)
  68 {
  69         int n;
  70
  71         n = lua_gettop(L);
  72         luaL_argcheck(L, n == 0, 1, "too many arguments");
  73         lua_pushinteger(L, getpid());
  74         return 1;
  75 }
  76
  77 #define REG_SIMPLE(n)   { #n, lua_ ## n }
  78 static const struct luaL_Reg sys_statlib[] = {
  79         REG_SIMPLE(chmod),
  80         { NULL, NULL },
  81 };
  82
  83 static const struct luaL_Reg unistdlib[] = {
  84         REG_SIMPLE(getpid),
  85         { NULL, NULL },
  86 };
  87 #undef REG_SIMPLE
  88
  89 int
  90 luaopen_posix_sys_stat(lua_State *L)
  91 {
  92         luaL_newlib(L, sys_statlib);
  93         return 1;
  94 }
  95
  96 int
  97 luaopen_posix_unistd(lua_State *L)
  98 {
  99         luaL_newlib(L, unistdlib);
 100         return 1;
 101 }