1 /*-
2 * Copyright (c) 2019 Kyle Evans <[email protected]>
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/stat.h>
28 #include <sys/utsname.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 static int
lua_uname(lua_State * L)134 lua_uname(lua_State *L)
135 {
136 struct utsname name;
137 int error, n;
138
139 n = lua_gettop(L);
140 luaL_argcheck(L, n == 0, 1, "too many arguments");
141
142 error = uname(&name);
143 if (error != 0) {
144 error = errno;
145 lua_pushnil(L);
146 lua_pushstring(L, strerror(error));
147 lua_pushinteger(L, error);
148 return (3);
149 }
150
151 lua_newtable(L);
152 #define setkv(f) do { \
153 lua_pushstring(L, name.f); \
154 lua_setfield(L, -2, #f); \
155 } while (0)
156 setkv(sysname);
157 setkv(nodename);
158 setkv(release);
159 setkv(version);
160 setkv(machine);
161 #undef setkv
162
163 return (1);
164 }
165
166 #define REG_SIMPLE(n) { #n, lua_ ## n }
167 static const struct luaL_Reg sys_statlib[] = {
168 REG_SIMPLE(chmod),
169 { NULL, NULL },
170 };
171
172 static const struct luaL_Reg sys_utsnamelib[] = {
173 REG_SIMPLE(uname),
174 { NULL, NULL },
175 };
176
177 static const struct luaL_Reg unistdlib[] = {
178 REG_SIMPLE(getpid),
179 REG_SIMPLE(chown),
180 { NULL, NULL },
181 };
182 #undef REG_SIMPLE
183
184 int
luaopen_posix_sys_stat(lua_State * L)185 luaopen_posix_sys_stat(lua_State *L)
186 {
187 luaL_newlib(L, sys_statlib);
188 return 1;
189 }
190
191 int
luaopen_posix_sys_utsname(lua_State * L)192 luaopen_posix_sys_utsname(lua_State *L)
193 {
194 luaL_newlib(L, sys_utsnamelib);
195 return 1;
196 }
197
198 int
luaopen_posix_unistd(lua_State * L)199 luaopen_posix_unistd(lua_State *L)
200 {
201 luaL_newlib(L, unistdlib);
202 return 1;
203 }
204