savedir.c revision 131554
153469Sobrien/* savedir.c -- save the list of files in a directory in a string 2131554Stjr Copyright (C) 1990, 1997, 1998, 1999, 2000, 2001 Free Software Foundation, Inc. 353469Sobrien 453469Sobrien This program is free software; you can redistribute it and/or modify 553469Sobrien it under the terms of the GNU General Public License as published by 653469Sobrien the Free Software Foundation; either version 2, or (at your option) 753469Sobrien any later version. 853469Sobrien 953469Sobrien This program is distributed in the hope that it will be useful, 1053469Sobrien but WITHOUT ANY WARRANTY; without even the implied warranty of 1153469Sobrien MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 1253469Sobrien GNU General Public License for more details. 1353469Sobrien 1453469Sobrien You should have received a copy of the GNU General Public License 1553469Sobrien along with this program; if not, write to the Free Software Foundation, 1653469Sobrien Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ 1753469Sobrien 1853469Sobrien/* Written by David MacKenzie <djm@gnu.ai.mit.edu>. */ 1953469Sobrien 2053469Sobrien#if HAVE_CONFIG_H 2153469Sobrien# include <config.h> 2253469Sobrien#endif 2353469Sobrien 2453469Sobrien#include <sys/types.h> 2553469Sobrien 2653469Sobrien#if HAVE_UNISTD_H 2753469Sobrien# include <unistd.h> 2853469Sobrien#endif 2953469Sobrien 3053469Sobrien#if HAVE_DIRENT_H 3153469Sobrien# include <dirent.h> 3253469Sobrien# define NAMLEN(dirent) strlen((dirent)->d_name) 3353469Sobrien#else 3453469Sobrien# define dirent direct 3553469Sobrien# define NAMLEN(dirent) (dirent)->d_namlen 3653469Sobrien# if HAVE_SYS_NDIR_H 3753469Sobrien# include <sys/ndir.h> 3853469Sobrien# endif 3953469Sobrien# if HAVE_SYS_DIR_H 4053469Sobrien# include <sys/dir.h> 4153469Sobrien# endif 4253469Sobrien# if HAVE_NDIR_H 4353469Sobrien# include <ndir.h> 4453469Sobrien# endif 4553469Sobrien#endif 4653469Sobrien 4753469Sobrien#ifdef CLOSEDIR_VOID 4853469Sobrien/* Fake a return value. */ 4953469Sobrien# define CLOSEDIR(d) (closedir (d), 0) 5053469Sobrien#else 5153469Sobrien# define CLOSEDIR(d) closedir (d) 5253469Sobrien#endif 5353469Sobrien 5453469Sobrien#ifdef STDC_HEADERS 5553469Sobrien# include <stdlib.h> 5653469Sobrien# include <string.h> 5753469Sobrien#else 5853469Sobrienchar *malloc (); 5953469Sobrienchar *realloc (); 6053469Sobrien#endif 6153469Sobrien#ifndef NULL 6253469Sobrien# define NULL 0 6353469Sobrien#endif 6453469Sobrien 6553469Sobrien#ifndef stpcpy 6653469Sobrienchar *stpcpy (); 6753469Sobrien#endif 6853469Sobrien 69131554Stjr#include <fnmatch.h> 7053469Sobrien#include "savedir.h" 7153469Sobrien 72131554Stjrchar *path; 73131554Stjrsize_t pathlen; 74131554Stjr 75131554Stjrstatic int 76131554Stjrisdir1 (const char *dir, const char *file) 77131554Stjr{ 78131554Stjr int status; 79131554Stjr int slash; 80131554Stjr size_t dirlen = strlen (dir); 81131554Stjr size_t filelen = strlen (file); 82131554Stjr if ((dirlen + filelen + 2) > pathlen) 83131554Stjr { 84131554Stjr path = calloc (dirlen + 1 + filelen + 1, sizeof (*path)); 85131554Stjr pathlen = dirlen + filelen + 2; 86131554Stjr } 87131554Stjr strcpy (path, dir); 88131554Stjr slash = (path[dirlen] != '/'); 89131554Stjr path[dirlen] = '/'; 90131554Stjr strcpy (path + dirlen + slash , file); 91131554Stjr status = isdir (path); 92131554Stjr return status; 93131554Stjr} 94131554Stjr 9553469Sobrien/* Return a freshly allocated string containing the filenames 9653469Sobrien in directory DIR, separated by '\0' characters; 9753469Sobrien the end is marked by two '\0' characters in a row. 9853469Sobrien NAME_SIZE is the number of bytes to initially allocate 9953469Sobrien for the string; it will be enlarged as needed. 10053469Sobrien Return NULL if DIR cannot be opened or if out of memory. */ 10153469Sobrienchar * 102131554Stjrsavedir (const char *dir, off_t name_size, struct exclude *included_patterns, 103131554Stjr struct exclude *excluded_patterns) 10453469Sobrien{ 10553469Sobrien DIR *dirp; 10653469Sobrien struct dirent *dp; 10753469Sobrien char *name_space; 10853469Sobrien char *namep; 10953469Sobrien 11053469Sobrien dirp = opendir (dir); 11153469Sobrien if (dirp == NULL) 11253469Sobrien return NULL; 11353469Sobrien 11456230Sru /* Be sure name_size is at least `1' so there's room for 11556230Sru the final NUL byte. */ 116126432Sache if (name_size <= 0) 117126432Sache name_size = 1; 11856230Sru 11953469Sobrien name_space = (char *) malloc (name_size); 12053469Sobrien if (name_space == NULL) 12153469Sobrien { 12253469Sobrien closedir (dirp); 12353469Sobrien return NULL; 12453469Sobrien } 12553469Sobrien namep = name_space; 12653469Sobrien 12753469Sobrien while ((dp = readdir (dirp)) != NULL) 12853469Sobrien { 12953469Sobrien /* Skip "." and ".." (some NFS filesystems' directories lack them). */ 13053469Sobrien if (dp->d_name[0] != '.' 13153469Sobrien || (dp->d_name[1] != '\0' 13253469Sobrien && (dp->d_name[1] != '.' || dp->d_name[2] != '\0'))) 13353469Sobrien { 13456915Sru off_t size_needed = (namep - name_space) + NAMLEN (dp) + 2; 13553469Sobrien 136131554Stjr if ((included_patterns || excluded_patterns) 137131554Stjr && !isdir1 (dir, dp->d_name)) 138131554Stjr { 139131554Stjr if (included_patterns 140131554Stjr && !excluded_filename (included_patterns, dp->d_name, 0)) 141131554Stjr continue; 142131554Stjr if (excluded_patterns 143131554Stjr && excluded_filename (excluded_patterns, dp->d_name, 0)) 144131554Stjr continue; 145131554Stjr } 146131554Stjr 14753469Sobrien if (size_needed > name_size) 14853469Sobrien { 14953469Sobrien char *new_name_space; 15053469Sobrien 15153469Sobrien while (size_needed > name_size) 15253469Sobrien name_size += 1024; 15353469Sobrien 15453469Sobrien new_name_space = realloc (name_space, name_size); 15553469Sobrien if (new_name_space == NULL) 15653469Sobrien { 15753469Sobrien closedir (dirp); 15853469Sobrien return NULL; 15953469Sobrien } 16053469Sobrien namep += new_name_space - name_space; 16153469Sobrien name_space = new_name_space; 16253469Sobrien } 16353469Sobrien namep = stpcpy (namep, dp->d_name) + 1; 16453469Sobrien } 16553469Sobrien } 16653469Sobrien *namep = '\0'; 16753469Sobrien if (CLOSEDIR (dirp)) 16853469Sobrien { 16953469Sobrien free (name_space); 17053469Sobrien return NULL; 17153469Sobrien } 172131554Stjr if (path) 173131554Stjr { 174131554Stjr free (path); 175131554Stjr path = NULL; 176131554Stjr pathlen = 0; 177131554Stjr } 17853469Sobrien return name_space; 17953469Sobrien} 180