diff options
Diffstat (limited to 'dir.c')
| -rw-r--r-- | dir.c | 427 |
1 files changed, 427 insertions, 0 deletions
@@ -0,0 +1,427 @@ +/************************************************ + + dir.c - + + $Author$ + $Date$ + created at: Wed Jan 5 09:51:01 JST 1994 + + Copyright (C) 1993-1996 Yukihiro Matsumoto + +************************************************/ + +#include "ruby.h" + +#include <sys/types.h> +#ifdef HAVE_UNISTD_H +#include <unistd.h> +#endif + +#ifdef HAVE_SYS_PARAM_H +# include <sys/param.h> +#else +# define MAXPATHLEN 1024 +#endif + +#if HAVE_DIRENT_H +# include <dirent.h> +# define NAMLEN(dirent) strlen((dirent)->d_name) +#else +# define dirent direct +# define NAMLEN(dirent) (dirent)->d_namlen +# if HAVE_SYS_NDIR_H +# include <sys/ndir.h> +# endif +# if HAVE_SYS_DIR_H +# include <sys/dir.h> +# endif +# if HAVE_NDIR_H +# include <ndir.h> +# endif +# ifdef NT +# include "missing/dir.h" +# endif +#endif + +#include <errno.h> + +#ifndef NT +char *getenv(); +#endif + +static VALUE cDir; + +static void +free_dir(dir) + DIR *dir; +{ + if (dir) closedir(dir); +} + +static VALUE +dir_s_open(dir_class, dirname) + VALUE dir_class; + struct RString *dirname; +{ + VALUE obj; + DIR *dirp; + + Check_SafeStr(dirname); + + dirp = opendir(dirname->ptr); + if (dirp == NULL) { + if (errno == EMFILE || errno == ENFILE) { + gc(); + dirp = opendir(dirname->ptr); + } + if (dirp == NULL) { + rb_sys_fail(dirname->ptr); + } + } + + obj = Data_Wrap_Struct(dir_class, 0, free_dir, dirp); + + return obj; +} + +static void +dir_closed() +{ + Fail("closed directory"); +} + +#define GetDIR(obj, dirp) {\ + Data_Get_Struct(obj, DIR, dirp);\ + if (dirp == NULL) dir_closed();\ +} + +static VALUE +dir_each(dir) + VALUE dir; +{ + DIR *dirp; + struct dirent *dp; + VALUE file; + + GetDIR(dir, dirp); + for (dp = readdir(dirp); dp != NULL; dp = readdir(dirp)) { + file = str_taint(str_new(dp->d_name, NAMLEN(dp))); + rb_yield(file); + } + return dir; +} + +static VALUE +dir_tell(dir) + VALUE dir; +{ + DIR *dirp; + int pos; + +#if !defined(__CYGWIN32__) + GetDIR(dir, dirp); + pos = telldir(dirp); + return int2inum(pos); +#else + rb_notimplement(); +#endif +} + +static VALUE +dir_seek(dir, pos) + VALUE dir, pos; +{ + DIR *dirp; + +#if !defined(__CYGWIN32__) + GetDIR(dir, dirp); + seekdir(dirp, NUM2INT(pos)); + return dir; +#else + rb_notimplement(); +#endif +} + +static VALUE +dir_rewind(dir) + VALUE dir; +{ + DIR *dirp; + + GetDIR(dir, dirp); + rewinddir(dirp); + return dir; +} + +static VALUE +dir_close(dir) + VALUE dir; +{ + DIR *dirp; + + Data_Get_Struct(dir, DIR, dirp); + if (dirp == NULL) dir_closed(); + closedir(dirp); + DATA_PTR(dir) = NULL; + + return Qnil; +} + +static VALUE +dir_s_chdir(argc, argv, obj) + int argc; + VALUE *argv; + VALUE obj; +{ + VALUE path; + char *dist = ""; + + rb_secure(2); + rb_scan_args(argc, argv, "01", &path); + if (path) { + Check_SafeStr(path); + dist = RSTRING(path)->ptr; + } + else { + dist = getenv("HOME"); + if (!dist) { + dist = getenv("LOGDIR"); + } + } + + if (chdir(dist) < 0) + rb_sys_fail(0); + + return INT2FIX(0); +} + +static VALUE +dir_s_getwd(dir) + VALUE dir; +{ + extern char *getwd(); + char path[MAXPATHLEN]; + +#ifdef HAVE_GETCWD + if (getcwd(path, sizeof(path)) == 0) rb_sys_fail(path); +#else + if (getwd(path) == 0) rb_sys_fail(path); +#endif + + return str_taint(str_new2(path)); +} + +static VALUE +dir_s_chroot(dir, path) + VALUE dir, path; +{ +#if !defined(DJGPP) && !defined(__CYGWIN32__) && !defined(NT) && !defined(__human68k__) + rb_secure(2); + Check_SafeStr(path); + + if (chroot(RSTRING(path)->ptr) == -1) + rb_sys_fail(0); + + return INT2FIX(0); +#else + rb_notimplement(); +#endif +} + +static VALUE +dir_s_mkdir(argc, argv, obj) + int argc; + VALUE *argv; + VALUE obj; +{ + VALUE path, vmode; + int mode; + + rb_secure(2); + if (rb_scan_args(argc, argv, "11", &path, &vmode) == 2) { + mode = NUM2INT(vmode); + } + else { + mode = 0777; + } + + Check_SafeStr(path); +#ifndef NT + if (mkdir(RSTRING(path)->ptr, mode) == -1) + rb_sys_fail(RSTRING(path)->ptr); +#else + if (mkdir(RSTRING(path)->ptr) == -1) + rb_sys_fail(RSTRING(path)->ptr); +#endif + + return INT2FIX(0); +} + +static VALUE +dir_s_rmdir(obj, dir) + VALUE obj; + struct RString *dir; +{ + rb_secure(2); + Check_SafeStr(dir); + if (rmdir(dir->ptr) < 0) + rb_sys_fail(dir->ptr); + + return TRUE; +} + +#define isdelim(c) ((c)==' '||(c)=='\t'||(c)=='\n'||(c)=='\0') + +char **glob_filename(); +extern char *glob_error_return; + +static void +push_globs(ary, s) + VALUE ary; + char *s; +{ + char **fnames, **ff; + + fnames = glob_filename(s); + if (fnames == (char**)-1) rb_sys_fail(s); + ff = fnames; + while (*ff) { + ary_push(ary, str_taint(str_new2(*ff))); + free(*ff); + ff++; + } + if (fnames != &glob_error_return) { + free(fnames); + } +} + +static void +push_braces(ary, s) + VALUE ary; + char *s; +{ + char buf[MAXPATHLEN]; + char *p, *t, *b; + char *lbrace, *rbrace; + + p = s; + lbrace = rbrace = 0; + while (*p) { + if (*p == '{') { + lbrace = p; + break; + } + p++; + } + while (*p) { + if (*p == '}' && lbrace) { + rbrace = p; + break; + } + p++; + } + + if (lbrace) { + memcpy(buf, s, lbrace-s); + b = buf + (lbrace-s); + p = lbrace; + while (*p != '}') { + t = p + 1; + for (p = t; *p!='}' && *p!=','; p++) { + /* skip inner braces */ + if (*p == '{') while (*p!='}') p++; + } + memcpy(b, t, p-t); + strcpy(b+(p-t), rbrace+1); + push_braces(ary, buf); + } + } + else { + push_globs(ary, s); + } +} + +static VALUE +dir_s_glob(dir, str) + VALUE dir; + struct RString *str; +{ + char *p, *pend; + char buf[MAXPATHLEN]; + char *t, *t0; + int nest; + VALUE ary; + + Check_SafeStr(str); + + ary = ary_new(); + + p = str->ptr; + pend = p + str->len; + + while (p < pend) { + t = buf; + while (p < pend && isdelim(*p)) p++; + while (p < pend && !isdelim(*p)) { + *t++ = *p++; + } + *t = '\0'; + t0 = buf; + nest = 0; + while (t0 < t) { + if (*t0 == '{') nest+=2; + if (*t0 == '}') nest+=3; + t0++; + } + if (nest == 0) { + push_globs(ary, buf); + } + else if (nest % 5 == 0) { + push_braces(ary, buf); + } + /* else unmatched braces */ + } + return ary; +} + +static VALUE +dir_foreach(io, dirname) + VALUE io; + struct RString *dirname; +{ + VALUE dir; + + dir = dir_s_open(cDir, dirname); + return rb_ensure(dir_each, dir, dir_close, dir); +} + +void +Init_Dir() +{ + extern VALUE mEnumerable; + + cDir = rb_define_class("Dir", cObject); + + rb_include_module(cDir, mEnumerable); + + rb_define_singleton_method(cDir, "open", dir_s_open, 1); + rb_define_singleton_method(cDir, "foreach", dir_foreach, 1); + + rb_define_method(cDir,"each", dir_each, 0); + rb_define_method(cDir,"rewind", dir_rewind, 0); + rb_define_method(cDir,"tell", dir_tell, 0); + rb_define_method(cDir,"seek", dir_seek, 1); + rb_define_method(cDir,"close", dir_close, 0); + + rb_define_singleton_method(cDir,"chdir", dir_s_chdir, -1); + rb_define_singleton_method(cDir,"getwd", dir_s_getwd, 0); + rb_define_singleton_method(cDir,"pwd", dir_s_getwd, 0); + rb_define_singleton_method(cDir,"chroot", dir_s_chroot, 1); + rb_define_singleton_method(cDir,"mkdir", dir_s_mkdir, -1); + rb_define_singleton_method(cDir,"rmdir", dir_s_rmdir, 1); + rb_define_singleton_method(cDir,"delete", dir_s_rmdir, 1); + rb_define_singleton_method(cDir,"unlink", dir_s_rmdir, 1); + + rb_define_singleton_method(cDir,"glob", dir_s_glob, 1); + rb_define_singleton_method(cDir,"[]", dir_s_glob, 1); +} |