+++ /dev/null
-/*
- * 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
-}
projects / project / luci.git / blobdiff