1 #include <u.h>
2 #define NOPLAN9DEFINES
3 #include <libc.h>
4 #include <sys/types.h>
5 #include <sys/stat.h>
6 #include <dirent.h>
7 #include <pwd.h>
8 #include <grp.h>
9 
10 #if defined(__FreeBSD__) || defined(__OpenBSD__)
11 #include <sys/disklabel.h>
12 #include <sys/ioctl.h>
13 #define _HAVEDISKLABEL
14 #endif
15 
16 #if defined(__OpenBSD__)
17 static int diskdev[] = {
18 	151,	/* aacd */
19 	116,	/* ad */
20 	157,	/* ar */
21 	118,	/* afd */
22 	133,	/* amrd */
23 	13,	/* da */
24 	102,	/* fla */
25 	109,	/* idad */
26 	95,	/* md */
27 	131,	/* mlxd */
28 	168,	/* pst */
29 	147,	/* twed */
30 	43,	/* vn */
31 	3,	/* wd */
32 	87,	/* wfd */
33 	4,	/* da on FreeBSD 5 */
34 };
35 static int
36 isdisk(struct stat *st)
37 {
38 	int i, dev;
39 
40 	if(!S_ISCHR(st->st_mode))
41 		return 0;
42 	dev = major(st->st_rdev);
43 	for(i=0; i<nelem(diskdev); i++)
44 		if(diskdev[i] == dev)
45 			return 1;
46 	return 0;
47 }
48 #endif
49 
50 #if defined(__FreeBSD__)	/* maybe OpenBSD too? */
51 char *diskdev[] = {
52 	"aacd",
53 	"ad",
54 	"ar",
55 	"afd",
56 	"amrd",
57 	"da",
58 	"fla",
59 	"idad",
60 	"md",
61 	"mlxd",
62 	"pst",
63 	"twed",
64 	"vn",
65 	"wd",
66 	"wfd",
67 	"da",
68 };
69 static int
70 isdisk(struct stat *st)
71 {
72 	char *name;
73 	int i, len;
74 	
75 	if(!S_ISCHR(st->st_mode))
76 		return 0;
77 	name = devname(st->st_rdev, S_IFCHR);
78 	for(i=0; i<nelem(diskdev); i++){
79 		len = strlen(diskdev[i]);
80 		if(strncmp(diskdev[i], name, len) == 0 && isdigit((uchar)name[len]))
81 			return 1;
82 	}
83 	return 0;
84 }
85 #endif
86 
87 
88 #if defined(__linux__)
89 #include <linux/hdreg.h>
90 #include <linux/fs.h>
91 #include <sys/ioctl.h>
92 #undef major
93 #define major(dev) ((int)(((dev) >> 8) & 0xff))
94 static vlong
95 disksize(int fd, int dev)
96 {
97 	u64int u64;
98 	long l;
99 	struct hd_geometry geo;
100 
101 	memset(&geo, 0, sizeof geo);
102 	l = 0;
103 	u64 = 0;
104 #ifdef BLKGETSIZE64
105 	if(ioctl(fd, BLKGETSIZE64, &u64) >= 0)
106 		return u64;
107 #endif
108 	if(ioctl(fd, BLKGETSIZE, &l) >= 0)
109 		return l*512;
110 	if(ioctl(fd, HDIO_GETGEO, &geo) >= 0)
111 		return (vlong)geo.heads*geo.sectors*geo.cylinders*512;
112 	return 0;
113 }
114 #define _HAVEDISKSIZE
115 #endif
116 
117 #if !defined(__linux__) && !defined(__sun__)
118 #define _HAVESTGEN
119 #endif
120 
121 int _p9usepwlibrary = 1;
122 /*
123  * Caching the last group and passwd looked up is
124  * a significant win (stupidly enough) on most systems.
125  * It's not safe for threaded programs, but neither is using
126  * getpwnam in the first place, so I'm not too worried.
127  */
128 int
129 _p9dir(struct stat *lst, struct stat *st, char *name, Dir *d, char **str, char *estr)
130 {
131 	char *s;
132 	char tmp[20];
133 	static struct group *g;
134 	static struct passwd *p;
135 	static int gid, uid;
136 	int sz, fd;
137 
138 	fd = -1;
139 	USED(fd);
140 	sz = 0;
141 	if(d)
142 		memset(d, 0, sizeof *d);
143 
144 	/* name */
145 	s = strrchr(name, '/');
146 	if(s)
147 		s++;
148 	if(!s || !*s)
149 		s = name;
150 	if(*s == '/')
151 		s++;
152 	if(*s == 0)
153 		s = "/";
154 	if(d){
155 		if(*str + strlen(s)+1 > estr)
156 			d->name = "oops";
157 		else{
158 			strcpy(*str, s);
159 			d->name = *str;
160 			*str += strlen(*str)+1;
161 		}
162 	}
163 	sz += strlen(s)+1;
164 
165 	/* user */
166 	if(p && st->st_uid == uid && p->pw_uid == uid)
167 		;
168 	else if(_p9usepwlibrary){
169 		p = getpwuid(st->st_uid);
170 		uid = st->st_uid;
171 	}
172 	if(p == nil || st->st_uid != uid || p->pw_uid != uid){
173 		snprint(tmp, sizeof tmp, "%d", (int)st->st_uid);
174 		s = tmp;
175 	}else
176 		s = p->pw_name;
177 	sz += strlen(s)+1;
178 	if(d){
179 		if(*str+strlen(s)+1 > estr)
180 			d->uid = "oops";	
181 		else{
182 			strcpy(*str, s);
183 			d->uid = *str;
184 			*str += strlen(*str)+1;
185 		}
186 	}
187 
188 	/* group */
189 	if(g && st->st_gid == gid && g->gr_gid == gid)
190 		;
191 	else if(_p9usepwlibrary){
192 		g = getgrgid(st->st_gid);
193 		gid = st->st_gid;
194 	}
195 	if(g == nil || st->st_gid != gid || g->gr_gid != gid){
196 		snprint(tmp, sizeof tmp, "%d", (int)st->st_gid);
197 		s = tmp;
198 	}else
199 		s = g->gr_name;
200 	sz += strlen(s)+1;
201 	if(d){
202 		if(*str + strlen(s)+1 > estr)
203 			d->gid = "oops";	
204 		else{
205 			strcpy(*str, s);
206 			d->gid = *str;
207 			*str += strlen(*str)+1;
208 		}
209 	}
210 
211 	if(d){
212 		d->type = 'M';
213 
214 		d->muid = "";
215 		d->qid.path = ((uvlong)st->st_dev<<32) | st->st_ino;
216 #ifdef _HAVESTGEN
217 		d->qid.vers = st->st_gen;
218 #endif
219 		if(d->qid.vers == 0)
220 			d->qid.vers = st->st_mtime + st->st_ctime;
221 		d->mode = st->st_mode&0777;
222 		d->atime = st->st_atime;
223 		d->mtime = st->st_mtime;
224 		d->length = st->st_size;
225 
226 		if(S_ISDIR(st->st_mode)){
227 			d->length = 0;
228 			d->mode |= DMDIR;
229 			d->qid.type = QTDIR;
230 		}
231 		if(S_ISLNK(lst->st_mode))	/* yes, lst not st */
232 			d->mode |= DMSYMLINK;
233 		if(S_ISFIFO(st->st_mode))
234 			d->mode |= DMNAMEDPIPE;
235 		if(S_ISSOCK(st->st_mode))
236 			d->mode |= DMSOCKET;
237 		if(S_ISBLK(st->st_mode)){
238 			d->mode |= DMDEVICE;
239 			d->qid.path = ('b'<<16)|st->st_rdev;
240 		}
241 		if(S_ISCHR(st->st_mode)){
242 			d->mode |= DMDEVICE;
243 			d->qid.path = ('c'<<16)|st->st_rdev;
244 		}
245 		/* fetch real size for disks */
246 #ifdef _HAVEDISKSIZE
247 		if(S_ISBLK(st->st_mode) && (fd = open(name, O_RDONLY)) >= 0){
248 			d->length = disksize(fd, major(st->st_dev));
249 			close(fd);
250 		}
251 #endif
252 #ifdef _HAVEDISKLABEL
253 		if(isdisk(st)){
254 			int fd, n;
255 			struct disklabel lab;
256 
257 			if((fd = open(name, O_RDONLY)) < 0)
258 				goto nosize;
259 			if(ioctl(fd, DIOCGDINFO, &lab) < 0)
260 				goto nosize;
261 			n = minor(st->st_rdev)&7;
262 			if(n >= lab.d_npartitions)
263 				goto nosize;
264 
265 			d->length = (vlong)(lab.d_partitions[n].p_size) * lab.d_secsize;
266 
267 		nosize:
268 			if(fd >= 0)
269 				close(fd);
270 		}
271 #endif
272 	}
273 
274 	return sz;
275 }
276