Blame

Date:
Message:
Today's changes. More changes.
Actions:
History | Blob | Raw File
  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 fd04aace 2003-11-23 devnull extern int _p9dir(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 fd04aace 2003-11-23 devnull 	struct stat st;

 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 1c8efa18 2003-12-04 devnull 		if(de->d_name[0] == 0)

 84 1c8efa18 2003-12-04 devnull 			/* nothing */ {}

 85 1c8efa18 2003-12-04 devnull 		else if(stat(de->d_name, &st) < 0)

 86 1c253ceb 2003-11-23 devnull 			de->d_name[0] = 0;

 87 fd04aace 2003-11-23 devnull 		else

 88 fd04aace 2003-11-23 devnull 			nstr += _p9dir(&st, de->d_name, nil, nil, nil);

 89 fd04aace 2003-11-23 devnull 		p += de->d_reclen;

 90 fd04aace 2003-11-23 devnull 	}

 91 fd04aace 2003-11-23 devnull 

 92 fd04aace 2003-11-23 devnull 	d = malloc(sizeof(Dir)*n+nstr);

 93 fd04aace 2003-11-23 devnull 	if(d == nil){

 94 fd04aace 2003-11-23 devnull 		fchdir(oldwd);

 95 fd04aace 2003-11-23 devnull 		close(oldwd);

 96 fd04aace 2003-11-23 devnull 		return -1;

 97 fd04aace 2003-11-23 devnull 	}

 98 fd04aace 2003-11-23 devnull 	str = (char*)&d[n];

 99 fd04aace 2003-11-23 devnull 	estr = str+nstr;

100 fd04aace 2003-11-23 devnull 

101 fd04aace 2003-11-23 devnull 	p = buf;

102 fd04aace 2003-11-23 devnull 	m = 0;

103 fd04aace 2003-11-23 devnull 	for(i=0; i<n; i++){

104 fd04aace 2003-11-23 devnull 		de = (struct dirent*)p;

105 1c253ceb 2003-11-23 devnull 		if(de->d_name[0] != 0 && stat(de->d_name, &st) >= 0)

106 fd04aace 2003-11-23 devnull 			_p9dir(&st, de->d_name, &d[m++], &str, estr);

107 fd04aace 2003-11-23 devnull 		p += de->d_reclen;

108 fd04aace 2003-11-23 devnull 	}

109 fd04aace 2003-11-23 devnull 

110 fd04aace 2003-11-23 devnull 	fchdir(oldwd);

111 fd04aace 2003-11-23 devnull 	close(oldwd);

112 fd04aace 2003-11-23 devnull 	*dp = d;

113 fd04aace 2003-11-23 devnull 	return m;

114 fd04aace 2003-11-23 devnull }

115 fd04aace 2003-11-23 devnull 

116 fd04aace 2003-11-23 devnull long

117 fd04aace 2003-11-23 devnull dirread(int fd, Dir **dp)

118 fd04aace 2003-11-23 devnull {

119 fd04aace 2003-11-23 devnull 	char *buf;

120 fd04aace 2003-11-23 devnull 	struct stat st;

121 fd04aace 2003-11-23 devnull 	int n;

122 fd04aace 2003-11-23 devnull 

123 fd04aace 2003-11-23 devnull 	*dp = 0;

124 fd04aace 2003-11-23 devnull 

125 fd04aace 2003-11-23 devnull 	if(fstat(fd, &st) < 0)

126 fd04aace 2003-11-23 devnull 		return -1;

127 fd04aace 2003-11-23 devnull 

128 fd04aace 2003-11-23 devnull 	if(st.st_blksize < 8192)

129 fd04aace 2003-11-23 devnull 		st.st_blksize = 8192;

130 fd04aace 2003-11-23 devnull 

131 fd04aace 2003-11-23 devnull 	buf = malloc(st.st_blksize);

132 fd04aace 2003-11-23 devnull 	if(buf == nil)

133 fd04aace 2003-11-23 devnull 		return -1;

134 fd04aace 2003-11-23 devnull 

135 912fba95 2003-11-24 devnull 	n = mygetdents(fd, (void*)buf, st.st_blksize);

136 fd04aace 2003-11-23 devnull 	if(n < 0){

137 fd04aace 2003-11-23 devnull 		free(buf);

138 fd04aace 2003-11-23 devnull 		return -1;

139 fd04aace 2003-11-23 devnull 	}

140 fd04aace 2003-11-23 devnull 	n = dirpackage(fd, buf, n, dp);

141 fd04aace 2003-11-23 devnull 	free(buf);

142 fd04aace 2003-11-23 devnull 	return n;

143 fd04aace 2003-11-23 devnull }

144 fd04aace 2003-11-23 devnull 

145 fd04aace 2003-11-23 devnull 

146 fd04aace 2003-11-23 devnull long

147 fd04aace 2003-11-23 devnull dirreadall(int fd, Dir **d)

148 fd04aace 2003-11-23 devnull {

149 fd04aace 2003-11-23 devnull 	uchar *buf, *nbuf;

150 fd04aace 2003-11-23 devnull 	long n, ts;

151 fd04aace 2003-11-23 devnull 	struct stat st;

152 fd04aace 2003-11-23 devnull 

153 fd04aace 2003-11-23 devnull 	if(fstat(fd, &st) < 0)

154 fd04aace 2003-11-23 devnull 		return -1;

155 fd04aace 2003-11-23 devnull 

156 fd04aace 2003-11-23 devnull 	if(st.st_blksize < 8192)

157 fd04aace 2003-11-23 devnull 		st.st_blksize = 8192;

158 fd04aace 2003-11-23 devnull 

159 fd04aace 2003-11-23 devnull 	buf = nil;

160 fd04aace 2003-11-23 devnull 	ts = 0;

161 fd04aace 2003-11-23 devnull 	for(;;){

162 fd04aace 2003-11-23 devnull 		nbuf = realloc(buf, ts+st.st_blksize);

163 fd04aace 2003-11-23 devnull 		if(nbuf == nil){

164 fd04aace 2003-11-23 devnull 			free(buf);

165 fd04aace 2003-11-23 devnull 			return -1;

166 fd04aace 2003-11-23 devnull 		}

167 fd04aace 2003-11-23 devnull 		buf = nbuf;

168 912fba95 2003-11-24 devnull 		n = mygetdents(fd, (void*)(buf+ts), st.st_blksize);

169 fd04aace 2003-11-23 devnull 		if(n <= 0)

170 fd04aace 2003-11-23 devnull 			break;

171 fd04aace 2003-11-23 devnull 		ts += n;

172 fd04aace 2003-11-23 devnull 	}

173 fd04aace 2003-11-23 devnull 	if(ts >= 0)

174 8ad51794 2004-03-25 devnull 		ts = dirpackage(fd, (char*)buf, ts, d);

175 fd04aace 2003-11-23 devnull 	free(buf);

176 fd04aace 2003-11-23 devnull 	if(ts == 0 && n < 0)

177 fd04aace 2003-11-23 devnull 		return -1;

178 fd04aace 2003-11-23 devnull 	return ts;

179 fd04aace 2003-11-23 devnull }