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
44static int
45lua_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
66static int
67lua_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
122static int
123lua_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 }
134static const struct luaL_Reg sys_statlib[] = {
135	REG_SIMPLE(chmod),
136	{ NULL, NULL },
137};
138
139static const struct luaL_Reg unistdlib[] = {
140	REG_SIMPLE(getpid),
141	REG_SIMPLE(chown),
142	{ NULL, NULL },
143};
144#undef REG_SIMPLE
145
146int
147luaopen_posix_sys_stat(lua_State *L)
148{
149	luaL_newlib(L, sys_statlib);
150	return 1;
151}
152
153int
154luaopen_posix_unistd(lua_State *L)
155{
156	luaL_newlib(L, unistdlib);
157	return 1;
158}
159