flm01/mote/v2/openwrt/package/luci/libs/nixio/src/fs.c

567 lines
13 KiB
C
Raw Normal View History

/*
* nixio - Linux I/O library for lua
*
* Copyright (C) 2009 Steven Barth <steven@midlink.org>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "nixio.h"
#include <libgen.h>
#include <string.h>
#include <unistd.h>
#include <limits.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/time.h>
#include <dirent.h>
/* Reads argument from given index and transforms it into a mode bitfield */
int nixio__check_mode(lua_State *L, int idx, int def) {
if (lua_isnoneornil(L, idx) && def > 0) {
return def;
} else if (lua_isstring(L, idx) && lua_objlen(L, idx) == 9) {
int mode = 0;
const char *modestr = lua_tostring(L, idx);
int i;
for (i=0; i<9; i++) {
if (i % 3 == 0) { /* read flags */
if (modestr[i] == 'r') {
mode |= 1 << (8 - i);
} else if (modestr[i] != '-') {
break;
}
} else if (i % 3 == 1) { /* write flags */
if (modestr[i] == 'w') {
mode |= 1 << (8 - i);
} else if (modestr[i] != '-') {
break;
}
} else if (i == 2) {
if (modestr[i] == 'x') {
mode |= 00100;
} else if (modestr[i] == 's') {
mode |= 04100;
} else if (modestr[i] == 'S') {
mode |= 04000;
} else if (modestr[i] != '-') {
break;
}
} else if (i == 5) {
if (modestr[i] == 'x') {
mode |= 00010;
} else if (modestr[i] == 's') {
mode |= 02010;
} else if (modestr[i] == 'S') {
mode |= 02000;
} else if (modestr[i] != '-') {
break;
}
} else if (i == 8) {
if (modestr[i] == 'x') {
mode |= 00001;
} else if (modestr[i] == 't') {
mode |= 01001;
} else if (modestr[i] == 'T') {
mode |= 01000;
} else if (modestr[i] != '-') {
break;
}
}
}
if (i == 9) { /* successfully parsed */
return mode;
}
} else if (lua_isnumber(L, idx)) {
int decmode = lua_tointeger(L, idx);
int s = (decmode % 10000) / 1000;
int u = (decmode % 1000) / 100;
int g = (decmode % 100) / 10;
int o = (decmode % 10);
if (s>=0 && s<=7 && u>=0 && u<=7 && g>=0 && g<=7 && o>=0 && o<=7) {
return (s << 9) + (u << 6) + (g << 3) + o;
}
}
return luaL_argerror(L, idx, "supported values: [0-7]?[0-7][0-7][0-7], "
"[-r][-w][-xsS][-r][-w][-xsS][-r][-w][-xtT]");
}
/* Transforms a mode into the modestring */
int nixio__mode_write(int mode, char *modestr) {
if (modestr) {
modestr[0] = (mode & 00400) ? 'r' : '-';
modestr[1] = (mode & 00200) ? 'w' : '-';
modestr[2] = ((mode & 04100) == 04100) ? 's' :
(mode & 04000) ? 'S' : (mode & 00100) ? 'x' : '-';
modestr[3] = (mode & 00040) ? 'r' : '-';
modestr[4] = (mode & 00020) ? 'w' : '-';
modestr[5] = ((mode & 02010) == 02010) ? 's' :
(mode & 02000) ? 'S' : (mode & 00010) ? 'x' : '-';
modestr[6] = (mode & 00004) ? 'r' : '-';
modestr[7] = (mode & 00002) ? 'w' : '-';
modestr[8] = ((mode & 01001) == 01001) ? 't' :
(mode & 01000) ? 'T' : (mode & 00001) ? 'x' : '-';
}
return (mode & 00007) + ((mode & 00070) >> 3) * 10 +
((mode & 00700) >> 6) * 100 + ((mode & 07000) >> 9) * 1000;
}
static int nixio_access(lua_State *L) {
const char *path = luaL_checkstring(L, 1);
int mode = F_OK;
for (const char *s = luaL_optstring(L, 2, "f"); *s; s++) {
if (*s == 'r') {
mode |= R_OK;
} else if (*s == 'w') {
mode |= W_OK;
} else if (*s == 'x') {
mode |= X_OK;
} else if (*s != 'f') {
return luaL_argerror(L, 2, "supported values: [frwx]");
}
}
return nixio__pstatus(L, !access(path, mode));
}
static int nixio_basename(lua_State *L) {
const char *path = luaL_checkstring(L, 1);
char base[PATH_MAX];
base[PATH_MAX-1] = 0;
strncpy(base, path, PATH_MAX-1);
lua_pushstring(L, basename(base));
return 1;
}
static int nixio_dirname(lua_State *L) {
const char *path = luaL_checkstring(L, 1);
char base[PATH_MAX];
base[PATH_MAX-1] = 0;
strncpy(base, path, PATH_MAX-1);
lua_pushstring(L, dirname(base));
return 1;
}
static int nixio_realpath(lua_State *L) {
const char *path = luaL_checkstring(L, 1);
char real[PATH_MAX];
if (!realpath(path, real)) {
return nixio__perror(L);
} else {
lua_pushstring(L, real);
return 1;
}
}
static int nixio_remove(lua_State *L) {
return nixio__pstatus(L, !remove(luaL_checkstring(L, 1)));
}
static int nixio_unlink(lua_State *L) {
return nixio__pstatus(L, !unlink(luaL_checkstring(L, 1)));
}
static int nixio_rename(lua_State *L) {
return nixio__pstatus(L,
!rename(luaL_checkstring(L, 1), luaL_checkstring(L, 2)));
}
static int nixio_rmdir(lua_State *L) {
return nixio__pstatus(L, !rmdir(luaL_checkstring(L, 1)));
}
static int nixio_mkdir(lua_State *L) {
return nixio__pstatus(L,
!mkdir(luaL_checkstring(L, 1), nixio__check_mode(L, 2, 0777)));
}
static int nixio_chmod(lua_State *L) {
return nixio__pstatus(L,
!chmod(luaL_checkstring(L, 1), nixio__check_mode(L, 2, -1)));
}
static int nixio_dir__gc(lua_State *L) {
DIR **dirp = lua_touserdata(L, 1);
if (dirp && *dirp) {
closedir(*dirp);
*dirp = NULL;
}
return 0;
}
static int nixio_dir__iter(lua_State *L) {
DIR **dirp = lua_touserdata(L, lua_upvalueindex(1));
struct dirent *entry;
const char *n = NULL;
if (*dirp) {
do {
entry = readdir(*dirp);
n = (entry) ? entry->d_name : NULL;
} while(n && n[0] == '.' && (n[1] == 0 || (n[1] == '.' && n[2] == 0)));
}
if (n) {
lua_pushstring(L, n);
} else {
if (*dirp) {
closedir(*dirp);
*dirp = NULL;
}
lua_pushnil(L);
}
return 1;
}
static int nixio_dir(lua_State *L) {
const char *path = luaL_optstring(L, 1, ".");
DIR **dirp = lua_newuserdata(L, sizeof(DIR *));
*dirp = opendir(path);
if (!*dirp) {
return nixio__perror(L);
} else {
luaL_getmetatable(L, NIXIO_DIR_META);
lua_setmetatable(L, -2);
lua_pushcclosure(L, nixio_dir__iter, 1);
return 1;
}
}
static int nixio_link(lua_State *L) {
return nixio__pstatus(L,
!link(luaL_checkstring(L, 1), luaL_checkstring(L, 2)));
}
static int nixio_utimes(lua_State *L) {
const char *path = luaL_checkstring(L, 1);
if (lua_gettop(L) < 2 || (lua_isnoneornil(L, 2) && lua_isnoneornil(L, 3))) {
return nixio__pstatus(L, !utimes(path, NULL));
} else {
double atime = nixio__checknumber(L, 2);
double mtime = nixio__optnumber(L, 3, atime);
struct timeval times[2];
times[0].tv_sec = atime;
times[0].tv_usec = 0;
times[1].tv_sec = mtime;
times[1].tv_usec = 0;
return nixio__pstatus(L, !utimes(path, times));
}
}
int nixio__push_stat(lua_State *L, nixio_stat_t *buf) {
lua_createtable(L, 0, 15);
lua_pushinteger(L, buf->st_dev);
lua_setfield(L, -2, "dev");
lua_pushinteger(L, buf->st_ino);
lua_setfield(L, -2, "ino");
if (S_ISREG(buf->st_mode)) {
lua_pushliteral(L, "reg");
} else if (S_ISDIR(buf->st_mode)) {
lua_pushliteral(L, "dir");
} else if (S_ISCHR(buf->st_mode)) {
lua_pushliteral(L, "chr");
} else if (S_ISBLK(buf->st_mode)) {
lua_pushliteral(L, "blk");
} else if (S_ISFIFO(buf->st_mode)) {
lua_pushliteral(L, "fifo");
} else if (S_ISLNK(buf->st_mode)) {
lua_pushliteral(L, "lnk");
} else if (S_ISSOCK(buf->st_mode)) {
lua_pushliteral(L, "sock");
} else {
lua_pushliteral(L, "unknown");
}
lua_setfield(L, -2, "type");
char modestr[9];
lua_pushinteger(L, nixio__mode_write(buf->st_mode, modestr));
lua_setfield(L, -2, "modedec");
lua_pushlstring(L, modestr, 9);
lua_setfield(L, -2, "modestr");
lua_pushinteger(L, buf->st_nlink);
lua_setfield(L, -2, "nlink");
lua_pushinteger(L, buf->st_uid);
lua_setfield(L, -2, "uid");
lua_pushinteger(L, buf->st_gid);
lua_setfield(L, -2, "gid");
lua_pushinteger(L, buf->st_rdev);
lua_setfield(L, -2, "rdev");
nixio__pushnumber(L, buf->st_size);
lua_setfield(L, -2, "size");
lua_pushinteger(L, buf->st_atime);
lua_setfield(L, -2, "atime");
lua_pushinteger(L, buf->st_mtime);
lua_setfield(L, -2, "mtime");
lua_pushinteger(L, buf->st_ctime);
lua_setfield(L, -2, "ctime");
#ifndef __WINNT__
lua_pushinteger(L, buf->st_blksize);
lua_setfield(L, -2, "blksize");
lua_pushinteger(L, buf->st_blocks);
lua_setfield(L, -2, "blocks");
#endif
return 1;
}
static int nixio_stat(lua_State *L) {
nixio_stat_t buf;
if (stat(luaL_checkstring(L, 1), &buf)) {
return nixio__perror(L);
} else {
nixio__push_stat(L, &buf);
if (lua_isstring(L, 2)) {
lua_getfield(L, -1, lua_tostring(L, 2));
}
return 1;
}
}
static int nixio_lstat(lua_State *L) {
nixio_stat_t buf;
if (stat(luaL_checkstring(L, 1), &buf)) {
return nixio__perror(L);
} else {
nixio__push_stat(L, &buf);
if (lua_isstring(L, 2)) {
lua_getfield(L, -1, lua_tostring(L, 2));
}
return 1;
}
}
#ifndef __WINNT__
static int nixio_chown(lua_State *L) {
return nixio__pstatus(L,
!chown(
luaL_checkstring(L, 1),
lua_isnoneornil(L, 2) ? -1 : nixio__check_user(L, 2),
lua_isnoneornil(L, 3) ? -1 : nixio__check_group(L, 3)
)
);
}
static int nixio_lchown(lua_State *L) {
return nixio__pstatus(L,
!lchown(
luaL_checkstring(L, 1),
lua_isnoneornil(L, 2) ? -1 : nixio__check_user(L, 2),
lua_isnoneornil(L, 3) ? -1 : nixio__check_group(L, 3)
)
);
}
static int nixio_mkfifo(lua_State *L) {
return nixio__pstatus(L,
!mkfifo(luaL_checkstring(L, 1), nixio__check_mode(L, 2, -1)));
}
static int nixio_symlink(lua_State *L) {
return nixio__pstatus(L,
!symlink(luaL_checkstring(L, 1), luaL_checkstring(L, 2)));
}
static int nixio_readlink(lua_State *L) {
char dest[PATH_MAX];
ssize_t res = readlink(luaL_checkstring(L, 1), dest, sizeof(dest));
if (res < 0) {
return nixio__perror(L);
} else {
lua_pushlstring(L, dest, res);
return 1;
}
}
#include <glob.h>
typedef struct {
glob_t gl;
size_t pos;
int freed;
} nixio_glob_t;
static int nixio_glob__iter(lua_State *L) {
nixio_glob_t *globres = lua_touserdata(L, lua_upvalueindex(1));
if (!globres->freed && globres->pos < globres->gl.gl_pathc) {
lua_pushstring(L, globres->gl.gl_pathv[(globres->pos)++]);
} else {
if (!globres->freed) {
globfree(&globres->gl);
globres->freed = 1;
}
lua_pushnil(L);
}
return 1;
}
static int nixio_glob__gc(lua_State *L) {
nixio_glob_t *globres = lua_touserdata(L, 1);
if (globres && !globres->freed) {
globres->freed = 1;
globfree(&globres->gl);
}
return 0;
}
static int nixio_glob(lua_State *L) {
const char *pattern = luaL_optstring(L, 1, "*");
nixio_glob_t *globres = lua_newuserdata(L, sizeof(nixio_glob_t));
if (!globres) {
return luaL_error(L, NIXIO_OOM);
}
globres->pos = 0;
globres->freed = 0;
int globstat = glob(pattern, 0, NULL, &globres->gl);
if (globstat == GLOB_NOMATCH) {
lua_pushcfunction(L, nixio__nulliter);
lua_pushinteger(L, 0);
} else if (globstat) {
return nixio__perror(L);
} else {
luaL_getmetatable(L, NIXIO_GLOB_META);
lua_setmetatable(L, -2);
lua_pushcclosure(L, nixio_glob__iter, 1);
lua_pushinteger(L, globres->gl.gl_pathc);
}
return 2;
}
#include <sys/statvfs.h>
static int nixio__push_statvfs(lua_State *L, struct statvfs *buf) {
lua_createtable(L, 0, 12);
nixio__pushnumber(L, buf->f_bavail);
lua_setfield(L, -2, "bavail");
nixio__pushnumber(L, buf->f_bfree);
lua_setfield(L, -2, "bfree");
nixio__pushnumber(L, buf->f_blocks);
lua_setfield(L, -2, "blocks");
nixio__pushnumber(L, buf->f_bsize);
lua_setfield(L, -2, "bsize");
nixio__pushnumber(L, buf->f_frsize);
lua_setfield(L, -2, "frsize");
nixio__pushnumber(L, buf->f_favail);
lua_setfield(L, -2, "favail");
nixio__pushnumber(L, buf->f_ffree);
lua_setfield(L, -2, "ffree");
nixio__pushnumber(L, buf->f_files);
lua_setfield(L, -2, "files");
nixio__pushnumber(L, buf->f_flag);
lua_setfield(L, -2, "flag");
nixio__pushnumber(L, buf->f_fsid);
lua_setfield(L, -2, "fsid");
nixio__pushnumber(L, buf->f_namemax);
lua_setfield(L, -2, "namemax");
return 1;
}
static int nixio_statvfs(lua_State *L) {
struct statvfs buf;
if (statvfs(luaL_optstring(L, 1, "."), &buf)) {
return nixio__perror(L);
} else {
return nixio__push_statvfs(L, &buf);
}
}
#endif /* !__WINNT__ */
/* module table */
static const luaL_reg R[] = {
#ifndef __WINNT__
{"glob", nixio_glob},
{"mkfifo", nixio_mkfifo},
{"symlink", nixio_symlink},
{"readlink", nixio_readlink},
{"chown", nixio_chown},
{"lchown", nixio_lchown},
{"statvfs", nixio_statvfs},
#endif
{"chmod", nixio_chmod},
{"access", nixio_access},
{"basename", nixio_basename},
{"dir", nixio_dir},
{"dirname", nixio_dirname},
{"realpath", nixio_realpath},
{"mkdir", nixio_mkdir},
{"rmdir", nixio_rmdir},
{"link", nixio_link},
{"unlink", nixio_unlink},
{"utimes", nixio_utimes},
{"rename", nixio_rename},
{"remove", nixio_remove},
{"stat", nixio_stat},
{"lstat", nixio_lstat},
{NULL, NULL}
};
void nixio_open_fs(lua_State *L) {
lua_newtable(L);
luaL_register(L, NULL, R);
lua_setfield(L, -2, "fs");
luaL_newmetatable(L, NIXIO_DIR_META);
lua_pushcfunction(L, nixio_dir__gc);
lua_setfield(L, -2, "__gc");
lua_pop(L, 1);
#ifndef __WINNT__
luaL_newmetatable(L, NIXIO_GLOB_META);
lua_pushcfunction(L, nixio_glob__gc);
lua_setfield(L, -2, "__gc");
lua_pop(L, 1);
#endif
}