move Rust sources to top level, C sources into c_src

[bcachefs-tools-debian] / c_src / cmd_attr.c

diff --git a/c_src/cmd_attr.c b/c_src/cmd_attr.c
new file mode 100644 (file)
index 0000000..bde9d8f
--- /dev/null
+++ b/c_src/cmd_attr.c
@@ -0,0 +1,119 @@
+#include <dirent.h>
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/xattr.h>
+#include <unistd.h>
+
+#include "libbcachefs/bcachefs_ioctl.h"
+
+#include "cmds.h"
+#include "libbcachefs.h"
+
+static void propagate_recurse(int dirfd)
+{
+       DIR *dir = fdopendir(dirfd);
+       struct dirent *d;
+
+       if (!dir) {
+               fprintf(stderr, "fdopendir() error: %m\n");
+               return;
+       }
+
+       while ((errno = 0), (d = readdir(dir))) {
+               if (!strcmp(d->d_name, ".") ||
+                   !strcmp(d->d_name, ".."))
+                       continue;
+
+               int ret = ioctl(dirfd, BCHFS_IOC_REINHERIT_ATTRS,
+                           d->d_name);
+               if (ret < 0) {
+                       fprintf(stderr, "error propagating attributes to %s: %m\n",
+                               d->d_name);
+                       continue;
+               }
+
+               if (!ret) /* did no work */
+                       continue;
+
+               struct stat st = xfstatat(dirfd, d->d_name,
+                                         AT_SYMLINK_NOFOLLOW);
+               if (!S_ISDIR(st.st_mode))
+                       continue;
+
+               int fd = openat(dirfd, d->d_name, O_RDONLY);
+               if (fd < 0) {
+                       fprintf(stderr, "error opening %s: %m\n", d->d_name);
+                       continue;
+               }
+               propagate_recurse(fd);
+               close(fd);
+       }
+
+       if (errno)
+               die("readdir error: %m");
+}
+
+static void do_setattr(char *path, struct bch_opt_strs opts)
+{
+       unsigned i;
+
+       for (i = 0; i < bch2_opts_nr; i++) {
+               if (!opts.by_id[i])
+                       continue;
+
+               char *n = mprintf("bcachefs.%s", bch2_opt_table[i].attr.name);
+
+               if (setxattr(path, n, opts.by_id[i], strlen(opts.by_id[i]), 0))
+                       die("setxattr error: %m");
+
+               free(n);
+       }
+
+       struct stat st = xstat(path);
+       if (!S_ISDIR(st.st_mode))
+               return;
+
+       int dirfd = open(path, O_RDONLY);
+       if (dirfd < 0)
+               die("error opening %s: %m", path);
+
+       propagate_recurse(dirfd);
+       close(dirfd);
+}
+
+static void setattr_usage(void)
+{
+       puts("bcachefs setattr - set attributes on files in a bcachefs filesystem\n"
+            "Usage: bcachefs setattr [OPTIONS]... <files>\n"
+            "\n"
+            "Options:");
+
+       bch2_opts_usage(OPT_INODE);
+       puts("  -h            Display this help and exit\n"
+            "Report bugs to <linux-bcachefs@vger.kernel.org>");
+}
+
+int cmd_setattr(int argc, char *argv[])
+{
+       struct bch_opt_strs opts =
+               bch2_cmdline_opts_get(&argc, argv, OPT_INODE);
+       unsigned i;
+
+       for (i = 1; i < argc; i++)
+               if (argv[i][0] == '-') {
+                       printf("invalid option %s\n", argv[i]);
+                       setattr_usage();
+                       exit(EXIT_FAILURE);
+               }
+
+       if (argc <= 1)
+               die("Please supply one or more files");
+
+       for (i = 1; i < argc; i++)
+               do_setattr(argv[i], opts);
+       bch2_opt_strs_free(&opts);
+
+       return 0;
+}