1 fd04aace 2003-11-23 devnull #include <u.h>
  2 f7012583 2003-11-25 devnull #define NOPLAN9DEFINES
  3 fd04aace 2003-11-23 devnull #include <libc.h>
  4 fd04aace 2003-11-23 devnull #include <sys/stat.h>
  5 fd04aace 2003-11-23 devnull #include <dirent.h>
  6 fd04aace 2003-11-23 devnull 
  7 2acd6fa6 2005-02-08 devnull extern int _p9dir(struct stat*, struct stat*, char*, Dir*, char**, char*);
  8 fd04aace 2003-11-23 devnull 
  9 64bcfff3 2003-11-25 devnull #if defined(__linux__)
 10 fd04aace 2003-11-23 devnull static int
 11 64bcfff3 2003-11-25 devnull mygetdents(int fd, struct dirent *buf, int n)
 12 d2c4ee9e 2003-11-24 devnull {
 13 64bcfff3 2003-11-25 devnull 	off_t off;
 14 e95a7088 2003-12-09 devnull 	int nn;
 15 64bcfff3 2003-11-25 devnull 
 16 03480d73 2004-01-09 devnull 	/* This doesn't match the man page, but it works in Debian with a 2.2 kernel */
 17 64bcfff3 2003-11-25 devnull 	off = p9seek(fd, 0, 1);
 18 64bcfff3 2003-11-25 devnull 	nn = getdirentries(fd, (void*)buf, n, &off);
 19 64bcfff3 2003-11-25 devnull 	return nn;
 20 64bcfff3 2003-11-25 devnull }
 21 8ad51794 2004-03-25 devnull #elif defined(__APPLE__) || defined(__FreeBSD__) || defined(__OpenBSD__)
 22 64bcfff3 2003-11-25 devnull static int
 23 64bcfff3 2003-11-25 devnull mygetdents(int fd, struct dirent *buf, int n)
 24 64bcfff3 2003-11-25 devnull {
 25 912fba95 2003-11-24 devnull 	long off;
 26 669250d1 2003-12-03 devnull 	return getdirentries(fd, (void*)buf, n, &off);
 27 d2c4ee9e 2003-11-24 devnull }
 28 64bcfff3 2003-11-25 devnull #elif defined(__sun__)
 29 64bcfff3 2003-11-25 devnull static int
 30 64bcfff3 2003-11-25 devnull mygetdents(int fd, struct dirent *buf, int n)
 31 64bcfff3 2003-11-25 devnull {
 32 64bcfff3 2003-11-25 devnull 	return getdents(fd, (void*)buf, n);
 33 64bcfff3 2003-11-25 devnull }
 34 669250d1 2003-12-03 devnull #endif
 35 d2c4ee9e 2003-11-24 devnull 
 36 d2c4ee9e 2003-11-24 devnull static int
 37 fd04aace 2003-11-23 devnull countde(char *p, int n)
 38 fd04aace 2003-11-23 devnull {
 39 fd04aace 2003-11-23 devnull 	char *e;
 40 fd04aace 2003-11-23 devnull 	int m;
 41 fd04aace 2003-11-23 devnull 	struct dirent *de;
 42 fd04aace 2003-11-23 devnull 
 43 fd04aace 2003-11-23 devnull 	e = p+n;
 44 fd04aace 2003-11-23 devnull 	m = 0;
 45 fd04aace 2003-11-23 devnull 	while(p < e){
 46 fd04aace 2003-11-23 devnull 		de = (struct dirent*)p;
 47 fd04aace 2003-11-23 devnull 		if(de->d_reclen <= 4+2+2+1 || p+de->d_reclen > e)
 48 fd04aace 2003-11-23 devnull 			break;
 49 1c253ceb 2003-11-23 devnull 		if(de->d_name[0]=='.' && de->d_name[1]==0)
 50 1c253ceb 2003-11-23 devnull 			de->d_name[0] = 0;
 51 1c253ceb 2003-11-23 devnull 		else if(de->d_name[0]=='.' && de->d_name[1]=='.' && de->d_name[2]==0)
 52 1c253ceb 2003-11-23 devnull 			de->d_name[0] = 0;
 53 1c8efa18 2003-12-04 devnull 		m++;
 54 fd04aace 2003-11-23 devnull 		p += de->d_reclen;
 55 fd04aace 2003-11-23 devnull 	}
 56 fd04aace 2003-11-23 devnull 	return m;
 57 fd04aace 2003-11-23 devnull }
 58 fd04aace 2003-11-23 devnull 
 59 fd04aace 2003-11-23 devnull static int
 60 fd04aace 2003-11-23 devnull dirpackage(int fd, char *buf, int n, Dir **dp)
 61 fd04aace 2003-11-23 devnull {
 62 fd04aace 2003-11-23 devnull 	int oldwd;
 63 fd04aace 2003-11-23 devnull 	char *p, *str, *estr;
 64 fd04aace 2003-11-23 devnull 	int i, nstr, m;
 65 fd04aace 2003-11-23 devnull 	struct dirent *de;
 66 2acd6fa6 2005-02-08 devnull 	struct stat st, lst;
 67 fd04aace 2003-11-23 devnull 	Dir *d;
 68 fd04aace 2003-11-23 devnull 
 69 fd04aace 2003-11-23 devnull 	n = countde(buf, n);
 70 fd04aace 2003-11-23 devnull 	if(n <= 0)
 71 fd04aace 2003-11-23 devnull 		return n;
 72 fd04aace 2003-11-23 devnull 
 73 fd04aace 2003-11-23 devnull 	if((oldwd = open(".", O_RDONLY)) < 0)
 74 fd04aace 2003-11-23 devnull 		return -1;
 75 fd04aace 2003-11-23 devnull 	if(fchdir(fd) < 0)
 76 fd04aace 2003-11-23 devnull 		return -1;
 77 fd04aace 2003-11-23 devnull 		
 78 fd04aace 2003-11-23 devnull 	p = buf;
 79 fd04aace 2003-11-23 devnull 	nstr = 0;
 80 1c8efa18 2003-12-04 devnull 
 81 fd04aace 2003-11-23 devnull 	for(i=0; i<n; i++){
 82 fd04aace 2003-11-23 devnull 		de = (struct dirent*)p;
 83 46199d11 2005-02-08 devnull 		memset(&lst, 0, sizeof lst);
 84 1c8efa18 2003-12-04 devnull 		if(de->d_name[0] == 0)
 85 1c8efa18 2003-12-04 devnull 			/* nothing */ {}
 86 0a229052 2005-02-08 devnull 		else if(lstat(de->d_name, &lst) < 0)
 87 1c253ceb 2003-11-23 devnull 			de->d_name[0] = 0;
 88 0a229052 2005-02-08 devnull 		else{
 89 0a229052 2005-02-08 devnull 			st = lst;
 90 46199d11 2005-02-08 devnull 			if(S_ISLNK(lst.st_mode))
 91 0a229052 2005-02-08 devnull 				stat(de->d_name, &st);
 92 0a229052 2005-02-08 devnull 			nstr += _p9dir(&lst, &st, de->d_name, nil, nil, nil);
 93 0a229052 2005-02-08 devnull 		}
 94 fd04aace 2003-11-23 devnull 		p += de->d_reclen;
 95 fd04aace 2003-11-23 devnull 	}
 96 fd04aace 2003-11-23 devnull 
 97 fd04aace 2003-11-23 devnull 	d = malloc(sizeof(Dir)*n+nstr);
 98 fd04aace 2003-11-23 devnull 	if(d == nil){
 99 fd04aace 2003-11-23 devnull 		fchdir(oldwd);
100 fd04aace 2003-11-23 devnull 		close(oldwd);
101 fd04aace 2003-11-23 devnull 		return -1;
102 fd04aace 2003-11-23 devnull 	}
103 fd04aace 2003-11-23 devnull 	str = (char*)&d[n];
104 fd04aace 2003-11-23 devnull 	estr = str+nstr;
105 fd04aace 2003-11-23 devnull 
106 fd04aace 2003-11-23 devnull 	p = buf;
107 fd04aace 2003-11-23 devnull 	m = 0;
108 fd04aace 2003-11-23 devnull 	for(i=0; i<n; i++){
109 fd04aace 2003-11-23 devnull 		de = (struct dirent*)p;
110 46199d11 2005-02-08 devnull 		if(de->d_name[0] != 0 && lstat(de->d_name, &lst) >= 0){
111 2acd6fa6 2005-02-08 devnull 			st = lst;
112 2acd6fa6 2005-02-08 devnull 			if((lst.st_mode&S_IFMT) == S_IFLNK)
113 2acd6fa6 2005-02-08 devnull 				stat(de->d_name, &st);
114 2acd6fa6 2005-02-08 devnull 			_p9dir(&lst, &st, de->d_name, &d[m++], &str, estr);
115 2acd6fa6 2005-02-08 devnull 		}
116 fd04aace 2003-11-23 devnull 		p += de->d_reclen;
117 fd04aace 2003-11-23 devnull 	}
118 fd04aace 2003-11-23 devnull 
119 fd04aace 2003-11-23 devnull 	fchdir(oldwd);
120 fd04aace 2003-11-23 devnull 	close(oldwd);
121 fd04aace 2003-11-23 devnull 	*dp = d;
122 fd04aace 2003-11-23 devnull 	return m;
123 fd04aace 2003-11-23 devnull }
124 fd04aace 2003-11-23 devnull 
125 fd04aace 2003-11-23 devnull long
126 fd04aace 2003-11-23 devnull dirread(int fd, Dir **dp)
127 fd04aace 2003-11-23 devnull {
128 fd04aace 2003-11-23 devnull 	char *buf;
129 fd04aace 2003-11-23 devnull 	struct stat st;
130 fd04aace 2003-11-23 devnull 	int n;
131 fd04aace 2003-11-23 devnull 
132 fd04aace 2003-11-23 devnull 	*dp = 0;
133 fd04aace 2003-11-23 devnull 
134 fd04aace 2003-11-23 devnull 	if(fstat(fd, &st) < 0)
135 fd04aace 2003-11-23 devnull 		return -1;
136 fd04aace 2003-11-23 devnull 
137 fd04aace 2003-11-23 devnull 	if(st.st_blksize < 8192)
138 fd04aace 2003-11-23 devnull 		st.st_blksize = 8192;
139 fd04aace 2003-11-23 devnull 
140 fd04aace 2003-11-23 devnull 	buf = malloc(st.st_blksize);
141 fd04aace 2003-11-23 devnull 	if(buf == nil)
142 fd04aace 2003-11-23 devnull 		return -1;
143 fd04aace 2003-11-23 devnull 
144 912fba95 2003-11-24 devnull 	n = mygetdents(fd, (void*)buf, st.st_blksize);
145 fd04aace 2003-11-23 devnull 	if(n < 0){
146 fd04aace 2003-11-23 devnull 		free(buf);
147 fd04aace 2003-11-23 devnull 		return -1;
148 fd04aace 2003-11-23 devnull 	}
149 fd04aace 2003-11-23 devnull 	n = dirpackage(fd, buf, n, dp);
150 fd04aace 2003-11-23 devnull 	free(buf);
151 fd04aace 2003-11-23 devnull 	return n;
152 fd04aace 2003-11-23 devnull }
153 fd04aace 2003-11-23 devnull 
154 fd04aace 2003-11-23 devnull 
155 fd04aace 2003-11-23 devnull long
156 fd04aace 2003-11-23 devnull dirreadall(int fd, Dir **d)
157 fd04aace 2003-11-23 devnull {
158 fd04aace 2003-11-23 devnull 	uchar *buf, *nbuf;
159 fd04aace 2003-11-23 devnull 	long n, ts;
160 fd04aace 2003-11-23 devnull 	struct stat st;
161 fd04aace 2003-11-23 devnull 
162 fd04aace 2003-11-23 devnull 	if(fstat(fd, &st) < 0)
163 fd04aace 2003-11-23 devnull 		return -1;
164 fd04aace 2003-11-23 devnull 
165 fd04aace 2003-11-23 devnull 	if(st.st_blksize < 8192)
166 fd04aace 2003-11-23 devnull 		st.st_blksize = 8192;
167 fd04aace 2003-11-23 devnull 
168 fd04aace 2003-11-23 devnull 	buf = nil;
169 fd04aace 2003-11-23 devnull 	ts = 0;
170 fd04aace 2003-11-23 devnull 	for(;;){
171 fd04aace 2003-11-23 devnull 		nbuf = realloc(buf, ts+st.st_blksize);
172 fd04aace 2003-11-23 devnull 		if(nbuf == nil){
173 fd04aace 2003-11-23 devnull 			free(buf);
174 fd04aace 2003-11-23 devnull 			return -1;
175 fd04aace 2003-11-23 devnull 		}
176 fd04aace 2003-11-23 devnull 		buf = nbuf;
177 912fba95 2003-11-24 devnull 		n = mygetdents(fd, (void*)(buf+ts), st.st_blksize);
178 fd04aace 2003-11-23 devnull 		if(n <= 0)
179 fd04aace 2003-11-23 devnull 			break;
180 fd04aace 2003-11-23 devnull 		ts += n;
181 fd04aace 2003-11-23 devnull 	}
182 fd04aace 2003-11-23 devnull 	if(ts >= 0)
183 8ad51794 2004-03-25 devnull 		ts = dirpackage(fd, (char*)buf, ts, d);
184 fd04aace 2003-11-23 devnull 	free(buf);
185 fd04aace 2003-11-23 devnull 	if(ts == 0 && n < 0)
186 fd04aace 2003-11-23 devnull 		return -1;
187 fd04aace 2003-11-23 devnull 	return ts;
188 fd04aace 2003-11-23 devnull }