ronnie
2022-10-23 cb8ede114f8c3e5ead5b294f66344b8a42004745
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
 * Copyright (c) 1997,2007 Andrew G. Morgan  <morgan@kernel.org>
 *
 * This displays the capabilities of a given file.
 */
 
#define _XOPEN_SOURCE 500
 
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <dirent.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/capability.h>
 
#include <ftw.h>
 
static int verbose = 0;
static int recursive = 0;
 
static void usage(void)
{
    fprintf(stderr,
       "usage: getcap [-v] [-r] [-h] <filename> [<filename> ...]\n"
       "\n"
       "\tdisplays the capabilities on the queried file(s).\n"
   );
    exit(1);
}
 
static int do_getcap(const char *fname, const struct stat *stbuf,
            int tflag, struct FTW* ftwbuf)
{
    cap_t cap_d;
    char *result;
 
    if (tflag != FTW_F) {
   if (verbose) {
       printf("%s (Not a regular file)\n", fname);
   }
   return 0;
    }
 
    cap_d = cap_get_file(fname);
    if (cap_d == NULL) {
   if (errno != ENODATA) {
       fprintf(stderr, "Failed to get capabilities of file `%s' (%s)\n",
           fname, strerror(errno));
   } else if (verbose) {
       printf("%s\n", fname);
   }
   return 0;
    }
 
    result = cap_to_text(cap_d, NULL);
    if (!result) {
   fprintf(stderr,
       "Failed to get capabilities of human readable format at `%s' (%s)\n",
       fname, strerror(errno));
   cap_free(cap_d);
   return 0;
    }
    printf("%s %s\n", fname, result);
    cap_free(cap_d);
    cap_free(result);
 
    return 0;
}
 
int main(int argc, char **argv)
{
    int i, c;
 
    while ((c = getopt(argc, argv, "rvh")) > 0) {
   switch(c) {
   case 'r':
       recursive = 1;
       break;
   case 'v':
       verbose = 1;
       break;
   default:
       usage();
   }
    }
 
    if (!argv[optind])
   usage();
 
    for (i=optind; argv[i] != NULL; i++) {
   struct stat stbuf;
 
   if (lstat(argv[i], &stbuf) != 0) {
       fprintf(stderr, "%s (%s)\n", argv[i], strerror(errno));
   } else if (recursive) {
       nftw(argv[i], do_getcap, 20, FTW_PHYS);
   } else {
       int tflag = S_ISREG(stbuf.st_mode) ? FTW_F :
       (S_ISLNK(stbuf.st_mode) ? FTW_SL : FTW_NS);
       do_getcap(argv[i], &stbuf, tflag, 0);
   }
    }
 
    return 0;
}