sbase/du.c

200 lines
3.5 KiB
C
Raw Normal View History

/* See LICENSE file for copyright and license details. */
2015-02-14 15:02:41 -05:00
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "util.h"
static size_t blksize = 512;
static char file[PATH_MAX];
static size_t depth = -1;
static size_t curdepth = 0;
static int aflag = 0;
static int dflag = 0;
static int sflag = 0;
static int kflag = 0;
static int hflag = 0;
static int HLPflag = 'P';
2014-01-23 16:16:05 -05:00
static char *
xrealpath(const char *pathname, char *resolved)
{
char *r;
r = realpath(pathname, resolved);
if (!r)
2014-12-21 16:56:52 -05:00
eprintf("realpath: %s:", pathname);
2014-01-23 16:16:05 -05:00
return r;
}
static void
print(size_t n, char *path)
{
if (hflag)
printf("%s\t%s\n", humansize(n * blksize), path);
else
printf("%lu\t%s\n", n, path);
}
static char *
push(const char *path)
{
char *cwd;
cwd = agetcwd();
if (chdir(path) < 0)
eprintf("chdir: %s:", path);
return cwd;
}
static void
pop(char *path)
{
if (chdir(path) < 0)
eprintf("chdir: %s:", path);
free(path);
}
static size_t
nblks(struct stat *st)
{
return (512 * st->st_blocks + blksize - 1) / blksize;
}
static size_t
du(const char *path, char follow)
{
struct dirent *dent;
struct stat st;
DIR *dp;
size_t n = 0, m, t;
int r;
char *cwd;
if (lstat(path, &st) < 0)
eprintf("stat: %s:", path);
n = nblks(&st);
if (!(S_ISDIR(st.st_mode) ||
(follow != 'P' && S_ISLNK(st.st_mode) &&
stat(path, &st) == 0 && S_ISDIR(st.st_mode))))
2013-10-17 13:03:01 -04:00
goto done;
follow = follow == 'H' ? 'P' : follow;
2013-10-17 13:03:01 -04:00
dp = opendir(path);
if (!dp) {
weprintf("opendir %s:", path);
2013-10-17 13:03:01 -04:00
goto done;
}
cwd = push(path);
while ((dent = readdir(dp))) {
if (strcmp(dent->d_name, ".") == 0 ||
strcmp(dent->d_name, "..") == 0)
continue;
if (lstat(dent->d_name, &st) < 0)
eprintf("stat: %s:", dent->d_name);
if (S_ISDIR(st.st_mode) ||
(follow != 'P' && S_ISLNK(st.st_mode) &&
stat(dent->d_name, &st) == 0 && S_ISDIR(st.st_mode))) {
t = curdepth;
curdepth++;
n += du(dent->d_name, follow);
curdepth = t;
2013-10-17 13:03:01 -04:00
continue;
}
m = nblks(&st);
2013-10-17 13:03:01 -04:00
n += m;
if (aflag && !sflag) {
2014-01-23 15:05:01 -05:00
if (S_ISLNK(st.st_mode)) {
r = snprintf(file, sizeof(file), "%s/%s",
cwd, dent->d_name);
if (r >= sizeof(file) || r < 0)
eprintf("path too long\n");
2014-01-23 15:05:01 -05:00
} else {
2014-01-23 16:16:05 -05:00
xrealpath(dent->d_name, file);
2014-01-23 15:05:01 -05:00
}
if (!dflag || (depth != -1 && curdepth < depth))
print(m, file);
}
}
2013-10-17 13:03:01 -04:00
pop(cwd);
closedir(dp);
2013-10-17 13:03:01 -04:00
done:
if (!sflag && (!dflag || (depth != -1 && curdepth <= depth)))
2014-01-23 16:16:05 -05:00
print(n, xrealpath(path, file));
return n;
}
static void
usage(void)
{
eprintf("usage: %s [-a | -s] [-d depth] [-h] [-k] [-H | -L | -P] [file ...]\n", argv0);
}
int
main(int argc, char *argv[])
{
size_t n;
char *bsize;
ARGBEGIN {
case 'a':
aflag = 1;
break;
case 'd':
dflag = 1;
depth = estrtonum(EARGF(usage()), 0, MIN(LLONG_MAX, SIZE_MAX));
break;
case 's':
sflag = 1;
break;
case 'k':
kflag = 1;
break;
case 'h':
hflag = 1;
break;
case 'H':
case 'L':
case 'P':
HLPflag = ARGC();
break;
default:
usage();
} ARGEND;
if ((aflag && sflag) || (dflag && sflag))
usage();
bsize = getenv("BLOCKSIZE");
if (bsize)
blksize = estrtonum(bsize, 0, LONG_MAX);
if (kflag)
blksize = 1024;
if (argc < 1) {
n = du(".", HLPflag);
if (sflag)
print(n, xrealpath(".", file));
} else {
for (; argc > 0; argc--, argv++) {
curdepth = 0;
n = du(argv[0], HLPflag);
if (sflag)
print(n, xrealpath(argv[0], file));
}
}
return 0;
}