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 /*
81  * Caching the last group and passwd looked up is
82  * a significant win (stupidly enough) on most systems.
83  * It's not safe for threaded programs, but neither is using
84  * getpwnam in the first place, so I'm not too worried.
85  */
86 int
87 _p9dir(struct stat *lst, struct stat *st, char *name, Dir *d, char **str, char *estr)
88 {
89 	char *s;
90 	char tmp[20];
91 	static struct group *g;
92 	static struct passwd *p;
93 	static int gid, uid;
94 	int sz, fd;
95 
96 	fd = -1;
97 	USED(fd);
98 	sz = 0;
99 	if(d)
100 		memset(d, 0, sizeof *d);
101 
102 	/* name */
103 	s = strrchr(name, '/');
104 	if(s)
105 		s++;
106 	if(!s || !*s)
107 		s = name;
108 	if(*s == '/')
109 		s++;
110 	if(*s == 0)
111 		s = "/";
112 	if(d){
113 		if(*str + strlen(s)+1 > estr)
114 			d->name = "oops";
115 		else{
116 			strcpy(*str, s);
117 			d->name = *str;
118 			*str += strlen(*str)+1;
119 		}
120 	}
121 	sz += strlen(s)+1;
122 
123 	/* user */
124 	if(p && st->st_uid == uid && p->pw_uid == uid)
125 		;
126 	else{
127 		p = getpwuid(st->st_uid);
128 		uid = st->st_uid;
129 	}
130 	if(p == nil){
131 		snprint(tmp, sizeof tmp, "%d", (int)st->st_uid);
132 		s = tmp;
133 	}else
134 		s = p->pw_name;
135 	sz += strlen(s)+1;
136 	if(d){
137 		if(*str+strlen(s)+1 > estr)
138 			d->uid = "oops";	
139 		else{
140 			strcpy(*str, s);
141 			d->uid = *str;
142 			*str += strlen(*str)+1;
143 		}
144 	}
145 
146 	/* group */
147 	if(g && st->st_gid == gid && g->gr_gid == gid)
148 		;
149 	else{
150 		g = getgrgid(st->st_gid);
151 		gid = st->st_gid;
152 	}
153 	if(g == nil){
154 		snprint(tmp, sizeof tmp, "%d", (int)st->st_gid);
155 		s = tmp;
156 	}else
157 		s = g->gr_name;
158 	sz += strlen(s)+1;
159 	if(d){
160 		if(*str + strlen(s)+1 > estr)
161 			d->gid = "oops";	
162 		else{
163 			strcpy(*str, s);
164 			d->gid = *str;
165 			*str += strlen(*str)+1;
166 		}
167 	}
168 
169 	if(d){
170 		d->type = 'M';
171 
172 		d->muid = "";
173 		d->qid.path = ((uvlong)st->st_dev<<32) | st->st_ino;
174 #ifdef _HAVESTGEN
175 		d->qid.vers = st->st_gen;
176 #endif
177 		if(d->qid.vers == 0)
178 			d->qid.vers = st->st_mtime + st->st_ctime;
179 		d->mode = st->st_mode&0777;
180 		d->atime = st->st_atime;
181 		d->mtime = st->st_mtime;
182 		d->length = st->st_size;
183 
184 		if(S_ISDIR(st->st_mode)){
185 			d->length = 0;
186 			d->mode |= DMDIR;
187 			d->qid.type = QTDIR;
188 		}
189 		if(S_ISLNK(lst->st_mode))	/* yes, lst not st */
190 			d->mode |= DMSYMLINK;
191 		if(S_ISFIFO(st->st_mode))
192 			d->mode |= DMNAMEDPIPE;
193 		if(S_ISSOCK(st->st_mode))
194 			d->mode |= DMSOCKET;
195 		if(S_ISBLK(st->st_mode)){
196 			d->mode |= DMDEVICE;
197 			d->qid.path = ('b'<<16)|st->st_rdev;
198 		}
199 		if(S_ISCHR(st->st_mode)){
200 			d->mode |= DMDEVICE;
201 			d->qid.path = ('c'<<16)|st->st_rdev;
202 		}
203 		/* fetch real size for disks */
204 #ifdef _HAVEDISKSIZE
205 		if(S_ISBLK(st->st_mode) && (fd = open(name, O_RDONLY)) >= 0){
206 			d->length = disksize(fd, major(st->st_dev));
207 			close(fd);
208 		}
209 #endif
210 #ifdef _HAVEDISKLABEL
211 		if(isdisk(st)){
212 			int fd, n;
213 			struct disklabel lab;
214 
215 			if((fd = open(name, O_RDONLY)) < 0)
216 				goto nosize;
217 			if(ioctl(fd, DIOCGDINFO, &lab) < 0)
218 				goto nosize;
219 			n = minor(st->st_rdev)&7;
220 			if(n >= lab.d_npartitions)
221 				goto nosize;
222 
223 			d->length = (vlong)(lab.d_partitions[n].p_size) * lab.d_secsize;
224 
225 		nosize:
226 			if(fd >= 0)
227 				close(fd);
228 		}
229 #endif
230 	}
231 
232 	return sz;
233 }
234