ann.py: Various improvements.
[utils.git] / pdu.c
1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <string.h>
4 #include <unistd.h>
5 #include <errno.h>
6 #include <dirent.h>
7 #include <stdarg.h>
8 #include <stdint.h>
9 #include <sys/stat.h>
10 #include <dpar.h>
11 #include <stdatomic.h>
12
13 struct subdir {
14     struct subdir *parent;
15     char *path;
16     off_t sum;
17     int rv, subjobs, summary;
18 };
19
20 static int summary = 0;
21 static int retval = 0;
22
23 static char *sprintf2(char *fmt, ...)
24 {
25     char *ret;
26     size_t sz;
27     FILE *fp;
28     va_list args;
29     
30     fp = open_memstream(&ret, &sz);
31     va_start(args, fmt);
32     vfprintf(fp, fmt, args);
33     va_end(args);
34     fclose(fp);
35     return(ret);
36 }
37
38 static void printres(struct subdir *job)
39 {
40     printf("%ji\t%s\n", (intmax_t)(job->sum / 1024), job->path);
41 }
42
43 static void finish(struct subdir *job)
44 {
45     if(atomic_fetch_sub(&job->subjobs, 1) == 1) {
46         if(job->summary)
47             printres(job);
48         if(job->parent) {
49             atomic_fetch_add(&job->parent->sum, job->sum);
50             finish(job->parent);
51         }
52         free(job->path);
53         free(job);
54     }
55 }
56
57 static void dodir(void *jobp)
58 {
59     struct subdir *job = jobp;
60     DIR *dp;
61     struct dirent *dent, dbuf;
62     struct stat sb;
63     struct subdir *nj;
64     char *ep;
65     
66     if((dp = opendir(job->path)) == NULL) {
67         fprintf(stderr, "pdu: %s: %s\n", job->path, strerror(errno));
68         job->rv = 1;
69         return;
70     }
71     job->subjobs = 1;
72     while(1) {
73         if(readdir_r(dp, &dbuf, &dent)) {
74             fprintf(stderr, "pdu: reading %s: %s\n", job->path, strerror(errno));
75             job->rv = 1;
76             break;
77         }
78         if(!dent)
79             break;
80         if(!strcmp(dent->d_name, ".") || !strcmp(dent->d_name, ".."))
81             continue;
82         ep = sprintf2("%s/%s", job->path, dent->d_name);
83         if(stat(ep, &sb)) {
84             fprintf(stderr, "pdu: stat %s: %s\n", ep, strerror(errno));
85         } else {
86             if(S_ISDIR(sb.st_mode)) {
87                 *(nj = malloc(sizeof(*nj))) = (struct subdir) {
88                     .path = strdup(ep),
89                     .summary = !summary,
90                     .parent = job,
91                 };
92                 atomic_fetch_add(&job->subjobs, 1);
93                 atomic_fetch_add(&job->sum, (off_t)sb.st_blocks * 512);
94                 submitiowork(dodir, nj);
95             } else if(S_ISREG(sb.st_mode)) {
96                 atomic_fetch_add(&job->sum, (off_t)sb.st_blocks * 512);
97             }
98         }
99         free(ep);
100     }
101     closedir(dp);
102     finish(job);
103 }
104
105 static void dodu(char *path)
106 {
107     struct subdir *job;
108     
109     *(job = malloc(sizeof(*job))) = (struct subdir) {
110         .path = strdup(path),
111         .summary = 1,
112     };
113     submitiowork(dodir, job);
114 }
115
116 static void usage(FILE *out)
117 {
118     fprintf(out, "usage: pdu [-sh] [DIRECTORY...]\n");
119 }
120
121 int main(int argc, char **argv)
122 {
123     int i, c;
124     
125     while((c = getopt(argc, argv, "hs")) != -1) {
126         switch(c) {
127         case 's':
128             summary = 1;
129             break;
130         case 'h':
131             usage(stdout);
132             exit(0);
133         default:
134             usage(stderr);
135             exit(1);
136         }
137     }
138     if(optind < argc) {
139         for(i = optind; i < argc; i++)
140             dodu(argv[i]);
141     } else {
142         dodu(".");
143     }
144     joinwork();
145     return(retval);
146 }
147
148 /*
149  * Local Variables:
150  * compile-command: "gcc -Wall -g -o pdu pdu.c -ldpar"
151  * End:
152  */