1 #include <u.h>
2 #include <libc.h>
3 #include <bio.h>
4 
5 #define dirbuf p9dirbuf	/* avoid conflict on sun */
6 
7 typedef struct NDir NDir;
8 struct NDir
9 {
10 	Dir *d;
11 	char	*prefix;
12 };
13 
14 int	errs = 0;
15 int	dflag;
16 int	lflag;
17 int	mflag;
18 int	nflag;
19 int	pflag;
20 int	qflag;
21 int	Qflag;
22 int	rflag;
23 int	sflag;
24 int	tflag;
25 int	uflag;
26 int	Fflag;
27 int	ndirbuf;
28 int	ndir;
29 NDir*	dirbuf;
30 int	ls(char*, int);
31 int	compar(NDir*, NDir*);
32 char*	asciitime(long);
33 char*	darwx(long);
34 void	rwx(long, char*);
35 void	growto(long);
36 void	dowidths(Dir*);
37 void	format(Dir*, char*);
38 void	output(void);
39 ulong	clk;
40 int	swidth;			/* max width of -s size */
41 int	qwidth;			/* max width of -q version */
42 int	vwidth;			/* max width of dev */
43 int	uwidth;			/* max width of userid */
44 int	mwidth;			/* max width of muid */
45 int	glwidth;		/* max width of groupid and length */
46 Biobuf	bin;
47 
48 void
49 main(int argc, char *argv[])
50 {
51 	int i;
52 
53 	Binit(&bin, 1, OWRITE);
54 	ARGBEGIN{
55 	case 'F':	Fflag++; break;
56 	case 'd':	dflag++; break;
57 	case 'l':	lflag++; break;
58 	case 'm':	mflag++; break;
59 	case 'n':	nflag++; break;
60 	case 'p':	pflag++; break;
61 	case 'q':	qflag++; break;
62 	case 'Q':	Qflag++; break;
63 	case 'r':	rflag++; break;
64 	case 's':	sflag++; break;
65 	case 't':	tflag++; break;
66 	case 'u':	uflag++; break;
67 	default:	fprint(2, "usage: ls [-dlmnpqrstuFQ] [file ...]\n");
68 			exits("usage");
69 	}ARGEND
70 
71 	doquote = needsrcquote;
72 	quotefmtinstall();
73 	fmtinstall('M', dirmodefmt);
74 
75 	if(lflag)
76 		clk = time(0);
77 	if(argc == 0)
78 		errs = ls(".", 0);
79 	else for(i=0; i<argc; i++)
80 		errs |= ls(argv[i], 1);
81 	output();
82 	exits(errs? "errors" : 0);
83 }
84 
85 int
86 ls(char *s, int multi)
87 {
88 	int fd;
89 	long i, n;
90 	char *p;
91 	Dir *db;
92 
93 	for(;;) {
94 		p = utfrrune(s, '/');
95 		if(p == 0 || p[1] != 0 || p == s)
96 			break;
97 		*p = 0;
98 	}
99 	db = dirstat(s);
100 	if(db == nil){
101     error:
102 		fprint(2, "ls: %s: %r\n", s);
103 		return 1;
104 	}
105 	if(db->qid.type&QTDIR && dflag==0){
106 		free(db);
107 		db = nil;
108 		output();
109 		fd = open(s, OREAD);
110 		if(fd == -1)
111 			goto error;
112 		n = dirreadall(fd, &db);
113 		if(n < 0)
114 			goto error;
115 		growto(ndir+n);
116 		for(i=0; i<n; i++){
117 			dirbuf[ndir+i].d = db+i;
118 			dirbuf[ndir+i].prefix = multi? s : 0;
119 		}
120 		ndir += n;
121 		close(fd);
122 		output();
123 	}else{
124 		growto(ndir+1);
125 		dirbuf[ndir].d = db;
126 		dirbuf[ndir].prefix = 0;
127 		p = utfrrune(s, '/');
128 		if(p){
129 			dirbuf[ndir].prefix = s;
130 			*p = 0;
131 			/* restore original name; don't use result of stat */
132 			dirbuf[ndir].d->name = strdup(p+1);
133 		}
134 		ndir++;
135 	}
136 	return 0;
137 }
138 
139 void
140 output(void)
141 {
142 	int i;
143 	char buf[4096];
144 	char *s;
145 
146 	if(!nflag && dirbuf!=0)
147 		qsort(dirbuf, ndir, sizeof dirbuf[0], (int (*)(const void*, const void*))compar);
148 	for(i=0; i<ndir; i++)
149 		dowidths(dirbuf[i].d);
150 	for(i=0; i<ndir; i++) {
151 		if(!pflag && (s = dirbuf[i].prefix)) {
152 			if(strcmp(s, "/") ==0)	/* / is a special case */
153 				s = "";
154 			sprint(buf, "%s/%s", s, dirbuf[i].d->name);
155 			format(dirbuf[i].d, buf);
156 		} else
157 			format(dirbuf[i].d, dirbuf[i].d->name);
158 	}
159 	ndir = 0;
160 	Bflush(&bin);
161 }
162 
163 void
164 dowidths(Dir *db)
165 {
166 	char buf[256];
167 	int n;
168 
169 	if(sflag) {
170 		n = sprint(buf, "%llud", (db->length+1023)/1024);
171 		if(n > swidth)
172 			swidth = n;
173 	}
174 	if(qflag) {
175 		n = sprint(buf, "%lud", db->qid.vers);
176 		if(n > qwidth)
177 			qwidth = n;
178 	}
179 	if(mflag) {
180 		n = snprint(buf, sizeof buf, "[%s]", db->muid);
181 		if(n > mwidth)
182 			mwidth = n;
183 	}
184 	if(lflag) {
185 		n = sprint(buf, "%ud", db->dev);
186 		if(n > vwidth)
187 			vwidth = n;
188 		n = strlen(db->uid);
189 		if(n > uwidth)
190 			uwidth = n;
191 		n = sprint(buf, "%llud", db->length);
192 		n += strlen(db->gid);
193 		if(n > glwidth)
194 			glwidth = n;
195 	}
196 }
197 
198 char*
199 fileflag(Dir *db)
200 {
201 	if(Fflag == 0)
202 		return "";
203 	if(QTDIR & db->qid.type)
204 		return "/";
205 	if(0111 & db->mode)
206 		return "*";
207 	return "";
208 }
209 
210 void
211 format(Dir *db, char *name)
212 {
213 	int i;
214 
215 	if(sflag)
216 		Bprint(&bin, "%*llud ",
217 			swidth, (db->length+1023)/1024);
218 	if(mflag){
219 		Bprint(&bin, "[%s] ", db->muid);
220 		for(i=2+strlen(db->muid); i<mwidth; i++)
221 			Bprint(&bin, " ");
222 	}
223 	if(qflag)
224 		Bprint(&bin, "(%.16llux %*lud %.2ux) ",
225 			db->qid.path,
226 			qwidth, db->qid.vers,
227 			db->qid.type);
228 	if(lflag)
229 		Bprint(&bin,
230 			"%M %C %*ud %*s %s %*llud %s ",
231 			db->mode, db->type,
232 			vwidth, db->dev,
233 			-uwidth, db->uid,
234 			db->gid,
235 			(int)(glwidth-strlen(db->gid)), db->length,
236 			asciitime(uflag? db->atime : db->mtime));
237 	Bprint(&bin,
238 		Qflag? "%s%s\n" : "%q%s\n",
239 		name, fileflag(db));
240 }
241 
242 void
243 growto(long n)
244 {
245 	if(n <= ndirbuf)
246 		return;
247 	ndirbuf = n;
248 	dirbuf=(NDir *)realloc(dirbuf, ndirbuf*sizeof(NDir));
249 	if(dirbuf == 0){
250 		fprint(2, "ls: malloc fail\n");
251 		exits("malloc fail");
252 	}
253 }
254 
255 int
256 compar(NDir *a, NDir *b)
257 {
258 	long i;
259 	Dir *ad, *bd;
260 
261 	ad = a->d;
262 	bd = b->d;
263 
264 	if(tflag){
265 		if(uflag)
266 			i = bd->atime-ad->atime;
267 		else
268 			i = bd->mtime-ad->mtime;
269 	}else{
270 		if(a->prefix && b->prefix){
271 			i = strcmp(a->prefix, b->prefix);
272 			if(i == 0)
273 				i = strcmp(ad->name, bd->name);
274 		}else if(a->prefix){
275 			i = strcmp(a->prefix, bd->name);
276 			if(i == 0)
277 				i = 1;	/* a is longer than b */
278 		}else if(b->prefix){
279 			i = strcmp(ad->name, b->prefix);
280 			if(i == 0)
281 				i = -1;	/* b is longer than a */
282 		}else
283 			i = strcmp(ad->name, bd->name);
284 	}
285 	if(i == 0)
286 		i = (ad<bd? -1 : 1);
287 	if(rflag)
288 		i = -i;
289 	return i;
290 }
291 
292 char*
293 asciitime(long l)
294 {
295 	static char buf[32];
296 	char *t;
297 
298 	t = ctime(l);
299 	/* 6 months in the past or a day in the future */
300 	if(l<clk-180L*24*60*60 || clk+24L*60*60<l){
301 		memmove(buf, t+4, 7);		/* month and day */
302 		memmove(buf+7, t+23, 5);		/* year */
303 	}else
304 		memmove(buf, t+4, 12);		/* skip day of week */
305 	buf[12] = 0;
306 	return buf;
307 }