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 #include <sys/stat.h>
29
30 #include <errno.h>
31 #include <grp.h>
32 #include <pwd.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
lua_chmod(lua_State * L)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
lua_chown(lua_State * L)67 lua_chown(lua_State *L)
68 {
69 int n;
70 const char *path;
71 uid_t owner = (uid_t) -1;
72 gid_t group = (gid_t) -1;
73
74 n = lua_gettop(L);
75 luaL_argcheck(L, n > 1, n,
76 "chown takes at least two arguments");
77 path = luaL_checkstring(L, 1);
78 if (lua_isinteger(L, 2))
79 owner = (uid_t) lua_tointeger(L, 2);
80 else if (lua_isstring(L, 2)) {
81 struct passwd *p = getpwnam(lua_tostring(L, 2));
82 if (p != NULL)
83 owner = p->pw_uid;
84 else
85 return (luaL_argerror(L, 2,
86 lua_pushfstring(L, "unknown user %s",
87 lua_tostring(L, 2))));
88 } else if (!lua_isnoneornil(L, 2)) {
89 const char *type = luaL_typename(L, 2);
90 return (luaL_argerror(L, 2,
91 lua_pushfstring(L, "integer or string expected, got %s",
92 type)));
93 }
94
95 if (lua_isinteger(L, 3))
96 group = (gid_t) lua_tointeger(L, 3);
97 else if (lua_isstring(L, 3)) {
98 struct group *g = getgrnam(lua_tostring(L, 3));
99 if (g != NULL)
100 group = g->gr_gid;
101 else
102 return (luaL_argerror(L, 3,
103 lua_pushfstring(L, "unknown group %s",
104 lua_tostring(L, 3))));
105 } else if (!lua_isnoneornil(L, 3)) {
106 const char *type = luaL_typename(L, 3);
107 return (luaL_argerror(L, 3,
108 lua_pushfstring(L, "integer or string expected, got %s",
109 type)));
110 }
111
112 if (chown(path, owner, group) == -1) {
113 lua_pushnil(L);
114 lua_pushstring(L, strerror(errno));
115 lua_pushinteger(L, errno);
116 return (3);
117 }
118 lua_pushinteger(L, 0);
119 return (1);
120 }
121
122 static int
lua_getpid(lua_State * L)123 lua_getpid(lua_State *L)
124 {
125 int n;
126
127 n = lua_gettop(L);
128 luaL_argcheck(L, n == 0, 1, "too many arguments");
129 lua_pushinteger(L, getpid());
130 return 1;
131 }
132
133 #define REG_SIMPLE(n) { #n, lua_ ## n }
134 static const struct luaL_Reg sys_statlib[] = {
135 REG_SIMPLE(chmod),
136 { NULL, NULL },
137 };
138
139 static const struct luaL_Reg unistdlib[] = {
140 REG_SIMPLE(getpid),
141 REG_SIMPLE(chown),
142 { NULL, NULL },
143 };
144 #undef REG_SIMPLE
145
146 int
luaopen_posix_sys_stat(lua_State * L)147 luaopen_posix_sys_stat(lua_State *L)
148 {
149 luaL_newlib(L, sys_statlib);
150 return 1;
151 }
152
153 int
luaopen_posix_unistd(lua_State * L)154 luaopen_posix_unistd(lua_State *L)
155 {
156 luaL_newlib(L, unistdlib);
157 return 1;
158 }
159