Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
tag: r1_1b9
Fetching contributors…

Cannot retrieve contributors at this time

426 lines (359 sloc) 7.477 kB
/************************************************
dir.c -
$Author$
$Date$
created at: Wed Jan 5 09:51:01 JST 1994
Copyright (C) 1993-1998 Yukihiro Matsumoto
************************************************/
#include "ruby.h"
#include <sys/types.h>
#include <sys/stat.h>
#ifdef HAVE_SYS_PARAM_H
# include <sys/param.h>
#else
# define MAXPATHLEN 1024
#endif
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#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, dirname;
{
VALUE obj;
DIR *dirp;
Check_SafeStr(dirname);
dirp = opendir(RSTRING(dirname)->ptr);
if (dirp == NULL) {
if (errno == EMFILE || errno == ENFILE) {
gc_gc();
dirp = opendir(RSTRING(dirname)->ptr);
}
if (dirp == NULL) {
rb_sys_fail(RSTRING(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 (!NIL_P(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;
{
char path[MAXPATHLEN];
#ifdef HAVE_GETCWD
if (getcwd(path, sizeof(path)) == 0) rb_sys_fail(path);
#else
extern char *getwd();
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, dir;
{
rb_secure(2);
Check_SafeStr(dir);
if (rmdir(RSTRING(dir)->ptr) < 0)
rb_sys_fail(RSTRING(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, vstr)
VALUE dir, vstr;
{
char *p, *pend;
char buf[MAXPATHLEN];
char *t, *t0;
int nest;
VALUE ary;
struct RString *str;
Check_SafeStr(vstr);
str = RSTRING(vstr);
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, 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);
}
Jump to Line
Something went wrong with that request. Please try again.