1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49#define FOR_ls
50#include "toys.h"
51
52
53
54
55
56GLOBALS(
57 long l;
58 char *color;
59
60 struct dirtree *files, *singledir;
61 unsigned screen_width;
62 int nl_title;
63 char *escmore;
64)
65
66
67static int crunch_qb(FILE *out, int cols, int wc)
68{
69 unsigned len = 1;
70 char buf[32];
71
72 if (toys.optflags&FLAG_q) *buf = '?';
73 else {
74 if (wc<256) *buf = wc;
75
76 else len = wcrtomb(buf, wc, 0);
77 if (toys.optflags&FLAG_b) {
78 char *to = buf, *from = buf+24;
79 int i, j;
80
81 memcpy(from, to, 8);
82 for (i = 0; i<len; i++) {
83 *to++ = '\\';
84 if (strchr(TT.escmore, from[i])) *to++ = from[i];
85 else if (-1 != (j = stridx("\\\a\b\033\f\n\r\t\v", from[i])))
86 *to++ = "\\abefnrtv"[j];
87 else to += sprintf(to, "%03o", from[i]);
88 }
89 len = to-buf;
90 }
91 }
92
93 if (cols<len) len = cols;
94 if (out) fwrite(buf, len, 1, out);
95
96 return len;
97}
98
99
100static int strwidth(char *s)
101{
102 return crunch_str(&s, INT_MAX, 0, TT.escmore, crunch_qb);
103}
104
105static char endtype(struct stat *st)
106{
107 mode_t mode = st->st_mode;
108 if ((toys.optflags&(FLAG_F|FLAG_p)) && S_ISDIR(mode)) return '/';
109 if (toys.optflags & FLAG_F) {
110 if (S_ISLNK(mode)) return '@';
111 if (S_ISREG(mode) && (mode&0111)) return '*';
112 if (S_ISFIFO(mode)) return '|';
113 if (S_ISSOCK(mode)) return '=';
114 }
115 return 0;
116}
117
118static int numlen(long long ll)
119{
120 return snprintf(0, 0, "%llu", ll);
121}
122
123static int print_with_h(char *s, long long value, int units)
124{
125 if (toys.optflags&FLAG_h) return human_readable(s, value*units, 0);
126 else return sprintf(s, "%lld", value);
127}
128
129
130
131static void entrylen(struct dirtree *dt, unsigned *len)
132{
133 struct stat *st = &(dt->st);
134 unsigned flags = toys.optflags;
135 char tmp[64];
136
137 *len = strwidth(dt->name);
138 if (endtype(st)) ++*len;
139 if (flags & FLAG_m) ++*len;
140
141 len[1] = (flags & FLAG_i) ? numlen(st->st_ino) : 0;
142 if (flags & (FLAG_l|FLAG_o|FLAG_n|FLAG_g)) {
143 unsigned fn = flags & FLAG_n;
144
145 len[2] = numlen(st->st_nlink);
146 len[3] = fn ? numlen(st->st_uid) : strwidth(getusername(st->st_uid));
147 len[4] = fn ? numlen(st->st_gid) : strwidth(getgroupname(st->st_gid));
148 if (S_ISBLK(st->st_mode) || S_ISCHR(st->st_mode)) {
149
150
151 len[5] = numlen(dev_major(st->st_rdev))+5;
152 } else len[5] = print_with_h(tmp, st->st_size, 1);
153 }
154
155 len[6] = (flags & FLAG_s) ? print_with_h(tmp, st->st_blocks, 512) : 0;
156 len[7] = (flags & FLAG_Z) ? strwidth((char *)dt->extra) : 0;
157}
158
159static int compare(void *a, void *b)
160{
161 struct dirtree *dta = *(struct dirtree **)a;
162 struct dirtree *dtb = *(struct dirtree **)b;
163 int ret = 0, reverse = (toys.optflags & FLAG_r) ? -1 : 1;
164
165 if (toys.optflags & FLAG_S) {
166 if (dta->st.st_size > dtb->st.st_size) ret = -1;
167 else if (dta->st.st_size < dtb->st.st_size) ret = 1;
168 }
169 if (toys.optflags & FLAG_t) {
170 if (dta->st.st_mtime > dtb->st.st_mtime) ret = -1;
171 else if (dta->st.st_mtime < dtb->st.st_mtime) ret = 1;
172 }
173 if (!ret) ret = strcmp(dta->name, dtb->name);
174 return ret * reverse;
175}
176
177
178
179static int filter(struct dirtree *new)
180{
181 int flags = toys.optflags;
182
183
184 if (flags == (FLAG_1|FLAG_f)) {
185 xprintf("%s\n", new->name);
186 return 0;
187 }
188
189 if (flags & FLAG_Z) {
190 if (!CFG_TOYBOX_LSM_NONE) {
191
192
193
194
195 int fd = openat(dirtree_parentfd(new), new->name,
196 O_PATH|(O_NOFOLLOW*!(toys.optflags&FLAG_L)));
197
198 if (fd != -1) {
199 if (-1 == lsm_fget_context(fd, (char **)&new->extra) && errno == EBADF)
200 {
201 char hack[32];
202
203
204
205
206
207
208
209 sprintf(hack, "/proc/self/fd/%d", fd);
210 lsm_lget_context(hack, (char **)&new->extra);
211 }
212 close(fd);
213 }
214 }
215 if (CFG_TOYBOX_LSM_NONE || !new->extra) new->extra = (long)xstrdup("?");
216 }
217
218 if (flags & FLAG_u) new->st.st_mtime = new->st.st_atime;
219 if (flags & FLAG_c) new->st.st_mtime = new->st.st_ctime;
220 new->st.st_blocks >>= 1;
221
222 if (flags & (FLAG_a|FLAG_f)) return DIRTREE_SAVE;
223 if (!(flags & FLAG_A) && new->name[0]=='.') return 0;
224
225 return dirtree_notdotdot(new) & DIRTREE_SAVE;
226}
227
228
229
230
231static unsigned long next_column(unsigned long ul, unsigned long dtlen,
232 unsigned columns, unsigned *xpos)
233{
234 unsigned long transition;
235 unsigned height, widecols;
236
237
238 if (!(toys.optflags & FLAG_C)) {
239 *xpos = ul % columns;
240 return ul;
241 }
242
243
244
245
246 height = (dtlen+columns-1)/columns;
247
248
249
250 if (height*columns - dtlen >= height) {
251 *xpos = columns;
252 return 0;
253 }
254
255
256 widecols = dtlen % height;
257 if (!widecols) widecols = height;
258 transition = widecols * columns;
259 if (ul < transition) {
260 *xpos = ul % columns;
261 return (*xpos*height) + (ul/columns);
262 }
263
264 ul -= transition;
265 *xpos = ul % (columns-1);
266
267 return (*xpos*height) + widecols + (ul/(columns-1));
268}
269
270static int color_from_mode(mode_t mode)
271{
272 int color = 0;
273
274 if (S_ISDIR(mode)) color = 256+34;
275 else if (S_ISLNK(mode)) color = 256+36;
276 else if (S_ISBLK(mode) || S_ISCHR(mode)) color = 256+33;
277 else if (S_ISREG(mode) && (mode&0111)) color = 256+32;
278 else if (S_ISFIFO(mode)) color = 33;
279 else if (S_ISSOCK(mode)) color = 256+35;
280
281 return color;
282}
283
284
285
286
287static void listfiles(int dirfd, struct dirtree *indir)
288{
289 struct dirtree *dt, **sort;
290 unsigned long dtlen, ul = 0;
291 unsigned width, flags = toys.optflags, totals[8], len[8], totpad = 0,
292 *colsizes = (unsigned *)toybuf, columns = sizeof(toybuf)/4;
293 char tmp[64];
294
295 if (-1 == dirfd) {
296 perror_msg_raw(indir->name);
297
298 return;
299 }
300
301 memset(totals, 0, sizeof(totals));
302 if (CFG_TOYBOX_DEBUG) memset(len, 0, sizeof(len));
303
304
305 if (!indir->parent) {
306
307
308 dt = indir->child;
309 if (dt && S_ISDIR(dt->st.st_mode) && !dt->next && !(flags&(FLAG_d|FLAG_R)))
310 {
311 listfiles(open(dt->name, 0), TT.singledir = dt);
312
313 return;
314 }
315
316
317 for (;dt; dt = dt->next) filter(dt);
318 if (flags == (FLAG_1|FLAG_f)) return;
319
320
321 } else dirtree_recurse(indir, filter, dup(dirfd),
322 DIRTREE_SYMFOLLOW*!!(flags&FLAG_L));
323
324
325
326
327 for (sort = 0;;sort = xmalloc(dtlen*sizeof(void *))) {
328 for (dtlen = 0, dt = indir->child; dt; dt = dt->next, dtlen++)
329 if (sort) sort[dtlen] = dt;
330 if (sort || !dtlen) break;
331 }
332
333
334 if (indir->parent && (TT.singledir!=indir || (flags&FLAG_R)))
335 {
336 char *path = dirtree_path(indir, 0);
337
338 if (TT.nl_title++) xputc('\n');
339 xprintf("%s:\n", path);
340 free(path);
341 }
342
343
344 if (!(flags & FLAG_f)) {
345 unsigned long long blocks = 0;
346
347 qsort(sort, dtlen, sizeof(void *), (void *)compare);
348 for (ul = 0; ul<dtlen; ul++) {
349 entrylen(sort[ul], len);
350 for (width = 0; width<8; width++)
351 if (len[width]>totals[width]) totals[width] = len[width];
352 blocks += sort[ul]->st.st_blocks;
353 }
354 totpad = totals[1]+!!totals[1]+totals[6]+!!totals[6]+totals[7]+!!totals[7];
355 if ((flags&(FLAG_h|FLAG_l|FLAG_o|FLAG_n|FLAG_g|FLAG_s)) && indir->parent) {
356 print_with_h(tmp, blocks, 512);
357 xprintf("total %s\n", tmp);
358 }
359 }
360
361
362 if (flags & (FLAG_C|FLAG_x)) {
363
364
365
366 if (columns > TT.screen_width/2) columns = TT.screen_width/2;
367 if (columns > dtlen) columns = dtlen;
368
369
370 for (;columns > 1; columns--) {
371 unsigned c, totlen = columns;
372
373 memset(colsizes, 0, columns*sizeof(unsigned));
374 for (ul=0; ul<dtlen; ul++) {
375 entrylen(sort[next_column(ul, dtlen, columns, &c)], len);
376 *len += totpad+1;
377 if (c == columns) break;
378
379 if (*len > colsizes[c]) {
380 totlen += (*len)-colsizes[c];
381 colsizes[c] = *len;
382 if (totlen > TT.screen_width) break;
383 }
384 }
385
386 if (ul == dtlen) break;
387 }
388 }
389
390
391 width = 0;
392 for (ul = 0; ul<dtlen; ul++) {
393 int ii;
394 unsigned curcol, color = 0;
395 unsigned long next = next_column(ul, dtlen, columns, &curcol);
396 struct stat *st = &(sort[next]->st);
397 mode_t mode = st->st_mode;
398 char et = endtype(st), *ss;
399
400
401 if (S_ISDIR(mode) && !indir->parent && !(flags & FLAG_d)) continue;
402 TT.nl_title=1;
403
404
405 entrylen(sort[next], len);
406 if (ul) {
407 int mm = !!(flags & FLAG_m);
408
409 if (mm) xputc(',');
410 if (flags & (FLAG_C|FLAG_x)) {
411 if (!curcol) xputc('\n');
412 } else if ((flags & FLAG_1) || width+1+*len > TT.screen_width) {
413 xputc('\n');
414 width = 0;
415 } else {
416 printf(" "+mm, 0);
417 width += 2-mm;
418 }
419 }
420 width += *len;
421
422 if (flags & FLAG_i) printf("%*lu ", totals[1], (unsigned long)st->st_ino);
423
424 if (flags & FLAG_s) {
425 print_with_h(tmp, st->st_blocks, 512);
426 printf("%*s ", totals[6], tmp);
427 }
428
429 if (flags & (FLAG_l|FLAG_o|FLAG_n|FLAG_g)) {
430 struct tm *tm;
431
432
433 mode_to_string(mode, tmp);
434 printf("%s% *ld", tmp, totals[2]+1, (long)st->st_nlink);
435
436
437 if (!(flags&FLAG_g)) {
438 putchar(' ');
439 ii = -totals[3];
440 if (flags&FLAG_n) printf("%*u", ii, (unsigned)st->st_uid);
441 else draw_trim_esc(getusername(st->st_uid), ii, abs(ii), TT.escmore,
442 crunch_qb);
443 }
444
445
446 if (!(flags&FLAG_o)) {
447 putchar(' ');
448 ii = -totals[4];
449 if (flags&FLAG_n) printf("%*u", ii, (unsigned)st->st_gid);
450 else draw_trim_esc(getgroupname(st->st_gid), ii, abs(ii), TT.escmore,
451 crunch_qb);
452 }
453
454 if (flags & FLAG_Z)
455 printf(" %-*s", -(int)totals[7], (char *)sort[next]->extra);
456
457
458 if (S_ISCHR(st->st_mode) || S_ISBLK(st->st_mode))
459 printf("% *d,% 4d", totals[5]-4, dev_major(st->st_rdev),
460 dev_minor(st->st_rdev));
461 else {
462 print_with_h(tmp, st->st_size, 1);
463 printf("%*s", totals[5]+1, tmp);
464 }
465
466
467 tm = localtime(&(st->st_mtime));
468 strftime(tmp, sizeof(tmp), "%F %H:%M", tm);
469 if (TT.l>1) {
470 char *s = tmp+strlen(tmp);
471
472 s += sprintf(s, ":%02d.%09d ", tm->tm_sec, (int)st->st_mtim.tv_nsec);
473 strftime(s, sizeof(tmp)-(s-tmp), "%z", tm);
474 }
475 printf(" %s ", tmp);
476 } else if (flags & FLAG_Z)
477 printf("%-*s ", (int)totals[7], (char *)sort[next]->extra);
478
479 if (flags & FLAG_color) {
480 color = color_from_mode(st->st_mode);
481 if (color) printf("\033[%d;%dm", color>>8, color&255);
482 }
483
484 ss = sort[next]->name;
485 crunch_str(&ss, INT_MAX, stdout, TT.escmore, crunch_qb);
486 if (color) printf("\033[0m");
487
488 if ((flags & (FLAG_l|FLAG_o|FLAG_n|FLAG_g)) && S_ISLNK(mode)) {
489 printf(" -> ");
490 if (flags & FLAG_color) {
491 struct stat st2;
492
493 if (fstatat(dirfd, sort[next]->symlink, &st2, 0)) color = 256+31;
494 else color = color_from_mode(st2.st_mode);
495
496 if (color) printf("\033[%d;%dm", color>>8, color&255);
497 }
498
499 printf("%s", sort[next]->symlink);
500 if (color) printf("\033[0m");
501 }
502
503 if (et) xputc(et);
504
505
506 if (flags & (FLAG_C|FLAG_x)) {
507 curcol = colsizes[curcol]-(*len)-totpad;
508 if (curcol < 255) printf("%*c", curcol, ' ');
509 }
510 }
511
512 if (width) xputc('\n');
513
514
515
516 for (ul = 0; ul<dtlen; free(sort[ul++])) {
517 if ((flags & FLAG_d) || !S_ISDIR(sort[ul]->st.st_mode)) continue;
518
519
520 if (!indir->parent || ((flags&FLAG_R) && dirtree_notdotdot(sort[ul])))
521 listfiles(openat(dirfd, sort[ul]->name, 0), sort[ul]);
522 free((void *)sort[ul]->extra);
523 }
524 free(sort);
525 if (dirfd != AT_FDCWD) close(dirfd);
526}
527
528void ls_main(void)
529{
530 char **s, *noargs[] = {".", 0};
531 struct dirtree *dt;
532
533 if (toys.optflags&FLAG_full_time) {
534 toys.optflags |= FLAG_l;
535 TT.l = 2;
536 }
537
538
539 if (isatty(1)) {
540 if (!(toys.optflags&FLAG_show_control_chars)) toys.optflags |= FLAG_b;
541 if (toys.optflags&(FLAG_l|FLAG_o|FLAG_n|FLAG_g)) toys.optflags |= FLAG_1;
542 else if (!(toys.optflags&(FLAG_1|FLAG_x|FLAG_m))) toys.optflags |= FLAG_C;
543 } else {
544 if (!(toys.optflags & FLAG_m)) toys.optflags |= FLAG_1;
545 if (TT.color) toys.optflags ^= FLAG_color;
546 }
547
548 TT.screen_width = 80;
549 terminal_size(&TT.screen_width, NULL);
550 if (TT.screen_width<2) TT.screen_width = 2;
551 if (toys.optflags&FLAG_b) TT.escmore = " \\";
552
553
554
555
556 if (toys.optflags & FLAG_d) toys.optflags &= ~FLAG_R;
557
558
559
560
561 TT.files = dirtree_add_node(0, 0, 0);
562 TT.files->dirfd = AT_FDCWD;
563 for (s = *toys.optargs ? toys.optargs : noargs; *s; s++) {
564 int sym = !(toys.optflags&(FLAG_l|FLAG_d|FLAG_F))
565 || (toys.optflags&(FLAG_L|FLAG_H));
566
567 dt = dirtree_add_node(0, *s, DIRTREE_SYMFOLLOW*sym);
568
569
570 if (dt) dlist_add_nomalloc((void *)&TT.files->child, (void *)dt);
571 else toys.exitval = 1;
572 }
573
574
575 dlist_terminate(TT.files->child);
576 for (dt = TT.files->child; dt; dt = dt->next) dt->parent = TT.files;
577
578
579 listfiles(AT_FDCWD, TT.files);
580
581 if (CFG_TOYBOX_FREE) free(TT.files);
582}
583