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