mirror of
				https://github.com/ruby/ruby.git
				synced 2022-11-09 12:17:21 -05:00 
			
		
		
		
	creating brand new exception object of the receiver. * eval.c (rb_eval_string_wrap): extend new ruby_top_self, not original self. * eval.c (rb_eval_cmd): respect ruby_wrapper if set. * eval.c (eval): do not update ruby_class unless scope is not provided. * eval.c (eval): preserve wrapper information. * eval.c (proc_invoke): ditto. * eval.c (block_pass): ditto. * parse.y (void_expr): too much warnings for void context (e.g. foo[1] that can be mere Proc call). * error.c (rb_name_error): new function to raise NameError with name attribute set. * eval.c (rb_f_missing): set name and args in the exception object. [new] * error.c (name_name): NameError#name - new method. * error.c (nometh_args): NoMethodError#args - new method. * lex.c (rb_reserved_word): lex_state after tRESCUE should be EXPR_MID. * gc.c (add_heap): allocation size of the heap unit is doubled for each allocation. * dir.c (isdelim): space, tab, and newline are no longer delimiters for glob patterns. * eval.c (svalue_to_avalue): new conversion scheme between single value and array values. * eval.c (avalue_to_svalue): ditto. * eval.c (rb_eval): REXPAND now uses avalue_to_svalue(), return and yield too. * eval.c (rb_yield_0): use avalue_to_svalue(). * eval.c (proc_invoke): Proc#call gives avaules, whereas Proc#yield gives mvalues. * eval.c (bmcall): convert given value (svalue) to avalue. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@1555 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
		
			
				
	
	
		
			991 lines
		
	
	
	
		
			19 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			991 lines
		
	
	
	
		
			19 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/**********************************************************************
 | 
						|
 | 
						|
  dir.c -
 | 
						|
 | 
						|
  $Author$
 | 
						|
  $Date$
 | 
						|
  created at: Wed Jan  5 09:51:01 JST 1994
 | 
						|
 | 
						|
  Copyright (C) 1993-2001 Yukihiro Matsumoto
 | 
						|
  Copyright (C) 2000  Network Applied Communication Laboratory, Inc.
 | 
						|
  Copyright (C) 2000  Information-technology Promotion Agency, Japan
 | 
						|
 | 
						|
**********************************************************************/
 | 
						|
 | 
						|
#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 defined HAVE_DIRENT_H && !defined NT
 | 
						|
# include <dirent.h>
 | 
						|
# define NAMLEN(dirent) strlen((dirent)->d_name)
 | 
						|
#elif defined HAVE_DIRECT_H && !defined NT
 | 
						|
# include <direct.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
 | 
						|
# if defined(NT)
 | 
						|
#  include "win32/dir.h"
 | 
						|
# endif
 | 
						|
#endif
 | 
						|
 | 
						|
#include <errno.h>
 | 
						|
 | 
						|
#ifndef HAVE_STDLIB_H
 | 
						|
char *getenv();
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef HAVE_STRING_H
 | 
						|
char *strchr _((char*,char));
 | 
						|
#endif
 | 
						|
 | 
						|
#include <ctype.h>
 | 
						|
 | 
						|
#ifndef HAVE_LSTAT
 | 
						|
#define lstat(path,st) stat(path,st)
 | 
						|
#endif
 | 
						|
 | 
						|
#define FNM_NOESCAPE	0x01
 | 
						|
#define FNM_PATHNAME	0x02
 | 
						|
#define FNM_PERIOD	0x04
 | 
						|
#define FNM_CASEFOLD	0x08
 | 
						|
 | 
						|
#define FNM_NOMATCH	1
 | 
						|
#define FNM_ERROR	2
 | 
						|
 | 
						|
#define downcase(c) (nocase && ISUPPER(c) ? tolower(c) : (c))
 | 
						|
 | 
						|
#if defined DOSISH
 | 
						|
#define isdirsep(c) ((c) == '/' || (c) == '\\')
 | 
						|
static char *
 | 
						|
find_dirsep(s)
 | 
						|
    char *s;
 | 
						|
{
 | 
						|
    while (*s) {
 | 
						|
	if (isdirsep(*s))
 | 
						|
	    return s;
 | 
						|
	s++;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
#else
 | 
						|
#define isdirsep(c) ((c) == '/')
 | 
						|
#define find_dirsep(s) strchr(s, '/')
 | 
						|
#endif
 | 
						|
 | 
						|
static char *
 | 
						|
range(pat, test, flags)
 | 
						|
    char *pat;
 | 
						|
    char test;
 | 
						|
    int flags;
 | 
						|
{
 | 
						|
    int not, ok = 0;
 | 
						|
    int nocase = flags & FNM_CASEFOLD;
 | 
						|
    int escape = !(flags & FNM_NOESCAPE);
 | 
						|
 | 
						|
    not = *pat == '!' || *pat == '^';
 | 
						|
    if (not)
 | 
						|
	pat++;
 | 
						|
 | 
						|
    test = downcase(test);
 | 
						|
 | 
						|
    while (*pat) {
 | 
						|
	int cstart, cend;
 | 
						|
	cstart = cend = *pat++;
 | 
						|
	if (cstart == ']')
 | 
						|
	    return ok == not ? 0 : pat;
 | 
						|
        else if (escape && cstart == '\\')
 | 
						|
	    cstart = cend = *pat++;
 | 
						|
	if (*pat == '-' && pat[1] != ']') {
 | 
						|
	    if (escape && pat[1] == '\\')
 | 
						|
		pat++;
 | 
						|
	    cend = pat[1];
 | 
						|
	    if (!cend)
 | 
						|
		return 0;
 | 
						|
	    pat += 2;
 | 
						|
	}
 | 
						|
	if (downcase(cstart) <= test && test <= downcase(cend))
 | 
						|
	    ok = 1;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
#define ISDIRSEP(c) (pathname && isdirsep(c))
 | 
						|
#define PERIOD(s) (period && *(s) == '.' && \
 | 
						|
		  ((s) == string || ISDIRSEP((s)[-1])))
 | 
						|
static int
 | 
						|
fnmatch(pat, string, flags)
 | 
						|
    const char *pat;
 | 
						|
    const char *string;
 | 
						|
    int flags;
 | 
						|
{
 | 
						|
    int c;
 | 
						|
    int test;
 | 
						|
    const char *s = string;
 | 
						|
    int escape = !(flags & FNM_NOESCAPE);
 | 
						|
    int pathname = flags & FNM_PATHNAME;
 | 
						|
    int period = flags & FNM_PERIOD;
 | 
						|
    int nocase = flags & FNM_CASEFOLD;
 | 
						|
 | 
						|
    while (c = *pat++) {
 | 
						|
	switch (c) {
 | 
						|
	case '?':
 | 
						|
	    if (!*s || ISDIRSEP(*s) || PERIOD(s))
 | 
						|
		return FNM_NOMATCH;
 | 
						|
	    s++;
 | 
						|
	    break;
 | 
						|
	case '*':
 | 
						|
	    while ((c = *pat++) == '*')
 | 
						|
		;
 | 
						|
 | 
						|
	    if (PERIOD(s))
 | 
						|
		return FNM_NOMATCH;
 | 
						|
 | 
						|
	    if (!c) {
 | 
						|
		if (pathname && find_dirsep(s))
 | 
						|
		    return FNM_NOMATCH;
 | 
						|
		else
 | 
						|
		    return 0;
 | 
						|
	    }
 | 
						|
	    else if (ISDIRSEP(c)) {
 | 
						|
		s = find_dirsep(s);
 | 
						|
		if (s)
 | 
						|
		    break;
 | 
						|
		return FNM_NOMATCH;
 | 
						|
	    }
 | 
						|
 | 
						|
	    test = escape && c == '\\' ? *pat : c;
 | 
						|
	    test = downcase(test);
 | 
						|
	    pat--;
 | 
						|
	    while (*s) {
 | 
						|
		if ((c == '[' || downcase(*s) == test) &&
 | 
						|
		    !fnmatch(pat, s, flags & ~FNM_PERIOD))
 | 
						|
		    return 0;
 | 
						|
		else if (ISDIRSEP(*s))
 | 
						|
		    break;
 | 
						|
		s++;
 | 
						|
	    }
 | 
						|
	    return FNM_NOMATCH;
 | 
						|
      
 | 
						|
	case '[':
 | 
						|
	    if (!*s || ISDIRSEP(*s) || PERIOD(s))
 | 
						|
		return FNM_NOMATCH;
 | 
						|
	    pat = range(pat, *s, flags);
 | 
						|
	    if (!pat)
 | 
						|
		return FNM_NOMATCH;
 | 
						|
	    s++;
 | 
						|
	    break;
 | 
						|
 | 
						|
	case '\\':
 | 
						|
	    if (escape
 | 
						|
#if defined DOSISH
 | 
						|
		&& *pat && strchr("*?[\\", *pat)
 | 
						|
#endif
 | 
						|
		) {
 | 
						|
		c = *pat;
 | 
						|
		if (!c)
 | 
						|
		    c = '\\';
 | 
						|
		else
 | 
						|
		    pat++;
 | 
						|
	    }
 | 
						|
	    /* FALLTHROUGH */
 | 
						|
 | 
						|
	default:
 | 
						|
#if defined DOSISH
 | 
						|
	    if (ISDIRSEP(c) && isdirsep(*s))
 | 
						|
		;
 | 
						|
	    else
 | 
						|
#endif
 | 
						|
	    if(downcase(c) != downcase(*s))
 | 
						|
		return FNM_NOMATCH;
 | 
						|
	    s++;
 | 
						|
	    break;
 | 
						|
	}
 | 
						|
    }
 | 
						|
    return !*s ? 0 : FNM_NOMATCH;
 | 
						|
}
 | 
						|
 | 
						|
VALUE rb_cDir;
 | 
						|
 | 
						|
static void
 | 
						|
free_dir(dir)
 | 
						|
    DIR *dir;
 | 
						|
{
 | 
						|
    if (dir) closedir(dir);
 | 
						|
}
 | 
						|
 | 
						|
static VALUE dir_close _((VALUE));
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_s_new(argc, argv, klass)
 | 
						|
    int argc;
 | 
						|
    VALUE *argv;
 | 
						|
    VALUE klass;
 | 
						|
{
 | 
						|
    VALUE obj = Data_Wrap_Struct(klass, 0, free_dir, 0);
 | 
						|
 | 
						|
    rb_obj_call_init(obj, argc, argv);
 | 
						|
 | 
						|
    return obj;
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_initialize(dir, dirname)
 | 
						|
    VALUE dir, dirname;
 | 
						|
{
 | 
						|
    DIR *dirp;
 | 
						|
 | 
						|
    SafeStringValue(dirname);
 | 
						|
    if (DATA_PTR(dir)) closedir(DATA_PTR(dir));
 | 
						|
    DATA_PTR(dir) = NULL;
 | 
						|
    dirp = opendir(RSTRING(dirname)->ptr);
 | 
						|
    if (dirp == NULL) {
 | 
						|
	if (errno == EMFILE || errno == ENFILE) {
 | 
						|
	    rb_gc();
 | 
						|
	    dirp = opendir(RSTRING(dirname)->ptr);
 | 
						|
	}
 | 
						|
	if (dirp == NULL) {
 | 
						|
	    rb_sys_fail(RSTRING(dirname)->ptr);
 | 
						|
	}
 | 
						|
    }
 | 
						|
    DATA_PTR(dir) = dirp;
 | 
						|
 | 
						|
    return dir;
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_s_open(klass, dirname)
 | 
						|
    VALUE klass, dirname;
 | 
						|
{
 | 
						|
    VALUE dir = Data_Wrap_Struct(klass, 0, free_dir, 0);
 | 
						|
 | 
						|
    dir_initialize(dir, dirname);
 | 
						|
    if (rb_block_given_p()) {
 | 
						|
	return rb_ensure(rb_yield, dir, dir_close, dir);
 | 
						|
    }
 | 
						|
 | 
						|
    return dir;
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
dir_closed()
 | 
						|
{
 | 
						|
    rb_raise(rb_eIOError, "closed directory");
 | 
						|
}
 | 
						|
 | 
						|
#define GetDIR(obj, dirp) {\
 | 
						|
    Data_Get_Struct(obj, DIR, dirp);\
 | 
						|
    if (dirp == NULL) dir_closed();\
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_read(dir)
 | 
						|
    VALUE dir;
 | 
						|
{
 | 
						|
    DIR *dirp;
 | 
						|
    struct dirent *dp;
 | 
						|
 | 
						|
    GetDIR(dir, dirp);
 | 
						|
    errno = 0;
 | 
						|
    dp = readdir(dirp);
 | 
						|
    if (dp)
 | 
						|
	return rb_tainted_str_new(dp->d_name, NAMLEN(dp));
 | 
						|
    else if (errno == 0) {	/* end of stream */
 | 
						|
	return Qnil;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
	rb_sys_fail(0);
 | 
						|
    }
 | 
						|
    return Qnil;		/* not reached */
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_each(dir)
 | 
						|
    VALUE dir;
 | 
						|
{
 | 
						|
    DIR *dirp;
 | 
						|
    struct dirent *dp;
 | 
						|
 | 
						|
    GetDIR(dir, dirp);
 | 
						|
    for (dp = readdir(dirp); dp != NULL; dp = readdir(dirp)) {
 | 
						|
	rb_yield(rb_tainted_str_new(dp->d_name, NAMLEN(dp)));
 | 
						|
	if (DATA_PTR(dir) == NULL) dir_closed();
 | 
						|
    }
 | 
						|
    return dir;
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_tell(dir)
 | 
						|
    VALUE dir;
 | 
						|
{
 | 
						|
#ifdef HAVE_TELLDIR
 | 
						|
    DIR *dirp;
 | 
						|
    long pos;
 | 
						|
 | 
						|
    GetDIR(dir, dirp);
 | 
						|
    pos = telldir(dirp);
 | 
						|
    return rb_int2inum(pos);
 | 
						|
#else
 | 
						|
    rb_notimplement();
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_seek(dir, pos)
 | 
						|
    VALUE dir, pos;
 | 
						|
{
 | 
						|
    DIR *dirp;
 | 
						|
 | 
						|
#ifdef HAVE_SEEKDIR
 | 
						|
    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 void
 | 
						|
dir_chdir(path)
 | 
						|
    const char *path;
 | 
						|
{
 | 
						|
    if (chdir(path) < 0)
 | 
						|
	rb_sys_fail(path);
 | 
						|
}
 | 
						|
 | 
						|
static int chdir_blocking = 0;
 | 
						|
 | 
						|
static VALUE
 | 
						|
chdir_restore(path)
 | 
						|
    const char *path;
 | 
						|
{
 | 
						|
    chdir_blocking--;
 | 
						|
    dir_chdir(path);
 | 
						|
    return Qnil;
 | 
						|
}
 | 
						|
 | 
						|
#ifdef HAVE_GETCWD
 | 
						|
#define GETCWD(path) if (getcwd(path, sizeof(path)) == 0) rb_sys_fail(path)
 | 
						|
#else
 | 
						|
#define GETCWD(path) if (getwd(path) == 0) rb_sys_fail(path)
 | 
						|
#endif
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_s_chdir(argc, argv, obj)
 | 
						|
    int argc;
 | 
						|
    VALUE *argv;
 | 
						|
    VALUE obj;
 | 
						|
{
 | 
						|
    VALUE path = Qnil;
 | 
						|
    char *dist = "";
 | 
						|
 | 
						|
    rb_secure(2);
 | 
						|
    if (rb_scan_args(argc, argv, "01", &path) == 1) {
 | 
						|
	SafeStringValue(path);
 | 
						|
	dist = RSTRING(path)->ptr;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
	dist = getenv("HOME");
 | 
						|
	if (!dist) {
 | 
						|
	    dist = getenv("LOGDIR");
 | 
						|
	}
 | 
						|
    }
 | 
						|
 | 
						|
    if (chdir_blocking > 0)
 | 
						|
	rb_warn("chdir during chdir block");
 | 
						|
 | 
						|
    if (rb_block_given_p()) {
 | 
						|
	char cwd[MAXPATHLEN];
 | 
						|
 | 
						|
	GETCWD(cwd);
 | 
						|
	chdir_blocking++;
 | 
						|
	dir_chdir(dist);
 | 
						|
	return rb_ensure(rb_yield, path, chdir_restore, (VALUE)cwd);
 | 
						|
    }
 | 
						|
    dir_chdir(dist);
 | 
						|
 | 
						|
    return INT2FIX(0);
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_s_getwd(dir)
 | 
						|
    VALUE dir;
 | 
						|
{
 | 
						|
    char path[MAXPATHLEN];
 | 
						|
 | 
						|
    GETCWD(path);
 | 
						|
    return rb_tainted_str_new2(path);
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_s_chroot(dir, path)
 | 
						|
    VALUE dir, path;
 | 
						|
{
 | 
						|
#if defined(HAVE_CHROOT) && !defined(__CHECKER__)
 | 
						|
    rb_secure(2);
 | 
						|
    SafeStringValue(path);
 | 
						|
 | 
						|
    if (chroot(RSTRING(path)->ptr) == -1)
 | 
						|
	rb_sys_fail(RSTRING(path)->ptr);
 | 
						|
 | 
						|
    return INT2FIX(0);
 | 
						|
#else
 | 
						|
    rb_notimplement();
 | 
						|
    return Qnil;		/* not reached */
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_s_mkdir(argc, argv, obj)
 | 
						|
    int argc;
 | 
						|
    VALUE *argv;
 | 
						|
    VALUE obj;
 | 
						|
{
 | 
						|
    VALUE path, vmode;
 | 
						|
    int mode;
 | 
						|
 | 
						|
    if (rb_scan_args(argc, argv, "11", &path, &vmode) == 2) {
 | 
						|
	mode = NUM2INT(vmode);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
	mode = 0777;
 | 
						|
    }
 | 
						|
 | 
						|
    SafeStringValue(path);
 | 
						|
    rb_secure(2);
 | 
						|
#if !defined(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;
 | 
						|
{
 | 
						|
    SafeStringValue(dir);
 | 
						|
    rb_secure(2);
 | 
						|
    if (rmdir(RSTRING(dir)->ptr) < 0)
 | 
						|
	rb_sys_fail(RSTRING(dir)->ptr);
 | 
						|
 | 
						|
    return INT2FIX(0);
 | 
						|
}
 | 
						|
 | 
						|
/* Return nonzero if S has any special globbing chars in it.  */
 | 
						|
static int
 | 
						|
has_magic(s, send, flags)
 | 
						|
     char *s, *send;
 | 
						|
     int flags;
 | 
						|
{
 | 
						|
    register char *p = s;
 | 
						|
    register char c;
 | 
						|
    int open = 0;
 | 
						|
    int escape = !(flags & FNM_NOESCAPE);
 | 
						|
 | 
						|
    while ((c = *p++) != '\0') {
 | 
						|
	switch (c) {
 | 
						|
	  case '?':
 | 
						|
	  case '*':
 | 
						|
	    return Qtrue;
 | 
						|
 | 
						|
	  case '[':	/* Only accept an open brace if there is a close */
 | 
						|
	    open++;	/* brace to match it.  Bracket expressions must be */
 | 
						|
	    continue;	/* complete, according to Posix.2 */
 | 
						|
	  case ']':
 | 
						|
	    if (open)
 | 
						|
		return Qtrue;
 | 
						|
	    continue;
 | 
						|
 | 
						|
	  case '\\':
 | 
						|
	    if (escape && *p++ == '\0')
 | 
						|
		return Qfalse;
 | 
						|
	}
 | 
						|
 | 
						|
	if (send && p >= send) break;
 | 
						|
    }
 | 
						|
    return Qfalse;
 | 
						|
}
 | 
						|
 | 
						|
static char*
 | 
						|
extract_path(p, pend)
 | 
						|
    char *p, *pend;
 | 
						|
{
 | 
						|
    char *alloc;
 | 
						|
    int len;
 | 
						|
 | 
						|
    len = pend - p;
 | 
						|
    alloc = ALLOC_N(char, len+1);
 | 
						|
    memcpy(alloc, p, len);
 | 
						|
    if (len > 1 && pend[-1] == '/'
 | 
						|
#if defined DOSISH
 | 
						|
    && pend[-2] != ':'
 | 
						|
#endif
 | 
						|
    ) {
 | 
						|
	alloc[len-1] = 0;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
	alloc[len] = 0;
 | 
						|
    }
 | 
						|
 | 
						|
    return alloc;
 | 
						|
}
 | 
						|
 | 
						|
static char*
 | 
						|
extract_elem(path)
 | 
						|
    char *path;
 | 
						|
{
 | 
						|
    char *pend;
 | 
						|
 | 
						|
    pend = strchr(path, '/');
 | 
						|
    if (!pend) pend = path + strlen(path);
 | 
						|
 | 
						|
    return extract_path(path, pend);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
remove_backslashes(p)
 | 
						|
    char *p;
 | 
						|
{
 | 
						|
#if defined DOSISH
 | 
						|
    char *pend = p + strlen(p);
 | 
						|
    char *t = p;
 | 
						|
 | 
						|
    while (p < pend) {
 | 
						|
	if (*p == '\\') {
 | 
						|
	    if (++p == pend) break;
 | 
						|
	}
 | 
						|
	*t++ = *p++;
 | 
						|
    }
 | 
						|
    *t = '\0';
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
#ifndef S_ISDIR
 | 
						|
#   define S_ISDIR(m) ((m & S_IFMT) == S_IFDIR)
 | 
						|
#endif
 | 
						|
 | 
						|
static void
 | 
						|
glob_helper(path, flags, func, arg)
 | 
						|
    char *path;
 | 
						|
    int flags;
 | 
						|
    void (*func)();
 | 
						|
    VALUE arg;
 | 
						|
{
 | 
						|
    struct stat st;
 | 
						|
    char *p, *m;
 | 
						|
 | 
						|
    if (!has_magic(path, 0, flags)) {
 | 
						|
	remove_backslashes(path);
 | 
						|
	if (stat(path, &st) == 0) {
 | 
						|
	    (*func)(path, arg);
 | 
						|
	}
 | 
						|
	else if (errno != ENOENT) {
 | 
						|
	    /* In case stat error is other than ENOENT and
 | 
						|
	       we may want to know what is wrong. */
 | 
						|
	    rb_sys_warning(path);
 | 
						|
	}
 | 
						|
	return;
 | 
						|
    }
 | 
						|
 | 
						|
    p = path;
 | 
						|
    while (p) {
 | 
						|
	if (*p == '/') p++;
 | 
						|
	m = strchr(p, '/');
 | 
						|
	if (has_magic(p, m, flags)) {
 | 
						|
	    char *dir, *base, *magic, *buf;
 | 
						|
	    DIR *dirp;
 | 
						|
	    struct dirent *dp;
 | 
						|
	    int recursive = 0;
 | 
						|
 | 
						|
	    struct d_link {
 | 
						|
		char *path;
 | 
						|
		struct d_link *next;
 | 
						|
	    } *tmp, *link = 0;
 | 
						|
 | 
						|
	    base = extract_path(path, p);
 | 
						|
	    if (path == p) dir = ".";
 | 
						|
	    else dir = base;
 | 
						|
 | 
						|
	    magic = extract_elem(p);
 | 
						|
	    if (stat(dir, &st) < 0) {
 | 
						|
	        if (errno != ENOENT) rb_sys_warning(dir);
 | 
						|
	        free(base);
 | 
						|
	        break;
 | 
						|
	    }
 | 
						|
	    if (S_ISDIR(st.st_mode)) {
 | 
						|
		if (m && strcmp(magic, "**") == 0) {
 | 
						|
		    recursive = 1;
 | 
						|
		    buf = ALLOC_N(char, strlen(base)+strlen(m)+3);
 | 
						|
		    sprintf(buf, "%s%s", base, *base ? m : m+1);
 | 
						|
		    glob_helper(buf, flags, func, arg);
 | 
						|
		    free(buf);
 | 
						|
		}
 | 
						|
		dirp = opendir(dir);
 | 
						|
		if (dirp == NULL) {
 | 
						|
		    rb_sys_warning(dir);
 | 
						|
		    free(base);
 | 
						|
		    break;
 | 
						|
		}
 | 
						|
	    }
 | 
						|
	    else {
 | 
						|
		free(base);
 | 
						|
		break;
 | 
						|
	    }
 | 
						|
	    
 | 
						|
#if defined DOSISH
 | 
						|
#define BASE (*base && !((isdirsep(*base) && !base[1]) || (base[1] == ':' && isdirsep(base[2]) && !base[3])))
 | 
						|
#else
 | 
						|
#define BASE (*base && !(*base == '/' && !base[1]))
 | 
						|
#endif
 | 
						|
 | 
						|
	    for (dp = readdir(dirp); dp != NULL; dp = readdir(dirp)) {
 | 
						|
		if (recursive) {
 | 
						|
		    if (strcmp(".", dp->d_name) == 0 || strcmp("..", dp->d_name) == 0)
 | 
						|
			continue;
 | 
						|
		    buf = ALLOC_N(char, strlen(base)+NAMLEN(dp)+strlen(m)+6);
 | 
						|
		    sprintf(buf, "%s%s%s", base, (BASE)?"/":"", dp->d_name);
 | 
						|
		    if (lstat(buf, &st) < 0) {
 | 
						|
			if (errno != ENOENT) rb_sys_warning(buf);
 | 
						|
			continue;
 | 
						|
		    }
 | 
						|
		    if (S_ISDIR(st.st_mode)) {
 | 
						|
		        strcat(buf, "/**");
 | 
						|
			strcat(buf, m);
 | 
						|
			glob_helper(buf, flags, func, arg);
 | 
						|
		    }
 | 
						|
		    free(buf);
 | 
						|
		    continue;
 | 
						|
		}
 | 
						|
		if (fnmatch(magic, dp->d_name, flags) == 0) {
 | 
						|
		    buf = ALLOC_N(char, strlen(base)+NAMLEN(dp)+2);
 | 
						|
		    sprintf(buf, "%s%s%s", base, (BASE)?"/":"", dp->d_name);
 | 
						|
		    if (!m) {
 | 
						|
			(*func)(buf, arg);
 | 
						|
			free(buf);
 | 
						|
			continue;
 | 
						|
		    }
 | 
						|
		    tmp = ALLOC(struct d_link);
 | 
						|
		    tmp->path = buf;
 | 
						|
		    tmp->next = link;
 | 
						|
		    link = tmp;
 | 
						|
		}
 | 
						|
	    }
 | 
						|
	    closedir(dirp);
 | 
						|
	    free(base);
 | 
						|
	    free(magic);
 | 
						|
	    if (link) {
 | 
						|
		while (link) {
 | 
						|
		    if (stat(link->path, &st) == 0) {
 | 
						|
			if (S_ISDIR(st.st_mode)) {
 | 
						|
			    int len = strlen(link->path);
 | 
						|
			    int mlen = strlen(m);
 | 
						|
			    char *t = ALLOC_N(char, len+mlen+1);
 | 
						|
 | 
						|
			    sprintf(t, "%s%s", link->path, m);
 | 
						|
			    glob_helper(t, flags, func, arg);
 | 
						|
			    free(t);
 | 
						|
			}
 | 
						|
			tmp = link;
 | 
						|
			link = link->next;
 | 
						|
			free(tmp->path);
 | 
						|
			free(tmp);
 | 
						|
		    }
 | 
						|
		    else {
 | 
						|
			rb_sys_warning(link->path);
 | 
						|
		    }
 | 
						|
		}
 | 
						|
		break;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
	p = m;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
rb_glob(path, func, arg)
 | 
						|
    char *path;
 | 
						|
    void (*func) _((const char*, VALUE));
 | 
						|
    VALUE arg;
 | 
						|
{
 | 
						|
    glob_helper(path, FNM_PERIOD, func, arg);
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
rb_globi(path, func, arg)
 | 
						|
    char *path;
 | 
						|
    void (*func)();
 | 
						|
    VALUE arg;
 | 
						|
{
 | 
						|
    glob_helper(path, FNM_PERIOD|FNM_CASEFOLD, func, arg);
 | 
						|
}
 | 
						|
 | 
						|
static void push_pattern _((const char *path, VALUE ary));
 | 
						|
 | 
						|
static void
 | 
						|
push_pattern(path, ary)
 | 
						|
    const char *path;
 | 
						|
    VALUE ary;
 | 
						|
{
 | 
						|
    VALUE str = rb_tainted_str_new2(path);
 | 
						|
 | 
						|
    if (ary) {
 | 
						|
	rb_ary_push(ary, str);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
	rb_yield(str);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
push_globs(ary, s)
 | 
						|
    VALUE ary;
 | 
						|
    char *s;
 | 
						|
{
 | 
						|
    rb_glob(s, push_pattern, ary);
 | 
						|
}
 | 
						|
 | 
						|
static void
 | 
						|
push_braces(ary, s)
 | 
						|
    VALUE ary;
 | 
						|
    char *s;
 | 
						|
{
 | 
						|
    char buffer[MAXPATHLEN], *buf = buffer;
 | 
						|
    char *p, *t, *b;
 | 
						|
    char *lbrace, *rbrace;
 | 
						|
    int nest = 0;
 | 
						|
 | 
						|
    p = s;
 | 
						|
    lbrace = rbrace = 0;
 | 
						|
    while (*p) {
 | 
						|
	if (*p == '{') {
 | 
						|
	    lbrace = p;
 | 
						|
	    break;
 | 
						|
	}
 | 
						|
	p++;
 | 
						|
    }
 | 
						|
    while (*p) {
 | 
						|
	if (*p == '{') nest++;
 | 
						|
	if (*p == '}' && --nest == 0) {
 | 
						|
	    rbrace = p;
 | 
						|
	    break;
 | 
						|
	}
 | 
						|
	p++;
 | 
						|
    }
 | 
						|
 | 
						|
    if (lbrace) {
 | 
						|
	int len = strlen(s);
 | 
						|
	if (len >= MAXPATHLEN)
 | 
						|
	    buf = xmalloc(len + 1);
 | 
						|
	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);
 | 
						|
	}
 | 
						|
	if (buf != buffer)
 | 
						|
	    free(buf);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
	push_globs(ary, s);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
#define isdelim(c) ((c)=='\0')
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_s_glob(dir, str)
 | 
						|
    VALUE dir, str;
 | 
						|
{
 | 
						|
    char *p, *pend;
 | 
						|
    char buffer[MAXPATHLEN], *buf;
 | 
						|
    char *t;
 | 
						|
    int nest;
 | 
						|
    VALUE ary = 0;
 | 
						|
 | 
						|
    SafeStringValue(str);
 | 
						|
    if (!rb_block_given_p()) {
 | 
						|
	ary = rb_ary_new();
 | 
						|
    }
 | 
						|
    if (RSTRING(str)->len >= MAXPATHLEN) {
 | 
						|
	buf = xmalloc(RSTRING(str)->len + 1);
 | 
						|
    } else {
 | 
						|
	buf = buffer;
 | 
						|
    }
 | 
						|
 | 
						|
    p = RSTRING(str)->ptr;
 | 
						|
    pend = p + RSTRING(str)->len;
 | 
						|
 | 
						|
    while (p < pend) {
 | 
						|
	t = buf;
 | 
						|
	nest = 0;
 | 
						|
	while (p < pend && isdelim(*p)) p++;
 | 
						|
	while (p < pend && !isdelim(*p)) {
 | 
						|
	    if (*p == '{') nest+=2;
 | 
						|
	    if (*p == '}') nest+=3;
 | 
						|
	    if (*p == '\\') {
 | 
						|
		*t++ = *p++;
 | 
						|
		if (p == pend) break;
 | 
						|
	    }
 | 
						|
	    *t++ = *p++;
 | 
						|
	}
 | 
						|
	*t = '\0';
 | 
						|
	if (nest == 0) {
 | 
						|
	    push_globs(ary, buf);
 | 
						|
	}
 | 
						|
	else if (nest % 5 == 0) {
 | 
						|
	    push_braces(ary, buf);
 | 
						|
	}
 | 
						|
	/* else unmatched braces */
 | 
						|
    }
 | 
						|
    if (buf != buffer)
 | 
						|
	free(buf);
 | 
						|
    if (ary && RARRAY(ary)->len == 0) {
 | 
						|
	rb_warning("no matches found: %s", RSTRING(str)->ptr);
 | 
						|
    }
 | 
						|
    return ary;
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_foreach(io, dirname)
 | 
						|
    VALUE io, dirname;
 | 
						|
{
 | 
						|
    VALUE dir;
 | 
						|
 | 
						|
    dir = rb_funcall(rb_cDir, rb_intern("open"), 1, dirname);
 | 
						|
    rb_ensure(dir_each, dir, dir_close, dir);
 | 
						|
    return Qnil;
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
dir_entries(io, dirname)
 | 
						|
    VALUE io, dirname;
 | 
						|
{
 | 
						|
    VALUE dir;
 | 
						|
 | 
						|
    dir = rb_funcall(rb_cDir, rb_intern("open"), 1, dirname);
 | 
						|
    return rb_ensure(rb_Array, dir, dir_close, dir);
 | 
						|
}
 | 
						|
 | 
						|
static VALUE
 | 
						|
file_s_fnmatch(argc, argv, obj)
 | 
						|
    int argc;
 | 
						|
    VALUE *argv;
 | 
						|
    VALUE obj;
 | 
						|
{
 | 
						|
    VALUE pattern, path;
 | 
						|
    VALUE rflags;
 | 
						|
    int flags;
 | 
						|
 | 
						|
    if (rb_scan_args(argc, argv, "21", &pattern, &path, &rflags) == 3)
 | 
						|
	flags = NUM2INT(rflags);
 | 
						|
    else
 | 
						|
	flags = 0;
 | 
						|
 | 
						|
    StringValue(pattern);
 | 
						|
    StringValue(path);
 | 
						|
 | 
						|
    if (fnmatch(RSTRING(pattern)->ptr, RSTRING(path)->ptr, flags) == 0)
 | 
						|
	return Qtrue;
 | 
						|
 | 
						|
    return Qfalse;
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
Init_Dir()
 | 
						|
{
 | 
						|
    rb_cDir = rb_define_class("Dir", rb_cObject);
 | 
						|
 | 
						|
    rb_include_module(rb_cDir, rb_mEnumerable);
 | 
						|
 | 
						|
    rb_define_singleton_method(rb_cDir, "new", dir_s_new, -1);
 | 
						|
    rb_define_singleton_method(rb_cDir, "open", dir_s_open, 1);
 | 
						|
    rb_define_singleton_method(rb_cDir, "foreach", dir_foreach, 1);
 | 
						|
    rb_define_singleton_method(rb_cDir, "entries", dir_entries, 1);
 | 
						|
 | 
						|
    rb_define_method(rb_cDir,"initialize", dir_initialize, 1);
 | 
						|
    rb_define_method(rb_cDir,"read", dir_read, 0);
 | 
						|
    rb_define_method(rb_cDir,"each", dir_each, 0);
 | 
						|
    rb_define_method(rb_cDir,"rewind", dir_rewind, 0);
 | 
						|
    rb_define_method(rb_cDir,"tell", dir_tell, 0);
 | 
						|
    rb_define_method(rb_cDir,"seek", dir_seek, 1);
 | 
						|
    rb_define_method(rb_cDir,"pos", dir_tell, 0);
 | 
						|
    rb_define_method(rb_cDir,"pos=", dir_seek, 1);
 | 
						|
    rb_define_method(rb_cDir,"close", dir_close, 0);
 | 
						|
 | 
						|
    rb_define_singleton_method(rb_cDir,"chdir", dir_s_chdir, -1);
 | 
						|
    rb_define_singleton_method(rb_cDir,"getwd", dir_s_getwd, 0);
 | 
						|
    rb_define_singleton_method(rb_cDir,"pwd", dir_s_getwd, 0);
 | 
						|
    rb_define_singleton_method(rb_cDir,"chroot", dir_s_chroot, 1);
 | 
						|
    rb_define_singleton_method(rb_cDir,"mkdir", dir_s_mkdir, -1);
 | 
						|
    rb_define_singleton_method(rb_cDir,"rmdir", dir_s_rmdir, 1);
 | 
						|
    rb_define_singleton_method(rb_cDir,"delete", dir_s_rmdir, 1);
 | 
						|
    rb_define_singleton_method(rb_cDir,"unlink", dir_s_rmdir, 1);
 | 
						|
 | 
						|
    rb_define_singleton_method(rb_cDir,"glob", dir_s_glob, 1);
 | 
						|
    rb_define_singleton_method(rb_cDir,"[]", dir_s_glob, 1);
 | 
						|
 | 
						|
    rb_define_singleton_method(rb_cFile,"fnmatch", file_s_fnmatch, -1);
 | 
						|
    rb_define_singleton_method(rb_cFile,"fnmatch?", file_s_fnmatch, -1);
 | 
						|
 | 
						|
    rb_file_const("FNM_NOESCAPE", INT2FIX(FNM_NOESCAPE));
 | 
						|
    rb_file_const("FNM_PATHNAME", INT2FIX(FNM_PATHNAME));
 | 
						|
    rb_file_const("FNM_PERIOD", INT2FIX(FNM_PERIOD));
 | 
						|
    rb_file_const("FNM_CASEFOLD", INT2FIX(FNM_CASEFOLD));
 | 
						|
}
 |