mirror of
https://github.com/mruby/mruby
synced 2026-06-08 16:11:16 +00:00
674 lines
19 KiB
C
674 lines
19 KiB
C
/*
|
|
** file.c - File class
|
|
*/
|
|
|
|
#include <mruby.h>
|
|
#include <mruby/class.h>
|
|
#include <mruby/data.h>
|
|
#include <mruby/string.h>
|
|
#include <mruby/ext/io.h>
|
|
#include <mruby/error.h>
|
|
#include <mruby/presym.h>
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
#include <windows.h>
|
|
#include <io.h>
|
|
#define NULL_FILE "NUL"
|
|
#define UNLINK _unlink
|
|
#define GETCWD _getcwd
|
|
#define CHMOD(a, b) 0
|
|
#define MAXPATHLEN 1024
|
|
#if !defined(PATH_MAX)
|
|
#define PATH_MAX _MAX_PATH
|
|
#endif
|
|
#define realpath(N,R) _fullpath((R),(N),_MAX_PATH)
|
|
#include <direct.h>
|
|
#else
|
|
#define NULL_FILE "/dev/null"
|
|
#include <unistd.h>
|
|
#define UNLINK unlink
|
|
#define GETCWD getcwd
|
|
#define CHMOD(a, b) chmod(a,b)
|
|
#include <sys/file.h>
|
|
#ifndef __DJGPP__
|
|
#include <libgen.h>
|
|
#endif
|
|
#include <sys/param.h>
|
|
#include <pwd.h>
|
|
#endif
|
|
|
|
#define FILE_SEPARATOR "/"
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
#define PATH_SEPARATOR ";"
|
|
#define FILE_ALT_SEPARATOR "\\"
|
|
#define VOLUME_SEPARATOR ":"
|
|
#else
|
|
#define PATH_SEPARATOR ":"
|
|
#endif
|
|
|
|
#ifndef LOCK_SH
|
|
#define LOCK_SH 1
|
|
#endif
|
|
#ifndef LOCK_EX
|
|
#define LOCK_EX 2
|
|
#endif
|
|
#ifndef LOCK_NB
|
|
#define LOCK_NB 4
|
|
#endif
|
|
#ifndef LOCK_UN
|
|
#define LOCK_UN 8
|
|
#endif
|
|
|
|
#if !defined(_WIN32) || defined(MRB_MINGW32_LEGACY)
|
|
typedef struct stat mrb_stat;
|
|
# define mrb_stat(path, sb) stat(path, sb)
|
|
# define mrb_fstat(fd, sb) fstat(fd, sb)
|
|
#elif defined MRB_INT32
|
|
typedef struct _stat32 mrb_stat;
|
|
# define mrb_stat(path, sb) _stat32(path, sb)
|
|
# define mrb_fstat(fd, sb) _fstat32(fd, sb)
|
|
#else
|
|
typedef struct _stat64 mrb_stat;
|
|
# define mrb_stat(path, sb) _stat64(path, sb)
|
|
# define mrb_fstat(fd, sb) _fstat64(fd, sb)
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
static int
|
|
flock(int fd, int operation)
|
|
{
|
|
HANDLE h = (HANDLE)_get_osfhandle(fd);
|
|
DWORD flags;
|
|
flags = ((operation & LOCK_NB) ? LOCKFILE_FAIL_IMMEDIATELY : 0)
|
|
| ((operation & LOCK_SH) ? LOCKFILE_EXCLUSIVE_LOCK : 0);
|
|
OVERLAPPED ov = (OVERLAPPED){0};
|
|
return LockFileEx(h, flags, 0, 0xffffffff, 0xffffffff, &ov) ? 0 : -1;
|
|
}
|
|
#endif
|
|
|
|
static mrb_value
|
|
mrb_file_s_umask(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
/* nothing to do on windows */
|
|
return mrb_fixnum_value(0);
|
|
|
|
#else
|
|
mrb_int mask, omask;
|
|
if (mrb_get_args(mrb, "|i", &mask) == 0) {
|
|
omask = umask(0);
|
|
umask(omask);
|
|
}
|
|
else {
|
|
omask = umask(mask);
|
|
}
|
|
return mrb_fixnum_value(omask);
|
|
#endif
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_s_unlink(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
const mrb_value *argv;
|
|
mrb_int argc;
|
|
|
|
mrb_get_args(mrb, "*", &argv, &argc);
|
|
for (int i = 0; i < argc; i++) {
|
|
mrb_value pathv = argv[i];
|
|
mrb_ensure_string_type(mrb, pathv);
|
|
const char *utf8_path = RSTRING_CSTR(mrb, pathv);
|
|
char *path = mrb_locale_from_utf8(utf8_path, -1);
|
|
if (UNLINK(path) < 0) {
|
|
mrb_locale_free(path);
|
|
mrb_sys_fail(mrb, utf8_path);
|
|
}
|
|
mrb_locale_free(path);
|
|
}
|
|
return mrb_fixnum_value(argc);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_s_rename(mrb_state *mrb, mrb_value obj)
|
|
{
|
|
mrb_value from, to;
|
|
|
|
mrb_get_args(mrb, "SS", &from, &to);
|
|
char *src = mrb_locale_from_utf8(RSTRING_CSTR(mrb, from), -1);
|
|
char *dst = mrb_locale_from_utf8(RSTRING_CSTR(mrb, to), -1);
|
|
if (rename(src, dst) < 0) {
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
if (CHMOD(dst, 0666) == 0 && UNLINK(dst) == 0 && rename(src, dst) == 0) {
|
|
mrb_locale_free(src);
|
|
mrb_locale_free(dst);
|
|
return mrb_fixnum_value(0);
|
|
}
|
|
#endif
|
|
mrb_locale_free(src);
|
|
mrb_locale_free(dst);
|
|
mrb_sys_fail(mrb, RSTRING_CSTR(mrb, mrb_format(mrb, "(%v, %v)", from, to)));
|
|
return mrb_fixnum_value(-1); /* not reached */
|
|
}
|
|
mrb_locale_free(src);
|
|
mrb_locale_free(dst);
|
|
return mrb_fixnum_value(0);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_dirname(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
char dname[_MAX_DIR], vname[_MAX_DRIVE];
|
|
char buffer[_MAX_DRIVE + _MAX_DIR];
|
|
const char *utf8_path;
|
|
mrb_get_args(mrb, "z", &utf8_path);
|
|
char *path = mrb_locale_from_utf8(utf8_path, -1);
|
|
_splitpath(path, vname, dname, NULL, NULL);
|
|
snprintf(buffer, _MAX_DRIVE + _MAX_DIR, "%s%s", vname, dname);
|
|
mrb_locale_free(path);
|
|
size_t ridx = strlen(buffer);
|
|
if (ridx == 0) {
|
|
strncpy(buffer, ".", 2); /* null terminated */
|
|
}
|
|
else if (ridx > 1) {
|
|
ridx--;
|
|
while (ridx > 0 && (buffer[ridx] == '/' || buffer[ridx] == '\\')) {
|
|
buffer[ridx] = '\0'; /* remove last char */
|
|
ridx--;
|
|
}
|
|
}
|
|
return mrb_str_new_cstr(mrb, buffer);
|
|
#else
|
|
mrb_value s;
|
|
mrb_get_args(mrb, "S", &s);
|
|
char *path = mrb_locale_from_utf8(mrb_str_to_cstr(mrb, s), -1);
|
|
char *dname;
|
|
if ((dname = dirname(path)) == NULL) {
|
|
mrb_locale_free(path);
|
|
mrb_sys_fail(mrb, "dirname");
|
|
}
|
|
mrb_locale_free(path);
|
|
return mrb_str_new_cstr(mrb, dname);
|
|
#endif
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_basename(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
// NOTE: Do not use mrb_locale_from_utf8 here
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
char bname[_MAX_DIR];
|
|
char extname[_MAX_EXT];
|
|
char buffer[_MAX_DIR + _MAX_EXT];
|
|
mrb_value s;
|
|
|
|
mrb_get_args(mrb, "S", &s);
|
|
char *path = mrb_str_to_cstr(mrb, s);
|
|
size_t ridx = strlen(path);
|
|
if (ridx > 0) {
|
|
ridx--;
|
|
while (ridx > 0 && (path[ridx] == '/' || path[ridx] == '\\')) {
|
|
path[ridx] = '\0';
|
|
ridx--;
|
|
}
|
|
if (strncmp(path, "/", 2) == 0) {
|
|
return mrb_str_new_cstr(mrb, path);
|
|
}
|
|
}
|
|
_splitpath((const char*)path, NULL, NULL, bname, extname);
|
|
snprintf(buffer, _MAX_DIR + _MAX_EXT, "%s%s", bname, extname);
|
|
return mrb_str_new_cstr(mrb, buffer);
|
|
#else
|
|
mrb_value s;
|
|
mrb_get_args(mrb, "S", &s);
|
|
char *path = mrb_str_to_cstr(mrb, s);
|
|
char *bname;
|
|
if ((bname = basename(path)) == NULL) {
|
|
mrb_sys_fail(mrb, "basename");
|
|
}
|
|
if (strncmp(bname, "//", 3) == 0) bname[1] = '\0'; /* patch for Cygwin */
|
|
return mrb_str_new_cstr(mrb, bname);
|
|
#endif
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_realpath(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
mrb_value pathname, dir_string;
|
|
mrb_int argc = mrb_get_args(mrb, "S|S", &pathname, &dir_string);
|
|
if (argc == 2) {
|
|
mrb_value s = mrb_str_dup(mrb, dir_string);
|
|
s = mrb_str_append(mrb, s, mrb_str_new_cstr(mrb, FILE_SEPARATOR));
|
|
s = mrb_str_append(mrb, s, pathname);
|
|
pathname = s;
|
|
}
|
|
char *cpath = mrb_locale_from_utf8(RSTRING_CSTR(mrb, pathname), -1);
|
|
mrb_value result = mrb_str_new_capa(mrb, PATH_MAX);
|
|
if (realpath(cpath, RSTRING_PTR(result)) == NULL) {
|
|
mrb_locale_free(cpath);
|
|
mrb_sys_fail(mrb, RSTRING_CSTR(mrb, pathname));
|
|
return result; /* not reached */
|
|
}
|
|
mrb_locale_free(cpath);
|
|
mrb_str_resize(mrb, result, strlen(RSTRING_PTR(result)));
|
|
return result;
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file__getwd(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
char buf[MAXPATHLEN];
|
|
|
|
mrb->c->ci->mid = 0;
|
|
if (GETCWD(buf, MAXPATHLEN) == NULL) {
|
|
mrb_sys_fail(mrb, "getcwd(2)");
|
|
}
|
|
char *utf8 = mrb_utf8_from_locale(buf, -1);
|
|
mrb_value path = mrb_str_new_cstr(mrb, utf8);
|
|
mrb_utf8_free(utf8);
|
|
return path;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
#define IS_FILESEP(x) (x == (*(char*)(FILE_SEPARATOR)) || x == (*(char*)(FILE_ALT_SEPARATOR)))
|
|
#define IS_VOLSEP(x) (x == (*(char*)(VOLUME_SEPARATOR)))
|
|
#define IS_DEVICEID(x) (x == '.' || x == '?')
|
|
#define CHECK_UNCDEV_PATH (IS_FILESEP(path[0]) && IS_FILESEP(path[1]))
|
|
|
|
static int
|
|
is_absolute_traditional_path(const char *path, size_t len)
|
|
{
|
|
if (len < 3) return 0;
|
|
return (ISALPHA(path[0]) && IS_VOLSEP(path[1]) && IS_FILESEP(path[2]));
|
|
}
|
|
|
|
static int
|
|
is_absolute_unc_path(const char *path, size_t len)
|
|
{
|
|
if (len < 2) return 0;
|
|
return (CHECK_UNCDEV_PATH && !IS_DEVICEID(path[2]));
|
|
}
|
|
|
|
static int
|
|
is_absolute_device_path(const char *path, size_t len)
|
|
{
|
|
if (len < 4) return 0;
|
|
return (CHECK_UNCDEV_PATH && IS_DEVICEID(path[2]) && IS_FILESEP(path[3]));
|
|
}
|
|
|
|
static int
|
|
mrb_file_is_absolute_path(const char *path)
|
|
{
|
|
size_t len = strlen(path);
|
|
if (IS_FILESEP(path[0])) return 1;
|
|
if (len > 0)
|
|
return (
|
|
is_absolute_traditional_path(path, len) ||
|
|
is_absolute_unc_path(path, len) ||
|
|
is_absolute_device_path(path, len)
|
|
);
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
#undef IS_FILESEP
|
|
#undef IS_VOLSEP
|
|
#undef IS_DEVICEID
|
|
#undef CHECK_UNCDEV_PATH
|
|
|
|
#else
|
|
static int
|
|
mrb_file_is_absolute_path(const char *path)
|
|
{
|
|
return (path[0] == *(char*)(FILE_SEPARATOR));
|
|
}
|
|
#endif
|
|
|
|
static mrb_value
|
|
mrb_file__gethome(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
char *home;
|
|
mrb_value path;
|
|
|
|
mrb->c->ci->mid = 0;
|
|
mrb_value username;
|
|
|
|
mrb_int argc = mrb_get_args(mrb, "|S", &username);
|
|
if (argc == 0) {
|
|
home = getenv("HOME");
|
|
if (home == NULL) {
|
|
return mrb_nil_value();
|
|
}
|
|
#ifdef _WIN32
|
|
home = getenv("USERPROFILE");
|
|
if (home == NULL) {
|
|
return mrb_nil_value();
|
|
}
|
|
#endif
|
|
if (!mrb_file_is_absolute_path(home)) {
|
|
mrb_raise(mrb, E_ARGUMENT_ERROR, "non-absolute home");
|
|
}
|
|
}
|
|
#if defined(_WIN32) || defined(MRB_IO_NO_PWNAM)
|
|
else {
|
|
return mrb_nil_value();
|
|
}
|
|
#else
|
|
else {
|
|
const char *cuser = RSTRING_CSTR(mrb, username);
|
|
struct passwd *pwd = getpwnam(cuser);
|
|
if (pwd == NULL) {
|
|
return mrb_nil_value();
|
|
}
|
|
home = pwd->pw_dir;
|
|
if (!mrb_file_is_absolute_path(home)) {
|
|
mrb_raisef(mrb, E_ARGUMENT_ERROR, "non-absolute home of ~%v", username);
|
|
}
|
|
}
|
|
#endif
|
|
home = mrb_utf8_from_locale(home, -1);
|
|
path = mrb_str_new_cstr(mrb, home);
|
|
mrb_utf8_free(home);
|
|
return path;
|
|
}
|
|
|
|
#define TIME_OVERFLOW_P(a) (sizeof(time_t) >= sizeof(mrb_int) && ((a) > MRB_INT_MAX || (a) < MRB_INT_MIN))
|
|
#define TIME_T_UINT (~(time_t)0 > 0)
|
|
#if defined(MRB_USE_BITINT)
|
|
#define TIME_BIGTIME(mrb, a) \
|
|
return (TIME_T_UINT ? mrb_bint_new_uint64((mrb), (uint64_t)(a)) \
|
|
: mrb_bint_new_int64(mrb, (int64_t)(a)))
|
|
#elif !defined(MRB_NO_FLOAT)
|
|
#define TIME_BIGTIME(mrb,a) return mrb_float_value((mrb), (mrb_float)(a))
|
|
#else
|
|
#define TIME_BIGTIME(mrb, a) mrb_raise(mrb, E_IO_ERROR, #a " overflow")
|
|
#endif
|
|
|
|
static mrb_value
|
|
mrb_file_atime(mrb_state *mrb, mrb_value self)
|
|
{
|
|
int fd = mrb_io_fileno(mrb, self);
|
|
mrb_stat st;
|
|
|
|
mrb->c->ci->mid = 0;
|
|
if (mrb_fstat(fd, &st) == -1)
|
|
mrb_sys_fail(mrb, "atime");
|
|
if (TIME_OVERFLOW_P(st.st_atime)) {
|
|
TIME_BIGTIME(mrb, st.st_atime);
|
|
}
|
|
return mrb_int_value(mrb, (mrb_int)st.st_atime);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_ctime(mrb_state *mrb, mrb_value self)
|
|
{
|
|
int fd = mrb_io_fileno(mrb, self);
|
|
mrb_stat st;
|
|
|
|
mrb->c->ci->mid = 0;
|
|
if (mrb_fstat(fd, &st) == -1)
|
|
mrb_sys_fail(mrb, "ctime");
|
|
if (TIME_OVERFLOW_P(st.st_ctime)) {
|
|
TIME_BIGTIME(mrb, st. st_ctime);
|
|
}
|
|
return mrb_int_value(mrb, (mrb_int)st.st_ctime);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_mtime(mrb_state *mrb, mrb_value self)
|
|
{
|
|
int fd = mrb_io_fileno(mrb, self);
|
|
mrb_stat st;
|
|
|
|
mrb->c->ci->mid = 0;
|
|
if (mrb_fstat(fd, &st) == -1)
|
|
mrb_sys_fail(mrb, "mtime");
|
|
if (TIME_OVERFLOW_P(st.st_mtime)) {
|
|
TIME_BIGTIME(mrb, st. st_mtime);
|
|
}
|
|
return mrb_int_value(mrb, (mrb_int)st.st_mtime);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_flock(mrb_state *mrb, mrb_value self)
|
|
{
|
|
#if defined(sun)
|
|
mrb_raise(mrb, E_NOTIMP_ERROR, "flock is not supported on Illumos/Solaris/Windows");
|
|
#else
|
|
mrb_int operation;
|
|
|
|
mrb_get_args(mrb, "i", &operation);
|
|
int fd = mrb_io_fileno(mrb, self);
|
|
|
|
while (flock(fd, (int)operation) == -1) {
|
|
switch (errno) {
|
|
case EINTR:
|
|
/* retry */
|
|
break;
|
|
case EAGAIN: /* NetBSD */
|
|
#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
|
|
case EWOULDBLOCK: /* FreeBSD OpenBSD Linux */
|
|
#endif
|
|
if (operation & LOCK_NB) {
|
|
return mrb_false_value();
|
|
}
|
|
/* FALLTHRU - should not happen */
|
|
default:
|
|
mrb_sys_fail(mrb, "flock failed");
|
|
break;
|
|
}
|
|
}
|
|
#endif
|
|
return mrb_fixnum_value(0);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_size(mrb_state *mrb, mrb_value self)
|
|
{
|
|
mrb_stat st;
|
|
int fd = mrb_io_fileno(mrb, self);
|
|
if (mrb_fstat(fd, &st) == -1) {
|
|
mrb_raise(mrb, E_RUNTIME_ERROR, "fstat failed");
|
|
}
|
|
|
|
if (st.st_size > MRB_INT_MAX) {
|
|
#ifdef MRB_NO_FLOAT
|
|
mrb_raise(mrb, E_RUNTIME_ERROR, "File#size too large for MRB_NO_FLOAT");
|
|
#else
|
|
return mrb_float_value(mrb, (mrb_float)st.st_size);
|
|
#endif
|
|
}
|
|
|
|
return mrb_int_value(mrb, (mrb_int)st.st_size);
|
|
}
|
|
|
|
static int
|
|
mrb_ftruncate(int fd, mrb_int length)
|
|
{
|
|
#ifndef _WIN32
|
|
return ftruncate(fd, (off_t)length);
|
|
#else
|
|
__int64 cur;
|
|
HANDLE file = (HANDLE)_get_osfhandle(fd);
|
|
if (file == INVALID_HANDLE_VALUE) {
|
|
return -1;
|
|
}
|
|
|
|
cur = _lseeki64(fd, 0, SEEK_CUR);
|
|
if (cur == -1) return -1;
|
|
|
|
if (_lseeki64(fd, (__int64)length, SEEK_SET) == -1) return -1;
|
|
|
|
if (!SetEndOfFile(file)) {
|
|
errno = EINVAL; /* TODO: GetLastError to errno */
|
|
return -1;
|
|
}
|
|
|
|
if (_lseeki64(fd, cur, SEEK_SET) == -1) return -1;
|
|
|
|
return 0;
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_truncate(mrb_state *mrb, mrb_value self)
|
|
{
|
|
mrb_value lenv = mrb_get_arg1(mrb);
|
|
int fd = mrb_io_fileno(mrb, self);
|
|
mrb_int length = mrb_as_int(mrb, lenv);
|
|
if (mrb_ftruncate(fd, length) != 0) {
|
|
mrb_raise(mrb, E_IO_ERROR, "ftruncate failed");
|
|
}
|
|
|
|
return mrb_fixnum_value(0);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_s_symlink(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
mrb_raise(mrb, E_NOTIMP_ERROR, "symlink is not supported on this platform");
|
|
#else
|
|
mrb_value from, to;
|
|
|
|
mrb_get_args(mrb, "SS", &from, &to);
|
|
const char *src = mrb_locale_from_utf8(RSTRING_CSTR(mrb, from), -1);
|
|
const char *dst = mrb_locale_from_utf8(RSTRING_CSTR(mrb, to), -1);
|
|
if (symlink(src, dst) == -1) {
|
|
mrb_locale_free(src);
|
|
mrb_locale_free(dst);
|
|
mrb_sys_fail(mrb, RSTRING_CSTR(mrb, mrb_format(mrb, "(%v, %v)", from, to)));
|
|
}
|
|
mrb_locale_free(src);
|
|
mrb_locale_free(dst);
|
|
#endif
|
|
return mrb_fixnum_value(0);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_s_chmod(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
mrb_int mode;
|
|
mrb_int argc;
|
|
const mrb_value *filenames;
|
|
int ai = mrb_gc_arena_save(mrb);
|
|
|
|
mrb_get_args(mrb, "i*", &mode, &filenames, &argc);
|
|
for (int i = 0; i < argc; i++) {
|
|
const char *utf8_path = RSTRING_CSTR(mrb, filenames[i]);
|
|
char *path = mrb_locale_from_utf8(utf8_path, -1);
|
|
if (CHMOD(path, mode) == -1) {
|
|
mrb_locale_free(path);
|
|
mrb_sys_fail(mrb, utf8_path);
|
|
}
|
|
mrb_locale_free(path);
|
|
mrb_gc_arena_restore(mrb, ai);
|
|
}
|
|
|
|
return mrb_fixnum_value(argc);
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_file_s_readlink(mrb_state *mrb, mrb_value klass)
|
|
{
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
mrb_raise(mrb, E_NOTIMP_ERROR, "readlink is not supported on this platform");
|
|
return mrb_nil_value(); // unreachable
|
|
#else
|
|
const char *path;
|
|
char *buf, *tmp;
|
|
size_t bufsize = 100;
|
|
ssize_t rc;
|
|
mrb_value ret;
|
|
|
|
mrb_get_args(mrb, "z", &path);
|
|
tmp = mrb_locale_from_utf8(path, -1);
|
|
|
|
buf = (char*)mrb_malloc(mrb, bufsize);
|
|
while ((rc = readlink(tmp, buf, bufsize)) == (ssize_t)bufsize && rc != -1) {
|
|
bufsize *= 2;
|
|
buf = (char*)mrb_realloc(mrb, buf, bufsize);
|
|
}
|
|
mrb_locale_free(tmp);
|
|
if (rc == -1) {
|
|
mrb_free(mrb, buf);
|
|
mrb_sys_fail(mrb, path);
|
|
}
|
|
tmp = mrb_utf8_from_locale(buf, -1);
|
|
ret = mrb_str_new(mrb, tmp, rc);
|
|
mrb_utf8_free(tmp);
|
|
mrb_free(mrb, buf);
|
|
|
|
return ret;
|
|
#endif
|
|
}
|
|
|
|
void
|
|
mrb_init_file(mrb_state *mrb)
|
|
{
|
|
struct RClass *io = mrb_class_get_id(mrb, MRB_SYM(IO));
|
|
struct RClass *file = mrb_define_class_id(mrb, MRB_SYM(File), io);
|
|
MRB_SET_INSTANCE_TT(file, MRB_TT_CDATA);
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(umask), mrb_file_s_umask, MRB_ARGS_OPT(1));
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(delete), mrb_file_s_unlink, MRB_ARGS_ANY());
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(unlink), mrb_file_s_unlink, MRB_ARGS_ANY());
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(rename), mrb_file_s_rename, MRB_ARGS_REQ(2));
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(symlink), mrb_file_s_symlink, MRB_ARGS_REQ(2));
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(chmod), mrb_file_s_chmod, MRB_ARGS_REQ(1) | MRB_ARGS_REST());
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(readlink), mrb_file_s_readlink, MRB_ARGS_REQ(1));
|
|
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(dirname), mrb_file_dirname, MRB_ARGS_REQ(1));
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(basename), mrb_file_basename, MRB_ARGS_REQ(1));
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(realpath), mrb_file_realpath, MRB_ARGS_REQ(1)|MRB_ARGS_OPT(1));
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(_getwd), mrb_file__getwd, MRB_ARGS_NONE());
|
|
mrb_define_class_method_id(mrb, file, MRB_SYM(_gethome), mrb_file__gethome, MRB_ARGS_OPT(1));
|
|
|
|
mrb_define_method_id(mrb, file, MRB_SYM(flock), mrb_file_flock, MRB_ARGS_REQ(1));
|
|
mrb_define_method_id(mrb, file, MRB_SYM(_atime), mrb_file_atime, MRB_ARGS_NONE());
|
|
mrb_define_method_id(mrb, file, MRB_SYM(_ctime), mrb_file_ctime, MRB_ARGS_NONE());
|
|
mrb_define_method_id(mrb, file, MRB_SYM(_mtime), mrb_file_mtime, MRB_ARGS_NONE());
|
|
mrb_define_method_id(mrb, file, MRB_SYM(size), mrb_file_size, MRB_ARGS_NONE());
|
|
mrb_define_method_id(mrb, file, MRB_SYM(truncate), mrb_file_truncate, MRB_ARGS_REQ(1));
|
|
|
|
struct RClass *cnst = mrb_define_module_under_id(mrb, file, MRB_SYM(Constants));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(LOCK_SH), mrb_fixnum_value(LOCK_SH));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(LOCK_EX), mrb_fixnum_value(LOCK_EX));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(LOCK_UN), mrb_fixnum_value(LOCK_UN));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(LOCK_NB), mrb_fixnum_value(LOCK_NB));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(SEPARATOR), mrb_str_new_cstr(mrb, FILE_SEPARATOR));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(PATH_SEPARATOR), mrb_str_new_cstr(mrb, PATH_SEPARATOR));
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(ALT_SEPARATOR), mrb_str_new_cstr(mrb, FILE_ALT_SEPARATOR));
|
|
#else
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(ALT_SEPARATOR), mrb_nil_value());
|
|
#endif
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(NULL), mrb_str_new_cstr(mrb, NULL_FILE));
|
|
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(RDONLY), mrb_fixnum_value(MRB_O_RDONLY));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(WRONLY), mrb_fixnum_value(MRB_O_WRONLY));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(RDWR), mrb_fixnum_value(MRB_O_RDWR));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(APPEND), mrb_fixnum_value(MRB_O_APPEND));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(CREAT), mrb_fixnum_value(MRB_O_CREAT));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(EXCL), mrb_fixnum_value(MRB_O_EXCL));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(TRUNC), mrb_fixnum_value(MRB_O_TRUNC));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(NONBLOCK), mrb_fixnum_value(MRB_O_NONBLOCK));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(NOCTTY), mrb_fixnum_value(MRB_O_NOCTTY));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(BINARY), mrb_fixnum_value(MRB_O_BINARY));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(SHARE_DELETE), mrb_fixnum_value(MRB_O_SHARE_DELETE));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(SYNC), mrb_fixnum_value(MRB_O_SYNC));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(DSYNC), mrb_fixnum_value(MRB_O_DSYNC));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(RSYNC), mrb_fixnum_value(MRB_O_RSYNC));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(NOFOLLOW), mrb_fixnum_value(MRB_O_NOFOLLOW));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(NOATIME), mrb_fixnum_value(MRB_O_NOATIME));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(DIRECT), mrb_fixnum_value(MRB_O_DIRECT));
|
|
mrb_define_const_id(mrb, cnst, MRB_SYM(TMPFILE), mrb_fixnum_value(MRB_O_TMPFILE));
|
|
}
|