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